From 5f73c836a6b308e03a5765967d0c7b9bf6817342 Mon Sep 17 00:00:00 2001 From: Philippe PRADOS Date: Wed, 24 Jul 2024 15:14:48 +0200 Subject: [PATCH] openai[small]: Add the new model: gpt-4o-mini (#24594) --- libs/partners/openai/langchain_openai/llms/base.py | 1 + 1 file changed, 1 insertion(+) diff --git a/libs/partners/openai/langchain_openai/llms/base.py b/libs/partners/openai/langchain_openai/llms/base.py index 1f66c1c2f8..e1d0938f01 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,