From 2dc64ed0a8d3a56bc27a6f56637a744f806bd49b Mon Sep 17 00:00:00 2001 From: ikun-11451 <334495606@qq.com> Date: Sat, 29 Nov 2025 19:49:26 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E6=9C=AC=E5=9C=B0=E8=AF=AD?= =?UTF-8?q?=E9=9F=B3=E8=AF=86=E5=88=AB=E6=8F=92=E4=BB=B6=E7=9A=84=E4=BE=9D?= =?UTF-8?q?=E8=B5=96=E5=AF=BC=E5=85=A5=E9=97=AE=E9=A2=98=E5=96=B5~?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/plugins/built_in/stt_whisper_plugin/__init__.py | 1 + src/plugins/built_in/stt_whisper_plugin/plugin.py | 4 ++-- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/src/plugins/built_in/stt_whisper_plugin/__init__.py b/src/plugins/built_in/stt_whisper_plugin/__init__.py index 5cde004ae..c6bd46bd7 100644 --- a/src/plugins/built_in/stt_whisper_plugin/__init__.py +++ b/src/plugins/built_in/stt_whisper_plugin/__init__.py @@ -6,4 +6,5 @@ __plugin_meta__ = PluginMetadata( usage="在 bot_config.toml 中将 asr_provider 设置为 'local' 即可启用", version="0.1.0", author="Elysia", + python_dependencies=["openai-whisper"], ) diff --git a/src/plugins/built_in/stt_whisper_plugin/plugin.py b/src/plugins/built_in/stt_whisper_plugin/plugin.py index 78ed6d8ad..0baa325d4 100644 --- a/src/plugins/built_in/stt_whisper_plugin/plugin.py +++ b/src/plugins/built_in/stt_whisper_plugin/plugin.py @@ -1,8 +1,6 @@ import asyncio from typing import ClassVar -import whisper - from src.common.logger import get_logger from src.plugin_system import BasePlugin, ComponentInfo, register_plugin from src.plugin_system.base.base_tool import BaseTool @@ -35,6 +33,8 @@ class LocalASRTool(BaseTool): if _whisper_model is None and not _is_loading: _is_loading = True try: + import whisper + model_size = plugin_config.get("whisper", {}).get("model_size", "tiny") device = plugin_config.get("whisper", {}).get("device", "cpu") logger.info(f"正在预加载 Whisper ASR 模型: {model_size} ({device})")