diff --git a/templates/components/googleaistudio.jsonnet b/templates/components/googleaistudio.jsonnet index 4088ceef..c2a40f2c 100644 --- a/templates/components/googleaistudio.jsonnet +++ b/templates/components/googleaistudio.jsonnet @@ -13,7 +13,7 @@ local prompts = import "prompts/mixtral.jsonnet"; create:: function(engine) - local envSecrets = engine.envSecrets("bedrock-credentials") + local envSecrets = engine.envSecrets("googleaistudio-key") .with_env_var("GOOGLE_AI_STUDIO_KEY", "googleaistudio-key"); local container = diff --git a/trustgraph-flow/trustgraph/query/doc_embeddings/pinecone/service.py b/trustgraph-flow/trustgraph/query/doc_embeddings/pinecone/service.py index 3fcbfb21..b8502143 100755 --- a/trustgraph-flow/trustgraph/query/doc_embeddings/pinecone/service.py +++ b/trustgraph-flow/trustgraph/query/doc_embeddings/pinecone/service.py @@ -131,9 +131,14 @@ def add_args(parser): ) parser.add_argument( - '-t', '--store-uri', - default=default_store_uri, - help=f'Milvus store URI (default: {default_store_uri})' + '-a', '--api-key', + default=default_api_key, + help='Pinecone API key. (default from PINECONE_API_KEY)' + ) + + parser.add_argument( + '-u', '--url', + help='Pinecone URL. If unspecified, serverless is used' ) def run():