diff --git a/trustgraph-flow/trustgraph/agent/react/agent_manager.py b/trustgraph-flow/trustgraph/agent/react/agent_manager.py
index 90bc445c..87cee33d 100644
--- a/trustgraph-flow/trustgraph/agent/react/agent_manager.py
+++ b/trustgraph-flow/trustgraph/agent/react/agent_manager.py
@@ -241,8 +241,8 @@ class AgentManager:
             logger.info("DEBUG: StreamingReActParser created")
 
             # Create async chunk callback that feeds parser and sends collected chunks
-            async def on_chunk(text):
-                logger.info(f"DEBUG: on_chunk called with {len(text)} chars")
+            async def on_chunk(text, end_of_stream):
+                logger.info(f"DEBUG: on_chunk called with {len(text)} chars, end_of_stream={end_of_stream}")
 
                 # Track what we had before
                 prev_thought_count = len(thought_chunks)