From 8745329d2c3d1dba7cf75fa5648931261f8b110a Mon Sep 17 00:00:00 2001 From: xunbu Date: Tue, 15 Jul 2025 13:04:50 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E4=BA=86cache=E7=9A=84bug?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- docutranslate/__init__.py | 2 +- docutranslate/app.py | 8 ++++++-- docutranslate/cacher/document_cacher.py | 4 ++-- 3 files changed, 9 insertions(+), 5 deletions(-) diff --git a/docutranslate/__init__.py b/docutranslate/__init__.py index 201ca17..17be052 100644 --- a/docutranslate/__init__.py +++ b/docutranslate/__init__.py @@ -1,4 +1,4 @@ -__version__="0.3.1b1" +__version__="0.3.1" diff --git a/docutranslate/app.py b/docutranslate/app.py index 902de6a..7a81fd3 100644 --- a/docutranslate/app.py +++ b/docutranslate/app.py @@ -969,8 +969,10 @@ async def temp_translate( examples=["test.txt", "test.md", "test.pdf"]), file_content: str = Body(..., description="文件内容，可以是纯文本或Base64编码的字符串。"), to_lang: str = Body("中文", description="目标语言。", examples=["中文", "英文", "English"]), - concurrent: int = Body(30, description="ai翻译请求并发数"), - custom_prompt_translate: str | None = Body(None, description="翻译自定义提示词") + concurrent: int = Body(default_params["concurrent"], description="ai翻译请求并发数"), + temperature:float|None = Body(default_params["temperature"], description="ai翻译请求温度"), + chunk_size:int =Body(default_params["chunk_size"],description="文本分块大小（bytes）"), + custom_prompt_translate: str | None = Body(None, description="翻译自定义提示词",example="人名保持原文不翻译"), ): """一个用于快速测试的同步翻译接口。""" @@ -986,6 +988,8 @@ async def temp_translate( model_id=model_id, mineru_token=mineru_token, concurrent=concurrent, + temperature=temperature, + chunk_size=chunk_size, ) try: diff --git a/docutranslate/cacher/document_cacher.py b/docutranslate/cacher/document_cacher.py index 5e934c8..b89ab05 100644 --- a/docutranslate/cacher/document_cacher.py +++ b/docutranslate/cacher/document_cacher.py @@ -3,7 +3,7 @@ from collections import OrderedDict from docutranslate.converter import Document -CACHE_NUM=os.getenv("DOCUTRANSLATE_CACHE_NUM",default=10) +CACHE_NUM=os.getenv("DOCUTRANSLATE_CACHE_NUM",default="10") class DocumentCacher: def __init__(self): @@ -18,7 +18,7 @@ class DocumentCacher: def cache_result(self, result: str, document: Document, formula: bool, code: bool, convert_engin: str): hash_code = self._get_hashcode(document, formula, code, convert_engin) - if len(self.cache_dict)>=CACHE_NUM: + if len(self.cache_dict)>=int(CACHE_NUM): self.cache_dict.popitem(last=False) self.cache_dict[hash_code] = result return result