Upload patch_frontend.py

patch_frontend.py

@@ -1,54 +1,71 @@
-"""Patch frontend ChatInput.tsx: Fix model IDs and rename dropdown labels."""
+"""Patch frontend ChatInput.tsx + model.ts: Fix model IDs and rename dropdown labels."""
 
-FILE = "/source/frontend/src/components/Chat/ChatInput.tsx"
+# === Step 1: Patch model.ts constants ===
+MODEL_FILE = "/source/frontend/src/utils/model.ts"
+with open(MODEL_FILE, "r") as f:
+    model_content = f.read()
+
+model_content = model_content.replace(
+    "export const CLAUDE_OPUS_48_MODEL_PATH = 'anthropic/claude-opus-4.8:fal-ai';",
+    "export const CLAUDE_OPUS_48_MODEL_PATH = 'openrouter/owl-alpha';"
+)
+model_content = model_content.replace(
+    "export const KIMI_K27_CODE_MODEL_PATH = 'moonshotai/Kimi-K2.7-Code:novita';",
+    "export const KIMI_K27_CODE_MODEL_PATH = 'deepseek-ai/DeepSeek-V4-Pro';"
+)
+model_content = model_content.replace(
+    "export const MINIMAX_M3_MODEL_PATH = 'MiniMaxAI/MiniMax-M3:novita';",
+    "export const MINIMAX_M3_MODEL_PATH = 'deepseek-ai/DeepSeek-V4-Flash';"
+)
+model_content = model_content.replace(
+    "export const GLM_52_MODEL_PATH = 'zai-org/GLM-5.2:novita';",
+    "export const GLM_52_MODEL_PATH = 'openai/deepseek/deepseek-v4-flash';"
+)
+model_content = model_content.replace(
+    "export const DEEPSEEK_V4_PRO_MODEL_PATH = 'deepseek-ai/DeepSeek-V4-Pro:novita';",
+    "export const DEEPSEEK_V4_PRO_MODEL_PATH = 'nvidia/nemotron-3-super-120b-a12b:free';"
+)
+
+with open(MODEL_FILE, "w") as f:
+    f.write(model_content)
+print("✅ model.ts constants updated")
 
+# === Step 2: Patch ChatInput.tsx names and add new models ===
+FILE = "/source/frontend/src/components/Chat/ChatInput.tsx"
 with open(FILE, "r") as f:
     content = f.read()
 
-# === Row 2: Claude Opus 4.6 → Owl Alpha ===
-content = content.replace("'Claude Opus 4.6'", "'Owl Alpha'")
-content = content.replace('"Claude Opus 4.6"', '"Owl Alpha"')
-content = content.replace("'Claude Opus 4'", "'Owl Alpha'")
-content = content.replace('"Claude Opus 4"', '"Owl Alpha"')
-content = content.replace("'Claude Sonnet 4.5'", "'Owl Beta'")
-content = content.replace('"Claude Sonnet 4.5"', '"Owl Beta"')
-content = content.replace("claude.label ?? option.name", "option.name")
-content = content.replace("claude.label || option.name", "option.name")
-content = content.replace("description: 'Anthropic'", "description: '🦉 Top Reasoning'")
-
-# === Row 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('"Kimi K2.6"', '"DeepSeek V4 Pro"')
+# Update model names (Row 1: Claude → Owl Alpha)
+content = content.replace("'Claude Opus 4.8'", "'Owl Alpha'")
+content = content.replace('"Claude Opus 4.8"', '"Owl Alpha"')
 
-# === Row 4: MiniMax M2.7 → DeepSeek V4 Flash, Novita → HF Inference ===
-content = content.replace("MiniMaxAI/MiniMax-M2.7", "deepseek-ai/DeepSeek-V4-Flash")
-content = content.replace("'MiniMax M2.7'", "'DeepSeek V4 Flash'")
-content = content.replace('"MiniMax M2.7"', '"DeepSeek V4 Flash"')
-content = content.replace("description: 'Novita'", "description: 'HF Inference'", 1)
+# Row 3: Kimi K2.7 Code → DeepSeek V4 Pro
+content = content.replace("'Kimi K2.7 Code'", "'DeepSeek V4 Pro'")
+content = content.replace('"Kimi K2.7 Code"', '"DeepSeek V4 Pro"')
 
-# === Row 5: GLM 5.1 → DeepSeek V4 Flash, Together → Openrouter ===
-content = content.replace("zai-org/GLM-5.1", "openai/deepseek/deepseek-v4-flash")
-content = content.replace("'GLM 5.1'", "'DeepSeek V4 Flash'")
-content = content.replace('"GLM 5.1"', '"DeepSeek V4 Flash"')
-content = content.replace("description: 'Together'", "description: 'Openrouter'")
+# Row 4: MiniMax M3 → DeepSeek V4 Flash
+content = content.replace("'MiniMax M3'", "'DeepSeek V4 Flash'")
+content = content.replace('"MiniMax M3"', '"DeepSeek V4 Flash"')
 
-# === Row 6: DeepSeek V4 Pro (DeepInfra) → nvidia/nemotron-3-super-120b-a12b:free ===
-content = content.replace("deepseek-ai/DeepSeek-V4-Pro:deepinfra", "nvidia/nemotron-3-super-120b-a12b:free")
-content = content.replace("description: 'DeepInfra'", "description: 'Openrouter Free'")
-# Find the name for row 6 - it's the one with id 'deepseek-v4-pro'
-# In source: id: 'deepseek-v4-pro', name: 'DeepSeek V4 Pro'
-# We need to change name to show the nvidia model path
-content = content.replace("id: 'deepseek-v4-pro',\n    name: 'DeepSeek V4 Pro'", "id: 'nemotron-120b',\n    name: 'nvidia/nemotron-3-super-120b-a12b:free'")
+# Row 5: GLM 5.2 → DeepSeek V4 Flash (also rename)
+content = content.replace("'GLM 5.2'", "'DeepSeek V4 Flash'")
+content = content.replace('"GLM 5.2"', '"DeepSeek V4 Flash"')
 
-# === Row 1 description: second Novita (if any remains) ===
-content = content.replace("description: 'Novita'", "description: 'HF Inference'")
+# Row 6: DeepSeek V4 Pro → Nemotron (use unique id+name pattern to avoid collision)
+# This must be done AFTER the Kimi→DeepSeek V4 Pro replacement above
+content = content.replace(
+    "  {\n    id: 'deepseek-v4-pro',\n    name: 'DeepSeek V4 Pro',\n    modelPath: DEEPSEEK_V4_PRO_MODEL_PATH,\n    avatarUrl: getHfAvatarUrl('deepseek-ai/DeepSeek-V4-Pro'),\n  },",
+    "  {\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  },"
+)
 
-# === Update generic descriptions ===
-content = content.replace("'Best coding model'", "'⚡ Mạnh nhất · Code + Reasoning'")
-content = content.replace("'Best all-around model'", "'⚡ Mạnh nhất · Code + Reasoning'")
+# Update isClaudePath to avoid obsolete function
+content = content.replace(
+    "export function isClaudePath(modelPath: string | undefined): boolean {\n  return !!modelPath && modelPath.includes('anthropic');\n}",
+    "export function isClaudePath(modelPath: string | undefined): boolean {\n  return !!modelPath && modelPath.includes('openrouter');\n}"
+)
 
-# === Add custom models at end of array ===
+# === Add custom models at end of DEFAULT_MODEL_OPTIONS array ===
+# ModelOption interface has no 'description' field — only: id, name, modelPath, avatarUrl, recommended?
 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))
@@ -57,7 +74,6 @@ if idx_end > 0:
     new_models = """  {
     id: 'gemma-4-31b',
     name: 'Gemma 4 31B',
-    description: 'HF Inference',
     modelPath: 'google/gemma-4-31B-it',
     avatarUrl: 'https://huggingface.co/api/avatars/google',
     recommended: true,
@@ -65,14 +81,12 @@ if idx_end > 0:
   {
     id: 'tencent-hy3',
     name: 'Tencent HY3',
-    description: 'Openrouter Free',
     modelPath: 'openai/tencent/hy3:free',
     avatarUrl: 'https://huggingface.co/api/avatars/tencent',
   },
   {
     id: 'gemma-3-1b',
     name: 'Gemma 3 1B',
-    description: 'HF Inference',
     modelPath: 'google/gemma-3-1b-it',
     avatarUrl: 'https://huggingface.co/api/avatars/google',
     recommended: true,
@@ -80,7 +94,6 @@ if idx_end > 0:
   {
     id: 'qwen3-coder-next',
     name: 'Qwen3 Coder Next',
-    description: 'HF Inference',
     modelPath: 'Qwen/Qwen3-Coder-Next',
     avatarUrl: 'https://huggingface.co/api/avatars/Qwen',
     recommended: true,
@@ -88,7 +101,6 @@ if idx_end > 0:
   {
     id: 'gemini-2.0-flash',
     name: 'Gemini 2.0 Flash',
-    description: 'Openrouter',
     modelPath: 'openai/google/gemini-2.0-flash-001',
     avatarUrl: 'https://huggingface.co/api/avatars/google',
   },
@@ -98,15 +110,11 @@ if idx_end > 0:
 with open(FILE, "w") as f:
     f.write(content)
 
-print("✅ Frontend patched:")
-print("  Row 1: DeepSeek V4 Pro | HF Inference")
-print("  Row 2: Owl Alpha | 🦉 Top Reasoning")
-print("  Row 3: GPT-5.5 | OpenAI")
-print("  Row 4: DeepSeek V4 Flash | HF Inference (was Novita)")
-print("  Row 5: DeepSeek V4 Flash | Openrouter (was Together)")
-print("  Row 6: nvidia/nemotron-3-super-120b-a12b:free | Openrouter Free")
-print("  Row 7: Gemma 4 31B | HF Inference (NEW)")
-print("  Row 8: Tencent HY3 | Openrouter Free (NEW)")
-print("  Row 9: Gemma 3 1B | HF Inference")
-print("  Row 10: Qwen3 Coder Next | HF Inference")
-print("  Row 11: Gemini 2.0 Flash | Openrouter")
+print("✅ Frontend patched for current upstream:")
+print("  Row 1 (Claude): Owl Alpha")
+print("  Row 2 (GPT-5.5): unchanged")
+print("  Row 3 (Kimi): DeepSeek V4 Pro")
+print("  Row 4 (MiniMax): DeepSeek V4 Flash")
+print("  Row 5 (GLM): DeepSeek V4 Flash (OR)")
+print("  Row 6 (DeepSeek): nvidia/nemotron-3-super-120b-a12b:free")
+print("  NEW: Gemma 4 31B, Tencent HY3, Gemma 3 1B, Qwen3 Coder Next, Gemini 2.0 Flash")