├── .gitignore ├── .idea ├── .gitignore ├── inspectionProfiles │ ├── Project_Default.xml │ └── profiles_settings.xml ├── misc.xml ├── modules.xml ├── rag_with_chat.iml └── vcs.xml ├── Dockerfile ├── LICENSE ├── README.md ├── benchmark ├── bench_data.json ├── benchmark.py └── model_serve.py ├── chatgpt_proxy.py ├── config.py ├── data ├── car_user_manual.pdf ├── gold_result.json └── test_question.json ├── example_test.py ├── generate_answer.py ├── images ├── .DS_Store ├── image_RiYKWHwtQa.png ├── image_fChhMjnifo.png └── image_tL0rUhQiZB.png ├── models ├── Baichuan2-7B-Chat │ └── README.md ├── Qwen2-7B-Instruct │ └── README.md └── chatglm3-6b │ └── README.md ├── pdf_parse.py ├── pre_train_model ├── bce-reranker-base_v1 │ └── README.md ├── bge-m3 │ └── README.md ├── bge-reranker-large │ └── README.md ├── m3e-large │ └── README.md └── text2vec-base-chinese │ └── README.md ├── requirements.txt ├── rerank_model.py ├── retriever ├── bge_retriever.py ├── bm25_retriever.py ├── m3e_retriever.py └── tfidf_retriever.py ├── run.py ├── test_score.py └── vllm_model.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangzg1/rag_with_chat/HEAD/.gitignore -------------------------------------------------------------------------------- /.idea/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangzg1/rag_with_chat/HEAD/.idea/.gitignore -------------------------------------------------------------------------------- /.idea/inspectionProfiles/Project_Default.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangzg1/rag_with_chat/HEAD/.idea/inspectionProfiles/Project_Default.xml -------------------------------------------------------------------------------- /.idea/inspectionProfiles/profiles_settings.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangzg1/rag_with_chat/HEAD/.idea/inspectionProfiles/profiles_settings.xml -------------------------------------------------------------------------------- /.idea/misc.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangzg1/rag_with_chat/HEAD/.idea/misc.xml -------------------------------------------------------------------------------- /.idea/modules.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangzg1/rag_with_chat/HEAD/.idea/modules.xml -------------------------------------------------------------------------------- /.idea/rag_with_chat.iml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangzg1/rag_with_chat/HEAD/.idea/rag_with_chat.iml -------------------------------------------------------------------------------- /.idea/vcs.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangzg1/rag_with_chat/HEAD/.idea/vcs.xml -------------------------------------------------------------------------------- /Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangzg1/rag_with_chat/HEAD/Dockerfile -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangzg1/rag_with_chat/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangzg1/rag_with_chat/HEAD/README.md -------------------------------------------------------------------------------- /benchmark/bench_data.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangzg1/rag_with_chat/HEAD/benchmark/bench_data.json -------------------------------------------------------------------------------- /benchmark/benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangzg1/rag_with_chat/HEAD/benchmark/benchmark.py -------------------------------------------------------------------------------- /benchmark/model_serve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangzg1/rag_with_chat/HEAD/benchmark/model_serve.py -------------------------------------------------------------------------------- /chatgpt_proxy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangzg1/rag_with_chat/HEAD/chatgpt_proxy.py -------------------------------------------------------------------------------- /config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangzg1/rag_with_chat/HEAD/config.py -------------------------------------------------------------------------------- /data/car_user_manual.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangzg1/rag_with_chat/HEAD/data/car_user_manual.pdf -------------------------------------------------------------------------------- /data/gold_result.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangzg1/rag_with_chat/HEAD/data/gold_result.json -------------------------------------------------------------------------------- /data/test_question.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangzg1/rag_with_chat/HEAD/data/test_question.json -------------------------------------------------------------------------------- /example_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangzg1/rag_with_chat/HEAD/example_test.py -------------------------------------------------------------------------------- /generate_answer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangzg1/rag_with_chat/HEAD/generate_answer.py -------------------------------------------------------------------------------- /images/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangzg1/rag_with_chat/HEAD/images/.DS_Store -------------------------------------------------------------------------------- /images/image_RiYKWHwtQa.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangzg1/rag_with_chat/HEAD/images/image_RiYKWHwtQa.png -------------------------------------------------------------------------------- /images/image_fChhMjnifo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangzg1/rag_with_chat/HEAD/images/image_fChhMjnifo.png -------------------------------------------------------------------------------- /images/image_tL0rUhQiZB.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangzg1/rag_with_chat/HEAD/images/image_tL0rUhQiZB.png -------------------------------------------------------------------------------- /models/Baichuan2-7B-Chat/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangzg1/rag_with_chat/HEAD/models/Baichuan2-7B-Chat/README.md -------------------------------------------------------------------------------- /models/Qwen2-7B-Instruct/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangzg1/rag_with_chat/HEAD/models/Qwen2-7B-Instruct/README.md -------------------------------------------------------------------------------- /models/chatglm3-6b/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangzg1/rag_with_chat/HEAD/models/chatglm3-6b/README.md -------------------------------------------------------------------------------- /pdf_parse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangzg1/rag_with_chat/HEAD/pdf_parse.py -------------------------------------------------------------------------------- /pre_train_model/bce-reranker-base_v1/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangzg1/rag_with_chat/HEAD/pre_train_model/bce-reranker-base_v1/README.md -------------------------------------------------------------------------------- /pre_train_model/bge-m3/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangzg1/rag_with_chat/HEAD/pre_train_model/bge-m3/README.md -------------------------------------------------------------------------------- /pre_train_model/bge-reranker-large/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangzg1/rag_with_chat/HEAD/pre_train_model/bge-reranker-large/README.md -------------------------------------------------------------------------------- /pre_train_model/m3e-large/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangzg1/rag_with_chat/HEAD/pre_train_model/m3e-large/README.md -------------------------------------------------------------------------------- /pre_train_model/text2vec-base-chinese/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangzg1/rag_with_chat/HEAD/pre_train_model/text2vec-base-chinese/README.md -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangzg1/rag_with_chat/HEAD/requirements.txt -------------------------------------------------------------------------------- /rerank_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangzg1/rag_with_chat/HEAD/rerank_model.py -------------------------------------------------------------------------------- /retriever/bge_retriever.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangzg1/rag_with_chat/HEAD/retriever/bge_retriever.py -------------------------------------------------------------------------------- /retriever/bm25_retriever.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangzg1/rag_with_chat/HEAD/retriever/bm25_retriever.py -------------------------------------------------------------------------------- /retriever/m3e_retriever.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangzg1/rag_with_chat/HEAD/retriever/m3e_retriever.py -------------------------------------------------------------------------------- /retriever/tfidf_retriever.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangzg1/rag_with_chat/HEAD/retriever/tfidf_retriever.py -------------------------------------------------------------------------------- /run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangzg1/rag_with_chat/HEAD/run.py -------------------------------------------------------------------------------- /test_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangzg1/rag_with_chat/HEAD/test_score.py -------------------------------------------------------------------------------- /vllm_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhangzg1/rag_with_chat/HEAD/vllm_model.py --------------------------------------------------------------------------------