apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-05-25 23:35:12 +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( parser.add_argument(
'-e', '--endpoint', '-e', '--endpoint',
default=default_endpoint,
help=f'LLM model endpoint' help=f'LLM model endpoint'
) )
parser.add_argument( parser.add_argument(
'-k', '--token', '-k', '--token',
default=default_token,
help=f'LLM model 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( parser.add_argument(
'-k', '--api-key', '-k', '--api-key',
default=default_api_key,
help=f'Claude 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( parser.add_argument(
'-k', '--api-key', '-k', '--api-key',
default=default_api_key,
help=f'Cohere 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( parser.add_argument(
'-k', '--api-key', '-k', '--api-key',
default=default_api_key,
help=f'GoogleAIStudio 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( parser.add_argument(
'-k', '--api-key', '-k', '--api-key',
default=default_api_key,
help=f'OpenAI API key' help=f'OpenAI API key'
) )

2
trustgraph-vertexai/trustgraph/model/text_completion/vertexai/llm.py
View file

@ -39,7 +39,7 @@ default_model = 'gemini-1.0-pro-001'
default_region = 'us-central1' default_region = 'us-central1'
default_temperature = 0.0 default_temperature = 0.0
default_max_output = 8192 default_max_output = 8192
default_private_key = os.getenv("VERTEXAI_KEY") default_private_key = "private.json"
class Processor(ConsumerProducer): class Processor(ConsumerProducer):