apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-04-27 17:36:23 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix agent streaming tool failure (#602)

Browse source 
* Fix agent streaming linkage

* Update tests
This commit is contained in:
cybermaggedon 2026-01-06 23:00:50 +00:00 • committed by GitHub
parent 53cf5fd7f9
commit 16a5cf966a
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 9 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

11
tests/integration/test_agent_streaming_integration.py
View file

@ -47,8 +47,9 @@ Args: {
"}"
]
for chunk in chunks:
await chunk_callback(chunk)
for i, chunk in enumerate(chunks):
is_final = (i == len(chunks) - 1)
await chunk_callback(chunk, is_final)
return full_text
else:
@ -312,8 +313,10 @@ Final Answer: AI is the simulation of human intelligence in machines."""
call_count += 1
if streaming and chunk_callback:
for chunk in response.split():
await chunk_callback(chunk + " ")
chunks = response.split()
for i, chunk in enumerate(chunks):
is_final = (i == len(chunks) - 1)
await chunk_callback(chunk + " ", is_final)
return response
return response