From 024fd26fa88d6ee1c059d9555f5bb645dfba6af7 Mon Sep 17 00:00:00 2001 From: Artur Gaspar Date: Wed, 18 Dec 2024 05:30:55 -0300 Subject: [PATCH] feat: gpt-4o-mini support --- ai_eval/llm.py | 1 + 1 file changed, 1 insertion(+) diff --git a/ai_eval/llm.py b/ai_eval/llm.py index 0c956e7..2d4cbc7 100644 --- a/ai_eval/llm.py +++ b/ai_eval/llm.py @@ -12,6 +12,7 @@ class SupportedModels(Enum): """ GPT4O = "gpt-4o" + GPT4O_MINI = "gpt-4o-mini" GEMINI_PRO = "gemini/gemini-pro" CLAUDE_SONNET = "claude-3-5-sonnet-20240620" LLAMA = "ollama/llama2"