代码格式化

.idea/workspace.xml

@@ -5,13 +5,8 @@
   </component>
   <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/Agents/__init__.py" beforeDir="false" afterPath="$PROJECT_DIR$/docutranslate/Agents/__init__.py" afterDir="false" />
-      <change beforePath="$PROJECT_DIR$/README.md" beforeDir="false" afterPath="$PROJECT_DIR$/README.md" 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/translater.py" beforeDir="false" afterPath="$PROJECT_DIR$/docutranslate/translater.py" afterDir="false" />
-      <change beforePath="$PROJECT_DIR$/docutranslate/utils/markdown_splitter.py" beforeDir="false" afterPath="$PROJECT_DIR$/docutranslate/utils/markdown_splitter.py" 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>
     <option name="SHOW_DIALOG" value="false" />
     <option name="HIGHLIGHT_CONFLICTS" value="true" />
@@ -402,7 +397,7 @@
       <workItem from="1746788668813" duration="25000" />
       <workItem from="1746791230782" duration="3129000" />
       <workItem from="1746799824552" duration="317000" />
-      <workItem from="1746801217905" duration="14612000" />
+      <workItem from="1746801217905" duration="14698000" />
     </task>
     <servers />
   </component>

docutranslate/Agents/__init__.py

@@ -1,2 +1,2 @@
-from .agent import Agent,AgentArgs
+from .agent import Agent, AgentArgs
-from .markdown_agent import MDRefineAgent,MDTranslateAgent
+from .markdown_agent import MDRefineAgent, MDTranslateAgent

docutranslate/translater.py

@@ -2,15 +2,17 @@ from pathlib import Path
 from typing import Literal
 
 import markdown2
-# import mdformat
 
+from docutranslate.Agents import Agent, AgentArgs
 from docutranslate.Agents import MDRefineAgent, MDTranslateAgent
-from docutranslate.Agents.agent import Agent, AgentArgs
 from docutranslate.utils.convert import file2markdown_embed_images
 from docutranslate.utils.markdown_splitter import split_markdown_text
 from docutranslate.utils.markdown_utils import uris2placeholder, placeholder2_uris, MaskDict
 
 
+# import mdformat
+
+
 class FileTranslater:
     def __init__(self, file_path: Path | str | None = None, chunksize: int = 4096, base_url="", key=None,
                  model_id="", temperature=0.7, max_concurrent=20, docling_artifact: Path | str | None = None,
@@ -69,7 +71,8 @@ class FileTranslater:
         return result
 
     def read_file(self, file_path: Path | str | None = None, formula=False, code=False, save=False,
-                  save_format: Literal["markdown", "html"] = "markdown", refine=False, refine_agent:Agent|None=None):
+                  save_format: Literal["markdown", "html"] = "markdown", refine=False,
+                  refine_agent: Agent | None = None):
         if file_path is None:
             if self.file_path is None:
                 raise Exception("未设置文件路径")
@@ -104,7 +107,7 @@ class FileTranslater:
         print("markdown已修正")
         return self.markdown
 
-    def translate_markdown_by_agent(self, translate_agent: Agent|None=None):
+    def translate_markdown_by_agent(self, translate_agent: Agent | None = None):
         print("正在翻译markdown")
 
         chuncks = self._split_markdown_into_chunks()

docutranslate/utils/markdown_utils.py

@@ -3,7 +3,6 @@ import threading
 import uuid
 
 
-
 class MaskDict:
     def __init__(self):
         self._dict = {}
@@ -33,8 +32,9 @@ class MaskDict:
         with self._lock:
             return item in self._dict
 
+
 # def uris2placeholder(markdown:str, mask_dict:MaskDict):
-        ##替换整个uri
+##替换整个uri
 #     def uri2placeholder(match: re.Match):
 #         id = mask_dict.create_id()
 #         mask_dict.set(id, match.group())
@@ -44,7 +44,7 @@ class MaskDict:
 #     markdown = re.sub(uri_pattern, uri2placeholder, markdown)
 #     return markdown
 
-def uris2placeholder(markdown:str, mask_dict:MaskDict):
+def uris2placeholder(markdown: str, mask_dict: MaskDict):
     ##只替换uri里的链接部分，保留标题
     def uri2placeholder(match: re.Match):
         id = mask_dict.create_id()
@@ -54,10 +54,12 @@ def uris2placeholder(markdown:str, mask_dict:MaskDict):
     uri_pattern = r'(!?\[.*?\])\((.*?)\)'
     markdown = re.sub(uri_pattern, uri2placeholder, markdown)
     return markdown
-def placeholder2_uris(markdown:str, mask_dict:MaskDict):
+
-    def placeholder2uri(match:re.Match):
+
-        id=match.group(1)
+def placeholder2_uris(markdown: str, mask_dict: MaskDict):
-        uri=mask_dict.get(id)
+    def placeholder2uri(match: re.Match):
+        id = match.group(1)
+        uri = mask_dict.get(id)
         if uri is None:
             return match.group()
         return uri
@@ -67,6 +69,5 @@ def placeholder2_uris(markdown:str, mask_dict:MaskDict):
     return markdown
 
 
-
 if __name__ == '__main__':
     pass