NexQuant

mirror of https://github.com/NicolasBohn/NexQuant.git synced 2026-08-03 02:17:43 +00:00

Files

T

WinstonLiyt 59cdf946a7 fix: fix some bugs in rag (#399 )

* fix some bugs in rag

* fix a bug in costeer

2024-09-30 02:38:33 +08:00

extract_knowledge.py

feat: add normal rag into framework (#360 )

2024-09-27 01:36:12 +08:00

graph.py

fix: refine some codes (#353 )

2024-09-27 00:17:28 +08:00

prompts.yaml

fix: fix some bugs in rag (#399 )

2024-09-30 02:38:33 +08:00

README.md

feat: Added support for loading and storing RAG in Kaggle scenarios. (#269 )

2024-09-19 11:33:33 +08:00

vector_base.py

fix: fix some bugs in rag (#399 )

2024-09-30 02:38:33 +08:00

README.md

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 main function in extract_knowledge.py.
Then, create a vector base in vector_base.py and save it.
Finally, add the field KG_RAG_PATH="xxx.pkl" (the path to the saved vector base) in your .env file.