fix: add xiaomi_mimo to LlmProviders enum to fix router support (#18819)

Added XIAOMI_MIMO to the LlmProviders enum in types/utils.py.
The provider was already configured in providers.json but was
missing from the enum, causing "Unsupported provider" errors
when using it in Router/Proxy configurations.

Also added comprehensive unit tests to prevent regression.

litellm/types/utils.py

@@ -3029,6 +3029,7 @@ class LlmProviders(str, Enum):
     NANOGPT = "nano-gpt"
     POE = "poe"
     CHUTES = "chutes"
+    XIAOMI_MIMO = "xiaomi_mimo"
 
 
 

tests/test_litellm/llms/openai_like/test_xiaomi_mimo.py

@@ -0,0 +1,150 @@
+"""
+Tests for Xiaomi MiMo provider configuration and integration.
+Related to issue #18794
+"""
+
+import os
+import sys
+from unittest.mock import MagicMock, patch
+
+try:
+    import pytest
+except ImportError:
+    pytest = None
+
+# Add workspace to path
+workspace_path = os.path.abspath(os.path.join(os.path.dirname(__file__), "../../../.."))
+sys.path.insert(0, workspace_path)
+
+import litellm
+
+
+class TestXiaomiMiMoProviderConfig:
+    """Test Xiaomi MiMo provider configuration"""
+
+    def test_xiaomi_mimo_in_provider_list(self):
+        """Test that xiaomi_mimo is in the provider list (fixes #18794)"""
+        from litellm import LlmProviders
+
+        # Verify xiaomi_mimo is in the enum
+        assert hasattr(LlmProviders, 'XIAOMI_MIMO')
+        assert LlmProviders.XIAOMI_MIMO.value == 'xiaomi_mimo'
+
+        # Verify it's in the provider list
+        assert 'xiaomi_mimo' in litellm.provider_list
+
+    def test_xiaomi_mimo_json_config_exists(self):
+        """Test that xiaomi_mimo is configured in providers.json"""
+        from litellm.llms.openai_like.json_loader import JSONProviderRegistry
+
+        # Verify xiaomi_mimo is loaded
+        assert JSONProviderRegistry.exists("xiaomi_mimo")
+
+        # Get xiaomi_mimo config
+        xiaomi_mimo = JSONProviderRegistry.get("xiaomi_mimo")
+        assert xiaomi_mimo is not None
+        assert xiaomi_mimo.base_url == "https://api.xiaomimimo.com/v1"
+        assert xiaomi_mimo.api_key_env == "XIAOMI_MIMO_API_KEY"
+        assert xiaomi_mimo.param_mappings.get("max_completion_tokens") == "max_tokens"
+
+    def test_xiaomi_mimo_provider_resolution(self):
+        """Test that provider resolution finds xiaomi_mimo"""
+        from litellm.litellm_core_utils.get_llm_provider_logic import get_llm_provider
+
+        model, provider, api_key, api_base = get_llm_provider(
+            model="xiaomi_mimo/mimo-v2-flash",
+            custom_llm_provider=None,
+            api_base=None,
+            api_key=None,
+        )
+
+        assert model == "mimo-v2-flash"
+        assert provider == "xiaomi_mimo"
+        assert api_base == "https://api.xiaomimimo.com/v1"
+
+    def test_xiaomi_mimo_router_config(self):
+        """Test that xiaomi_mimo can be used in Router configuration (fixes #18794)"""
+        from litellm import Router
+
+        # This should not raise "Unsupported provider - xiaomi_mimo"
+        router = Router(
+            model_list=[
+                {
+                    "model_name": "mimo-v2-flash",
+                    "litellm_params": {
+                        "model": "xiaomi_mimo/mimo-v2-flash",
+                        "api_key": "test-key",
+                    },
+                }
+            ]
+        )
+
+        # Verify the deployment was created successfully
+        assert len(router.model_list) == 1
+        assert router.model_list[0]["model_name"] == "mimo-v2-flash"
+
+
+class TestXiaomiMiMoIntegration:
+    """Integration tests for Xiaomi MiMo provider"""
+
+    def test_xiaomi_mimo_completion_basic(self):
+        """Test basic completion call to Xiaomi MiMo"""
+        # Skip test if API key not set in environment
+        if not os.environ.get("XIAOMI_MIMO_API_KEY"):
+            if pytest:
+                pytest.skip("XIAOMI_MIMO_API_KEY not set")
+            return
+
+        try:
+            response = litellm.completion(
+                model="xiaomi_mimo/mimo-v2-flash",
+                messages=[{"role": "user", "content": "Say 'test successful' and nothing else"}],
+                max_tokens=10,
+            )
+
+            # Verify response structure
+            assert response is not None
+            assert hasattr(response, "choices")
+            assert len(response.choices) > 0
+            assert hasattr(response.choices[0], "message")
+            assert hasattr(response.choices[0].message, "content")
+            assert response.choices[0].message.content is not None
+
+            # Check that we got a response
+            content = response.choices[0].message.content.lower()
+            assert len(content) > 0
+
+            print(f"✓ Xiaomi MiMo completion successful: {response.choices[0].message.content}")
+
+        except Exception as e:
+            if pytest:
+                pytest.fail(f"Xiaomi MiMo completion failed: {str(e)}")
+            else:
+                raise
+
+
+if __name__ == "__main__":
+    # Run basic tests
+    print("Testing Xiaomi MiMo Provider...")
+
+    test_config = TestXiaomiMiMoProviderConfig()
+
+    print("\n1. Testing provider in list...")
+    test_config.test_xiaomi_mimo_in_provider_list()
+    print("   ✓ xiaomi_mimo in provider list")
+
+    print("\n2. Testing JSON config...")
+    test_config.test_xiaomi_mimo_json_config_exists()
+    print("   ✓ xiaomi_mimo JSON config loaded")
+
+    print("\n3. Testing provider resolution...")
+    test_config.test_xiaomi_mimo_provider_resolution()
+    print("   ✓ Provider resolution works")
+
+    print("\n4. Testing router configuration...")
+    test_config.test_xiaomi_mimo_router_config()
+    print("   ✓ Router configuration works (issue #18794 fixed)")
+
+    print("\n" + "="*50)
+    print("✓ All configuration tests passed!")
+    print("="*50)