Spaces:
Running
Running
Anirudh Esthuri
commited on
Commit
·
2d070b4
1
Parent(s):
4d77281
Improve AWS credential validation and use AWS_DEFAULT_REGION
Browse files
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=
|
| 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 |
)
|