From c6b8a4684c201310ddac2868c1f1f780ccc56805 Mon Sep 17 00:00:00 2001 From: Cyber MacGeddon Date: Wed, 27 Aug 2025 00:25:24 +0100 Subject: [PATCH] Use collection field from request when loading a knowledge core --- trustgraph-flow/trustgraph/cores/knowledge.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/trustgraph-flow/trustgraph/cores/knowledge.py b/trustgraph-flow/trustgraph/cores/knowledge.py index 898e8e15..77477343 100644 --- a/trustgraph-flow/trustgraph/cores/knowledge.py +++ b/trustgraph-flow/trustgraph/cores/knowledge.py @@ -248,6 +248,9 @@ class KnowledgeManager: await ge_pub.start() async def publish_triples(t): + # Override collection with request collection + if hasattr(t, 'metadata') and hasattr(t.metadata, 'collection'): + t.metadata.collection = request.collection or "default" await t_pub.send(None, t) logger.debug("Publishing triples...") @@ -260,6 +263,9 @@ class KnowledgeManager: ) async def publish_ge(g): + # Override collection with request collection + if hasattr(g, 'metadata') and hasattr(g.metadata, 'collection'): + g.metadata.collection = request.collection or "default" await ge_pub.send(None, g) logger.debug("Publishing graph embeddings...")