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

优化提示词

Browse Source
This commit is contained in:
xunbu
2025-05-12 20:20:57 +08:00
parent 257f6a0679
commit a9df39eb08
4 changed files with 29 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
.idea/workspace.xml generated
View File

@@ -5,11 +5,10 @@
</component> </component>
<component name="ChangeListManager"> <component name="ChangeListManager">
<list default="true" id="6b18b44a-df57-4212-a857-9e291ebe5dd2" name="更改" comment=""> <list default="true" id="6b18b44a-df57-4212-a857-9e291ebe5dd2" name="更改" comment="">
<change afterPath="$PROJECT_DIR$/docutranslate/DocuTranslate.png" afterDir="false" />
<change beforePath="$PROJECT_DIR$/.idea/workspace.xml" beforeDir="false" afterPath="$PROJECT_DIR$/.idea/workspace.xml" afterDir="false" /> <change beforePath="$PROJECT_DIR$/.idea/workspace.xml" beforeDir="false" afterPath="$PROJECT_DIR$/.idea/workspace.xml" afterDir="false" />
<change beforePath="$PROJECT_DIR$/docutranslate/translater.py" beforeDir="false" afterPath="$PROJECT_DIR$/docutranslate/translater.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$/pyproject.toml" beforeDir="false" afterPath="$PROJECT_DIR$/pyproject.toml" afterDir="false" /> <change beforePath="$PROJECT_DIR$/pyproject.toml" beforeDir="false" afterPath="$PROJECT_DIR$/pyproject.toml" afterDir="false" />
<change beforePath="$PROJECT_DIR$/uv.lock" beforeDir="false" afterPath="$PROJECT_DIR$/uv.lock" afterDir="false" />
</list> </list>
<option name="SHOW_DIALOG" value="false" /> <option name="SHOW_DIALOG" value="false" />
<option name="HIGHLIGHT_CONFLICTS" value="true" /> <option name="HIGHLIGHT_CONFLICTS" value="true" />
@@ -81,10 +80,10 @@
<component name="RecentsManager"> <component name="RecentsManager">
<key name="CopyFile.RECENT_KEYS"> <key name="CopyFile.RECENT_KEYS">
<recent name="C:\Users\jxgm\Desktop\FileTranslate\tests\files" /> <recent name="C:\Users\jxgm\Desktop\FileTranslate\tests\files" />
<recent name="C:\Users\jxgm\Desktop\FileTranslate\dist\app" />
<recent name="C:\Users\jxgm\Desktop\FileTranslate\tests" /> <recent name="C:\Users\jxgm\Desktop\FileTranslate\tests" />
<recent name="C:\Users\jxgm\Desktop\FileTranslate\docutranslate" /> <recent name="C:\Users\jxgm\Desktop\FileTranslate\docutranslate" />
<recent name="C:\Users\jxgm\Desktop\FileTranslate\tests\备份" /> <recent name="C:\Users\jxgm\Desktop\FileTranslate\tests\备份" />
<recent name="C:\Users\jxgm\Desktop\FileTranslate\dist\app\_internal" />
</key> </key>
<key name="MoveFile.RECENT_KEYS"> <key name="MoveFile.RECENT_KEYS">
<recent name="C:\Users\jxgm\Desktop\FileTranslate\dist\app" /> <recent name="C:\Users\jxgm\Desktop\FileTranslate\dist\app" />
@@ -636,7 +635,8 @@
<workItem from="1746963064346" duration="9090000" /> <workItem from="1746963064346" duration="9090000" />
<workItem from="1747008194598" duration="13116000" /> <workItem from="1747008194598" duration="13116000" />
<workItem from="1747028178509" duration="751000" /> <workItem from="1747028178509" duration="751000" />
<workItem from="1747046024264" duration="2159000" /> <workItem from="1747046024264" duration="2911000" />
<workItem from="1747050264594" duration="2135000" />
</task> </task>
<servers /> <servers />
</component> </component>
@@ -645,7 +645,7 @@
</component> </component>
<component name="com.intellij.coverage.CoverageDataManagerImpl"> <component name="com.intellij.coverage.CoverageDataManagerImpl">
<SUITE FILE_PATH="coverage/filetranslate$agent_utils.coverage" NAME="agent_utils 覆盖结果" MODIFIED="1746708534311" 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/filetranslate$agent_utils.coverage" NAME="agent_utils 覆盖结果" MODIFIED="1746708534311" 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/filetranslate$test.coverage" NAME="test 覆盖结果" MODIFIED="1747048348489" 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="1747052174571" 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/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/filetranslate$test1.coverage" NAME="test1 覆盖结果" MODIFIED="1746936018440" 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$test1.coverage" NAME="test1 覆盖结果" MODIFIED="1746936018440" 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/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" /> <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" />

8
docutranslate/agents/agent.py
View File

@@ -22,10 +22,10 @@ TIMEOUT = 500
class Agent: class Agent:
def __init__(self, baseurl="", key="", model_id="", system_prompt="", temperature=0.7, max_concurrent=6,timeout:int=TIMEOUT): def __init__(self, baseurl:str="", key:str="xx", model_id:str="", system_prompt:str="", temperature=0.7, max_concurrent=6,timeout:int=TIMEOUT):
self.baseurl = baseurl self.baseurl = baseurl.strip()
self.key = key self.key = key.strip()
self.model_id = model_id self.model_id = model_id.strip()
self.system_prompt = system_prompt self.system_prompt = system_prompt
self.temperature = temperature self.temperature = temperature
self.client_async = httpx.AsyncClient() self.client_async = httpx.AsyncClient()

27
docutranslate/agents/markdown_agent.py
View File

@@ -8,17 +8,18 @@ class MDRefineAgent(Agent):
super().__init__(**kwargs) super().__init__(**kwargs)
self.system_prompt=r""" self.system_prompt=r"""
# 角色 # 角色
你是一个修正markdown文本的专家。 你是一个修正markdown文本的专家
# 工作 # 工作
找到markdown片段的不合理之处。 找到markdown片段的不合理之处
对于缺失、中断的句子，应该查看缺失的语句是否可能被错误的放在了其他位置，并通过句子拼接修复不合理之处。 对于缺失、中断的句子，应该查看缺失的语句是否可能被错误的放在了其他位置，并通过句子拼接修复不合理之处
去掉异常字词，修复错误格式。 去掉异常字词，修复错误格式
# 要求 # 要求
If refine is unnecessary, return the original text. If refine is unnecessary, return the original text.
NO explanations. NO notes. NO explanations. NO notes.
不要修改标题的级别（如一级标题不要修改为二级标题） 不要修改标题的级别（如一级标题不要修改为二级标题）
形如<ph-abc123>的占位符不要改变。 形如<ph-abc123>的占位符不要改变
code、latex和HTML保持结构。 code、latex和HTML保持结构
对于修复后的latex文本，用$包裹以构建合法表达式
# 输出 # 输出
修正后的markdown纯文本（不是markdown代码块） 修正后的markdown纯文本（不是markdown代码块）
# 示例 # 示例
@@ -30,9 +31,9 @@ code、latex和HTML保持结构。
你叫什么名字 你叫什么名字
## 去掉异常字词(保持占位符和latex符号) ## 去掉异常字词(保持占位符和latex符号)
输入： 输入：
一道\题@#目<ph-12asd2>:\(x_1+1=2\) 一道\题@#目<ph-12asd2>:c_0+1=2，\(c 0\)等于几
输出: 输出:
一道题目<ph-12asd2>:：\(x_1+1=2\) 一道题目<ph-12asd2>:：$c_0+1=2$，\(c_0\)等于几
\no_think""" \no_think"""
@@ -49,7 +50,8 @@ You are a professional, authentic machine translation engine.
If translation is unnecessary (e.g. proper nouns, codes, etc.), return the original text. If translation is unnecessary (e.g. proper nouns, codes, etc.), return the original text.
NO explanations. NO notes. NO explanations. NO notes.
不要修改标题的级别（如一级标题不要修改为二级标题） 不要修改标题的级别（如一级标题不要修改为二级标题）
形如<ph-abc123>的占位符不要改变。 文献名保持原文
形如<ph-abc123>的占位符不要改变
code、latex和HTML只翻译说明文字，其余保持原文 code、latex和HTML只翻译说明文字，其余保持原文
# 输出 # 输出
翻译后的markdown纯文本（不是markdown代码块） 翻译后的markdown纯文本（不是markdown代码块）
@@ -59,4 +61,11 @@ code、latex和HTML只翻译说明文字，其余保持原文
hello<ph-aaaaaa>, what's your name? hello<ph-aaaaaa>, what's your name?
输出： 输出：
你好<ph-aaaaaa>，你叫什么名字？ 你好<ph-aaaaaa>，你叫什么名字？
## 文献名不翻译
输入：
[2] M. Castro, B. Liskov, et al. Practical byzantine fault tolerance. In OSDI,
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.
\\no_think""" \\no_think"""

2
pyproject.toml
View File

@@ -1,6 +1,6 @@
[project] [project]
name = "docutranslate" name = "docutranslate"
version = "0.2.0" version = "0.2.2"
description = "文件翻译工具" description = "文件翻译工具"
readme = "README.md" readme = "README.md"
requires-python = ">=3.10" requires-python = ">=3.10"