feat: Handle AWS_PROFILE for Bedrock models in validate_environment

This commit is contained in:
lentil32 (aider) 2025-03-16 15:27:36 +09:00 committed by lentil32
parent 4f4b10fd86
commit cc33fc2822
No known key found for this signature in database
GPG key ID: ED44A29A1A3B09B1

View file

@ -572,6 +572,21 @@ class Model(ModelSettings):
model = self.name
res = litellm.validate_environment(model)
# If missing AWS credential keys but AWS_PROFILE is set, consider AWS credentials valid
if res["missing_keys"] and any(
key in ["AWS_ACCESS_KEY_ID", "AWS_SECRET_ACCESS_KEY"] for key in res["missing_keys"]
):
if model.startswith("bedrock/") or model.startswith("us.anthropic."):
if os.environ.get("AWS_PROFILE"):
res["missing_keys"] = [
k
for k in res["missing_keys"]
if k not in ["AWS_ACCESS_KEY_ID", "AWS_SECRET_ACCESS_KEY"]
]
if not res["missing_keys"]:
res["keys_in_environment"] = True
if res["keys_in_environment"]:
return res
if res["missing_keys"]: