From bd92aac28052e3172698b19496275202abb520a2 Mon Sep 17 00:00:00 2001
From: Soulter <905617992@qq.com>
Date: Sat, 29 Mar 2025 18:31:07 +0800
Subject: [PATCH] =?UTF-8?q?=E2=9C=A8=20feat:=20=E6=94=AF=E6=8C=81=20/llm?=
 =?UTF-8?q?=20=E6=8C=87=E4=BB=A4=E5=BF=AB=E6=8D=B7=E5=90=AF=E5=81=9C=20LLM?=
 =?UTF-8?q?=20=E5=8A=9F=E8=83=BD=20#296?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 packages/astrbot/main.py | 21 +++++++++++++++++----
 1 file changed, 17 insertions(+), 4 deletions(-)

diff --git a/packages/astrbot/main.py b/packages/astrbot/main.py
index 30d6d99c..fc5ee3f3 100644
--- a/packages/astrbot/main.py
+++ b/packages/astrbot/main.py
@@ -88,6 +88,7 @@ class Main(star.Star):
 /alter_cmd: 设置指令权限(op)
 
 [大模型]
+/llm: 开启/关闭 LLM
 /provider: 大模型提供商
 /model: 模型列表
 /ls: 对话列表
@@ -96,7 +97,7 @@ class Main(star.Star):
 /switch 序号: 切换对话
 /rename 新名字: 重命名当前对话
 /del: 删除当前会话对话(op)
-/reset: 重置 LLM 会话(op)
+/reset: 重置 LLM 会话
 /history: 当前对话的对话记录
 /persona: 人格情景(op)
 /tool ls: 函数工具
@@ -106,6 +107,20 @@ class Main(star.Star):
 
         event.set_result(MessageEventResult().message(msg).use_t2i(False))
 
+    @filter.command("llm")
+    async def llm(self, event: AstrMessageEvent):
+        """开启/关闭 LLM"""
+        cfg = self.context.get_config()
+        enable = cfg["provider_settings"]["enable"]
+        if enable:
+            cfg["provider_settings"]["enable"] = False
+            status = "关闭"
+        else:
+            cfg["provider_settings"]["enable"] = True
+            status = "开启"
+        cfg.save_config()
+        yield event.plain_result(f"{status} LLM 聊天功能。")
+
     @filter.command_group("tool")
     def tool(self):
         pass
@@ -529,9 +544,7 @@ UID: {user_id} 此 ID 可用于设置管理员。
             try:
                 models = await self.context.get_using_provider().get_models()
             except BaseException as e:
-                err_msg = api_key_pattern.sub(
-                    "key=***", str(e)
-                )
+                err_msg = api_key_pattern.sub("key=***", str(e))
                 message.set_result(
                     MessageEventResult()
                     .message("获取模型列表失败: " + err_msg)