From aacc96064f6a12ced096b364078f6e0b3c2cb208 Mon Sep 17 00:00:00 2001
From: Cyber MacGeddon <cybermaggedon@gmail.com>
Date: Mon, 12 Jan 2026 17:41:54 +0000
Subject: [PATCH] Fix non-streaming failure in RAG services

---
 trustgraph-flow/trustgraph/retrieval/document_rag/rag.py | 1 +
 trustgraph-flow/trustgraph/retrieval/graph_rag/rag.py    | 1 +
 2 files changed, 2 insertions(+)

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}