Skip to content

Xiaomi Mimo Model Support#730

Open
m-prabhu wants to merge 1 commit intoTauricResearch:mainfrom
m-prabhu:feature-xiaomimimo-model-support
Open

Xiaomi Mimo Model Support#730
m-prabhu wants to merge 1 commit intoTauricResearch:mainfrom
m-prabhu:feature-xiaomimimo-model-support

Conversation

@m-prabhu
Copy link
Copy Markdown

@m-prabhu m-prabhu commented May 5, 2026

Added inbuilt support for Xiaomi Mimo Models

Copy link
Copy Markdown

@gemini-code-assist gemini-code-assist Bot left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Code Review

This pull request integrates the MiMo LLM provider into the system. The changes include adding the MIMO_API_KEY to environment configurations, updating the LLM provider list and factory, and defining a model catalog for MiMo. Feedback suggests removing the 'xiaomi/' prefix from model identifiers to ensure compatibility with native API endpoints and ensuring consistency in model validation logic by excluding MiMo from the VALID_MODELS definition.

Comment on lines +104 to +117
"mimo": {
"quick": [
("MiMo-V2.5 - Native omnimodal, 1M context, cost-efficient", "xiaomi/mimo-v2.5"),
("MiMo-V2-Flash - Open-source 309B MoE, fast reasoning", "xiaomi/mimo-v2-flash"),
("MiMo-V2-Omni - Multimodal (image/video/audio)", "xiaomi/mimo-v2-omni"),
("Custom model ID", "custom"),
],
"deep": [
("MiMo-V2.5-Pro - Flagship, 1T params, 1M context, best agent perf", "xiaomi/mimo-v2.5-pro"),
("MiMo-V2.5 - Native omnimodal, 1M context, cost-efficient", "xiaomi/mimo-v2.5"),
("MiMo-V2-Pro - Previous flagship, 1T params, 1M context", "xiaomi/mimo-v2-pro"),
("Custom model ID", "custom"),
],
},
Copy link
Copy Markdown

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

high

The model identifiers for MiMo (e.g., xiaomi/mimo-v2.5) include a xiaomi/ prefix. While this format is common on aggregator platforms like OpenRouter, native provider endpoints (like the one configured in openai_client.py: https://token-plan-sgp.xiaomimimo.com/v1) typically expect the model name without the provider prefix. Based on the pattern used for other native providers in this catalog (such as Qwen and GLM), these should likely be updated to remove the prefix to ensure compatibility with the native API.

Suggested change
"mimo": {
"quick": [
("MiMo-V2.5 - Native omnimodal, 1M context, cost-efficient", "xiaomi/mimo-v2.5"),
("MiMo-V2-Flash - Open-source 309B MoE, fast reasoning", "xiaomi/mimo-v2-flash"),
("MiMo-V2-Omni - Multimodal (image/video/audio)", "xiaomi/mimo-v2-omni"),
("Custom model ID", "custom"),
],
"deep": [
("MiMo-V2.5-Pro - Flagship, 1T params, 1M context, best agent perf", "xiaomi/mimo-v2.5-pro"),
("MiMo-V2.5 - Native omnimodal, 1M context, cost-efficient", "xiaomi/mimo-v2.5"),
("MiMo-V2-Pro - Previous flagship, 1T params, 1M context", "xiaomi/mimo-v2-pro"),
("Custom model ID", "custom"),
],
},
"mimo": {
"quick": [
("MiMo-V2.5 - Native omnimodal, 1M context, cost-efficient", "mimo-v2.5"),
("MiMo-V2-Flash - Open-source 309B MoE, fast reasoning", "mimo-v2-flash"),
("MiMo-V2-Omni - Multimodal (image/video/audio)", "mimo-v2-omni"),
("Custom model ID", "custom"),
],
"deep": [
("MiMo-V2.5-Pro - Flagship, 1T params, 1M context, best agent perf", "mimo-v2.5-pro"),
("MiMo-V2.5 - Native omnimodal, 1M context, cost-efficient", "mimo-v2.5"),
("MiMo-V2-Pro - Previous flagship, 1T params, 1M context", "mimo-v2-pro"),
("Custom model ID", "custom"),
],
},

provider_lower = provider.lower()

if provider_lower in ("ollama", "openrouter"):
if provider_lower in ("ollama", "openrouter", "mimo"):
Copy link
Copy Markdown

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

medium

Adding mimo to the bypass list here allows any model ID to pass validation. However, mimo was not added to the exclusion list in the VALID_MODELS dictionary comprehension (lines 6-10). This creates an inconsistency where VALID_MODELS contains data for mimo that is never used because of this early return. For consistency with ollama and openrouter, mimo should also be excluded from the VALID_MODELS definition.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant