apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-04-29 18:36:22 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

tg-put-kg-core command (#369)

Browse source
This commit is contained in:
cybermaggedon 2025-05-07 11:13:21 +01:00 • committed by GitHub
parent 8080b54328
commit f7123ac57f
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
7 changed files with 298 additions and 22 deletions
Download patch file Download diff file Expand all files Collapse all files

37
trustgraph-flow/trustgraph/gateway/dispatch/knowledge.py
View file

@ -1,13 +1,14 @@
import base64
from ... schema import KnowledgeRequest, KnowledgeResponse
from ... schema import KnowledgeRequest, KnowledgeResponse, Triples
from ... schema import GraphEmbeddings, Metadata, EntityEmbeddings
from ... schema import knowledge_request_queue
from ... schema import knowledge_response_queue
from . requestor import ServiceRequestor
from . serialize import serialize_graph_embeddings
from . serialize import serialize_triples
from . serialize import serialize_triples, to_subgraph, to_value
from . serialize import to_document_metadata, to_processing_metadata
class KnowledgeRequestor(ServiceRequestor):
@ -26,10 +27,42 @@ class KnowledgeRequestor(ServiceRequestor):
def to_request(self, body):
if "triples" in body:
triples = Triples(
metadata=Metadata(
id = body["triples"]["metadata"]["id"],
metadata = to_subgraph(body["triples"]["metadata"]["metadata"]),
user = body["triples"]["metadata"]["user"],
),
triples = to_subgraph(body["triples"]["triples"]),
)
else:
triples = None
if "graph-embeddings" in body:
ge = GraphEmbeddings(
metadata = Metadata(
id = body["graph-embeddings"]["metadata"]["id"],
metadata = to_subgraph(body["graph-embeddings"]["metadata"]["metadata"]),
user = body["graph-embeddings"]["metadata"]["user"],
),
entities=[
EntityEmbeddings(
entity = to_value(ent["entity"]),
vectors = ent["vectors"],
)
for ent in body["graph-embeddings"]["entities"]
]
)
else:
ge = None
return KnowledgeRequest(
operation = body.get("operation", None),
user = body.get("user", None),
id = body.get("id", None),
triples = triples,
graph_embeddings = ge,
)
def from_response(self, message):