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

Fix more naming problems

Browse source
This commit is contained in:
Cyber MacGeddon 2024-07-23 21:08:21 +01:00
parent 9a62413d49
commit 3f6d2d879a
2 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

2
trustgraph/kg/extract_relationships/extract.py
View file

@ -87,7 +87,7 @@ class Processor(ConsumerProducer):
def emit_vec(self, ent, vec): def emit_vec(self, ent, vec):
r = VectorsAssociation(entity=ent, vectors=vec) r = GraphEmbeddings(entity=ent, vectors=vec)
self.vec_prod.send(r) self.vec_prod.send(r)
def handle(self, msg): def handle(self, msg):

4
trustgraph/storage/graph_embeddings/milvus/write.py
View file

@ -3,7 +3,7 @@
Accepts entity/vector pairs and writes them to a Milvus store. Accepts entity/vector pairs and writes them to a Milvus store.
""" """
from .... schema import VectorsAssociation from .... schema import GraphEmbeddings
from .... schema import graph_embeddings_store_queue from .... schema import graph_embeddings_store_queue
from .... log_level import LogLevel from .... log_level import LogLevel
from .... triple_vectors import TripleVectors from .... triple_vectors import TripleVectors
@ -27,7 +27,7 @@ class Processor(Consumer):
**params | { **params | {
"input_queue": input_queue, "input_queue": input_queue,
"subscriber": subscriber, "subscriber": subscriber,
"input_schema": VectorsAssociation, "input_schema": GraphEmbeddings,
"store_uri": store_uri, "store_uri": store_uri,
} }
) )