Leon/docutranslate
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

translator增加glossary_dict的配置

Browse Source
This commit is contained in:
xunbu
2025-08-27 21:11:02 +08:00
parent 78a4525108
commit 1ffed8f280
9 changed files with 22 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
docutranslate/translator/ai_translator/md_translator.py
View File

@@ -29,7 +29,8 @@ class MDTranslator(AiTranslator):
thinking=config.thinking,
max_concurrent=config.concurrent,
timeout=config.timeout,
logger=self.logger)
logger=self.logger,
glossary_dict=config.glossary_dict)
self.translate_agent = MDTranslateAgent(agent_config)
def translate(self, document: MarkdownDocument) -> Self:
@@ -37,7 +38,7 @@ class MDTranslator(AiTranslator):
with MDMaskUrisContext(document):
chunks: list[str] = split_markdown_text(document.content.decode(), self.chunk_size)
self.logger.info(f"markdown分为{len(chunks)}块")
result: list[str] = self.translate_agent.send_prompts(chunks)
result: list[str] = self.translate_agent.send_chunks(chunks)
content = join_markdown_texts(result)
# 做一些加强鲁棒性的操作
content = content.replace(r'\（', r'\(')
@@ -52,7 +53,7 @@ class MDTranslator(AiTranslator):
with MDMaskUrisContext(document):
chunks: list[str] = split_markdown_text(document.content.decode(), self.chunk_size)
self.logger.info(f"markdown分为{len(chunks)}块")
result: list[str] = await self.translate_agent.send_prompts_async(chunks)
result: list[str] = await self.translate_agent.send_chunks_async(chunks)
def run():
content = join_markdown_texts(result)