Merge pull request #221 from PDFMathTranslate/dev

feat(qwenmt): add model name check

Author: awwaawwa

pdf2zh_next/__init__.py

@@ -30,7 +30,7 @@
 
 # from pdf2zh_next.high_level import translate, translate_stream
 
-__version__ = "2.6.3"
+__version__ = "2.6.4"
 __author__ = "Byaidu, awwaawwa"
 __license__ = "AGPL-3.0"
 __maintainer__ = "awwaawwa"

pdf2zh_next/const.py

@@ -1,4 +1,4 @@
-__version__ = "2.6.3"
+__version__ = "2.6.4"
 __major_version__ = "2"
 __config_file_version__ = "3"
 

pdf2zh_next/main.py

@@ -14,7 +14,7 @@
 from pdf2zh_next.config import ConfigManager
 from pdf2zh_next.high_level import do_translate_file_async
 
-__version__ = "2.6.3"
+__version__ = "2.6.4"
 
 logger = logging.getLogger(__name__)
 

pdf2zh_next/translator/translator_impl/qwenmt.py

@@ -37,9 +37,14 @@ def __init__(
         self.prompt_token_count = AtomicInteger()
         self.completion_token_count = AtomicInteger()
 
+        if "qwen-mt" not in self.model:
+            raise ValueError(
+                f"Model {self.model} is not a Qwen-MT model, Other Qwen models should use AliyunDashScope or OpenAICompatible."
+            )
+
     def lang_mapping(self, input_lang: str) -> str:
         """
-        Mapping the language code to the language code that Aliyun Qwen-Mt model supports.
+        Mapping the language code to the language code that Aliyun Qwen-MT model supports.
         Since all existings languagues codes used in gui.py are able to be mapped, the original
         languague code will not be checked.
         """

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "pdf2zh-next"
-version = "2.6.3"
+version = "2.6.4"
 description = "Latex PDF Translator"
 authors = [
     { name = "awwaawwa", email = "[email protected]" },
@@ -83,7 +83,7 @@ max-line-length = 88
 
 
 [bumpver]
-current_version = "2.6.3"
+current_version = "2.6.4"
 version_pattern = "MAJOR.MINOR.PATCH[.PYTAGNUM]"
 
 [bumpver.file_patterns]