Fix running issues

trustgraph-flow/trustgraph/embeddings/vectorize/vectorize.py

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

trustgraph-flow/trustgraph/extract/kg/definitions/extract.py

@@ -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,

trustgraph-flow/trustgraph/extract/kg/relationships/extract.py

@@ -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

trustgraph-flow/trustgraph/storage/graph_embeddings/milvus/write.py

@@ -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):

trustgraph-flow/trustgraph/storage/graph_embeddings/pinecone/write.py

@@ -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 },
                     }
                 ]
 

trustgraph-flow/trustgraph/storage/graph_embeddings/qdrant/write.py

@@ -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,
                             }
                         )
                     ]