diff --git a/db/request_logs.go b/db/request_logs.go index ca37caa..65f9185 100644 --- 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 index f79563d..51b787d 100644 --- 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":