@aigne/doc-smith 0.8.11-beta.6 → 0.8.11-beta.7

package/.release-please-manifest.json

@@ -1,3 +1,3 @@
 {
-  ".": "0.8.11-beta.6"
+  ".": "0.8.11-beta.7"
 }

package/CHANGELOG.md

@@ -1,5 +1,18 @@
 # Changelog
 
+## [0.8.11-beta.7](https://github.com/AIGNE-io/aigne-doc-smith/compare/v0.8.11-beta.6...v0.8.11-beta.7) (2025-10-03)
+
+
+### Features
+
+* enhance document generator and updater by fs tools ([#156](https://github.com/AIGNE-io/aigne-doc-smith/issues/156)) ([64ee91c](https://github.com/AIGNE-io/aigne-doc-smith/commit/64ee91c59498b4fd788e287cb81bbc640d042bcc))
+* use shared context for faster structure and doc tuning ([#157](https://github.com/AIGNE-io/aigne-doc-smith/issues/157)) ([e5cb30e](https://github.com/AIGNE-io/aigne-doc-smith/commit/e5cb30ec114d51a19f0086b3e4aaf2ec70aea046))
+
+
+### Bug Fixes
+
+* update docs and refine custom component prompts ([#160](https://github.com/AIGNE-io/aigne-doc-smith/issues/160)) ([233f3ca](https://github.com/AIGNE-io/aigne-doc-smith/commit/233f3ca449646b39a3e493d2bb1fe59e50376827))
+
 ## [0.8.11-beta.6](https://github.com/AIGNE-io/aigne-doc-smith/compare/v0.8.11-beta.5...v0.8.11-beta.6) (2025-10-02)
 
 

package/README.md

@@ -107,7 +107,7 @@ DocSmith supports multiple AI providers:
 ```bash
 # Switch models effortlessly
 aigne doc generate --model google:gemini-2.5-pro
-aigne doc generate --model claude:claude-3-5-sonnet
+aigne doc generate --model anthropic:claude-sonnet-4-5
 aigne doc generate --model openai:gpt-4o
 ```
 

package/agents/generate/document-structure-tools/add-document.mjs

@@ -4,32 +4,47 @@ import {
   validateAddDocumentInput,
 } from "../../../types/document-structure-schema.mjs";
 
-export default async function addDocument(input) {
+export default async function addDocument(input, options) {
   // Validate input using Zod schema
   const validation = validateAddDocumentInput(input);
   if (!validation.success) {
-    console.log(`⚠️  Cannot add document: ${validation.error}`);
-    return { documentStructure: input.documentStructure };
+    const errorMessage = `Cannot add document: ${validation.error}`;
+    console.log(`⚠️  ${errorMessage}`);
+    return {
+      documentStructure: input.documentStructure,
+      error: { message: errorMessage },
+    };
   }
 
-  const { documentStructure, title, description, path, parentId, sourceIds } = validation.data;
+  const { title, description, path, parentId, sourceIds } = validation.data;
+  let documentStructure = options?.context?.userContext?.currentStructure;
+
+  if (!documentStructure) {
+    documentStructure = input.documentStructure;
+  }
 
   // Validate parent exists if parentId is provided
   if (parentId && parentId !== "null") {
     const parentExists = documentStructure.some((item) => item.path === parentId);
     if (!parentExists) {
-      console.log(`⚠️  Cannot add document: Parent document '${parentId}' not found.`);
-      return { documentStructure };
+      const errorMessage = `Cannot add document: Parent document '${parentId}' not found.`;
+      console.log(`⚠️  ${errorMessage}`);
+      return {
+        documentStructure,
+        error: { message: errorMessage },
+      };
     }
   }
 
   // Check if document with same path already exists
   const existingDocument = documentStructure.find((item) => item.path === path);
   if (existingDocument) {
-    console.log(
-      `⚠️  Cannot add document: A document with path '${path}' already exists. Choose a different path.`,
-    );
-    return { documentStructure };
+    const errorMessage = `Cannot add document: A document with path '${path}' already exists. Choose a different path.`;
+    console.log(`⚠️  ${errorMessage}`);
+    return {
+      documentStructure,
+      error: { message: errorMessage },
+    };
   }
 
   // Create new document object
@@ -44,8 +59,18 @@ export default async function addDocument(input) {
   // Add the document to the structure
   const updatedStructure = [...documentStructure, newDocument];
 
+  const successMessage = `addDocument executed successfully.
+  Successfully added document '${title}' with path '${path}'.
+  Check if the latest version of documentStructure meets user feedback, if so, just return 'success'.`;
+
+  // update shared document structure
+  if (options?.context?.userContext) {
+    options.context.userContext.currentStructure = updatedStructure;
+  }
+
   return {
     documentStructure: updatedStructure,
+    message: successMessage,
     addedDocument: newDocument,
   };
 }

package/agents/generate/document-structure-tools/delete-document.mjs

@@ -4,23 +4,34 @@ import {
   validateDeleteDocumentInput,
 } from "../../../types/document-structure-schema.mjs";
 
-export default async function deleteDocument(input) {
+export default async function deleteDocument(input, options) {
   // Validate input using Zod schema
   const validation = validateDeleteDocumentInput(input);
   if (!validation.success) {
-    console.log(`⚠️  Cannot delete document: ${validation.error}`);
-    return { documentStructure: input.documentStructure };
+    const errorMessage = `Cannot delete document: ${validation.error}`;
+    console.log(`⚠️  ${errorMessage}`);
+    return {
+      documentStructure: input.documentStructure,
+      error: { message: errorMessage },
+    };
   }
 
-  const { documentStructure, path } = validation.data;
+  const { path } = validation.data;
+  let documentStructure = options?.context?.userContext?.currentStructure;
+
+  if (!documentStructure) {
+    documentStructure = input.documentStructure;
+  }
 
   // Find the document to delete
   const documentIndex = documentStructure.findIndex((item) => item.path === path);
   if (documentIndex === -1) {
-    console.log(
-      `⚠️  Cannot delete document: Document '${path}' does not exist. Please choose an existing document to delete.`,
-    );
-    return { documentStructure };
+    const errorMessage = `Cannot delete document: Document '${path}' does not exist. Please choose an existing document to delete.`;
+    console.log(`⚠️  ${errorMessage}`);
+    return {
+      documentStructure,
+      error: { message: errorMessage },
+    };
   }
 
   const documentToDelete = documentStructure[documentIndex];
@@ -28,17 +39,29 @@ export default async function deleteDocument(input) {
   // Check if any other documents have this document as parent
   const childDocuments = documentStructure.filter((item) => item.parentId === path);
   if (childDocuments.length > 0) {
-    console.log(
-      `⚠️  Cannot delete document: Document '${path}' has ${childDocuments.length} child document(s): ${childDocuments.map((p) => p.path).join(", ")}. Please first move or delete these child documents.`,
-    );
-    return { documentStructure };
+    const errorMessage = `Cannot delete document: Document '${path}' has ${childDocuments.length} child document(s): ${childDocuments.map((p) => p.path).join(", ")}. Please first move or delete these child documents.`;
+    console.log(`⚠️  ${errorMessage}`);
+    return {
+      documentStructure,
+      error: { message: errorMessage },
+    };
   }
 
   // Remove the document from the structure
   const updatedStructure = documentStructure.filter((_, index) => index !== documentIndex);
 
+  const successMessage = `deleteDocument executed successfully.
+  Successfully deleted document '${documentToDelete.title}' with path '${path}'.
+  Check if the latest version of documentStructure meets user feedback, if so, just return 'success'.`;
+
+  // update shared document structure
+  if (options?.context?.userContext) {
+    options.context.userContext.currentStructure = updatedStructure;
+  }
+
   return {
     documentStructure: updatedStructure,
+    message: successMessage,
     deletedDocument: documentToDelete,
   };
 }

package/agents/generate/document-structure-tools/move-document.mjs

@@ -4,23 +4,34 @@ import {
   validateMoveDocumentInput,
 } from "../../../types/document-structure-schema.mjs";
 
-export default async function moveDocument(input) {
+export default async function moveDocument(input, options) {
   // Validate input using Zod schema
   const validation = validateMoveDocumentInput(input);
   if (!validation.success) {
-    console.log(`Error: Cannot move document - ${validation.error}`);
-    return { documentStructure: input.documentStructure };
+    const errorMessage = `Cannot move document: ${validation.error}`;
+    console.log(`⚠️  ${errorMessage}`);
+    return {
+      documentStructure: input.documentStructure,
+      error: { message: errorMessage },
+    };
   }
 
-  const { documentStructure, path, newParentId } = validation.data;
+  const { path, newParentId } = validation.data;
+  let documentStructure = options?.context?.userContext?.currentStructure;
+
+  if (!documentStructure) {
+    documentStructure = input.documentStructure;
+  }
 
   // Find the document to move
   const documentIndex = documentStructure.findIndex((item) => item.path === path);
   if (documentIndex === -1) {
-    console.log(
-      `Error: Cannot move document - Document '${path}' does not exist. Please select an existing document to move.`,
-    );
-    return { documentStructure };
+    const errorMessage = `Cannot move document: Document '${path}' does not exist. Please select an existing document to move.`;
+    console.log(`⚠️  ${errorMessage}`);
+    return {
+      documentStructure,
+      error: { message: errorMessage },
+    };
   }
 
   const documentToMove = documentStructure[documentIndex];
@@ -34,10 +45,12 @@ export default async function moveDocument(input) {
   ) {
     const newParentExists = documentStructure.some((item) => item.path === newParentId);
     if (!newParentExists) {
-      console.log(
-        `Error: Cannot move document - Target parent document '${newParentId}' does not exist. Please select an existing parent document.`,
-      );
-      return { documentStructure };
+      const errorMessage = `Cannot move document: Target parent document '${newParentId}' does not exist. Please select an existing parent document.`;
+      console.log(`⚠️  ${errorMessage}`);
+      return {
+        documentStructure,
+        error: { message: errorMessage },
+      };
     }
 
     // Check for circular dependency: the new parent cannot be a descendant of the document being moved
@@ -52,10 +65,12 @@ export default async function moveDocument(input) {
     };
 
     if (isDescendant(path, newParentId)) {
-      console.log(
-        `Error: Cannot move document - Moving '${path}' under '${newParentId}' would create an invalid hierarchy. Please select a parent that is not nested under the document being moved.`,
-      );
-      return { documentStructure };
+      const errorMessage = `Cannot move document: Moving '${path}' under '${newParentId}' would create an invalid hierarchy. Please select a parent that is not nested under the document being moved.`;
+      console.log(`⚠️  ${errorMessage}`);
+      return {
+        documentStructure,
+        error: { message: errorMessage },
+      };
     }
   }
 
@@ -69,10 +84,21 @@ export default async function moveDocument(input) {
   const updatedStructure = [...documentStructure];
   updatedStructure[documentIndex] = updatedDocument;
 
+  const newParentText = newParentId ? `'${newParentId}'` : "root level";
+  const successMessage = `moveDocument executed successfully.
+  Successfully moved document '${documentToMove.title}' to ${newParentText}.
+  Check if the latest version of documentStructure meets user feedback, if so, just return 'success'.`;
+
+  // update shared document structure
+  if (options?.context?.userContext) {
+    options.context.userContext.currentStructure = updatedStructure;
+  }
+
   return {
     documentStructure: updatedStructure,
+    message: successMessage,
     originalDocument: documentToMove,
-    updatedDocument,
+    updatedDocument: updatedDocument,
   };
 }
 

package/agents/generate/document-structure-tools/update-document.mjs

@@ -4,23 +4,34 @@ import {
   validateUpdateDocumentInput,
 } from "../../../types/document-structure-schema.mjs";
 
-export default async function updateDocument(input) {
+export default async function updateDocument(input, options) {
   // Validate input using Zod schema
   const validation = validateUpdateDocumentInput(input);
   if (!validation.success) {
-    console.log(`⚠️  Cannot update document: ${validation.error}`);
-    return { documentStructure: input.documentStructure };
+    const errorMessage = `Cannot update document: ${validation.error}`;
+    console.log(`⚠️  ${errorMessage}`);
+    return {
+      documentStructure: input.documentStructure,
+      error: { message: errorMessage },
+    };
   }
 
-  const { documentStructure, path, title, description, sourceIds } = validation.data;
+  const { path, title, description, sourceIds } = validation.data;
+  let documentStructure = options?.context?.userContext?.currentStructure;
+
+  if (!documentStructure) {
+    documentStructure = input.documentStructure;
+  }
 
   // Find the document to update
   const documentIndex = documentStructure.findIndex((item) => item.path === path);
   if (documentIndex === -1) {
-    console.log(
-      `⚠️  Cannot update document: Document '${path}' does not exist. Choose an existing document to update.`,
-    );
-    return { documentStructure };
+    const errorMessage = `Cannot update document: Document '${path}' does not exist. Choose an existing document to update.`;
+    console.log(`⚠️  ${errorMessage}`);
+    return {
+      documentStructure,
+      error: { message: errorMessage },
+    };
   }
 
   const originalDocument = documentStructure[documentIndex];
@@ -37,10 +48,26 @@ export default async function updateDocument(input) {
   const updatedStructure = [...documentStructure];
   updatedStructure[documentIndex] = updatedDocument;
 
+  const updates = [];
+  if (title !== undefined) updates.push(`title to '${title}'`);
+  if (description !== undefined) updates.push("description");
+  if (sourceIds !== undefined) updates.push("sourceIds");
+  const updatesText = updates.length > 0 ? updates.join(", ") : "properties";
+
+  const successMessage = `updateDocument executed successfully.
+  Successfully updated ${updatesText} for document '${path}'.
+  Check if the latest version of documentStructure meets user feedback, if so, just return 'success'.`;
+
+  // update shared document structure
+  if (options?.context?.userContext) {
+    options.context.userContext.currentStructure = updatedStructure;
+  }
+
   return {
     documentStructure: updatedStructure,
-    originalDocument,
-    updatedDocument,
+    message: successMessage,
+    originalDocument: originalDocument,
+    updatedDocument: updatedDocument,
   };
 }
 

package/agents/generate/update-document-structure.yaml

@@ -28,13 +28,7 @@ input_schema:
   required:
     - documentStructure
     - feedback
-output_schema:
-  type: object
-  properties:
-    documentStructure: ../schema/document-structure.yaml
-    operationSummary:
-      type: string
-      description: Summary of the operations performed on the documentation structure
+output_key: message
 skills:
   - ./document-structure-tools/add-document.mjs
   - ./document-structure-tools/delete-document.mjs

package/agents/generate/user-review-document-structure.mjs

@@ -97,6 +97,9 @@ export default async function userReviewDocumentStructure({ documentStructure, .
 
   const MAX_ITERATIONS = 100;
   let iterationCount = 0;
+
+  // share current structure with updateDocumentStructure agent
+  options.context.userContext.currentStructure = currentStructure;
   while (iterationCount < MAX_ITERATIONS) {
     iterationCount++;
 
@@ -132,16 +135,14 @@ export default async function userReviewDocumentStructure({ documentStructure, .
 
     try {
       // Call refineDocumentStructure agent with feedback
-      const result = await options.context.invoke(refineAgent, {
+      await options.context.invoke(refineAgent, {
         ...rest,
         feedback: feedback.trim(),
         documentStructure: currentStructure,
         userPreferences,
       });
 
-      if (result.documentStructure) {
-        currentStructure = result.documentStructure;
-      }
+      currentStructure = options.context.userContext.currentStructure;
 
       // Check if feedback should be saved as user preference
       const feedbackRefinerAgent = options.context.agents["checkFeedbackRefiner"];

package/agents/translate/translate-document.yaml

@@ -21,12 +21,4 @@ input_schema:
   required:
     - language
     - content
-output_schema:
-  type: object
-  properties:
-    translation:
-      type: string
-      description: Translated text
-    language:
-      type: string
-      description: Language code of the translation
+output_key: translation

package/agents/update/check-update-is-single.mjs

@@ -10,7 +10,8 @@ export default async function checkUpdateIsSingle({ selectedDocs, ...rest }, opt
   let targetAgent;
   let agentName;
 
-  if (selectedDocs.length === 1) {
+  if (selectedDocs.length === 1 && !rest.reset) {
+    // Single doc update without reset
     agentName = "updateSingleDocument";
     targetAgent = options.context.agents["updateSingleDocument"];
   } else {

package/agents/update/document-tools/update-document-content.mjs

@@ -5,18 +5,24 @@ import {
   validateUpdateDocumentContentInput,
 } from "../../../types/document-schema.mjs";
 
-export default async function updateDocumentContent(input) {
+export default async function updateDocumentContent(input, options) {
+  // Get originalContent from shared context, fallback to input
+  let originalContent = options?.context?.userContext?.currentContent;
+
+  if (!originalContent) {
+    originalContent = input.originalContent;
+  }
+
   // Validate input using Zod schema
   const validation = validateUpdateDocumentContentInput(input);
   if (!validation.success) {
     return {
       success: false,
-      error: validation.error,
-      message: "Invalid input parameters",
+      error: { message: validation.error },
     };
   }
 
-  const { originalContent, diffPatch } = validation.data;
+  const { diffPatch } = validation.data;
 
   try {
     // Parse and validate diff patch
@@ -24,8 +30,7 @@ export default async function updateDocumentContent(input) {
     if (!parsedDiff.success) {
       return {
         success: false,
-        error: parsedDiff.error,
-        message: "Invalid diff format: No valid hunks found or parsing failed",
+        error: { message: parsedDiff.error },
       };
     }
 
@@ -34,8 +39,7 @@ export default async function updateDocumentContent(input) {
     if (!fixedDiff.success) {
       return {
         success: false,
-        error: fixedDiff.error,
-        message: "Cannot fix diff line number issues",
+        error: { message: fixedDiff.error },
       };
     }
 
@@ -48,21 +52,27 @@ export default async function updateDocumentContent(input) {
     if (result === false) {
       return {
         success: false,
-        error: "Failed to apply patch",
-        message: "Diff patch could not be applied",
+        error: { message: "Failed to apply patch" },
       };
     }
 
+    // Update shared context with new content if options is provided
+    if (options?.context?.userContext) {
+      options.context.userContext.currentContent = result;
+    }
+
     return {
       success: true,
-      updatedContent: result,
-      message: "Document content updated successfully",
+      updatedContent: `<page_content>
+${result}
+</page_content>`,
+      message:
+        "Document content updated successfully.\nCheck if updatedContent meets user feedback, if so, just return 'success'.",
     };
   } catch (error) {
     return {
       success: false,
-      error: error.message,
-      message: "Failed to update document content",
+      error: { message: error.message },
     };
   }
 }