fix: closing streaming contexts for reasoning at done event

Author: IdkwhatImD0ing

examples/tutorials/10_agentic/10_temporal/010_agent_chat/project/workflow.py

@@ -241,7 +241,7 @@ async def on_task_event_send(self, params: SendEventParams) -> None:
                     # Ask the model to include a short reasoning summary
                     reasoning=Reasoning(effort="medium", summary="detailed"),
                 ),
-                tools=[CALCULATOR_TOOL],
+                # tools=[CALCULATOR_TOOL],
             )
             if self._state:
                 # Update the state with the final input list if available

src/agentex/lib/core/services/adk/providers/openai.py

@@ -830,16 +830,26 @@ async def run_agent_streamed_auto_send(
                                 item_id = event.data.item_id
                                 summary_index = event.data.summary_index
 
-                                # Note: We don't close the streaming context here since there might be more deltas
-                                # The context will be closed when the reasoning item is completely done
+                                # Finish the streaming context for this reasoning item
+                                if item_id in item_id_to_streaming_context and item_id in unclosed_item_ids:
+                                    streaming_context = item_id_to_streaming_context[
+                                        item_id
+                                    ]
+                                    await streaming_context.close()
+                                    unclosed_item_ids.remove(item_id)
 
                             elif isinstance(event.data, ResponseReasoningTextDoneEvent):
                                 # Handle reasoning content text completion
                                 item_id = event.data.item_id
                                 content_index = event.data.content_index
 
-                                # Note: We don't close the streaming context here since there might be more deltas
-                                # The context will be closed when the reasoning item is completely done
+                                # Finish the streaming context for this reasoning item
+                                if item_id in item_id_to_streaming_context and item_id in unclosed_item_ids:
+                                    streaming_context = item_id_to_streaming_context[
+                                        item_id
+                                    ]
+                                    await streaming_context.close()
+                                    unclosed_item_ids.remove(item_id)
 
                             elif isinstance(event.data, ResponseOutputItemDoneEvent):
                                 # Handle item completion