mirror of
https://github.com/hwchase17/langchain.git
synced 2025-08-15 07:36:08 +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.llms import AzureOpenAI, OpenAI
|
||||
|
||||
__all__ = [
|
||||
"OpenAI",
|
||||
"ChatOpenAI",
|
||||
"ChatOpenAIV1",
|
||||
"OpenAIEmbeddings",
|
||||
"AzureOpenAI",
|
||||
"AzureChatOpenAI",
|
||||
|
@ -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"]
|
||||
|
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 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"
|
||||
|
||||
|
@ -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)
|
||||
|
@ -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 = [
|
||||
|
@ -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)
|
||||
|
Loading…
Reference in New Issue
Block a user