diff --git a/langchain/chat_models/openai.py b/langchain/chat_models/openai.py index 6fe9cc90..6c18b9c8 100644 --- a/langchain/chat_models/openai.py +++ b/langchain/chat_models/openai.py @@ -466,12 +466,12 @@ class ChatOpenAI(BaseChatModel): if sys.version_info[1] <= 7: return super().get_num_tokens_from_messages(messages) model, encoding = self._get_encoding_model() - if model == "gpt-3.5-turbo-0301": + if model.startswith("gpt-3.5-turbo"): # every message follows {role/name}\n{content}\n tokens_per_message = 4 # if there's a name, the role is omitted tokens_per_name = -1 - elif model == "gpt-4-0314": + elif model.startswith("gpt-4"): tokens_per_message = 3 tokens_per_name = 1 else: