Skip to content
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
160 changes: 79 additions & 81 deletions google/genai/_interactions/types/content_delta.py
Original file line number Diff line number Diff line change
Expand Up @@ -32,37 +32,37 @@
__all__ = [
"ContentDelta",
"Delta",
"DeltaTextDelta",
"DeltaImageDelta",
"DeltaAudioDelta",
"DeltaDocumentDelta",
"DeltaVideoDelta",
"DeltaThoughtSummaryDelta",
"DeltaThoughtSummaryDeltaContent",
"DeltaThoughtSignatureDelta",
"DeltaFunctionCallDelta",
"DeltaFunctionResultDelta",
"DeltaFunctionResultDeltaResult",
"DeltaFunctionResultDeltaResultItems",
"DeltaFunctionResultDeltaResultItemsItem",
"DeltaCodeExecutionCallDelta",
"DeltaCodeExecutionResultDelta",
"DeltaURLContextCallDelta",
"DeltaURLContextResultDelta",
"DeltaGoogleSearchCallDelta",
"DeltaGoogleSearchResultDelta",
"DeltaMCPServerToolCallDelta",
"DeltaMCPServerToolResultDelta",
"DeltaMCPServerToolResultDeltaResult",
"DeltaMCPServerToolResultDeltaResultItems",
"DeltaMCPServerToolResultDeltaResultItemsItem",
"DeltaFileSearchCallDelta",
"DeltaFileSearchResultDelta",
"DeltaFileSearchResultDeltaResult",
"DeltaText",
"DeltaImage",
"DeltaAudio",
"DeltaDocument",
"DeltaVideo",
"DeltaThoughtSummary",
"DeltaThoughtSummaryContent",
"DeltaThoughtSignature",
"DeltaFunctionCall",
"DeltaFunctionResult",
"DeltaFunctionResultResult",
"DeltaFunctionResultResultItems",
"DeltaFunctionResultResultItemsItem",
"DeltaCodeExecutionCall",
"DeltaCodeExecutionResult",
"DeltaURLContextCall",
"DeltaURLContextResult",
"DeltaGoogleSearchCall",
"DeltaGoogleSearchResult",
"DeltaMCPServerToolCall",
"DeltaMCPServerToolResult",
"DeltaMCPServerToolResultResult",
"DeltaMCPServerToolResultResultItems",
"DeltaMCPServerToolResultResultItemsItem",
"DeltaFileSearchCall",
"DeltaFileSearchResult",
"DeltaFileSearchResultResult",
]


class DeltaTextDelta(BaseModel):
class DeltaText(BaseModel):
text: str

type: Literal["text"]
Expand All @@ -71,7 +71,7 @@ class DeltaTextDelta(BaseModel):
"""Citation information for model-generated content."""


class DeltaImageDelta(BaseModel):
class DeltaImage(BaseModel):
type: Literal["image"]

data: Optional[str] = None
Expand All @@ -84,7 +84,7 @@ class DeltaImageDelta(BaseModel):
uri: Optional[str] = None


class DeltaAudioDelta(BaseModel):
class DeltaAudio(BaseModel):
type: Literal["audio"]

data: Optional[str] = None
Expand All @@ -94,7 +94,7 @@ class DeltaAudioDelta(BaseModel):
uri: Optional[str] = None


class DeltaDocumentDelta(BaseModel):
class DeltaDocument(BaseModel):
type: Literal["document"]

data: Optional[str] = None
Expand All @@ -104,7 +104,7 @@ class DeltaDocumentDelta(BaseModel):
uri: Optional[str] = None


class DeltaVideoDelta(BaseModel):
class DeltaVideo(BaseModel):
type: Literal["video"]

data: Optional[str] = None
Expand All @@ -129,26 +129,24 @@ class DeltaVideoDelta(BaseModel):
uri: Optional[str] = None


DeltaThoughtSummaryDeltaContent: TypeAlias = Annotated[
Union[TextContent, ImageContent], PropertyInfo(discriminator="type")
]
DeltaThoughtSummaryContent: TypeAlias = Annotated[Union[TextContent, ImageContent], PropertyInfo(discriminator="type")]


class DeltaThoughtSummaryDelta(BaseModel):
class DeltaThoughtSummary(BaseModel):
type: Literal["thought_summary"]

content: Optional[DeltaThoughtSummaryDeltaContent] = None
content: Optional[DeltaThoughtSummaryContent] = None
"""A text content block."""


class DeltaThoughtSignatureDelta(BaseModel):
class DeltaThoughtSignature(BaseModel):
type: Literal["thought_signature"]

signature: Optional[str] = None
"""Signature to match the backend source to be part of the generation."""


class DeltaFunctionCallDelta(BaseModel):
class DeltaFunctionCall(BaseModel):
id: str
"""A unique ID for this specific tool call."""

Expand All @@ -162,21 +160,21 @@ class DeltaFunctionCallDelta(BaseModel):
"""A signature hash for backend validation."""


DeltaFunctionResultDeltaResultItemsItem: TypeAlias = Union[TextContent, ImageContent]
DeltaFunctionResultResultItemsItem: TypeAlias = Union[TextContent, ImageContent]


class DeltaFunctionResultDeltaResultItems(BaseModel):
items: Optional[List[DeltaFunctionResultDeltaResultItemsItem]] = None
class DeltaFunctionResultResultItems(BaseModel):
items: Optional[List[DeltaFunctionResultResultItemsItem]] = None


DeltaFunctionResultDeltaResult: TypeAlias = Union[DeltaFunctionResultDeltaResultItems, str, object]
DeltaFunctionResultResult: TypeAlias = Union[DeltaFunctionResultResultItems, str, object]


class DeltaFunctionResultDelta(BaseModel):
class DeltaFunctionResult(BaseModel):
call_id: str
"""ID to match the ID from the function call block."""

result: DeltaFunctionResultDeltaResult
result: DeltaFunctionResultResult
"""Tool call result delta."""

type: Literal["function_result"]
Expand All @@ -189,7 +187,7 @@ class DeltaFunctionResultDelta(BaseModel):
"""A signature hash for backend validation."""


class DeltaCodeExecutionCallDelta(BaseModel):
class DeltaCodeExecutionCall(BaseModel):
id: str
"""A unique ID for this specific tool call."""

Expand All @@ -202,7 +200,7 @@ class DeltaCodeExecutionCallDelta(BaseModel):
"""A signature hash for backend validation."""


class DeltaCodeExecutionResultDelta(BaseModel):
class DeltaCodeExecutionResult(BaseModel):
call_id: str
"""ID to match the ID from the function call block."""

Expand All @@ -216,7 +214,7 @@ class DeltaCodeExecutionResultDelta(BaseModel):
"""A signature hash for backend validation."""


class DeltaURLContextCallDelta(BaseModel):
class DeltaURLContextCall(BaseModel):
id: str
"""A unique ID for this specific tool call."""

Expand All @@ -229,7 +227,7 @@ class DeltaURLContextCallDelta(BaseModel):
"""A signature hash for backend validation."""


class DeltaURLContextResultDelta(BaseModel):
class DeltaURLContextResult(BaseModel):
call_id: str
"""ID to match the ID from the function call block."""

Expand All @@ -243,7 +241,7 @@ class DeltaURLContextResultDelta(BaseModel):
"""A signature hash for backend validation."""


class DeltaGoogleSearchCallDelta(BaseModel):
class DeltaGoogleSearchCall(BaseModel):
id: str
"""A unique ID for this specific tool call."""

Expand All @@ -256,7 +254,7 @@ class DeltaGoogleSearchCallDelta(BaseModel):
"""A signature hash for backend validation."""


class DeltaGoogleSearchResultDelta(BaseModel):
class DeltaGoogleSearchResult(BaseModel):
call_id: str
"""ID to match the ID from the function call block."""

Expand All @@ -270,7 +268,7 @@ class DeltaGoogleSearchResultDelta(BaseModel):
"""A signature hash for backend validation."""


class DeltaMCPServerToolCallDelta(BaseModel):
class DeltaMCPServerToolCall(BaseModel):
id: str
"""A unique ID for this specific tool call."""

Expand All @@ -286,21 +284,21 @@ class DeltaMCPServerToolCallDelta(BaseModel):
"""A signature hash for backend validation."""


DeltaMCPServerToolResultDeltaResultItemsItem: TypeAlias = Union[TextContent, ImageContent]
DeltaMCPServerToolResultResultItemsItem: TypeAlias = Union[TextContent, ImageContent]


class DeltaMCPServerToolResultDeltaResultItems(BaseModel):
items: Optional[List[DeltaMCPServerToolResultDeltaResultItemsItem]] = None
class DeltaMCPServerToolResultResultItems(BaseModel):
items: Optional[List[DeltaMCPServerToolResultResultItemsItem]] = None


DeltaMCPServerToolResultDeltaResult: TypeAlias = Union[DeltaMCPServerToolResultDeltaResultItems, str, object]
DeltaMCPServerToolResultResult: TypeAlias = Union[DeltaMCPServerToolResultResultItems, str, object]


class DeltaMCPServerToolResultDelta(BaseModel):
class DeltaMCPServerToolResult(BaseModel):
call_id: str
"""ID to match the ID from the function call block."""

result: DeltaMCPServerToolResultDeltaResult
result: DeltaMCPServerToolResultResult
"""Tool call result delta."""

type: Literal["mcp_server_tool_result"]
Expand All @@ -313,7 +311,7 @@ class DeltaMCPServerToolResultDelta(BaseModel):
"""A signature hash for backend validation."""


class DeltaFileSearchCallDelta(BaseModel):
class DeltaFileSearchCall(BaseModel):
id: str
"""A unique ID for this specific tool call."""

Expand All @@ -323,7 +321,7 @@ class DeltaFileSearchCallDelta(BaseModel):
"""A signature hash for backend validation."""


class DeltaFileSearchResultDeltaResult(BaseModel):
class DeltaFileSearchResultResult(BaseModel):
"""The result of the File Search."""

file_search_store: Optional[str] = None
Expand All @@ -336,33 +334,33 @@ class DeltaFileSearchResultDeltaResult(BaseModel):
"""The title of the search result."""


class DeltaFileSearchResultDelta(BaseModel):
class DeltaFileSearchResult(BaseModel):
type: Literal["file_search_result"]

result: Optional[List[DeltaFileSearchResultDeltaResult]] = None
result: Optional[List[DeltaFileSearchResultResult]] = None


Delta: TypeAlias = Annotated[
Union[
DeltaTextDelta,
DeltaImageDelta,
DeltaAudioDelta,
DeltaDocumentDelta,
DeltaVideoDelta,
DeltaThoughtSummaryDelta,
DeltaThoughtSignatureDelta,
DeltaFunctionCallDelta,
DeltaFunctionResultDelta,
DeltaCodeExecutionCallDelta,
DeltaCodeExecutionResultDelta,
DeltaURLContextCallDelta,
DeltaURLContextResultDelta,
DeltaGoogleSearchCallDelta,
DeltaGoogleSearchResultDelta,
DeltaMCPServerToolCallDelta,
DeltaMCPServerToolResultDelta,
DeltaFileSearchCallDelta,
DeltaFileSearchResultDelta,
DeltaText,
DeltaImage,
DeltaAudio,
DeltaDocument,
DeltaVideo,
DeltaThoughtSummary,
DeltaThoughtSignature,
DeltaFunctionCall,
DeltaFunctionResult,
DeltaCodeExecutionCall,
DeltaCodeExecutionResult,
DeltaURLContextCall,
DeltaURLContextResult,
DeltaGoogleSearchCall,
DeltaGoogleSearchResult,
DeltaMCPServerToolCall,
DeltaMCPServerToolResult,
DeltaFileSearchCall,
DeltaFileSearchResult,
],
PropertyInfo(discriminator="type"),
]
Expand Down