Upload patch_models.py

patch_models.py

@@ -16,26 +16,25 @@ if "NO_TOOLS_MODELS" in content:
     content = re.sub(r'NO_TOOLS_MODELS\s*=\s*\{[^}]*\}\n?', '', content)
     print("OK: Removed old NO_TOOLS_MODELS")
 
-# Fix model IDs for OpenRouter
-content = content.replace(
-    'KIMI_K27_CODE_MODEL_ID = "moonshotai/Kimi-K2.7-Code:novita"',
-    'KIMI_K27_CODE_MODEL_ID = "deepseek-ai/DeepSeek-V4-Pro"'
-)
-content = content.replace(
-    'MINIMAX_M3_MODEL_ID = "MiniMaxAI/MiniMax-M3:novita"',
-    'MINIMAX_M3_MODEL_ID = "deepseek-ai/DeepSeek-V4-Flash"'
-)
-content = content.replace(
-    'GLM_52_MODEL_ID = "zai-org/GLM-5.2:novita"',
-    'GLM_52_MODEL_ID = "openai/deepseek/deepseek-v4-flash"'
-)
-content = content.replace(
-    'DEEPSEEK_V4_PRO_MODEL_ID = "deepseek-ai/DeepSeek-V4-Pro:novita"',
-    'DEEPSEEK_V4_PRO_MODEL_ID = "nvidia/nemotron-3-super-120b-a12b:free"'
-)
-
-# Add missing constants before DEEPSEEK_V4_PRO_MODEL_ID line
-insert_before = '''TENCENT_HY3_FREE_MODEL_ID = "openai/tencent/hy3:free"
+# Fix model IDs for OpenRouter - use exact full-line patterns to avoid accidental partial matches
+replacements = [
+    ('KIMI_K27_CODE_MODEL_ID = "moonshotai/Kimi-K2.7-Code:novita"',
+     'KIMI_K27_CODE_MODEL_ID = "deepseek-ai/DeepSeek-V4-Pro"'),
+    ('MINIMAX_M3_MODEL_ID = "MiniMaxAI/MiniMax-M3:novita"',
+     'MINIMAX_M3_MODEL_ID = "deepseek-ai/DeepSeek-V4-Flash"'),
+    ('GLM_52_MODEL_ID = "zai-org/GLM-5.2:novita"',
+     'GLM_52_MODEL_ID = "openai/deepseek/deepseek-v4-flash"'),
+    ('DEEPSEEK_V4_PRO_MODEL_ID = "deepseek-ai/DeepSeek-V4-Pro:novita"',
+     'DEEPSEEK_V4_PRO_MODEL_ID = "nvidia/nemotron-3-super-120b-a12b:free"'),
+]
+for old, new in replacements:
+    content = content.replace(old, new)
+print("OK: Model ID replacements applied")
+
+# Add missing constants BEFORE the specific line DEEPSEEK_V4_PRO_MODEL_ID = "..."
+if "TENCENT_HY3_FREE_MODEL_ID" not in content:
+    target_line = 'DEEPSEEK_V4_PRO_MODEL_ID = "nvidia/nemotron-3-super-120b-a12b:free"'
+    insert_block = '''TENCENT_HY3_FREE_MODEL_ID = "openai/tencent/hy3:free"
 GEMMA_4_31B_FREE_MODEL_ID = "openai/google/gemma-4-31b-it:free"
 LLAMA_3_3_70B_FREE_MODEL_ID = "openai/meta-llama/llama-3.3-70b-instruct:free"
 LLAMA_3_1_8B_MODEL_ID = "openai/meta-llama/llama-3.1-8b-instruct"
@@ -44,16 +43,20 @@ LAGUNA_S21_FREE_MODEL_ID = "openai/poolside/laguna-s-2.1:free"
 NEX_N2_MINI_MODEL_ID = "openai/nex-agi/nex-n2-mini"
 LING_3_0_FLASH_FREE_MODEL_ID = "openai/inclusionai/ling-3.0-flash:free"
 '''
-if "TENCENT_HY3_FREE_MODEL_ID" not in content:
-    content = content.replace('DEEPSEEK_V4_PRO_MODEL_ID', insert_before + 'DEEPSEEK_V4_PRO_MODEL_ID')
+    # Only replace the first occurrence (the variable definition)
+    if target_line in content:
+        content = content.replace(target_line, insert_block + target_line)
+        print("OK: Added missing constants")
+    else:
+        print("WARN: Could not find target line")
 
-# Replace all references to latest deepseek models with 0731 version
+# Replace all references to ~deepseek deepseek models with 0731 version (quoted strings only)
 content = content.replace('"~deepseek/deepseek-v4-flash-latest"', '"deepseek/deepseek-v4-flash-0731"')
 content = content.replace('"~deepseek/deepseek-v4-flash-0731"', '"deepseek/deepseek-v4-flash-0731"')
 
 with open(IDS_FILE, "w") as f:
     f.write(content)
-print("OK: model_ids.py patched")
+print("OK: model_ids.py written")
 
 ast.parse(content)  # validate syntax
 print("OK: model_ids.py syntax OK")