github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-08-06 19:48:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

remove type ignores for eugene

Browse Source
This commit is contained in:
Chester Curme 2025-07-24 13:31:50 -04:00
parent b94f23883f
commit dd8057a034
1 changed files with 8 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
libs/core/langchain_core/messages/utils.py
View File

@ -300,38 +300,37 @@ def _create_message_from_message_type(
def _convert_from_v1_message(message: MessageV1) -> BaseMessage: def _convert_from_v1_message(message: MessageV1) -> BaseMessage:
# type ignores here are because AIMessageV1.content is a list of dicts. AIMessage content = cast("Union[str, list[str | dict]]", message.content)
# content expects str or list[str | dict].
if isinstance(message, AIMessageV1): if isinstance(message, AIMessageV1):
return AIMessage( return AIMessage(
content=message.content, # type: ignore[arg-type] 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=message.response_metadata, response_metadata=cast(dict, message.response_metadata),
) )
if isinstance(message, AIMessageChunkV1): if isinstance(message, AIMessageChunkV1):
return AIMessageChunk( return AIMessageChunk(
content=message.content, # type: ignore[arg-type] content=content,
id=message.id, id=message.id,
name=message.name, name=message.name,
tool_call_chunks=message.tool_call_chunks, tool_call_chunks=message.tool_call_chunks,
response_metadata=message.response_metadata, response_metadata=cast(dict, message.response_metadata),
) )
if isinstance(message, HumanMessageV1): if isinstance(message, HumanMessageV1):
return HumanMessage( return HumanMessage(
content=message.content, # type: ignore[arg-type] content=content,
id=message.id, id=message.id,
name=message.name, name=message.name,
) )
if isinstance(message, SystemMessageV1): if isinstance(message, SystemMessageV1):
return SystemMessage( return SystemMessage(
content=message.content, # type: ignore[arg-type] content=content,
id=message.id, id=message.id,
) )
if isinstance(message, ToolMessageV1): if isinstance(message, ToolMessageV1):
return ToolMessage( return ToolMessage(
content=message.content, # type: ignore[arg-type] content=content,
id=message.id, id=message.id,
) )
message = f"Unsupported message type: {type(message)}" message = f"Unsupported message type: {type(message)}"