blob42/langchain
1
0
Fork 0
forked from Archives/langchain
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Harrison/add warning to openaichat (#1781)

Browse Source
This commit is contained in:
Harrison Chase 2023-03-19 09:43:56 -07:00 committed by GitHub
parent 3161ced4bc
commit 779790167e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
langchain/llms/openai.py
View File

@ -3,6 +3,7 @@ from __future__ import annotations
import logging import logging
import sys import sys
import warnings
from typing import ( from typing import (
Any, Any,
Callable, Callable,
@ -165,6 +166,11 @@ class BaseOpenAI(BaseLLM, BaseModel):
"""Initialize the OpenAI object.""" """Initialize the OpenAI object."""
model_name = data.get("model_name", "") 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"):
warnings.warn(
"You are trying to use a chat model. This way of initializing it is "
"no longer supported. Instead, please use: "
"`from langchain.chat_models import ChatOpenAI`"
)
return OpenAIChat(**data) return OpenAIChat(**data)
return super().__new__(cls) return super().__new__(cls)
@ -599,6 +605,11 @@ class OpenAIChat(BaseLLM, BaseModel):
"due to an old version of the openai package. Try upgrading it " "due to an old version of the openai package. Try upgrading it "
"with `pip install --upgrade openai`." "with `pip install --upgrade openai`."
) )
warnings.warn(
"You are trying to use a chat model. This way of initializing it is "
"no longer supported. Instead, please use: "
"`from langchain.chat_models import ChatOpenAI`"
)
return values return values
@property @property