diff --git a/trustgraph-flow/trustgraph/model/text_completion/azure/llm.py b/trustgraph-flow/trustgraph/model/text_completion/azure/llm.py index 949eeae5..cbc638ec 100755 --- a/trustgraph-flow/trustgraph/model/text_completion/azure/llm.py +++ b/trustgraph-flow/trustgraph/model/text_completion/azure/llm.py @@ -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' ) diff --git a/trustgraph-flow/trustgraph/model/text_completion/claude/llm.py b/trustgraph-flow/trustgraph/model/text_completion/claude/llm.py index 08e2828d..d784f1f8 100755 --- a/trustgraph-flow/trustgraph/model/text_completion/claude/llm.py +++ b/trustgraph-flow/trustgraph/model/text_completion/claude/llm.py @@ -180,6 +180,7 @@ class Processor(ConsumerProducer): 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 index 8e1f4f7c..8b72f9db 100755 --- a/trustgraph-flow/trustgraph/model/text_completion/cohere/llm.py +++ b/trustgraph-flow/trustgraph/model/text_completion/cohere/llm.py @@ -167,6 +167,7 @@ class Processor(ConsumerProducer): 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 index 47a75927..345a5441 100644 --- a/trustgraph-flow/trustgraph/model/text_completion/googleaistudio/llm.py +++ b/trustgraph-flow/trustgraph/model/text_completion/googleaistudio/llm.py @@ -195,6 +195,7 @@ class Processor(ConsumerProducer): 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 index b63ad43b..849e78ea 100755 --- a/trustgraph-flow/trustgraph/model/text_completion/openai/llm.py +++ b/trustgraph-flow/trustgraph/model/text_completion/openai/llm.py @@ -190,6 +190,7 @@ class Processor(ConsumerProducer): 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 index 76d517a1..ecdbae06 100755 --- 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_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):