deepseek-translate-server 0.1.0__tar.gz

deepseek_translate_server-0.1.0/.gitignore

@@ -0,0 +1,24 @@
+# Python
+__pycache__/
+*.py[cod]
+*$py.class
+*.so
+.Python
+.venv/
+venv/
+ENV/
+env/
+dist/
+build/
+*.egg-info/
+.pytest_cache/
+
+# IDE
+.vscode/
+.idea/
+*.swp
+*.swo
+
+# OS
+.DS_Store
+Thumbs.db

deepseek_translate_server-0.1.0/PKG-INFO

@@ -0,0 +1,84 @@
+Metadata-Version: 2.4
+Name: deepseek-translate-server
+Version: 0.1.0
+Summary: MCP server for translating Markdown to Chinese via DeepSeek
+Requires-Python: >=3.13
+Requires-Dist: fastmcp>=3.4.2
+Requires-Dist: openai>=1.65.0
+Description-Content-Type: text/markdown
+
+# deepseek-translate MCP Server
+
+将 Markdown 文本翻译为中文的 MCP server，基于 DeepSeek API。
+
+## 安装
+
+```bash
+uv sync
+```
+
+## 运行
+
+```bash
+uv run server.py
+# 或安装后
+uvx deepseek-translate-server
+```
+
+## 环境变量
+
+- `DEEPSEEK_API_KEY`：必需，DeepSeek API key。
+
+> `uvx` 启动时不会自动加载 pdf2md 目录下的 `.env`。需要在 MCP host 配置里通过 `env` 传入，或在启动 server 的 shell 环境中 export。
+
+## 工具
+
+### `translate_md`
+
+翻译单段 Markdown 文本或单个文件。
+
+参数：
+
+- `text`: Markdown 文本（与 `file_path` 二选一）
+- `file_path`: 本地 Markdown 文件路径（与 `text` 二选一）
+- `model`: DeepSeek 模型，默认 `deepseek-v4-flash`
+- `temperature`: 采样温度，默认 `0.3`
+- `thinking`: 是否开启 reasoning，默认 `False`
+
+返回：翻译后的 Markdown 文本。
+
+### `translate_md_dir`
+
+并发翻译目录下所有 Markdown 文件。
+
+参数：
+
+- `dir_path`: 目录路径
+- `pattern`: 文件匹配模式，默认 `*.md`
+- `model`: DeepSeek 模型，默认 `deepseek-v4-flash`
+- `temperature`: 采样温度，默认 `0.3`
+- `thinking`: 是否开启 reasoning，默认 `False`
+- `concurrency`: 最大并发数，默认 `100`
+
+行为：
+
+- 优先翻译 `dir_path/pages/` 下的文件；若不存在则翻译 `dir_path` 根目录文件
+- 对每个匹配文件生成 `{stem}_trans.md`，放在与源文件相同目录
+- 按文件名排序合并为 `dir_path/full_trans.md`
+- 返回合并后的文件路径
+
+## 接入 MCP Host
+
+```json
+{
+  "mcpServers": {
+    "deepseek-translate": {
+      "command": "uvx",
+      "args": ["deepseek-translate-server"],
+      "env": {
+        "DEEPSEEK_API_KEY": "your-api-key"
+      }
+    }
+  }
+}
+```

deepseek_translate_server-0.1.0/README.md

@@ -0,0 +1,75 @@
+# deepseek-translate MCP Server
+
+将 Markdown 文本翻译为中文的 MCP server，基于 DeepSeek API。
+
+## 安装
+
+```bash
+uv sync
+```
+
+## 运行
+
+```bash
+uv run server.py
+# 或安装后
+uvx deepseek-translate-server
+```
+
+## 环境变量
+
+- `DEEPSEEK_API_KEY`：必需，DeepSeek API key。
+
+> `uvx` 启动时不会自动加载 pdf2md 目录下的 `.env`。需要在 MCP host 配置里通过 `env` 传入，或在启动 server 的 shell 环境中 export。
+
+## 工具
+
+### `translate_md`
+
+翻译单段 Markdown 文本或单个文件。
+
+参数：
+
+- `text`: Markdown 文本（与 `file_path` 二选一）
+- `file_path`: 本地 Markdown 文件路径（与 `text` 二选一）
+- `model`: DeepSeek 模型，默认 `deepseek-v4-flash`
+- `temperature`: 采样温度，默认 `0.3`
+- `thinking`: 是否开启 reasoning，默认 `False`
+
+返回：翻译后的 Markdown 文本。
+
+### `translate_md_dir`
+
+并发翻译目录下所有 Markdown 文件。
+
+参数：
+
+- `dir_path`: 目录路径
+- `pattern`: 文件匹配模式，默认 `*.md`
+- `model`: DeepSeek 模型，默认 `deepseek-v4-flash`
+- `temperature`: 采样温度，默认 `0.3`
+- `thinking`: 是否开启 reasoning，默认 `False`
+- `concurrency`: 最大并发数，默认 `100`
+
+行为：
+
+- 优先翻译 `dir_path/pages/` 下的文件；若不存在则翻译 `dir_path` 根目录文件
+- 对每个匹配文件生成 `{stem}_trans.md`，放在与源文件相同目录
+- 按文件名排序合并为 `dir_path/full_trans.md`
+- 返回合并后的文件路径
+
+## 接入 MCP Host
+
+```json
+{
+  "mcpServers": {
+    "deepseek-translate": {
+      "command": "uvx",
+      "args": ["deepseek-translate-server"],
+      "env": {
+        "DEEPSEEK_API_KEY": "your-api-key"
+      }
+    }
+  }
+}
+```

deepseek_translate_server-0.1.0/pyproject.toml

@@ -0,0 +1,26 @@
+[project]
+name = "deepseek-translate-server"
+version = "0.1.0"
+description = "MCP server for translating Markdown to Chinese via DeepSeek"
+readme = "README.md"
+requires-python = ">=3.13"
+dependencies = [
+    "fastmcp>=3.4.2",
+    "openai>=1.65.0",
+]
+
+[project.scripts]
+deepseek-translate-server = "server:main"
+
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[tool.hatch.build.targets.wheel]
+packages = ["server.py"]
+
+[dependency-groups]
+dev = [
+    "pytest>=9.1.0",
+    "pytest-asyncio>=1.4.0",
+]

deepseek_translate_server-0.1.0/server.py

@@ -0,0 +1,200 @@
+"""DeepSeek Translate MCP Server — Markdown 文本翻译为中文.
+
+暴露 tool:
+  - translate_md      : 翻译单段 Markdown 文本或单个文件
+  - translate_md_dir  : 并发翻译目录下所有 Markdown 文件，输出 _trans.md 并合并
+
+环境变量:
+  DEEPSEEK_API_KEY  — 必需，DeepSeek API key
+"""
+
+import asyncio
+import fnmatch
+import os
+import re
+from pathlib import Path
+
+from fastmcp import FastMCP
+from openai import AsyncOpenAI
+
+mcp = FastMCP("DeepSeekTranslate")
+
+DEFAULT_MODEL = "deepseek-v4-flash"
+DEFAULT_TEMPERATURE = 0.3
+DEFAULT_TIMEOUT = 600
+DEFAULT_CONCURRENCY = 100
+
+_SYSTEM_PROMPT = """你是英文学术论文翻译助手，将 Markdown 文本翻译为地道的中文学术表达。
+
+必须原封不动保留的元素（仅翻译其周围的自然语言）：
+- 数学公式、LaTeX、代码块
+- 引用标注、URL、DOI、邮箱、文件路径
+- 人名、机构名、地名、约定俗成的算法/模型/框架名
+- Markdown 图片语法中的路径（alt 文字可译）
+
+翻译要求：
+- 学术中文风格，术语全文统一
+- 仅对核心概念或关键结论使用 **加粗**
+- 只输出翻译后的 Markdown，不加任何前言/后注
+- 保持原文空行、缩进和换行结构
+"""
+
+
+def _extract_page_index(filename: str) -> int:
+    match = re.search(r"(\d+)", filename)
+    return int(match.group(1)) if match else -1
+
+
+def _get_api_key() -> str:
+    key = os.environ.get("DEEPSEEK_API_KEY", "")
+    if not key:
+        raise ValueError("DEEPSEEK_API_KEY 环境变量未设置")
+    return key
+
+
+def _create_client(timeout: int = DEFAULT_TIMEOUT) -> AsyncOpenAI:
+    return AsyncOpenAI(
+        api_key=_get_api_key(),
+        base_url="https://api.deepseek.com",
+        timeout=timeout,
+    )
+
+
+def _build_kwargs(text: str, model: str, temperature: float, thinking: bool) -> dict[str, object]:
+    kwargs: dict[str, object] = {
+        "model": model,
+        "messages": [
+            {"role": "system", "content": _SYSTEM_PROMPT},
+            {"role": "user", "content": text},
+        ],
+        "temperature": temperature,
+        "stream": False,
+    }
+    if not thinking:
+        kwargs["extra_body"] = {"thinking": {"type": "disabled"}}
+    return kwargs
+
+
+async def _translate_one(
+    client: AsyncOpenAI,
+    text: str,
+    model: str,
+    temperature: float,
+    thinking: bool,
+) -> str:
+    """翻译单段文本，返回翻译后的 Markdown。"""
+    kwargs = _build_kwargs(text, model, temperature, thinking)
+    response = await client.chat.completions.create(**kwargs)
+    msg = response.choices[0].message
+    return msg.content or ""
+
+
+@mcp.tool
+async def translate_md(
+    text: str | None = None,
+    file_path: str | None = None,
+    model: str = DEFAULT_MODEL,
+    temperature: float = DEFAULT_TEMPERATURE,
+    thinking: bool = False,
+) -> str:
+    """将 Markdown 文本翻译为中文。
+
+    提供 text 或 file_path，二选一。
+
+    Args:
+        text: 待翻译的 Markdown 文本。
+        file_path: 本地 Markdown 文件路径。
+        model: DeepSeek 模型名称，默认 deepseek-v4-flash。
+        temperature: 采样温度，默认 0.3。
+        thinking: 是否开启 reasoning，默认 False。
+    """
+    if not text and not file_path:
+        raise ValueError("必须提供 text 或 file_path 其中之一")
+    if text and file_path:
+        raise ValueError("text 和 file_path 只能提供一个")
+
+    if file_path:
+        src = Path(file_path)
+        if not src.exists():
+            raise FileNotFoundError(f"文件不存在: {file_path}")
+        text = src.read_text(encoding="utf-8")
+
+    client = _create_client()
+    return await _translate_one(client, text, model, temperature, thinking)
+
+
+@mcp.tool
+async def translate_md_dir(
+    dir_path: str,
+    pattern: str = "*.md",
+    model: str = DEFAULT_MODEL,
+    temperature: float = DEFAULT_TEMPERATURE,
+    thinking: bool = False,
+    concurrency: int = DEFAULT_CONCURRENCY,
+) -> str:
+    """并发翻译目录下所有 Markdown 文件，输出 _trans.md 后缀文件并合并。
+
+    如果 dir_path/pages/ 存在，则翻译其中的文件；否则翻译 dir_path 根目录文件。
+    单页翻译结果与源文件放在同一目录，合并结果保存为 dir_path/full_trans.md。
+
+    Args:
+        dir_path: Markdown 文件所在目录。
+        pattern: 文件匹配模式，默认 *.md。
+        model: DeepSeek 模型名称，默认 deepseek-v4-flash。
+        temperature: 采样温度，默认 0.3。
+        thinking: 是否开启 reasoning，默认 False。
+        concurrency: 最大并发数，默认 100。
+    """
+    directory = Path(dir_path)
+    if not directory.is_dir():
+        raise NotADirectoryError(f"目录不存在: {dir_path}")
+
+    # 优先使用 pages/ 子目录（paddle-ocr 的默认输出结构）
+    source_dir = directory / "pages"
+    if not source_dir.is_dir():
+        source_dir = directory
+
+    files = sorted(
+        [
+            f
+            for f in source_dir.iterdir()
+            if f.is_file()
+            and fnmatch.fnmatch(f.name, pattern)
+            and not f.name.endswith("_trans.md")
+        ],
+        key=lambda p: _extract_page_index(p.name),
+    )
+    if not files:
+        raise ValueError(f"目录下没有匹配 {pattern} 的文件")
+
+    client = _create_client()
+    semaphore = asyncio.Semaphore(concurrency)
+    results: list[tuple[Path, str]] = []
+
+    async def _translate_file(src: Path) -> tuple[Path, str]:
+        text = src.read_text(encoding="utf-8")
+        async with semaphore:
+            translated = await _translate_one(client, text, model, temperature, thinking)
+        return src, translated
+
+    tasks = [asyncio.create_task(_translate_file(f)) for f in files]
+    for fut in asyncio.as_completed(tasks):
+        src, translated = await fut
+        out_path = src.with_stem(src.stem + "_trans")
+        out_path.write_text(translated, encoding="utf-8")
+        results.append((src, translated))
+
+    # 按原文件顺序合并
+    results.sort(key=lambda x: _extract_page_index(x[0].name))
+    merged_text = "\n\n---\n\n".join([t for _, t in results])
+    merged_path = directory / "full_trans.md"
+    merged_path.write_text(merged_text, encoding="utf-8")
+    return str(merged_path)
+
+
+def main():
+    mcp.run()
+
+
+if __name__ == "__main__":
+    main()

deepseek_translate_server-0.1.0/test_server.py

@@ -0,0 +1,13 @@
+"""DeepSeek Translate MCP server 基础测试."""
+
+import pytest
+
+from server import mcp
+
+
+@pytest.mark.asyncio
+async def test_list_tools():
+    tools = await mcp.list_tools()
+    names = {t.name for t in tools}
+    assert "translate_md" in names
+    assert "translate_md_dir" in names