This is the official codebase for GastroBot.

GastroBot is a RAG framework Chinese gastric disease chatbot. It uses 25 gastric disease-related guideline documents as a corpus, fine-tuned the embedding model, and is developed based on llama-index. An overview of GastroBot architecture can be seen below:

Set up conda environment and clone the github repo

# create a new environment
$ conda create --name rag python=3.10.13 -y
$ conda activate rag
# install requirements
$ pip install -r requirements.txt

Executing the code in the build_index.ipynb file will build the index locally. The built index is saved in the vector_index folder.

$ streamlit run app.py

Open the browser and enter the address http://127.0.0.1:port/ Start Q&A with it！

• llamaindex: https://www.llamaindex.ai/