Add o1-preview and o1-mini

Author: ajhai

llmstack/apps/types/agent.py

@@ -16,6 +16,8 @@ class AgentModel(StrEnum):
     GPT_4_1106_PREVIEW = "gpt-4-1106-preview"
     GPT_4_O = "gpt-4o"
     GPT_4_O_MINI = "gpt-4o-mini"
+    O1_PREVIEW = "o1-preview"
+    O1_MINI = "o1-mini"
 
 
 class AgentConfigSchema(BaseSchema):

llmstack/play/actors/agent.py

@@ -146,15 +146,19 @@ def __init__(
         if configured_model == "gpt-4o-mini":
             self._base_price_per_message = 20
         elif configured_model == "gpt-4o":
-            self._base_price_per_message = 100
+            self._base_price_per_message = 200
         elif configured_model == "gpt-4-32k":
             self._base_price_per_message = 500
         elif configured_model == "gpt-4":
             self._base_price_per_message = 300
         elif configured_model == "gpt-4-turbo-latest":
-            self._base_price_per_message = 200
+            self._base_price_per_message = 300
         elif configured_model == "gpt-4-turbo":
+            self._base_price_per_message = 300
+        elif configured_model == "o1-mini":
             self._base_price_per_message = 200
+        elif configured_model == "o1-preview":
+            self._base_price_per_message = 1000
         else:
             self._base_price_per_message = 150
 

llmstack/processors/providers/azure/azure_chat_completions.py

@@ -21,6 +21,8 @@ class ChatCompletionsModel(StrEnum):
     GPT_3_5 = "gpt-35-turbo"
     GPT_3_5_16 = "gpt-35-turbo-16k"
     GPT_4_32 = "gpt-4-32k"
+    O1_PREVIEW = "o1-preview"
+    O1_MINI = "o1-mini"
 
 
 class Role(StrEnum):

llmstack/processors/providers/openai/chat_completions.py

@@ -32,6 +32,8 @@ class ChatCompletionsModel(StrEnum):
     GPT_3_5_16K = "gpt-3.5-turbo-16k"
     GPT_4_LATEST = "gpt-4-0125-preview"
     GPT_4_1106_PREVIEW = "gpt-4-1106-preview"
+    O1_PREVIEW = "o1-preview"
+    O1_MINI = "o1-mini"
 
     def model_name(self):
         return self.value

llmstack/processors/providers/openai/chat_completions_vision.py

@@ -33,6 +33,8 @@ class ChatCompletionsVisionModel(StrEnum):
     GPT_4_1106_VISION_PREVIEW = "gpt-4-1106-vision-preview"
     GPT_4_O = "gpt-4o"
     GPT_4_O_MINI = "gpt-4o-mini"
+    O1_PREVIEW = "o1-preview"
+    O1_MINI = "o1-mini"
 
     def model_name(self):
         return self.value

llmstack/processors/providers/promptly/web_browser.py

@@ -187,6 +187,8 @@ class Model(StrEnum):
     GPT_4_32K = "gpt-4-32k"
     GPT_4_LATEST = "gpt-4-turbo-latest"
     GPT_4_V_LATEST = "gpt-4-vision-latest"
+    O1_PREVIEW = "o1-preview"
+    O1_MINI = "o1-mini"
 
 
 class GoogleVisionModel(StrEnum):