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"]
@@ -16,7 +16,7 @@ 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"]
 
-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}
@@ -227,4 +227,4 @@ def gpt4stream(messages,model,api_keys):
         if "```json"  in cunk:
             helper.q.put_nowait("RESULT: "+cunk)
         else:
-            helper.q.put_nowait("END")
+            helper.q.put_nowait("END")