diff --git a/rag/src/config/config.py b/rag/src/config/config.py
index 5b81b72..366cf85 100644
--- a/rag/src/config/config.py
+++ b/rag/src/config/config.py
@@ -1,37 +1,38 @@
-import os
-
-cur_dir = os.path.dirname(os.path.abspath(__file__))                # config
-src_dir = os.path.dirname(cur_dir)                                  # src
-base_dir = os.path.dirname(src_dir)                                 # base
-model_repo = 'ajupyter/EmoLLM_aiwei'
-
-# model
-model_dir = os.path.join(base_dir, 'model')                         # model
-embedding_path = os.path.join(model_dir, 'gte-small-zh')            # embedding
-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')        # 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
-system_prompt = """
-	你是一个拥有丰富心理学知识的温柔邻家温柔大姐姐艾薇，我有一些心理问题，请你用专业的知识和温柔、可爱、俏皮、的口吻帮我解决，回复中可以穿插一些可爱的Emoji表情符号或者文本符号。\n
-"""
-prompt_template = """
-	{system_prompt}
-	根据下面检索回来的信息，回答问题。
-	{content}
-	问题：{query}
+import os
+
+cur_dir = os.path.dirname(os.path.abspath(__file__))                # config
+src_dir = os.path.dirname(cur_dir)                                  # src
+base_dir = os.path.dirname(src_dir)                                 # base
+model_repo = 'ajupyter/EmoLLM_aiwei'
+
+# model
+model_dir = os.path.join(base_dir, 'model')                         # model
+embedding_path = os.path.join(model_dir, 'embedding_model')         # embedding
+rerank_path = os.path.join(model_dir, 'rerank_model')  	        	# embedding
+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')        # 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
+system_prompt = """
+	你是一个拥有丰富心理学知识的温柔邻家温柔大姐姐艾薇，我有一些心理问题，请你用专业的知识和温柔、可爱、俏皮、的口吻帮我解决，回复中可以穿插一些可爱的Emoji表情符号或者文本符号。\n
+"""
+prompt_template = """
+	{system_prompt}
+	根据下面检索回来的信息，回答问题。
+	{content}
+	问题：{query}
 """
\ No newline at end of file