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

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

@@ -48,13 +48,13 @@ function formatDocumentStructure(structure) {
 
 function printDocumentStructure(structure) {
   console.log(`\n  ${"-".repeat(50)}`);
-  console.log("  Current Document Structure");
+  console.log("  Current Documentation Structure");
   console.log(`  ${"-".repeat(50)}`);
 
   const { rootNodes, printNode } = formatDocumentStructure(structure);
 
   if (rootNodes.length === 0) {
-    console.log("  No document structure found.");
+    console.log("  No documentation structure found.");
   } else {
     rootNodes.forEach((node) => {
       printNode(node);
@@ -64,19 +64,19 @@ function printDocumentStructure(structure) {
 }
 
 export default async function userReviewDocumentStructure({ documentStructure, ...rest }, options) {
-  // Check if document structure exists
+  // Check if documentation structure exists
   if (!documentStructure || !Array.isArray(documentStructure) || documentStructure.length === 0) {
-    console.log("No document structure was generated to review.");
+    console.log("No documentation structure was generated to review.");
     return { documentStructure };
   }
 
-  // Print current document structure in a user-friendly format
+  // Print current documentation structure in a user-friendly format
   printDocumentStructure(documentStructure);
 
-  // Ask user if they want to review the document structure
+  // Ask user if they want to review the documentation structure
   const needReview = await options.prompts.select({
     message:
-      "Would you like to optimize the document structure?\n  You can edit titles, reorganize sections.",
+      "Would you like to optimize the documentation structure?\n  You can edit titles, reorganize sections.",
     choices: [
       {
         name: "Looks good - proceed with current structure",
@@ -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++;
 
@@ -117,7 +120,7 @@ export default async function userReviewDocumentStructure({ documentStructure, .
     const refineAgent = options.context.agents["updateDocumentStructure"];
     if (!refineAgent) {
       console.log(
-        "Unable to process your feedback - the document structure update feature is unavailable.",
+        "Unable to process your feedback - the documentation structure update feature is unavailable.",
       );
       console.log("Please try again later or contact support if this continues.");
       break;
@@ -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"];
@@ -157,7 +158,7 @@ export default async function userReviewDocumentStructure({ documentStructure, .
         }
       }
 
-      // Print current document structure in a user-friendly format
+      // Print current documentation structure in a user-friendly format
       printDocumentStructure(currentStructure);
     } catch (error) {
       console.error("Error processing your feedback:");
@@ -172,4 +173,4 @@ export default async function userReviewDocumentStructure({ documentStructure, .
   return { documentStructure: currentStructure };
 }
 
-userReviewDocumentStructure.taskTitle = "User review and modify document structure";
+userReviewDocumentStructure.taskTitle = "User review and modify documentation structure";

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/batch-generate-document.yaml

@@ -8,7 +8,7 @@ input_schema:
   properties:
     datasources:
       type: string
-      description: Context for document structure generation, used to assist generate document structure
+      description: Context for documentation structure generation, used to assist generate documentation structure
     documentExecutionStructure: ../schema/document-execution-structure.yaml
     modifiedFiles:
       type: array

package/agents/update/check-document.mjs

@@ -36,10 +36,10 @@ export default async function checkDocument(
     detailGenerated = false;
   }
 
-  // Check if sourceIds have changed by comparing with original document structure
+  // Check if sourceIds have changed by comparing with original documentation structure
   let sourceIdsChanged = false;
   if (originalDocumentStructure && sourceIds) {
-    // Find the original node in the document structure
+    // Find the original node in the documentation structure
     const originalNode = originalDocumentStructure.find((node) => node.path === path);
 
     if (originalNode?.sourceIds) {

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 },
     };
   }
 }

package/agents/update/fs-tools/glob.mjs

@@ -0,0 +1,184 @@
+import fs from "node:fs";
+import fsPromises from "node:fs/promises";
+import path from "node:path";
+
+/**
+ * Sorts files by modification time (newest first), then alphabetically
+ */
+function sortFilesByModTime(files, basePath) {
+  return files
+    .map((file) => {
+      const fullPath = path.isAbsolute(file) ? file : path.resolve(basePath, file);
+      try {
+        const stats = fs.statSync(fullPath);
+        return {
+          path: file,
+          fullPath,
+          mtime: stats.mtimeMs || 0,
+        };
+      } catch {
+        return {
+          path: file,
+          fullPath,
+          mtime: 0,
+        };
+      }
+    })
+    .sort((a, b) => {
+      // Sort by modification time (newest first), then alphabetically
+      if (b.mtime !== a.mtime) {
+        return b.mtime - a.mtime;
+      }
+      return a.path.localeCompare(b.path);
+    })
+    .map((item) => item.path);
+}
+
+/**
+ * Filters out common ignore patterns
+ */
+function shouldIgnoreFile(filePath) {
+  const ignorePatterns = [
+    "node_modules",
+    ".git",
+    ".DS_Store",
+    ".vscode",
+    ".idea",
+    "dist",
+    "build",
+    "*.log",
+    "coverage",
+    ".nyc_output",
+    ".cache",
+  ];
+
+  const normalizedPath = filePath.replace(/\\/g, "/");
+
+  return ignorePatterns.some((pattern) => {
+    if (pattern.includes("*")) {
+      // Simple wildcard matching for patterns like "*.log"
+      const regex = new RegExp(pattern.replace(/\*/g, ".*"));
+      return regex.test(path.basename(normalizedPath));
+    } else {
+      // Directory or file name matching
+      return (
+        normalizedPath.includes(`/${pattern}/`) ||
+        normalizedPath.endsWith(`/${pattern}`) ||
+        normalizedPath.startsWith(`${pattern}/`) ||
+        normalizedPath === pattern
+      );
+    }
+  });
+}
+
+export default async function glob({
+  pattern,
+  case_sensitive = false,
+  respect_git_ignore = true,
+  limit = 100,
+}) {
+  let result = [];
+  let error = null;
+  const searchDir = process.cwd();
+
+  try {
+    // Validate required parameters
+    if (!pattern || typeof pattern !== "string" || pattern.trim() === "") {
+      throw new Error("Pattern parameter is required and cannot be empty");
+    }
+
+    // Use Node.js built-in glob
+    const globOptions = {
+      cwd: searchDir,
+      nodir: true, // Only return files, not directories
+      dot: true, // Include hidden files
+    };
+
+    // Note: Node.js fs.glob doesn't support case_sensitive option directly
+    // We'll handle case sensitivity in post-processing if needed
+    const iter = fsPromises.glob(pattern, globOptions);
+    const files = [];
+
+    for await (const file of iter) {
+      if (files.length >= limit) break;
+
+      // Apply ignore filters
+      if (shouldIgnoreFile(file)) {
+        continue;
+      }
+
+      // Handle case sensitivity if needed
+      if (case_sensitive === false) {
+        // Node.js glob is case-sensitive by default on most systems
+        // For case-insensitive matching, we rely on the pattern itself
+        // or the filesystem behavior
+      }
+
+      files.push(file);
+    }
+
+    // Sort files by modification time (newest first)
+    const sortedFiles = sortFilesByModTime(files, searchDir);
+
+    // Build result message
+    let message;
+    if (sortedFiles.length === 0) {
+      message = `No files found matching pattern "${pattern}"`;
+    } else {
+      const fileCount = sortedFiles.length;
+      const truncated = files.length >= limit;
+      message = `Found ${fileCount}${truncated ? "+" : ""} file(s) matching "${pattern}"`;
+      message += ", sorted by modification time (newest first):";
+    }
+
+    result = {
+      files: sortedFiles,
+      count: sortedFiles.length,
+      message,
+      truncated: files.length >= limit,
+    };
+  } catch (err) {
+    error = err;
+    result = {
+      files: [],
+      count: 0,
+      message: `Error during glob search: ${err.message}`,
+      truncated: false,
+    };
+  }
+
+  return {
+    command: "glob",
+    arguments: {
+      pattern,
+      case_sensitive,
+      respect_git_ignore,
+      limit,
+    },
+    result,
+    error: error && { message: error.message },
+  };
+}
+
+glob.input_schema = {
+  type: "object",
+  properties: {
+    pattern: {
+      type: "string",
+      description: 'The glob pattern to match files against (e.g., "**/*.js", "src/**/*.ts")',
+    },
+    case_sensitive: {
+      type: "boolean",
+      description: "Optional: Whether the search should be case-sensitive (defaults to false)",
+    },
+    respect_git_ignore: {
+      type: "boolean",
+      description: "Optional: Whether to respect .gitignore patterns (defaults to true)",
+    },
+    limit: {
+      type: "number",
+      description: "Optional: Maximum number of files to return (defaults to 100)",
+    },
+  },
+  required: ["pattern"],
+};