Merge pull request #89 from lloydchang/lloydchang-patch-6

feat: add gpt-4o, gpt-4o-mini, o1-preview, o1-mini

Author: andreashappe

pyproject.toml

@@ -28,7 +28,7 @@ dependencies = [
 	'Mako == 1.3.2',
 	'requests == 2.32.0',
 	'rich == 13.7.1',
-	'tiktoken == 0.6.0',
+	'tiktoken == 0.8.0',
 	'instructor == 1.3.5',
 	'PyYAML == 6.0.1',
 	'python-dotenv == 1.0.1',

src/hackingBuddyGPT/utils/openai/openai_llm.py

@@ -95,3 +95,31 @@ class GPT4(OpenAIConnection):
 class GPT4Turbo(OpenAIConnection):
     model: str = "gpt-4-turbo-preview"
     context_size: int = 128000
+
+
+@configurable("openai/gpt-4o", "OpenAI GPT-4o")
+@dataclass
+class GPT4oMini(OpenAIConnection):
+    model: str = "gpt-4o"
+    context_size: int = 128000
+
+
+@configurable("openai/gpt-4o-mini", "OpenAI GPT-4o-mini")
+@dataclass
+class GPT4oMini(OpenAIConnection):
+    model: str = "gpt-4o-mini"
+    context_size: int = 128000
+
+
+@configurable("openai/o1-preview", "OpenAI o1-preview")
+@dataclass
+class O1Preview(OpenAIConnection):
+    model: str = "o1-preview"
+    context_size: int = 128000
+
+
+@configurable("openai/o1-mini", "OpenAI o1-mini")
+@dataclass
+class O1Mini(OpenAIConnection):
+    model: str = "o1-mini"
+    context_size: int = 128000