Skip to content

Latest commit

 

History

History
48 lines (35 loc) · 4.46 KB

File metadata and controls

48 lines (35 loc) · 4.46 KB

QA with LLM and RAG (Retrieval Augmented Generation)

This project is a Question Answering application with Large Language Models (LLMs) and Amazon DocumentDB (with MongoDB Compatibility). An application using the RAG(Retrieval Augmented Generation) approach retrieves information most relevant to the user’s request from the enterprise knowledge base or content, bundles it as context along with the user’s request as a prompt, and then sends it to the LLM to get a GenAI response.

LLMs have limitations around the maximum word count for the input prompt, therefore choosing the right passages among thousands or millions of documents in the enterprise, has a direct impact on the LLM’s accuracy.

In this project, Amazon DocumentDB (with MongoDB Compatibility) is used for knowledge base.

The overall architecture is like this:

rag_with_bedrock_docdb_arch

Overall Workflow

  1. Deploy the cdk stacks (For more information, see here).
    • An Amazon DocumentDB (with MongoDB Compatibility) to store embeddings.
    • An SageMaker Studio for RAG application and data ingestion to the Amazon DocumentDB (with MongoDB Compatibility).
  2. Open JupyterLab in SageMaker Studio and then open a new terminal.
  3. Run the following commands on the terminal to clone the code repository for this project:
    git clone --depth=1 https://github.com/aws-samples/rag-with-amazon-bedrock-and-documentdb.git
    
  4. Open data_ingestion_to_documentdb.ipynb notebook and Run it. (For more information, see here)
  5. Run Streamlit application. (For more information, see here)

References

Security

See CONTRIBUTING for more information.

License

This library is licensed under the MIT-0 License. See the LICENSE file.