Upload patch_models.py

patch_models.py

@@ -18,7 +18,7 @@ AVAILABLE_MODELS.extend([
     {"id": "openai/nvidia/nemotron-3-super-120b-a12b:free", "label": "Nemotron 3 Super 120B", "provider": "openrouter", "tier": "free"},
     {"id": "google/gemma-3-1b-it", "label": "Gemma 3 1B", "provider": "huggingface", "tier": "free", "recommended": True},
     {"id": "Qwen/Qwen3-Coder-Next", "label": "Qwen3 Coder Next", "provider": "huggingface", "tier": "free", "recommended": True},
-    {"id": "openai/owl-alpha:free", "label": "Owl Alpha", "provider": "openrouter", "tier": "free", "recommended": True},
+    {"id": "openai/openrouter/owl-alpha:free", "label": "Owl Alpha", "provider": "openrouter", "tier": "free", "recommended": True},
     {"id": "openai/google/gemini-2.0-flash-001", "label": "Gemini 2.0 Flash", "provider": "openrouter", "tier": "free"},
     {"id": "openai/google/gemma-4-31b-it:free", "label": "Gemma 4 31B", "provider": "openrouter", "tier": "free", "recommended": True},
     {"id": "openai/nvidia/nemotron-3-ultra-550b-a55b:free", "label": "Nemotron 3 Ultra 550B", "provider": "openrouter", "tier": "free", "recommended": True},
@@ -63,10 +63,9 @@ new_block = """    hf_model = normalized_model
     # === PATCH: Route OpenRouter-only models to openrouter.ai ===
     # Models with openai/<org>/<model> (3+ segments) are OpenRouter-only.
     # openai/gpt-5.5:fal-ai (2 segments) stays on HF Router.
-    # OpenRouter free tier uses :free suffix in the model ID.
-    # OpenRouter requires OPENROUTER_API_KEY or a user HF token.
-    _or_prefixes = ("openai/google/", "openai/deepseek/", "openai/nvidia/", "openai/poolside/")
-    if normalized_model.startswith(_or_prefixes) or normalized_model.startswith("openai/owl-alpha"):
+    # OpenRouter free tier uses :free suffix. Requires OPENROUTER_API_KEY.
+    _or_prefixes = ("openai/google/", "openai/deepseek/", "openai/nvidia/", "openai/poolside/", "openai/openrouter/")
+    if normalized_model.startswith(_or_prefixes):
         return {
             "model": normalized_model,
             "api_base": "https://openrouter.ai/api/v1",
@@ -82,7 +81,7 @@ new_block = """    hf_model = normalized_model
 
 if old_block in llm_content:
     llm_content = llm_content.replace(old_block, new_block)
-    print("✅ Patched _resolve_llm_params: OpenRouter routing with :free models")
+    print("✅ Patched _resolve_llm_params: +openai/openrouter/ prefix for Owl Alpha")
 else:
     print("⚠ Could not find target block in llm_params.py")
 
@@ -97,5 +96,5 @@ with open(LLM_PARAMS_FILE, "w") as f:
     f.write(llm_content)
 
 print("✅ Backend patched successfully")
-print("  - AVAILABLE_MODELS: :free suffix added to all OpenRouter free models")
-print("  - _resolve_llm_params: OpenRouter routing with :free suffix preserved")
+print("  - AVAILABLE_MODELS match frontend modelPaths")
+print("  - OpenRouter routing includes openai/openrouter/ prefix")