AstrBot 4.5.1__py3-none-any.whl → 4.5.2__py3-none-any.whl

astrbot/core/provider/sources/vllm_rerank_source.py

@@ -1,8 +1,10 @@
 import aiohttp
+
 from astrbot import logger
+
+from ..entities import ProviderType, RerankResult
 from ..provider import RerankProvider
 from ..register import register_provider_adapter
-from ..entities import ProviderType, RerankResult
 
 
 @register_provider_adapter(
@@ -30,7 +32,10 @@ class VLLMRerankProvider(RerankProvider):
         )
 
     async def rerank(
-        self, query: str, documents: list[str], top_n: int | None = None
+        self,
+        query: str,
+        documents: list[str],
+        top_n: int | None = None,
     ) -> list[RerankResult]:
         payload = {
             "query": query,
@@ -40,14 +45,15 @@ class VLLMRerankProvider(RerankProvider):
         if top_n is not None:
             payload["top_n"] = top_n
         async with self.client.post(
-            f"{self.base_url}/v1/rerank", json=payload
+            f"{self.base_url}/v1/rerank",
+            json=payload,
         ) as response:
             response_data = await response.json()
             results = response_data.get("results", [])
 
             if not results:
                 logger.warning(
-                    f"Rerank API 返回了空的列表数据。原始响应: {response_data}"
+                    f"Rerank API 返回了空的列表数据。原始响应: {response_data}",
                 )
 
             return [

astrbot/core/provider/sources/volcengine_tts.py

@@ -1,18 +1,23 @@
-import uuid
+import asyncio
 import base64
 import json
 import os
 import traceback
-import asyncio
+import uuid
+
 import aiohttp
-from ..provider import TTSProvider
+
+from astrbot import logger
+
 from ..entities import ProviderType
+from ..provider import TTSProvider
 from ..register import register_provider_adapter
-from astrbot import logger
 
 
 @register_provider_adapter(
-    "volcengine_tts", "火山引擎 TTS", provider_type=ProviderType.TEXT_TO_SPEECH
+    "volcengine_tts",
+    "火山引擎 TTS",
+    provider_type=ProviderType.TEXT_TO_SPEECH,
 )
 class ProviderVolcengineTTS(TTSProvider):
     def __init__(self, provider_config: dict, provider_settings: dict) -> None:
@@ -23,7 +28,8 @@ class ProviderVolcengineTTS(TTSProvider):
         self.voice_type = provider_config.get("volcengine_voice_type", "")
         self.speed_ratio = provider_config.get("volcengine_speed_ratio", 1.0)
         self.api_base = provider_config.get(
-            "api_base", "https://openspeech.bytedance.com/api/v1/tts"
+            "api_base",
+            "https://openspeech.bytedance.com/api/v1/tts",
         )
         self.timeout = provider_config.get("timeout", 20)
 
@@ -66,43 +72,44 @@ class ProviderVolcengineTTS(TTSProvider):
         logger.debug(f"请求体: {json.dumps(payload, ensure_ascii=False)[:100]}...")
 
         try:
-            async with aiohttp.ClientSession() as session:
-                async with session.post(
+            async with (
+                aiohttp.ClientSession() as session,
+                session.post(
                     self.api_base,
                     data=json.dumps(payload),
                     headers=headers,
                     timeout=self.timeout,
-                ) as response:
-                    logger.debug(f"响应状态码: {response.status}")
-
-                    response_text = await response.text()
-                    logger.debug(f"响应内容: {response_text[:200]}...")
+                ) as response,
+            ):
+                logger.debug(f"响应状态码: {response.status}")
 
-                    if response.status == 200:
-                        resp_data = json.loads(response_text)
+                response_text = await response.text()
+                logger.debug(f"响应内容: {response_text[:200]}...")
 
-                        if "data" in resp_data:
-                            audio_data = base64.b64decode(resp_data["data"])
+                if response.status == 200:
+                    resp_data = json.loads(response_text)
 
-                            os.makedirs("data/temp", exist_ok=True)
+                    if "data" in resp_data:
+                        audio_data = base64.b64decode(resp_data["data"])
 
-                            file_path = f"data/temp/volcengine_tts_{uuid.uuid4()}.mp3"
+                        os.makedirs("data/temp", exist_ok=True)
 
-                            loop = asyncio.get_running_loop()
-                            await loop.run_in_executor(
-                                None, lambda: open(file_path, "wb").write(audio_data)
-                            )
+                        file_path = f"data/temp/volcengine_tts_{uuid.uuid4()}.mp3"
 
-                            return file_path
-                        else:
-                            error_msg = resp_data.get("message", "未知错误")
-                            raise Exception(f"火山引擎 TTS API 返回错误: {error_msg}")
-                    else:
-                        raise Exception(
-                            f"火山引擎 TTS API 请求失败: {response.status}, {response_text}"
+                        loop = asyncio.get_running_loop()
+                        await loop.run_in_executor(
+                            None,
+                            lambda: open(file_path, "wb").write(audio_data),
                         )
 
+                        return file_path
+                    error_msg = resp_data.get("message", "未知错误")
+                    raise Exception(f"火山引擎 TTS API 返回错误: {error_msg}")
+                raise Exception(
+                    f"火山引擎 TTS API 请求失败: {response.status}, {response_text}",
+                )
+
         except Exception as e:
             error_details = traceback.format_exc()
             logger.debug(f"火山引擎 TTS 异常详情: {error_details}")
-            raise Exception(f"火山引擎 TTS 异常: {str(e)}")
+            raise Exception(f"火山引擎 TTS 异常: {e!s}")

astrbot/core/provider/sources/whisper_api_source.py

@@ -1,13 +1,16 @@
-import uuid
 import os
-from openai import AsyncOpenAI, NOT_GIVEN
-from ..provider import STTProvider
-from ..entities import ProviderType
-from astrbot.core.utils.io import download_file
-from ..register import register_provider_adapter
+import uuid
+
+from openai import NOT_GIVEN, AsyncOpenAI
+
 from astrbot.core import logger
-from astrbot.core.utils.tencent_record_helper import tencent_silk_to_wav
 from astrbot.core.utils.astrbot_path import get_astrbot_data_path
+from astrbot.core.utils.io import download_file
+from astrbot.core.utils.tencent_record_helper import tencent_silk_to_wav
+
+from ..entities import ProviderType
+from ..provider import STTProvider
+from ..register import register_provider_adapter
 
 
 @register_provider_adapter(
@@ -26,11 +29,11 @@ class ProviderOpenAIWhisperAPI(STTProvider):
 
         self.client = AsyncOpenAI(
             api_key=self.chosen_api_key,
-            base_url=provider_config.get("api_base", None),
+            base_url=provider_config.get("api_base"),
             timeout=provider_config.get("timeout", NOT_GIVEN),
         )
 
-        self.set_model(provider_config.get("model", None))
+        self.set_model(provider_config.get("model"))
 
     async def _is_silk_file(self, file_path):
         silk_header = b"SILK"
@@ -39,11 +42,10 @@ class ProviderOpenAIWhisperAPI(STTProvider):
 
         if silk_header in file_header:
             return True
-        else:
-            return False
+        return False
 
     async def get_text(self, audio_url: str) -> str:
-        """only supports mp3, mp4, mpeg, m4a, wav, webm"""
+        """Only supports mp3, mp4, mpeg, m4a, wav, webm"""
         is_tencent = False
 
         if audio_url.startswith("http"):

astrbot/core/provider/sources/whisper_selfhosted_source.py

@@ -1,14 +1,17 @@
-import uuid
-import os
 import asyncio
+import os
+import uuid
+
 import whisper
-from ..provider import STTProvider
-from ..entities import ProviderType
-from astrbot.core.utils.io import download_file
-from ..register import register_provider_adapter
+
 from astrbot.core import logger
-from astrbot.core.utils.tencent_record_helper import tencent_silk_to_wav
 from astrbot.core.utils.astrbot_path import get_astrbot_data_path
+from astrbot.core.utils.io import download_file
+from astrbot.core.utils.tencent_record_helper import tencent_silk_to_wav
+
+from ..entities import ProviderType
+from ..provider import STTProvider
+from ..register import register_provider_adapter
 
 
 @register_provider_adapter(
@@ -23,14 +26,16 @@ class ProviderOpenAIWhisperSelfHost(STTProvider):
         provider_settings: dict,
     ) -> None:
         super().__init__(provider_config, provider_settings)
-        self.set_model(provider_config.get("model", None))
+        self.set_model(provider_config.get("model"))
         self.model = None
 
     async def initialize(self):
         loop = asyncio.get_event_loop()
         logger.info("下载或者加载 Whisper 模型中，这可能需要一些时间 ...")
         self.model = await loop.run_in_executor(
-            None, whisper.load_model, self.model_name
+            None,
+            whisper.load_model,
+            self.model_name,
         )
         logger.info("Whisper 模型加载完成。")
 
@@ -41,8 +46,7 @@ class ProviderOpenAIWhisperSelfHost(STTProvider):
 
         if silk_header in file_header:
             return True
-        else:
-            return False
+        return False
 
     async def get_text(self, audio_url: str) -> str:
         loop = asyncio.get_event_loop()

astrbot/core/provider/sources/xinference_rerank_source.py

@@ -1,10 +1,12 @@
 from xinference_client.client.restful.async_restful_client import (
     AsyncClient as Client,
 )
+
 from astrbot import logger
+
+from ..entities import ProviderType, RerankResult
 from ..provider import RerankProvider
 from ..register import register_provider_adapter
-from ..entities import ProviderType, RerankResult
 
 
 @register_provider_adapter(
@@ -23,7 +25,8 @@ class XinferenceRerankProvider(RerankProvider):
         self.model_name = provider_config.get("rerank_model", "BAAI/bge-reranker-base")
         self.api_key = provider_config.get("rerank_api_key")
         self.launch_model_if_not_running = provider_config.get(
-            "launch_model_if_not_running", False
+            "launch_model_if_not_running",
+            False,
         )
         self.client = None
         self.model = None
@@ -42,7 +45,7 @@ class XinferenceRerankProvider(RerankProvider):
             for uid, model_spec in running_models.items():
                 if model_spec.get("model_name") == self.model_name:
                     logger.info(
-                        f"Model '{self.model_name}' is already running with UID: {uid}"
+                        f"Model '{self.model_name}' is already running with UID: {uid}",
                     )
                     self.model_uid = uid
                     break
@@ -51,12 +54,13 @@ class XinferenceRerankProvider(RerankProvider):
                 if self.launch_model_if_not_running:
                     logger.info(f"Launching {self.model_name} model...")
                     self.model_uid = await self.client.launch_model(
-                        model_name=self.model_name, model_type="rerank"
+                        model_name=self.model_name,
+                        model_type="rerank",
                     )
                     logger.info("Model launched.")
                 else:
                     logger.warning(
-                        f"Model '{self.model_name}' is not running and auto-launch is disabled. Provider will not be available."
+                        f"Model '{self.model_name}' is not running and auto-launch is disabled. Provider will not be available.",
                     )
                     return
 
@@ -66,12 +70,16 @@ class XinferenceRerankProvider(RerankProvider):
         except Exception as e:
             logger.error(f"Failed to initialize Xinference model: {e}")
             logger.debug(
-                f"Xinference initialization failed with exception: {e}", exc_info=True
+                f"Xinference initialization failed with exception: {e}",
+                exc_info=True,
             )
             self.model = None
 
     async def rerank(
-        self, query: str, documents: list[str], top_n: int | None = None
+        self,
+        query: str,
+        documents: list[str],
+        top_n: int | None = None,
     ) -> list[RerankResult]:
         if not self.model:
             logger.error("Xinference rerank model is not initialized.")
@@ -83,7 +91,7 @@ class XinferenceRerankProvider(RerankProvider):
 
             if not results:
                 logger.warning(
-                    f"Rerank API returned an empty list. Original response: {response}"
+                    f"Rerank API returned an empty list. Original response: {response}",
                 )
 
             return [

astrbot/core/provider/sources/xinference_stt_provider.py

@@ -1,15 +1,18 @@
-import uuid
 import os
+import uuid
+
 import aiohttp
 from xinference_client.client.restful.async_restful_client import (
     AsyncClient as Client,
 )
-from ..provider import STTProvider
-from ..entities import ProviderType
-from ..register import register_provider_adapter
+
 from astrbot.core import logger
-from astrbot.core.utils.tencent_record_helper import tencent_silk_to_wav
 from astrbot.core.utils.astrbot_path import get_astrbot_data_path
+from astrbot.core.utils.tencent_record_helper import tencent_silk_to_wav
+
+from ..entities import ProviderType
+from ..provider import STTProvider
+from ..register import register_provider_adapter
 
 
 @register_provider_adapter(
@@ -28,7 +31,8 @@ class ProviderXinferenceSTT(STTProvider):
         self.model_name = provider_config.get("model", "whisper-large-v3")
         self.api_key = provider_config.get("api_key")
         self.launch_model_if_not_running = provider_config.get(
-            "launch_model_if_not_running", False
+            "launch_model_if_not_running",
+            False,
         )
         self.client = None
         self.model_uid = None
@@ -46,7 +50,7 @@ class ProviderXinferenceSTT(STTProvider):
             for uid, model_spec in running_models.items():
                 if model_spec.get("model_name") == self.model_name:
                     logger.info(
-                        f"Model '{self.model_name}' is already running with UID: {uid}"
+                        f"Model '{self.model_name}' is already running with UID: {uid}",
                     )
                     self.model_uid = uid
                     break
@@ -55,19 +59,21 @@ class ProviderXinferenceSTT(STTProvider):
                 if self.launch_model_if_not_running:
                     logger.info(f"Launching {self.model_name} model...")
                     self.model_uid = await self.client.launch_model(
-                        model_name=self.model_name, model_type="audio"
+                        model_name=self.model_name,
+                        model_type="audio",
                     )
                     logger.info("Model launched.")
                 else:
                     logger.warning(
-                        f"Model '{self.model_name}' is not running and auto-launch is disabled. Provider will not be available."
+                        f"Model '{self.model_name}' is not running and auto-launch is disabled. Provider will not be available.",
                     )
                     return
 
         except Exception as e:
             logger.error(f"Failed to initialize Xinference model: {e}")
             logger.debug(
-                f"Xinference initialization failed with exception: {e}", exc_info=True
+                f"Xinference initialization failed with exception: {e}",
+                exc_info=True,
             )
 
     async def get_text(self, audio_url: str) -> str:
@@ -90,16 +96,15 @@ class ProviderXinferenceSTT(STTProvider):
                             audio_bytes = await resp.read()
                         else:
                             logger.error(
-                                f"Failed to download audio from {audio_url}, status: {resp.status}"
+                                f"Failed to download audio from {audio_url}, status: {resp.status}",
                             )
                             return ""
+            elif os.path.exists(audio_url):
+                with open(audio_url, "rb") as f:
+                    audio_bytes = f.read()
             else:
-                if os.path.exists(audio_url):
-                    with open(audio_url, "rb") as f:
-                        audio_bytes = f.read()
-                else:
-                    logger.error(f"File not found: {audio_url}")
-                    return ""
+                logger.error(f"File not found: {audio_url}")
+                return ""
 
             if not audio_bytes:
                 logger.error("Audio bytes are empty.")
@@ -145,23 +150,28 @@ class ProviderXinferenceSTT(STTProvider):
             data = aiohttp.FormData()
             data.add_field("model", self.model_uid)
             data.add_field(
-                "file", audio_bytes, filename="audio.wav", content_type="audio/wav"
+                "file",
+                audio_bytes,
+                filename="audio.wav",
+                content_type="audio/wav",
             )
 
             async with self.client.session.post(
-                url, data=data, headers=headers, timeout=self.timeout
+                url,
+                data=data,
+                headers=headers,
+                timeout=self.timeout,
             ) as resp:
                 if resp.status == 200:
                     result = await resp.json()
                     text = result.get("text", "")
                     logger.debug(f"Xinference STT result: {text}")
                     return text
-                else:
-                    error_text = await resp.text()
-                    logger.error(
-                        f"Xinference STT transcription failed with status {resp.status}: {error_text}"
-                    )
-                    return ""
+                error_text = await resp.text()
+                logger.error(
+                    f"Xinference STT transcription failed with status {resp.status}: {error_text}",
+                )
+                return ""
 
         except Exception as e:
             logger.error(f"Xinference STT failed: {e}")

astrbot/core/star/__init__.py

@@ -1,10 +1,11 @@
-from .star import StarMetadata, star_map, star_registry
-from .star_manager import PluginManager
-from .context import Context
-from astrbot.core.provider import Provider
-from astrbot.core.utils.command_parser import CommandParserMixin
 from astrbot.core import html_renderer
+from astrbot.core.provider import Provider
 from astrbot.core.star.star_tools import StarTools
+from astrbot.core.utils.command_parser import CommandParserMixin
+
+from .context import Context
+from .star import StarMetadata, star_map, star_registry
+from .star_manager import PluginManager
 
 
 class Star(CommandParserMixin):
@@ -36,24 +37,28 @@ class Star(CommandParserMixin):
         )
 
     async def html_render(
-        self, tmpl: str, data: dict, return_url=True, options: dict | None = None
+        self,
+        tmpl: str,
+        data: dict,
+        return_url=True,
+        options: dict | None = None,
     ) -> str:
         """渲染 HTML"""
         return await html_renderer.render_custom_template(
-            tmpl, data, return_url=return_url, options=options
+            tmpl,
+            data,
+            return_url=return_url,
+            options=options,
         )
 
     async def initialize(self):
         """当插件被激活时会调用这个方法"""
-        pass
 
     async def terminate(self):
         """当插件被禁用、重载插件时会调用这个方法"""
-        pass
 
     def __del__(self):
         """[Deprecated] 当插件被禁用、重载插件时会调用这个方法"""
-        pass
 
 
-__all__ = ["Star", "StarMetadata", "PluginManager", "Context", "Provider", "StarTools"]
+__all__ = ["Context", "PluginManager", "Provider", "Star", "StarMetadata", "StarTools"]

astrbot/core/star/config.py

@@ -1,23 +1,20 @@
-"""
-此功能已过时，参考 https://astrbot.app/dev/plugin.html#%E6%B3%A8%E5%86%8C%E6%8F%92%E4%BB%B6%E9%85%8D%E7%BD%AE-beta
-"""
+"""此功能已过时，参考 https://astrbot.app/dev/plugin.html#%E6%B3%A8%E5%86%8C%E6%8F%92%E4%BB%B6%E9%85%8D%E7%BD%AE-beta"""
 
-from typing import Union
-import os
 import json
+import os
+
 from astrbot.core.utils.astrbot_path import get_astrbot_data_path
 
 
-def load_config(namespace: str) -> Union[dict, bool]:
-    """
-    从配置文件中加载配置。
+def load_config(namespace: str) -> dict | bool:
+    """从配置文件中加载配置。
     namespace: str, 配置的唯一识别符，也就是配置文件的名字。
     返回值: 当配置文件存在时，返回 namespace 对应配置文件的内容dict，否则返回 False。
     """
     path = os.path.join(get_astrbot_data_path(), "config", f"{namespace}.json")
     if not os.path.exists(path):
         return False
-    with open(path, "r", encoding="utf-8-sig") as f:
+    with open(path, encoding="utf-8-sig") as f:
         ret = {}
         data = json.load(f)
         for k in data:
@@ -26,8 +23,7 @@ def load_config(namespace: str) -> Union[dict, bool]:
 
 
 def put_config(namespace: str, name: str, key: str, value, description: str):
-    """
-    将配置项写入以namespace为名字的配置文件，如果key不存在于目标配置文件中。当前 value 仅支持 str, int, float, bool, list 类型（暂不支持 dict）。
+    """将配置项写入以namespace为名字的配置文件，如果key不存在于目标配置文件中。当前 value 仅支持 str, int, float, bool, list 类型（暂不支持 dict）。
     namespace: str, 配置的唯一识别符，也就是配置文件的名字。
     name: str, 配置项的显示名字。
     key: str, 配置项的键。
@@ -51,7 +47,7 @@ def put_config(namespace: str, name: str, key: str, value, description: str):
     if not os.path.exists(path):
         with open(path, "w", encoding="utf-8-sig") as f:
             f.write("{}")
-    with open(path, "r", encoding="utf-8-sig") as f:
+    with open(path, encoding="utf-8-sig") as f:
         d = json.load(f)
     assert isinstance(d, dict)
     if key not in d:
@@ -69,8 +65,7 @@ def put_config(namespace: str, name: str, key: str, value, description: str):
 
 
 def update_config(namespace: str, key: str, value):
-    """
-    更新配置文件中的配置项。
+    """更新配置文件中的配置项。
     namespace: str, 配置的唯一识别符，也就是配置文件的名字。
     key: str, 配置项的键。
     value: str, int, float, bool, list, 配置项的值。
@@ -78,7 +73,7 @@ def update_config(namespace: str, key: str, value):
     path = os.path.join(get_astrbot_data_path(), "config", f"{namespace}.json")
     if not os.path.exists(path):
         raise FileNotFoundError(f"配置文件 {namespace}.json 不存在。")
-    with open(path, "r", encoding="utf-8-sig") as f:
+    with open(path, encoding="utf-8-sig") as f:
         d = json.load(f)
     assert isinstance(d, dict)
     if key not in d: