This commit is contained in:
Cyber MacGeddon 2025-04-19 19:52:21 +01:00
parent 8d082aed87
commit 42ea9ed370
3 changed files with 5 additions and 4 deletions

View file

@ -12,7 +12,6 @@ from .. schema import config_request_queue, config_response_queue
from .. schema import config_push_queue from .. schema import config_push_queue
from .. log_level import LogLevel from .. log_level import LogLevel
from . async_processor import AsyncProcessor from . async_processor import AsyncProcessor
from . subscriber import Subscriber
from . flow import Flow from . flow import Flow
# Parent class for configurable processors, configured with flows by # Parent class for configurable processors, configured with flows by

View file

@ -44,6 +44,8 @@ class RequestResponse(Subscriber):
id = str(uuid.uuid4()) id = str(uuid.uuid4())
print("Request", id, "...", flush=True)
q = await self.subscribe(id) q = await self.subscribe(id)
try: try:
@ -58,6 +60,8 @@ class RequestResponse(Subscriber):
timeout=timeout timeout=timeout
) )
print("Got response.", flush=True)
return resp return resp
except Exception as e: except Exception as e:

View file

@ -58,12 +58,10 @@ class Processor(FlowProcessor):
for entity in v.entities: for entity in v.entities:
resp = await flow("embeddings-request").embed( vectors = await flow("embeddings-request").embed(
text = entity.context text = entity.context
) )
vectors = resp.vectors
entities.append( entities.append(
EntityEmbeddings( EntityEmbeddings(
entity=entity.entity, entity=entity.entity,