将主函数中的文件路径硬编码更改为从环境变量加载

2025-07-23 21:54:22 +08:00
parent d8078c6d09
commit 466fae53c9
1 changed files with 2 additions and 1 deletions
--- a/src/add_chunk_cli_pdf_img.py
+++ b/src/add_chunk_cli_pdf_img.py
@@ -380,7 +380,8 @@ def main():
    dataset.id = bucket_name
    chunk_id = object_name
    """
-    file_path = "F:\\Synology_nas\\SynologyDrive\\大模型\\厦门市城市道路开口设置指引DB3502T 141-2024\\"
+    file_path = os.getenv("FILE_PATH")
+    
    pdf_dict, txt_dict = pair_pdf_and_txt(file_path, file_path)
    
    if not pdf_dict: