From ce7a4ae416bc1bf406f14aa9774752ffbf885eb4 Mon Sep 17 00:00:00 2001 From: zealot52099 Date: Mon, 18 Mar 2024 10:31:34 +0800 Subject: [PATCH] add more directories --- rag/src/config/config.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/rag/src/config/config.py b/rag/src/config/config.py index b84327f..d803d64 100644 --- a/rag/src/config/config.py +++ b/rag/src/config/config.py @@ -13,11 +13,16 @@ llm_path = os.path.join(model_dir, 'pythia-14m') # llm # data data_dir = os.path.join(base_dir, 'data') # data knowledge_json_path = os.path.join(data_dir, 'knowledge.json') # json -knowledge_pkl_path = os.path.join(data_dir, 'knowledge.pkl') # pickle +knowledge_pkl_path = os.path.join(data_dir, 'knowledge.pkl') # pkl +doc_dir = os.path.join(data_dir, 'txt') +qa_dir = os.path.join(data_dir, 'json') # log log_dir = os.path.join(base_dir, 'log') # log log_path = os.path.join(log_dir, 'log.log') # file +# vector DB +vector_db_dir = os.path.join(data_dir, 'vector_db.pkl') + select_num = 3 retrieval_num = 10 \ No newline at end of file