Merge pull request #27 from hako884/main

Fix: kendra_retriever_samples/kendra_chat_open_ai.py
aws-samples · Aug 14, 2023 · 0c72951 · 0c72951
2 parents 813868b + b184dff
commit 0c72951
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.