From 579d14fbc136bd5abf1e343915846358c987911f Mon Sep 17 00:00:00 2001 From: Jacob Lee Date: Mon, 18 Sep 2023 15:55:13 -0700 Subject: [PATCH] Allow 3.5-turbo instruct models in the OpenAI LLM class (#10750) @baskaryan @hwchase17 --- libs/langchain/langchain/llms/openai.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/libs/langchain/langchain/llms/openai.py b/libs/langchain/langchain/llms/openai.py index 837256ec99..33b312a5da 100644 --- a/libs/langchain/langchain/llms/openai.py +++ b/libs/langchain/langchain/llms/openai.py @@ -196,7 +196,9 @@ class BaseOpenAI(BaseLLM): def __new__(cls, **data: Any) -> Union[OpenAIChat, BaseOpenAI]: # type: ignore """Initialize the OpenAI object.""" model_name = data.get("model_name", "") - if model_name.startswith("gpt-3.5-turbo") or model_name.startswith("gpt-4"): + if ( + model_name.startswith("gpt-3.5-turbo") or model_name.startswith("gpt-4") + ) and not model_name.endswith("-instruct"): warnings.warn( "You are trying to use a chat model. This way of initializing it is " "no longer supported. Instead, please use: "