akiko19191 commited on
Commit
0c18fe7
·
verified ·
1 Parent(s): bca0e76

Update utils/llms.py

Browse files
Files changed (1) hide show
  1. utils/llms.py +2 -2
utils/llms.py CHANGED
@@ -5,7 +5,7 @@ import random
5
  import json
6
  import os
7
 
8
- from g4f.Provider import DeepInfraChat,LambdaChat,TypeGPT
9
 
10
  gemini_api_keys=json.loads(os.environ.get("GEMINI_KEY_LIST"))
11
  groq_api_keys=json.loads(os.environ.get("GROQ_API_KEYS"))
@@ -18,7 +18,7 @@ deepinframodels=["meta-llama/Llama-4-Maverick-17B-128E-Instruct-FP8","microsoft/
18
  chutes_models={'Llama-4-Maverick-17B-128E-Instruct-FP8':'chutesai/Llama-4-Maverick-17B-128E-Instruct-FP8', "Qwen3-235B":"Qwen/Qwen3-235B-A22B","MAI-DS-R1-FP8":"microsoft/MAI-DS-R1-FP8","DeepSeek-V3-0324":"deepseek-ai/DeepSeek-V3-0324","deepseek-reasoner":"deepseek-ai/DeepSeek-R1-0528","GLM-4-32B-0414":"THUDM/GLM-4-32B-0414","GLM-Z1-32B-0414":"THUDM/GLM-Z1-32B-0414","DeepSeek-R1T-Chimera":"tngtech/DeepSeek-R1T-Chimera", "DeepSeek-R1-Zero":"deepseek-ai/DeepSeek-R1-Zero"}
19
  github_models={"gpt4.1":"gpt-4.1","gpt-4o":"gpt-4o","o4-mini":"o4-mini"}
20
 
21
- REASONING_CORRESPONDANCE = {"DeepSeekR1-LAMBDA":LambdaChat,"DeepSeekR1":DeepInfraChat,"deepseek-slow":TypeGPT}
22
  os.environ["GEMINI_API_KEY"] =random.choice(gemini_api_keys)
23
 
24
  REASONING_QWQ = {"qwq-32b":DeepInfraChat}
 
5
  import json
6
  import os
7
 
8
+ from g4f.Provider import DeepInfraChat,LambdaChat
9
 
10
  gemini_api_keys=json.loads(os.environ.get("GEMINI_KEY_LIST"))
11
  groq_api_keys=json.loads(os.environ.get("GROQ_API_KEYS"))
 
18
  chutes_models={'Llama-4-Maverick-17B-128E-Instruct-FP8':'chutesai/Llama-4-Maverick-17B-128E-Instruct-FP8', "Qwen3-235B":"Qwen/Qwen3-235B-A22B","MAI-DS-R1-FP8":"microsoft/MAI-DS-R1-FP8","DeepSeek-V3-0324":"deepseek-ai/DeepSeek-V3-0324","deepseek-reasoner":"deepseek-ai/DeepSeek-R1-0528","GLM-4-32B-0414":"THUDM/GLM-4-32B-0414","GLM-Z1-32B-0414":"THUDM/GLM-Z1-32B-0414","DeepSeek-R1T-Chimera":"tngtech/DeepSeek-R1T-Chimera", "DeepSeek-R1-Zero":"deepseek-ai/DeepSeek-R1-Zero"}
19
  github_models={"gpt4.1":"gpt-4.1","gpt-4o":"gpt-4o","o4-mini":"o4-mini"}
20
 
21
+ REASONING_CORRESPONDANCE = {"DeepSeekR1-LAMBDA":LambdaChat,"DeepSeekR1":DeepInfraChat}
22
  os.environ["GEMINI_API_KEY"] =random.choice(gemini_api_keys)
23
 
24
  REASONING_QWQ = {"qwq-32b":DeepInfraChat}