@aigne/doc-smith 0.9.10 → 0.9.11-beta

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

@@ -1,303 +0,0 @@
-import { applyPatch } from "diff";
-import {
-  getUpdateDocumentContentInputJsonSchema,
-  getUpdateDocumentContentOutputJsonSchema,
-  validateUpdateDocumentContentInput,
-} from "../../../types/document-schema.mjs";
-import { userContextAt } from "../../../utils/utils.mjs";
-
-export default async function updateDocumentContent(input, options) {
-  // Validate input using Zod schema
-  const validation = validateUpdateDocumentContentInput(input);
-  if (!validation.success) {
-    return {
-      success: false,
-      error: { message: validation.error },
-    };
-  }
-
-  const { diffPatch, path } = validation.data;
-
-  // Get originalContent from shared context using path, fallback to input
-  const contentContext = userContextAt(options, `currentContents.${path}`);
-  let originalContent = contentContext.get();
-
-  if (!originalContent) {
-    originalContent = input.originalContent;
-  }
-
-  try {
-    // Parse and validate diff patch
-    const parsedDiff = parseDiffPatch(diffPatch);
-    if (!parsedDiff.success) {
-      return {
-        success: false,
-        error: { message: parsedDiff.error },
-      };
-    }
-
-    // Check and fix line number issues
-    const fixedDiff = fixLineNumberIssues(originalContent, parsedDiff.hunks);
-    if (!fixedDiff.success) {
-      return {
-        success: false,
-        error: { message: fixedDiff.error },
-      };
-    }
-
-    // Reconstruct the fixed diff patch
-    const fixedPatch = reconstructDiffPatch(fixedDiff.hunks);
-
-    // Apply the diff patch using the diff library
-    const result = applyPatch(originalContent, fixedPatch);
-
-    if (result === false) {
-      return {
-        success: false,
-        error: { message: "Failed to apply patch" },
-      };
-    }
-
-    // Update shared context with new content using path
-    contentContext.set(result);
-
-    return {
-      success: true,
-      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: { message: error.message },
-    };
-  }
-}
-
-function parseDiffPatch(diffPatch) {
-  try {
-    const hunks = [];
-    const lines = diffPatch.split("\n");
-    let currentHunk = null;
-
-    for (let i = 0; i < lines.length; i++) {
-      const line = lines[i];
-
-      // Parse hunk header: @@ -oldStart,oldCount +newStart,newCount @@
-      const hunkMatch = line.match(/^@@\s+-(\d+)(?:,(\d+))?\s+\+(\d+)(?:,(\d+))?\s+@@/);
-      if (hunkMatch) {
-        if (currentHunk) {
-          hunks.push(currentHunk);
-        }
-        currentHunk = {
-          oldStart: parseInt(hunkMatch[1], 10),
-          oldCount: parseInt(hunkMatch[2], 10) || 1,
-          newStart: parseInt(hunkMatch[3], 10),
-          newCount: parseInt(hunkMatch[4], 10) || 1,
-          changes: [],
-        };
-        continue;
-      }
-
-      // Parse changes
-      if (currentHunk && line.length > 0) {
-        const changeType = line[0];
-        const content = line.slice(1);
-
-        if (changeType === "-") {
-          currentHunk.changes.push({ type: "remove", content });
-        } else if (changeType === "+") {
-          currentHunk.changes.push({ type: "add", content });
-        } else if (changeType === " ") {
-          currentHunk.changes.push({ type: "context", content });
-        }
-      }
-    }
-
-    if (currentHunk) {
-      hunks.push(currentHunk);
-    }
-
-    if (hunks.length === 0) {
-      return { success: false, error: "No valid hunks found in diff" };
-    }
-
-    return { success: true, hunks };
-  } catch (error) {
-    return { success: false, error: `Failed to parse diff: ${error.message}` };
-  }
-}
-
-function fixLineNumberIssues(originalContent, hunks) {
-  try {
-    const originalLines = originalContent.split("\n");
-    const fixedHunks = [];
-
-    for (const hunk of hunks) {
-      // Extract context and removed lines to find best match
-      const contextAndRemoved = hunk.changes
-        .filter((change) => change.type === "context" || change.type === "remove")
-        .map((change) => change.content);
-
-      if (contextAndRemoved.length === 0) {
-        // No context to match against, keep original
-        fixedHunks.push(hunk);
-        continue;
-      }
-
-      // Find best matching position in original content
-      const bestMatch = findBestMatch(originalLines, contextAndRemoved, hunk.oldStart - 1);
-
-      if (bestMatch.found) {
-        const fixedHunk = {
-          ...hunk,
-          oldStart: bestMatch.position + 1,
-          newStart: bestMatch.position + 1,
-          // Fix line counts based on actual changes
-          oldCount: calculateOldCount(hunk.changes),
-          newCount: calculateNewCount(hunk.changes),
-        };
-        fixedHunks.push(fixedHunk);
-      } else {
-        // Try fuzzy matching
-        const fuzzyMatch = findFuzzyMatch(originalLines, contextAndRemoved);
-        if (fuzzyMatch.found) {
-          const fixedHunk = {
-            ...hunk,
-            oldStart: fuzzyMatch.position + 1,
-            newStart: fuzzyMatch.position + 1,
-            // Fix line counts based on actual changes
-            oldCount: calculateOldCount(hunk.changes),
-            newCount: calculateNewCount(hunk.changes),
-          };
-          fixedHunks.push(fixedHunk);
-        } else {
-          return {
-            success: false,
-            error: `Cannot find matching context for hunk at line ${hunk.oldStart}`,
-          };
-        }
-      }
-    }
-
-    return { success: true, hunks: fixedHunks };
-  } catch (error) {
-    return { success: false, error: `Failed to fix line numbers: ${error.message}` };
-  }
-}
-
-function findBestMatch(originalLines, targetLines, startPosition) {
-  // Try exact match at expected position first
-  if (startPosition >= 0 && startPosition + targetLines.length <= originalLines.length) {
-    let matches = 0;
-    for (let i = 0; i < targetLines.length; i++) {
-      if (originalLines[startPosition + i] === targetLines[i]) {
-        matches++;
-      }
-    }
-
-    if (matches === targetLines.length) {
-      return { found: true, position: startPosition };
-    }
-  }
-
-  // Try nearby positions (within 10 lines)
-  const NEARBY_SEARCH_RANGE = 10; // Maximum number of lines to search before/after expected position
-  for (let offset = 1; offset <= NEARBY_SEARCH_RANGE; offset++) {
-    // Try before
-    const beforePos = startPosition - offset;
-    if (beforePos >= 0 && beforePos + targetLines.length <= originalLines.length) {
-      let matches = 0;
-      for (let i = 0; i < targetLines.length; i++) {
-        if (originalLines[beforePos + i] === targetLines[i]) {
-          matches++;
-        }
-      }
-      if (matches === targetLines.length) {
-        return { found: true, position: beforePos };
-      }
-    }
-
-    // Try after
-    const afterPos = startPosition + offset;
-    if (afterPos >= 0 && afterPos + targetLines.length <= originalLines.length) {
-      let matches = 0;
-      for (let i = 0; i < targetLines.length; i++) {
-        if (originalLines[afterPos + i] === targetLines[i]) {
-          matches++;
-        }
-      }
-      if (matches === targetLines.length) {
-        return { found: true, position: afterPos };
-      }
-    }
-  }
-
-  return { found: false };
-}
-
-function findFuzzyMatch(originalLines, targetLines) {
-  // Find the best partial match
-  let bestMatch = { found: false, position: 0, score: 0 };
-
-  for (let pos = 0; pos <= originalLines.length - targetLines.length; pos++) {
-    let matches = 0;
-    for (let i = 0; i < targetLines.length; i++) {
-      if (originalLines[pos + i] === targetLines[i]) {
-        matches++;
-      }
-    }
-
-    const score = matches / targetLines.length;
-    const FUZZY_MATCH_THRESHOLD = 0.7; // 70% similarity threshold for fuzzy matching
-    if (score > bestMatch.score && score >= FUZZY_MATCH_THRESHOLD) {
-      bestMatch = { found: true, position: pos, score };
-    }
-  }
-
-  return bestMatch;
-}
-
-function calculateOldCount(changes) {
-  const contextCount = changes.filter((c) => c.type === "context").length;
-  const removeCount = changes.filter((c) => c.type === "remove").length;
-  return contextCount + removeCount;
-}
-
-function calculateNewCount(changes) {
-  const contextCount = changes.filter((c) => c.type === "context").length;
-  const addCount = changes.filter((c) => c.type === "add").length;
-  return contextCount + addCount;
-}
-
-function reconstructDiffPatch(hunks) {
-  let patchContent = "";
-
-  for (const hunk of hunks) {
-    // Add hunk header
-    const oldRange = hunk.oldCount === 1 ? hunk.oldStart : `${hunk.oldStart},${hunk.oldCount}`;
-    const newRange = hunk.newCount === 1 ? hunk.newStart : `${hunk.newStart},${hunk.newCount}`;
-    patchContent += `@@ -${oldRange} +${newRange} @@\n`;
-
-    // Add changes
-    for (const change of hunk.changes) {
-      if (change.type === "context") {
-        patchContent += ` ${change.content}\n`;
-      } else if (change.type === "remove") {
-        patchContent += `-${change.content}\n`;
-      } else if (change.type === "add") {
-        patchContent += `+${change.content}\n`;
-      }
-    }
-  }
-
-  return patchContent;
-}
-
-updateDocumentContent.inputSchema = getUpdateDocumentContentInputJsonSchema();
-updateDocumentContent.outputSchema = getUpdateDocumentContentOutputJsonSchema();
-
-updateDocumentContent.description = "Apply diff patch to update markdown document content";

package/agents/update/generate-diagram.yaml

@@ -1,80 +0,0 @@
-type: team
-task_render_mode: collapse
-name: generateDiagram
-skills:
-  # Step 1: Analyze document content to determine diagram type and style
-  - ../create/analyze-diagram-type.mjs
-  # Step 2: Transform output for image generation agent (map aspectRatio to ratio, add size, pass existingImage and feedback)
-  - type: transform
-    jsonata: |
-      $merge([
-        $,
-        {
-          "ratio": aspectRatio,
-          "size": "1K",
-          "existingImage": existingImage,
-          "useImageToImage": useImageToImage,
-          "feedback": feedback
-        }
-      ])
-  # Step 3: Generate diagram image directly with unified agent
-  - ../create/generate-diagram-image.yaml
-  # Step 4: Replace placeholder with generated image
-  - ../create/replace-d2-with-image.mjs
-input_schema:
-  type: object
-  properties:
-    documentContent:
-      type: string
-      description: The **raw text content** of the current document. (**Note:** This is the original document and **does not include** any diagram source code.)
-    locale:
-      type: string
-      description: Language for diagram labels and text
-      default: en
-    diagramming:
-      type: object
-      description: Diagramming configuration
-      properties:
-        style:
-          type: string
-          description: Default diagram style to use when no style is specified in feedback
-    feedback:
-      type: string
-      description: User feedback that may contain style or type preferences, or diagram index (e.g., 'use anthropomorphic style', 'update the second diagram')
-      default: ""
-    path:
-      type: string
-      description: Document path (e.g., "guides/getting-started.md") used for generating image filename
-    docsDir:
-      type: string
-      description: Documentation directory where assets will be saved (relative to project root)
-    intentAnalysis:
-      type: object
-      description: Analysis results from analyzeFeedbackIntent containing intentType, diagramInfo, generationMode, and changes
-      properties:
-        intentType:
-          type: string
-          enum: ["addDiagram", "updateDiagram", "deleteDiagram", "updateDocument"]
-        diagramInfo:
-          type: object
-          nullable: true
-          properties:
-            path: { type: "string" }
-            index: { type: "number" }
-            markdown: { type: "string" }
-        generationMode:
-          type: string
-          enum: ["image-to-image", "text-only", "add-new", "remove-image"]
-        changes:
-          type: array
-          items: { type: "string" }
-  required:
-    - documentContent
-output_schema:
-  type: object
-  properties:
-    content:
-      type: string
-      description: The document content with D2 code blocks replaced by generated diagram image.
-  required:
-    - content

package/agents/update/generate-document.yaml

@@ -1,70 +0,0 @@
-name: generateDocument
-description: Comprehensive content generator for documentation that creates detailed, high-quality document content from various data sources with intelligent analysis.
-instructions:
-  - role: system
-    url: ../../prompts/detail/generate/system-prompt.md
-  - role: user
-    url: ../../prompts/detail/generate/user-prompt.md
-auto_reorder_system_messages: true
-auto_merge_system_messages: true
-input_schema:
-  type: object
-  properties:
-    rules:
-      type: string
-      description: User's content generation rules and requirements
-    locale:
-      type: string
-      description: User language, such as zh, en
-    detailDataSource:
-      type: string
-      description: Source data and context for document content generation
-    targetAudience:
-      type: string
-      description: Target audience for document content
-    nodeName:
-      type: string
-      description: Document node name for content generation
-    originalDocumentStructure: ../schema/document-structure.yaml
-    title:
-      type: string
-    description:
-      type: string
-    path:
-      type: string
-    originalContent:
-      type: string
-      description: The original content of the document before update
-    parentId:
-      type:
-        - string
-        - "null"
-    glossary:
-      type: string
-      description: Glossary
-    additionalInformation:
-      type: string
-      description: Additional supplementary information
-    intentType:
-      type:
-        - string
-        - "null"
-      description: User intent type analyzed from feedback (addDiagram, updateDiagram, deleteDiagram, updateDocument). Can be null if not specified.
-      enum: ["addDiagram", "updateDiagram", "deleteDiagram", "updateDocument", null]
-  required:
-    - rules
-    - detailDataSource
-    - originalDocumentStructure
-output_key: content
-afs:
-  modules:
-    - module: system-fs
-      options:
-        name: sources
-        localPath: .
-        description: |
-          Codebase of the project to be documented used as context for document generation,
-          should search and read as needed while generating document content
-keep_text_in_tool_uses: false
-# skills:
-#   - ./generate-diagram.yaml

package/agents/update/handle-document-update.yaml

@@ -1,103 +0,0 @@
-type: team
-name: handleDocumentUpdate
-description: Update a document in a batch
-skills:
-  - ../utils/transform-detail-data-sources.mjs
-  # Check if document generation should be skipped (if content already exists)
-  # If intentType is diagram-related and content exists, mark skipGenerateDocument so generation can be short-circuited
-  - url: ../utils/skip-if-content-exists.mjs
-  # Generate document content (skipped when skipGenerateDocument is true and content exists)
-  - type: team
-    task_render_mode: collapse
-    name: generateDocumentContent
-    skills:
-      - url: ../utils/find-user-preferences-by-path.mjs
-        default_input:
-          scope: document
-      - ../utils/generate-document-or-skip.mjs
-    reflection:
-      reviewer: ../utils/check-detail-result.mjs
-      is_approved: isApproved
-      max_iterations: 5
-      return_last_on_max_iterations: true
-    task_title: Generate document for '{{ title }}'
-  # Ensure content, documentContent, and originalContent are set
-  - type: transform
-    jsonata: |
-      $merge([
-        $,
-        { 
-          "documentContent": content,
-          "originalContent": content
-        }
-      ])
-  - ./check-generate-diagram.mjs
-  # - ../generate/merge-diagram.yaml
-  - ../utils/save-doc.mjs
-input_schema:
-  type: object
-  properties:
-    nodeName:
-      type: string
-      description: Node name
-      default: Section
-    locale:
-      type: string
-      description: Language
-    targetAudience:
-      type: string
-      description: Target audience
-    glossary:
-      type: string
-      description: Glossary
-    doc-path:
-      type: string
-      description: Current path
-    feedback:
-      type: string
-      description: Feedback
-    sources:
-      type: array
-      items:
-        type: string
-        description: Source code path
-    sourcesPath:
-      type: array
-      items:
-        type: string
-        description: Source code path
-      default:
-        - ./
-    includePatterns:
-      type: array
-      items:
-        type: string
-        description: Included file name patterns
-    excludePatterns:
-      type: array
-      items:
-        type: string
-        description: Excluded file name patterns
-    outputDir:
-      type: string
-      description: Output directory
-      default: ./aigne-docs/output
-    docsDir:
-      type: string
-      description: Documentation directory
-      default: ./aigne-docs/docs
-    additionalInformation:
-      type: string
-      description: Additional information
-output_schema:
-  type: object
-  properties:
-    title:
-      type: string
-    description:
-      type: string
-    path:
-      type: string
-    content:
-      type: string
-mode: sequential

package/agents/update/index.yaml

@@ -1,69 +0,0 @@
-type: team
-name: update
-alias:
-  - up
-description: Update specific documents and their translations. Use --diagram to filter and select documents with diagrams, or --diagram-all to auto-update all diagrams.
-skills:
-  - url: ../init/index.mjs
-    default_input:
-      skipIfExists: true
-      checkOnly: true
-  - url: ../init/check.mjs
-  - ../utils/load-sources.mjs
-  - type: transform
-    task_render_mode: hide
-    jsonata: |
-      $merge([
-        $,
-        {
-          'documentStructure': originalDocumentStructure
-        }
-      ])
-  # Check if --diagram or --diagram-all flag is set
-  - url: ../update/check-diagram-flag.mjs
-  - url: ../utils/choose-docs.mjs
-    default_input:
-      requiredFeedback: false
-  - ../utils/format-document-structure.mjs
-  - ../utils/ensure-document-icons.mjs
-  - ../media/load-media-description.mjs
-  - ../update/check-update-is-single.mjs
-  - ../update/save-and-translate-document.mjs
-  - url: ../utils/action-success.mjs
-    default_input:
-      action: "✅ Documents updated successfully!\n💡 Looks good? Run `aigne doc publish` to go live."
-input_schema:
-  type: object
-  properties:
-    glossary:
-      type: string
-      description: Glossary file for consistent terminology (use @filename.md)
-    docs:
-      type: array
-      items:
-        type: string
-      description: Documents to update
-    feedback:
-      type: string
-      description: Tell us what to change in this content
-    reset:
-      type: boolean
-      description: Start fresh - ignore previous versions
-    diagram:
-      type: ["boolean", "string"]
-      description: "Flag to update diagrams: true for user selection (can also use --diagram CLI arg)"
-    "diagram-all":
-      type: ["boolean", "string"]
-      description: "Flag to auto-select all documents with diagrams (can also use --diagram-all CLI arg)"
-output_schema:
-  type: object
-  properties:
-    title:
-      type: string
-    description:
-      type: string
-    path:
-      type: string
-    content:
-      type: string
-mode: sequential

package/agents/update/pre-check-generate-diagram.yaml

@@ -1,44 +0,0 @@
-name: preCheckGenerateDiagram
-description: Pre-check for generating diagram
-model:
-  reasoning_effort: 1
-instructions:
-  url: ../../prompts/detail/diagram/pre-check.md
-input_schema:
-  type: object
-  properties:
-    documentContent:
-      type: string
-      description: Source content of the document
-    feedback:
-      type: string
-      description: User generation feedback
-    detailFeedback:
-      type: string
-      description: Current document detail feedback
-    previousGenerationContent:
-      type: string
-      description: Previous document content
-  required:
-    - documentContent
-output_schema:
-  type: object
-  properties:
-    content:
-      type: string
-      description: Document content
-    details:
-      type: array
-      description: Detailed needDiagram score
-      items:
-        type: object
-        properties:
-          type:
-            type: string
-          score:
-            type: number
-          reason:
-            type: string
-  required:
-    - content
-