From 695bdc2adb343f145944c0bf9d5fc1313c3a7e62 Mon Sep 17 00:00:00 2001
From: HuangHai <10402852@qq.com>
Date: Mon, 7 Jul 2025 07:41:58 +0800
Subject: [PATCH] 'commit'

---
 dsRagAnything/T1_Train.py | 18 +++++++++++-------
 1 file changed, 11 insertions(+), 7 deletions(-)

diff --git a/dsRagAnything/T1_Train.py b/dsRagAnything/T1_Train.py
index a6f52177..164a981c 100644
--- a/dsRagAnything/T1_Train.py
+++ b/dsRagAnything/T1_Train.py
@@ -6,34 +6,38 @@ from Util.RagUtil import create_llm_model_func, create_vision_model_func, create
 
 
 async def main():
+    # 要处理的文件路径
+    file_path = "./Txt/驿来特平台安全.docx"
+
+    # 索引生成目录
+    WORKING_DIR = "./rag_storage"
 
+    # 指定最终的索引生成目录，启动索引生成
     config = RAGAnythingConfig(
-        working_dir="./rag_storage",
+        working_dir=WORKING_DIR,
         mineru_parse_method="auto",
         enable_image_processing=True,
         enable_table_processing=True,
         enable_equation_processing=True,
     )
-
+    # 自定义的大模型函数
     llm_model_func = create_llm_model_func()
+    # 自定义的可视模型函数
     vision_model_func = create_vision_model_func(llm_model_func)
+    # 自定义的嵌入函数
     embedding_func = create_embedding_func()
-
     rag = RAGAnything(
         config=config,
         llm_model_func=llm_model_func,
         vision_model_func=vision_model_func,
         embedding_func=embedding_func,
     )
-
-    file_path = "./Txt/驿来特平台安全.docx"
     await rag.process_document_complete(
         file_path=file_path,
         output_dir="./output",
         parse_method="auto"
     )
-
-    print("Processing complete.")
+    print("文档解析索引完成！")
 
 if __name__ == "__main__":
     asyncio.run(main())