From f11d845dee355709b41dec36dcc7c74f7b90c7ec Mon Sep 17 00:00:00 2001 From: Piyush Jain Date: Wed, 12 Jul 2023 20:47:35 -0700 Subject: [PATCH] Fixed validation error when credentials_profile_name, or region_name is not passed (#7629) ## Summary This PR corrects the checks for credentials_profile_name, and region_name attributes. This was causing validation exceptions when either of these values were missing during creation of the retriever class. Fixes #7571 #### Requested reviewers: @baskaryan --- langchain/retrievers/kendra.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/langchain/retrievers/kendra.py b/langchain/retrievers/kendra.py index 796623ead..94b72759e 100644 --- a/langchain/retrievers/kendra.py +++ b/langchain/retrievers/kendra.py @@ -213,14 +213,14 @@ class AmazonKendraRetriever(BaseRetriever): try: import boto3 - if values["credentials_profile_name"] is not None: + if values.get("credentials_profile_name"): session = boto3.Session(profile_name=values["credentials_profile_name"]) else: # use default credentials session = boto3.Session() client_params = {} - if values["region_name"] is not None: + if values.get("region_name"): client_params["region_name"] = values["region_name"] values["client"] = session.client("kendra", **client_params)