update models

shivanker · Jan 31, 2025 · b5b8aae · b5b8aae
1 parent 59cadca
commit b5b8aae
Show file tree

Hide file tree

Showing 2 changed files with 31 additions and 21 deletions.
diff --git a/lite_llms.py b/lite_llms.py
@@ -6,15 +6,17 @@ class TextModel(Enum):
     GPT_4O = "gpt-4o"
     GPT_4O_MINI = "gpt-4o-mini"
     GPT_4_TURBO = "gpt-4-turbo"
-    O1_PREVIEW = "o1-preview"
+    O1 = "o1"
     O1_MINI = "o1-mini"
     CLAUDE_3_OPUS = "claude-3-opus-20240229"
     CLAUDE_35_SONNET = "claude-3-5-sonnet-20241022"
-    CLAUDE_3_HAIKU = "claude-3-haiku-20240307"
-    GEMINI_15_PRO = "gemini/gemini-1.5-pro-latest"
-    GEMINI_15_FLASH = "gemini/gemini-1.5-flash-latest"
+    CLAUDE_35_HAIKU = "claude-3-5-haiku-20241022"
+    GEMINI_2_PRO = "gemini/gemini-exp-1206"
+    GEMINI_2_FLASH = "gemini/gemini-2.0-flash-exp"
+    GEMINI_FLASH_THINKING = "gemini/gemini-2.0-flash-thinking-exp-01-21"
     # GROQ_LLAMA3_70B = "groq/llama3-70b-8192"
     # GROQ_LLAMA3_8B = "groq/llama3-8b-8192"
     LLAMA3_70B = "fireworks_ai/llama-v3-70b-instruct"
     LLAMA31_405B = "fireworks_ai/llama-v3p1-405b-instruct"
     LLAMA31_8B = "fireworks_ai/llama-v3p1-8b-instruct"
+    DEEPSEEK_R1 = "fireworks_ai/deepseek-r1"
diff --git a/session.py b/session.py
@@ -230,9 +230,9 @@ def process_command(self, text, say=lambda text: None):
             say(text="Session has been reset.")
         elif cmd in ("\\who?", "\\who", "\\llm", "\\model"):
             say(text=f"You are currently chatting with {self.model.value}.")
-        elif cmd in ["\\o1", "\\o1-preview"]:
-            self.model = TextModel.O1_PREVIEW
-            say(text="Model set to O1 Preview.")
+        elif cmd == "\\o1":
+            self.model = TextModel.O1
+            say(text="Model set to O1.")
         elif cmd in ["\\o1-mini", "\\o1mini", "\\mini"]:
             self.model = TextModel.O1_MINI
             say(text="Model set to O1 Mini.")
@@ -264,14 +264,20 @@ def process_command(self, text, say=lambda text: None):
             self.model = TextModel.CLAUDE_35_SONNET
             say(text="Model set to Claude 3.5 Sonnet.")
         elif cmd == "\\haiku":
-            self.model = TextModel.CLAUDE_3_HAIKU
-            say(text="Model set to Claude 3 Haiku.")
+            self.model = TextModel.CLAUDE_35_HAIKU
+            say(text="Model set to Claude 3.5 Haiku.")
         elif cmd == "\\gemini":
-            self.model = TextModel.GEMINI_15_PRO
-            say(text="Model set to Gemini 1.5 Pro.")
+            self.model = TextModel.GEMINI_2_PRO
+            say(text="Model set to Gemini 2.0 Pro.")
         elif cmd == "\\flash":
-            self.model = TextModel.GEMINI_15_FLASH
-            say(text="Model set to Gemini 1.5 Flash.")
+            self.model = TextModel.GEMINI_2_FLASH
+            say(text="Model set to Gemini 2.0 Flash.")
+        elif cmd == "\\flash-thinking":
+            self.model = TextModel.GEMINI_FLASH_THINKING
+            say(text="Model set to Gemini 2.0 Flash Thinking.")
+        elif cmd == "\\deepseek":
+            self.model = TextModel.DEEPSEEK_R1
+            say(text="Model set to Deepseek R1.")
         elif cmd == "\\stream":
             self.streaming_mode ^= True
             say(
@@ -286,16 +292,18 @@ def process_command(self, text, say=lambda text: None):
         elif cmd == "\\help":
             say(
                 f"""
-{HELP_PREAMBLE} I am a basic chatbot to quickly use GPT4, Claude, LLaMA & Gemini in one place. The chat is organized in sessions. Once you reset a session, all the previous conversation is lost. I am incapable of analyzing images or writing code right now, but feel free to upload PDFs, text files, or link to any websites, and I'll try to scrape whatever text I can. Here's the full list of available commands you can use:\n
+{HELP_PREAMBLE} I am a basic chatbot to quickly use GPT4, Claude, LLaMA & Gemini in one place. The chat is organized in sessions. Once you reset a session, all the previous conversation is lost. I am incapable of analyzing images or writing code right now, but feel free to upload PDFs, text files, or link to any websites, and I'll try to scrape whatever text I can. Note that model changes preserve the session so far. Here's the full list of available commands you can use:\n
 - \\reset: Reset the chat session. Preserves the previous LLM you were chatting with.\n
 - \\who: Returns the name of the chat model you are chatting with.\n
-- \\o1: Use O1 Preview for future messages. Preserves the session so far.\n
-- \\o1mini: Use O1 Mini for future messages. Preserves the session so far.\n
-- \\gpt4o: Use GPT-4o (Omni) for future messages. Preserves the session so far.\n
-- \\sonnet: Use Claude 3.5 Sonnet for future messages. Preserves the session so far.\n
-- \\llama: Use LLaMA-3.1 405B for future messages. Preserves the session so far.\n
-- \\gemini: Use Gemini 1.5 Pro for future messages. Preserves the session so far.\n
-- \\flash: Use Gemini 1.5 Flash for future messages. Preserves the session so far.\n
+- \\o1: Use O1 for future messages.\n
+- \\o1mini: Use O1 Mini for future messages.\n
+- \\gpt4o: Use GPT-4o (Omni) for future messages.\n
+- \\sonnet: Use Claude 3.5 Sonnet for future messages.\n
+- \\llama: Use LLaMA-3.1 405B for future messages.\n
+- \\gemini: Use Gemini 2.0 Pro for future messages.\n
+- \\flash: Use Gemini 2.0 Flash for future messages.\n
+- \\flash-thinking: Use Gemini 2.0 Flash Thinking for future messages.\n
+- \\deepseek: Use Deepseek R1 for future messages.\n
 - \\stream: Toggle streaming mode. In streaming mode, the bot will send you a message every time it generates a new token.\n
 - \\extract: [debug] Extract text from a URL or a YT video.\n
                 """