apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-07-24 12:41:02 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

Updated streaming tests

Browse source
This commit is contained in:
Cyber MacGeddon 2026-01-06 19:47:06 +00:00
parent 10f9489f97
commit b34c29ad15
1 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
trustgraph-base/trustgraph/base/prompt_client.py
View file

@ -51,7 +51,8 @@ class PromptClient(RequestResponse):
end_stream = getattr(resp, 'end_of_stream', False) 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 last_text = resp.text
# Call chunk callback if provided with both chunk and end_of_stream flag # Call chunk callback if provided with both chunk and end_of_stream flag
if chunk_callback: if chunk_callback: