36 lines
1.2 KiB
Python
36 lines
1.2 KiB
Python
import base64
|
|
|
|
from src.config.config import global_config
|
|
from src.llm_models.utils_model import LLMRequest
|
|
|
|
from src.common.logger import get_logger
|
|
from rich.traceback import install
|
|
install(extra_lines=3)
|
|
|
|
logger = get_logger("chat_voice")
|
|
|
|
async def get_voice_text(voice_base64: str) -> str:
|
|
"""获取音频文件描述"""
|
|
if not global_config.voice.enable_asr:
|
|
logger.warning("语音识别未启用，无法处理语音消息")
|
|
return "[语音]"
|
|
try:
|
|
# 解码base64音频数据
|
|
# 确保base64字符串只包含ASCII字符
|
|
if isinstance(voice_base64, str):
|
|
voice_base64 = voice_base64.encode("ascii", errors="ignore").decode("ascii")
|
|
voice_bytes = base64.b64decode(voice_base64)
|
|
_llm = LLMRequest(model=global_config.model.voice, request_type="voice")
|
|
text = await _llm.generate_response_for_voice(voice_bytes)
|
|
if text is None:
|
|
logger.warning("未能生成语音文本")
|
|
return "[语音(文本生成失败)]"
|
|
|
|
logger.debug(f"描述是{text}")
|
|
|
|
return f"[语音：{text}]"
|
|
except Exception as e:
|
|
logger.error(f"语音转文字失败: {str(e)}")
|
|
return "[语音]"
|
|
|