apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-07-23 20:21:03 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

Update tests

Browse source
This commit is contained in:
Cyber MacGeddon 2026-01-06 22:58:44 +00:00
parent 04f870ee2a
commit 63f41cff1a
1 changed files with 7 additions and 4 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: for i, chunk in enumerate(chunks):
await chunk_callback(chunk) is_final = (i == len(chunks) - 1)
await chunk_callback(chunk, is_final)
return full_text return full_text
else: else:
@ -312,8 +313,10 @@ Final Answer: AI is the simulation of human intelligence in machines."""
call_count += 1 call_count += 1
if streaming and chunk_callback: if streaming and chunk_callback:
for chunk in response.split(): chunks = response.split()
await chunk_callback(chunk + " ") for i, chunk in enumerate(chunks):
is_final = (i == len(chunks) - 1)
await chunk_callback(chunk + " ", is_final)
return response return response
return response return response