remove TODOs

This commit is contained in:
Mason Daugherty 2025-08-06 14:12:35 -04:00
parent bc494570a0
commit 8fc1d5bb12
No known key found for this signature in database

View File

@ -498,9 +498,7 @@ def _convert_from_v0_to_v1(message: BaseMessage) -> MessageV1:
content.append( content.append(
ToolCallV1( ToolCallV1(
type="tool_call", type="tool_call",
name=tool_call[ name=tool_call["name"],
"name" # TODO: this is the name of the message, not the tool, so remove? # noqa: E501
],
args=tool_call["args"], args=tool_call["args"],
id=tool_call.get("id", ""), id=tool_call.get("id", ""),
) )
@ -522,7 +520,7 @@ def _convert_from_v0_to_v1(message: BaseMessage) -> MessageV1:
content = [create_text_block(str(message.content))] content = [create_text_block(str(message.content))]
return ToolMessageV1( return ToolMessageV1(
content=content, content=content,
name=message.name, # TODO: this is the name of the message, not the tool, so remove? # noqa: E501 name=message.name,
tool_call_id=message.tool_call_id, tool_call_id=message.tool_call_id,
status=message.status, status=message.status,
) )