Skip to content

Commit 8c2ebb9

Browse files
Copilothiyouga
andcommitted
Improve docstrings to clarify response format
Co-authored-by: hiyouga <16256802+hiyouga@users.noreply.github.com>
1 parent 9b8d05b commit 8c2ebb9

1 file changed

Lines changed: 6 additions & 3 deletions

File tree

src_py/agent_adapter/client.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -55,10 +55,12 @@ async def stream_generate(
5555
tool_choice: Tool usage preference (none/auto/required or list of tool names)
5656
5757
Yields:
58-
Message chunks from the streaming response
58+
Message chunks from the streaming response. Each chunk is a dictionary
59+
with the response data. The actual structure depends on the SDK implementation.
5960
"""
6061
# TODO: Implement actual SDK integration
61-
# For now, return a simple mock response
62+
# For now, return a simple mock response as a dict
63+
# Future implementations should return SDK-specific response objects
6264
yield {"type": "text", "content": "Mock response chunk"}
6365

6466
async def stream_generate_stateful(
@@ -87,7 +89,8 @@ async def stream_generate_stateful(
8789
tool_choice: Tool usage preference (none/auto/required or list of tool names)
8890
8991
Yields:
90-
Message chunks from the streaming response
92+
Message chunks from the streaming response. Each chunk is a dictionary
93+
with the response data. The actual structure depends on the SDK implementation.
9194
"""
9295
# Add message to history
9396
self._message_history.append(message)

0 commit comments

Comments
 (0)