fix: fix stream event model reference

Author: stainless-app[bot]

.stats.yml

@@ -1,4 +1,4 @@
 configured_endpoints: 109
 openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/llamastack%2Fllama-stack-client-4337a6181c2db17737133e944b4b660a5e00ea10dce6be3252918e39451e9b5f.yml
 openapi_spec_hash: a0bc8f4b5f45bc5741fed8eaa61171c3
-config_hash: 03aab396899c7d9aa3fba867ce54824b
+config_hash: 47ef2eb62d188340f22eb6dea3693f15

src/llama_stack_client/resources/alpha/agents/turn.py

@@ -22,6 +22,7 @@
 from ....types.alpha.agents import turn_create_params, turn_resume_params
 from ....types.alpha.agents.turn import Turn
 from ....types.alpha.tool_response_param import ToolResponseParam
+from ....types.alpha.agents.agent_turn_response_stream_chunk import AgentTurnResponseStreamChunk
 
 __all__ = ["TurnResource", "AsyncTurnResource"]
 
@@ -108,7 +109,7 @@ def create(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> Stream[object]:
+    ) -> Stream[AgentTurnResponseStreamChunk]:
         """
         Create a new turn for an agent.
 
@@ -153,7 +154,7 @@ def create(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> Turn | Stream[object]:
+    ) -> Turn | Stream[AgentTurnResponseStreamChunk]:
         """
         Create a new turn for an agent.
 
@@ -198,7 +199,7 @@ def create(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> Turn | Stream[object]:
+    ) -> Turn | Stream[AgentTurnResponseStreamChunk]:
         if not agent_id:
             raise ValueError(f"Expected a non-empty value for `agent_id` but received {agent_id!r}")
         if not session_id:
@@ -222,7 +223,7 @@ def create(
             ),
             cast_to=Turn,
             stream=stream or False,
-            stream_cls=Stream[object],
+            stream_cls=Stream[AgentTurnResponseStreamChunk],
         )
 
     def retrieve(
@@ -317,7 +318,7 @@ def resume(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> Stream[object]:
+    ) -> Stream[AgentTurnResponseStreamChunk]:
         """Resume an agent turn with executed tool call responses.
 
         When a Turn has the
@@ -355,7 +356,7 @@ def resume(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> Turn | Stream[object]:
+    ) -> Turn | Stream[AgentTurnResponseStreamChunk]:
         """Resume an agent turn with executed tool call responses.
 
         When a Turn has the
@@ -393,7 +394,7 @@ def resume(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> Turn | Stream[object]:
+    ) -> Turn | Stream[AgentTurnResponseStreamChunk]:
         if not agent_id:
             raise ValueError(f"Expected a non-empty value for `agent_id` but received {agent_id!r}")
         if not session_id:
@@ -416,7 +417,7 @@ def resume(
             ),
             cast_to=Turn,
             stream=stream or False,
-            stream_cls=Stream[object],
+            stream_cls=Stream[AgentTurnResponseStreamChunk],
         )
 
 
@@ -502,7 +503,7 @@ async def create(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> AsyncStream[object]:
+    ) -> AsyncStream[AgentTurnResponseStreamChunk]:
         """
         Create a new turn for an agent.
 
@@ -547,7 +548,7 @@ async def create(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> Turn | AsyncStream[object]:
+    ) -> Turn | AsyncStream[AgentTurnResponseStreamChunk]:
         """
         Create a new turn for an agent.
 
@@ -592,7 +593,7 @@ async def create(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> Turn | AsyncStream[object]:
+    ) -> Turn | AsyncStream[AgentTurnResponseStreamChunk]:
         if not agent_id:
             raise ValueError(f"Expected a non-empty value for `agent_id` but received {agent_id!r}")
         if not session_id:
@@ -616,7 +617,7 @@ async def create(
             ),
             cast_to=Turn,
             stream=stream or False,
-            stream_cls=AsyncStream[object],
+            stream_cls=AsyncStream[AgentTurnResponseStreamChunk],
         )
 
     async def retrieve(
@@ -711,7 +712,7 @@ async def resume(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> AsyncStream[object]:
+    ) -> AsyncStream[AgentTurnResponseStreamChunk]:
         """Resume an agent turn with executed tool call responses.
 
         When a Turn has the
@@ -749,7 +750,7 @@ async def resume(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> Turn | AsyncStream[object]:
+    ) -> Turn | AsyncStream[AgentTurnResponseStreamChunk]:
         """Resume an agent turn with executed tool call responses.
 
         When a Turn has the
@@ -787,7 +788,7 @@ async def resume(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> Turn | AsyncStream[object]:
+    ) -> Turn | AsyncStream[AgentTurnResponseStreamChunk]:
         if not agent_id:
             raise ValueError(f"Expected a non-empty value for `agent_id` but received {agent_id!r}")
         if not session_id:
@@ -810,7 +811,7 @@ async def resume(
             ),
             cast_to=Turn,
             stream=stream or False,
-            stream_cls=AsyncStream[object],
+            stream_cls=AsyncStream[AgentTurnResponseStreamChunk],
         )
 
 

src/llama_stack_client/types/alpha/agents/__init__.py

@@ -7,8 +7,10 @@
 from .turn_create_params import TurnCreateParams as TurnCreateParams
 from .turn_resume_params import TurnResumeParams as TurnResumeParams
 from .session_list_params import SessionListParams as SessionListParams
+from .turn_response_event import TurnResponseEvent as TurnResponseEvent
 from .session_create_params import SessionCreateParams as SessionCreateParams
 from .session_list_response import SessionListResponse as SessionListResponse
 from .step_retrieve_response import StepRetrieveResponse as StepRetrieveResponse
 from .session_create_response import SessionCreateResponse as SessionCreateResponse
 from .session_retrieve_params import SessionRetrieveParams as SessionRetrieveParams
+from .agent_turn_response_stream_chunk import AgentTurnResponseStreamChunk as AgentTurnResponseStreamChunk

src/llama_stack_client/types/alpha/agents/agent_turn_response_stream_chunk.py

@@ -0,0 +1,11 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from ...._models import BaseModel
+from .turn_response_event import TurnResponseEvent
+
+__all__ = ["AgentTurnResponseStreamChunk"]
+
+
+class AgentTurnResponseStreamChunk(BaseModel):
+    event: TurnResponseEvent
+    """Individual event in the agent turn response stream"""

src/llama_stack_client/types/alpha/agents/turn_response_event.py

@@ -0,0 +1,160 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import Dict, List, Union, Optional
+from typing_extensions import Literal, Annotated, TypeAlias
+
+from .turn import Turn
+from ...._utils import PropertyInfo
+from ...._models import BaseModel
+from ..inference_step import InferenceStep
+from ..shield_call_step import ShieldCallStep
+from ...shared.tool_call import ToolCall
+from ..tool_execution_step import ToolExecutionStep
+from ..memory_retrieval_step import MemoryRetrievalStep
+
+__all__ = [
+    "TurnResponseEvent",
+    "Payload",
+    "PayloadAgentTurnResponseStepStartPayload",
+    "PayloadAgentTurnResponseStepProgressPayload",
+    "PayloadAgentTurnResponseStepProgressPayloadDelta",
+    "PayloadAgentTurnResponseStepProgressPayloadDeltaTextDelta",
+    "PayloadAgentTurnResponseStepProgressPayloadDeltaImageDelta",
+    "PayloadAgentTurnResponseStepProgressPayloadDeltaToolCallDelta",
+    "PayloadAgentTurnResponseStepProgressPayloadDeltaToolCallDeltaToolCall",
+    "PayloadAgentTurnResponseStepCompletePayload",
+    "PayloadAgentTurnResponseStepCompletePayloadStepDetails",
+    "PayloadAgentTurnResponseTurnStartPayload",
+    "PayloadAgentTurnResponseTurnCompletePayload",
+    "PayloadAgentTurnResponseTurnAwaitingInputPayload",
+]
+
+
+class PayloadAgentTurnResponseStepStartPayload(BaseModel):
+    event_type: Literal["step_start"]
+    """Type of event being reported"""
+
+    step_id: str
+    """Unique identifier for the step within a turn"""
+
+    step_type: Literal["inference", "tool_execution", "shield_call", "memory_retrieval"]
+    """Type of step being executed"""
+
+    metadata: Optional[Dict[str, Union[bool, float, str, List[object], object, None]]] = None
+    """(Optional) Additional metadata for the step"""
+
+
+class PayloadAgentTurnResponseStepProgressPayloadDeltaTextDelta(BaseModel):
+    text: str
+    """The incremental text content"""
+
+    type: Literal["text"]
+    """Discriminator type of the delta. Always "text" """
+
+
+class PayloadAgentTurnResponseStepProgressPayloadDeltaImageDelta(BaseModel):
+    image: str
+    """The incremental image data as bytes"""
+
+    type: Literal["image"]
+    """Discriminator type of the delta. Always "image" """
+
+
+PayloadAgentTurnResponseStepProgressPayloadDeltaToolCallDeltaToolCall: TypeAlias = Union[str, ToolCall]
+
+
+class PayloadAgentTurnResponseStepProgressPayloadDeltaToolCallDelta(BaseModel):
+    parse_status: Literal["started", "in_progress", "failed", "succeeded"]
+    """Current parsing status of the tool call"""
+
+    tool_call: PayloadAgentTurnResponseStepProgressPayloadDeltaToolCallDeltaToolCall
+    """Either an in-progress tool call string or the final parsed tool call"""
+
+    type: Literal["tool_call"]
+    """Discriminator type of the delta. Always "tool_call" """
+
+
+PayloadAgentTurnResponseStepProgressPayloadDelta: TypeAlias = Annotated[
+    Union[
+        PayloadAgentTurnResponseStepProgressPayloadDeltaTextDelta,
+        PayloadAgentTurnResponseStepProgressPayloadDeltaImageDelta,
+        PayloadAgentTurnResponseStepProgressPayloadDeltaToolCallDelta,
+    ],
+    PropertyInfo(discriminator="type"),
+]
+
+
+class PayloadAgentTurnResponseStepProgressPayload(BaseModel):
+    delta: PayloadAgentTurnResponseStepProgressPayloadDelta
+    """Incremental content changes during step execution"""
+
+    event_type: Literal["step_progress"]
+    """Type of event being reported"""
+
+    step_id: str
+    """Unique identifier for the step within a turn"""
+
+    step_type: Literal["inference", "tool_execution", "shield_call", "memory_retrieval"]
+    """Type of step being executed"""
+
+
+PayloadAgentTurnResponseStepCompletePayloadStepDetails: TypeAlias = Annotated[
+    Union[InferenceStep, ToolExecutionStep, ShieldCallStep, MemoryRetrievalStep],
+    PropertyInfo(discriminator="step_type"),
+]
+
+
+class PayloadAgentTurnResponseStepCompletePayload(BaseModel):
+    event_type: Literal["step_complete"]
+    """Type of event being reported"""
+
+    step_details: PayloadAgentTurnResponseStepCompletePayloadStepDetails
+    """Complete details of the executed step"""
+
+    step_id: str
+    """Unique identifier for the step within a turn"""
+
+    step_type: Literal["inference", "tool_execution", "shield_call", "memory_retrieval"]
+    """Type of step being executed"""
+
+
+class PayloadAgentTurnResponseTurnStartPayload(BaseModel):
+    event_type: Literal["turn_start"]
+    """Type of event being reported"""
+
+    turn_id: str
+    """Unique identifier for the turn within a session"""
+
+
+class PayloadAgentTurnResponseTurnCompletePayload(BaseModel):
+    event_type: Literal["turn_complete"]
+    """Type of event being reported"""
+
+    turn: Turn
+    """Complete turn data including all steps and results"""
+
+
+class PayloadAgentTurnResponseTurnAwaitingInputPayload(BaseModel):
+    event_type: Literal["turn_awaiting_input"]
+    """Type of event being reported"""
+
+    turn: Turn
+    """Turn data when waiting for external tool responses"""
+
+
+Payload: TypeAlias = Annotated[
+    Union[
+        PayloadAgentTurnResponseStepStartPayload,
+        PayloadAgentTurnResponseStepProgressPayload,
+        PayloadAgentTurnResponseStepCompletePayload,
+        PayloadAgentTurnResponseTurnStartPayload,
+        PayloadAgentTurnResponseTurnCompletePayload,
+        PayloadAgentTurnResponseTurnAwaitingInputPayload,
+    ],
+    PropertyInfo(discriminator="event_type"),
+]
+
+
+class TurnResponseEvent(BaseModel):
+    payload: Payload
+    """Event-specific payload containing event data"""