diff --git a/trustgraph-flow/trustgraph/retrieval/document_rag/rag.py b/trustgraph-flow/trustgraph/retrieval/document_rag/rag.py
index 14d71d97..6490562a 100755
--- a/trustgraph-flow/trustgraph/retrieval/document_rag/rag.py
+++ b/trustgraph-flow/trustgraph/retrieval/document_rag/rag.py
@@ -128,6 +128,7 @@ class Processor(FlowProcessor):
                 await flow("response").send(
                     DocumentRagResponse(
                         response = response,
+                        end_of_stream = True,
                         error = None
                     ),
                     properties = {"id": id}
diff --git a/trustgraph-flow/trustgraph/retrieval/graph_rag/rag.py b/trustgraph-flow/trustgraph/retrieval/graph_rag/rag.py
index d159dbae..d8bfbddb 100755
--- a/trustgraph-flow/trustgraph/retrieval/graph_rag/rag.py
+++ b/trustgraph-flow/trustgraph/retrieval/graph_rag/rag.py
@@ -171,6 +171,7 @@ class Processor(FlowProcessor):
                 await flow("response").send(
                     GraphRagResponse(
                         response = response,
+                        end_of_stream = True,
                         error = None
                     ),
                     properties = {"id": id}