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

small fix

Browse Source
This commit is contained in:
xunbu
2025-05-28 11:07:18 +08:00
parent 3c9b94150c
commit affec6c1b5
3 changed files with 14 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
.idea/workspace.xml generated
View File

@@ -6,12 +6,8 @@
<component name="ChangeListManager">
<list default="true" id="6b18b44a-df57-4212-a857-9e291ebe5dd2" name="更改" comment="">
<change beforePath="$PROJECT_DIR$/.idea/workspace.xml" beforeDir="false" afterPath="$PROJECT_DIR$/.idea/workspace.xml" afterDir="false" />
<change beforePath="$PROJECT_DIR$/docutranslate/__init__.py" beforeDir="false" afterPath="$PROJECT_DIR$/docutranslate/__init__.py" afterDir="false" />
<change beforePath="$PROJECT_DIR$/docutranslate/agents/agent.py" beforeDir="false" afterPath="$PROJECT_DIR$/docutranslate/agents/agent.py" afterDir="false" />
<change beforePath="$PROJECT_DIR$/docutranslate/agents/markdown_agent.py" beforeDir="false" afterPath="$PROJECT_DIR$/docutranslate/agents/markdown_agent.py" afterDir="false" />
<change beforePath="$PROJECT_DIR$/docutranslate/app.py" beforeDir="false" afterPath="$PROJECT_DIR$/docutranslate/app.py" afterDir="false" />
<change beforePath="$PROJECT_DIR$/docutranslate/static/index.html" beforeDir="false" afterPath="$PROJECT_DIR$/docutranslate/static/index.html" afterDir="false" />
<change beforePath="$PROJECT_DIR$/docutranslate/translater.py" beforeDir="false" afterPath="$PROJECT_DIR$/docutranslate/translater.py" afterDir="false" />
</list>
<option name="SHOW_DIALOG" value="false" />
<option name="HIGHLIGHT_CONFLICTS" value="true" />
@@ -87,7 +83,7 @@
"RunOnceActivity.ShowReadmeOnStart": "true",
"RunOnceActivity.TerminalTabsStorage.copyFrom.TerminalArrangementManager": "true",
"RunOnceActivity.git.unshallow": "true",
"git-widget-placeholder": "dev",
"git-widget-placeholder": "main",
"last_opened_file_path": "C:/Users/jxgm/Desktop/translate/docutranslate",
"list.type.of.created.stylesheet": "CSS",
"node.js.detected.package.eslint": "true",
@@ -617,7 +613,7 @@
</option>
</component>
<component name="com.intellij.coverage.CoverageDataManagerImpl">
<SUITE FILE_PATH="coverage/filetranslate$app_test__1_.coverage" NAME="app_test (1) 覆盖结果" MODIFIED="1748398439791" SOURCE_PROVIDER="com.intellij.coverage.DefaultCoverageFileProvider" RUNNER="coverage.py" COVERAGE_BY_TEST_ENABLED="false" COVERAGE_TRACING_ENABLED="false" WORKING_DIRECTORY="$PROJECT_DIR$/tests" />
<SUITE FILE_PATH="coverage/filetranslate$app_test__1_.coverage" NAME="app_test (1) 覆盖结果" MODIFIED="1748401114236" SOURCE_PROVIDER="com.intellij.coverage.DefaultCoverageFileProvider" RUNNER="coverage.py" COVERAGE_BY_TEST_ENABLED="false" COVERAGE_TRACING_ENABLED="false" WORKING_DIRECTORY="$PROJECT_DIR$/tests" />
<SUITE FILE_PATH="coverage/filetranslate$test.coverage" NAME="test 覆盖结果" MODIFIED="1747472297913" SOURCE_PROVIDER="com.intellij.coverage.DefaultCoverageFileProvider" RUNNER="coverage.py" COVERAGE_BY_TEST_ENABLED="false" COVERAGE_TRACING_ENABLED="false" WORKING_DIRECTORY="$PROJECT_DIR$/tests" />
<SUITE FILE_PATH="coverage/filetranslate$convert.coverage" NAME="convert 覆盖结果" MODIFIED="1746963490689" SOURCE_PROVIDER="com.intellij.coverage.DefaultCoverageFileProvider" RUNNER="coverage.py" COVERAGE_BY_TEST_ENABLED="false" COVERAGE_TRACING_ENABLED="false" WORKING_DIRECTORY="$PROJECT_DIR$/docutranslate/utils" />
<SUITE FILE_PATH="coverage/PDFtranslate$PDFtranslater__1_.coverage" NAME="PDFtranslater (1) 覆盖结果" MODIFIED="1746633258205" SOURCE_PROVIDER="com.intellij.coverage.DefaultCoverageFileProvider" RUNNER="coverage.py" COVERAGE_BY_TEST_ENABLED="false" COVERAGE_TRACING_ENABLED="false" WORKING_DIRECTORY="$PROJECT_DIR$/pdftranslate_packages" />

2
docutranslate/agents/agent.py
View File

@@ -96,7 +96,7 @@ class Agent:
system_prompt: str | None = None,
max_concurrent: int | None = None # 新增参数，默认并发数为5
) -> list[str]:
print(f"system_prompt:{self.system_prompt}")
# print(f"system_prompt:{self.system_prompt}")
max_concurrent = self.max_concurrent if max_concurrent is None else max_concurrent
total = len(prompts)
count = 0

22
docutranslate/agents/markdown_agent.py
View File

@@ -17,7 +17,7 @@ class MDRefineAgent(Agent):
如果修正不必要，则返回原文。
不要解释，不要注释。
不要修改标题的级别（如一级标题不要修改为二级标题）
形如<ph-abc123>的占位符不要改变
形如<ph-ads231>的占位符不要改变【非常重要】
code、latex和HTML保持结构
所有公式（包括短公式）都应该是latex公式
修复不正确的latex公式，行内公式要用$正确包裹以构造合法latex表达式
@@ -38,8 +38,8 @@ code、latex和HTML保持结构
一道题目<ph-12asd2>:$c_0+1=2$，$c_0$等于几
{$c_0$,$c_1$,$c^2$}是一个集合"""
if custom_prompt:
self.system_prompt += "\n#其余要求（可能为背景或指令）\n" + custom_prompt
self.system_prompt+=r'\no_think'
self.system_prompt += "\n#补充重要规则或背景如下(应遵守):\n" + custom_prompt + '\n'
self.system_prompt += r'\no_think'
class MDTranslateAgent(Agent):
@@ -55,20 +55,20 @@ class MDTranslateAgent(Agent):
翻译要有专业性和高质量
没有任何解释和注释。
引用的参考文献名和其作者名保持原文不翻译
形如<ph-abc123>的占位符不要改变
形如<ph-ads231>的占位符不要改变【非常重要】
code、latex和HTML只翻译说明文字，其余保持原文
公式必须表示为合法的latex公式,行内公式需被$正确包裹
删除明显异常的无意义字符
删除明显异常的字符
# 输出
翻译后的markdown译文纯文本（不是markdown代码块，无任何多余文字）
# 示例
## 英文翻译为中文
输入：
hello<ph-aaaaaa>, what's your nam*@e?
![photo title](<ph-abcde>)
hello<ph-12asd2>, what's your nam*@e?
![photo title](<ph-abcdde>)
输出：
你好<ph-aaaaaa>，你叫什么名字？
![图像标题](<ph-abcde>)
你好<ph-12asd2>，你叫什么名字？
![图像标题](<ph-abcdde>)
## 公式要为合法latex（行内公式应正确包裹）
输入：
The equation is E=mc 2. This is famous.
@@ -84,5 +84,5 @@ volume 99, pages 173–186, 1999.
[2] M. Castro, B. Liskov, et al. Practical byzantine fault tolerance. In OSDI,
volume 99, pages 173–186, 1999."""
if custom_prompt:
self.system_prompt += "\n#其余要求（可能为背景或指令）\n" + custom_prompt
self.system_prompt+=r'\no_think'
self.system_prompt += "\n#补充重要规则或背景如下(应遵守):\n" + custom_prompt + '\n'
self.system_prompt += r'\no_think'