'commit'

dsLightRag/T1_Train.py

@@ -4,17 +4,12 @@ from Util.DocxUtil import get_docx_content_by_pandoc
 from Util.LightRagUtil import configure_logging, initialize_rag
 import os
 
-# 化学
-#WORKING_DIR = "./Topic/Chemistry"
-#docx_file = 'static/Txt/化学方程式.docx'
-
-# 苏轼
-#WORKING_DIR = "./Topic/Chinese"
-#docx_file = 'static/Txt/苏轼.docx'
-
 # 数学
-WORKING_DIR = "./Topic/Math"
-docx_file = 'static/Txt/小学数学教学中的若干问题.docx'
+KEMU = 'Math'  # Chinese,Chemistry
+
+# 组装文件路径
+WORKING_DIR = "./Topic/" + KEMU
+docx_file = 'static/Txt/' + KEMU + '.docx'
 
 async def main():
     # 注释掉或删除以下清理代码