Fixing some more pulsar API mismatches

trustgraph-flow/trustgraph/gateway/dispatch/agent.py

@@ -6,12 +6,12 @@ from . requestor import ServiceRequestor
 
 class AgentRequestor(ServiceRequestor):
     def __init__(
-            self, pulsar_client, request_queue, response_queue, timeout,
+            self, backend, request_queue, response_queue, timeout,
             consumer, subscriber,
     ):
 
         super(AgentRequestor, self).__init__(
-            pulsar_client=pulsar_client,
+            backend=backend,
             request_queue=request_queue,
             response_queue=response_queue,
             request_schema=AgentRequest,

trustgraph-flow/trustgraph/gateway/dispatch/document_embeddings_import.py

@@ -15,7 +15,7 @@ logger = logging.getLogger(__name__)
 class DocumentEmbeddingsImport:
 
     def __init__(
-            self, ws, running, pulsar_client, queue
+            self, ws, running, backend, queue
     ):
 
         self.ws = ws
@@ -23,7 +23,7 @@ class DocumentEmbeddingsImport:
         self.translator = DocumentEmbeddingsTranslator()
         
         self.publisher = Publisher(
-            pulsar_client, topic = queue, schema = DocumentEmbeddings
+            backend, topic = queue, schema = DocumentEmbeddings
         )
 
     async def start(self):

trustgraph-flow/trustgraph/gateway/dispatch/document_rag.py

@@ -6,12 +6,12 @@ from . requestor import ServiceRequestor
 
 class DocumentRagRequestor(ServiceRequestor):
     def __init__(
-            self, pulsar_client, request_queue, response_queue, timeout,
+            self, backend, request_queue, response_queue, timeout,
             consumer, subscriber,
     ):
 
         super(DocumentRagRequestor, self).__init__(
-            pulsar_client=pulsar_client,
+            backend=backend,
             request_queue=request_queue,
             response_queue=response_queue,
             request_schema=DocumentRagQuery,

trustgraph-flow/trustgraph/gateway/dispatch/embeddings.py

@@ -6,12 +6,12 @@ from . requestor import ServiceRequestor
 
 class EmbeddingsRequestor(ServiceRequestor):
     def __init__(
-            self, pulsar_client, request_queue, response_queue, timeout,
+            self, backend, request_queue, response_queue, timeout,
             consumer, subscriber,
     ):
 
         super(EmbeddingsRequestor, self).__init__(
-            pulsar_client=pulsar_client,
+            backend=backend,
             request_queue=request_queue,
             response_queue=response_queue,
             request_schema=EmbeddingsRequest,

trustgraph-flow/trustgraph/gateway/dispatch/entity_contexts_import.py

@@ -16,14 +16,14 @@ logger = logging.getLogger(__name__)
 class EntityContextsImport:
 
     def __init__(
-            self, ws, running, pulsar_client, queue
+            self, ws, running, backend, queue
     ):
 
         self.ws = ws
         self.running = running
         
         self.publisher = Publisher(
-            pulsar_client, topic = queue, schema = EntityContexts
+            backend, topic = queue, schema = EntityContexts
         )
 
     async def start(self):

trustgraph-flow/trustgraph/gateway/dispatch/graph_embeddings_import.py

@@ -16,14 +16,14 @@ logger = logging.getLogger(__name__)
 class GraphEmbeddingsImport:
 
     def __init__(
-            self, ws, running, pulsar_client, queue
+            self, ws, running, backend, queue
     ):
 
         self.ws = ws
         self.running = running
         
         self.publisher = Publisher(
-            pulsar_client, topic = queue, schema = GraphEmbeddings
+            backend, topic = queue, schema = GraphEmbeddings
         )
 
     async def start(self):

trustgraph-flow/trustgraph/gateway/dispatch/graph_embeddings_query.py

@@ -6,12 +6,12 @@ from . requestor import ServiceRequestor
 
 class GraphEmbeddingsQueryRequestor(ServiceRequestor):
     def __init__(
-            self, pulsar_client, request_queue, response_queue, timeout,
+            self, backend, request_queue, response_queue, timeout,
             consumer, subscriber,
     ):
 
         super(GraphEmbeddingsQueryRequestor, self).__init__(
-            pulsar_client=pulsar_client,
+            backend=backend,
             request_queue=request_queue,
             response_queue=response_queue,
             request_schema=GraphEmbeddingsRequest,

trustgraph-flow/trustgraph/gateway/dispatch/graph_rag.py

@@ -6,12 +6,12 @@ from . requestor import ServiceRequestor
 
 class GraphRagRequestor(ServiceRequestor):
     def __init__(
-            self, pulsar_client, request_queue, response_queue, timeout,
+            self, backend, request_queue, response_queue, timeout,
             consumer, subscriber,
     ):
 
         super(GraphRagRequestor, self).__init__(
-            pulsar_client=pulsar_client,
+            backend=backend,
             request_queue=request_queue,
             response_queue=response_queue,
             request_schema=GraphRagQuery,

trustgraph-flow/trustgraph/gateway/dispatch/mcp_tool.py

@@ -6,12 +6,12 @@ from . requestor import ServiceRequestor
 
 class McpToolRequestor(ServiceRequestor):
     def __init__(
-            self, pulsar_client, request_queue, response_queue, timeout,
+            self, backend, request_queue, response_queue, timeout,
             consumer, subscriber,
     ):
 
         super(McpToolRequestor, self).__init__(
-            pulsar_client=pulsar_client,
+            backend=backend,
             request_queue=request_queue,
             response_queue=response_queue,
             request_schema=ToolRequest,

trustgraph-flow/trustgraph/gateway/dispatch/nlp_query.py

@@ -5,12 +5,12 @@ from . requestor import ServiceRequestor
 
 class NLPQueryRequestor(ServiceRequestor):
     def __init__(
-            self, pulsar_client, request_queue, response_queue, timeout,
+            self, backend, request_queue, response_queue, timeout,
             consumer, subscriber,
     ):
 
         super(NLPQueryRequestor, self).__init__(
-            pulsar_client=pulsar_client,
+            backend=backend,
             request_queue=request_queue,
             response_queue=response_queue,
             request_schema=QuestionToStructuredQueryRequest,

trustgraph-flow/trustgraph/gateway/dispatch/objects_import.py

@@ -15,14 +15,14 @@ logger = logging.getLogger(__name__)
 class ObjectsImport:
 
     def __init__(
-            self, ws, running, pulsar_client, queue
+            self, ws, running, backend, queue
     ):
 
         self.ws = ws
         self.running = running
         
         self.publisher = Publisher(
-            pulsar_client, topic = queue, schema = ExtractedObject
+            backend, topic = queue, schema = ExtractedObject
         )
 
     async def start(self):

trustgraph-flow/trustgraph/gateway/dispatch/objects_query.py

@@ -5,12 +5,12 @@ from . requestor import ServiceRequestor
 
 class ObjectsQueryRequestor(ServiceRequestor):
     def __init__(
-            self, pulsar_client, request_queue, response_queue, timeout,
+            self, backend, request_queue, response_queue, timeout,
             consumer, subscriber,
     ):
 
         super(ObjectsQueryRequestor, self).__init__(
-            pulsar_client=pulsar_client,
+            backend=backend,
             request_queue=request_queue,
             response_queue=response_queue,
             request_schema=ObjectsQueryRequest,

trustgraph-flow/trustgraph/gateway/dispatch/prompt.py

@@ -8,12 +8,12 @@ from . requestor import ServiceRequestor
 
 class PromptRequestor(ServiceRequestor):
     def __init__(
-            self, pulsar_client, request_queue, response_queue, timeout,
+            self, backend, request_queue, response_queue, timeout,
             consumer, subscriber,
     ):
 
         super(PromptRequestor, self).__init__(
-            pulsar_client=pulsar_client,
+            backend=backend,
             request_queue=request_queue,
             response_queue=response_queue,
             request_schema=PromptRequest,

trustgraph-flow/trustgraph/gateway/dispatch/structured_diag.py

@@ -5,12 +5,12 @@ from . requestor import ServiceRequestor
 
 class StructuredDiagRequestor(ServiceRequestor):
     def __init__(
-            self, pulsar_client, request_queue, response_queue, timeout,
+            self, backend, request_queue, response_queue, timeout,
             consumer, subscriber,
     ):
 
         super(StructuredDiagRequestor, self).__init__(
-            pulsar_client=pulsar_client,
+            backend=backend,
             request_queue=request_queue,
             response_queue=response_queue,
             request_schema=StructuredDataDiagnosisRequest,

trustgraph-flow/trustgraph/gateway/dispatch/structured_query.py

@@ -5,12 +5,12 @@ from . requestor import ServiceRequestor
 
 class StructuredQueryRequestor(ServiceRequestor):
     def __init__(
-            self, pulsar_client, request_queue, response_queue, timeout,
+            self, backend, request_queue, response_queue, timeout,
             consumer, subscriber,
     ):
 
         super(StructuredQueryRequestor, self).__init__(
-            pulsar_client=pulsar_client,
+            backend=backend,
             request_queue=request_queue,
             response_queue=response_queue,
             request_schema=StructuredQueryRequest,

trustgraph-flow/trustgraph/gateway/dispatch/triples_import.py

@@ -16,14 +16,14 @@ logger = logging.getLogger(__name__)
 class TriplesImport:
 
     def __init__(
-            self, ws, running, pulsar_client, queue
+            self, ws, running, backend, queue
     ):
 
         self.ws = ws
         self.running = running
         
         self.publisher = Publisher(
-            pulsar_client, topic = queue, schema = Triples
+            backend, topic = queue, schema = Triples
         )
 
     async def start(self):

trustgraph-flow/trustgraph/gateway/dispatch/triples_query.py

@@ -6,12 +6,12 @@ from . requestor import ServiceRequestor
 
 class TriplesQueryRequestor(ServiceRequestor):
     def __init__(
-            self, pulsar_client, request_queue, response_queue, timeout,
+            self, backend, request_queue, response_queue, timeout,
             consumer, subscriber,
     ):
 
         super(TriplesQueryRequestor, self).__init__(
-            pulsar_client=pulsar_client,
+            backend=backend,
             request_queue=request_queue,
             response_queue=response_queue,
             request_schema=TriplesQueryRequest,