From fc713057e3d34ab51c3b0e375a4fa29483919e4a Mon Sep 17 00:00:00 2001 From: Cyber MacGeddon Date: Mon, 12 Aug 2024 16:48:17 +0100 Subject: [PATCH] Add schema for querying graph embeddings --- trustgraph/schema.py | 17 +++++++++++++++++ 1 file changed, 17 insertions(+) diff --git a/trustgraph/schema.py b/trustgraph/schema.py index a84a5880..02e799d7 100644 --- a/trustgraph/schema.py +++ b/trustgraph/schema.py @@ -71,6 +71,23 @@ graph_embeddings_store_queue = topic('graph-embeddings-store') ############################################################################ +# Graph embeddings query + +class GraphEmbeddingsQueryRequest(Record): + vectors = Array(Array(Double())) + +class GraphEmbeddingsQueryResponse(Record): + vectors = Array(Value()) + +graph_embeddings_request_queue = topic( + 'graph-embeddings', kind='non-persistent', namespace='request' +) +graph_embeddings_response_queue = topic( + 'graph-embeddings-response', kind='non-persistent', namespace='response', +) + +############################################################################ + # Graph triples class Triple(Record):