fix: update error handling in Translator class and remove unused import

TensoRaws · Feb 6, 2025 · e199c12 · e199c12
1 parent e8501dd
commit e199c12
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git 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
@@ -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