From c4679e67d874e91f3fd57cecc24505148d79b71a Mon Sep 17 00:00:00 2001 From: based Date: Sun, 28 Jan 2024 14:49:00 +1000 Subject: [PATCH] only output aws keys with bedrock enabled when running -proxyoutput --- main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/main.py b/main.py index 666010e..a3df788 100644 --- a/main.py +++ b/main.py @@ -278,7 +278,7 @@ def output_keys(): # ai21 and vertex keys aren't supported in proxies so no point outputting them, filtered azure keys should be excluded. print("OPENAI_KEY=" + ','.join(key.api_key for key in valid_oai_keys)) print("ANTHROPIC_KEY=" + ','.join(key.api_key for key in valid_anthropic_keys)) - print("AWS_CREDENTIALS=" + ','.join(f"{key.api_key}:{key.region}" for key in valid_aws_keys if not key.useless)) + print("AWS_CREDENTIALS=" + ','.join(f"{key.api_key}:{key.region}" for key in valid_aws_keys if not key.useless and key.bedrock_enabled)) print("GOOGLE_AI_KEY=" + ','.join(key.api_key for key in valid_makersuite_keys)) print("AZURE_CREDENTIALS=" + ','.join(f"{key.api_key.split(':')[0]}:{key.best_deployment}:{key.api_key.split(':')[1]}" for key in valid_azure_keys if key.unfiltered)) print("MISTRAL_AI_KEY=" + ','.join(key.api_key for key in valid_mistral_keys))