Upload patch_models.py

patch_models.py

@@ -57,11 +57,12 @@ marker = """    if local_model_provider(normalized_model) is not None:
 replacement = """    if local_model_provider(normalized_model) is not None:
         return _resolve_local_model_params(normalized_model, reasoning_effort, strict)
 
-    # === PATCH: Route "openai/..." models to OpenRouter ===
-    # These are OpenRouter-specific models (Laguna, Nemotron free, Owl Alpha, etc.)
-    # that aren't served by the HF Router. LiteLLM openai/ prefix + custom api_base
-    # makes it send the bare model name to OpenRouter.
-    if normalized_model.startswith("openai/"):
+    # === PATCH: Route OpenRouter-only models to openrouter.ai ===
+    # Models with openai/<org>/<model> (3+ segments) are OpenRouter-only
+    # and must NOT go through HF Router, which doesn't serve them.
+    # openai/gpt-5.5:fal-ai (1 segment) stays on HF Router as before.
+    _or_prefixes = ("openai/google/", "openai/deepseek/", "openai/nvidia/", "openai/poolside/")
+    if normalized_model.startswith(_or_prefixes) or normalized_model == "openai/owl-alpha":
         return {
             "model": normalized_model,
             "api_base": "https://openrouter.ai/api/v1",
@@ -89,6 +90,5 @@ with open(LLM_PARAMS_FILE, "w") as f:
 
 print("✅ Backend patched successfully")
 print("  - AVAILABLE_MODELS: 12 new models")
-print("  - Owl Alpha id: openai/owl-alpha (matches frontend modelPath)")
-print("  - _resolve_llm_params: openai/ prefix → OpenRouter routing")
-print("  - DEFAULT_FREE_MODEL_ID: deepseek-ai/DeepSeek-V4-Pro")
+print("  - _resolve_llm_params: openai/google/|deepseek/|nvidia/|poolside/ + owl-alpha → OpenRouter")
+print("  - openai/gpt-5.5:fal-ai stays on HF Router (correct)")