mirror of
https://github.com/trustgraph-ai/trustgraph.git
synced 2026-07-23 20:21:03 +02:00
Updated streaming tests
This commit is contained in:
parent
10f9489f97
commit
b34c29ad15
1 changed files with 2 additions and 1 deletions
|
|
@ -51,7 +51,8 @@ class PromptClient(RequestResponse):
|
|||
|
||||
end_stream = getattr(resp, 'end_of_stream', False)
|
||||
|
||||
if resp.text:
|
||||
# Always call callback if there's text OR if it's the final message
|
||||
if resp.text is not None:
|
||||
last_text = resp.text
|
||||
# Call chunk callback if provided with both chunk and end_of_stream flag
|
||||
if chunk_callback:
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue