mirror of
https://github.com/hwchase17/langchain.git
synced 2025-08-15 23:57:21 +00:00
remove unnecessary tool population
This commit is contained in:
parent
704db3741a
commit
ed006a7444
@ -36,7 +36,6 @@ from langchain_core.messages.ai import AIMessage, AIMessageChunk
|
|||||||
from langchain_core.messages.base import BaseMessage, BaseMessageChunk
|
from langchain_core.messages.base import BaseMessage, BaseMessageChunk
|
||||||
from langchain_core.messages.chat import ChatMessage, ChatMessageChunk
|
from langchain_core.messages.chat import ChatMessage, ChatMessageChunk
|
||||||
from langchain_core.messages.content_blocks import ContentBlock
|
from langchain_core.messages.content_blocks import ContentBlock
|
||||||
from langchain_core.messages.content_blocks import ToolCall as ToolCallV1
|
|
||||||
from langchain_core.messages.function import FunctionMessage, FunctionMessageChunk
|
from langchain_core.messages.function import FunctionMessage, FunctionMessageChunk
|
||||||
from langchain_core.messages.human import HumanMessage, HumanMessageChunk
|
from langchain_core.messages.human import HumanMessage, HumanMessageChunk
|
||||||
from langchain_core.messages.modifier import RemoveMessage
|
from langchain_core.messages.modifier import RemoveMessage
|
||||||
@ -492,17 +491,6 @@ def _convert_from_v0_to_v1(message: BaseMessage) -> MessageV1:
|
|||||||
return HumanMessageV1(content, name=message.name)
|
return HumanMessageV1(content, name=message.name)
|
||||||
if isinstance(message, AIMessage): # Checking for v0 AIMessage
|
if isinstance(message, AIMessage): # Checking for v0 AIMessage
|
||||||
content = [create_text_block(str(message.content))]
|
content = [create_text_block(str(message.content))]
|
||||||
# Handle tool calls from v0 messages
|
|
||||||
if message.tool_calls:
|
|
||||||
for tool_call in message.tool_calls:
|
|
||||||
content.append(
|
|
||||||
ToolCallV1(
|
|
||||||
type="tool_call",
|
|
||||||
name=tool_call["name"],
|
|
||||||
args=tool_call["args"],
|
|
||||||
id=tool_call.get("id", ""),
|
|
||||||
)
|
|
||||||
)
|
|
||||||
|
|
||||||
# Construct ResponseMetadata TypedDict from v0 response_metadata dict
|
# Construct ResponseMetadata TypedDict from v0 response_metadata dict
|
||||||
# Since ResponseMetadata has total=False, we can safely cast the dict
|
# Since ResponseMetadata has total=False, we can safely cast the dict
|
||||||
@ -512,6 +500,7 @@ def _convert_from_v0_to_v1(message: BaseMessage) -> MessageV1:
|
|||||||
name=message.name,
|
name=message.name,
|
||||||
usage_metadata=message.usage_metadata,
|
usage_metadata=message.usage_metadata,
|
||||||
response_metadata=response_metadata,
|
response_metadata=response_metadata,
|
||||||
|
tool_calls=message.tool_calls,
|
||||||
)
|
)
|
||||||
if isinstance(message, SystemMessage): # Checking for v0 SystemMessage
|
if isinstance(message, SystemMessage): # Checking for v0 SystemMessage
|
||||||
content = [create_text_block(str(message.content))]
|
content = [create_text_block(str(message.content))]
|
||||||
|
Loading…
Reference in New Issue
Block a user