refactor(openai): move chat model to v1 namespace (#32407)

This commit is contained in:
ccurme 2025-08-05 14:23:29 -03:00 committed by GitHub
parent 71f0138885
commit 4559997e35
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
9 changed files with 20 additions and 8 deletions

View File

@ -1,11 +1,10 @@
from langchain_openai.chat_models import AzureChatOpenAI, ChatOpenAI, ChatOpenAIV1
from langchain_openai.chat_models import AzureChatOpenAI, ChatOpenAI
from langchain_openai.embeddings import AzureOpenAIEmbeddings, OpenAIEmbeddings
from langchain_openai.llms import AzureOpenAI, OpenAI
__all__ = [
"OpenAI",
"ChatOpenAI",
"ChatOpenAIV1",
"OpenAIEmbeddings",
"AzureOpenAI",
"AzureChatOpenAI",

View File

@ -1,5 +1,4 @@
from langchain_openai.chat_models.azure import AzureChatOpenAI
from langchain_openai.chat_models.base import ChatOpenAI
from langchain_openai.chat_models.base_v1 import ChatOpenAI as ChatOpenAIV1
__all__ = ["ChatOpenAI", "AzureChatOpenAI", "ChatOpenAIV1"]
__all__ = ["ChatOpenAI", "AzureChatOpenAI"]

View File

@ -0,0 +1,3 @@
from langchain_openai.v1.chat_models import ChatOpenAI
__all__ = ["ChatOpenAI"]

View File

@ -0,0 +1,3 @@
from langchain_openai.v1.chat_models.base import ChatOpenAI
__all__ = ["ChatOpenAI"]

View File

@ -20,7 +20,8 @@ from langchain_core.v1.messages import HumanMessage as HumanMessageV1
from pydantic import BaseModel
from typing_extensions import TypedDict
from langchain_openai import ChatOpenAI, ChatOpenAIV1
from langchain_openai import ChatOpenAI
from langchain_openai.v1 import ChatOpenAI as ChatOpenAIV1
MODEL_NAME = "gpt-4o-mini"

View File

@ -1,7 +1,10 @@
from langchain_openai.chat_models import __all__
from langchain_openai.v1.chat_models import __all__ as v1_all
EXPECTED_ALL = ["ChatOpenAI", "ChatOpenAIV1", "AzureChatOpenAI"]
EXPECTED_ALL = ["ChatOpenAI", "AzureChatOpenAI"]
EXPECTED_ALL_V1 = ["ChatOpenAI"]
def test_all_imports() -> None:
assert sorted(EXPECTED_ALL) == sorted(__all__)
assert sorted(EXPECTED_ALL_V1) == sorted(v1_all)

View File

@ -37,7 +37,8 @@ from openai.types.responses.response_usage import (
from openai.types.shared.reasoning import Reasoning
from openai.types.shared.response_format_text import ResponseFormatText
from langchain_openai import ChatOpenAI, ChatOpenAIV1
from langchain_openai import ChatOpenAI
from langchain_openai.v1 import ChatOpenAI as ChatOpenAIV1
from tests.unit_tests.chat_models.test_base import MockSyncContextManager
responses_stream = [

View File

@ -1,15 +1,18 @@
from langchain_openai import __all__
from langchain_openai.v1 import __all__ as v1_all
EXPECTED_ALL = [
"OpenAI",
"ChatOpenAI",
"ChatOpenAIV1",
"OpenAIEmbeddings",
"AzureOpenAI",
"AzureChatOpenAI",
"AzureOpenAIEmbeddings",
]
EXPECTED_ALL_V1 = ["ChatOpenAI"]
def test_all_imports() -> None:
assert sorted(EXPECTED_ALL) == sorted(__all__)
assert sorted(EXPECTED_ALL_V1) == sorted(v1_all)