address comment

Author: sid-rl

src/runloop_api_client/sdk/async_axon.py

@@ -26,6 +26,9 @@ class AsyncAxon:
         >>> runloop = AsyncRunloopSDK()
         >>> axon = await runloop.axon.create()
         >>> await axon.publish(event_type="task_done", origin="AGENT_EVENT", payload="{}", source="my-agent")
+        >>> async with await axon.subscribe_sse() as stream:
+        ...     async for event in stream:
+        ...         print(event.event_type, event.payload)
     """
 
     def __init__(self, client: AsyncRunloop, axon_id: str) -> None:

src/runloop_api_client/sdk/axon.py

@@ -26,6 +26,9 @@ class Axon:
         >>> runloop = RunloopSDK()
         >>> axon = runloop.axon.create()
         >>> axon.publish(event_type="task_done", origin="AGENT_EVENT", payload="{}", source="my-agent")
+        >>> with axon.subscribe_sse() as stream:
+        ...     for event in stream:
+        ...         print(event.event_type, event.payload)
     """
 
     def __init__(self, client: Runloop, axon_id: str) -> None: