add chatgpt-4o-latest support

defog-ai · Jan 21, 2025 · 8e0745b · 8e0745b
1 parent 6d9f208
commit 8e0745b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/defog_utils/utils_llm.py b/defog_utils/utils_llm.py
@@ -4,6 +4,7 @@
 from typing import Dict, List, Optional, Any
 
 LLM_COSTS_PER_TOKEN = {
+    "chatgpt-4o": {"input_cost_per1k": 0.0025, "output_cost_per1k": 0.01},
     "gpt-4o": {"input_cost_per1k": 0.0025, "output_cost_per1k": 0.01},
     "gpt-4o-mini": {"input_cost_per1k": 0.00015, "output_cost_per1k": 0.0006},
     "o1": {"input_cost_per1k": 0.015, "output_cost_per1k": 0.06},

diff --git a/defog_utils/utils_multi_llm.py b/defog_utils/utils_multi_llm.py
@@ -24,7 +24,7 @@ def map_model_to_chat_fn(model: str) -> Callable:
         return chat_anthropic
     if model.startswith("gemini"):
         return chat_gemini
-    if model.startswith("gpt") or model.startswith("o1"):
+    if model.startswith("gpt") or model.startswith("o1") or model.startswith("chatgpt"):
         return chat_openai
     if model.startswith("deepseek"):
         return chat_openai
@@ -45,7 +45,7 @@ def map_model_to_chat_fn_async(model: str) -> Callable:
         return chat_anthropic_async
     if model.startswith("gemini"):
         return chat_gemini_async
-    if model.startswith("gpt") or model.startswith("o1"):
+    if model.startswith("gpt") or model.startswith("o1") or model.startswith("chatgpt"):
         return chat_openai_async
     if model.startswith("deepseek"):
         return chat_openai_async