FACTS About Building Retrieval Augmented Generation-based Chatbots
Rama Akkiraju,Anbang Xu,Deepak Bora,Tan Yu,Lu An,Vishal Seth,Aaditya Shukla,Pritam Gundecha,Hridhay Mehta,Ashwin Jha,Prithvi Raj,Abhinav Balasubramanian,Murali Maram,Guru Muthusamy,Shivakesh Reddy Annepally,Sidney Knowles,Min Du,Nick Burnett,Sean Javiya,Ashok Marannan,Mamta Kumari,Surbhi Jha,Ethan Dereszenski,Anupam Chakraborty,Subhash Ranjan,Amina Terfai,Anoop Surya,Tracey Mercer,Vinodh Kumar Thanigachalam,Tamar Bar,Sanjana Krishnan,Samy Kilaru,Jasmine Jaksic,Nave Algarici,Jacob Liberman,Joey Conway,Sonu Nayyar,Justin Boitano
2024-07-11
Abstract:Enterprise chatbots, powered by generative AI, are emerging as key applications to enhance employee productivity. Retrieval Augmented Generation (RAG), Large Language Models (LLMs), and orchestration frameworks like Langchain and Llamaindex are crucial for building these chatbots. However, creating effective enterprise chatbots is challenging and requires meticulous RAG pipeline engineering. This includes fine-tuning embeddings and LLMs, extracting documents from vector databases, rephrasing queries, reranking results, designing prompts, honoring document access controls, providing concise responses, including references, safeguarding personal information, and building orchestration agents. We present a framework for building RAG-based chatbots based on our experience with three NVIDIA chatbots: for IT/HR benefits, financial earnings, and general content. Our contributions are three-fold: introducing the FACTS framework (Freshness, Architectures, Cost, Testing, Security), presenting fifteen RAG pipeline control points, and providing empirical results on accuracy-latency tradeoffs between large and small LLMs. To the best of our knowledge, this is the first paper of its kind that provides a holistic view of the factors as well as solutions for building secure enterprise-grade chatbots."
Machine Learning,Computation and Language