Upload patch_models.py

patch_models.py

@@ -49,18 +49,22 @@ with open(AGENT_FILE, "w") as f:
 with open(LLM_PARAMS_FILE) as f:
     llm_content = f.read()
 
-marker = """    if local_model_provider(normalized_model) is not None:
-        return _resolve_local_model_params(normalized_model, reasoning_effort, strict)
-
-    hf_model = normalized_model"""
-
-replacement = """    if local_model_provider(normalized_model) is not None:
-        return _resolve_local_model_params(normalized_model, reasoning_effort, strict)
+# The OpenRouter patch must go AFTER `api_key` is resolved.
+# We replace the HF Router assignment block to add OR routing.
+old_block = """    hf_model = normalized_model
+    api_key = _resolve_hf_router_token(session_hf_token)
+    params = {
+        "model": f"openai/{hf_model}",
+        "api_base": HF_ROUTER_BASE_URL,
+        "api_key": api_key,
+    }"""
+
+new_block = """    hf_model = normalized_model
+    api_key = _resolve_hf_router_token(session_hf_token)
 
     # === 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.
+    # Models with openai/<org>/<model> (3+ segments) are OpenRouter-only.
+    # openai/gpt-5.5:fal-ai (2 segments) stays on HF Router.
     _or_prefixes = ("openai/google/", "openai/deepseek/", "openai/nvidia/", "openai/poolside/")
     if normalized_model.startswith(_or_prefixes) or normalized_model == "openai/owl-alpha":
         return {
@@ -70,13 +74,17 @@ replacement = """    if local_model_provider(normalized_model) is not None:
         }
     # === END PATCH ===
 
-    hf_model = normalized_model"""
+    params = {
+        "model": f"openai/{hf_model}",
+        "api_base": HF_ROUTER_BASE_URL,
+        "api_key": api_key,
+    }"""
 
-if marker in llm_content:
-    llm_content = llm_content.replace(marker, replacement)
-    print("✅ Patched _resolve_llm_params: OpenRouter routing added")
+if old_block in llm_content:
+    llm_content = llm_content.replace(old_block, new_block)
+    print("✅ Patched _resolve_llm_params: OpenRouter routing added (after api_key)")
 else:
-    print("⚠ Could not find marker in llm_params.py")
+    print("⚠ Could not find target block in llm_params.py")
 
 try:
     ast.parse(llm_content)
@@ -90,5 +98,5 @@ with open(LLM_PARAMS_FILE, "w") as f:
 
 print("✅ Backend patched successfully")
 print("  - AVAILABLE_MODELS: 12 new models")
-print("  - _resolve_llm_params: openai/google/|deepseek/|nvidia/|poolside/ + owl-alpha → OpenRouter")
-print("  - openai/gpt-5.5:fal-ai stays on HF Router (correct)")
+print("  - _resolve_llm_params: OpenRouter check after api_key resolves")
+print("  - No more UnboundLocalError on api_key")