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

Core entity context flow in place

Browse source
This commit is contained in:
Cyber MacGeddon 2024-12-30 11:53:05 +00:00
parent 61a3e45e70
commit c3904b6772
6 changed files with 92 additions and 81 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -35,20 +35,6 @@ chunk_ingest_queue = topic('chunk-load')
############################################################################
# Chunk embeddings are an embeddings associated with a text chunk
class EntityContext(Record):
entity = Value()
context = String()
class EntityContexts(Record):
metadata = Metadata()
entities = Array(EntityContext())
entity_contexts_ingest_queue = topic('entity-contexts-load')
############################################################################
# Doc embeddings query
class DocumentEmbeddingsRequest(Record):
@ -65,3 +51,4 @@ document_embeddings_request_queue = topic(
document_embeddings_response_queue = topic(
'doc-embeddings', kind='non-persistent', namespace='response',
)

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

@ -7,12 +7,28 @@ from . metadata import Metadata
############################################################################
# Entity context are an entity associated with textual context
class EntityContext(Record):
entity = Value()
context = String()
# This is a 'batching' mechanism for the above data
class EntityContexts(Record):
metadata = Metadata()
entities = Array(EntityContext())
entity_contexts_ingest_queue = topic('entity-contexts-load')
############################################################################
# Graph embeddings are embeddings associated with a graph entity
class EntityEmbeddings(Record):
entity = Value()
vectors = Array(Array(Double()))
# This is a 'batching' mechanism for the above data
class GraphEmbeddings(Record):
metadata = Metadata()
entities = Array(EntityEmbeddings())