diff --git a/libs/partners/openai/langchain_openai/llms/base.py b/libs/partners/openai/langchain_openai/llms/base.py index 1f66c1c2f82..e1d0938f018 100644 --- a/libs/partners/openai/langchain_openai/llms/base.py +++ b/libs/partners/openai/langchain_openai/llms/base.py @@ -537,6 +537,7 @@ class BaseOpenAI(BaseLLM): max_tokens = openai.modelname_to_contextsize("gpt-3.5-turbo-instruct") """ model_token_mapping = { + "gpt-4o-mini": 128_000, "gpt-4o": 128_000, "gpt-4o-2024-05-13": 128_000, "gpt-4": 8192,