This repo is an implementation of a locally hosted chatbot specifically focused on question answering over the WPILib documentation. Built with LangChain and FastAPI.
The app leverages LangChain's streaming support and async API to update the page in real time for multiple users.
- Install dependencies:
pip install -r requirements.txt
- Run
ingest.sh
to ingest WPILib docs data into the vectorstore (only needs to be done once). - Run the app:
make start
- To enable tracing, make sure
langchain-server
is running locally and passtracing=True
toget_chain
inmain.py
. You can find more documentation here.
- To enable tracing, make sure
- Open localhost:9000 in your browser.
There are two components: ingestion and question-answering.
Ingestion has the following steps:
- Pull html from documentation site
- Load html with LangChain's ReadTheDocs Loader
- Split documents with LangChain's TextSplitter
- Create a vectorstore of embeddings, using LangChain's vectorstore wrapper (with OpenAI's embeddings and FAISS vectorstore).
Question-Answering has the following steps, all handled by ConversationalRetrievalChain:
- Given the chat history and new user input, determine what a standalone question would be (using ChatGPT).
- Given that standalone question, look up relevant documents from the vectorstore.
- Pass the standalone question and relevant documents to ChatGPT to generate a final answer.