增加了Dify和AngthingLLM接口，构建携带知识库的翻译过程，提高翻译质量

pdf2zh/converter.py

@@ -31,7 +31,7 @@ from pdf2zh.translator import (
     AzureTranslator,
     TencentTranslator,
     DifyTranslator,
-    AngthingLLMTranslator,
+    AnythingLLMTranslator,
 )
 from pymupdf import Font
 
@@ -145,7 +145,7 @@ class TranslateConverter(PDFConverterEx):
         service_name = param[0]
         service_model = param[1] if len(param) > 1 else None
         for translator in [GoogleTranslator, BingTranslator, DeepLTranslator, DeepLXTranslator, OllamaTranslator, AzureOpenAITranslator,
-                           OpenAITranslator, ZhipuTranslator, SiliconTranslator, GeminiTranslator, AzureTranslator, TencentTranslator, DifyTranslator, AngthingLLMTranslator]:
+                           OpenAITranslator, ZhipuTranslator, SiliconTranslator, GeminiTranslator, AzureTranslator, TencentTranslator, DifyTranslator, AnythingLLMTranslator]:
             if service_name == translator.name:
                 self.translator = translator(lang_in, lang_out, service_model)
         if not self.translator:

pdf2zh/gui.py

@@ -18,7 +18,7 @@ from pdf2zh.translator import (
     AzureTranslator,
     TencentTranslator,
     DifyTranslator,
-    AngthingLLMTranslator,
+    AnythingLLMTranslator,
 )
 
 import gradio as gr
@@ -41,7 +41,7 @@ service_map: dict[str, BaseTranslator] = {
     "Azure": AzureTranslator,
     "Tencent": TencentTranslator,
     "Dify": DifyTranslator,
-    "AnythingLLM": AngthingLLMTranslator,
+    "AnythingLLM": AnythingLLMTranslator,
 }
 lang_map = {
     "Chinese": "zh",

pdf2zh/translator.py

@@ -352,17 +352,17 @@ class TencentTranslator(BaseTranslator):
         self.req.SourceText = text
         resp: TextTranslateResponse = self.client.TextTranslate(self.req)
         return resp.TargetText
-class AngthingLLMTranslator(BaseTranslator):
-    name = "angthingllm"
+class AnythingLLMTranslator(BaseTranslator):
+    name = "anythingllm"
     envs = {
-        "AngthingLLM_URL": None,
-        "AngthingLLM_APIKEY": "api_key",
+        "AnythingLLM_URL": None,
+        "AnythingLLM_APIKEY": "api_key",
     }
 
     def __init__(self, lang_out, lang_in, model):
         super().__init__(lang_out, lang_in, model)
-        self.api_url = os.getenv("AngthingLLM_URL", self.envs["AngthingLLM_URL"])
-        self.api_key = os.getenv("AngthingLLM_APIKEY", self.envs["AngthingLLM_APIKEY"])
+        self.api_url = os.getenv("AnythingLLM_URL", self.envs["AnythingLLM_URL"])
+        self.api_key = os.getenv("AnythingLLM_APIKEY", self.envs["AnythingLLM_APIKEY"])
         self.headers = {
             "accept": "application/json",
             "Authorization": f"Bearer {self.api_key}",