chore: create union for working with message content (#939)

2025-06-12 00:01:33 -07:00 · 2025-06-12 00:01:33 -07:00 · d69b290b65
commit d69b290b65
parent 185fa49fd1
1 changed files with 25 additions and 51 deletions
--- a/examples/servers/simple-streamablehttp/mcp_simple_streamablehttp/server.py
+++ b/examples/servers/simple-streamablehttp/mcp_simple_streamablehttp/server.py
@ -45,14 +45,7 @@ def main(
    app = Server("mcp-streamable-http-demo")

    @app.call_tool()
-    async def call_tool(
-        name: str, arguments: dict
-    ) -> list[
-        types.TextContent
-        | types.ImageContent
-        | types.AudioContent
-        | types.EmbeddedResource
-    ]:
+    async def call_tool(name: str, arguments: dict) -> list[types.Content]:
        ctx = app.request_context
        interval = arguments.get("interval", 1.0)
        count = arguments.get("count", 5)