github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-08-01 09:04:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

GPT-4 support (#1778)

Browse Source
This commit is contained in:
Nils Durner 2023-03-19 17:29:44 +01:00 committed by GitHub
parent 3d6fcb85dc
commit 3161ced4bc
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
langchain/llms/openai.py
View File

@ -163,7 +163,8 @@ class BaseOpenAI(BaseLLM, BaseModel):
def __new__(cls, **data: Any) -> Union[OpenAIChat, BaseOpenAI]: # type: ignore
"""Initialize the OpenAI object."""
if data.get("model_name", "").startswith("gpt-3.5-turbo"):
model_name = data.get("model_name", "")
if model_name.startswith("gpt-3.5-turbo") or model_name.startswith("gpt-4"):
return OpenAIChat(**data)
return super().__new__(cls)