Fix env var passing (#120)

trustgraph-ai · Oct 15, 2024 · 04617a8 · 04617a8
1 parent 25983d1
commit 04617a8
Show file tree

Hide file tree

Showing 6 changed files with 7 additions and 1 deletion.
diff --git a/trustgraph-flow/trustgraph/model/text_completion/azure/llm.py b/trustgraph-flow/trustgraph/model/text_completion/azure/llm.py
@@ -208,11 +208,13 @@ def add_args(parser):
 
         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'
         )
 

diff --git a/trustgraph-flow/trustgraph/model/text_completion/claude/llm.py b/trustgraph-flow/trustgraph/model/text_completion/claude/llm.py
@@ -180,6 +180,7 @@ def add_args(parser):
 
         parser.add_argument(
             '-k', '--api-key',
+            default=default_api_key,
             help=f'Claude API key'
         )
 

diff --git a/trustgraph-flow/trustgraph/model/text_completion/cohere/llm.py b/trustgraph-flow/trustgraph/model/text_completion/cohere/llm.py
@@ -167,6 +167,7 @@ def add_args(parser):
 
         parser.add_argument(
             '-k', '--api-key',
+            default=default_api_key,
             help=f'Cohere API key'
         )
 

diff --git a/trustgraph-flow/trustgraph/model/text_completion/googleaistudio/llm.py b/trustgraph-flow/trustgraph/model/text_completion/googleaistudio/llm.py
@@ -195,6 +195,7 @@ def add_args(parser):
 
         parser.add_argument(
             '-k', '--api-key',
+            default=default_api_key,
             help=f'GoogleAIStudio API key'
         )
 

diff --git a/trustgraph-flow/trustgraph/model/text_completion/openai/llm.py b/trustgraph-flow/trustgraph/model/text_completion/openai/llm.py
@@ -190,6 +190,7 @@ def add_args(parser):
 
         parser.add_argument(
             '-k', '--api-key',
+            default=default_api_key,
             help=f'OpenAI API key'
         )
 

diff --git a/trustgraph-vertexai/trustgraph/model/text_completion/vertexai/llm.py b/trustgraph-vertexai/trustgraph/model/text_completion/vertexai/llm.py
@@ -39,7 +39,7 @@
 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):