mirror of
https://github.com/hwchase17/langchain.git
synced 2025-08-16 08:06:14 +00:00
refactor(openai): move chat model to v1 namespace (#32407)
This commit is contained in:
parent
71f0138885
commit
4559997e35
@ -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.embeddings import AzureOpenAIEmbeddings, OpenAIEmbeddings
|
||||||
from langchain_openai.llms import AzureOpenAI, OpenAI
|
from langchain_openai.llms import AzureOpenAI, OpenAI
|
||||||
|
|
||||||
__all__ = [
|
__all__ = [
|
||||||
"OpenAI",
|
"OpenAI",
|
||||||
"ChatOpenAI",
|
"ChatOpenAI",
|
||||||
"ChatOpenAIV1",
|
|
||||||
"OpenAIEmbeddings",
|
"OpenAIEmbeddings",
|
||||||
"AzureOpenAI",
|
"AzureOpenAI",
|
||||||
"AzureChatOpenAI",
|
"AzureChatOpenAI",
|
||||||
|
@ -1,5 +1,4 @@
|
|||||||
from langchain_openai.chat_models.azure import AzureChatOpenAI
|
from langchain_openai.chat_models.azure import AzureChatOpenAI
|
||||||
from langchain_openai.chat_models.base import ChatOpenAI
|
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"]
|
||||||
|
3
libs/partners/openai/langchain_openai/v1/__init__.py
Normal file
3
libs/partners/openai/langchain_openai/v1/__init__.py
Normal file
@ -0,0 +1,3 @@
|
|||||||
|
from langchain_openai.v1.chat_models import ChatOpenAI
|
||||||
|
|
||||||
|
__all__ = ["ChatOpenAI"]
|
@ -0,0 +1,3 @@
|
|||||||
|
from langchain_openai.v1.chat_models.base import ChatOpenAI
|
||||||
|
|
||||||
|
__all__ = ["ChatOpenAI"]
|
@ -20,7 +20,8 @@ from langchain_core.v1.messages import HumanMessage as HumanMessageV1
|
|||||||
from pydantic import BaseModel
|
from pydantic import BaseModel
|
||||||
from typing_extensions import TypedDict
|
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"
|
MODEL_NAME = "gpt-4o-mini"
|
||||||
|
|
||||||
|
@ -1,7 +1,10 @@
|
|||||||
from langchain_openai.chat_models import __all__
|
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:
|
def test_all_imports() -> None:
|
||||||
assert sorted(EXPECTED_ALL) == sorted(__all__)
|
assert sorted(EXPECTED_ALL) == sorted(__all__)
|
||||||
|
assert sorted(EXPECTED_ALL_V1) == sorted(v1_all)
|
||||||
|
@ -37,7 +37,8 @@ from openai.types.responses.response_usage import (
|
|||||||
from openai.types.shared.reasoning import Reasoning
|
from openai.types.shared.reasoning import Reasoning
|
||||||
from openai.types.shared.response_format_text import ResponseFormatText
|
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
|
from tests.unit_tests.chat_models.test_base import MockSyncContextManager
|
||||||
|
|
||||||
responses_stream = [
|
responses_stream = [
|
||||||
|
@ -1,15 +1,18 @@
|
|||||||
from langchain_openai import __all__
|
from langchain_openai import __all__
|
||||||
|
from langchain_openai.v1 import __all__ as v1_all
|
||||||
|
|
||||||
EXPECTED_ALL = [
|
EXPECTED_ALL = [
|
||||||
"OpenAI",
|
"OpenAI",
|
||||||
"ChatOpenAI",
|
"ChatOpenAI",
|
||||||
"ChatOpenAIV1",
|
|
||||||
"OpenAIEmbeddings",
|
"OpenAIEmbeddings",
|
||||||
"AzureOpenAI",
|
"AzureOpenAI",
|
||||||
"AzureChatOpenAI",
|
"AzureChatOpenAI",
|
||||||
"AzureOpenAIEmbeddings",
|
"AzureOpenAIEmbeddings",
|
||||||
]
|
]
|
||||||
|
|
||||||
|
EXPECTED_ALL_V1 = ["ChatOpenAI"]
|
||||||
|
|
||||||
|
|
||||||
def test_all_imports() -> None:
|
def test_all_imports() -> None:
|
||||||
assert sorted(EXPECTED_ALL) == sorted(__all__)
|
assert sorted(EXPECTED_ALL) == sorted(__all__)
|
||||||
|
assert sorted(EXPECTED_ALL_V1) == sorted(v1_all)
|
||||||
|
Loading…
Reference in New Issue
Block a user