Spaces:
Running
Running
update
Browse files
app.py
CHANGED
|
@@ -2290,7 +2290,7 @@ AVAILABLE_MODELS = [
|
|
| 2290 |
},
|
| 2291 |
{
|
| 2292 |
"name": "Kimi K2 Thinking",
|
| 2293 |
-
"id": "
|
| 2294 |
"description": "Moonshot Kimi K2 Thinking model for advanced reasoning and code generation"
|
| 2295 |
}
|
| 2296 |
]
|
|
@@ -2427,7 +2427,7 @@ def get_inference_client(model_id, provider="auto"):
|
|
| 2427 |
api_key=os.getenv("MOONSHOT_API_KEY"),
|
| 2428 |
base_url="https://api.moonshot.ai/v1",
|
| 2429 |
)
|
| 2430 |
-
elif model_id == "
|
| 2431 |
# Use HuggingFace InferenceClient with Novita provider for Kimi K2 Thinking
|
| 2432 |
provider = "novita"
|
| 2433 |
elif model_id == "stealth-model-1":
|
|
@@ -2472,9 +2472,6 @@ def get_real_model_id(model_id: str) -> str:
|
|
| 2472 |
raise ValueError("STEALTH_MODEL_1_ID environment variable is required for Carrot model")
|
| 2473 |
|
| 2474 |
return real_model_id
|
| 2475 |
-
elif model_id == "kimi-k2-thinking":
|
| 2476 |
-
# Return the full model name with provider tag for Kimi K2 Thinking
|
| 2477 |
-
return "moonshotai/Kimi-K2-Thinking:novita"
|
| 2478 |
return model_id
|
| 2479 |
|
| 2480 |
# Type definitions
|
|
|
|
| 2290 |
},
|
| 2291 |
{
|
| 2292 |
"name": "Kimi K2 Thinking",
|
| 2293 |
+
"id": "moonshotai/Kimi-K2-Thinking:novita",
|
| 2294 |
"description": "Moonshot Kimi K2 Thinking model for advanced reasoning and code generation"
|
| 2295 |
}
|
| 2296 |
]
|
|
|
|
| 2427 |
api_key=os.getenv("MOONSHOT_API_KEY"),
|
| 2428 |
base_url="https://api.moonshot.ai/v1",
|
| 2429 |
)
|
| 2430 |
+
elif model_id == "moonshotai/Kimi-K2-Thinking:novita":
|
| 2431 |
# Use HuggingFace InferenceClient with Novita provider for Kimi K2 Thinking
|
| 2432 |
provider = "novita"
|
| 2433 |
elif model_id == "stealth-model-1":
|
|
|
|
| 2472 |
raise ValueError("STEALTH_MODEL_1_ID environment variable is required for Carrot model")
|
| 2473 |
|
| 2474 |
return real_model_id
|
|
|
|
|
|
|
|
|
|
| 2475 |
return model_id
|
| 2476 |
|
| 2477 |
# Type definitions
|