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

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):
r = VectorsAssociation(entity=ent, vectors=vec)
r = GraphEmbeddings(entity=ent, vectors=vec)
self.vec_prod.send(r)
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.
"""
from .... schema import VectorsAssociation
from .... schema import GraphEmbeddings
from .... schema import graph_embeddings_store_queue
from .... log_level import LogLevel
from .... triple_vectors import TripleVectors
@ -27,7 +27,7 @@ class Processor(Consumer):
**params | {
"input_queue": input_queue,
"subscriber": subscriber,
"input_schema": VectorsAssociation,
"input_schema": GraphEmbeddings,
"store_uri": store_uri,
}
)