fix(realtime): update assistant message status to completed when audio is done

Author: DevAbdullah90

src/agents/realtime/session.py

@@ -275,11 +275,21 @@ async def on_event(self, event: RealtimeModelEvent) -> None:
                 )
             )
         elif event.type == "audio_done":
+            # Update history status to completed
+            item_id = event.item_id
+            for i, item in enumerate(self._history):
+                if item.item_id == item_id and hasattr(item, "status"):
+                    self._history[i] = item.model_copy(update={"status": "completed"})
+                    break
+
             await self._put_event(
                 RealtimeAudioEnd(
                     info=self._event_info, item_id=event.item_id, content_index=event.content_index
                 )
             )
+            await self._put_event(
+                RealtimeHistoryUpdated(info=self._event_info, history=self._history)
+            )
         elif event.type == "input_audio_transcription_completed":
             prev_len = len(self._history)
             self._history = RealtimeSession._get_new_history(self._history, event)

tests/realtime/test_session.py

@@ -382,8 +382,15 @@ async def test_audio_events_transformation(self, mock_model, mock_agent):
         done_event = RealtimeModelAudioDoneEvent(item_id="item_1", content_index=0)
         await session.on_event(done_event)
 
-        # Should have 6 events total (2 per event: raw + transformed)
-        assert session._event_queue.qsize() == 6
+        # Should have 7 events total:
+        # 1. raw audio_event
+        # 2. transformed RealtimeAudio
+        # 3. raw interrupted_event
+        # 4. transformed RealtimeAudioInterrupted
+        # 5. raw done_event
+        # 6. transformed RealtimeAudioEnd
+        # 7. RealtimeHistoryUpdated (from audio_done)
+        assert session._event_queue.qsize() == 7
 
         # Check audio event transformation
         await session._event_queue.get()  # raw event
@@ -401,6 +408,10 @@ async def test_audio_events_transformation(self, mock_model, mock_agent):
         done_session_event = await session._event_queue.get()
         assert isinstance(done_session_event, RealtimeAudioEnd)
 
+        # Check history updated event
+        history_updated_event = await session._event_queue.get()
+        assert isinstance(history_updated_event, RealtimeHistoryUpdated)
+
     @pytest.mark.asyncio
     async def test_turn_events_transformation(self, mock_model, mock_agent):
         """Test that turn start/end events are properly transformed"""