This is the repo for the generative AI question-answering chatbot on Pokemon data using retrieval-augmented generation (RAG) in Talk to your Data with Retrieval-Augmented Generation (RAG) or: How I Built a Pokémon Chatbot.
- LLM: Mistral 7B
- embedding model: voyage 2 from Voyage AI
- vector DB: Chroma DB
- framework: LangChain
- dataset: Master Pokemon Dataset and Corpus
Kaggle Notebooks with GPU T4 x2
Dataset: https://www.kaggle.com/datasets/hanicier/master-pokemon-dataset-with-corpus-version