Yiled json HIL in stream

openbraininstitute · Jan 14, 2025 · 5ac20b8 · 5ac20b8
1 parent 151e028
commit 5ac20b8
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/neuroagent/stream.py b/src/neuroagent/stream.py
@@ -1,5 +1,6 @@
 """Wrapper around streaming methods to reinitiate connections due to the way fastAPI StreamingResponse works."""
 
+import json
 from typing import Any, AsyncIterator
 
 from fastapi import Request
@@ -51,7 +52,7 @@ async def stream_agent_response(
             yield chunk
         # Final chunk that contains the whole response
         elif chunk.hil_messages:
-            yield str(
+            yield json.dumps(
                 [hil_message.model_dump_json() for hil_message in chunk.hil_messages]
             )