From faa50cbcdc52282169e7afbeaf5d54e64bb61006 Mon Sep 17 00:00:00 2001 From: Cyber MacGeddon Date: Tue, 11 Feb 2025 23:03:16 +0000 Subject: [PATCH] GOt it working --- .../trustgraph/embeddings/document_embeddings/embeddings.py | 2 +- .../trustgraph/embeddings/graph_embeddings/embeddings.py | 2 +- trustgraph-flow/trustgraph/extract/kg/definitions/extract.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/trustgraph-flow/trustgraph/embeddings/document_embeddings/embeddings.py b/trustgraph-flow/trustgraph/embeddings/document_embeddings/embeddings.py index 9e8e1e10..6a4a4a67 100755 --- a/trustgraph-flow/trustgraph/embeddings/document_embeddings/embeddings.py +++ b/trustgraph-flow/trustgraph/embeddings/document_embeddings/embeddings.py @@ -73,7 +73,7 @@ class Processor(ConsumerProducer): chunks=embeds, ) - await self.producer.send(r) + await self.send(r) except Exception as e: print("Exception:", e, flush=True) diff --git a/trustgraph-flow/trustgraph/embeddings/graph_embeddings/embeddings.py b/trustgraph-flow/trustgraph/embeddings/graph_embeddings/embeddings.py index f76f3235..2cbe9907 100755 --- a/trustgraph-flow/trustgraph/embeddings/graph_embeddings/embeddings.py +++ b/trustgraph-flow/trustgraph/embeddings/graph_embeddings/embeddings.py @@ -77,7 +77,7 @@ class Processor(ConsumerProducer): entities=entities, ) - await self.producer.send(r) + await self.send(r) except Exception as e: print("Exception:", e, flush=True) diff --git a/trustgraph-flow/trustgraph/extract/kg/definitions/extract.py b/trustgraph-flow/trustgraph/extract/kg/definitions/extract.py index 62411cb5..c88005b7 100755 --- a/trustgraph-flow/trustgraph/extract/kg/definitions/extract.py +++ b/trustgraph-flow/trustgraph/extract/kg/definitions/extract.py @@ -110,7 +110,7 @@ class Processor(ConsumerProducer): metadata=metadata, entities=entities, ) - await self.ec_prod.send(t) + self.ec_prod.send(t) async def handle(self, msg):