Update utils/llms.py

utils/llms.py

@@ -6,7 +6,7 @@ from g4f.client import Client
 from litellm import completion
 import random
 
-from g4f.Provider import DeepInfraChat,Glider,Liaobots,Blackbox,ChatGptEs
+from g4f.Provider import DeepInfraChat,Glider,Liaobots,Blackbox,ChatGptEs,LambdaChat,TypeGPT
 os.environ["OPENROUTER_API_KEY"] = "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"
 os.environ["GROQ_API_KEY"] ="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"
 gemini_api_keys=["AIzaSyB7yKIdfW7Umv62G47BCdJjoHTJ9TeiAko","AIzaSyDtP05TyoIy9j0uPL7_wLEhgQEE75AZQSc","AIzaSyDOyjfqFhHmGlGJ2raX82XWTtmMcZxRshs"]
@@ -14,12 +14,12 @@ groq_api_keys=["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","gsk_b
 #[,"AIzaSyBPfR-HG_HeUgLF0LYW1XQgQUxFF6jF_0U","AIzaSyBz01gZCb9kzZF3lNHuwy_iajWhi9ivyDk"]]
 os.environ["GEMINI_API_KEY"] =random.choice(gemini_api_keys)
 os.environ["TOGETHERAI_API_KEY"] ="30bed0b842ed3268372d57f588c22452798e9af96aa8d3129ba745ef226282a8"
-DeepInfraChat.models = ["deepseek-ai/DeepSeek-R1-Turbo","Qwen/QwQ-32B","deepseek-ai/DeepSeek-R1"]
+DeepInfraChat.models = ["deepseek-ai/DeepSeek-R1-Turbo","Qwen/QwQ-32B","deepseek-ai/DeepSeek-R1","deepseek-ai/DeepSeek-V3-0324"]
 
-REASONING_CORRESPONDANCE = {"DeepSeek-R1-Glider":Glider,"DeepSeekR1":DeepInfraChat}
+REASONING_CORRESPONDANCE = {"DeepSeek-R1-Glider":Glider, "DeepSeekR1-LAMBDA":LambdaChat,"DeepSeekR1":DeepInfraChat,"deepseek-slow":TypeGPT}
 REASONING_QWQ = {"qwq-32b":DeepInfraChat}
 
-CHAT_CORRESPONDANCE = {"DeepSeek-V3":DeepInfraChat,"DeepSeek-V3-Blackbox":Blackbox}
+CHAT_CORRESPONDANCE = {"DeepSeek-V3":DeepInfraChat}
 
 client = Client()
 genai.configure(api_key="AIzaSyAQgAtQPpY0bQaCqCISGxeyF6tpDePx-Jg")
@@ -137,12 +137,12 @@ def gpt4stream(messages,model,api_keys):
         helper.q.put_nowait("RESULT: "+cunk)
 
 
-    elif model=="deepseek.r1":
+    elif model=="deepseek.r1" or model=="deepseek-chat":
         cunk=""
 
-        if "V3" in model:
+        if "chat" in model:
             providers = CHAT_CORRESPONDANCE
-            model_name="deepseek-v3"
+            model_name="deepseek-ai/DeepSeek-V3-0324"
         else:
             providers = REASONING_CORRESPONDANCE
             model_name="deepseek-r1"