Anirudh Esthuri commited on
Commit
2d070b4
·
1 Parent(s): 4d77281

Improve AWS credential validation and use AWS_DEFAULT_REGION

Browse files
Files changed (1) hide show
  1. llm.py +7 -4
llm.py CHANGED
@@ -27,19 +27,22 @@ def get_bedrock_client():
27
  """Get or create the Bedrock runtime client with proper error handling."""
28
  global _bedrock_runtime
29
  if _bedrock_runtime is None:
30
- aws_access_key = os.getenv("AWS_ACCESS_KEY_ID")
31
- aws_secret_key = os.getenv("AWS_SECRET_ACCESS_KEY")
 
32
 
33
  if not aws_access_key or not aws_secret_key:
34
  raise ValueError(
35
  "AWS credentials not found. Please set AWS_ACCESS_KEY_ID and AWS_SECRET_ACCESS_KEY "
36
- "as secrets in Hugging Face Spaces settings."
 
 
37
  )
38
 
39
  try:
40
  _bedrock_runtime = boto3.client(
41
  "bedrock-runtime",
42
- region_name="us-east-1",
43
  aws_access_key_id=aws_access_key,
44
  aws_secret_access_key=aws_secret_key
45
  )
 
27
  """Get or create the Bedrock runtime client with proper error handling."""
28
  global _bedrock_runtime
29
  if _bedrock_runtime is None:
30
+ aws_access_key = os.getenv("AWS_ACCESS_KEY_ID", "").strip()
31
+ aws_secret_key = os.getenv("AWS_SECRET_ACCESS_KEY", "").strip()
32
+ aws_region = os.getenv("AWS_DEFAULT_REGION", "us-east-1").strip()
33
 
34
  if not aws_access_key or not aws_secret_key:
35
  raise ValueError(
36
  "AWS credentials not found. Please set AWS_ACCESS_KEY_ID and AWS_SECRET_ACCESS_KEY "
37
+ "as secrets in Hugging Face Spaces settings. "
38
+ f"Current values: AWS_ACCESS_KEY_ID={'***' if aws_access_key else 'EMPTY'}, "
39
+ f"AWS_SECRET_ACCESS_KEY={'***' if aws_secret_key else 'EMPTY'}"
40
  )
41
 
42
  try:
43
  _bedrock_runtime = boto3.client(
44
  "bedrock-runtime",
45
+ region_name=aws_region,
46
  aws_access_key_id=aws_access_key,
47
  aws_secret_access_key=aws_secret_key
48
  )