@aigne/doc-smith 0.9.7 → 0.9.8-beta.1

package/agents/utils/save-doc-translation.mjs

@@ -1,4 +1,8 @@
 import { saveDocTranslation as _saveDocTranslation } from "../../utils/utils.mjs";
+import { readFileContent } from "../../utils/docs-finder-utils.mjs";
+import { getFileName } from "../../utils/utils.mjs";
+import { debug } from "../../utils/debug.mjs";
+import { syncDiagramToTranslations } from "../../utils/sync-diagram-to-translations.mjs";
 
 export default async function saveDocTranslation({
   path,
@@ -7,6 +11,7 @@ export default async function saveDocTranslation({
   language,
   labels,
   isShowMessage = false,
+  locale,
 }) {
   await _saveDocTranslation({
     path,
@@ -16,6 +21,35 @@ export default async function saveDocTranslation({
     labels,
   });
 
+  // Sync diagram images from main document to translations
+  // This ensures all images (including diagrams) in the main document are synced to translation files
+  if (path && docsDir && locale) {
+    try {
+      // Read main document content (it should already be saved)
+      const mainFileName = getFileName(path, locale);
+      const mainContent = await readFileContent(docsDir, mainFileName);
+
+      if (mainContent) {
+        const syncResult = await syncDiagramToTranslations(
+          mainContent,
+          path,
+          docsDir,
+          locale,
+          "sync",
+        );
+
+        if (syncResult.updated > 0) {
+          debug(
+            `✅ Synced diagram images to ${syncResult.updated} translation file(s) after translation`,
+          );
+        }
+      }
+    } catch (error) {
+      // Don't fail the translation if sync fails
+      debug(`⚠️  Failed to sync diagram images after translation: ${error.message}`);
+    }
+  }
+
   if (isShowMessage) {
     const message = `✅ Translation completed successfully.`;
     return { message };

package/agents/utils/save-doc.mjs

@@ -1,5 +1,6 @@
 import { shutdownMermaidWorkerPool } from "../../utils/mermaid-worker-pool.mjs";
 import { saveDoc as _saveDoc } from "../../utils/utils.mjs";
+import { debug } from "../../utils/debug.mjs";
 
 export default async function saveDoc({
   path,
@@ -9,6 +10,8 @@ export default async function saveDoc({
   locale,
   feedback,
   isShowMessage = false,
+  intentType,
+  originalContent,
   ...rest
 }) {
   await _saveDoc({
@@ -19,6 +22,43 @@ export default async function saveDoc({
     locale,
   });
 
+  // Sync diagram changes to translation documents if needed
+  // Only sync for diagram-related operations (addDiagram, updateDiagram, deleteDiagram)
+  if (
+    docsDir &&
+    path &&
+    intentType &&
+    ["addDiagram", "updateDiagram", "deleteDiagram"].includes(intentType)
+  ) {
+    try {
+      const { syncDiagramToTranslations } = await import(
+        "../../utils/sync-diagram-to-translations.mjs"
+      );
+
+      // Determine operation type for sync
+      // deleteDiagram -> "delete" (process even if 0 diagrams)
+      // addDiagram/updateDiagram -> "update" (skip if 0 diagrams)
+      const operationType = intentType === "deleteDiagram" ? "delete" : "update";
+
+      const syncResult = await syncDiagramToTranslations(
+        content,
+        path,
+        docsDir,
+        locale || "en",
+        operationType,
+      );
+
+      if (syncResult.updated > 0) {
+        debug(
+          `✅ Synced diagram changes to ${syncResult.updated} translation file(s) for ${intentType}`,
+        );
+      }
+    } catch (error) {
+      // Don't fail the operation if sync fails
+      debug(`⚠️  Failed to sync diagram to translations: ${error.message}`);
+    }
+  }
+
   if (isShowMessage) {
     // Shutdown mermaid worker pool to ensure clean exit
     try {
@@ -39,6 +79,8 @@ export default async function saveDoc({
     locale,
     feedback,
     isShowMessage,
+    intentType,
+    originalContent,
     ...rest,
   };
 }

package/agents/utils/skip-if-content-exists.mjs

@@ -0,0 +1,27 @@
+/**
+ * Determine whether to skip document generation when the document already exists.
+ * If intentType is diagram-related *and* content is already present, mark that
+ * generation should be skipped so downstream agents can short-circuit.
+ */
+export default async function skipIfContentExists(input) {
+  const { intentType, content } = input;
+
+  const isDiagramIntent =
+    intentType && ["addDiagram", "updateDiagram", "deleteDiagram"].includes(intentType);
+  const shouldSkipGeneration = Boolean(isDiagramIntent && content);
+
+  return {
+    ...input,
+    skipGenerateDocument: shouldSkipGeneration,
+    // Ensure downstream steps have content available when skipping
+    ...(shouldSkipGeneration
+      ? {
+          content,
+          documentContent: content,
+          originalContent: content,
+        }
+      : {}),
+  };
+}
+
+skipIfContentExists.task_render_mode = "hide";

package/aigne.yaml

@@ -105,10 +105,15 @@ agents:
   - ./agents/evaluate/code-snippet.mjs
 
   # Diagram
-  - ./agents/create/merge-diagram.yaml
+  - ./agents/create/analyze-diagram-type.mjs
+  - ./agents/create/analyze-diagram-type-llm.yaml
+  - ./agents/create/generate-diagram-image.yaml
+  - ./agents/create/replace-d2-with-image.mjs
   - ./agents/update/generate-diagram.yaml
   - ./agents/update/check-generate-diagram.mjs
   - ./agents/update/pre-check-generate-diagram.yaml
+  - ./agents/update/check-sync-image-flag.mjs
+  - ./agents/update/sync-images-and-exit.mjs
 cli:
   chat: ./agents/chat/index.mjs
   agents: