diff --git a/trustgraph-flow/trustgraph/gateway/document_embeddings_stream.py b/trustgraph-flow/trustgraph/gateway/document_embeddings_stream.py index a32c3e68..790ea27e 100644 --- a/trustgraph-flow/trustgraph/gateway/document_embeddings_stream.py +++ b/trustgraph-flow/trustgraph/gateway/document_embeddings_stream.py @@ -27,7 +27,8 @@ class DocumentEmbeddingsStreamEndpoint(SocketEndpoint): self.subscriber = Subscriber( self.pulsar_client, document_embeddings_store_queue, "api-gateway", "api-gateway", - schema=JsonSchema(DocumentEmbeddings) + schema=JsonSchema(DocumentEmbeddings), + pulsar_api_key=self.pulsar_api_key ) async def listener(self, ws, running): diff --git a/trustgraph-flow/trustgraph/processing/processing.py b/trustgraph-flow/trustgraph/processing/processing.py index e5e9593f..5352776a 100644 --- a/trustgraph-flow/trustgraph/processing/processing.py +++ b/trustgraph-flow/trustgraph/processing/processing.py @@ -69,6 +69,7 @@ class Processing: params = { "pulsar_host": self.pulsar_host, + "pulsar_api_key": self.pulsar_api_key, "log_level": str(self.log_level), } @@ -161,6 +162,7 @@ def run(): try: p = Processing( pulsar_host=args.pulsar_host, + pulsar_api_key=args.pulsar_api_key, file=args.file, log_level=args.log_level, )