apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-05-25 23:35:12 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Use collection field from request when loading a knowledge core

Browse source
This commit is contained in:
Cyber MacGeddon 2025-08-27 00:25:24 +01:00
parent e5b9b4976a
commit c6b8a4684c
1 changed files with 6 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

6
trustgraph-flow/trustgraph/cores/knowledge.py
View file

@ -248,6 +248,9 @@ class KnowledgeManager:
await ge_pub.start() await ge_pub.start()
async def publish_triples(t): 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) await t_pub.send(None, t)
logger.debug("Publishing triples...") logger.debug("Publishing triples...")
@ -260,6 +263,9 @@ class KnowledgeManager:
) )
async def publish_ge(g): 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) await ge_pub.send(None, g)
logger.debug("Publishing graph embeddings...") logger.debug("Publishing graph embeddings...")