From a5cce29f2271a4a8cde761c0e112c39e0548a374 Mon Sep 17 00:00:00 2001
From: Lynn <lynn_inf@hotmail.com>
Date: Wed, 17 Jun 2026 19:02:33 +0800
Subject: [PATCH] Fix: add mimo (#16136)

### What problem does this PR solve?

Add chat model factory for Xiaomi model.

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
---
 rag/llm/chat_model.py | 9 +++++++++
 1 file changed, 9 insertions(+)

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"