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,