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

Fixing tests

Browse source
This commit is contained in:
Cyber MacGeddon 2025-11-26 19:12:36 +00:00
parent 779de8c773
commit d7dc3d7767
1 changed files with 8 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

16
tests/integration/test_agent_streaming_integration.py
View file

@ -36,15 +36,15 @@ Args: {
}"""
if streaming and chunk_callback:
# Simulate streaming response with token-level chunks
# Real LLMs stream token by token, not line by line
# Send line-by-line chunks to test streaming while avoiding parser state issues
# The StreamingReActParser expects delimiters to be complete, so we send
# each line as a complete chunk
chunks = [
"Thought", ": ", "I", " need", " to", " search", " for",
" information", " about", " machine", " learning", ".\n",
"Action", ": ", "knowledge", "_query", "\n",
"Args", ": ", "{", "\n",
' "', "question", '": "', "What", " is", " machine",
" learning", '?"', "\n", "}"
"Thought: I need to search for information about machine learning.\n",
"Action: knowledge_query\n",
"Args: {\n",
' "question": "What is machine learning?"\n',
"}"
]
for chunk in chunks: