Add gpt-4o-mini to MaxLLMTokensMap (#356)

getzep · Sep 11, 2024 · 12a69d7 · 12a69d7
1 parent 60a3a11
commit 12a69d7
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/pkg/llms/llm_base.go b/pkg/llms/llm_base.go
@@ -185,6 +185,7 @@ var MaxLLMTokensMap = map[string]int{
 	"gpt-4-1106-preview":         128_000,
 	"gpt-4-turbo":                128_000,
 	"gpt-4o":                     128_000,
+	"gpt-4o-mini":                128_000,
 	"claude-instant-1":           100_000,
 	"claude-2":                   100_000,
 	"claude-3-5-sonnet-20240620": 200_000,