bep40 commited on
Commit
eaebe20
·
verified ·
1 Parent(s): e06f11b

Fix backend: owl-alpha with openai/ prefix

Browse files
Files changed (1) hide show
  1. patch_models.py +5 -5
patch_models.py CHANGED
@@ -12,10 +12,10 @@ patch = '''
12
  import os as _os
13
  AVAILABLE_MODELS.extend([
14
  {"id": "deepseek-ai/DeepSeek-V4-Pro", "label": "DeepSeek V4 Pro", "provider": "huggingface", "tier": "free", "recommended": True},
15
- {"id": "deepseek-ai/DeepSeek-V4-Flash", "label": "DeepSeek V4 Flash", "provider": "huggingface", "tier": "free"},
16
- {"id": "openai/deepseek/deepseek-v4-flash", "label": "DeepSeek V4 Flash (OR)", "provider": "openrouter", "tier": "free"},
17
  {"id": "Qwen/Qwen3-Coder-Next", "label": "Qwen3 Coder Next", "provider": "huggingface", "tier": "free", "recommended": True},
18
- {"id": "openrouter/owl-alpha", "label": "Owl Alpha", "provider": "openrouter", "tier": "free", "recommended": True},
 
 
19
  {"id": "openai/google/gemini-2.0-flash-001", "label": "Gemini 2.0 Flash", "provider": "openrouter", "tier": "free"},
20
  ])
21
  PREMIUM_MODEL_IDS = set() # No premium gate
@@ -27,9 +27,9 @@ if target in content:
27
  content = content.replace(target, target + patch)
28
  print("✅ Added models to AVAILABLE_MODELS")
29
 
30
- # Fix title generation - use huggingface/ prefix for litellm
31
  old_title = 'model="openai/openai/gpt-oss-120b:cerebras",\n api_base="https://router.huggingface.co/v1",\n api_key=api_key,'
32
- new_title = 'model="huggingface/deepseek-ai/DeepSeek-V4-Pro",\n api_key=api_key,'
33
  if old_title in content:
34
  content = content.replace(old_title, new_title)
35
  content = content.replace(' reasoning_effort="low",\n', '')
 
12
  import os as _os
13
  AVAILABLE_MODELS.extend([
14
  {"id": "deepseek-ai/DeepSeek-V4-Pro", "label": "DeepSeek V4 Pro", "provider": "huggingface", "tier": "free", "recommended": True},
 
 
15
  {"id": "Qwen/Qwen3-Coder-Next", "label": "Qwen3 Coder Next", "provider": "huggingface", "tier": "free", "recommended": True},
16
+ {"id": "openai/openrouter/owl-alpha", "label": "Owl Alpha Openrouter", "provider": "openrouter", "tier": "free", "recommended": True},
17
+ {"id": "MiniMaxAI/MiniMax-M2.7", "label": "MiniMax M2.7", "provider": "huggingface", "tier": "free"},
18
+ {"id": "zai-org/GLM-5.1", "label": "GLM 5.1", "provider": "huggingface", "tier": "free"},
19
  {"id": "openai/google/gemini-2.0-flash-001", "label": "Gemini 2.0 Flash", "provider": "openrouter", "tier": "free"},
20
  ])
21
  PREMIUM_MODEL_IDS = set() # No premium gate
 
27
  content = content.replace(target, target + patch)
28
  print("✅ Added models to AVAILABLE_MODELS")
29
 
30
+ # Fix title generation
31
  old_title = 'model="openai/openai/gpt-oss-120b:cerebras",\n api_base="https://router.huggingface.co/v1",\n api_key=api_key,'
32
+ new_title = 'model="deepseek-ai/DeepSeek-V4-Pro",\n api_base=_os.environ.get("OPENAI_API_BASE", "https://router.huggingface.co/v1"),\n api_key=_os.environ.get("OPENAI_API_KEY", api_key),'
33
  if old_title in content:
34
  content = content.replace(old_title, new_title)
35
  content = content.replace(' reasoning_effort="low",\n', '')