apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-07-16 16:51:02 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Update schema for doc embeddings

Browse source
This commit is contained in:
Cyber MacGeddon 2025-01-04 18:59:58 +00:00
parent e8f7600d74
commit 5227e3ac58
1 changed files with 15 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

15
trustgraph-base/trustgraph/schema/documents.py
View file

@ -35,6 +35,21 @@ chunk_ingest_queue = topic('chunk-load')
############################################################################
# Document embeddings are embeddings associated with a chunk
class ChunkEmbeddings(Record):
chunk = Bytes()
vectors = Array(Array(Double()))
# This is a 'batching' mechanism for the above data
class DocumentEmbeddings(Record):
metadata = Metadata()
chunks = Array(ChunkEmbeddings())
document_embeddings_store_queue = topic('document-embeddings-store')
############################################################################
# Doc embeddings query
class DocumentEmbeddingsRequest(Record):