From e199c12ef6ba98780c2829f144f714e1b9a78555 Mon Sep 17 00:00:00 2001 From: NULL204 Date: Thu, 6 Feb 2025 16:31:13 +0800 Subject: [PATCH] fix: update error handling in Translator class and remove unused import --- yuisub/__init__.py | 2 +- yuisub/llm.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/yuisub/__init__.py b/yuisub/__init__.py index b9597db..864e007 100644 --- a/yuisub/__init__.py +++ b/yuisub/__init__.py @@ -1,5 +1,5 @@ from yuisub.bangumi import BGM, bangumi # noqa: F401 from yuisub.llm import Summarizer, Translator # noqa: F401 -from yuisub.prompt import ORIGIN, ZH # noqa: F401 +from yuisub.prompt import ZH # noqa: F401 from yuisub.sub import advertisement, bilingual, load, translate # noqa: F401 from yuisub.translator import SubtitleTranslator # noqa: F401 diff --git a/yuisub/llm.py b/yuisub/llm.py index 0b59893..bac1d4d 100644 --- a/yuisub/llm.py +++ b/yuisub/llm.py @@ -54,8 +54,8 @@ async def ask(self, question: str) -> ZH: raise e except Exception as e: - print(f"Unknown Error: {e} return original question: {question.origin}") - return ZH(zh=question.origin) + print(f"Unknown Error: {e} return original question: {question}") + return ZH(zh=question) return zh