├── .gitignore ├── README.md ├── configs ├── FactKG.json ├── metaQA_1hop.json ├── metaQA_2hop.json └── metaQA_3hop.json ├── ollama_server.sh ├── pipeline ├── FactKG_index.py ├── FactKG_query.py ├── metaQA_index.py ├── metaQA_query1hop.py ├── metaQA_query2hop.py └── metaQA_query3hop.py ├── prompts ├── answer_FactKG.py ├── answer_metaQA.py ├── rewrite_FactKG.py └── rewrite_metaQA.py └── src ├── dataset.py ├── embedding_model.py ├── indexer.py ├── llm.py ├── retriever.py ├── utils.py └── vecdb.py /.gitignore: -------------------------------------------------------------------------------- 1 | data/ 2 | results/ 3 | test/ 4 | 5 | Milvus/ 6 | 7 | __pycache__/ 8 | .vscode/ -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YZ-Cai/SimGRAG/HEAD/README.md -------------------------------------------------------------------------------- /configs/FactKG.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YZ-Cai/SimGRAG/HEAD/configs/FactKG.json -------------------------------------------------------------------------------- /configs/metaQA_1hop.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YZ-Cai/SimGRAG/HEAD/configs/metaQA_1hop.json -------------------------------------------------------------------------------- /configs/metaQA_2hop.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YZ-Cai/SimGRAG/HEAD/configs/metaQA_2hop.json -------------------------------------------------------------------------------- /configs/metaQA_3hop.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YZ-Cai/SimGRAG/HEAD/configs/metaQA_3hop.json -------------------------------------------------------------------------------- /ollama_server.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YZ-Cai/SimGRAG/HEAD/ollama_server.sh -------------------------------------------------------------------------------- /pipeline/FactKG_index.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YZ-Cai/SimGRAG/HEAD/pipeline/FactKG_index.py -------------------------------------------------------------------------------- /pipeline/FactKG_query.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YZ-Cai/SimGRAG/HEAD/pipeline/FactKG_query.py -------------------------------------------------------------------------------- /pipeline/metaQA_index.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YZ-Cai/SimGRAG/HEAD/pipeline/metaQA_index.py -------------------------------------------------------------------------------- /pipeline/metaQA_query1hop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YZ-Cai/SimGRAG/HEAD/pipeline/metaQA_query1hop.py -------------------------------------------------------------------------------- /pipeline/metaQA_query2hop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YZ-Cai/SimGRAG/HEAD/pipeline/metaQA_query2hop.py -------------------------------------------------------------------------------- /pipeline/metaQA_query3hop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YZ-Cai/SimGRAG/HEAD/pipeline/metaQA_query3hop.py -------------------------------------------------------------------------------- /prompts/answer_FactKG.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YZ-Cai/SimGRAG/HEAD/prompts/answer_FactKG.py -------------------------------------------------------------------------------- /prompts/answer_metaQA.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YZ-Cai/SimGRAG/HEAD/prompts/answer_metaQA.py -------------------------------------------------------------------------------- /prompts/rewrite_FactKG.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YZ-Cai/SimGRAG/HEAD/prompts/rewrite_FactKG.py -------------------------------------------------------------------------------- /prompts/rewrite_metaQA.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YZ-Cai/SimGRAG/HEAD/prompts/rewrite_metaQA.py -------------------------------------------------------------------------------- /src/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YZ-Cai/SimGRAG/HEAD/src/dataset.py -------------------------------------------------------------------------------- /src/embedding_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YZ-Cai/SimGRAG/HEAD/src/embedding_model.py -------------------------------------------------------------------------------- /src/indexer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YZ-Cai/SimGRAG/HEAD/src/indexer.py -------------------------------------------------------------------------------- /src/llm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YZ-Cai/SimGRAG/HEAD/src/llm.py -------------------------------------------------------------------------------- /src/retriever.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YZ-Cai/SimGRAG/HEAD/src/retriever.py -------------------------------------------------------------------------------- /src/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YZ-Cai/SimGRAG/HEAD/src/utils.py -------------------------------------------------------------------------------- /src/vecdb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YZ-Cai/SimGRAG/HEAD/src/vecdb.py --------------------------------------------------------------------------------