Restore 3d39bbc + change: Kimi→DeepSeek-V4-Pro, MiniMax→deepseek-v4-flash(OR), GLM→DeepSeek-V4-Flash(HF)

patch_models.py

@@ -1,67 +1,72 @@
-"""Patch backend: modify existing AVAILABLE_MODELS entries + fix title generation."""
+"""Patch backend: add new model IDs to AVAILABLE_MODELS + fix title generation."""
 import re, os
 
 AGENT_FILE = "/app/backend/routes/agent.py"
 with open(AGENT_FILE, "r") as f:
     content = f.read()
 
-# Replace existing model entries in _available_models()
+# Add new models to AVAILABLE_MODELS
+target = "AVAILABLE_MODELS = _available_models()\n"
+patch = '''
+# === PATCH: Add models ===
+import os as _os
+AVAILABLE_MODELS.extend([
+    {"id": "openai/openrouter/owl-alpha", "label": "Owl Alpha", "provider": "openrouter", "tier": "free", "recommended": True},
+    {"id": "openai/nvidia/nemotron-3-super-120b-a12b:free", "label": "Nemotron 3 Super 120B", "provider": "openrouter", "tier": "free", "recommended": True},
+    {"id": "openai/deepseek/deepseek-v4-pro", "label": "DeepSeek V4 Pro (OR)", "provider": "openrouter", "tier": "free", "recommended": True},
+    {"id": "Qwen/Qwen3-Coder-Next", "label": "Qwen3 Coder Next", "provider": "huggingface", "tier": "free", "recommended": True},
+    {"id": "openai/xiaomi/mimo-v2.5", "label": "Mimo V2.5", "provider": "openrouter", "tier": "free"},
+    {"id": "google/gemma-3-1b-it", "label": "Gemma 3 1B", "provider": "huggingface", "tier": "free"},
+    {"id": "openai/google/gemini-2.0-flash-001", "label": "Gemini 2.0 Flash", "provider": "openrouter", "tier": "free"},
+])
+PREMIUM_MODEL_IDS = set()  # No premium gate
+DEFAULT_FREE_MODEL_ID = "openai/openrouter/owl-alpha"
+# === END PATCH ===
+'''
+
+if target in content:
+    content = content.replace(target, target + patch)
+    print("✅ Added models to AVAILABLE_MODELS")
+
+# Change existing models in _available_models():
 # 1. Kimi K2.6 → DeepSeek V4 Pro (HF Inference)
-content = content.replace(
-    '"moonshotai/Kimi-K2.6"',
-    '"deepseek-ai/DeepSeek-V4-Pro"'
-)
-content = content.replace(
-    '"Kimi K2.6"',
-    '"DeepSeek V4 Pro"'
-)
+content = content.replace('"id": "moonshotai/Kimi-K2.6"', '"id": "deepseek-ai/DeepSeek-V4-Pro"')
+content = content.replace('"label": "Kimi K2.6"', '"label": "DeepSeek V4 Pro"')
+print("✅ Changed Kimi K2.6 → deepseek-ai/DeepSeek-V4-Pro")
 
-# 4. MiniMax M2.7 (novita) → deepseek/deepseek-v4-flash (openai/ prefix = OpenRouter)
-content = content.replace(
-    '"MiniMaxAI/MiniMax-M2.7"',
-    '"openai/deepseek/deepseek-v4-flash"'
-)
-content = content.replace(
-    '"MiniMax M2.7"',
-    '"DeepSeek V4 Flash (OR)"'
-)
+# 4. MiniMax M2.7 → deepseek/deepseek-v4-flash (OpenRouter)
+content = content.replace('"id": "MiniMaxAI/MiniMax-M2.7"', '"id": "openai/deepseek/deepseek-v4-flash"')
+content = content.replace('"label": "MiniMax M2.7"', '"label": "DeepSeek V4 Flash (OR)"')
+print("✅ Changed MiniMax M2.7 → openai/deepseek/deepseek-v4-flash")
 
-# 5. GLM 5.1 (together) → DeepSeek-V4-Flash (HF Inference)
-content = content.replace(
-    '"zai-org/GLM-5.1"',
-    '"deepseek-ai/DeepSeek-V4-Flash"'
-)
-content = content.replace(
-    '"GLM 5.1"',
-    '"DeepSeek V4 Flash (HF)"'
-)
+# 5. GLM 5.1 → DeepSeek-V4-Flash (HF Inference)
+content = content.replace('"id": "zai-org/GLM-5.1"', '"id": "deepseek-ai/DeepSeek-V4-Flash"')
+content = content.replace('"label": "GLM 5.1"', '"label": "DeepSeek V4 Flash"')
+print("✅ Changed GLM 5.1 → deepseek-ai/DeepSeek-V4-Flash")
 
-# Also fix DEFAULT_FREE_MODEL_ID since Kimi no longer exists
-content = content.replace(
-    'DEFAULT_FREE_MODEL_ID = "moonshotai/Kimi-K2.6"',
-    'DEFAULT_FREE_MODEL_ID = "deepseek-ai/DeepSeek-V4-Pro"'
-)
+# Rename Claude → Owl Alpha
+content = content.replace('"label": "Claude Opus 4.6"', '"label": "Owl Alpha (OpenRouter)"')
+content = content.replace('return session_manager.config.model_name', 'return "openai/openrouter/owl-alpha"')
+print("✅ Renamed Claude → Owl Alpha")
 
 # Fix title generation
 old_patterns = [
     'model="openai/openai/gpt-oss-120b:cerebras",\n            api_base="https://router.huggingface.co/v1",\n            api_key=api_key,',
 ]
-new_title_call = 'model="openrouter/openrouter/owl-alpha",\n            api_key=_os.environ.get("OPENROUTER_API_KEY", api_key),'
+
+new_title_call = 'model="openai/openrouter/owl-alpha",\n            api_base="https://openrouter.ai/api/v1",\n            api_key=_os.environ.get("OPENROUTER_API_KEY", api_key),'
 
 for old in old_patterns:
     if old in content:
         content = content.replace(old, new_title_call)
-        # Need to add import os as _os
-        if '_os.environ' in content and 'import os as _os' not in content:
-            content = content.replace(
-                'AVAILABLE_MODELS = _available_models()\n',
-                'AVAILABLE_MODELS = _available_models()\nimport os as _os\n'
-            )
         print("✅ Title generation patched")
         break
 
 content = content.replace('            reasoning_effort="low",\n', '')
 
+# Also fix DEFAULT_FREE_MODEL_ID in original code
+content = content.replace('DEFAULT_FREE_MODEL_ID = "moonshotai/Kimi-K2.6"', 'DEFAULT_FREE_MODEL_ID = "deepseek-ai/DeepSeek-V4-Pro"')
+
 import ast
 try:
     ast.parse(content)
@@ -73,6 +78,3 @@ except SyntaxError as e:
 with open(AGENT_FILE, "w") as f:
     f.write(content)
 print("✅ Backend patched")
-print("  - Kimi K2.6 → deepseek-ai/DeepSeek-V4-Pro (HF)")
-print("  - MiniMax M2.7 → openai/deepseek/deepseek-v4-flash (OR)")
-print("  - GLM 5.1 → deepseek-ai/DeepSeek-V4-Flash (HF)")