Upload patch_frontend.py

patch_frontend.py

@@ -1,4 +1,4 @@
-"""Patch frontend: Remove Owl Alpha, add models with openai/ prefix for OpenRouter routing."""
+"""Patch frontend: Replace DEFAULT_MODEL_OPTIONS with OpenRouter models."""
 
 import re
 import sys
@@ -6,7 +6,7 @@ import sys
 FILE = "/source/frontend/src/components/Chat/ChatInput.tsx"
 MODEL_FILE = "/source/frontend/src/utils/model.ts"
 
-# === Step 1: Patch model.ts constants ===
+# === Step 1: Patch model.ts ===
 with open(MODEL_FILE, "r") as f:
     model_content = f.read()
 
@@ -35,7 +35,7 @@ print("OK: model.ts constants updated")
 with open(FILE, "r", encoding="utf-8") as f:
     content = f.read()
 
-# Fix imports — replace model constants imports with just isClaudePath
+# Fix imports
 import_pattern = re.compile(
     r'import\s*\{[^}]+\}\s*from\s*[\'"]@/utils/model[\'"];'
 )
@@ -43,22 +43,20 @@ match = import_pattern.search(content)
 if match:
     content = content.replace(match.group(0), "import { isClaudePath } from '@/utils/model';")
     print("OK: Replaced imports")
-else:
-    print("OK: Imports not found")
 
 # Remove Owl Alpha references
 content = content.replace("'Owl Alpha'", "'Claude Opus 4.8'")
 content = content.replace('"Owl Alpha"', '"Claude Opus 4.8"')
 
-# Row 3: Kimi K2.7 Code → DeepSeek V4 Pro
+# Row 3: Kimi → DeepSeek V4 Pro
 content = content.replace("'Kimi K2.7 Code'", "'DeepSeek V4 Pro'")
 content = content.replace('"Kimi K2.7 Code"', '"DeepSeek V4 Pro"')
 
-# Row 4: MiniMax M3 → DeepSeek V4 Flash
+# Row 4: MiniMax → DeepSeek V4 Flash
 content = content.replace("'MiniMax M3'", "'DeepSeek V4 Flash'")
 content = content.replace('"MiniMax M3"', '"DeepSeek V4 Flash"')
 
-# Row 5: GLM 5.2 → DeepSeek V4 Flash
+# Row 5: GLM → DeepSeek V4 Flash
 content = content.replace("'GLM 5.2'", "'DeepSeek V4 Flash'")
 content = content.replace('"GLM 5.2"', '"DeepSeek V4 Flash"')
 
@@ -68,24 +66,7 @@ content = content.replace(
     "  {\n    id: 'nemotron-120b',\n    name: 'nvidia/nemotron-3-super-120b-a12b:free',\n    modelPath: DEEPSEEK_V4_PRO_MODEL_PATH,\n    avatarUrl: 'https://huggingface.co/api/avatars/nvidia',\n  },"
 )
 
-# Remove ALL old custom model blocks
-old_blocks = [
-    'modelPath: \'google/gemma-4-31b-it:free\'',
-    'modelPath: \'tencent/hy3:free\'',
-    'modelPath: \'tencent/hy3\'',
-    'modelPath: \'meta-llama/llama-3.1-8b-instruct\'',
-    'modelPath: \'meta-llama/llama-3.3-70b-instruct:free\'',
-    'modelPath: \'google/gemma-2-9b-it\'',
-    'modelPath: \'nvidia/llama-3.1-nemotron-70b-instruct\'',
-    'modelPath: \'openai/tencent/hy3:free\'',
-    'id: \'gemma-4-31b\'',
-    'id: \'llama-3.1-8b\'',
-    'id: \'llama-3.3-70b\'',
-    'id: \'nemotron-70b\'',
-    'id: \'gemma-2-9b\'',
-    'id: \'tencent-hy3\'',
-]
-# Use simpler approach: find and replace entire DEFAULT_MODEL_OPTIONS array
+# Replace entire DEFAULT_MODEL_OPTIONS array
 pattern = r'const DEFAULT_MODEL_OPTIONS: ModelOption\[\] = \[.*?\];'
 match = re.search(pattern, content, re.DOTALL)
 
@@ -94,6 +75,13 @@ if not match:
     sys.exit(1)
 
 new_array = """const DEFAULT_MODEL_OPTIONS: ModelOption[] = [
+  {
+    id: 'tencent-hy3',
+    name: 'Tencent HY3:free',
+    modelPath: 'openai/tencent/hy3:free',
+    avatarUrl: 'https://huggingface.co/api/avatars/tencent',
+    recommended: true,
+  },
   {
     id: 'gemma-4-31b',
     name: 'Gemma 4 31B:free',
@@ -108,18 +96,19 @@ new_array = """const DEFAULT_MODEL_OPTIONS: ModelOption[] = [
     avatarUrl: 'https://huggingface.co/api/avatars/meta-llama',
     recommended: true,
   },
+  {
+    id: 'laguna-m1',
+    name: 'Laguna M.1:free',
+    modelPath: 'openai/poolside/laguna-m.1:free',
+    avatarUrl: 'https://huggingface.co/api/avatars/poolside',
+    recommended: true,
+  },
   {
     id: 'llama-3.1-8b',
     name: 'Llama 3.1 8B',
     modelPath: 'openai/meta-llama/llama-3.1-8b-instruct',
     avatarUrl: 'https://huggingface.co/api/avatars/meta-llama',
   },
-  {
-    id: 'tencent-hy3',
-    name: 'Tencent HY3:free',
-    modelPath: 'openai/tencent/hy3:free',
-    avatarUrl: 'https://huggingface.co/api/avatars/tencent',
-  },
   {
     id: 'qwen3-coder-next',
     name: 'Qwen3 Coder Next',
@@ -139,7 +128,7 @@ content = content[:match.start()] + new_array + content[match.end():]
 # Update DEFAULT_MODEL_PATH
 dm_path = re.search(r'const DEFAULT_MODEL_PATH\s*=.*?;', content)
 if dm_path:
-    content = content.replace(dm_path.group(0), "const DEFAULT_MODEL_PATH = 'openai/google/gemma-4-31b-it:free';")
+    content = content.replace(dm_path.group(0), "const DEFAULT_MODEL_PATH = 'openai/tencent/hy3:free';")
     print("OK: Replaced DEFAULT_MODEL_PATH")
 else:
     print("OK: DEFAULT_MODEL_PATH not found")
@@ -147,4 +136,5 @@ else:
 with open(FILE, "w", encoding="utf-8") as f:
     f.write(content)
 
-print("OK: Frontend patched - Gemma 4 31B:free DEFAULT (openai/ prefix for OpenRouter routing)")
+print("OK: Frontend patched - Tencent HY3:free DEFAULT")
+print("  Models: HY3:free, Gemma 4 31B:free, Llama 3.3 70B:free, Laguna M.1:free, Llama 3.1 8B")