diff --git a/docutranslate/core/schemas.py b/docutranslate/core/schemas.py index 4aa69d0..0a9c71e 100644 --- a/docutranslate/core/schemas.py +++ b/docutranslate/core/schemas.py @@ -25,6 +25,13 @@ from docutranslate.agents.agent import ThinkingMode from docutranslate.agents.thinking.thinking_factory import ProviderType from docutranslate.translator import default_params +# --- 公共类型定义 --- +WorkflowType = Literal[ + "auto", "markdown_based", "txt", "json", "xlsx", "docx", + "srt", "epub", "html", "ass", "pptx" +] +InsertMode = Literal["replace", "append", "prepend"] + class GlossaryAgentConfigPayload(BaseModel): base_url: str = Field( diff --git a/docutranslate/sdk.py b/docutranslate/sdk.py index df0c16c..a376ecd 100644 --- a/docutranslate/sdk.py +++ b/docutranslate/sdk.py @@ -10,24 +10,14 @@ from typing import Optional, Literal, Dict, Any, List, Union from pydantic import TypeAdapter -from docutranslate.core.schemas import TranslatePayload, GlossaryAgentConfigPayload +from docutranslate.agents.agent import ThinkingMode +from docutranslate.agents.provider import ProviderType + +from docutranslate.core.schemas import TranslatePayload, GlossaryAgentConfigPayload, WorkflowType, InsertMode from docutranslate.core.factory import create_workflow_from_payload from docutranslate.translator import default_params from docutranslate.global_values.conditional_import import DOCLING_EXIST -# --- 类型定义 --- -WorkflowType = Literal[ - "auto", "markdown_based", "txt", "json", "xlsx", "docx", - "srt", "epub", "html", "ass", "pptx" -] -ThinkingMode = Literal["default", "enable", "disable"] -InsertMode = Literal["replace", "append", "prepend"] -ProviderType = Literal[ - "ollama", "open.bigmodel.cn", "dashscope.aliyuncs.com", - "ark.cn-beijing.volces.com", "generativelanguage.googleapis.com", - "api.siliconflow.cn", "api.302.ai" -] - # --- 映射配置 --- _WORKFLOW_MAPPINGS = { "markdown_based": {"save": "save_as_markdown_zip", "export": "export_to_markdown_zip"}, @@ -92,7 +82,7 @@ class TranslationResult: return self._workflow -class DocuTranslate: +class Client: """ DocuTranslate SDK。 """