Merge pull request #67 from aws-samples/development

Development

kendra_retriever_samples/app.py

@@ -79,7 +79,7 @@ def read_properties_file(filename):
         else:
             raise Exception("Unsupported LLM: ", sys.argv[1])
     else:
-        raise Exception("Usage: streamlit run app.py <anthropic|flanxl|flanxxl|openai|bedrock_titan|bedrock_claude|bedrock|claudev2>")
+        raise Exception("Usage: streamlit run app.py <anthropic|flanxl|flanxxl|openai|falcon40b|llama2|bedrock_titan|bedrock_claude|bedrock_claudev2>")
 
 if 'chat_history' not in st.session_state:
     st.session_state['chat_history'] = []

kendra_retriever_samples/kendra_chat_bedrock_claudev2.py

@@ -23,17 +23,22 @@ class bcolors:
 def build_chain():
   region = os.environ["AWS_REGION"]
   kendra_index_id = os.environ["KENDRA_INDEX_ID"]
-  credentials_profile_name = os.environ['AWS_PROFILE']
 
-  print(credentials_profile_name)
-
-
-  llm = Bedrock(
+  if "AWS_PROFILE" in os.environ:
+    credentials_profile_name = os.environ['AWS_PROFILE']
+    print("Using " + credentials_profile_name + " profile.")
+    llm = Bedrock(
       credentials_profile_name=credentials_profile_name,
       region_name = region,
       model_kwargs={"max_tokens_to_sample":300,"temperature":1,"top_k":250,"top_p":0.999,"anthropic_version":"bedrock-2023-05-31"},
       model_id="anthropic.claude-v2"
-  )
+    )
+  else:
+    llm = Bedrock(
+      region_name = region,
+      model_kwargs={"max_tokens_to_sample":300,"temperature":1,"top_k":250,"top_p":0.999,"anthropic_version":"bedrock-2023-05-31"},
+      model_id="anthropic.claude-v2"
+    )
 
   retriever = AmazonKendraRetriever(index_id=kendra_index_id,top_k=5,region_name=region)
 

kendra_retriever_samples/requirements.txt

@@ -1,4 +1,4 @@
-langchain==0.0.326
+langchain==0.0.329
 boto3>=1.28.27
 openai
 anthropic