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

Fix env var passing

Browse source
This commit is contained in:
Cyber MacGeddon 2024-10-15 19:35:17 +01:00
parent 25983d1557
commit 2d2389f947
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'
)

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_temperature = 0.0
default_max_output = 8192
default_private_key = os.getenv("VERTEXAI_KEY")
default_private_key = "private.json"
class Processor(ConsumerProducer):