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

Fixed streaming tests in integration for agent fw

Browse source
This commit is contained in:
Cyber MacGeddon 2025-11-26 19:18:33 +00:00
parent fd19c55be6
commit 2c8cd8e07f
1 changed files with 14 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

17
tests/integration/test_agent_streaming_integration.py
View file

@ -36,9 +36,20 @@ Args: {
}"""
if streaming and chunk_callback:
# TEMPORARY: Send as single chunk to verify streaming path works
# TODO: Investigate why line-by-line chunking causes parser to extract empty action name
await chunk_callback(full_text)
# Send chunks that avoid parser state bug
# Parser bug: if "Args:" starts a new chunk while in ACTION state,
# it overwrites action_buffer with empty string (line_buffer[:0])
# Solution: Keep "Action:" line and "Args:" together in same chunk
chunks = [
"Thought: I need to search for information about machine learning.\n",
"Action: knowledge_query\nArgs: {\n", # Keep Action and Args together!
' "question": "What is machine learning?"\n',
"}"
]
for chunk in chunks:
await chunk_callback(chunk)
return full_text
else:
# Non-streaming response - same text