chore(prebuilt): remove structured tool support from ToolNode (#5902)

Remove structured tool support from ToolNode We'll handle structured tools directly in the call_model nodes.
2026-10-01 22:15:11 +02:00 · 2025-08-13 22:32:21 -04:00
parent 50601dc02c
commit e6d71a586d
2 changed files with 10 additions and 164 deletions
@@ -1482,94 +1482,3 @@ def test_tool_node_stream_writer() -> None:
            },
        ),
    ]
-
-
-def test_structured_output_tools_sync() -> None:
-    """Test that ToolNode handles Pydantic model classes as structured output tools."""
-
-    class OutputSchema(BaseModel):
-        name: str
-        age: int
-        location: str
-
-    tool_node = ToolNode([OutputSchema])
-
-    # Test that the structured output tool is registered correctly
-    assert "OutputSchema" in tool_node.structured_output_tools
-
-    # Create a tool call that matches the schema
-    tool_call = {
-        "name": "OutputSchema",
-        "args": {"name": "Alice", "age": 30, "location": "NYC"},
-        "id": "call_123",
-        "type": "tool_call",
-    }
-
-    # Test sync execution
-    result = tool_node.invoke(
-        {"messages": [AIMessage(content="", tool_calls=[tool_call])]}
-    )
-
-    # Should return a Command with structured response
-    assert isinstance(result, list)
-    assert len(result) == 1
-    command = result[0]
-    assert isinstance(command, Command)
-
-    # Check the update structure
-    assert "messages" in command.update
-    assert "structured_response" in command.update
-
-    # Check the tool message
-    tool_message = command.update["messages"][0]
-    assert isinstance(tool_message, ToolMessage)
-    assert tool_message.name == "OutputSchema"
-    assert tool_message.tool_call_id == "call_123"
-
-    # Check the structured response
-    structured_response = command.update["structured_response"]
-    assert isinstance(structured_response, OutputSchema)
-    assert structured_response.name == "Alice"
-    assert structured_response.age == 30
-    assert structured_response.location == "NYC"
-
-
-async def test_structured_output_tools_async() -> None:
-    """Test that ToolNode handles Pydantic model classes as structured output tools."""
-
-    class OutputSchema(BaseModel):
-        name: str
-        age: int
-        location: str
-
-    tool_node = ToolNode([OutputSchema])
-
-    # Test that the structured output tool is registered correctly
-    assert "OutputSchema" not in tool_node.tools_by_name
-    assert "OutputSchema" in tool_node.structured_output_tools
-
-    # Create a tool call that matches the schema
-    tool_call = {
-        "name": "OutputSchema",
-        "args": {"name": "Alice", "age": 30, "location": "NYC"},
-        "id": "call_123",
-        "type": "tool_call",
-    }
-
-    # Test async execution
-    result_async = await tool_node.ainvoke(
-        {"messages": [AIMessage(content="", tool_calls=[tool_call])]}
-    )
-
-    # Should produce the same result
-    assert isinstance(result_async, list)
-    assert len(result_async) == 1
-    command_async = result_async[0]
-    assert isinstance(command_async, Command)
-    assert "structured_response" in command_async.update
-
-    structured_response_async = command_async.update["structured_response"]
-    assert isinstance(structured_response_async, OutputSchema)
-    assert structured_response_async.name == "Alice"
-    assert structured_response_async.age == 30
-    assert structured_response_async.location == "NYC"