aws-solutions-library-samples / guidance-for-natural-language-queries-of-relational-databases-on-aws

Demonstration of Natural Language Query (NLQ) of an Amazon RDS for PostgreSQL database, using SageMaker JumpStart, Amazon Bedrock, LangChain, Streamlit, and Chroma.
https://aws.amazon.com/solutions/guidance/natural-language-queries-of-relational-databases-on-aws/
MIT No Attribution
58 stars 10 forks source link

Update Code to Reflect Changes to LangChain Base Class #52

Closed garystafford-aws closed 10 months ago

garystafford-aws commented 10 months ago

Update the application code to reflect changes to the LangChain base class:

from langchain import FewShotPromptTemplate, PromptTemplate, SQLDatabase

to

from langchain.sql_database import SQLDatabase
from langchain.prompts import FewShotPromptTemplate, PromptTemplate