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

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