fix: address CodeRabbit review comments on PR #3

- Import ResponseCreatedEvent and reset _reasoning_snapshot to "" when
  a ResponseCreatedEvent is received inside the retry stream loop, fixing
  the bug where snapshot text would be duplicated across retries
- In test_reasoning_delta_event_type_field: add found=False flag and
  assert found after the loop so the test properly fails when no
  ReasoningDeltaEvent is emitted

Author: Aditya Singh

src/agents/run_internal/run_loop.py

@@ -11,7 +11,7 @@
 from collections.abc import Awaitable, Callable, Mapping
 from typing import Any, TypeVar, cast
 
-from openai.types.responses import Response, ResponseCompletedEvent, ResponseOutputItemDoneEvent
+from openai.types.responses import Response, ResponseCompletedEvent, ResponseCreatedEvent, ResponseOutputItemDoneEvent
 from openai.types.responses.response_reasoning_text_delta_event import (
     ResponseReasoningTextDeltaEvent,
 )
@@ -1295,6 +1295,12 @@ async def rewind_model_request() -> None:
     async for event in retry_stream:
         streamed_result._event_queue.put_nowait(RawResponsesStreamEvent(data=event))
 
+        # Reset the reasoning snapshot at the start of each new response attempt
+        # (e.g., after a retry) so the snapshot field never contains stale text
+        # from a failed previous attempt.
+        if isinstance(event, ResponseCreatedEvent):
+            _reasoning_snapshot = ""
+
         # Emit a ReasoningDeltaEvent for reasoning/thinking deltas so consumers don't have
         # to unwrap the raw event themselves.
         if isinstance(event, ResponseReasoningSummaryTextDeltaEvent):

tests/test_reasoning_delta_stream_event.py

@@ -97,10 +97,13 @@ async def test_reasoning_delta_event_type_field() -> None:
     agent = Agent(name="A", model=model)
     result = Runner.run_streamed(agent, input="hi")
 
+    found = False
     async for event in result.stream_events():
         if isinstance(event, ReasoningDeltaEvent):
             assert event.type == "reasoning_delta"
+            found = True
             break
+    assert found, "Expected at least one ReasoningDeltaEvent but none were emitted"
 
 
 @pytest.mark.asyncio