kennethreitz/langchain
1
0
Fork 0
mirror of https://github.com/kennethreitz/langchain.git synced 2026-06-05 23:00:18 +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
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
langchain/llms/openai.py
+2 -1
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)