From 42ea9ed370867ea0c80d84493653d5e76bfbc21c Mon Sep 17 00:00:00 2001 From: Cyber MacGeddon Date: Sat, 19 Apr 2025 19:52:21 +0100 Subject: [PATCH] Tweaking --- trustgraph-base/trustgraph/base/flow_processor.py | 1 - trustgraph-base/trustgraph/base/request_response_spec.py | 4 ++++ .../trustgraph/embeddings/graph_embeddings/embeddings.py | 4 +--- 3 files changed, 5 insertions(+), 4 deletions(-) diff --git a/trustgraph-base/trustgraph/base/flow_processor.py b/trustgraph-base/trustgraph/base/flow_processor.py index df33f911..0552eb9c 100644 --- a/trustgraph-base/trustgraph/base/flow_processor.py +++ b/trustgraph-base/trustgraph/base/flow_processor.py @@ -12,7 +12,6 @@ from .. schema import config_request_queue, config_response_queue from .. schema import config_push_queue from .. log_level import LogLevel from . async_processor import AsyncProcessor -from . subscriber import Subscriber from . flow import Flow # Parent class for configurable processors, configured with flows by diff --git a/trustgraph-base/trustgraph/base/request_response_spec.py b/trustgraph-base/trustgraph/base/request_response_spec.py index 16932baa..8c881550 100644 --- a/trustgraph-base/trustgraph/base/request_response_spec.py +++ b/trustgraph-base/trustgraph/base/request_response_spec.py @@ -44,6 +44,8 @@ class RequestResponse(Subscriber): id = str(uuid.uuid4()) + print("Request", id, "...", flush=True) + q = await self.subscribe(id) try: @@ -58,6 +60,8 @@ class RequestResponse(Subscriber): timeout=timeout ) + print("Got response.", flush=True) + return resp except Exception as e: diff --git a/trustgraph-flow/trustgraph/embeddings/graph_embeddings/embeddings.py b/trustgraph-flow/trustgraph/embeddings/graph_embeddings/embeddings.py index 3189a79d..043be3a7 100755 --- a/trustgraph-flow/trustgraph/embeddings/graph_embeddings/embeddings.py +++ b/trustgraph-flow/trustgraph/embeddings/graph_embeddings/embeddings.py @@ -58,12 +58,10 @@ class Processor(FlowProcessor): for entity in v.entities: - resp = await flow("embeddings-request").embed( + vectors = await flow("embeddings-request").embed( text = entity.context ) - vectors = resp.vectors - entities.append( EntityEmbeddings( entity=entity.entity,