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

Added missing api_key references

Browse source
This commit is contained in:
Tyler O 2025-02-11 15:30:53 +00:00 • committed by Cyber MacGeddon
parent 0464423b04
commit aef1ad7f39
2 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
trustgraph-flow/trustgraph/gateway/document_embeddings_stream.py
View file

@ -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):

2
trustgraph-flow/trustgraph/processing/processing.py
View file

@ -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,
)