mirror of
https://github.com/aljazceru/mcp-python-sdk.git
synced 2025-12-19 23:04:25 +01:00
chore: create union for working with message content (#939)
This commit is contained in:
@@ -667,11 +667,14 @@ class EmbeddedResource(BaseModel):
|
||||
model_config = ConfigDict(extra="allow")
|
||||
|
||||
|
||||
Content = TextContent | ImageContent | AudioContent | EmbeddedResource
|
||||
|
||||
|
||||
class PromptMessage(BaseModel):
|
||||
"""Describes a message returned as part of a prompt."""
|
||||
|
||||
role: Role
|
||||
content: TextContent | ImageContent | AudioContent | EmbeddedResource
|
||||
content: Content
|
||||
model_config = ConfigDict(extra="allow")
|
||||
|
||||
|
||||
@@ -787,7 +790,7 @@ class CallToolRequest(Request[CallToolRequestParams, Literal["tools/call"]]):
|
||||
class CallToolResult(Result):
|
||||
"""The server's response to a tool call."""
|
||||
|
||||
content: list[TextContent | ImageContent | AudioContent | EmbeddedResource]
|
||||
content: list[Content]
|
||||
isError: bool = False
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user