From 95ed4cc3611d2f63d3a47e34204d92bb7f766a49 Mon Sep 17 00:00:00 2001
From: xunbu <xunbu3@qq.com>
Date: Thu, 4 Sep 2025 17:11:55 +0800
Subject: [PATCH] =?UTF-8?q?=E7=BB=9F=E8=AE=A1=E4=B8=BA=E8=A7=A3=E5=86=B3?=
 =?UTF-8?q?=E7=9A=84=E9=94=99=E8=AF=AF=E6=95=B0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 docutranslate/agents/agent.py | 27 ++++++++++++++++++++++++++-
 1 file changed, 26 insertions(+), 1 deletion(-)

diff --git a/docutranslate/agents/agent.py b/docutranslate/agents/agent.py
index 0e016c0..5145c44 100644
--- a/docutranslate/agents/agent.py
+++ b/docutranslate/agents/agent.py
@@ -123,6 +123,9 @@ class Agent:
         self.thinking = config.thinking
         self.logger = config.logger or global_logger
         self.total_error_counter = TotalErrorCounter(logger=self.logger)
+        # 新增：用于统计最终未解决的错误
+        self.unresolved_error_lock = Lock()
+        self.unresolved_error_count = 0
 
     def _add_thinking_mode(self, data: dict):
         if self.domain not in self._think_factory:
@@ -234,6 +237,9 @@ class Agent:
         else:
             if should_retry:
                 self.logger.error(f"所有重试均失败，已达到重试次数上限。")
+                # 新增：当所有重试失败后，增加未解决错误计数
+                with self.unresolved_error_lock:
+                    self.unresolved_error_count += 1
 
             if best_partial_result:
                 self.logger.info("所有重试失败，但存在部分翻译结果，将使用该结果。")
@@ -256,6 +262,10 @@ class Agent:
             f"base-url:{self.baseurl},model-id:{self.model_id},concurrent:{max_concurrent},temperature:{self.temperature}")
         self.logger.info(f"预计发送{total}个请求，并发请求数:{max_concurrent}")
         self.total_error_counter.max_errors_count = len(prompts) // MAX_REQUESTS_PER_ERROR
+
+        # 新增：在每次批量发送前重置计数器
+        self.unresolved_error_count = 0
+
         count = 0
         semaphore = asyncio.Semaphore(max_concurrent)
         tasks = []
@@ -283,6 +293,10 @@ class Agent:
                 tasks.append(task)
 
             results = await asyncio.gather(*tasks, return_exceptions=False)
+
+            # 新增：在所有任务完成后打印未解决的错误总数
+            self.logger.info(f"所有请求处理完毕。未解决的错误总数: {self.unresolved_error_count}")
+
             return results
 
     def send(self, client: httpx.Client, prompt: str, system_prompt: None | str = None, retry=True, retry_count=0,
@@ -362,6 +376,9 @@ class Agent:
         else:
             if should_retry:
                 self.logger.error(f"所有重试均失败，已达到重试次数上限。")
+                # 新增：当所有重试失败后，增加未解决错误计数
+                with self.unresolved_error_lock:
+                    self.unresolved_error_count += 1
 
             if best_partial_result:
                 self.logger.info("所有重试失败，但存在部分翻译结果，将使用该结果。")
@@ -391,6 +408,10 @@ class Agent:
             f"base-url:{self.baseurl},model-id:{self.model_id},concurrent:{self.max_concurrent},temperature:{self.temperature}")
         self.logger.info(f"预计发送{len(prompts)}个请求，并发请求数:{self.max_concurrent}")
         self.total_error_counter.max_errors_count = len(prompts) // MAX_REQUESTS_PER_ERROR
+
+        # 新增：在每次批量发送前重置计数器
+        self.unresolved_error_count = 0
+
         counter = PromptsCounter(len(prompts), self.logger)
 
         system_prompts = itertools.repeat(system_prompt, len(prompts))
@@ -408,8 +429,12 @@ class Agent:
                                                 result_handlers,
                                                 error_result_handlers)
                 output_list = list(results_iterator)
+
+        # 新增：在所有任务完成后打印未解决的错误总数
+        self.logger.info(f"所有请求处理完毕。未解决的错误总数: {self.unresolved_error_count}")
+
         return output_list
 
 
 if __name__ == '__main__':
-    pass
+    pass
\ No newline at end of file