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})")