diff --git a/src/responses/schemas.py b/src/responses/schemas.py index 5736b1b..1db5499 100644 --- a/src/responses/schemas.py +++ b/src/responses/schemas.py @@ -71,7 +71,7 @@ class FunctionCallOutputItem(CustomBaseModel): # Union type for all output items # Type: ignore because Pydantic handles union types specially -OutputItem = MessageOutputItem | ReasoningOutputItem | FunctionCallOutputItem # type: ignore +OutputItem = MessageOutputItem | ReasoningOutputItem | FunctionCallOutputItem # ============================================================================ diff --git a/src/responses/streaming.py b/src/responses/streaming.py index 85d62e2..fa167f8 100644 --- a/src/responses/streaming.py +++ b/src/responses/streaming.py @@ -137,7 +137,7 @@ class StreamingCoordinator: async def stream_response_with_steward( self, - request: "ResponseRequest", # type: ignore # Forward reference + request: "ResponseRequest", # Forward reference ) -> AsyncGenerator[StreamEvent, None]: """ Stream response with Steward preprocessing and two-phase Tatlock execution. @@ -282,8 +282,8 @@ class StreamingCoordinator: async def _stream_direct_delegation( self, user_message: str, - recommendation: "StewardRecommendation", # type: ignore - tracker: "ToolCallTracker", # type: ignore + recommendation: "StewardRecommendation", + tracker: "ToolCallTracker", conversation_id: str, conversation_history: list | None = None, results: dict | None = None, @@ -387,7 +387,7 @@ class StreamingCoordinator: async def stream_response( self, - request: "ResponseRequest", # type: ignore # Forward reference + request: "ResponseRequest", # Forward reference ) -> AsyncGenerator[StreamEvent, None]: """ Coordinate streaming from agent to SSE events.