diff --git a/rag/llm/chat_model.py b/rag/llm/chat_model.py index da9e430174..8d616d7293 100644 --- a/rag/llm/chat_model.py +++ b/rag/llm/chat_model.py @@ -921,6 +921,15 @@ class OpenAI_APIChat(Base): super().__init__(key, model_name, base_url, **kwargs) +class Xiaomi(Base): + _FACTORY_NAME = "Xiaomi" + + def __init__(self, key, model_name, base_url, **kwargs): + if not base_url: + base_url = "https://api.xiaomimimo.com/v1" + super().__init__(key, model_name, base_url, **kwargs) + + class LeptonAIChat(Base): _FACTORY_NAME = "LeptonAI"