File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff 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 )
You can’t perform that action at this time.
0 commit comments