mirror of
https://github.com/hwchase17/langchain.git
synced 2025-08-14 15:16:21 +00:00
fix(core): refactor new types into top-level v1 namespace (#32403)
This commit is contained in:
parent
deae8cc164
commit
c36b123c8c
@ -7,7 +7,7 @@ from typing import TYPE_CHECKING, Any, Optional, Union
|
|||||||
|
|
||||||
from typing_extensions import Self
|
from typing_extensions import Self
|
||||||
|
|
||||||
from langchain_core.messages.v1 import AIMessage, AIMessageChunk, MessageV1
|
from langchain_core.v1.messages import AIMessage, AIMessageChunk, MessageV1
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from collections.abc import Sequence
|
from collections.abc import Sequence
|
||||||
|
@ -38,15 +38,15 @@ from langchain_core.callbacks.base import (
|
|||||||
from langchain_core.callbacks.stdout import StdOutCallbackHandler
|
from langchain_core.callbacks.stdout import StdOutCallbackHandler
|
||||||
from langchain_core.messages import BaseMessage, get_buffer_string
|
from langchain_core.messages import BaseMessage, get_buffer_string
|
||||||
from langchain_core.messages.utils import convert_from_v1_message
|
from langchain_core.messages.utils import convert_from_v1_message
|
||||||
from langchain_core.messages.v1 import (
|
from langchain_core.outputs import ChatGeneration, ChatGenerationChunk, LLMResult
|
||||||
|
from langchain_core.tracers.schemas import Run
|
||||||
|
from langchain_core.utils.env import env_var_is_set
|
||||||
|
from langchain_core.v1.messages import (
|
||||||
AIMessage,
|
AIMessage,
|
||||||
AIMessageChunk,
|
AIMessageChunk,
|
||||||
MessageV1,
|
MessageV1,
|
||||||
MessageV1Types,
|
MessageV1Types,
|
||||||
)
|
)
|
||||||
from langchain_core.outputs import ChatGeneration, ChatGenerationChunk, LLMResult
|
|
||||||
from langchain_core.tracers.schemas import Run
|
|
||||||
from langchain_core.utils.env import env_var_is_set
|
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from collections.abc import AsyncGenerator, Coroutine, Generator, Sequence
|
from collections.abc import AsyncGenerator, Coroutine, Generator, Sequence
|
||||||
|
@ -12,8 +12,8 @@ from langchain_core.callbacks.base import BaseCallbackHandler
|
|||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from langchain_core.agents import AgentAction, AgentFinish
|
from langchain_core.agents import AgentAction, AgentFinish
|
||||||
from langchain_core.messages import BaseMessage
|
from langchain_core.messages import BaseMessage
|
||||||
from langchain_core.messages.v1 import AIMessage, MessageV1
|
|
||||||
from langchain_core.outputs import LLMResult
|
from langchain_core.outputs import LLMResult
|
||||||
|
from langchain_core.v1.messages import AIMessage, MessageV1
|
||||||
|
|
||||||
|
|
||||||
class StreamingStdOutCallbackHandler(BaseCallbackHandler):
|
class StreamingStdOutCallbackHandler(BaseCallbackHandler):
|
||||||
|
@ -12,8 +12,8 @@ from langchain_core.callbacks import BaseCallbackHandler
|
|||||||
from langchain_core.messages import AIMessage
|
from langchain_core.messages import AIMessage
|
||||||
from langchain_core.messages.ai import UsageMetadata, add_usage
|
from langchain_core.messages.ai import UsageMetadata, add_usage
|
||||||
from langchain_core.messages.utils import convert_from_v1_message
|
from langchain_core.messages.utils import convert_from_v1_message
|
||||||
from langchain_core.messages.v1 import AIMessage as AIMessageV1
|
|
||||||
from langchain_core.outputs import ChatGeneration, LLMResult
|
from langchain_core.outputs import ChatGeneration, LLMResult
|
||||||
|
from langchain_core.v1.messages import AIMessage as AIMessageV1
|
||||||
|
|
||||||
|
|
||||||
class UsageMetadataCallbackHandler(BaseCallbackHandler):
|
class UsageMetadataCallbackHandler(BaseCallbackHandler):
|
||||||
|
@ -4,7 +4,7 @@ from collections.abc import Sequence
|
|||||||
from typing import Optional
|
from typing import Optional
|
||||||
|
|
||||||
from langchain_core.messages import BaseMessage
|
from langchain_core.messages import BaseMessage
|
||||||
from langchain_core.messages.v1 import MessageV1
|
from langchain_core.v1.messages import MessageV1
|
||||||
|
|
||||||
|
|
||||||
def _is_openai_data_block(block: dict) -> bool:
|
def _is_openai_data_block(block: dict) -> bool:
|
||||||
|
@ -28,10 +28,10 @@ from langchain_core.messages import (
|
|||||||
MessageLikeRepresentation,
|
MessageLikeRepresentation,
|
||||||
get_buffer_string,
|
get_buffer_string,
|
||||||
)
|
)
|
||||||
from langchain_core.messages.v1 import AIMessage as AIMessageV1
|
|
||||||
from langchain_core.prompt_values import PromptValue
|
from langchain_core.prompt_values import PromptValue
|
||||||
from langchain_core.runnables import Runnable, RunnableSerializable
|
from langchain_core.runnables import Runnable, RunnableSerializable
|
||||||
from langchain_core.utils import get_pydantic_field_names
|
from langchain_core.utils import get_pydantic_field_names
|
||||||
|
from langchain_core.v1.messages import AIMessage as AIMessageV1
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from langchain_core.outputs import LLMResult
|
from langchain_core.outputs import LLMResult
|
||||||
|
@ -13,13 +13,13 @@ from langchain_core.callbacks import (
|
|||||||
CallbackManagerForLLMRun,
|
CallbackManagerForLLMRun,
|
||||||
)
|
)
|
||||||
from langchain_core.language_models.chat_models import BaseChatModel, SimpleChatModel
|
from langchain_core.language_models.chat_models import BaseChatModel, SimpleChatModel
|
||||||
from langchain_core.language_models.v1.chat_models import BaseChatModelV1
|
|
||||||
from langchain_core.messages import AIMessage, AIMessageChunk, BaseMessage
|
from langchain_core.messages import AIMessage, AIMessageChunk, BaseMessage
|
||||||
from langchain_core.messages.v1 import AIMessage as AIMessageV1
|
|
||||||
from langchain_core.messages.v1 import AIMessageChunk as AIMessageChunkV1
|
|
||||||
from langchain_core.messages.v1 import MessageV1
|
|
||||||
from langchain_core.outputs import ChatGeneration, ChatGenerationChunk, ChatResult
|
from langchain_core.outputs import ChatGeneration, ChatGenerationChunk, ChatResult
|
||||||
from langchain_core.runnables import RunnableConfig
|
from langchain_core.runnables import RunnableConfig
|
||||||
|
from langchain_core.v1.chat_models import BaseChatModelV1
|
||||||
|
from langchain_core.v1.messages import AIMessage as AIMessageV1
|
||||||
|
from langchain_core.v1.messages import AIMessageChunk as AIMessageChunkV1
|
||||||
|
from langchain_core.v1.messages import MessageV1
|
||||||
|
|
||||||
|
|
||||||
class FakeMessagesListChatModel(BaseChatModel):
|
class FakeMessagesListChatModel(BaseChatModel):
|
||||||
|
@ -1 +0,0 @@
|
|||||||
"""LangChain v1.0 chat models."""
|
|
@ -40,12 +40,12 @@ from langchain_core.messages.human import HumanMessage, HumanMessageChunk
|
|||||||
from langchain_core.messages.modifier import RemoveMessage
|
from langchain_core.messages.modifier import RemoveMessage
|
||||||
from langchain_core.messages.system import SystemMessage, SystemMessageChunk
|
from langchain_core.messages.system import SystemMessage, SystemMessageChunk
|
||||||
from langchain_core.messages.tool import ToolCall, ToolMessage, ToolMessageChunk
|
from langchain_core.messages.tool import ToolCall, ToolMessage, ToolMessageChunk
|
||||||
from langchain_core.messages.v1 import AIMessage as AIMessageV1
|
from langchain_core.v1.messages import AIMessage as AIMessageV1
|
||||||
from langchain_core.messages.v1 import AIMessageChunk as AIMessageChunkV1
|
from langchain_core.v1.messages import AIMessageChunk as AIMessageChunkV1
|
||||||
from langchain_core.messages.v1 import HumanMessage as HumanMessageV1
|
from langchain_core.v1.messages import HumanMessage as HumanMessageV1
|
||||||
from langchain_core.messages.v1 import MessageV1, MessageV1Types
|
from langchain_core.v1.messages import MessageV1, MessageV1Types
|
||||||
from langchain_core.messages.v1 import SystemMessage as SystemMessageV1
|
from langchain_core.v1.messages import SystemMessage as SystemMessageV1
|
||||||
from langchain_core.messages.v1 import ToolMessage as ToolMessageV1
|
from langchain_core.v1.messages import ToolMessage as ToolMessageV1
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from langchain_text_splitters import TextSplitter
|
from langchain_text_splitters import TextSplitter
|
||||||
|
@ -18,10 +18,10 @@ from typing_extensions import override
|
|||||||
|
|
||||||
from langchain_core.language_models import LanguageModelOutput
|
from langchain_core.language_models import LanguageModelOutput
|
||||||
from langchain_core.messages import AnyMessage, BaseMessage
|
from langchain_core.messages import AnyMessage, BaseMessage
|
||||||
from langchain_core.messages.v1 import AIMessage, MessageV1, MessageV1Types
|
|
||||||
from langchain_core.outputs import ChatGeneration, Generation
|
from langchain_core.outputs import ChatGeneration, Generation
|
||||||
from langchain_core.runnables import Runnable, RunnableConfig, RunnableSerializable
|
from langchain_core.runnables import Runnable, RunnableConfig, RunnableSerializable
|
||||||
from langchain_core.runnables.config import run_in_executor
|
from langchain_core.runnables.config import run_in_executor
|
||||||
|
from langchain_core.v1.messages import AIMessage, MessageV1, MessageV1Types
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from langchain_core.prompt_values import PromptValue
|
from langchain_core.prompt_values import PromptValue
|
||||||
|
@ -13,7 +13,6 @@ from pydantic.v1 import BaseModel
|
|||||||
from typing_extensions import override
|
from typing_extensions import override
|
||||||
|
|
||||||
from langchain_core.exceptions import OutputParserException
|
from langchain_core.exceptions import OutputParserException
|
||||||
from langchain_core.messages.v1 import AIMessage
|
|
||||||
from langchain_core.output_parsers.format_instructions import JSON_FORMAT_INSTRUCTIONS
|
from langchain_core.output_parsers.format_instructions import JSON_FORMAT_INSTRUCTIONS
|
||||||
from langchain_core.output_parsers.transform import BaseCumulativeTransformOutputParser
|
from langchain_core.output_parsers.transform import BaseCumulativeTransformOutputParser
|
||||||
from langchain_core.outputs import Generation
|
from langchain_core.outputs import Generation
|
||||||
@ -22,6 +21,7 @@ from langchain_core.utils.json import (
|
|||||||
parse_json_markdown,
|
parse_json_markdown,
|
||||||
parse_partial_json,
|
parse_partial_json,
|
||||||
)
|
)
|
||||||
|
from langchain_core.v1.messages import AIMessage
|
||||||
|
|
||||||
# Union type needs to be last assignment to PydanticBaseModel to make mypy happy.
|
# Union type needs to be last assignment to PydanticBaseModel to make mypy happy.
|
||||||
PydanticBaseModel = Union[BaseModel, pydantic.BaseModel]
|
PydanticBaseModel = Union[BaseModel, pydantic.BaseModel]
|
||||||
|
@ -12,8 +12,8 @@ from typing import TYPE_CHECKING, TypeVar, Union
|
|||||||
from typing_extensions import override
|
from typing_extensions import override
|
||||||
|
|
||||||
from langchain_core.messages import BaseMessage
|
from langchain_core.messages import BaseMessage
|
||||||
from langchain_core.messages.v1 import AIMessage
|
|
||||||
from langchain_core.output_parsers.transform import BaseTransformOutputParser
|
from langchain_core.output_parsers.transform import BaseTransformOutputParser
|
||||||
|
from langchain_core.v1.messages import AIMessage
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from collections.abc import AsyncIterator, Iterator
|
from collections.abc import AsyncIterator, Iterator
|
||||||
|
@ -11,13 +11,13 @@ from pydantic.v1 import BaseModel as BaseModelV1
|
|||||||
from typing_extensions import override
|
from typing_extensions import override
|
||||||
|
|
||||||
from langchain_core.exceptions import OutputParserException
|
from langchain_core.exceptions import OutputParserException
|
||||||
from langchain_core.messages.v1 import AIMessage
|
|
||||||
from langchain_core.output_parsers import (
|
from langchain_core.output_parsers import (
|
||||||
BaseCumulativeTransformOutputParser,
|
BaseCumulativeTransformOutputParser,
|
||||||
BaseGenerationOutputParser,
|
BaseGenerationOutputParser,
|
||||||
)
|
)
|
||||||
from langchain_core.output_parsers.json import parse_partial_json
|
from langchain_core.output_parsers.json import parse_partial_json
|
||||||
from langchain_core.outputs import ChatGeneration, Generation
|
from langchain_core.outputs import ChatGeneration, Generation
|
||||||
|
from langchain_core.v1.messages import AIMessage
|
||||||
|
|
||||||
|
|
||||||
class OutputFunctionsParser(BaseGenerationOutputParser[Any]):
|
class OutputFunctionsParser(BaseGenerationOutputParser[Any]):
|
||||||
|
@ -12,11 +12,11 @@ from langchain_core.exceptions import OutputParserException
|
|||||||
from langchain_core.messages import AIMessage, InvalidToolCall
|
from langchain_core.messages import AIMessage, InvalidToolCall
|
||||||
from langchain_core.messages.tool import invalid_tool_call
|
from langchain_core.messages.tool import invalid_tool_call
|
||||||
from langchain_core.messages.tool import tool_call as create_tool_call
|
from langchain_core.messages.tool import tool_call as create_tool_call
|
||||||
from langchain_core.messages.v1 import AIMessage as AIMessageV1
|
|
||||||
from langchain_core.output_parsers.transform import BaseCumulativeTransformOutputParser
|
from langchain_core.output_parsers.transform import BaseCumulativeTransformOutputParser
|
||||||
from langchain_core.outputs import ChatGeneration, Generation
|
from langchain_core.outputs import ChatGeneration, Generation
|
||||||
from langchain_core.utils.json import parse_partial_json
|
from langchain_core.utils.json import parse_partial_json
|
||||||
from langchain_core.utils.pydantic import TypeBaseModel
|
from langchain_core.utils.pydantic import TypeBaseModel
|
||||||
|
from langchain_core.v1.messages import AIMessage as AIMessageV1
|
||||||
|
|
||||||
logger = logging.getLogger(__name__)
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
@ -8,13 +8,13 @@ from pydantic import SkipValidation
|
|||||||
from typing_extensions import override
|
from typing_extensions import override
|
||||||
|
|
||||||
from langchain_core.exceptions import OutputParserException
|
from langchain_core.exceptions import OutputParserException
|
||||||
from langchain_core.messages.v1 import AIMessage
|
|
||||||
from langchain_core.output_parsers import JsonOutputParser
|
from langchain_core.output_parsers import JsonOutputParser
|
||||||
from langchain_core.outputs import Generation
|
from langchain_core.outputs import Generation
|
||||||
from langchain_core.utils.pydantic import (
|
from langchain_core.utils.pydantic import (
|
||||||
PydanticBaseModel,
|
PydanticBaseModel,
|
||||||
TBaseModel,
|
TBaseModel,
|
||||||
)
|
)
|
||||||
|
from langchain_core.v1.messages import AIMessage
|
||||||
|
|
||||||
|
|
||||||
class PydanticOutputParser(JsonOutputParser, Generic[TBaseModel]):
|
class PydanticOutputParser(JsonOutputParser, Generic[TBaseModel]):
|
||||||
|
@ -12,7 +12,6 @@ from typing import (
|
|||||||
from typing_extensions import override
|
from typing_extensions import override
|
||||||
|
|
||||||
from langchain_core.messages import BaseMessage, BaseMessageChunk
|
from langchain_core.messages import BaseMessage, BaseMessageChunk
|
||||||
from langchain_core.messages.v1 import AIMessage, AIMessageChunk
|
|
||||||
from langchain_core.output_parsers.base import BaseOutputParser, T
|
from langchain_core.output_parsers.base import BaseOutputParser, T
|
||||||
from langchain_core.outputs import (
|
from langchain_core.outputs import (
|
||||||
ChatGeneration,
|
ChatGeneration,
|
||||||
@ -21,6 +20,7 @@ from langchain_core.outputs import (
|
|||||||
GenerationChunk,
|
GenerationChunk,
|
||||||
)
|
)
|
||||||
from langchain_core.runnables.config import run_in_executor
|
from langchain_core.runnables.config import run_in_executor
|
||||||
|
from langchain_core.v1.messages import AIMessage, AIMessageChunk
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from collections.abc import AsyncIterator, Iterator
|
from collections.abc import AsyncIterator, Iterator
|
||||||
|
@ -13,9 +13,9 @@ from typing_extensions import override
|
|||||||
from langchain_core.exceptions import OutputParserException
|
from langchain_core.exceptions import OutputParserException
|
||||||
from langchain_core.messages import BaseMessage
|
from langchain_core.messages import BaseMessage
|
||||||
from langchain_core.messages.utils import convert_from_v1_message
|
from langchain_core.messages.utils import convert_from_v1_message
|
||||||
from langchain_core.messages.v1 import AIMessage
|
|
||||||
from langchain_core.output_parsers.transform import BaseTransformOutputParser
|
from langchain_core.output_parsers.transform import BaseTransformOutputParser
|
||||||
from langchain_core.runnables.utils import AddableDict
|
from langchain_core.runnables.utils import AddableDict
|
||||||
|
from langchain_core.v1.messages import AIMessage
|
||||||
|
|
||||||
XML_FORMAT_INSTRUCTIONS = """The output should be formatted as a XML file.
|
XML_FORMAT_INSTRUCTIONS = """The output should be formatted as a XML file.
|
||||||
1. Output should conform to the tags below.
|
1. Output should conform to the tags below.
|
||||||
|
@ -23,11 +23,11 @@ from langchain_core.messages import (
|
|||||||
get_buffer_string,
|
get_buffer_string,
|
||||||
)
|
)
|
||||||
from langchain_core.messages import content_blocks as types
|
from langchain_core.messages import content_blocks as types
|
||||||
from langchain_core.messages.v1 import AIMessage as AIMessageV1
|
from langchain_core.v1.messages import AIMessage as AIMessageV1
|
||||||
from langchain_core.messages.v1 import HumanMessage as HumanMessageV1
|
from langchain_core.v1.messages import HumanMessage as HumanMessageV1
|
||||||
from langchain_core.messages.v1 import MessageV1, ResponseMetadata
|
from langchain_core.v1.messages import MessageV1, ResponseMetadata
|
||||||
from langchain_core.messages.v1 import SystemMessage as SystemMessageV1
|
from langchain_core.v1.messages import SystemMessage as SystemMessageV1
|
||||||
from langchain_core.messages.v1 import ToolMessage as ToolMessageV1
|
from langchain_core.v1.messages import ToolMessage as ToolMessageV1
|
||||||
|
|
||||||
|
|
||||||
def _convert_to_v1(message: BaseMessage) -> MessageV1:
|
def _convert_to_v1(message: BaseMessage) -> MessageV1:
|
||||||
|
@ -47,7 +47,6 @@ from langchain_core.callbacks import (
|
|||||||
Callbacks,
|
Callbacks,
|
||||||
)
|
)
|
||||||
from langchain_core.messages.tool import ToolCall, ToolMessage, ToolOutputMixin
|
from langchain_core.messages.tool import ToolCall, ToolMessage, ToolOutputMixin
|
||||||
from langchain_core.messages.v1 import ToolMessage as ToolMessageV1
|
|
||||||
from langchain_core.runnables import (
|
from langchain_core.runnables import (
|
||||||
RunnableConfig,
|
RunnableConfig,
|
||||||
RunnableSerializable,
|
RunnableSerializable,
|
||||||
@ -69,6 +68,7 @@ from langchain_core.utils.pydantic import (
|
|||||||
is_pydantic_v1_subclass,
|
is_pydantic_v1_subclass,
|
||||||
is_pydantic_v2_subclass,
|
is_pydantic_v2_subclass,
|
||||||
)
|
)
|
||||||
|
from langchain_core.v1.messages import ToolMessage as ToolMessageV1
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
import uuid
|
import uuid
|
||||||
|
@ -16,8 +16,8 @@ from typing_extensions import override
|
|||||||
|
|
||||||
from langchain_core.callbacks.base import AsyncCallbackHandler, BaseCallbackHandler
|
from langchain_core.callbacks.base import AsyncCallbackHandler, BaseCallbackHandler
|
||||||
from langchain_core.exceptions import TracerException # noqa: F401
|
from langchain_core.exceptions import TracerException # noqa: F401
|
||||||
from langchain_core.messages.v1 import AIMessage, AIMessageChunk, MessageV1
|
|
||||||
from langchain_core.tracers.core import _TracerCore
|
from langchain_core.tracers.core import _TracerCore
|
||||||
|
from langchain_core.v1.messages import AIMessage, AIMessageChunk, MessageV1
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from collections.abc import Sequence
|
from collections.abc import Sequence
|
||||||
|
@ -19,12 +19,6 @@ from typing import (
|
|||||||
from langchain_core.exceptions import TracerException
|
from langchain_core.exceptions import TracerException
|
||||||
from langchain_core.load import dumpd
|
from langchain_core.load import dumpd
|
||||||
from langchain_core.messages.utils import convert_from_v1_message
|
from langchain_core.messages.utils import convert_from_v1_message
|
||||||
from langchain_core.messages.v1 import (
|
|
||||||
AIMessage,
|
|
||||||
AIMessageChunk,
|
|
||||||
MessageV1,
|
|
||||||
MessageV1Types,
|
|
||||||
)
|
|
||||||
from langchain_core.outputs import (
|
from langchain_core.outputs import (
|
||||||
ChatGeneration,
|
ChatGeneration,
|
||||||
ChatGenerationChunk,
|
ChatGenerationChunk,
|
||||||
@ -32,6 +26,12 @@ from langchain_core.outputs import (
|
|||||||
LLMResult,
|
LLMResult,
|
||||||
)
|
)
|
||||||
from langchain_core.tracers.schemas import Run
|
from langchain_core.tracers.schemas import Run
|
||||||
|
from langchain_core.v1.messages import (
|
||||||
|
AIMessage,
|
||||||
|
AIMessageChunk,
|
||||||
|
MessageV1,
|
||||||
|
MessageV1Types,
|
||||||
|
)
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from collections.abc import Coroutine, Sequence
|
from collections.abc import Coroutine, Sequence
|
||||||
|
@ -19,7 +19,6 @@ from typing_extensions import NotRequired, TypedDict, override
|
|||||||
|
|
||||||
from langchain_core.callbacks.base import AsyncCallbackHandler
|
from langchain_core.callbacks.base import AsyncCallbackHandler
|
||||||
from langchain_core.messages import AIMessageChunk, BaseMessage, BaseMessageChunk
|
from langchain_core.messages import AIMessageChunk, BaseMessage, BaseMessageChunk
|
||||||
from langchain_core.messages.v1 import MessageV1
|
|
||||||
from langchain_core.outputs import (
|
from langchain_core.outputs import (
|
||||||
ChatGenerationChunk,
|
ChatGenerationChunk,
|
||||||
GenerationChunk,
|
GenerationChunk,
|
||||||
@ -39,15 +38,16 @@ from langchain_core.runnables.utils import (
|
|||||||
from langchain_core.tracers._streaming import _StreamingCallbackHandler
|
from langchain_core.tracers._streaming import _StreamingCallbackHandler
|
||||||
from langchain_core.tracers.memory_stream import _MemoryStream
|
from langchain_core.tracers.memory_stream import _MemoryStream
|
||||||
from langchain_core.utils.aiter import aclosing, py_anext
|
from langchain_core.utils.aiter import aclosing, py_anext
|
||||||
|
from langchain_core.v1.messages import MessageV1
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from collections.abc import AsyncIterator, Iterator, Sequence
|
from collections.abc import AsyncIterator, Iterator, Sequence
|
||||||
|
|
||||||
from langchain_core.documents import Document
|
from langchain_core.documents import Document
|
||||||
from langchain_core.messages.v1 import AIMessage as AIMessageV1
|
|
||||||
from langchain_core.messages.v1 import AIMessageChunk as AIMessageChunkV1
|
|
||||||
from langchain_core.runnables import Runnable, RunnableConfig
|
from langchain_core.runnables import Runnable, RunnableConfig
|
||||||
from langchain_core.tracers.log_stream import LogEntry
|
from langchain_core.tracers.log_stream import LogEntry
|
||||||
|
from langchain_core.v1.messages import AIMessage as AIMessageV1
|
||||||
|
from langchain_core.v1.messages import AIMessageChunk as AIMessageChunkV1
|
||||||
|
|
||||||
logger = logging.getLogger(__name__)
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
@ -22,14 +22,14 @@ from typing_extensions import override
|
|||||||
from langchain_core.env import get_runtime_environment
|
from langchain_core.env import get_runtime_environment
|
||||||
from langchain_core.load import dumpd
|
from langchain_core.load import dumpd
|
||||||
from langchain_core.messages.utils import convert_from_v1_message
|
from langchain_core.messages.utils import convert_from_v1_message
|
||||||
from langchain_core.messages.v1 import MessageV1Types
|
|
||||||
from langchain_core.tracers.base import BaseTracer
|
from langchain_core.tracers.base import BaseTracer
|
||||||
from langchain_core.tracers.schemas import Run
|
from langchain_core.tracers.schemas import Run
|
||||||
|
from langchain_core.v1.messages import MessageV1Types
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from langchain_core.messages import BaseMessage
|
from langchain_core.messages import BaseMessage
|
||||||
from langchain_core.messages.v1 import AIMessageChunk, MessageV1
|
|
||||||
from langchain_core.outputs import ChatGenerationChunk, GenerationChunk
|
from langchain_core.outputs import ChatGenerationChunk, GenerationChunk
|
||||||
|
from langchain_core.v1.messages import AIMessageChunk, MessageV1
|
||||||
|
|
||||||
logger = logging.getLogger(__name__)
|
logger = logging.getLogger(__name__)
|
||||||
_LOGGED = set()
|
_LOGGED = set()
|
||||||
|
@ -32,9 +32,9 @@ if TYPE_CHECKING:
|
|||||||
from collections.abc import AsyncIterator, Iterator, Sequence
|
from collections.abc import AsyncIterator, Iterator, Sequence
|
||||||
from uuid import UUID
|
from uuid import UUID
|
||||||
|
|
||||||
from langchain_core.messages.v1 import AIMessageChunk
|
|
||||||
from langchain_core.runnables.utils import Input, Output
|
from langchain_core.runnables.utils import Input, Output
|
||||||
from langchain_core.tracers.schemas import Run
|
from langchain_core.tracers.schemas import Run
|
||||||
|
from langchain_core.v1.messages import AIMessageChunk
|
||||||
|
|
||||||
|
|
||||||
class LogEntry(TypedDict):
|
class LogEntry(TypedDict):
|
||||||
|
1
libs/core/langchain_core/v1/__init__.py
Normal file
1
libs/core/langchain_core/v1/__init__.py
Normal file
@ -0,0 +1 @@
|
|||||||
|
"""LangChain v1.0 types."""
|
@ -52,10 +52,6 @@ from langchain_core.messages.utils import (
|
|||||||
convert_from_v1_message,
|
convert_from_v1_message,
|
||||||
convert_to_messages_v1,
|
convert_to_messages_v1,
|
||||||
)
|
)
|
||||||
from langchain_core.messages.v1 import AIMessage as AIMessageV1
|
|
||||||
from langchain_core.messages.v1 import AIMessageChunk as AIMessageChunkV1
|
|
||||||
from langchain_core.messages.v1 import HumanMessage as HumanMessageV1
|
|
||||||
from langchain_core.messages.v1 import MessageV1, add_ai_message_chunks
|
|
||||||
from langchain_core.outputs import (
|
from langchain_core.outputs import (
|
||||||
ChatGeneration,
|
ChatGeneration,
|
||||||
ChatGenerationChunk,
|
ChatGenerationChunk,
|
||||||
@ -71,6 +67,10 @@ from langchain_core.utils.function_calling import (
|
|||||||
convert_to_openai_tool,
|
convert_to_openai_tool,
|
||||||
)
|
)
|
||||||
from langchain_core.utils.pydantic import TypeBaseModel, is_basemodel_subclass
|
from langchain_core.utils.pydantic import TypeBaseModel, is_basemodel_subclass
|
||||||
|
from langchain_core.v1.messages import AIMessage as AIMessageV1
|
||||||
|
from langchain_core.v1.messages import AIMessageChunk as AIMessageChunkV1
|
||||||
|
from langchain_core.v1.messages import HumanMessage as HumanMessageV1
|
||||||
|
from langchain_core.v1.messages import MessageV1, add_ai_message_chunks
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from langchain_core.output_parsers.base import OutputParserLike
|
from langchain_core.output_parsers.base import OutputParserLike
|
@ -10,9 +10,9 @@ from typing_extensions import override
|
|||||||
from langchain_core.callbacks.base import AsyncCallbackHandler
|
from langchain_core.callbacks.base import AsyncCallbackHandler
|
||||||
from langchain_core.language_models import GenericFakeChatModel
|
from langchain_core.language_models import GenericFakeChatModel
|
||||||
from langchain_core.messages import AIMessage, BaseMessage
|
from langchain_core.messages import AIMessage, BaseMessage
|
||||||
from langchain_core.messages.v1 import AIMessageChunk as AIMessageChunkV1
|
|
||||||
from langchain_core.messages.v1 import MessageV1
|
|
||||||
from langchain_core.outputs import ChatGenerationChunk, GenerationChunk
|
from langchain_core.outputs import ChatGenerationChunk, GenerationChunk
|
||||||
|
from langchain_core.v1.messages import AIMessageChunk as AIMessageChunkV1
|
||||||
|
from langchain_core.v1.messages import MessageV1
|
||||||
|
|
||||||
|
|
||||||
class MyCustomAsyncHandler(AsyncCallbackHandler):
|
class MyCustomAsyncHandler(AsyncCallbackHandler):
|
||||||
|
@ -9,7 +9,7 @@ from typing_extensions import override
|
|||||||
|
|
||||||
from langchain_core.callbacks.base import AsyncCallbackHandler, BaseCallbackHandler
|
from langchain_core.callbacks.base import AsyncCallbackHandler, BaseCallbackHandler
|
||||||
from langchain_core.messages import BaseMessage
|
from langchain_core.messages import BaseMessage
|
||||||
from langchain_core.messages.v1 import MessageV1
|
from langchain_core.v1.messages import MessageV1
|
||||||
|
|
||||||
|
|
||||||
class BaseFakeCallbackHandler(BaseModel):
|
class BaseFakeCallbackHandler(BaseModel):
|
||||||
|
@ -15,9 +15,9 @@ from langchain_core.language_models import (
|
|||||||
ParrotFakeChatModel,
|
ParrotFakeChatModel,
|
||||||
)
|
)
|
||||||
from langchain_core.messages import AIMessage, AIMessageChunk, BaseMessage, HumanMessage
|
from langchain_core.messages import AIMessage, AIMessageChunk, BaseMessage, HumanMessage
|
||||||
from langchain_core.messages.v1 import AIMessageChunk as AIMessageChunkV1
|
|
||||||
from langchain_core.messages.v1 import MessageV1
|
|
||||||
from langchain_core.outputs import ChatGenerationChunk, GenerationChunk
|
from langchain_core.outputs import ChatGenerationChunk, GenerationChunk
|
||||||
|
from langchain_core.v1.messages import AIMessageChunk as AIMessageChunkV1
|
||||||
|
from langchain_core.v1.messages import MessageV1
|
||||||
from tests.unit_tests.stubs import (
|
from tests.unit_tests.stubs import (
|
||||||
_any_id_ai_message,
|
_any_id_ai_message,
|
||||||
_any_id_ai_message_chunk,
|
_any_id_ai_message_chunk,
|
||||||
|
@ -25,7 +25,6 @@ from langchain_core.messages import (
|
|||||||
HumanMessage,
|
HumanMessage,
|
||||||
SystemMessage,
|
SystemMessage,
|
||||||
)
|
)
|
||||||
from langchain_core.messages.v1 import AIMessageChunk as AIMessageChunkV1
|
|
||||||
from langchain_core.outputs import ChatGeneration, ChatGenerationChunk, ChatResult
|
from langchain_core.outputs import ChatGeneration, ChatGenerationChunk, ChatResult
|
||||||
from langchain_core.outputs.llm_result import LLMResult
|
from langchain_core.outputs.llm_result import LLMResult
|
||||||
from langchain_core.tracers import LogStreamCallbackHandler
|
from langchain_core.tracers import LogStreamCallbackHandler
|
||||||
@ -33,6 +32,7 @@ from langchain_core.tracers.base import BaseTracer
|
|||||||
from langchain_core.tracers.context import collect_runs
|
from langchain_core.tracers.context import collect_runs
|
||||||
from langchain_core.tracers.event_stream import _AstreamEventsCallbackHandler
|
from langchain_core.tracers.event_stream import _AstreamEventsCallbackHandler
|
||||||
from langchain_core.tracers.schemas import Run
|
from langchain_core.tracers.schemas import Run
|
||||||
|
from langchain_core.v1.messages import AIMessageChunk as AIMessageChunkV1
|
||||||
from tests.unit_tests.fake.callbacks import (
|
from tests.unit_tests.fake.callbacks import (
|
||||||
BaseFakeCallbackHandler,
|
BaseFakeCallbackHandler,
|
||||||
FakeAsyncCallbackHandler,
|
FakeAsyncCallbackHandler,
|
||||||
|
@ -1,6 +1,6 @@
|
|||||||
"""Unit tests for ResponseMetadata TypedDict."""
|
"""Unit tests for ResponseMetadata TypedDict."""
|
||||||
|
|
||||||
from langchain_core.messages.v1 import AIMessage, AIMessageChunk, ResponseMetadata
|
from langchain_core.v1.messages import AIMessage, AIMessageChunk, ResponseMetadata
|
||||||
|
|
||||||
|
|
||||||
class TestResponseMetadata:
|
class TestResponseMetadata:
|
||||||
|
@ -8,12 +8,12 @@ from langchain_core.exceptions import OutputParserException
|
|||||||
from langchain_core.language_models import GenericFakeChatModel
|
from langchain_core.language_models import GenericFakeChatModel
|
||||||
from langchain_core.language_models.fake_chat_models import GenericFakeChatModelV1
|
from langchain_core.language_models.fake_chat_models import GenericFakeChatModelV1
|
||||||
from langchain_core.messages import AIMessage
|
from langchain_core.messages import AIMessage
|
||||||
from langchain_core.messages.v1 import AIMessage as AIMessageV1
|
|
||||||
from langchain_core.output_parsers import (
|
from langchain_core.output_parsers import (
|
||||||
BaseGenerationOutputParser,
|
BaseGenerationOutputParser,
|
||||||
BaseTransformOutputParser,
|
BaseTransformOutputParser,
|
||||||
)
|
)
|
||||||
from langchain_core.outputs import ChatGeneration, Generation
|
from langchain_core.outputs import ChatGeneration, Generation
|
||||||
|
from langchain_core.v1.messages import AIMessage as AIMessageV1
|
||||||
|
|
||||||
|
|
||||||
def test_base_generation_parser() -> None:
|
def test_base_generation_parser() -> None:
|
||||||
|
@ -10,14 +10,14 @@ from langchain_core.messages import (
|
|||||||
BaseMessage,
|
BaseMessage,
|
||||||
ToolCallChunk,
|
ToolCallChunk,
|
||||||
)
|
)
|
||||||
from langchain_core.messages.v1 import AIMessage as AIMessageV1
|
|
||||||
from langchain_core.messages.v1 import AIMessageChunk as AIMessageChunkV1
|
|
||||||
from langchain_core.output_parsers.openai_tools import (
|
from langchain_core.output_parsers.openai_tools import (
|
||||||
JsonOutputKeyToolsParser,
|
JsonOutputKeyToolsParser,
|
||||||
JsonOutputToolsParser,
|
JsonOutputToolsParser,
|
||||||
PydanticToolsParser,
|
PydanticToolsParser,
|
||||||
)
|
)
|
||||||
from langchain_core.outputs import ChatGeneration
|
from langchain_core.outputs import ChatGeneration
|
||||||
|
from langchain_core.v1.messages import AIMessage as AIMessageV1
|
||||||
|
from langchain_core.v1.messages import AIMessageChunk as AIMessageChunkV1
|
||||||
|
|
||||||
STREAMED_MESSAGES: list = [
|
STREAMED_MESSAGES: list = [
|
||||||
AIMessageChunk(content=""),
|
AIMessageChunk(content=""),
|
||||||
|
@ -2529,7 +2529,7 @@
|
|||||||
'title': 'ToolMessage',
|
'title': 'ToolMessage',
|
||||||
'type': 'object',
|
'type': 'object',
|
||||||
}),
|
}),
|
||||||
'langchain_core__messages__v1__AIMessage': dict({
|
'langchain_core__v1__messages__AIMessage': dict({
|
||||||
'properties': dict({
|
'properties': dict({
|
||||||
'content': dict({
|
'content': dict({
|
||||||
'items': dict({
|
'items': dict({
|
||||||
@ -2655,7 +2655,7 @@
|
|||||||
'title': 'AIMessage',
|
'title': 'AIMessage',
|
||||||
'type': 'object',
|
'type': 'object',
|
||||||
}),
|
}),
|
||||||
'langchain_core__messages__v1__AIMessageChunk': dict({
|
'langchain_core__v1__messages__AIMessageChunk': dict({
|
||||||
'properties': dict({
|
'properties': dict({
|
||||||
'content': dict({
|
'content': dict({
|
||||||
'items': dict({
|
'items': dict({
|
||||||
@ -2781,7 +2781,7 @@
|
|||||||
'title': 'AIMessageChunk',
|
'title': 'AIMessageChunk',
|
||||||
'type': 'object',
|
'type': 'object',
|
||||||
}),
|
}),
|
||||||
'langchain_core__messages__v1__HumanMessage': dict({
|
'langchain_core__v1__messages__HumanMessage': dict({
|
||||||
'properties': dict({
|
'properties': dict({
|
||||||
'content': dict({
|
'content': dict({
|
||||||
'items': dict({
|
'items': dict({
|
||||||
@ -2869,7 +2869,7 @@
|
|||||||
'title': 'HumanMessage',
|
'title': 'HumanMessage',
|
||||||
'type': 'object',
|
'type': 'object',
|
||||||
}),
|
}),
|
||||||
'langchain_core__messages__v1__SystemMessage': dict({
|
'langchain_core__v1__messages__SystemMessage': dict({
|
||||||
'properties': dict({
|
'properties': dict({
|
||||||
'content': dict({
|
'content': dict({
|
||||||
'items': dict({
|
'items': dict({
|
||||||
@ -2969,7 +2969,7 @@
|
|||||||
'title': 'SystemMessage',
|
'title': 'SystemMessage',
|
||||||
'type': 'object',
|
'type': 'object',
|
||||||
}),
|
}),
|
||||||
'langchain_core__messages__v1__ToolMessage': dict({
|
'langchain_core__v1__messages__ToolMessage': dict({
|
||||||
'properties': dict({
|
'properties': dict({
|
||||||
'artifact': dict({
|
'artifact': dict({
|
||||||
'anyOf': list([
|
'anyOf': list([
|
||||||
@ -3123,19 +3123,19 @@
|
|||||||
]),
|
]),
|
||||||
}),
|
}),
|
||||||
dict({
|
dict({
|
||||||
'$ref': '#/$defs/langchain_core__messages__v1__AIMessage',
|
'$ref': '#/$defs/langchain_core__v1__messages__AIMessage',
|
||||||
}),
|
}),
|
||||||
dict({
|
dict({
|
||||||
'$ref': '#/$defs/langchain_core__messages__v1__AIMessageChunk',
|
'$ref': '#/$defs/langchain_core__v1__messages__AIMessageChunk',
|
||||||
}),
|
}),
|
||||||
dict({
|
dict({
|
||||||
'$ref': '#/$defs/langchain_core__messages__v1__HumanMessage',
|
'$ref': '#/$defs/langchain_core__v1__messages__HumanMessage',
|
||||||
}),
|
}),
|
||||||
dict({
|
dict({
|
||||||
'$ref': '#/$defs/langchain_core__messages__v1__SystemMessage',
|
'$ref': '#/$defs/langchain_core__v1__messages__SystemMessage',
|
||||||
}),
|
}),
|
||||||
dict({
|
dict({
|
||||||
'$ref': '#/$defs/langchain_core__messages__v1__ToolMessage',
|
'$ref': '#/$defs/langchain_core__v1__messages__ToolMessage',
|
||||||
}),
|
}),
|
||||||
]),
|
]),
|
||||||
'title': 'RunnableParallel<as_list,as_str>Input',
|
'title': 'RunnableParallel<as_list,as_str>Input',
|
||||||
|
@ -9412,19 +9412,19 @@
|
|||||||
]),
|
]),
|
||||||
}),
|
}),
|
||||||
dict({
|
dict({
|
||||||
'$ref': '#/definitions/langchain_core__messages__v1__AIMessage',
|
'$ref': '#/definitions/langchain_core__v1__messages__AIMessage',
|
||||||
}),
|
}),
|
||||||
dict({
|
dict({
|
||||||
'$ref': '#/definitions/langchain_core__messages__v1__AIMessageChunk',
|
'$ref': '#/definitions/langchain_core__v1__messages__AIMessageChunk',
|
||||||
}),
|
}),
|
||||||
dict({
|
dict({
|
||||||
'$ref': '#/definitions/langchain_core__messages__v1__HumanMessage',
|
'$ref': '#/definitions/langchain_core__v1__messages__HumanMessage',
|
||||||
}),
|
}),
|
||||||
dict({
|
dict({
|
||||||
'$ref': '#/definitions/langchain_core__messages__v1__SystemMessage',
|
'$ref': '#/definitions/langchain_core__v1__messages__SystemMessage',
|
||||||
}),
|
}),
|
||||||
dict({
|
dict({
|
||||||
'$ref': '#/definitions/langchain_core__messages__v1__ToolMessage',
|
'$ref': '#/definitions/langchain_core__v1__messages__ToolMessage',
|
||||||
}),
|
}),
|
||||||
]),
|
]),
|
||||||
'definitions': dict({
|
'definitions': dict({
|
||||||
@ -11521,7 +11521,7 @@
|
|||||||
'title': 'ToolMessage',
|
'title': 'ToolMessage',
|
||||||
'type': 'object',
|
'type': 'object',
|
||||||
}),
|
}),
|
||||||
'langchain_core__messages__v1__AIMessage': dict({
|
'langchain_core__v1__messages__AIMessage': dict({
|
||||||
'properties': dict({
|
'properties': dict({
|
||||||
'content': dict({
|
'content': dict({
|
||||||
'items': dict({
|
'items': dict({
|
||||||
@ -11646,7 +11646,7 @@
|
|||||||
'title': 'AIMessage',
|
'title': 'AIMessage',
|
||||||
'type': 'object',
|
'type': 'object',
|
||||||
}),
|
}),
|
||||||
'langchain_core__messages__v1__AIMessageChunk': dict({
|
'langchain_core__v1__messages__AIMessageChunk': dict({
|
||||||
'properties': dict({
|
'properties': dict({
|
||||||
'content': dict({
|
'content': dict({
|
||||||
'items': dict({
|
'items': dict({
|
||||||
@ -11771,7 +11771,7 @@
|
|||||||
'title': 'AIMessageChunk',
|
'title': 'AIMessageChunk',
|
||||||
'type': 'object',
|
'type': 'object',
|
||||||
}),
|
}),
|
||||||
'langchain_core__messages__v1__HumanMessage': dict({
|
'langchain_core__v1__messages__HumanMessage': dict({
|
||||||
'properties': dict({
|
'properties': dict({
|
||||||
'content': dict({
|
'content': dict({
|
||||||
'items': dict({
|
'items': dict({
|
||||||
@ -11858,7 +11858,7 @@
|
|||||||
'title': 'HumanMessage',
|
'title': 'HumanMessage',
|
||||||
'type': 'object',
|
'type': 'object',
|
||||||
}),
|
}),
|
||||||
'langchain_core__messages__v1__SystemMessage': dict({
|
'langchain_core__v1__messages__SystemMessage': dict({
|
||||||
'properties': dict({
|
'properties': dict({
|
||||||
'content': dict({
|
'content': dict({
|
||||||
'items': dict({
|
'items': dict({
|
||||||
@ -11957,7 +11957,7 @@
|
|||||||
'title': 'SystemMessage',
|
'title': 'SystemMessage',
|
||||||
'type': 'object',
|
'type': 'object',
|
||||||
}),
|
}),
|
||||||
'langchain_core__messages__v1__ToolMessage': dict({
|
'langchain_core__v1__messages__ToolMessage': dict({
|
||||||
'properties': dict({
|
'properties': dict({
|
||||||
'artifact': dict({
|
'artifact': dict({
|
||||||
'anyOf': list([
|
'anyOf': list([
|
||||||
|
@ -35,9 +35,9 @@ from langchain_core.messages.content_blocks import KNOWN_BLOCK_TYPES, ContentBlo
|
|||||||
from langchain_core.messages.tool import invalid_tool_call as create_invalid_tool_call
|
from langchain_core.messages.tool import invalid_tool_call as create_invalid_tool_call
|
||||||
from langchain_core.messages.tool import tool_call as create_tool_call
|
from langchain_core.messages.tool import tool_call as create_tool_call
|
||||||
from langchain_core.messages.tool import tool_call_chunk as create_tool_call_chunk
|
from langchain_core.messages.tool import tool_call_chunk as create_tool_call_chunk
|
||||||
from langchain_core.messages.v1 import AIMessage as AIMessageV1
|
|
||||||
from langchain_core.messages.v1 import AIMessageChunk as AIMessageChunkV1
|
|
||||||
from langchain_core.utils._merge import merge_lists
|
from langchain_core.utils._merge import merge_lists
|
||||||
|
from langchain_core.v1.messages import AIMessage as AIMessageV1
|
||||||
|
from langchain_core.v1.messages import AIMessageChunk as AIMessageChunkV1
|
||||||
|
|
||||||
|
|
||||||
def test_message_init() -> None:
|
def test_message_init() -> None:
|
||||||
|
@ -37,7 +37,6 @@ from langchain_core.callbacks.manager import (
|
|||||||
from langchain_core.documents import Document
|
from langchain_core.documents import Document
|
||||||
from langchain_core.messages import ToolCall, ToolMessage
|
from langchain_core.messages import ToolCall, ToolMessage
|
||||||
from langchain_core.messages.tool import ToolOutputMixin
|
from langchain_core.messages.tool import ToolOutputMixin
|
||||||
from langchain_core.messages.v1 import ToolMessage as ToolMessageV1
|
|
||||||
from langchain_core.retrievers import BaseRetriever
|
from langchain_core.retrievers import BaseRetriever
|
||||||
from langchain_core.runnables import (
|
from langchain_core.runnables import (
|
||||||
Runnable,
|
Runnable,
|
||||||
@ -69,6 +68,7 @@ from langchain_core.utils.pydantic import (
|
|||||||
_create_subset_model,
|
_create_subset_model,
|
||||||
create_model_v2,
|
create_model_v2,
|
||||||
)
|
)
|
||||||
|
from langchain_core.v1.messages import ToolMessage as ToolMessageV1
|
||||||
from tests.unit_tests.fake.callbacks import FakeCallbackHandler
|
from tests.unit_tests.fake.callbacks import FakeCallbackHandler
|
||||||
from tests.unit_tests.pydantic_utils import _schema
|
from tests.unit_tests.pydantic_utils import _schema
|
||||||
from tests.unit_tests.stubs import AnyStr
|
from tests.unit_tests.stubs import AnyStr
|
||||||
|
@ -12,11 +12,11 @@ from freezegun import freeze_time
|
|||||||
from langchain_core.callbacks import AsyncCallbackManager
|
from langchain_core.callbacks import AsyncCallbackManager
|
||||||
from langchain_core.exceptions import TracerException
|
from langchain_core.exceptions import TracerException
|
||||||
from langchain_core.messages import HumanMessage
|
from langchain_core.messages import HumanMessage
|
||||||
from langchain_core.messages.v1 import HumanMessage as HumanMessageV1
|
|
||||||
from langchain_core.messages.v1 import MessageV1
|
|
||||||
from langchain_core.outputs import LLMResult
|
from langchain_core.outputs import LLMResult
|
||||||
from langchain_core.tracers.base import AsyncBaseTracer
|
from langchain_core.tracers.base import AsyncBaseTracer
|
||||||
from langchain_core.tracers.schemas import Run
|
from langchain_core.tracers.schemas import Run
|
||||||
|
from langchain_core.v1.messages import HumanMessage as HumanMessageV1
|
||||||
|
from langchain_core.v1.messages import MessageV1
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from langchain_core.messages import BaseMessage
|
from langchain_core.messages import BaseMessage
|
||||||
|
@ -15,12 +15,12 @@ from langsmith import Client, traceable
|
|||||||
from langchain_core.callbacks import CallbackManager
|
from langchain_core.callbacks import CallbackManager
|
||||||
from langchain_core.exceptions import TracerException
|
from langchain_core.exceptions import TracerException
|
||||||
from langchain_core.messages import HumanMessage
|
from langchain_core.messages import HumanMessage
|
||||||
from langchain_core.messages.v1 import HumanMessage as HumanMessageV1
|
|
||||||
from langchain_core.messages.v1 import MessageV1
|
|
||||||
from langchain_core.outputs import LLMResult
|
from langchain_core.outputs import LLMResult
|
||||||
from langchain_core.runnables import chain as as_runnable
|
from langchain_core.runnables import chain as as_runnable
|
||||||
from langchain_core.tracers.base import BaseTracer
|
from langchain_core.tracers.base import BaseTracer
|
||||||
from langchain_core.tracers.schemas import Run
|
from langchain_core.tracers.schemas import Run
|
||||||
|
from langchain_core.v1.messages import HumanMessage as HumanMessageV1
|
||||||
|
from langchain_core.v1.messages import MessageV1
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from langchain_core.messages import BaseMessage
|
from langchain_core.messages import BaseMessage
|
||||||
|
@ -9,8 +9,8 @@ from langchain_core.messages import (
|
|||||||
BaseMessage,
|
BaseMessage,
|
||||||
)
|
)
|
||||||
from langchain_core.messages.utils import convert_from_v1_message
|
from langchain_core.messages.utils import convert_from_v1_message
|
||||||
from langchain_core.messages.v1 import AIMessage as AIMessageV1
|
|
||||||
from langchain_core.outputs import ChatGeneration, Generation
|
from langchain_core.outputs import ChatGeneration, Generation
|
||||||
|
from langchain_core.v1.messages import AIMessage as AIMessageV1
|
||||||
from typing_extensions import override
|
from typing_extensions import override
|
||||||
|
|
||||||
from langchain.agents.agent import AgentOutputParser
|
from langchain.agents.agent import AgentOutputParser
|
||||||
|
@ -3,8 +3,8 @@ from typing import Union
|
|||||||
from langchain_core.agents import AgentAction, AgentFinish
|
from langchain_core.agents import AgentAction, AgentFinish
|
||||||
from langchain_core.messages import BaseMessage
|
from langchain_core.messages import BaseMessage
|
||||||
from langchain_core.messages.utils import convert_from_v1_message
|
from langchain_core.messages.utils import convert_from_v1_message
|
||||||
from langchain_core.messages.v1 import AIMessage as AIMessageV1
|
|
||||||
from langchain_core.outputs import ChatGeneration, Generation
|
from langchain_core.outputs import ChatGeneration, Generation
|
||||||
|
from langchain_core.v1.messages import AIMessage as AIMessageV1
|
||||||
from typing_extensions import override
|
from typing_extensions import override
|
||||||
|
|
||||||
from langchain.agents.agent import MultiActionAgentOutputParser
|
from langchain.agents.agent import MultiActionAgentOutputParser
|
||||||
|
@ -10,8 +10,8 @@ from langchain_core.messages import (
|
|||||||
ToolCall,
|
ToolCall,
|
||||||
)
|
)
|
||||||
from langchain_core.messages.utils import convert_from_v1_message
|
from langchain_core.messages.utils import convert_from_v1_message
|
||||||
from langchain_core.messages.v1 import AIMessage as AIMessageV1
|
|
||||||
from langchain_core.outputs import ChatGeneration, Generation
|
from langchain_core.outputs import ChatGeneration, Generation
|
||||||
|
from langchain_core.v1.messages import AIMessage as AIMessageV1
|
||||||
from typing_extensions import override
|
from typing_extensions import override
|
||||||
|
|
||||||
from langchain.agents.agent import MultiActionAgentOutputParser
|
from langchain.agents.agent import MultiActionAgentOutputParser
|
||||||
|
@ -5,8 +5,8 @@ from collections.abc import AsyncIterator
|
|||||||
from typing import Any, Literal, Union, cast
|
from typing import Any, Literal, Union, cast
|
||||||
|
|
||||||
from langchain_core.callbacks import AsyncCallbackHandler
|
from langchain_core.callbacks import AsyncCallbackHandler
|
||||||
from langchain_core.messages.v1 import AIMessage
|
|
||||||
from langchain_core.outputs import LLMResult
|
from langchain_core.outputs import LLMResult
|
||||||
|
from langchain_core.v1.messages import AIMessage
|
||||||
from typing_extensions import override
|
from typing_extensions import override
|
||||||
|
|
||||||
# TODO If used by two LLM runs in parallel this won't work as expected
|
# TODO If used by two LLM runs in parallel this won't work as expected
|
||||||
|
@ -2,8 +2,8 @@ from __future__ import annotations
|
|||||||
|
|
||||||
from typing import Any, Optional, Union
|
from typing import Any, Optional, Union
|
||||||
|
|
||||||
from langchain_core.messages.v1 import AIMessage
|
|
||||||
from langchain_core.outputs import LLMResult
|
from langchain_core.outputs import LLMResult
|
||||||
|
from langchain_core.v1.messages import AIMessage
|
||||||
from typing_extensions import override
|
from typing_extensions import override
|
||||||
|
|
||||||
from langchain.callbacks.streaming_aiter import AsyncIteratorCallbackHandler
|
from langchain.callbacks.streaming_aiter import AsyncIteratorCallbackHandler
|
||||||
|
@ -7,8 +7,8 @@ from uuid import UUID
|
|||||||
|
|
||||||
from langchain_core.callbacks import base as base_callbacks
|
from langchain_core.callbacks import base as base_callbacks
|
||||||
from langchain_core.documents import Document
|
from langchain_core.documents import Document
|
||||||
from langchain_core.messages.v1 import AIMessage
|
|
||||||
from langchain_core.outputs import LLMResult
|
from langchain_core.outputs import LLMResult
|
||||||
|
from langchain_core.v1.messages import AIMessage
|
||||||
from typing_extensions import override
|
from typing_extensions import override
|
||||||
|
|
||||||
|
|
||||||
|
@ -6,7 +6,7 @@ from uuid import UUID
|
|||||||
|
|
||||||
from langchain_core.callbacks.base import AsyncCallbackHandler, BaseCallbackHandler
|
from langchain_core.callbacks.base import AsyncCallbackHandler, BaseCallbackHandler
|
||||||
from langchain_core.messages import BaseMessage
|
from langchain_core.messages import BaseMessage
|
||||||
from langchain_core.messages.v1 import MessageV1
|
from langchain_core.v1.messages import MessageV1
|
||||||
from pydantic import BaseModel
|
from pydantic import BaseModel
|
||||||
from typing_extensions import override
|
from typing_extensions import override
|
||||||
|
|
||||||
|
@ -6,9 +6,9 @@ from uuid import UUID
|
|||||||
|
|
||||||
from langchain_core.callbacks.base import AsyncCallbackHandler
|
from langchain_core.callbacks.base import AsyncCallbackHandler
|
||||||
from langchain_core.messages import AIMessage, AIMessageChunk, BaseMessage
|
from langchain_core.messages import AIMessage, AIMessageChunk, BaseMessage
|
||||||
from langchain_core.messages.v1 import AIMessageChunk as AIMessageChunkV1
|
|
||||||
from langchain_core.messages.v1 import MessageV1
|
|
||||||
from langchain_core.outputs import ChatGenerationChunk, GenerationChunk
|
from langchain_core.outputs import ChatGenerationChunk, GenerationChunk
|
||||||
|
from langchain_core.v1.messages import AIMessageChunk as AIMessageChunkV1
|
||||||
|
from langchain_core.v1.messages import MessageV1
|
||||||
from typing_extensions import override
|
from typing_extensions import override
|
||||||
|
|
||||||
from tests.unit_tests.llms.fake_chat_model import GenericFakeChatModel
|
from tests.unit_tests.llms.fake_chat_model import GenericFakeChatModel
|
||||||
|
@ -73,7 +73,7 @@ from typing import Any, Literal, Optional, Union, cast
|
|||||||
|
|
||||||
from langchain_core.messages import AIMessage, is_data_content_block
|
from langchain_core.messages import AIMessage, is_data_content_block
|
||||||
from langchain_core.messages import content_blocks as types
|
from langchain_core.messages import content_blocks as types
|
||||||
from langchain_core.messages.v1 import AIMessage as AIMessageV1
|
from langchain_core.v1.messages import AIMessage as AIMessageV1
|
||||||
|
|
||||||
_FUNCTION_CALL_IDS_MAP_KEY = "__openai_function_call_ids__"
|
_FUNCTION_CALL_IDS_MAP_KEY = "__openai_function_call_ids__"
|
||||||
|
|
||||||
|
@ -38,11 +38,6 @@ from langchain_core.callbacks import (
|
|||||||
)
|
)
|
||||||
from langchain_core.language_models import LanguageModelInput
|
from langchain_core.language_models import LanguageModelInput
|
||||||
from langchain_core.language_models.chat_models import LangSmithParams
|
from langchain_core.language_models.chat_models import LangSmithParams
|
||||||
from langchain_core.language_models.v1.chat_models import (
|
|
||||||
BaseChatModelV1,
|
|
||||||
agenerate_from_stream,
|
|
||||||
generate_from_stream,
|
|
||||||
)
|
|
||||||
from langchain_core.messages import (
|
from langchain_core.messages import (
|
||||||
InvalidToolCall,
|
InvalidToolCall,
|
||||||
ToolCall,
|
ToolCall,
|
||||||
@ -55,12 +50,6 @@ from langchain_core.messages.ai import (
|
|||||||
UsageMetadata,
|
UsageMetadata,
|
||||||
)
|
)
|
||||||
from langchain_core.messages.tool import tool_call_chunk
|
from langchain_core.messages.tool import tool_call_chunk
|
||||||
from langchain_core.messages.v1 import AIMessage as AIMessageV1
|
|
||||||
from langchain_core.messages.v1 import AIMessageChunk as AIMessageChunkV1
|
|
||||||
from langchain_core.messages.v1 import HumanMessage as HumanMessageV1
|
|
||||||
from langchain_core.messages.v1 import MessageV1, ResponseMetadata
|
|
||||||
from langchain_core.messages.v1 import SystemMessage as SystemMessageV1
|
|
||||||
from langchain_core.messages.v1 import ToolMessage as ToolMessageV1
|
|
||||||
from langchain_core.output_parsers import JsonOutputParser, PydanticOutputParser
|
from langchain_core.output_parsers import JsonOutputParser, PydanticOutputParser
|
||||||
from langchain_core.output_parsers.openai_tools import (
|
from langchain_core.output_parsers.openai_tools import (
|
||||||
JsonOutputKeyToolsParser,
|
JsonOutputKeyToolsParser,
|
||||||
@ -88,6 +77,17 @@ from langchain_core.utils.pydantic import (
|
|||||||
is_basemodel_subclass,
|
is_basemodel_subclass,
|
||||||
)
|
)
|
||||||
from langchain_core.utils.utils import _build_model_kwargs, from_env, secret_from_env
|
from langchain_core.utils.utils import _build_model_kwargs, from_env, secret_from_env
|
||||||
|
from langchain_core.v1.chat_models import (
|
||||||
|
BaseChatModelV1,
|
||||||
|
agenerate_from_stream,
|
||||||
|
generate_from_stream,
|
||||||
|
)
|
||||||
|
from langchain_core.v1.messages import AIMessage as AIMessageV1
|
||||||
|
from langchain_core.v1.messages import AIMessageChunk as AIMessageChunkV1
|
||||||
|
from langchain_core.v1.messages import HumanMessage as HumanMessageV1
|
||||||
|
from langchain_core.v1.messages import MessageV1, ResponseMetadata
|
||||||
|
from langchain_core.v1.messages import SystemMessage as SystemMessageV1
|
||||||
|
from langchain_core.v1.messages import ToolMessage as ToolMessageV1
|
||||||
from pydantic import BaseModel, ConfigDict, Field, SecretStr, model_validator
|
from pydantic import BaseModel, ConfigDict, Field, SecretStr, model_validator
|
||||||
from pydantic.v1 import BaseModel as BaseModelV1
|
from pydantic.v1 import BaseModel as BaseModelV1
|
||||||
from typing_extensions import Self
|
from typing_extensions import Self
|
||||||
|
@ -14,9 +14,9 @@ from langchain_core.messages import (
|
|||||||
HumanMessage,
|
HumanMessage,
|
||||||
MessageLikeRepresentation,
|
MessageLikeRepresentation,
|
||||||
)
|
)
|
||||||
from langchain_core.messages.v1 import AIMessage as AIMessageV1
|
from langchain_core.v1.messages import AIMessage as AIMessageV1
|
||||||
from langchain_core.messages.v1 import AIMessageChunk as AIMessageChunkV1
|
from langchain_core.v1.messages import AIMessageChunk as AIMessageChunkV1
|
||||||
from langchain_core.messages.v1 import HumanMessage as HumanMessageV1
|
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
|
||||||
|
|
||||||
|
@ -22,11 +22,11 @@ from langchain_core.messages import (
|
|||||||
)
|
)
|
||||||
from langchain_core.messages import content_blocks as types
|
from langchain_core.messages import content_blocks as types
|
||||||
from langchain_core.messages.ai import UsageMetadata
|
from langchain_core.messages.ai import UsageMetadata
|
||||||
from langchain_core.messages.v1 import AIMessage as AIMessageV1
|
|
||||||
from langchain_core.outputs import ChatGeneration, ChatResult
|
from langchain_core.outputs import ChatGeneration, ChatResult
|
||||||
from langchain_core.runnables import RunnableLambda
|
from langchain_core.runnables import RunnableLambda
|
||||||
from langchain_core.tracers.base import BaseTracer
|
from langchain_core.tracers.base import BaseTracer
|
||||||
from langchain_core.tracers.schemas import Run
|
from langchain_core.tracers.schemas import Run
|
||||||
|
from langchain_core.v1.messages import AIMessage as AIMessageV1
|
||||||
from openai.types.responses import ResponseOutputMessage, ResponseReasoningItem
|
from openai.types.responses import ResponseOutputMessage, ResponseReasoningItem
|
||||||
from openai.types.responses.response import IncompleteDetails, Response, ResponseUsage
|
from openai.types.responses.response import IncompleteDetails, Response, ResponseUsage
|
||||||
from openai.types.responses.response_error import ResponseError
|
from openai.types.responses.response_error import ResponseError
|
||||||
|
@ -2,7 +2,7 @@ from typing import Any, Optional
|
|||||||
from unittest.mock import MagicMock, patch
|
from unittest.mock import MagicMock, patch
|
||||||
|
|
||||||
from langchain_core.messages import AIMessageChunk, BaseMessageChunk
|
from langchain_core.messages import AIMessageChunk, BaseMessageChunk
|
||||||
from langchain_core.messages.v1 import AIMessageChunk as AIMessageChunkV1
|
from langchain_core.v1.messages import AIMessageChunk as AIMessageChunkV1
|
||||||
from openai.types.responses import (
|
from openai.types.responses import (
|
||||||
ResponseCompletedEvent,
|
ResponseCompletedEvent,
|
||||||
ResponseContentPartAddedEvent,
|
ResponseContentPartAddedEvent,
|
||||||
|
@ -6,7 +6,7 @@ from uuid import UUID
|
|||||||
|
|
||||||
from langchain_core.callbacks.base import AsyncCallbackHandler, BaseCallbackHandler
|
from langchain_core.callbacks.base import AsyncCallbackHandler, BaseCallbackHandler
|
||||||
from langchain_core.messages import BaseMessage
|
from langchain_core.messages import BaseMessage
|
||||||
from langchain_core.messages.v1 import MessageV1
|
from langchain_core.v1.messages import MessageV1
|
||||||
from pydantic import BaseModel
|
from pydantic import BaseModel
|
||||||
|
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user