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

This commit is contained in:
Luca Chang 2025-06-12 00:01:33 -07:00 • committed by GitHub
commit d69b290b65
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -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