From 6ef5bca042af24e18a10e39914e976b26659cd04 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Thu, 26 Oct 2023 18:14:28 -0700 Subject: [PATCH] fix(utils.py): adding bedrock to validate_environment() --- litellm/utils.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/litellm/utils.py b/litellm/utils.py index d7c71bf22..f7298dd19 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -1997,6 +1997,12 @@ def validate_environment(model: Optional[str]=None) -> dict: keys_in_environment = True else: missing_keys.append("NLP_CLOUD_API_KEY") + elif custom_llm_provider == "bedrock": + if "AWS_ACCESS_KEY_ID" in os.environ and "AWS_SECRET_ACCESS_KEY" in os.environ: + keys_in_environment = True + else: + missing_keys.append("AWS_ACCESS_KEY_ID") + missing_keys.append("AWS_SECRET_ACCESS_KEY") else: ## openai - chatcompletion + text completion if model in litellm.open_ai_chat_completion_models or litellm.open_ai_text_completion_models: