Usage
This folder implements a knowledge base using RAG based on Kaggle competitions. It allows you to store Kaggle competition experiences into the knowledge base, as well as store experimental experiences from RD-Agent.
- First, generate a knowledge base (in JSON format) by running the
mainfunction inextract_knowledge.py. - Then, create a vector base in
vector_base.pyand save it. - Finally, add the field
KG_RAG_PATH="xxx.pkl"(the path to the saved vector base) in your.envfile.