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

Feature/pulsar api key support (#308)

Browse source 
* Add pulsar API token check

* Added missing api_key references

---------

Co-authored-by: Tyler O <4535788+toliver38@users.noreply.github.com>
This commit is contained in:
cybermaggedon 2025-02-15 11:22:48 +00:00 • committed by GitHub
parent f7df2df266
commit 617eb7efd5
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
40 changed files with 173 additions and 21 deletions
Download patch file Download diff file Expand all files Collapse all files

1
trustgraph-flow/trustgraph/extract/kg/definitions/extract.py
View file

@ -79,6 +79,7 @@ class Processor(ConsumerProducer):
self.prompt = PromptClient(
pulsar_host=self.pulsar_host,
pulsar_api_key=self.pulsar_api_key,
input_queue=pr_request_queue,
output_queue=pr_response_queue,
subscriber = module + "-prompt",

1
trustgraph-flow/trustgraph/extract/kg/relationships/extract.py
View file

@ -54,6 +54,7 @@ class Processor(ConsumerProducer):
self.prompt = PromptClient(
pulsar_host=self.pulsar_host,
pulsar_api_key=self.pulsar_api_key,
input_queue=pr_request_queue,
output_queue=pr_response_queue,
subscriber = module + "-prompt",

1
trustgraph-flow/trustgraph/extract/kg/topics/extract.py
View file

@ -52,6 +52,7 @@ class Processor(ConsumerProducer):
self.prompt = PromptClient(
pulsar_host=self.pulsar_host,
pulsar_api_key=self.pulsar_api_key,
input_queue=pr_request_queue,
output_queue=pr_response_queue,
subscriber = module + "-prompt",

1
trustgraph-flow/trustgraph/extract/object/row/extract.py
View file

@ -112,6 +112,7 @@ class Processor(ConsumerProducer):
self.prompt = PromptClient(
pulsar_host=self.pulsar_host,
pulsar_api_key=self.pulsar_api_key,
input_queue=pr_request_queue,
output_queue=pr_response_queue,
subscriber = module + "-prompt",