Update the application code to reflect changes to the LangChain base class (#53)

* Update app_bedrock.py - Class Changes to LangChain

* Update app_openai.py - Class Changes to LangChain

* Update app_sagemaker.py - Class Changes to LangChain

Author: garystafford-aws

docker/app_bedrock.py

@@ -13,7 +13,8 @@
 import streamlit as st
 import yaml
 from botocore.exceptions import ClientError
-from langchain import FewShotPromptTemplate, PromptTemplate, SQLDatabase
+from langchain.sql_database import SQLDatabase
+from langchain.prompts import FewShotPromptTemplate, PromptTemplate
 from langchain.chains.sql_database.prompt import PROMPT_SUFFIX, _postgres_prompt
 from langchain.embeddings.huggingface import HuggingFaceEmbeddings
 from langchain.llms import Bedrock

docker/app_openai.py

@@ -17,7 +17,8 @@
 import streamlit as st
 import yaml
 from botocore.exceptions import ClientError
-from langchain import FewShotPromptTemplate, PromptTemplate, SQLDatabase
+from langchain.sql_database import SQLDatabase
+from langchain.prompts import FewShotPromptTemplate, PromptTemplate
 from langchain.chains.sql_database.prompt import PROMPT_SUFFIX, _postgres_prompt
 from langchain.chat_models import ChatOpenAI
 from langchain.embeddings.huggingface import HuggingFaceEmbeddings

docker/app_sagemaker.py

@@ -16,7 +16,8 @@
 import streamlit as st
 import yaml
 from botocore.exceptions import ClientError
-from langchain import FewShotPromptTemplate, PromptTemplate, SQLDatabase
+from langchain.sql_database import SQLDatabase
+from langchain.prompts import FewShotPromptTemplate, PromptTemplate
 from langchain.chains.sql_database.prompt import PROMPT_SUFFIX, _postgres_prompt
 from langchain.embeddings.huggingface import HuggingFaceEmbeddings
 from langchain.llms.sagemaker_endpoint import LLMContentHandler, SagemakerEndpoint