Merge pull request #607 from PiDanShouRouZhouXD/main

Update SakuraTranslator
zyddnys · Apr 18, 2024 · af29619 · af29619
2 parents b061cd6 + 6b3df6b
commit af29619
Show file tree

Hide file tree

Showing 4 changed files with 454 additions and 156 deletions.
diff --git a/manga_translator/translators/__init__.py b/manga_translator/translators/__init__.py
@@ -15,7 +15,7 @@
 from .selective import SelectiveOfflineTranslator, prepare as prepare_selective_translator
 from .none import NoneTranslator
 from .original import OriginalTranslator
-from .sakura import Sakura13BTranslator
+from .sakura import SakuraTranslator
 
 OFFLINE_TRANSLATORS = {
     'offline': SelectiveOfflineTranslator,
@@ -41,7 +41,7 @@
     'gpt4': GPT4Translator,
     'none': NoneTranslator,
     'original': OriginalTranslator,
-    'sakura': Sakura13BTranslator,
+    'sakura': SakuraTranslator,
     **OFFLINE_TRANSLATORS,
 }
 translator_cache = {}

diff --git a/manga_translator/translators/keys.py b/manga_translator/translators/keys.py
@@ -15,10 +15,11 @@
 OPENAI_HTTP_PROXY = os.getenv('OPENAI_HTTP_PROXY') # TODO: Replace with --proxy
 
 OPENAI_API_BASE = os.getenv('OPENAI_API_BASE', 'https://api.openai.com/v1') #使用api-for-open-llm例子 http://127.0.0.1:8000/v1
-SAKURA_API_BASE = os.getenv('SAKURA_API_BASE', 'http://127.0.0.1:8080/v1') #SAKURA API地址
 
+# sakura
+SAKURA_API_BASE = os.getenv('SAKURA_API_BASE', 'http://127.0.0.1:8080/v1') #SAKURA API地址
+SAKURA_VERSION = os.getenv('SAKURA_VERSION', '0.9') #SAKURA API版本，可选值：0.9、0.10，选择0.10则会加载术语表。
+SAKURA_DICT_PATH = os.getenv('SAKURA_DICT_PATH', './sakura_dict.txt') #SAKURA 术语表路径
 
-CAIYUN_TOKEN = os.getenv('CAIYUN_TOKEN', '') # 彩云小译API访问令牌
 
-SAKURA_API_KEY = os.getenv('SAKURA_API_KEY', '')
-SAKURA_API_BASE = os.getenv('SAKURA_API_BASE', 'http://127.0.0.1:5000/v1')
+CAIYUN_TOKEN = os.getenv('CAIYUN_TOKEN', '') # 彩云小译API访问令牌