add o1 costs

defog-ai · Dec 18, 2024 · f032c99 · f032c99
1 parent 60e00b0
commit f032c99
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/defog_utils/utils_llm.py b/defog_utils/utils_llm.py
@@ -6,6 +6,7 @@
 LLM_COSTS_PER_TOKEN = {
     "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},
     "o1-preview": {"input_cost_per1k": 0.015, "output_cost_per1k": 0.06},
     "o1-mini": {"input_cost_per1k": 0.003, "output_cost_per1k": 0.012},
     "gpt-4-turbo": {"input_cost_per1k": 0.01, "output_cost_per1k": 0.03},