apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-04-27 09:26:22 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix env var passing (#120)

Browse source
This commit is contained in:
cybermaggedon 2024-10-15 20:02:21 +01:00 • committed by GitHub
parent 25983d1557
commit 04617a81f7
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
6 changed files with 7 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
trustgraph-flow/trustgraph/model/text_completion/azure/llm.py
View file

@ -208,11 +208,13 @@ class Processor(ConsumerProducer):
parser.add_argument(
'-e', '--endpoint',
default=default_endpoint,
help=f'LLM model endpoint'
)
parser.add_argument(
'-k', '--token',
default=default_token,
help=f'LLM model token'
)

1
trustgraph-flow/trustgraph/model/text_completion/claude/llm.py
View file

@ -180,6 +180,7 @@ class Processor(ConsumerProducer):
parser.add_argument(
'-k', '--api-key',
default=default_api_key,
help=f'Claude API key'
)

1
trustgraph-flow/trustgraph/model/text_completion/cohere/llm.py
View file

@ -167,6 +167,7 @@ class Processor(ConsumerProducer):
parser.add_argument(
'-k', '--api-key',
default=default_api_key,
help=f'Cohere API key'
)

1
trustgraph-flow/trustgraph/model/text_completion/googleaistudio/llm.py
View file

@ -195,6 +195,7 @@ class Processor(ConsumerProducer):
parser.add_argument(
'-k', '--api-key',
default=default_api_key,
help=f'GoogleAIStudio API key'
)

1
trustgraph-flow/trustgraph/model/text_completion/openai/llm.py
View file

@ -190,6 +190,7 @@ class Processor(ConsumerProducer):
parser.add_argument(
'-k', '--api-key',
default=default_api_key,
help=f'OpenAI API key'
)