feat: add gpt-4o-mini as a model we can use (#148)

polyfact · Jan 3, 2025 · f4b1c95 · f4b1c95
1 parent 7ef46d5
commit f4b1c95
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/db/request_logs.go b/db/request_logs.go
@@ -42,6 +42,8 @@ func tokenToCredit(
 			return (inputTokenCount * 600) + (outputTokenCount * 1200)
 		case "gpt-4o":
 			return (inputTokenCount * 50) + (outputTokenCount * 150)
+		case "gpt-4o-mini":
+			return (inputTokenCount * 50) + (outputTokenCount * 150)
 		case "gpt-4-turbo":
 			return (inputTokenCount * 100) + (outputTokenCount * 300)
 		case "text-embedding-ada-002":

diff --git a/llm/provider.go b/llm/provider.go
@@ -46,6 +46,8 @@ func getAvailableModels(model string) (string, string) {
 		return "openai", "gpt-4-32k"
 	case "gpt-4o":
 		return "openai", "gpt-4o"
+	case "gpt-4o-mini":
+		return "openai", "gpt-4o-mini"
 	case "gpt-4-turbo":
 		return "openai", "gpt-4-turbo"
 	case "cohere":