Upload patch_frontend.py

patch_frontend.py

@@ -1,7 +1,12 @@
-"""Patch frontend: Remove Owl Alpha, add working free model Llama 3.3 70B:free."""
+"""Patch frontend: Remove Owl Alpha, add models with openai/ prefix for OpenRouter routing."""
 
-# === Step 1: Patch model.ts constants ===
+import re
+import sys
+
+FILE = "/source/frontend/src/components/Chat/ChatInput.tsx"
 MODEL_FILE = "/source/frontend/src/utils/model.ts"
+
+# === Step 1: Patch model.ts constants ===
 with open(MODEL_FILE, "r") as f:
     model_content = f.read()
 
@@ -24,13 +29,23 @@ model_content = model_content.replace(
 
 with open(MODEL_FILE, "w") as f:
     f.write(model_content)
-print("✅ model.ts constants updated")
+print("OK: model.ts constants updated")
 
 # === Step 2: Patch ChatInput.tsx ===
-FILE = "/source/frontend/src/components/Chat/ChatInput.tsx"
-with open(FILE, "r") as f:
+with open(FILE, "r", encoding="utf-8") as f:
     content = f.read()
 
+# Fix imports — replace model constants imports with just isClaudePath
+import_pattern = re.compile(
+    r'import\s*\{[^}]+\}\s*from\s*[\'"]@/utils/model[\'"];'
+)
+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"')
@@ -53,74 +68,57 @@ 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 broken model blocks
+# Remove ALL old custom model blocks
 old_blocks = [
-    """  {
-    id: 'tencent-hy3',
-    name: 'Tencent HY3',
-    modelPath: 'openai/tencent/hy3:free',
-    avatarUrl: 'https://huggingface.co/api/avatars/tencent',
-  },""",
-    """  {
+    '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
+pattern = r'const DEFAULT_MODEL_OPTIONS: ModelOption\[\] = \[.*?\];'
+match = re.search(pattern, content, re.DOTALL)
+
+if not match:
+    print("FAIL: Could not find DEFAULT_MODEL_OPTIONS array!")
+    sys.exit(1)
+
+new_array = """const DEFAULT_MODEL_OPTIONS: ModelOption[] = [
+  {
     id: 'gemma-4-31b',
     name: 'Gemma 4 31B:free',
-    modelPath: 'google/gemma-4-31b-it:free',
+    modelPath: 'openai/google/gemma-4-31b-it:free',
     avatarUrl: 'https://huggingface.co/api/avatars/google',
     recommended: true,
-  },""",
-    """  {
-    id: 'tencent-hy3',
-    name: 'Tencent HY3:free',
-    modelPath: 'tencent/hy3:free',
-    avatarUrl: 'https://huggingface.co/api/avatars/tencent',
-  },""",
-    """  {
-    id: 'llama-3.1-8b',
-    name: 'Llama 3.1 8B',
-    modelPath: 'meta-llama/llama-3.1-8b-instruct',
-    avatarUrl: 'https://huggingface.co/api/avatars/meta-llama',
-    recommended: true,
-  },""",
-    """  {
-    id: 'nemotron-70b',
-    name: 'Nemotron 70B',
-    modelPath: 'nvidia/llama-3.1-nemotron-70b-instruct',
-    avatarUrl: 'https://huggingface.co/api/avatars/nvidia',
-  },""",
-    """  {
-    id: 'gemma-2-9b',
-    name: 'Gemma 2 9B',
-    modelPath: 'google/gemma-2-9b-it',
-    avatarUrl: 'https://huggingface.co/api/avatars/google',
-  },""",
-    """  {
-    id: 'gemma-3-1b',
-    name: 'Gemma 3 1B',
-    modelPath: 'google/gemma-3-1b-it',
-    avatarUrl: 'https://huggingface.co/api/avatars/google',
-  },""",
-]
-for block in old_blocks:
-    content = content.replace(block, "")
-
-# === Add working free models at end of DEFAULT_MODEL_OPTIONS array ===
-idx_end = content.find("];", content.find("DEFAULT_MODEL_OPTIONS") if "DEFAULT_MODEL_OPTIONS" in content else 0)
-if idx_end == -1:
-    idx_end = content.rfind("];", 0, content.find("export") if "export" in content else len(content))
-
-if idx_end > 0:
-    new_models = """  {
+  },
+  {
     id: 'llama-3.3-70b',
     name: 'Llama 3.3 70B:free',
-    modelPath: 'meta-llama/llama-3.3-70b-instruct:free',
+    modelPath: 'openai/meta-llama/llama-3.3-70b-instruct:free',
     avatarUrl: 'https://huggingface.co/api/avatars/meta-llama',
     recommended: true,
   },
   {
-    id: 'gemma-3-1b',
-    name: 'Gemma 3 1B',
-    modelPath: 'google/gemma-3-1b-it',
-    avatarUrl: 'https://huggingface.co/api/avatars/google',
+    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',
@@ -134,11 +132,19 @@ if idx_end > 0:
     modelPath: 'openai/google/gemini-2.0-flash-001',
     avatarUrl: 'https://huggingface.co/api/avatars/google',
   },
-"""
-    content = content[:idx_end] + new_models + content[idx_end:]
+];"""
+
+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';")
+    print("OK: Replaced DEFAULT_MODEL_PATH")
+else:
+    print("OK: DEFAULT_MODEL_PATH not found")
 
-with open(FILE, "w") as f:
+with open(FILE, "w", encoding="utf-8") as f:
     f.write(content)
 
-print("✅ Frontend patched:")
-print("  NEW DEFAULT: Llama 3.3 70B:free (meta-llama/llama-3.3-70b-instruct:free) — free variant thật")
+print("OK: Frontend patched - Gemma 4 31B:free DEFAULT (openai/ prefix for OpenRouter routing)")