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/openrouter/owl-alpha", "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},
@@ -62,7 +62,7 @@ new_block = """    hf_model = normalized_model
 
     # === PATCH: Route OpenRouter models to openrouter.ai ===
     # OpenRouter expects the model name WITHOUT the openai/ prefix.
-    # E.g. "openai/openrouter/owl-alpha" → send "openrouter/owl-alpha"
+    # E.g. "openai/openrouter/owl-alpha:free" → send "openrouter/owl-alpha:free"
     _or_prefixes = ("openai/google/", "openai/deepseek/", "openai/nvidia/", "openai/poolside/", "openai/openrouter/")
     if normalized_model.startswith(_or_prefixes):
         # Strip the "openai/" prefix so OpenRouter gets the correct model ID
@@ -97,5 +97,4 @@ with open(LLM_PARAMS_FILE, "w") as f:
     f.write(llm_content)
 
 print("✅ Backend patched")
-print("  - OpenRouter models strip openai/ prefix before sending to OR API")
-print("  - Owl Alpha: sends 'openrouter/owl-alpha' to openrouter.ai")
+print("  - Owl Alpha: sends 'openrouter/owl-alpha:free' to openrouter.ai")