From 490f4a9ff0c7d3aaddb63831a6d9f0d7c262602b Mon Sep 17 00:00:00 2001 From: Ronald Li Date: Mon, 10 Jul 2023 15:02:36 +0800 Subject: [PATCH] Fixes KeyError in AmazonKendraRetriever initializer (#7464) ### Description argument variable client is marked as required in commit 81e5b1ad362e9e6ec955b6a54776322af82050d0 which breaks the default way of initialization providing only index_id. This commit avoid KeyError exception when it is initialized without a client variable ### Dependencies no dependency required --- langchain/retrievers/kendra.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/langchain/retrievers/kendra.py b/langchain/retrievers/kendra.py index 7d0e2084ac..796623ead5 100644 --- a/langchain/retrievers/kendra.py +++ b/langchain/retrievers/kendra.py @@ -207,7 +207,7 @@ class AmazonKendraRetriever(BaseRetriever): @root_validator(pre=True) def create_client(cls, values: Dict[str, Any]) -> Dict[str, Any]: - if values["client"] is not None: + if values.get("client") is not None: return values try: