Commit e034f4f

benny-dou <60535774+benny-dou@users.noreply.github.com>
2026-05-17 05:08:48
chore(subtitles): add option `enable_corrector` to `fetch_subtitle` function
1 parent 70116b8
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