From 848ea1eb63058b0e1f3cb93bd9050a42129d12a5 Mon Sep 17 00:00:00 2001
From: Raven95676 <Raven95676@gmail.com>
Date: Wed, 9 Apr 2025 16:37:19 +0800
Subject: [PATCH] =?UTF-8?q?=E6=8F=90=E5=8D=87=E5=81=A5=E5=A3=AE=E6=80=A7?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 astrbot/core/star/star_tools.py | 42 +++++++++++++++++++++++----------
 1 file changed, 29 insertions(+), 13 deletions(-)

diff --git a/astrbot/core/star/star_tools.py b/astrbot/core/star/star_tools.py
index 410a49eed..fb850aa5e 100644
--- a/astrbot/core/star/star_tools.py
+++ b/astrbot/core/star/star_tools.py
@@ -147,30 +147,46 @@ class StarTools:
         cls._context.unregister_llm_tool(name)
 
     @classmethod
-    def get_data_dir(cls) -> Path:
+    def get_data_dir(cls, plugin_name: Optional[str]) -> Path:
         """
-        为调用者创建并返回数据目录路径。
+        返回插件数据目录的绝对路径。
+
+        此方法会在 data/plugin_data 目录下为插件创建一个专属的数据目录。如果未提供插件名称，
+        会自动从调用栈中获取插件信息。
+
+        Args:
+            plugin_name: 可选的插件名称。如果为None，将自动检测调用者的插件名称。
 
         Returns:
-            Path: 数据目录的绝对路径
+            Path (Path): 插件数据目录的绝对路径，位于 data/plugin_data/{plugin_name}。
+
+        Raises:
+            RuntimeError: 当出现以下情况时抛出:
+                - 无法获取调用者模块信息
+                - 无法获取模块的元数据信息
+                - 创建目录失败（权限不足或其他IO错误）
         """
+        if not plugin_name:
+            frame = inspect.currentframe().f_back
+            module = inspect.getmodule(frame)
 
-        frame = inspect.currentframe().f_back
-        module = inspect.getmodule(frame)
+            if not module:
+                raise RuntimeError("无法获取调用者模块信息")
 
-        if not module:
-            raise RuntimeError("无法获取调用者模块信息")
+            metadata = star_map.get(module.__name__, None)
 
-        metadata = star_map.get(module.__name__, None)
+            if not metadata:
+                raise RuntimeError(f"无法获取模块 {module.__name__} 的元数据信息")
 
-        if not metadata:
-            raise RuntimeError(f"无法获取模块 {module.__name__} 的元数据信息")
+            plugin_name = metadata.name
 
-        data_dir = Path("data/plugin_data") / metadata.name
+        data_dir = Path("data/plugin_data") / plugin_name
 
         try:
             data_dir.mkdir(parents=True, exist_ok=True)
-        except PermissionError:
-            raise RuntimeError(f"无法创建目录 {data_dir}：权限不足")
+        except OSError as e:
+            if isinstance(e, PermissionError):
+                raise RuntimeError(f"无法创建目录 {data_dir}：权限不足") from e
+            raise RuntimeError(f"无法创建目录 {data_dir}：{e!s}") from e
 
         return data_dir.resolve()