Adding new openai 4o-mini model (#352)

* adding new openai 4o-mini model * updating config.yaml comment for available openai llm models * Update pkg/llms/llm_base.go Co-authored-by: ellipsis-dev[bot] <65095814+ellipsis-dev[bot]@users.noreply.github.com> --------- Co-authored-by: bricemacias-rocketcode <[email protected]> Co-authored-by: ellipsis-dev[bot] <65095814+ellipsis-dev[bot]@users.noreply.github.com>
getzep · Aug 11, 2024 · e160be0 · e160be0
1 parent 415979b
commit e160be0
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 9 deletions.
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; 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-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
@@ -152,14 +152,16 @@ func NewLLMError(message string, originalError error) *LLMError {
 }
 
 var ValidOpenAILLMs = map[string]bool{
-	"gpt-3.5-turbo":      true,
-	"gpt-4":              true,
-	"gpt-3.5-turbo-16k":  true,
-	"gpt-3.5-turbo-1106": true,
-	"gpt-4-32k":          true,
-	"gpt-4-1106-preview": true,
-	"gpt-4-turbo":        true,
-	"gpt-4o":             true,
+	"gpt-3.5-turbo":          true,
+	"gpt-4":                  true,
+	"gpt-3.5-turbo-16k":      true,
+	"gpt-3.5-turbo-1106":     true,
+	"gpt-4-32k":              true,
+	"gpt-4-1106-preview":     true,
+	"gpt-4-turbo":            true,
+	"gpt-4o":                 true,
+	"gpt-4o-mini":            true,
+	"gpt-4o-mini-2024-07-18": true,
 }
 
 var ValidAnthropicLLMs = map[string]bool{