mirror of
https://github.com/trustgraph-ai/trustgraph.git
synced 2026-04-26 17:06:22 +02:00
38826c7de1
* trustgraph-base .chunks / .documents confusion in the API * Added tests, fixed test failures in code * Fix file dup error * Fix contract error
55 lines
No EOL
1.4 KiB
Python
55 lines
No EOL
1.4 KiB
Python
from pulsar.schema import Record, String, Integer, Array, Double
|
|
|
|
from ..core.primitives import Error, Value, Triple
|
|
from ..core.topic import topic
|
|
|
|
############################################################################
|
|
|
|
# Graph embeddings query
|
|
|
|
class GraphEmbeddingsRequest(Record):
|
|
vectors = Array(Array(Double()))
|
|
limit = Integer()
|
|
user = String()
|
|
collection = String()
|
|
|
|
class GraphEmbeddingsResponse(Record):
|
|
error = Error()
|
|
entities = Array(Value())
|
|
|
|
############################################################################
|
|
|
|
# Graph triples query
|
|
|
|
class TriplesQueryRequest(Record):
|
|
user = String()
|
|
collection = String()
|
|
s = Value()
|
|
p = Value()
|
|
o = Value()
|
|
limit = Integer()
|
|
|
|
class TriplesQueryResponse(Record):
|
|
error = Error()
|
|
triples = Array(Triple())
|
|
|
|
############################################################################
|
|
|
|
# Doc embeddings query
|
|
|
|
class DocumentEmbeddingsRequest(Record):
|
|
vectors = Array(Array(Double()))
|
|
limit = Integer()
|
|
user = String()
|
|
collection = String()
|
|
|
|
class DocumentEmbeddingsResponse(Record):
|
|
error = Error()
|
|
chunks = Array(String())
|
|
|
|
document_embeddings_request_queue = topic(
|
|
"non-persistent://trustgraph/document-embeddings-request"
|
|
)
|
|
document_embeddings_response_queue = topic(
|
|
"non-persistent://trustgraph/document-embeddings-response"
|
|
) |