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
parent 185fa49fd1
commit d69b290b65
9 changed files with 25 additions and 51 deletions

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