feat: update translate docs workflow inputs

xiaoyu2er · xiaoyu2er · commit b472d5695965 · 2025-06-08T17:20:47.000-07:00
diff --git a/.github/actions/translate-docs/action.yml b/.github/actions/translate-docs/action.yml
@@ -2,8 +2,32 @@ name: "Translate Documentation Action"
 description: "Translates documentation files automatically using AI"
 
 inputs:
+  list_only:
+    description: "Only list file status without updating docs"
+    required: false
+    default: "false"
+  target_language:
+    description: "Specify the target language code for translation (e.g., 'zh-hans', 'fr', 'es')"
+    required: false
+    default: ""
+  pattern:
+    description: "File pattern to match for updating (e.g., '*.mdx' or 'docs/**/*.mdx')"
+    required: false
+    default: ""
+  max_files:
+    description: "Maximum number of files to process in one batch (per language)"
+    required: false
+    default: ""
+  concurrency:
+    description: "Number of concurrent translation tasks"
+    required: false
+    default: ""
+  model:
+    description: "DeepSeek model to use: 'deepseek-chat' or 'deepseek-reasoner'"
+    required: false
+    default: ""
   custom_arguments:
-    description: "Custom arguments to pass to the translation package command"
+    description: "Additional custom arguments to pass to the translation package command"
     required: false
     default: ""
   github_token:
@@ -97,8 +121,39 @@ runs:
       if: steps.check_branch.outputs.exists == 'false'
       shell: bash
       run: |
-        echo "Running translation with custom arguments: ${{ inputs.custom_arguments }}"
-        ${{ inputs.translation_command }} ${{ inputs.custom_arguments }}
+        # Build command with individual arguments
+        COMMAND="${{ inputs.translation_command }}"
+        
+        if [ "${{ inputs.list_only }}" = "true" ]; then
+          COMMAND="$COMMAND --list-only"
+        fi
+        
+        if [ -n "${{ inputs.target_language }}" ]; then
+          COMMAND="$COMMAND --target-language ${{ inputs.target_language }}"
+        fi
+        
+        if [ -n "${{ inputs.pattern }}" ]; then
+          COMMAND="$COMMAND --pattern ${{ inputs.pattern }}"
+        fi
+        
+        if [ -n "${{ inputs.max_files }}" ]; then
+          COMMAND="$COMMAND --max ${{ inputs.max_files }}"
+        fi
+        
+        if [ -n "${{ inputs.concurrency }}" ]; then
+          COMMAND="$COMMAND --concurrency ${{ inputs.concurrency }}"
+        fi
+        
+        if [ -n "${{ inputs.model }}" ]; then
+          COMMAND="$COMMAND --model ${{ inputs.model }}"
+        fi
+        
+        if [ -n "${{ inputs.custom_arguments }}" ]; then
+          COMMAND="$COMMAND ${{ inputs.custom_arguments }}"
+        fi
+        
+        echo "Running translation command: $COMMAND"
+        eval $COMMAND
       env:
         OPENAI_API_KEY: ${{ inputs.api_key }}
 
diff --git a/.github/workflows/translate-docs.yml b/.github/workflows/translate-docs.yml
@@ -17,8 +17,38 @@ on:
         required: false
         type: string
         default: 'docs/update-translations'
+      list_only:
+        description: 'Only list file status without updating docs'
+        required: false
+        type: boolean
+        default: false
+      target_language:
+        description: 'Specify the target language code for translation (e.g., "zh-hans", "fr", "es")'
+        required: false
+        type: string
+      pattern:
+        description: 'File pattern to match for updating (e.g., "*.mdx" or "docs/**/*.mdx")'
+        required: false
+        type: string
+      max_files:
+        description: 'Maximum number of files to process in one batch (per language)'
+        required: false
+        type: number
+      concurrency:
+        description: 'Number of concurrent translation tasks'
+        required: false
+        type: number
+        default: 10
+      model:
+        description: 'DeepSeek model to use'
+        required: false
+        type: choice
+        options:
+          - deepseek-chat
+          - deepseek-reasoner
+        default: deepseek-chat
       custom_arguments:
-        description: 'Custom arguments to pass to the translation package command. e.g., "-t zh-hans"'
+        description: 'Additional custom arguments to pass to the translation package command'
         required: false
         type: string
 
@@ -58,8 +88,15 @@ jobs:
           api_key: ${{ secrets.OPENAI_API_KEY }}
           # Optional inputs with their default values shown
           github_token: ${{ secrets.PAT_TOKEN }}  # Use PAT instead of GITHUB_TOKEN
-          custom_arguments: ${{ github.event.inputs.custom_arguments }}
           pr_branch: ${{ github.event.inputs.pr_branch || 'docs/update-translations' }}
+          # Translation-specific inputs
+          list_only: ${{ github.event.inputs.list_only }}
+          target_language: ${{ github.event.inputs.target_language }}
+          pattern: ${{ github.event.inputs.pattern }}
+          max_files: ${{ github.event.inputs.max_files }}
+          concurrency: ${{ github.event.inputs.concurrency }}
+          model: ${{ github.event.inputs.model }}
+          custom_arguments: ${{ github.event.inputs.custom_arguments }}
           # translation_command: 'pnpm run translate'
           base_branch: 'dev'
           # pr_branch: 'docs/update-translations'
diff --git a/packages/translate/src/main.ts b/packages/translate/src/main.ts
@@ -49,7 +49,7 @@ export async function main({
 
   logger.divider();
   logger.info(
-    `Translation for ${docsRoot} in languages: ${LANGUAGES.join(', ')} started!`,
+    `Translation for ${docsRoot} in languages: ${LANGUAGES.join(', ')} using ${model} started!`,
   );
 
   // Normalize paths and prepare patterns
@@ -157,6 +157,7 @@ export async function main({
       const { shouldUpdate, reason, chunks } = await getDocUpdateStatus({
         sourcePath,
         targetPath,
+        model,
       });
 
       if (verbose) {
diff --git a/packages/translate/src/utils.ts b/packages/translate/src/utils.ts
@@ -18,6 +18,7 @@ interface LangConfig {
 interface CheckFileUpdateParams {
   sourcePath: string;
   targetPath: string;
+  model?: DeepSeekModel;
 }
 
 interface BuildTranslationContextParams {
@@ -60,6 +61,7 @@ export function getLastModifiedTimeFromGit(filePath: string): Date {
 export async function getDocUpdateStatus({
   sourcePath,
   targetPath,
+  model = 'deepseek-chat',
 }: CheckFileUpdateParams): Promise<{
   shouldUpdate: boolean;
   chunks: 'N/A' | number;
@@ -75,8 +77,8 @@ export async function getDocUpdateStatus({
   }
 
   const sourceContent = await fs$.readFile(sourcePath, 'utf8');
-  const chunks = needsChunking(sourceContent)
-    ? splitIntoChunks(sourceContent).length
+  const chunks = needsChunking(sourceContent, model)
+    ? splitIntoChunks(sourceContent, model).length
     : 1;
 
   try {
@@ -179,7 +181,9 @@ export async function translateDoc({
   model = 'deepseek-chat',
 }: TranslateDocumentFileParams) {
   // Create directory if it doesn't exist
-  logger.debug(`Translating ${sourcePath} to ${targetPath}`);
+  logger.debug(
+    `Translating ${sourcePath} to ${targetPath} using model ${model}`,
+  );
   await fs$.mkdir(path.dirname(targetPath), { recursive: true });
 
   // Read source file
