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

Fix running issues

Browse source
This commit is contained in:
Cyber MacGeddon 2024-12-30 12:35:22 +00:00
parent caf58bb5f2
commit 4a5ca3197a
6 changed files with 10 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

2
trustgraph-flow/trustgraph/embeddings/vectorize/vectorize.py
View file

@ -78,7 +78,7 @@ class Processor(ConsumerProducer):
r = GraphEmbeddings(
metadata=v.metadata,
entities=entiities,
entities=entities,
)
self.producer.send(r)

4
trustgraph-flow/trustgraph/extract/kg/definitions/extract.py
View file

@ -6,7 +6,7 @@ entity/context definitions for embedding.
"""
import urllib.parse
import json
from pulsar.schema import JsonSchema
from .... schema import Chunk, Triple, Triples, Metadata, Value
from .... schema import EntityContext, EntityContexts
@ -68,7 +68,7 @@ class Processor(ConsumerProducer):
__class__.pubsub_metric.info({
"input_queue": input_queue,
"output_queue": output_queue,
"vector_queue": vector_queue,
"entity_context_queue": ec_queue,
"prompt_request_queue": pr_request_queue,
"prompt_response_queue": pr_response_queue,
"subscriber": subscriber,

2
trustgraph-flow/trustgraph/extract/kg/relationships/extract.py
View file

@ -6,8 +6,6 @@ graph edges.
"""
import urllib.parse
import os
from pulsar.schema import JsonSchema
from .... schema import Chunk, Triple, Triples
from .... schema import Metadata, Value

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

@ -40,9 +40,9 @@ class Processor(Consumer):
for entity in v.entities:
if entity.value != "" and entity.value is not None:
if entity.entity.value != "" and entity.entity.value is not None:
for vec in entity.vectors:
self.vecstore.insert(vec, entity.value)
self.vecstore.insert(vec, entity.entity.value)
@staticmethod
def add_args(parser):

5
trustgraph-flow/trustgraph/storage/graph_embeddings/pinecone/write.py
View file

@ -96,7 +96,8 @@ class Processor(Consumer):
for entity in v.entities:
if v.entity.value == "" or v.entity.value is None: continue
if entity.entity.value == "" or entity.entity.value is None:
continue
for vec in entity.vectors:
@ -128,7 +129,7 @@ class Processor(Consumer):
{
"id": id,
"values": vec,
"metadata": { "entity": entity.value },
"metadata": { "entity": entity.entity.value },
}
]

4
trustgraph-flow/trustgraph/storage/graph_embeddings/qdrant/write.py
View file

@ -71,7 +71,7 @@ class Processor(Consumer):
for entity in v.entities:
if entity.value == "" or entity.value is None: return
if entity.entity.value == "" or entity.entity.value is None: return
for vec in entity.vectors:
@ -88,7 +88,7 @@ class Processor(Consumer):
id=str(uuid.uuid4()),
vector=vec,
payload={
"entity": entity.value,
"entity": entity.entity.value,
}
)
]