Add docstring

Author: jiwon-oai

chatkit/agents.py

@@ -367,7 +367,18 @@ async def stream_agent_response(
     *,
     base64_to_generated_image: Callable[[str], Awaitable[GeneratedImage]] | None = None,
 ) -> AsyncIterator[ThreadStreamEvent]:
-    """Convert a streamed Agents SDK run into ChatKit ThreadStreamEvents."""
+    """Convert a streamed Agents SDK run into ChatKit ThreadStreamEvents.
+
+    Args:
+        context: The AgentContext to use for the stream.
+        result: The RunResultStreaming to convert.
+        base64_to_generated_image: A callable to convert a base64-encoded image generation result
+            to a GeneratedImage. If not provided, the generated base64 image data will be saved
+            as part of the GeneratedImageItem.
+
+    Returns:
+        An async iterator that yields thread stream events representing the run result.
+    """
     current_item_id = None
     current_tool_call = None
     ctx = context
@@ -731,7 +742,7 @@ async def generated_image_to_input(
         self, item: GeneratedImageItem
     ) -> TResponseInputItem | list[TResponseInputItem] | None:
         """
-        Convert a GeneratedImageItem into a message content part to send to the model.
+        Convert a GeneratedImageItem into input item(s) to send to the model.
         Required when generated images are enabled.
         """
         raise NotImplementedError(