Commit e034f4f
Changed files (1)
src
subtitles
src/subtitles/base.py
@@ -50,7 +50,7 @@ async def match_url(client: Client, message: Message) -> str:
@cache.memoize(ttl=120)
-async def fetch_subtitle(url: str, reference: str = "") -> dict:
+async def fetch_subtitle(url: str, reference: str = "", *, enable_corrector: bool = True) -> dict:
"""Fetch subtitles from Bilibili or YouTube.
Returns:
@@ -64,7 +64,7 @@ async def fetch_subtitle(url: str, reference: str = "") -> dict:
matched = await match_social_media_link(url)
if matched["platform"] == "bilibili":
resp = await bilibili_subtitle_and_summary(url)
- if resp.get("subtitles"):
+ if resp.get("subtitles") and enable_corrector:
resp["subtitles"] = await asr_corrector(resp["subtitles"], reference)
return resp
@@ -82,7 +82,7 @@ async def fetch_subtitle(url: str, reference: str = "") -> dict:
if not subtitles:
return {"error": "❌下载内嵌字幕失败\n🔄尝试使用语音转文字获取字幕"}
resp = to_transcription(subtitles)
- if resp.get("subtitles"):
+ if resp.get("subtitles") and enable_corrector:
resp["subtitles"] = await asr_corrector(resp["subtitles"], reference)
return resp