apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-07-23 20:21:03 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

Updating manager

Browse source
This commit is contained in:
Cyber MacGeddon 2025-05-02 15:21:44 +01:00
parent 54bd509d86
commit 73bced123b
1 changed files with 24 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

27
trustgraph-flow/trustgraph/gateway/dispatch/manager.py
View file

@ -2,6 +2,26 @@
import asyncio import asyncio
from . embeddings import EmbeddingsRequestor from . embeddings import EmbeddingsRequestor
from . agent import AgentRequestor
from . text_completion import TextCompletionRequestor
from . prompt import PromptRequestor
from . graph_rag import GraphRagRequestor
from . document_rag import DocumentRagRequestor
from . triples_query import TriplesQueryRequestor
from . embeddings import EmbeddingsRequestor
from . graph_embeddings_query import GraphEmbeddingsQueryRequestor
from . prompt import PromptRequestor
request_response_dispatchers = {
"agent": AgentRequestor,
"text-completion": TextCompletionRequestor,
"prompt": PromptRequestor,
"graph-rag": GraphRagRequestor,
"document-rag": DocumentRagRequestor,
"embeddings": EmbeddingsRequestor,
"graph-embeddings": GraphEmbeddingsQueryRequestor,
"triples-query": TriplesQueryRequestor,
}
class TestDispatcher: class TestDispatcher:
def __init__(self, pulsar_client, timeout=120): def __init__(self, pulsar_client, timeout=120):
@ -91,7 +111,6 @@ class DispatcherManager:
return TestDispatcher(pulsar_client = self.pulsar_client) return TestDispatcher(pulsar_client = self.pulsar_client)
def dispatch_flow_service(self): def dispatch_flow_service(self):
# return TestDispatcher2(pulsar_client = self.pulsar_client)
return self return self
def dispatch_socket_service(self): def dispatch_socket_service(self):
@ -102,6 +121,9 @@ class DispatcherManager:
flow = params.get("flow") flow = params.get("flow")
kind = params.get("kind") kind = params.get("kind")
if kind not in request_response_dispatchers:
raise RuntimeError("Invalid kind")
key = (flow, kind) key = (flow, kind)
if flow not in self.flows: if flow not in self.flows:
@ -112,7 +134,7 @@ class DispatcherManager:
qconfig = self.flows[flow]["interfaces"]["embeddings"] qconfig = self.flows[flow]["interfaces"]["embeddings"]
dispatcher = EmbeddingsRequestor( dispatcher = request_response_dispatchers[kind](
pulsar_client = self.pulsar_client, pulsar_client = self.pulsar_client,
request_queue = qconfig["request"], request_queue = qconfig["request"],
response_queue = qconfig["response"], response_queue = qconfig["response"],
@ -125,6 +147,5 @@ class DispatcherManager:
self.dispatchers[key] = dispatcher self.dispatchers[key] = dispatcher
print("CREATE")
return await dispatcher.process(data, responder) return await dispatcher.process(data, responder)