mirror of
https://github.com/hwchase17/langchain.git
synced 2025-08-16 08:06:14 +00:00
simplify response_metadata
This commit is contained in:
parent
3165954328
commit
b9028a16a3
@ -387,16 +387,12 @@ def convert_from_v1_message(message: MessageV1) -> BaseMessage:
|
|||||||
"""
|
"""
|
||||||
content = cast("Union[str, list[str | dict]]", message.content)
|
content = cast("Union[str, list[str | dict]]", message.content)
|
||||||
if isinstance(message, AIMessageV1):
|
if isinstance(message, AIMessageV1):
|
||||||
response_metadata: dict[str, Any] = {}
|
|
||||||
if message.response_metadata:
|
|
||||||
# Copy all fields from the v1 ResponseMetadata to a plain dict
|
|
||||||
response_metadata.update(message.response_metadata)
|
|
||||||
return AIMessage(
|
return AIMessage(
|
||||||
content=content,
|
content=content,
|
||||||
id=message.id,
|
id=message.id,
|
||||||
name=message.name,
|
name=message.name,
|
||||||
tool_calls=message.tool_calls,
|
tool_calls=message.tool_calls,
|
||||||
response_metadata=response_metadata,
|
response_metadata=cast("dict", message.response_metadata),
|
||||||
)
|
)
|
||||||
if isinstance(message, AIMessageChunkV1):
|
if isinstance(message, AIMessageChunkV1):
|
||||||
response_metadata_chunk: dict[str, Any] = {}
|
response_metadata_chunk: dict[str, Any] = {}
|
||||||
|
Loading…
Reference in New Issue
Block a user