diff --git a/libs/community/langchain_community/callbacks/openai_info.py b/libs/community/langchain_community/callbacks/openai_info.py index 5dee1e22ba3..42399eb5899 100644 --- a/libs/community/langchain_community/callbacks/openai_info.py +++ b/libs/community/langchain_community/callbacks/openai_info.py @@ -10,6 +10,22 @@ from langchain_core.messages import AIMessage from langchain_core.outputs import ChatGeneration, LLMResult MODEL_COST_PER_1K_TOKENS = { + # OpenAI o1 input + "o1": 0.015, + "o1-2024-12-17": 0.015, + "o1-cached": 0.0075, + "o1-2024-12-17-cached": 0.0075, + # OpenAI o1 output + "o1-completion": 0.06, + "o1-2024-12-17-completion": 0.06, + # OpenAI o3-mini input + "o3-mini": 0.0011, + "o3-mini-2025-01-31": 0.0011, + "o3-mini-cached": 0.00055, + "o3-mini-2025-01-31-cached": 0.00055, + # OpenAI o3-mini output + "o3-mini-completion": 0.0044, + "o3-mini-2025-01-31-completion": 0.0044, # OpenAI o1-preview input "o1-preview": 0.015, "o1-preview-cached": 0.0075,