diff --git a/defog_utils/utils_llm.py b/defog_utils/utils_llm.py
index e0ac2e1..60bae8b 100644
--- 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
index 10894a7..adbb043 100644
--- 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