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

Make schema changes

Browse source
This commit is contained in:
Cyber MacGeddon 2024-12-30 11:20:25 +00:00
parent 9942f63773
commit 61a3e45e70
2 changed files with 13 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -37,12 +37,15 @@ chunk_ingest_queue = topic('chunk-load')
# Chunk embeddings are an embeddings associated with a text chunk
class ChunkEmbeddings(Record):
metadata = Metadata()
vectors = Array(Array(Double()))
chunk = Bytes()
class EntityContext(Record):
entity = Value()
context = String()
chunk_embeddings_ingest_queue = topic('chunk-embeddings-load')
class EntityContexts(Record):
metadata = Metadata()
entities = Array(EntityContext())
entity_contexts_ingest_queue = topic('entity-contexts-load')
############################################################################

7
trustgraph-base/trustgraph/schema/graph.py
View file

@ -9,10 +9,13 @@ from . metadata import Metadata
# Graph embeddings are embeddings associated with a graph entity
class EntityEmbeddings(Record):
entity = Value()
vectors = Array(Array(Double()))
class GraphEmbeddings(Record):
metadata = Metadata()
vectors = Array(Array(Double()))
entity = Value()
entities = Array(EntityEmbeddings())
graph_embeddings_store_queue = topic('graph-embeddings-store')