Add support for `Tool.outputSchema` and `CallToolResult.structuredContent` #316

JMLX42 · 2025-07-14T12:44:40Z

Add support for:

Tool.outputSchema
CallToolResult.structuredContent

Motivation and Context

Implements #312

First step toward MCP 2025-06-18 support.

How Has This Been Tested?

Comprehensive unit tests for the new structured output features we implemented. The tests cover:

CallToolResult::structured() and CallToolResult::structured_error() methods
Tool output_schema field functionality
IntoCallToolResult trait implementation for Structured<T>
Mutual exclusivity validation between content and structured_content
Schema generation and serialization/deserialization

The tests are located in tests/test_structured_output.rs and provide good coverage of the core functionality we added.

Breaking Changes

Both Tool.outputSchema and CallToolResult.structuredContent are optional.

The only breaking change being that CallToolResult.content is now optional to support mutual exclusivity with structured_content.

Types of changes

Bug fix (non-breaking change which fixes an issue)
New feature (non-breaking change which adds functionality)
Breaking change (fix or feature that would cause existing functionality to change)
Documentation update

Checklist

I have read the MCP Documentation
My code follows the repository's style guidelines
New and existing tests pass locally
I have added appropriate error handling
I have added or updated documentation as needed

Additional context

None for now.

Task List

Core Data Structures

Add output_schema: Option<Arc<JsonObject>> field to Tool struct
Add structured_content: Option<Value> field to CallToolResult struct
Implement validation for mutually exclusive content/structuredContent fields
Add CallToolResult::structured() constructor method
Add CallToolResult::structured_error() constructor method

Macro Support

Parse function return types in #[tool] macro to generate output schemas
Support explicit output_schema attribute for manual schema specification
Generate schema using schemars for structured return types
Store output schema in generated tool metadata
Update tool_attr generation to include output_schema

Type Conversion Infrastructure

Create Structured<T> wrapper type for structured results
Implement IntoCallToolResult for Structured<T>
Implement IntoCallToolResult for types that should produce structured content
Add automatic JSON serialization for structured types
Implement schema validation in conversion logic

Tool Handler Updates

Update tool invocation to check for output_schema
Implement validation of structured output against schema
Handle conversion between Rust types and JSON values
Update error propagation for validation failures
Cache output schemas similar to input schemas
Update tool listing to include output schemas

Testing

Test Tool serialization/deserialization with output_schema
Test CallToolResult with structured_content
Test mutual exclusivity validation
Test schema validation for structured outputs
Test #[tool] macro with various return types
Test error cases (schema violations, invalid types)
Test backward compatibility with existing tools
Add integration tests for end-to-end scenarios

Documentation and Examples

Document Tool.outputSchema field usage
Document CallToolResult.structuredContent usage
Create example: simple tool with structured output
Create example: complex nested structures
Create example: error handling with structured content
Write migration guide for existing tools
Update API documentation
Add inline code documentation

Validation Improvements

Enforce structured_content usage when output_schema is defined
Forbid content field when output_schema is present
Ensure errors also use structured_content for tools with output_schema
Add comprehensive validation tests for the new strict behavior
Update IntoCallToolResult implementations for consistent error handling

Technical Considerations

Backward Compatibility

All changes must be backward compatible
Tools without output_schema continue to work as before
Clients that don't understand structured_content can still use content field

Performance

Schema generation should be cached
Validation should be efficient
Consider lazy evaluation where appropriate

Error Handling

Clear error messages for schema violations
Proper error propagation through the macro system
Graceful degradation when schemas can't be generated

Dependencies

schemars 1.0 for schema generation
serde_json for JSON manipulation
Existing MCP types and traits

Timeline Estimate

Core data structure updates: 2-3 hours
Macro enhancements: 4-6 hours
Type conversion and validation: 3-4 hours
Testing: 3-4 hours
Documentation: 2-3 hours

Total estimated time: 14-20 hours

References

- Add optional output_schema field to Tool struct for defining tool output structure - Update Tool::new() to initialize output_schema as None

- Add optional structured_content field for JSON object results - Make content field optional to support either structured or unstructured results - Add CallToolResult::structured() and structured_error() constructor methods

…ontent - Add validate() method to ensure content and structured_content are mutually exclusive - Implement custom Deserialize to enforce validation during deserialization - Update documentation to clarify the mutual exclusivity requirement

- Add output_schema field to ToolAttribute and ResolvedToolAttribute structs - Implement automatic output schema generation from return types - Support explicit output_schema attribute for manual specification - Generate schemas for Result<T, E> where T is not CallToolResult - Update tool generation to include output_schema in Tool struct

- Add Structured<T> wrapper type for explicit structured content - Implement IntoCallToolResult for Structured<T> with JSON serialization - Add support for Result<Structured<T>, E> conversions - Enable tools to return structured content through the trait system

- Handle Option<Vec<Content>> in CallToolResult.content - Add proper unwrapping for the optional content field - Fix compilation error in chat.rs

- Add output_schema field to Tool initialization in sampling_stdio example - Update test_tool_macros tests to handle Option<Vec<Content>> - Use as_ref() before calling first() on optional content field

- Add validate_against_schema function for basic type validation - Add note that full JSON Schema validation requires dedicated library - Document that actual validation should happen in tool handler

- Add output_schema field to Tool struct for defining output JSON schemas - Add structured_content field to CallToolResult (mutually exclusive with content) - Implement Structured<T> wrapper for type-safe structured outputs - Update #[tool] macro to automatically generate output schemas from return types - Add validation of structured outputs against their schemas - Update all examples and tests for breaking change (CallToolResult.content now Option) - Add comprehensive documentation and rustdoc - Add structured_output example demonstrating the feature BREAKING CHANGE: CallToolResult.content is now Option<Vec<Content>> instead of Vec<Content> Closes modelcontextprotocol#312

JMLX42 · 2025-07-14T16:20:25Z

cargo test fails on some doctest. I'm working on it.

The #[tool] macro requires Parameters<T> wrapper for tool inputs. This fixes the pre-existing broken doctest in the structured output documentation example.

JMLX42 · 2025-07-14T17:34:52Z

Some doctest were apparently failing on main already. I fixed them in cb28342

jokemanfire · 2025-07-15T02:08:06Z

crates/rmcp-macros/src/tool.rs

+                }
+            }
+            _ => None,
+        }


Is the complexity of the circle here a bit high?

Simplify this please.

@4t145 done in 70bf2b1

It seems that there are no modifications here

@jokemanfire my bad! Sorry.

Done in 43f08bf

JMLX42 · 2025-07-15T15:17:08Z

I am also working on this, but I would like to merge your pr firstly.

@4t145 I just pushed an update that should address all your comments.

- Add Default implementation for StructuredOutputServer - Fix collapsible else-if in simple-chat-client - No functional changes

Apply automatic formatting changes to: - examples/simple-chat-client/src/chat.rs - fix line wrapping - crates/rmcp-macros/src/tool.rs - format method chaining - examples/servers/src/structured_output.rs - reorder imports and format function signatures

JMLX42 · 2025-07-16T09:14:14Z

I'm fixing the formatting and doc issues.

IIRC the commits will be squashed so I won't fix the commit message lint issues unless asked to by a maintainer.

JMLX42 · 2025-07-16T09:16:07Z

I'm fixing the formatting and doc issues.

IIRC the commits will be squashed so I won't fix the commit message lint issues unless asked to by a maintainer.

@4t145 all done!

jokemanfire · 2025-07-17T01:31:27Z

we should create a 6-18 spec milestone ,for stracing the work before merging this pr @4t145 @alexhancock

crates/rmcp/src/handler/server/tool.rs

JMLX42 · 2025-07-22T15:26:45Z

@4t145 @jokemanfire FYI I already made an actual implementation that leverages this PR:

https://gitlab.com/lx-industries/rmcp-openapi/-/blob/4d6bdaac3780c8953051f5b90019227c01300bb0/src/tool_generator.rs#L223

and as far as I can tell, it works!

JMLX42 · 2025-07-22T18:34:32Z

Back to draft to add validation improvements:

Enforce structured_content usage when output_schema is defined
Forbid content field when output_schema is present
Ensure errors also use structured_content for tools with output_schema
Add comprehensive validation tests for the new strict behavior
Update IntoCallToolResult implementations for consistent error handling

JMLX42 · 2025-07-22T19:01:05Z

Back to draft to add validation improvements:

Enforce structured_content usage when output_schema is defined

Forbid content field when output_schema is present

Ensure errors also use structured_content for tools with output_schema

Add comprehensive validation tests for the new strict behavior

Update IntoCallToolResult implementations for consistent error handling

Enhanced Validation in ToolRouter::call() (crates/rmcp/src/handler/server/router/tool.rs:250-269):
- Added check that tools with output_schema must return structured_content (not None)
- Added check that tools with output_schema cannot use regular content field
- Maintained existing schema validation for structured_content
Added Comprehensive Tests (crates/rmcp/tests/test_structured_output.rs):
- test_output_schema_requires_structured_content() - verifies tools with schemas use structured content
- test_output_schema_forbids_regular_content() - documents the validation logic
- test_output_schema_error_must_be_structured() - covers error scenarios
- test_structured_content_schema_validation() - confirms schema validation works
Updated Documentation (TODO.md):
- Marked validation improvements as complete
- Added tracking for the new validation features
Created Example (examples/servers/src/strict_output_validation.rs):
- Demonstrates proper structured output usage
- Shows what would fail validation (commented out)

Key Benefits:

Consistency: Tools that declare structured output via output_schema must consistently use structured_content for both success and error cases
Type Safety: All responses from structured tools are validated against their declared schemas
Clear Contract: When a tool has output_schema, clients can rely on getting structured responses
Breaking Change: This enforces stricter validation, ensuring tools are correctly implemented

Validation Logic:

  if let Some(ref output_schema) = item.attr.output_schema {
      // When output_schema is defined, structured_content is required
      if result.structured_content.is_none() {
          return Err(crate::ErrorData::invalid_params(
              "Tool with output_schema must return structured_content",
              None
          ));
      }

      // Ensure content is not used when output_schema is defined
      if result.content.is_some() {
          return Err(crate::ErrorData::invalid_params(
              "Tool with output_schema cannot use content field",
              None
          ));
      }

      // Validate the structured content against the schema
      validate_against_schema(result.structured_content.as_ref().unwrap(), output_schema)?;
  }

The implementation successfully addresses the original issue and ensures that when a tool declares an output_schema, it must consistently return structured content, providing better type safety and
clearer contracts for MCP clients.

This commit implements strict validation to ensure tools with output_schema consistently use structured_content for both success and error responses. Changes: - Enhanced ToolRouter::call() validation to require structured_content when output_schema is present - Added validation that tools with output_schema cannot use regular content field - Added comprehensive tests covering the new strict validation behavior - Created example demonstrating proper structured output usage - Updated TODO.md to track validation improvements This ensures consistent response format and better type safety for MCP clients.

jokemanfire · 2025-07-23T01:18:48Z

TODO.md

@@ -0,0 +1,142 @@
+Add support for:
+


We may should not contain this doc.

@jokemanfire sorry about that! fixed in cf52be9

jokemanfire · 2025-07-23T01:19:29Z

crates/rmcp-macros/src/tool.rs

+                }
+            }
+            _ => None,
+        }


It seems that there are no modifications here

- Extract complex nested logic into dedicated helper function - Replace deeply nested if-else chains with functional approach - Use early returns and ? operator for cleaner code flow - Reduce 46 lines to 7 lines in main logic while improving readability

JMLX42 · 2025-07-23T12:23:24Z

@4t145 fixed formatting: 5109fcb

Sorry about that.

github-actions bot added T-documentation Documentation improvements T-core Core library changes T-model Model/data structure changes labels Jul 14, 2025

JMLX42 mentioned this pull request Jul 14, 2025

Structured tool output + schema #312

Open

JMLX42 marked this pull request as draft July 14, 2025 12:45

github-actions bot added T-dependencies Dependencies related changes T-test Testing related changes T-config Configuration file changes T-examples Example code changes T-handler Handler implementation changes T-macros Macro changes labels Jul 14, 2025

JMLX42 marked this pull request as ready for review July 14, 2025 14:09

JMLX42 added 8 commits July 14, 2025 17:55

feat: add output_schema field to Tool struct

c2f65cc

- Add optional output_schema field to Tool struct for defining tool output structure - Update Tool::new() to initialize output_schema as None

feat: add structured_content field to CallToolResult

5d24acc

- Add optional structured_content field for JSON object results - Make content field optional to support either structured or unstructured results - Add CallToolResult::structured() and structured_error() constructor methods

fix: update simple-chat-client example for optional content field

366d0af

- Handle Option<Vec<Content>> in CallToolResult.content - Add proper unwrapping for the optional content field - Fix compilation error in chat.rs

fix: update examples and tests for optional content field

b16fd38

- Add output_schema field to Tool initialization in sampling_stdio example - Update test_tool_macros tests to handle Option<Vec<Content>> - Use as_ref() before calling first() on optional content field

feat: implement basic schema validation in conversion logic

d82056f

- Add validate_against_schema function for basic type validation - Add note that full JSON Schema validation requires dedicated library - Document that actual validation should happen in tool handler

JMLX42 force-pushed the feature/output-schema branch from d9b8bfc to 8769ec5 Compare July 14, 2025 15:55

github-actions bot removed the T-documentation Documentation improvements label Jul 14, 2025

JMLX42 force-pushed the feature/output-schema branch from 8769ec5 to b174b63 Compare July 14, 2025 16:00

JMLX42 marked this pull request as draft July 14, 2025 16:06

fix: correct structured output doctest to use Parameters wrapper

cb28342

The #[tool] macro requires Parameters<T> wrapper for tool inputs. This fixes the pre-existing broken doctest in the structured output documentation example.

JMLX42 marked this pull request as ready for review July 14, 2025 17:34

jokemanfire reviewed Jul 15, 2025

View reviewed changes

JMLX42 added 3 commits July 15, 2025 17:20

fix: address clippy warnings

4001d65

- Add Default implementation for StructuredOutputServer - Fix collapsible else-if in simple-chat-client - No functional changes

style: apply cargo fmt

cff51c4

Apply automatic formatting changes to: - examples/simple-chat-client/src/chat.rs - fix line wrapping - crates/rmcp-macros/src/tool.rs - format method chaining - examples/servers/src/structured_output.rs - reorder imports and format function signatures

chore: fix formatting

33b4d59

chore: fix rustdoc redundant link warning

1274857

JMLX42 requested review from 4t145 and jokemanfire July 16, 2025 21:23

jokemanfire reviewed Jul 17, 2025

View reviewed changes

crates/rmcp/src/handler/server/tool.rs Show resolved Hide resolved

refactor: validate_against_schema

a1ef39e

JMLX42 requested a review from jokemanfire July 17, 2025 16:41

Merge branch 'modelcontextprotocol:main' into feature/output-schema

f752f66

JMLX42 marked this pull request as draft July 22, 2025 18:33

github-actions bot added the T-documentation Documentation improvements label Jul 22, 2025

JMLX42 marked this pull request as ready for review July 22, 2025 19:56

jokemanfire added this to the version 0.3.x milestone Jul 23, 2025

jokemanfire reviewed Jul 23, 2025

View reviewed changes

chore: remove TODO.md

cf52be9

github-actions bot removed the T-documentation Documentation improvements label Jul 23, 2025

JMLX42 requested a review from jokemanfire July 23, 2025 08:57

chore: run cargo fmt

5109fcb

jokemanfire approved these changes Jul 24, 2025

View reviewed changes

Add support for Tool.outputSchema and CallToolResult.structuredContent #316

Are you sure you want to change the base?

Add support for Tool.outputSchema and CallToolResult.structuredContent #316

Conversation

JMLX42 commented Jul 14, 2025 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Motivation and Context

How Has This Been Tested?

Breaking Changes

Types of changes

Checklist

Additional context

Task List

Core Data Structures

Macro Support

Type Conversion Infrastructure

Tool Handler Updates

Testing

Documentation and Examples

Validation Improvements

Technical Considerations

Backward Compatibility

Performance

Error Handling

Dependencies

Timeline Estimate

References

Uh oh!

JMLX42 commented Jul 14, 2025

Uh oh!

JMLX42 commented Jul 14, 2025

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

JMLX42 commented Jul 15, 2025 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Uh oh!

JMLX42 commented Jul 16, 2025

Uh oh!

JMLX42 commented Jul 16, 2025

Uh oh!

jokemanfire commented Jul 17, 2025

Uh oh!

Uh oh!

JMLX42 commented Jul 22, 2025

Uh oh!

JMLX42 commented Jul 22, 2025

Uh oh!

JMLX42 commented Jul 22, 2025

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

JMLX42 commented Jul 23, 2025

Uh oh!

Uh oh!

Add support for `Tool.outputSchema` and `CallToolResult.structuredContent` #316

Add support for `Tool.outputSchema` and `CallToolResult.structuredContent` #316

JMLX42 commented Jul 14, 2025 •

edited

Loading

JMLX42 commented Jul 15, 2025 •

edited

Loading