From ddb7ef4d93350b44b04e53986f2085c9f9e3cc57 Mon Sep 17 00:00:00 2001
From: minecraft1024a <wwwww95915@qq.com>
Date: Sat, 27 Sep 2025 14:13:09 +0800
Subject: [PATCH] =?UTF-8?q?feat(tool):=20=E4=B8=BA=E5=AD=A6=E4=B9=A0?=
 =?UTF-8?q?=E5=B7=A5=E5=85=B7=E6=B7=BB=E5=8A=A0=E7=BC=93=E5=AD=98=E6=B8=85?=
 =?UTF-8?q?=E7=90=86=E5=8A=9F=E8=83=BD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

为 lpmm_learning_tool.py 脚本添加了清理缓存的选项。
用户现在可以通过菜单选项 '6' 来删除 `temp/lpmm_cache` 目录下的所有临时文件。

此功能有助于释放磁盘空间，并可以在缓存数据陈旧或损坏时进行重置，提高了工具的可维护性。
---
 scripts/lpmm_learning_tool.py | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

diff --git a/scripts/lpmm_learning_tool.py b/scripts/lpmm_learning_tool.py
index 3fe26eb93..9a67a82ee 100644
--- a/scripts/lpmm_learning_tool.py
+++ b/scripts/lpmm_learning_tool.py
@@ -1,5 +1,6 @@
 import asyncio
 import os
+import shutil
 import sys
 import orjson
 import datetime
@@ -36,6 +37,21 @@ OPENIE_OUTPUT_DIR = os.path.join(ROOT_PATH, "data", "openie")
 TEMP_DIR = os.path.join(ROOT_PATH, "temp", "lpmm_cache")
 file_lock = Lock()
 
+# --- 缓存清理 ---
+
+def clear_cache():
+    """清理 lpmm_learning_tool.py 生成的缓存文件"""
+    logger.info("--- 开始清理缓存 ---")
+    if os.path.exists(TEMP_DIR):
+        try:
+            shutil.rmtree(TEMP_DIR)
+            logger.info(f"成功删除缓存目录: {TEMP_DIR}")
+        except OSError as e:
+            logger.error(f"删除缓存时出错: {e}")
+    else:
+        logger.info("缓存目录不存在，无需清理。")
+    logger.info("--- 缓存清理完成 ---")
+
 # --- 模块一：数据预处理 ---
 
 
@@ -274,6 +290,7 @@ def main():
     print("3. [数据导入] -> 从 openie 文件夹自动导入最新知识")
     print("4. [全流程] -> 按顺序执行 1 -> 2 -> 3")
     print("5. [指定导入] -> 从特定的 openie.json 文件导入知识")
+    print("6. [清理缓存] -> 删除所有已提取信息的缓存")
     print("0. [退出]")
     print("-" * 30)
     choice = input("请输入你的选择 (0-5): ").strip()
@@ -293,6 +310,8 @@ def main():
             asyncio.run(import_data())
     elif choice == "5":
         import_from_specific_file()
+    elif choice == "6":
+        clear_cache()
     elif choice == "0":
         sys.exit(0)
     else: