修复了cache的bug

This commit is contained in:
xunbu
2025-07-15 13:04:50 +08:00
parent c08bb84e3a
commit 8745329d2c
3 changed files with 9 additions and 5 deletions

View File

@@ -1,4 +1,4 @@
__version__="0.3.1b1"
__version__="0.3.1"

View File

@@ -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:

View File

@@ -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