mirror of
https://github.com/trustgraph-ai/trustgraph.git
synced 2026-05-06 22:02:37 +02:00
9b91d5eee3
* Starting to spawn base package * More package hacking * Bedrock and VertexAI * Parquet split * Updated templates * Utils
33 lines
841 B
Python
33 lines
841 B
Python
|
|
from pulsar.schema import Record, Bytes, String, Boolean, Integer, Array
|
|
from pulsar.schema import Double, Map
|
|
|
|
from . documents import Source
|
|
from . types import Value, RowSchema
|
|
from . topic import topic
|
|
|
|
############################################################################
|
|
|
|
# Object embeddings are embeddings associated with the primary key of an
|
|
# object
|
|
|
|
class ObjectEmbeddings(Record):
|
|
source = Source()
|
|
vectors = Array(Array(Double()))
|
|
name = String()
|
|
key_name = String()
|
|
id = String()
|
|
|
|
object_embeddings_store_queue = topic('object-embeddings-store')
|
|
|
|
############################################################################
|
|
|
|
# Stores rows of information
|
|
|
|
class Rows(Record):
|
|
source = Source()
|
|
row_schema = RowSchema()
|
|
rows = Array(Map(String()))
|
|
|
|
rows_store_queue = topic('rows-store')
|
|
|