From bb724e8c29b102427c9b995e757abc132925c160 Mon Sep 17 00:00:00 2001 From: xunbu Date: Wed, 4 Jun 2025 10:38:28 +0800 Subject: [PATCH] =?UTF-8?q?=E4=B8=80=E4=BA=9Bagent=E8=AF=B7=E6=B1=82?= =?UTF-8?q?=E9=94=99=E8=AF=AF=E7=9B=B4=E6=8E=A5=E6=8A=9B=E5=87=BA=E5=BC=82?= =?UTF-8?q?=E5=B8=B8=EF=BC=8C=E5=A2=9E=E5=A4=A7=E4=BA=86=E5=9D=97=E5=A4=A7?= =?UTF-8?q?=E5=B0=8F?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- docutranslate/agents/agent.py | 12 +++++------- docutranslate/translater.py | 2 +- 2 files changed, 6 insertions(+), 8 deletions(-) diff --git a/docutranslate/agents/agent.py b/docutranslate/agents/agent.py index 9d36f2a..9bc0ffd 100644 --- a/docutranslate/agents/agent.py +++ b/docutranslate/agents/agent.py @@ -107,12 +107,11 @@ class Agent: result = response.json()["choices"][0]["message"]["content"] return result except httpx.HTTPStatusError as e: - translater_logger.error(f"AI请求错误 (async): {e.response.status_code} - {e.response.text}") + raise Exception(f"AI请求错误 (async): {e.response.status_code} - {e.response.text}") except httpx.RequestError as e: - translater_logger.warning(Exception(f"AI请求连接错误 (async): {repr(e)}")) + translater_logger.warning(f"AI请求连接错误 (async): {repr(e)}") except (KeyError, IndexError) as e: - translater_logger.error(f"AI响应格式错误 (async): {repr(e)}") - return "" + raise Exception(f"AI响应格式错误 (async): {repr(e)}") # 如果没有正常获取结果则重试 if retry and retry_count < MAX_RETRY_COUNT: if total_retry_counter.add(): @@ -173,12 +172,11 @@ class Agent: result = response.json()["choices"][0]["message"]["content"] return result except httpx.HTTPStatusError as e: - translater_logger.error(f"AI请求错误 (sync): {e.response.status_code} - {e.response.text}") + raise Exception(f"AI请求错误 (sync): {e.response.status_code} - {e.response.text}") except httpx.RequestError as e: translater_logger.warning(f"AI请求连接错误 (sync): {repr(e)}\nprompt:{prompt}") except (KeyError, IndexError) as e: - translater_logger.error(f"AI响应格式错误 (sync): {repr(e)}") - return "" + raise Exception(f"AI响应格式错误 (sync): {repr(e)}") # 如果没有正常获取结果则重试 if retry and retry_count < MAX_RETRY_COUNT: if total_retry_counter.add(): diff --git a/docutranslate/translater.py b/docutranslate/translater.py index bcedb49..3836126 100644 --- a/docutranslate/translater.py +++ b/docutranslate/translater.py @@ -20,7 +20,7 @@ if DOCLING_FLAG: class FileTranslater: - def __init__(self, file_path: Path | str | None = None, chunksize: int = 2000, + def __init__(self, file_path: Path | str | None = None, chunksize: int = 3000, base_url="", key=None, model_id="", temperature=0.7, max_concurrent=30, timeout=2000, convert_engin: Literal["docling", "mineru"] = "mineru",