Add timestamp to bookkeeping data

Author: ajhai

llmstack/play/output_stream.py

@@ -4,6 +4,7 @@
 
 import asyncio
 import logging
+import time
 import uuid
 from collections import defaultdict
 from typing import Any, Dict, Type
@@ -230,7 +231,10 @@ def bookkeep(self, data: BaseModel) -> None:
         """
         Bookkeeping entry.
         """
-        self._bookkeeping_queue.put_nowait((self._stream_id, data.model_dump()))
+        timestamp = time.time()
+        timestamped_data = {**data.model_dump(), "timestamp": timestamp}
+
+        self._bookkeeping_queue.put_nowait((self._stream_id, timestamped_data))
 
     def error(self, error: Exception) -> None:
         """