From 6b6577006d21d27332d21304a0d4d08a531f98da Mon Sep 17 00:00:00 2001
From: anka <1350989414@qq.com>
Date: Wed, 26 Mar 2025 17:59:30 +0800
Subject: [PATCH] =?UTF-8?q?perf:=20=E6=A0=BC=E5=BC=8F=E5=8C=96?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 astrbot/core/db/po.py                                     | 8 ++++++++
 astrbot/core/pipeline/process_stage/method/llm_request.py | 6 +++---
 astrbot/core/pipeline/result_decorate/stage.py            | 4 +++-
 3 files changed, 14 insertions(+), 4 deletions(-)

diff --git a/astrbot/core/db/po.py b/astrbot/core/db/po.py
index 59041d6dd..49adb2781 100644
--- a/astrbot/core/db/po.py
+++ b/astrbot/core/db/po.py
@@ -6,6 +6,8 @@ from typing import List
 
 @dataclass
 class Platform:
+    """平台使用统计数据"""
+
     name: str
     count: int
     timestamp: int
@@ -13,6 +15,8 @@ class Platform:
 
 @dataclass
 class Provider:
+    """供应商使用统计数据"""
+
     name: str
     count: int
     timestamp: int
@@ -20,6 +24,8 @@ class Provider:
 
 @dataclass
 class Plugin:
+    """插件使用统计数据"""
+
     name: str
     count: int
     timestamp: int
@@ -27,6 +33,8 @@ class Plugin:
 
 @dataclass
 class Command:
+    """命令使用统计数据"""
+
     name: str
     count: int
     timestamp: int
diff --git a/astrbot/core/pipeline/process_stage/method/llm_request.py b/astrbot/core/pipeline/process_stage/method/llm_request.py
index 8d606d9e5..de2460bdc 100644
--- a/astrbot/core/pipeline/process_stage/method/llm_request.py
+++ b/astrbot/core/pipeline/process_stage/method/llm_request.py
@@ -58,9 +58,9 @@ class LLMRequestSubStage(Stage):
 
         if event.get_extra("provider_request"):
             req = event.get_extra("provider_request")
-            assert isinstance(req, ProviderRequest), (
-                "provider_request 必须是 ProviderRequest 类型。"
-            )
+            assert isinstance(
+                req, ProviderRequest
+            ), "provider_request 必须是 ProviderRequest 类型。"
 
             if req.conversation:
                 req.contexts = json.loads(req.conversation.history)
diff --git a/astrbot/core/pipeline/result_decorate/stage.py b/astrbot/core/pipeline/result_decorate/stage.py
index d7bb9583c..4894b2e03 100644
--- a/astrbot/core/pipeline/result_decorate/stage.py
+++ b/astrbot/core/pipeline/result_decorate/stage.py
@@ -156,7 +156,9 @@ class ResultDecorateStage(Stage):
                 self.ctx.astrbot_config["provider_tts_settings"]["enable"]
                 and result.is_llm_result()
             ):
-                tts_provider = self.ctx.plugin_manager.context.provider_manager.curr_tts_provider_inst
+                tts_provider = (
+                    self.ctx.plugin_manager.context.provider_manager.curr_tts_provider_inst
+                )
                 new_chain = []
                 for comp in result.chain:
                     if isinstance(comp, Plain) and len(comp.text) > 1: