chore: Add gpt-4o-2024-08-06 support (#353)

getzep · Aug 12, 2024 · 1d6fb46 · 1d6fb46
1 parent e160be0
commit 1d6fb46
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/config.yaml b/config.yaml
@@ -1,7 +1,7 @@
 llm:
   # openai or anthropic
   service: "openai"
-  # OpenAI: gpt-3.5-turbo, gpt-4, gpt-3.5-turbo-1106, gpt-3.5-turbo-16k, gpt-4-32k, gpt-4o-mini, gpt-4o-mini-2024-07-18; Anthropic: claude-instant-1 or claude-2
+  # OpenAI: gpt-3.5-turbo, gpt-4, gpt-3.5-turbo-1106, gpt-3.5-turbo-16k, gpt-4-32k, gpt-4o, gpt-4o-2024-08-06, gpt-4o-mini, gpt-4o-mini-2024-07-18; Anthropic: claude-instant-1 or claude-2
   model: "gpt-3.5-turbo-1106"
   ## OpenAI-specific settings
   # Only used for Azure OpenAI API

diff --git a/pkg/llms/llm_base.go b/pkg/llms/llm_base.go
@@ -160,6 +160,7 @@ var ValidOpenAILLMs = map[string]bool{
 	"gpt-4-1106-preview":     true,
 	"gpt-4-turbo":            true,
 	"gpt-4o":                 true,
+	"gpt-4o-2024-08-06":      true,
 	"gpt-4o-mini":            true,
 	"gpt-4o-mini-2024-07-18": true,
 }