Add schema for querying graph embeddings

This commit is contained in:
Cyber MacGeddon 2024-08-12 16:48:17 +01:00
parent 697214635b
commit fc713057e3

View file

@ -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 # Graph triples
class Triple(Record): class Triple(Record):