@aigne/doc-smith 0.9.10 → 0.9.11-beta

package/agents/localize/choose-language.mjs

@@ -1,107 +0,0 @@
-import { SUPPORTED_LANGUAGES } from "../../utils/constants/index.mjs";
-import { loadConfigFromFile, saveValueToConfig } from "../../utils/utils.mjs";
-
-/**
- * Interactively selects languages for translation.
- * @param {Object} params
- * @param {Array<string>} [params.langs] - Pre-selected languages.
- * @param {string} params.locale - The primary language code.
- * @param {Object} options - Options object with prompts.
- * @returns {Promise<Object>} The selected languages.
- */
-export default async function chooseLanguage({ langs, locale, selectedDocs }, options) {
-  let selectedLanguages = [];
-
-  // Load existing config to get current translation languages
-  const existingConfig = await loadConfigFromFile();
-  const currentTranslateLanguages = existingConfig?.translateLanguages || [];
-
-  // Get primary language from config or parameter
-  const primaryLanguage = locale || existingConfig?.locale || "en";
-
-  // Filter out the primary language from available choices (like init.mjs)
-  const availableTranslationLanguages = SUPPORTED_LANGUAGES.filter(
-    (lang) => lang.code !== primaryLanguage,
-  );
-
-  // If languages are provided as a parameter, validate them
-  if (langs && Array.isArray(langs) && langs.length > 0) {
-    const validLanguages = langs.filter((lang) =>
-      availableTranslationLanguages.some((availableLang) => availableLang.code === lang),
-    );
-
-    if (validLanguages.length > 0) {
-      selectedLanguages = validLanguages;
-    } else {
-      console.log(`⚠️  Invalid languages provided: ${langs.join(", ")}`);
-      console.log(
-        `Available languages: ${availableTranslationLanguages.map((l) => l.code).join(", ")}`,
-      );
-    }
-  }
-
-  // If no valid languages were provided, let user select from available languages
-  if (selectedLanguages.length === 0) {
-    // Create choices from available translation languages with labels
-    const choices = availableTranslationLanguages.map((lang) => ({
-      name: `${lang.label} - ${lang.sample}`,
-      value: lang.code,
-      checked: currentTranslateLanguages.includes(lang.code), // Default to previously selected languages
-    }));
-
-    selectedLanguages = await options.prompts.checkbox({
-      message: "Please select the languages you'd like to translate to:",
-      choices: choices,
-      validate: (answer) => {
-        if (answer.length === 0) {
-          return "You must select at least one language.";
-        }
-        return true;
-      },
-    });
-  }
-
-  if (selectedLanguages.length === 0) {
-    throw new Error("You must select at least one language to continue.");
-  }
-
-  // Find new languages that need to be added
-  const newLanguages = selectedLanguages.filter(
-    (lang) => !currentTranslateLanguages.includes(lang),
-  );
-
-  if (newLanguages.length > 0) {
-    // Add new languages to existing ones
-    const updatedTranslateLanguages = [...currentTranslateLanguages, ...newLanguages];
-    await saveValueToConfig("translateLanguages", updatedTranslateLanguages);
-  }
-
-  // Convert selectedLanguages to translates format
-  const translates = selectedLanguages.map((lang) => ({ language: lang }));
-
-  return {
-    selectedLanguages,
-    selectedDocs,
-    translates,
-  };
-}
-
-chooseLanguage.input_schema = {
-  type: "object",
-  properties: {
-    langs: {
-      type: "array",
-      items: {
-        type: "string",
-      },
-      description: "A list of pre-selected languages for translation.",
-    },
-
-    locale: {
-      type: "string",
-      description:
-        "The primary language code, which will be excluded from the translation options.",
-    },
-  },
-  required: [],
-};

package/agents/localize/record-translation-history.mjs

@@ -1,23 +0,0 @@
-import { recordUpdate } from "../../utils/history-utils.mjs";
-
-export default function recordTranslationHistory({ selectedPaths, feedback }) {
-  // Skip if no feedback provided
-  if (!feedback?.trim()) {
-    return {};
-  }
-
-  if (!Array.isArray(selectedPaths) || selectedPaths.length === 0) {
-    return {};
-  }
-
-  // Record translation history for this document
-  recordUpdate({
-    operation: "translation_update",
-    feedback: feedback.trim(),
-    docPaths: selectedPaths,
-  });
-
-  return {};
-}
-
-recordTranslationHistory.task_render_mode = "hide";

package/agents/localize/save-doc-translation-or-skip.mjs

@@ -1,18 +0,0 @@
-/**
- * Save translated document
- * Always saves the translation content, regardless of --diagram flag.
- * In --diagram mode, the translation content may only have updated diagram images,
- * but it still needs to be saved to persist the image changes.
- */
-export default async function saveDocTranslationOrSkip(input, options) {
-  // Always save translation content, whether it's a full translation or just diagram image updates
-  // The translation content (from set-review-content.mjs) already contains the updated diagram images
-  const saveDocTranslationAgent = options.context?.agents?.["saveDocTranslation"];
-  if (saveDocTranslationAgent) {
-    return await options.context.invoke(saveDocTranslationAgent, input);
-  }
-
-  return {};
-}
-
-saveDocTranslationOrSkip.task_render_mode = "hide";

package/agents/localize/set-review-content.mjs

@@ -1,58 +0,0 @@
-import { debug } from "../../utils/debug.mjs";
-import { d2CodeBlockRegex, diagramImageFullRegex } from "../../utils/d2-utils.mjs";
-
-/**
- * Replace cached diagram images in translated document and set reviewContent
- * This step runs after translate-document-wrapper.mjs to:
- * 1. Insert cached image translations into the translated document
- * 2. Set reviewContent from the final translation
- * Note: Each document has at most one diagram image
- */
-export default async function setReviewContent(input) {
-  let translation = input.translation || "";
-  const cachedImages = input.cachedDiagramImages || null;
-
-  // Replace cached diagram image if any (each document has at most one image)
-  if (cachedImages && cachedImages.length > 0) {
-    try {
-      const cachedImage = cachedImages[0]; // Only one image per document
-
-      // Find existing image in translated content
-      // Note: Translation process may copy content from main document, so we always need to
-      // replace with cached image to ensure the final document uses the correct language-specific image
-      const imageMatch = translation.match(diagramImageFullRegex);
-
-      if (imageMatch) {
-        translation = translation.replace(diagramImageFullRegex, cachedImage.translatedMarkdown);
-        debug(`✅ Replaced diagram image in translation with language-specific image`);
-      } else {
-        const d2Match = translation.match(d2CodeBlockRegex);
-        if (d2Match) {
-          translation = translation.replace(d2CodeBlockRegex, cachedImage.translatedMarkdown);
-          debug(`✅ Replaced D2 code block in translation with language-specific image`);
-        } else {
-          // No existing image, insert at the position from main document
-          const insertIndex = Math.min(cachedImage.mainImageIndex, translation.length);
-          translation =
-            translation.slice(0, insertIndex) +
-            "\n\n" +
-            cachedImage.translatedMarkdown +
-            "\n\n" +
-            translation.slice(insertIndex);
-          debug(`✅ Inserted diagram image at index ${insertIndex}`);
-        }
-      }
-    } catch (error) {
-      debug(`⚠️  Failed to replace cached diagram image: ${error.message}`);
-      // Continue with original translation if replacement fails
-    }
-  }
-
-  return {
-    ...input,
-    translation,
-    reviewContent: translation || input.content || "",
-  };
-}
-
-setReviewContent.task_render_mode = "hide";

package/agents/localize/translate-document-wrapper.mjs

@@ -1,34 +0,0 @@
-/**
- * Conditionally call translateDocument agent
- * If translation is already set and approved (--diagram mode with existing translation), skip calling translateDocument
- * Otherwise, call translateDocument agent normally (including --diagram mode when translation doesn't exist)
- */
-export default async function translateDocumentWrapper(input, options) {
-  // If translation is already set and approved, skip document translation
-  // This happens in --diagram mode when translation document exists (translate-diagram-images.mjs sets isApproved: true)
-  // In --diagram mode when translation doesn't exist, translate-diagram-images.mjs sets isApproved: false to allow translation
-  if (input.translation !== undefined && input.translation !== null && input.isApproved === true) {
-    return input;
-  }
-
-  // Otherwise, call translateDocument agent (YAML-defined AI agent)
-  const translateAgent = options.context?.agents?.["translateDocument"];
-  if (!translateAgent) {
-    throw new Error("translateDocument agent not found");
-  }
-
-  try {
-    const result = await options.context.invoke(translateAgent, input);
-    return {
-      ...input,
-      ...result,
-      translation: result?.translation || result,
-      // Preserve shouldTranslateDiagramsOnly flag
-      shouldTranslateDiagramsOnly: input.shouldTranslateDiagramsOnly,
-    };
-  } catch (error) {
-    throw new Error(`Failed to invoke translateDocument agent: ${error.message}`);
-  }
-}
-
-translateDocumentWrapper.task_render_mode = "hide";

package/agents/localize/translate-document.yaml

@@ -1,24 +0,0 @@
-name: translateDocument
-description: Translate content to another language
-instructions:
-  url: ../../prompts/translate/translate-document.md
-task_render_mode: collapse
-input_schema:
-  type: object
-  properties:
-    language:
-      type: string
-      description: Target language (e.g., 'zh' for Chinese, 'ja' for Japanese)
-    content:
-      type: string
-      description: Text content to translate
-    glossary:
-      type: string
-      description: Glossary for consistent terminology
-    feedback:
-      type: string
-      description: Tell us how to improve the translation style
-  required:
-    - language
-    - content
-output_key: translation

package/agents/localize/translate-multilingual.yaml

@@ -1,57 +0,0 @@
-type: team
-name: translateMultilingual
-description: Batch translate documents to multiple languages. Use --diagram to translate only diagram images without translating document content.
-skills:
-  # Step 1: Check --diagram flag and determine if we should translate diagrams only
-  - url: ./translate-or-skip-diagram.mjs
-  # Translate document content (only if translation is not already set)
-  - type: team
-    task_render_mode: collapse
-    name: translate
-    skills:
-      - url: ../utils/find-user-preferences-by-path.mjs
-        default_input:
-          scope: translation
-      # Step 2: Cache diagram images for translation (before document translation)
-      # Checks --diagram flag and timestamps to determine if images need translation
-      - url: ../../utils/translate-diagram-images.mjs
-      # Step 3: Translate document content
-      - url: ./translate-document-wrapper.mjs
-      # Step 4: Replace cached diagram images and set review content
-      - url: ./set-review-content.mjs
-    reflection:
-      reviewer: ../utils/check-detail-result.mjs
-      is_approved: isApproved
-      max_iterations: 5
-      return_last_on_max_iterations: true
-    task_title: Translate '{{ title }}' to '{{ language }}'
-  - url: ./save-doc-translation-or-skip.mjs
-input_schema:
-  type: object
-  properties:
-    translates:
-      type: array
-      items:
-        type: object
-        properties:
-          language:
-            type: string
-    content:
-      type: string
-    diagram:
-      type: boolean
-      description: Translate only diagram images without translating document content
-output_schema:
-  type: object
-  properties:
-    translates:
-      type: array
-      items:
-        type: object
-        properties:
-          language:
-            type: string
-          translation:
-            type: string
-iterate_on: translates
-concurrency: 3

package/agents/localize/translate-or-skip-diagram.mjs

@@ -1,52 +0,0 @@
-import { debug } from "../../utils/debug.mjs";
-
-/**
- * Check --diagram flag and conditionally translate document
- * If --diagram is set, skip document translation and only translate images
- * Otherwise, proceed with normal document translation
- */
-export default async function translateOrSkipDiagram(input) {
-  // Check if --diagram flag is set
-  let shouldTranslateDiagramsOnly = false;
-
-  if (process.argv) {
-    const hasDiagramFlag = process.argv.some((arg) => arg === "--diagram" || arg === "-d");
-    if (hasDiagramFlag) {
-      shouldTranslateDiagramsOnly = true;
-    }
-  }
-
-  if (input.diagram === true || input.diagram === "true") {
-    shouldTranslateDiagramsOnly = true;
-  }
-
-  if (
-    process.env.DOC_SMITH_TRANSLATE_DIAGRAMS_ONLY === "true" ||
-    process.env.DOC_SMITH_TRANSLATE_DIAGRAMS_ONLY === "1"
-  ) {
-    shouldTranslateDiagramsOnly = true;
-  }
-
-  // If --diagram flag is set, skip document translation
-  if (shouldTranslateDiagramsOnly) {
-    debug("⏭️  --diagram flag set: skipping document translation, only translating images");
-    // Set translation to content to skip actual translation
-    // Also set isApproved to true to skip reflection check
-    return {
-      ...input,
-      shouldTranslateDiagramsOnly: true,
-      translation: input.content || "",
-      reviewContent: input.content || "",
-      isApproved: true, // Skip reflection check
-    };
-  }
-
-  // Otherwise, proceed with normal translation
-  // Don't call translateDocument here - let translate-document-wrapper.mjs handle it
-  return {
-    ...input,
-    shouldTranslateDiagramsOnly: false,
-  };
-}
-
-translateOrSkipDiagram.task_render_mode = "hide";

package/agents/media/batch-generate-media-description.yaml

@@ -1,46 +0,0 @@
-type: team
-name: batchGenerateMediaDescription
-description: Batch generate media (image/video) descriptions with concurrency
-skills:
-  - url: ./generate-media-description.yaml
-task_render_mode: collapse
-task_title: Generate Media Description
-input_schema:
-  type: object
-  properties:
-    mediaToDescribe:
-      type: array
-      items:
-        type: object
-        properties:
-          name:
-            type: string
-          width:
-            type: number
-          height:
-            type: number
-          hash:
-            type: string
-          path:
-            type: string
-          type:
-            type: string
-          svgContent:
-            type: string
-          mediaFile:
-            type: array
-            items:
-              type: object
-              properties:
-                type:
-                  type: string
-                path:
-                  type: string
-                filename:
-                  type: string
-                mimeType:
-                  type: string
-      description: Array of media files (images/videos) that need descriptions
-iterate_on: mediaToDescribe
-concurrency: 5
-mode: sequential

package/agents/media/generate-media-description.yaml

@@ -1,50 +0,0 @@
-name: generateMediaDescription
-description: Generate description for a single media file (image/video)
-model: gemini-2.5-flash
-modalities: ["text", "image"]
-instructions:
-  - role: system
-    url: ../../prompts/media/media-description/system-prompt.md
-  - role: user
-    url: ../../prompts/media/media-description/user-prompt.md
-input_file_key: mediaFile
-include_input_in_output: true
-input_schema:
-  type: object
-  properties:
-    name:
-      type: string
-      description: Media file name
-    width:
-      type: number
-      description: Media width in pixels
-    height:
-      type: number
-      description: Media height in pixels
-    hash:
-      type: string
-    path:
-      type: string
-      description: Media path
-    type:
-      type: string
-      description: Media type (image/video)
-    svgContent:
-      type: string
-      description: SVG content
-    mediaFile:
-      type: array
-      items:
-        type: object
-        properties:
-          type:
-            type: string
-          path:
-            type: string
-          filename:
-            type: string
-          mimeType:
-            type: string
-  required:
-    - name
-output_key: description