Leon/docutranslate
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

修复model_version错误

Browse Source
This commit is contained in:
xunbu
2025-08-20 15:54:57 +08:00
parent 4caa7d8843
commit 5110bba898
3 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
docutranslate/app.py
View File

@@ -220,7 +220,7 @@ class MarkdownWorkflowParams(BaseWorkflowParams):
mineru_token: Optional[str] = Field(None, description="当 `convert_engine` 为 'mineru' 时必填的API令牌。") mineru_token: Optional[str] = Field(None, description="当 `convert_engine` 为 'mineru' 时必填的API令牌。")
formula_ocr: bool = Field(True, description="是否对公式进行OCR识别。对 `mineru` 和 `docling` 均有效。") formula_ocr: bool = Field(True, description="是否对公式进行OCR识别。对 `mineru` 和 `docling` 均有效。")
code_ocr: bool = Field(True, description="是否对代码块进行OCR识别。仅 `docling` 引擎有效。") code_ocr: bool = Field(True, description="是否对代码块进行OCR识别。仅 `docling` 引擎有效。")
model_version: Literal["pipline", "vlm"] = Field("vlm", description="Mineru模型的版本，'vlm'是更新的版本。仅 `mineru` 引擎有效。") model_version: Literal["pipeline", "vlm"] = Field("vlm", description="Mineru模型的版本，'vlm'是更新的版本。仅 `mineru` 引擎有效。")
@field_validator('mineru_token') @field_validator('mineru_token')
def check_mineru_token(cls, v, values): def check_mineru_token(cls, v, values):
@@ -1216,7 +1216,7 @@ async def temp_translate(
temperature: float = Body(default_params["temperature"]), temperature: float = Body(default_params["temperature"]),
thinking: ThinkingMode = Body(default_params["thinking"]), thinking: ThinkingMode = Body(default_params["thinking"]),
chunk_size: int = Body(default_params["chunk_size"]), custom_prompt: Optional[str] = Body(None), chunk_size: int = Body(default_params["chunk_size"]), custom_prompt: Optional[str] = Body(None),
model_version: Literal["pipline", "vlm"] = Body("vlm"), model_version: Literal["pipeline", "vlm"] = Body("vlm"),
): ):
file_name = Path(file_name) file_name = Path(file_name)
try: try:

2
docutranslate/converter/x2md/converter_mineru.py
View File

@@ -18,7 +18,7 @@ URL = 'https://mineru.net/api/v4/file-urls/batch'
class ConverterMineruConfig(X2MarkdownConverterConfig): class ConverterMineruConfig(X2MarkdownConverterConfig):
mineru_token: str mineru_token: str
formula_ocr: bool = True formula_ocr: bool = True
model_version: Literal["pipline", "vlm"] = "vlm" model_version: Literal["pipeline", "vlm"] = "vlm"
def gethash(self) -> Hashable: def gethash(self) -> Hashable:
return self.formula_ocr,self.model_version return self.formula_ocr,self.model_version

2
docutranslate/static/index.html
View File

File diff suppressed because one or more lines are too long