Fix: kendra_retriever_samples/kendra_chat_open_ai.py

aws-samples · Jul 18, 2023 · b184dff · b184dff
1 parent 67f85e5
commit b184dff
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/kendra_retriever_samples/kendra_chat_open_ai.py b/kendra_retriever_samples/kendra_chat_open_ai.py
@@ -13,7 +13,7 @@ def build_chain():
 
   llm = OpenAI(batch_size=5, temperature=0, max_tokens=300)
 
-  retriever = AmazonKendraRetriever(index_id=kendra_index_id)
+  retriever = AmazonKendraRetriever(index_id=kendra_index_id, region_name=region)
 
   prompt_template = """
   The following is a friendly conversation between a human and an AI.