Fix patch: ensure models are added correctly to the list

patch_models.py

@@ -1,118 +1,65 @@
-"""Build-time patch: Add models to dropdown + fix title generation for OpenRouter."""
-import re, os
+"""Build-time patch: Add Gemini + Qwen3 to models dropdown."""
+import os
 
 AGENT_FILE = "/app/backend/routes/agent.py"
 with open(AGENT_FILE, "r") as f:
     content = f.read()
 
-# 1. Add new models to _available_models (INSERT before closing bracket)
-# Find the last model entry and add our models before "]"
-insert_marker = """        {
-            "id": "deepseek-ai/DeepSeek-V4-Pro:deepinfra",
-            "label": "DeepSeek V4 Pro",
-            "provider": "huggingface",
-            "tier": "free",
-        },
-    ]
-    return models"""
+# Find the AVAILABLE_MODELS assignment and add models after it
+# The pattern: AVAILABLE_MODELS = _available_models()
+# We'll append our models directly to the list after it's created
 
-new_models_block = """        {
-            "id": "deepseek-ai/DeepSeek-V4-Pro:deepinfra",
-            "label": "DeepSeek V4 Pro",
-            "provider": "huggingface",
-            "tier": "free",
-        },
-        {
-            "id": "openai/deepseek/deepseek-chat-v3-0324",
-            "label": "DeepSeek V3 (OpenRouter)",
-            "provider": "openrouter",
-            "tier": "free",
-            "recommended": True,
-        },
-        {
-            "id": "openai/google/gemini-2.0-flash-001",
-            "label": "Gemini 2.0 Flash (OpenRouter)",
-            "provider": "openrouter",
-            "tier": "free",
-        },
-        {
-            "id": "openai/qwen/qwen3-30b-a3b",
-            "label": "Qwen3 30B (OpenRouter)",
-            "provider": "openrouter",
-            "tier": "free",
-        },
-    ]
-    return models"""
+patch_code = '''
 
-if insert_marker in content:
-    content = content.replace(insert_marker, new_models_block)
-    print("✅ Added DeepSeek V3, Gemini 2.0 Flash, Qwen3 30B to models list")
-else:
-    print("⚠️ Could not find insert marker - trying alternative")
-    # Alternative: find "]\\n    return models" and insert before it
-    content = content.replace(
-        "    ]\n    return models",
-        '''        {
-            "id": "openai/deepseek/deepseek-chat-v3-0324",
-            "label": "DeepSeek V3 (OpenRouter)",
-            "provider": "openrouter",
-            "tier": "free",
-            "recommended": True,
-        },
-        {
-            "id": "openai/google/gemini-2.0-flash-001",
-            "label": "Gemini 2.0 Flash (OpenRouter)",
-            "provider": "openrouter",
-            "tier": "free",
-        },
-        {
-            "id": "openai/qwen/qwen3-30b-a3b",
-            "label": "Qwen3 30B (OpenRouter)",
-            "provider": "openrouter",
-            "tier": "free",
-        },
-    ]
-    return models''',
-        1  # only first occurrence
-    )
-    print("✅ Added models via alternative method")
+# === PATCHED: Add OpenRouter models ===
+import os as _os
+_extra_models = [
+    {"id": "openai/google/gemini-2.0-flash-001", "label": "Gemini 2.0 Flash", "provider": "openrouter", "tier": "free", "recommended": True},
+    {"id": "openai/qwen/qwen3-30b-a3b", "label": "Qwen3 30B", "provider": "openrouter", "tier": "free"},
+]
+AVAILABLE_MODELS.extend(_extra_models)
+# === END PATCH ===
+'''
 
-# 2. Fix title generation: replace HF Router call with OpenRouter
-old_title = '''model="openai/openai/gpt-oss-120b:cerebras",
-            api_base="https://router.huggingface.co/v1",
-            api_key=api_key,'''
+# Insert after AVAILABLE_MODELS = _available_models()
+target = "AVAILABLE_MODELS = _available_models()\n"
+if target in content:
+    content = content.replace(target, target + patch_code)
+    print("✅ Added Gemini 2.0 Flash + Qwen3 30B after AVAILABLE_MODELS")
+else:
+    # Try without newline
+    target2 = "AVAILABLE_MODELS = _available_models()"
+    if target2 in content:
+        content = content.replace(target2, target2 + "\n" + patch_code)
+        print("✅ Added models (alt method)")
+    else:
+        print("❌ Cannot find AVAILABLE_MODELS assignment!")
+        import sys; sys.exit(1)
 
-new_title = '''model="openai/deepseek/deepseek-chat-v3-0324",
-            api_base=os.environ.get("OPENAI_API_BASE", "https://openrouter.ai/api/v1"),
-            api_key=os.environ.get("OPENAI_API_KEY", api_key),'''
+# Fix title generation to use OpenRouter
+old_title = 'model="openai/openai/gpt-oss-120b:cerebras",\n            api_base="https://router.huggingface.co/v1",\n            api_key=api_key,'
+new_title = 'model="openai/deepseek/deepseek-chat-v3-0324",\n            api_base=_os.environ.get("OPENAI_API_BASE", "https://openrouter.ai/api/v1"),\n            api_key=_os.environ.get("OPENAI_API_KEY", api_key),'
 
 if old_title in content:
     content = content.replace(old_title, new_title)
-    # Remove reasoning_effort (not supported by DeepSeek)
     content = content.replace('            reasoning_effort="low",\n', '')
-    print("✅ Title generation patched to use OpenRouter")
+    print("✅ Title generation fixed for OpenRouter")
 else:
-    print("���️ Title generation pattern not found")
-
-# 3. Add "import os" if missing
-if "\nimport os\n" not in content and "import os," not in content:
-    content = "import os\n" + content
-    print("✅ Added import os")
+    print("⚠️ Title pattern not found (may already be different)")
 
 # Verify syntax
 import ast
 try:
     ast.parse(content)
-    print("✅ Syntax valid")
+    print("✅ Syntax OK")
 except SyntaxError as e:
-    print(f"❌ Syntax error line {e.lineno}: {e.msg}")
+    print(f"❌ Syntax error: {e}")
     import sys; sys.exit(1)
 
 with open(AGENT_FILE, "w") as f:
     f.write(content)
 
-# Verify models in file
-assert "deepseek-chat-v3-0324" in content, "DeepSeek missing!"
-assert "gemini-2.0-flash" in content, "Gemini missing!"
-assert "qwen3-30b" in content, "Qwen missing!"
-print("\n✅ All patches applied successfully")
+# Final check
+assert "gemini-2.0-flash" in content, "Gemini not in file!"
+assert "qwen3-30b" in content, "Qwen3 not in file!"
+print("\n✅ Done - Gemini 2.0 Flash + Qwen3 30B added to dropdown")