From f005988e3181b15771d06a27478236753f22c0b1 Mon Sep 17 00:00:00 2001 From: ccurme Date: Sun, 13 Apr 2025 11:20:46 -0400 Subject: [PATCH] community[patch]: fix cost calculations for o3 in OpenAI callback (#30807) Resolves https://github.com/langchain-ai/langchain/issues/30795 --- .../community/langchain_community/callbacks/openai_info.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/libs/community/langchain_community/callbacks/openai_info.py b/libs/community/langchain_community/callbacks/openai_info.py index 42399eb5899..fbf3011e67a 100644 --- a/libs/community/langchain_community/callbacks/openai_info.py +++ b/libs/community/langchain_community/callbacks/openai_info.py @@ -218,12 +218,17 @@ def standardize_model_name( or model_name.startswith("gpt-3.5") or model_name.startswith("gpt-35") or model_name.startswith("o1-") + or model_name.startswith("o3-") or ("finetuned" in model_name and "legacy" not in model_name) ): return model_name + "-completion" if ( token_type == TokenType.PROMPT_CACHED - and (model_name.startswith("gpt-4o") or model_name.startswith("o1")) + and ( + model_name.startswith("gpt-4o") + or model_name.startswith("o1") + or model_name.startswith("o3") + ) and not (model_name.startswith("gpt-4o-2024-05-13")) ): return model_name + "-cached"