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

package/.aigne/doc-smith/config.yaml

@@ -67,8 +67,6 @@ docsDir: ./docs  # Directory to save generated documentation
 sourcesPath:  # Source code paths to analyze
   - ./
 lastGitHead: f3f14b93e6b2e6beb42b2f368c2560ab050dfd03
-appUrl: https://docsmith.aigne.io
 # ⚠️ Warning: boardId is auto-generated by system, please do not edit manually
 boardId: "docsmith"
-# Checkout ID for document deployment service
-checkoutId: ""
+appUrl: https://docsmith.aigne.io

package/.release-please-manifest.json

@@ -1,3 +1,3 @@
 {
-  ".": "0.8.11-beta.4"
+  ".": "0.8.11-beta.5"
 }

package/CHANGELOG.md

@@ -1,5 +1,13 @@
 # Changelog
 
+## [0.8.11-beta.5](https://github.com/AIGNE-io/aigne-doc-smith/compare/v0.8.11-beta.4...v0.8.11-beta.5) (2025-10-01)
+
+
+### Features
+
+* add user role requirement for publish and custom rule support ([#151](https://github.com/AIGNE-io/aigne-doc-smith/issues/151)) ([95866f9](https://github.com/AIGNE-io/aigne-doc-smith/commit/95866f9fcb2ca697da42e950e9011b29913726d4))
+* split d2 diagram generate as independent tool ([#152](https://github.com/AIGNE-io/aigne-doc-smith/issues/152)) ([8e9b811](https://github.com/AIGNE-io/aigne-doc-smith/commit/8e9b811dc6108bb19ab8a1853afb4cab92af1d62))
+
 ## [0.8.11-beta.4](https://github.com/AIGNE-io/aigne-doc-smith/compare/v0.8.11-beta.3...v0.8.11-beta.4) (2025-09-30)
 
 

package/agents/generate/check-d2-diagram-valid.mjs

@@ -0,0 +1,26 @@
+import { checkContent } from "../../utils/d2-utils.mjs";
+
+export default async function checkD2DiagramIsValid({ d2DiagramSourceCode }) {
+  try {
+    await checkContent({ content: d2DiagramSourceCode });
+    return {
+      isValid: true,
+    };
+  } catch (err) {
+    return {
+      isValid: false,
+      error: err.message,
+    };
+  }
+}
+
+checkD2DiagramIsValid.input_schema = {
+  type: "object",
+  properties: {
+    d2DiagramSourceCode: {
+      type: "string",
+      description: "Source code of d2 diagram",
+    },
+  },
+  required: ["d2DiagramSourceCode"],
+};

package/agents/generate/generate-d2-diagram.yaml

@@ -0,0 +1,23 @@
+name: drawD2Diagram
+description: Generate a D2 diagram from document content.
+instructions:
+  - role: system
+    url: ../../prompts/detail/d2-diagram/rules-system.md
+  - role: user
+    url: ../../prompts/detail/d2-diagram/rules-user.md
+input_schema:
+  type: object
+  properties:
+    documentContent:
+      type: string
+      description: Source code of current document (without the D2 diagram)
+  required:
+    - documentContent
+output_schema:
+  type: object
+  properties:
+    d2DiagramSourceCode:
+      type: string
+      description: Source code to draw D2 diagram
+  required:
+    - d2DiagramSourceCode

package/agents/generate/merge-d2-diagram.yaml

@@ -0,0 +1,39 @@
+name: mergeD2DiagramToDocument
+description: Merge D2 Diagram source code into document
+instructions: |
+  You are an AI assistant that helps to merge d2 diagram into document.
+
+  <datasources>
+  {{ content }}
+  </datasources>
+  <datasources>
+  {{ d2DiagramSourceCode }}
+  </datasources>
+
+  Given the source content of a document and the D2 diagram source code, your task is to:
+  - **Keep the original content as soon as possible.**
+  - D2 diagram source code should wrap by ```d2 and ``` in markdown format.
+  - Should find proper position to insert the D2 diagram in the document, usually after the first paragraph or after the section that describes the diagram.
+  - If there is no suitable position, append it to the end of the document.
+  - If there is no D2 diagram source code, return the original document content without any changes.
+input_schema:
+  type: object
+  properties:
+    content:
+      type: string
+      description: Source content of the document
+    d2DiagramSourceCode:
+      type: string
+      description: Source content of D2 Diagram
+  required:
+    - content
+    - d2DiagramSourceCode
+output_schema:
+  type: object
+  properties:
+    content:
+      type: string
+      description: Merged content of the document with D2 diagram
+  required:
+    - content
+

package/agents/init/index.mjs

@@ -76,7 +76,7 @@ export default async function init(
 
   // 1. Primary purpose - what's the main outcome you want readers to achieve?
   const purposeChoices = await options.prompts.checkbox({
-    message: "📝 [1/8]: What should your documentation help readers achieve?",
+    message: "📝 [1/9]: What should your documentation help readers achieve?",
     choices: Object.entries(DOCUMENT_STYLES)
       .filter(([key]) => key !== "custom") // Remove custom option for multiselect
       .map(([key, style]) => ({
@@ -124,7 +124,7 @@ export default async function init(
 
   // 2. Target audience - who will be reading this most often?
   const audienceChoices = await options.prompts.checkbox({
-    message: "👥 [2/8]: Who will be reading your documentation?",
+    message: "👥 [2/9]: Who will be reading your documentation?",
     choices: Object.entries(TARGET_AUDIENCES)
       .filter(([key]) => key !== "custom") // Remove custom option for multiselect
       .map(([key, audience]) => ({
@@ -143,7 +143,15 @@ export default async function init(
   // Save target audience choices as keys
   input.targetAudienceTypes = audienceChoices;
 
-  // 3. Reader knowledge level - what do readers typically know when they arrive?
+  // 3. Custom rules - any specific requirements for the documentation?
+  const rulesInput = await options.prompts.input({
+    message:
+      "📋 [3/9]: Any custom rules or requirements for your documentation? (Optional, press Enter to skip)",
+    default: "",
+  });
+  input.rules = rulesInput.trim();
+
+  // 4. Reader knowledge level - what do readers typically know when they arrive?
   // Determine default based on selected purposes using mapping
   const mappedPurpose = prioritizedPurposes.find(
     (purpose) => PURPOSE_TO_KNOWLEDGE_MAPPING[purpose],
@@ -158,7 +166,7 @@ export default async function init(
   );
 
   const knowledgeChoice = await options.prompts.select({
-    message: "🧠 [3/8]: How much do readers already know about your project?",
+    message: "🧠 [4/9]: How much do readers already know about your project?",
     choices: Object.entries(filteredKnowledgeOptions).map(([key, level]) => ({
       name: `${level.name}`,
       description: level.description,
@@ -203,7 +211,7 @@ export default async function init(
   );
 
   const depthChoice = await options.prompts.select({
-    message: "📊 [4/8]: How detailed should your documentation be?",
+    message: "📊 [5/9]: How detailed should your documentation be?",
     choices: Object.entries(filteredDepthOptions).map(([key, depth]) => ({
       name: `${depth.name}`,
       description: depth.description,
@@ -221,7 +229,7 @@ export default async function init(
 
   // Let user select primary language from supported list
   const primaryLanguageChoice = await options.prompts.select({
-    message: "🌐 [5/8]: What's your main documentation language?",
+    message: "🌐 [6/9]: What's your main documentation language?",
     choices: SUPPORTED_LANGUAGES.map((lang) => ({
       name: `${lang.label} - ${lang.sample}`,
       value: lang.code,
@@ -238,7 +246,7 @@ export default async function init(
   );
 
   const translateLanguageChoices = await options.prompts.checkbox({
-    message: "🔄 [6/8]: Which languages should we translate to?",
+    message: "🔄 [7/9]: Which languages should we translate to?",
     choices: availableTranslationLanguages.map((lang) => ({
       name: `${lang.label} - ${lang.sample}`,
       value: lang.code,
@@ -249,13 +257,13 @@ export default async function init(
 
   // 7. Documentation directory
   const docsDirInput = await options.prompts.input({
-    message: `📁 [7/8]: Where should we save your documentation?`,
+    message: `📁 [8/9]: Where should we save your documentation?`,
     default: `${outputPath}/docs`,
   });
   input.docsDir = docsDirInput.trim() || `${outputPath}/docs`;
 
   // 8. Content sources
-  console.log("\n🔍 [8/8]: Content Sources");
+  console.log("\n🔍 [9/9]: Content Sources");
   console.log(
     "What folders/files should we analyze for documentation? (e.g., ./src, ./docs, ./README.md)",
   );
@@ -400,7 +408,7 @@ export function generateYAML(input) {
     documentationDepth: input.documentationDepth || "",
 
     // Custom rules and target audience (empty for user to fill)
-    rules: "",
+    rules: input.rules || "",
     targetAudience: "",
 
     // Language settings

package/agents/publish/publish-docs.mjs

@@ -5,13 +5,17 @@ import chalk from "chalk";
 import fs from "fs-extra";
 
 import { getAccessToken } from "../../utils/auth-utils.mjs";
-import { DISCUSS_KIT_STORE_URL, TMP_DIR, TMP_DOCS_DIR } from "../../utils/constants/index.mjs";
+import {
+  DEFAULT_APP_URL,
+  DISCUSS_KIT_STORE_URL,
+  DOC_SMITH_DIR,
+  TMP_DIR,
+  TMP_DOCS_DIR,
+} from "../../utils/constants/index.mjs";
 import { beforePublishHook, ensureTmpDir } from "../../utils/d2-utils.mjs";
 import { deploy } from "../../utils/deploy.mjs";
 import { getGithubRepoUrl, loadConfigFromFile, saveValueToConfig } from "../../utils/utils.mjs";
 
-const DEFAULT_APP_URL = "https://docsmith.aigne.io";
-
 export default async function publishDocs(
   { docsDir: rawDocsDir, appUrl, boardId, projectName, projectDesc, projectLogo },
   options,
@@ -19,9 +23,7 @@ export default async function publishDocs(
   // move work dir to tmp-dir
   await ensureTmpDir();
 
-  const hasDocSmithBaseUrl = !!process.env.DOC_SMITH_BASE_URL;
-
-  const docsDir = join(".aigne", "doc-smith", TMP_DIR, TMP_DOCS_DIR);
+  const docsDir = join(DOC_SMITH_DIR, TMP_DIR, TMP_DOCS_DIR);
   await fs.rm(docsDir, { recursive: true, force: true });
   await fs.mkdir(docsDir, {
     recursive: true,
@@ -61,22 +63,18 @@ export default async function publishDocs(
           name: `${chalk.blue("Your existing website")} - Integrate and publish directly on your current site (setup required)`,
           value: "custom",
         },
-        ...(hasCachedCheckoutId && hasDocSmithBaseUrl
+        ...(hasCachedCheckoutId
           ? [
               {
-                name: `${chalk.yellow("Resume previous website setup")} - ${chalk.green("Already paid.")} Continue where you left off. Your payment is already processed.`,
+                name: `${chalk.yellow("Resume previous website setup")} - ${chalk.green("Already paid.")} Continue where you left off. Your payment has already been processed.`,
                 value: "new-instance-continue",
               },
             ]
           : []),
-        ...(hasDocSmithBaseUrl
-          ? [
-              {
-                name: `${chalk.blue("New website")} - ${chalk.yellow("Paid service.")} We'll help you set up a brand-new website with custom domain and hosting. Great if you want a professional presence.`,
-                value: "new-instance",
-              },
-            ]
-          : []),
+        {
+          name: `${chalk.blue("New website")} - ${chalk.yellow("Paid service.")} We'll help you set up a brand-new website with custom domain and hosting. Great if you want a professional presence.`,
+          value: "new-instance",
+        },
       ],
     });
 
@@ -100,7 +98,7 @@ export default async function publishDocs(
       });
       // Ensure appUrl has protocol
       appUrl = userInput.includes("://") ? userInput : `https://${userInput}`;
-    } else if (hasDocSmithBaseUrl && ["new-instance", "new-instance-continue"].includes(choice)) {
+    } else if (["new-instance", "new-instance-continue"].includes(choice)) {
       // Deploy a new Discuss Kit service
       try {
         let id = "";
@@ -118,8 +116,7 @@ export default async function publishDocs(
         token = ltToken;
       } catch (error) {
         const errorMsg = error?.message || "Unknown error occurred";
-        console.error(`${chalk.red("❌ Failed to publish to website:")} ${errorMsg}`);
-        return { message: `❌ Publish failed: ${errorMsg}` };
+        return { message: `${chalk.red("❌ Failed to publish to website:")} ${errorMsg}` };
       }
     }
   }
@@ -168,7 +165,7 @@ export default async function publishDocs(
       boardDesc: projectInfo.description,
       boardCover: projectInfo.icon,
       mediaFolder: rawDocsDir,
-      cacheFilePath: join(".aigne", "doc-smith", "upload-cache.yaml"),
+      cacheFilePath: join(DOC_SMITH_DIR, "upload-cache.yaml"),
       boardMeta,
     });
 

package/agents/update/generate-document.yaml

@@ -48,3 +48,28 @@ output_schema:
       type: string
   required:
     - content
+skills:
+  - type: team
+    task_render_mode: collapse
+    name: generateD2DiagramContent
+    skills:
+      - ../generate/generate-d2-diagram.yaml
+    reflection:
+      reviewer: ../generate/check-d2-diagram-valid.mjs
+      is_approved: isValid
+      max_iterations: 5
+      return_last_on_max_iterations: true
+    input_schema:
+      type: object
+      properties:
+        documentContent:
+          type: string
+          description: Source code of current document (without the D2 diagram)
+      required:
+        - documentContent
+    output_schema:
+      type: object
+      properties:
+        d2DiagramSourceCode:
+          type: string
+          description: Source code for the D2 diagram

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aigne/doc-smith",
-  "version": "0.8.11-beta.4",
+  "version": "0.8.11-beta.5",
   "description": "AI-driven documentation generation tool built on the AIGNE Framework",
   "publishConfig": {
     "access": "public"
@@ -19,7 +19,7 @@
     "@aigne/gemini": "^0.14.0",
     "@aigne/openai": "^0.16.0",
     "@aigne/publish-docs": "^0.11.0",
-    "@blocklet/payment-broker-client": "^1.20.20",
+    "@blocklet/payment-broker-client": "^1.21.5",
     "@terrastruct/d2": "^0.1.33",
     "chalk": "^5.5.0",
     "cli-highlight": "^2.1.11",

package/prompts/detail/{d2-chart/rules.md → d2-diagram/rules-system.md}

@@ -1,12 +1,51 @@
-<d2_generate_constraints>
 # D2 Diagram Generation Expert Guide
 
 ## Preamble: LLM Role and Core Objective
 
-You are an expert software architect and a master of the D2 (Declarative Diagramming) language. Your primary function is to translate abstract descriptions of software systems, components, and processes into precise, readable, and visually effective D2 diagram code.
+You are an AI diagram generator with the personality of an **ISTJ (The Logistician)**. 
+You are reliable, rule-abiding, and methodical. Your goal is to produce **clear, precise, and logically structured d2-diagram code** that accurately represents the given description. 
 
 Your core directive is to produce D2 code that is not only syntactically correct but also semantically meaningful and adheres to the highest standards of technical diagramming. The generated output must follow all instructions, constraints, and best practices detailed in this document. You will operate in a zero-tolerance mode for syntactical errors, especially concerning predefined keyword values. The fundamental principle is the separation of concerns: the logical structure of the diagram must be defined independently of its visual styling. The following chapters are structured to enforce this principle.
 
+You value **order, consistency, and factual accuracy** over abstract or decorative styles. 
+Your diagrams should focus on **readability, structural correctness, and practical use in technical documentation**.
+
+**ISTJ-style guiding principles for your diagram generation:**
+
+1. **Fact-Driven and Accurate:**  
+   - Adhere strictly to the provided description and rules.  
+   - Do not assume or add elements that are not explicitly described.  
+
+2. **Structured and Orderly:**  
+   - Organize diagram elements in a logical, hierarchical order (e.g., top-down for processes, left-to-right for data flows).  
+   - Group related nodes consistently.  
+   - Maintain clear flow and avoid unnecessary crossing lines.  
+
+3. **Clarity and Precision:**  
+   - Use simple, standard shapes and consistent node labeling.  
+   - Ensure every arrow or connection has a clear meaning (e.g., data flow, control flow).  
+   - Avoid ambiguous or decorative text.  
+
+4. **Standards and Consistency:**  
+   - Follow best practices for technical diagrams (e.g., rectangular boxes for processes, cylinders for data storage, labeled arrows for flows).  
+   - Maintain consistent spacing, alignment, and sizing in the diagram code.  
+
+5. **Practical and Maintainable:**  
+   - Ensure the generated d2 code is concise, easy to edit, and reproducible.  
+   - Provide comments in the d2 code (if necessary) to clarify sections or complex relationships.  
+   - Avoid unnecessary stylistic complexity that may hinder future maintenance.
+
+**Output Requirements:**
+- Output only valid d2-diagram code.
+- Do not include explanatory text outside of the code block.
+- Ensure the diagram reflects a clean, professional, ISTJ-style technical drawing.
+- output must be wrap with 
+```md
+\`\`\`d2\n...\n\`\`\`
+```
+
+
+
 ## Chapter 1: Core Instructions for D2 Diagram Generation
 
 This chapter establishes the foundational rules for generating the structure and logic of a D2 diagram. It prioritizes semantic correctness and adherence to diagramming principles over aesthetic concerns, which are addressed in Chapter 2.
@@ -1090,6 +1129,3 @@ Session.t2 -> Session.t2: "Update and remove status effects"
 Session.t2 -> Lua: "Trigger OnPlayerTurn"
 User.t2 <- Session.t2
 ```
-
-
-</d2_generate_constraints>

package/prompts/detail/d2-diagram/rules-user.md

@@ -0,0 +1,4 @@
+Follow the given rules and ISTJ style from your system instructions.
+
+Generate a d2-diagram that represents the following document content:
+{{documentContent}}

package/prompts/detail/document-rules.md

@@ -19,9 +19,8 @@ Documentation Generation Rules:
 - When describing multiple properties of the same object, wrap the outermost `<x-field>` elements with `<x-field-group>` elements. Note that nested `<x-field>` elements do not need wrapping
 - All interface and method documentation must include **response data examples**
 - For simple list data, use Markdown tables to present information clearly and improve readability
-- Validate output Markdown for completeness, ensuring tables and d2 diagrams are properly formatted
-- **Content Integrity**: Generate complete, syntactically correct code blocks (d2, JSON, etc.). Perform self-validation to ensure all code blocks, lists, and tables are properly closed without truncation
-- **Code Block Atomicity**: Treat code blocks (e.g., ```d2 ... ```) as indivisible units. Generate them completely from opening marker (```d2) to closing marker (```) without interruption
+- Validate output Markdown for completeness, ensuring tables are properly formatted
+- **Content Integrity**: Generate complete, syntactically correct code blocks (JSON, etc.). Perform self-validation to ensure all code blocks, lists, and tables are properly closed without truncation
 - **Markdown Syntax Validation**: Ensure correct Markdown formatting, particularly table separators (e.g., `|---|---|---|`) that match column counts
 - Use README files for reference only—extract the most current and comprehensive information directly from source code
 - Omit tag information from document headers as it's processed programmatically

package/prompts/detail/generate-document.md

@@ -37,8 +37,14 @@ Custom component generation rules:
 Custom code block generation rules:
 {% include "custom/custom-code-block.md" %}
 
-D2 Diagram Generation Expert Guide:
-{% include "d2-chart/rules.md" %}
+Diagram generate guide:
+Evaluate for each document whether diagrams are necessary.
+- Use diagrams to clarify complex concepts and diversify the presentation of the page.
+- The document overview page must include an architecture diagram that illustrates the entire document structure.
+- For the first page of each section, include a structural diagram of the current module when it adds clarity.
+- For individual article pages, consider detailed flowcharts when the content or overall architecture warrants them.
+- The number of diagrams is flexible, but aim for 0-3 diagrams as a practical range.
+
 </content_generation_rules>
 
 {% if glossary %}

package/prompts/detail/update-document.md

@@ -70,8 +70,6 @@ Custom component optimization rules:
 Custom code block optimization rules:
 {% include "custom/custom-code-block.md" %}
 
-D2 Diagram optimization rules:
-{% include "d2-chart/rules.md" %}
 </content_optimization_rules>
 
 {% if glossary %}

package/tests/agents/init/init.test.mjs

@@ -986,11 +986,12 @@ describe("init", () => {
         const mockResponses = {
           checkbox_1: ["getStarted", "findAnswers"], // Document purpose
           checkbox_2: ["developers"], // Target audience
-          select_3: "domainFamiliar", // Reader knowledge level
-          select_4: "balancedCoverage", // Documentation depth
-          select_5: "en", // Primary language
-          checkbox_6: ["zh", "ja"], // Translation languages
-          input_7: join(tempDir, "docs"), // Documentation directory
+          input_3: "Custom rules for documentation", // Custom rules
+          select_4: "domainFamiliar", // Reader knowledge level
+          select_5: "balancedCoverage", // Documentation depth
+          select_6: "en", // Primary language
+          checkbox_7: ["zh", "ja"], // Translation languages
+          input_8: join(tempDir, "docs"), // Documentation directory
           search: "", // Source paths (empty to finish)
         };
 
@@ -1023,6 +1024,7 @@ describe("init", () => {
 
         expect(config.documentPurpose).toEqual(["getStarted", "findAnswers"]);
         expect(config.targetAudienceTypes).toEqual(["developers"]);
+        expect(config.rules).toBe("Custom rules for documentation");
         expect(config.readerKnowledgeLevel).toBe("domainFamiliar");
         expect(config.documentationDepth).toBe("balancedCoverage");
         expect(config.locale).toBe("en");
@@ -1042,11 +1044,12 @@ describe("init", () => {
           checkbox_1: ["mixedPurpose"], // Document purpose - triggers follow-up
           checkbox: ["completeTasks", "findAnswers"], // Top priorities after mixedPurpose
           checkbox_2: ["developers", "devops"], // Target audience
-          select_3: "experiencedUsers", // Reader knowledge level
-          select_4: "comprehensive", // Documentation depth
-          select_5: "zh-CN", // Primary language
-          checkbox_6: ["en"], // Translation languages
-          input_7: join(tempDir, "documentation"), // Documentation directory
+          input_3: "Custom rules for documentation", // Custom rules
+          select_4: "experiencedUsers", // Reader knowledge level
+          select_5: "comprehensive", // Documentation depth
+          select_6: "zh-CN", // Primary language
+          checkbox_7: ["en"], // Translation languages
+          input_8: join(tempDir, "documentation"), // Documentation directory
           search: "", // Source paths (empty to finish)
         };
 
@@ -1071,6 +1074,7 @@ describe("init", () => {
 
         expect(config.documentPurpose).toEqual(["completeTasks", "findAnswers"]);
         expect(config.targetAudienceTypes).toEqual(["developers", "devops"]);
+        expect(config.rules).toBe("Custom rules for documentation");
         expect(config.readerKnowledgeLevel).toBe("experiencedUsers");
         expect(config.documentationDepth).toBe("comprehensive");
         expect(config.locale).toBe("zh-CN");
@@ -1087,11 +1091,12 @@ describe("init", () => {
         const mockResponses = {
           checkbox_1: ["getStarted"], // Document purpose
           checkbox_2: ["endUsers"], // Target audience
-          select_3: "completeBeginners", // Reader knowledge level
-          select_4: "essentialOnly", // Documentation depth
-          select_5: "en", // Primary language
-          checkbox_6: [], // No translation languages
-          input_7: join(tempDir, "simple-docs"), // Documentation directory
+          input_3: "Custom rules for documentation", // Custom rules
+          select_4: "completeBeginners", // Reader knowledge level
+          select_5: "essentialOnly", // Documentation depth
+          select_6: "en", // Primary language
+          checkbox_7: [], // No translation languages
+          input_8: join(tempDir, "simple-docs"), // Documentation directory
           search: "", // Source paths (empty to finish)
         };
 
@@ -1115,6 +1120,7 @@ describe("init", () => {
 
         expect(config.documentPurpose).toEqual(["getStarted"]);
         expect(config.targetAudienceTypes).toEqual(["endUsers"]);
+        expect(config.rules).toBe("Custom rules for documentation");
         expect(config.readerKnowledgeLevel).toBe("completeBeginners");
         expect(config.documentationDepth).toBe("essentialOnly");
         expect(config.locale).toBe("en");
@@ -1298,7 +1304,7 @@ describe("init", () => {
         let validateCalled = false;
         const mockPrompts = {
           checkbox: (options) => {
-            if (options.message.includes("[1/8]") && options.validate) {
+            if (options.message.includes("[1/9]") && options.validate) {
               // Test the validation function directly
               const validationResult = options.validate([]);
               expect(validationResult).toBe(
@@ -1336,10 +1342,10 @@ describe("init", () => {
         let audienceValidateCalled = false;
         const mockPrompts = {
           checkbox: (options) => {
-            if (options.message.includes("[1/8]")) {
+            if (options.message.includes("[1/9]")) {
               return Promise.resolve(["getStarted"]); // Valid document purpose
             }
-            if (options.message.includes("[2/8]") && options.validate) {
+            if (options.message.includes("[2/9]") && options.validate) {
               // Test the validation function for target audience
               const validationResult = options.validate([]);
               expect(validationResult).toBe("Please choose at least one audience.");
@@ -1374,7 +1380,7 @@ describe("init", () => {
         let priorityValidateCalled = false;
         const mockPrompts = {
           checkbox: (options) => {
-            if (options.message.includes("[1/8]")) {
+            if (options.message.includes("[1/9]")) {
               return Promise.resolve(["mixedPurpose"]); // Trigger follow-up question
             }
             // This is the follow-up priority selection

package/tests/agents/publish/publish-docs.test.mjs

@@ -15,6 +15,7 @@ import publishDocs from "../../../agents/publish/publish-docs.mjs";
 import * as authUtils from "../../../utils/auth-utils.mjs";
 import * as d2Utils from "../../../utils/d2-utils.mjs";
 import * as utils from "../../../utils/utils.mjs";
+import * as deployUtils from "../../../utils/deploy.mjs";
 
 // Mock all external dependencies
 const mockPublishDocs = {
@@ -51,6 +52,7 @@ describe("publish-docs", () => {
   let getGithubRepoUrlSpy;
   let loadConfigFromFileSpy;
   let saveValueToConfigSpy;
+  let deploySpy;
 
   beforeAll(() => {
     // Apply mocks for external dependencies only
@@ -98,6 +100,10 @@ describe("publish-docs", () => {
     );
     loadConfigFromFileSpy = spyOn(utils, "loadConfigFromFile").mockResolvedValue({});
     saveValueToConfigSpy = spyOn(utils, "saveValueToConfig").mockResolvedValue();
+    deploySpy = spyOn(deployUtils, "deploy").mockResolvedValue({
+      appUrl: "https://deployed.example.com",
+      token: "deploy-token",
+    });
 
     mockOptions = {
       prompts: {
@@ -125,6 +131,7 @@ describe("publish-docs", () => {
     getGithubRepoUrlSpy?.mockRestore();
     loadConfigFromFileSpy?.mockRestore();
     saveValueToConfigSpy?.mockRestore();
+    deploySpy?.mockRestore();
   });
 
   // BASIC FUNCTIONALITY TESTS
@@ -593,6 +600,98 @@ describe("publish-docs", () => {
     expect(mockOptions.prompts.select).not.toHaveBeenCalled();
   });
 
+  // RESUME PREVIOUS WEBSITE SETUP TESTS
+  test("should show resume option when checkoutId exists in config", async () => {
+    loadConfigFromFileSpy.mockResolvedValue({
+      checkoutId: "cached-checkout-123",
+    });
+    mockOptions.prompts.select.mockResolvedValue("default");
+
+    await publishDocs(
+      {
+        docsDir: "./docs",
+        appUrl: "https://docsmith.aigne.io",
+      },
+      mockOptions,
+    );
+
+    expect(mockOptions.prompts.select).toHaveBeenCalledWith(
+      expect.objectContaining({
+        message: "Select platform to publish your documents:",
+        choices: expect.arrayContaining([
+          expect.objectContaining({
+            name: expect.stringContaining("Resume previous website setup"),
+            value: "new-instance-continue",
+          }),
+        ]),
+      }),
+    );
+
+    // Verify the exact text content
+    const selectCall = mockOptions.prompts.select.mock.calls[0][0];
+    const resumeChoice = selectCall.choices.find(
+      (choice) => choice.value === "new-instance-continue",
+    );
+    expect(resumeChoice.name).toContain("Resume previous website setup");
+    expect(resumeChoice.name).toContain("Already paid.");
+    expect(resumeChoice.name).toContain(
+      "Continue where you left off. Your payment has already been processed.",
+    );
+  });
+
+  test("should not show resume option when no checkoutId in config", async () => {
+    loadConfigFromFileSpy.mockResolvedValue({});
+    mockOptions.prompts.select.mockResolvedValue("default");
+
+    await publishDocs(
+      {
+        docsDir: "./docs",
+        appUrl: "https://docsmith.aigne.io",
+      },
+      mockOptions,
+    );
+
+    expect(mockOptions.prompts.select).toHaveBeenCalledWith(
+      expect.objectContaining({
+        message: "Select platform to publish your documents:",
+        choices: expect.not.arrayContaining([
+          expect.objectContaining({
+            value: "new-instance-continue",
+          }),
+        ]),
+      }),
+    );
+  });
+
+  test("should handle resume previous website setup selection", async () => {
+    deploySpy.mockResolvedValue({
+      appUrl: "https://resumed.example.com",
+      token: "resume-token",
+    });
+
+    loadConfigFromFileSpy.mockResolvedValue({
+      checkoutId: "cached-checkout-123",
+      paymentUrl: "https://payment.example.com",
+    });
+    mockOptions.prompts.select.mockResolvedValue("new-instance-continue");
+
+    const consoleSpy = spyOn(console, "log").mockImplementation(() => {});
+
+    await publishDocs(
+      {
+        docsDir: "./docs",
+        appUrl: "https://docsmith.aigne.io",
+      },
+      mockOptions,
+    );
+
+    expect(consoleSpy).toHaveBeenCalledWith("\nResuming your previous website setup...");
+    expect(deploySpy).toHaveBeenCalledWith("cached-checkout-123", "https://payment.example.com");
+    expect(getAccessTokenSpy).toHaveBeenCalledWith("https://resumed.example.com", "resume-token");
+
+    consoleSpy.mockRestore();
+  });
+
   test("should handle URL validation edge cases", async () => {
     loadConfigFromFileSpy.mockResolvedValue({});
     mockOptions.prompts.select.mockResolvedValue("custom");

package/tests/agents/utils/check-detail-result.test.mjs

@@ -229,7 +229,7 @@ set -euo pipefail
 function deploy_app() {
     local app_name="$1"
     local version="$2"
-    
+
     echo "Deploying $app_name version $version"
     docker run --rm "$app_name:$version"
 }
@@ -337,19 +337,6 @@ This document demonstrates various programming language code blocks.`;
     });
   });
 
-  describe("D2 syntax validation", () => {
-    test("should handle D2 syntax errors", async () => {
-      const documentStructure = [];
-      const reviewContent =
-        "```d2\n" +
-        "invalid d2 syntax {{\n" + // Malformed D2
-        "```\n\n" +
-        "This has proper structure.";
-      const result = await checkDetailResult({ documentStructure, reviewContent });
-      expect(result.isApproved).toBe(false);
-    });
-  });
-
   describe("Advanced table edge cases", () => {
     test("should handle empty table cells correctly", async () => {
       const documentStructure = [];
@@ -660,7 +647,7 @@ flowchart TD
     A["1. Create Backend Implementation<br>api/src/providers/"]
     B["2. Add Backend Configuration<br>api/src/providers/models.ts"]
     C["3. Update Frontend Selector<br>src/pages/config/ai-providers/"]
-    
+
     A --> B --> C
 \`\`\`
 

package/tests/utils/auth-utils.test.mjs

@@ -336,7 +336,7 @@ describe("auth-utils", () => {
 
     // Test that openPage calls the mock open function
     await capturedOpenPage("https://auth.example.com");
-    expect(mockOpen).toHaveBeenCalledWith("https://auth.example.com/");
+    expect(mockOpen).toHaveBeenCalledWith("https://auth.example.com/?required_roles=owner%2Cadmin");
   });
 
   test("should handle authorization failure", async () => {

package/tests/utils/d2-utils.test.mjs

@@ -6,7 +6,7 @@ import path from "node:path";
 
 import Debug from "debug";
 
-import { TMP_ASSETS_DIR } from "../../utils/constants/index.mjs";
+import { DOC_SMITH_DIR, TMP_ASSETS_DIR, TMP_DIR } from "../../utils/constants/index.mjs";
 import {
   beforePublishHook,
   checkContent,
@@ -346,7 +346,7 @@ E -> F
       try {
         await checkContent({ content });
 
-        const assetDir = path.join(process.cwd(), ".aigne", "doc-smith", ".tmp", "assets", "d2");
+        const assetDir = path.join(process.cwd(), DOC_SMITH_DIR, TMP_DIR, TMP_ASSETS_DIR, "d2");
         const files = await readdir(assetDir);
         const d2File = files.find((file) => file.endsWith(".d2"));
         expect(d2File).toBeDefined();
@@ -365,7 +365,7 @@ E -> F
       try {
         await ensureTmpDir();
 
-        const tmpDir = path.join(tempDir, ".aigne", "doc-smith", ".tmp");
+        const tmpDir = path.join(tempDir, DOC_SMITH_DIR, TMP_DIR);
         const gitignorePath = path.join(tmpDir, ".gitignore");
 
         expect(existsSync(tmpDir)).toBe(true);
@@ -386,7 +386,7 @@ E -> F
         // First call
         await ensureTmpDir();
 
-        const tmpDir = path.join(tempDir, ".aigne", "doc-smith", ".tmp");
+        const tmpDir = path.join(tempDir, DOC_SMITH_DIR, TMP_DIR);
         const gitignorePath = path.join(tmpDir, ".gitignore");
 
         // Modify .gitignore to test if it gets overwritten

package/tests/utils/deploy.test.mjs

@@ -91,15 +91,8 @@ describe("deploy", () => {
 
     // Verify BrokerClient was constructed with correct config
     expect(mockBrokerClientConstructor).toHaveBeenCalledWith({
-      baseUrl: "",
       authToken: "mock-auth-token",
-      paymentLinkKey: "PAYMENT_LINK_ID",
-      timeout: 300000,
-      polling: {
-        interval: 3000,
-        maxAttempts: 100,
-        backoffStrategy: "linear",
-      },
+      baseUrl: "https://docsmith.aigne.io",
     });
 
     // Verify deploy was called with correct parameters
@@ -120,7 +113,6 @@ describe("deploy", () => {
           ACCESS_PREPARING: expect.any(Function),
           ACCESS_READY: expect.any(Function),
         }),
-        onError: expect.any(Function),
       }),
     );
 
@@ -348,7 +340,8 @@ describe("deploy", () => {
     // Verify BrokerClient was constructed with empty baseUrl
     expect(mockBrokerClientConstructor).toHaveBeenCalledWith(
       expect.objectContaining({
-        baseUrl: "",
+        authToken: "mock-auth-token",
+        baseUrl: "https://docsmith.aigne.io",
       }),
     );
   });

package/tests/utils/kroki-utils.test.mjs

@@ -6,7 +6,7 @@ import path from "node:path";
 
 import Debug from "debug";
 
-import { TMP_ASSETS_DIR } from "../../utils/constants/index.mjs";
+import { DOC_SMITH_DIR, TMP_ASSETS_DIR, TMP_DIR } from "../../utils/constants/index.mjs";
 import {
   beforePublishHook,
   checkD2Content,
@@ -451,7 +451,7 @@ E -> F
       try {
         await checkD2Content({ content });
 
-        const assetDir = path.join(process.cwd(), ".aigne", "doc-smith", ".tmp", "assets", "d2");
+        const assetDir = path.join(process.cwd(), DOC_SMITH_DIR, TMP_DIR, TMP_ASSETS_DIR, "d2");
         const files = await readdir(assetDir);
         const d2File = files.find((file) => file.endsWith(".d2"));
         expect(d2File).toBeDefined();
@@ -470,7 +470,7 @@ E -> F
       try {
         await ensureTmpDir();
 
-        const tmpDir = path.join(tempDir, ".aigne", "doc-smith", ".tmp");
+        const tmpDir = path.join(tempDir, DOC_SMITH_DIR, TMP_DIR);
         const gitignorePath = path.join(tmpDir, ".gitignore");
 
         expect(existsSync(tmpDir)).toBe(true);
@@ -491,7 +491,7 @@ E -> F
         // First call
         await ensureTmpDir();
 
-        const tmpDir = path.join(tempDir, ".aigne", "doc-smith", ".tmp");
+        const tmpDir = path.join(tempDir, DOC_SMITH_DIR, TMP_DIR);
         const gitignorePath = path.join(tmpDir, ".gitignore");
 
         // Modify .gitignore to test if it gets overwritten
@@ -632,7 +632,7 @@ E -> F
         await Promise.all(promises);
 
         // Should only create directory once
-        const tmpDir = path.join(tempDir, ".aigne", "doc-smith", ".tmp");
+        const tmpDir = path.join(tempDir, DOC_SMITH_DIR, TMP_DIR);
         expect(existsSync(tmpDir)).toBe(true);
 
         // .gitignore should be created properly

package/tests/utils/preferences-utils.test.mjs

@@ -3,6 +3,8 @@ import { existsSync } from "node:fs";
 import { mkdir, rm, writeFile } from "node:fs/promises";
 import { dirname, join } from "node:path";
 import { fileURLToPath } from "node:url";
+
+import { DOC_SMITH_DIR } from "../../utils/constants/index.mjs";
 import {
   addPreferenceRule,
   deactivateRule,
@@ -44,7 +46,7 @@ describe("preferences-utils", () => {
 
     test("should read existing preferences file", async () => {
       // Create preferences directory and file
-      const prefsDir = join(testDir, ".aigne", "doc-smith");
+      const prefsDir = join(testDir, DOC_SMITH_DIR);
       await mkdir(prefsDir, { recursive: true });
 
       await writeFile(
@@ -69,7 +71,7 @@ describe("preferences-utils", () => {
 
     test("should handle malformed YAML gracefully", async () => {
       // Create preferences directory and invalid file
-      const prefsDir = join(testDir, ".aigne", "doc-smith");
+      const prefsDir = join(testDir, DOC_SMITH_DIR);
       await mkdir(prefsDir, { recursive: true });
 
       await writeFile(join(prefsDir, "preferences.yml"), "invalid: yaml: content: [", "utf8");
@@ -85,7 +87,7 @@ describe("preferences-utils", () => {
 
       writePreferences(testPreferences);
 
-      const prefsDir = join(testDir, ".aigne", "doc-smith");
+      const prefsDir = join(testDir, DOC_SMITH_DIR);
       expect(existsSync(prefsDir)).toBe(true);
       expect(existsSync(join(prefsDir, "preferences.yml"))).toBe(true);
     });

package/tests/utils/save-value-to-config.test.mjs

@@ -5,6 +5,8 @@ import fsPromisesDefault, * as fsPromises from "node:fs/promises";
 import { mkdir, readFile, rm, writeFile } from "node:fs/promises";
 import { dirname, join } from "node:path";
 import { fileURLToPath } from "node:url";
+
+import { DOC_SMITH_DIR } from "../../utils/constants/index.mjs";
 import { saveValueToConfig } from "../../utils/utils.mjs";
 
 const __filename = fileURLToPath(import.meta.url);
@@ -12,7 +14,7 @@ const __dirname = dirname(__filename);
 
 // Test directory for isolated testing
 const TEST_DIR = join(__dirname, "temp-config-test");
-const TEST_CONFIG_DIR = join(TEST_DIR, ".aigne", "doc-smith");
+const TEST_CONFIG_DIR = join(TEST_DIR, DOC_SMITH_DIR);
 const TEST_CONFIG_PATH = join(TEST_CONFIG_DIR, "config.yaml");
 
 // Store original working directory

package/utils/auth-utils.mjs

@@ -15,6 +15,7 @@ import {
 } from "./blocklet.mjs";
 import {
   BLOCKLET_ADD_COMPONENT_DOCS,
+  DEFAULT_APP_URL,
   DISCUSS_KIT_DID,
   DISCUSS_KIT_STORE_URL,
   DOC_OFFICIAL_ACCESS_TOKEN,
@@ -97,6 +98,9 @@ export async function getAccessToken(appUrl, ltToken = "") {
       appLogo: "https://docsmith.aigne.io/image-bin/uploads/9645caf64b4232699982c4d940b03b90.svg",
       openPage: (pageUrl) => {
         const url = new URL(pageUrl);
+        if (url.hostname !== DEFAULT_APP_URL) {
+          url.searchParams.set("required_roles", "owner,admin");
+        }
         if (ltToken) {
           url.searchParams.set("__lt", ltToken);
         }

package/utils/constants/index.mjs

@@ -334,6 +334,9 @@ export const PAYMENT_KIT_DID = "z2qaCNvKMv5GjouKdcDWexv6WqtHbpNPQDnAk";
 
 export const DOC_OFFICIAL_ACCESS_TOKEN = "DOC_OFFICIAL_ACCESS_TOKEN";
 
+// Default application URL for the document deployment website.
+export const DEFAULT_APP_URL = "https://docsmith.aigne.io";
+
 // Discuss Kit related URLs
 export const DISCUSS_KIT_STORE_URL =
   "https://store.blocklet.dev/blocklets/z8ia1WEiBZ7hxURf6LwH21Wpg99vophFwSJdu";

package/utils/d2-utils.mjs

@@ -17,6 +17,8 @@ import { debug } from "./debug.mjs";
 import { iconMap } from "./icon-map.mjs";
 import { getContentHash } from "./utils.mjs";
 
+const codeBlockRegex = /```d2.*\n([\s\S]*?)```/g;
+
 export async function getChart({ content, strict }) {
   const d2 = new D2();
   const iconUrlList = Object.keys(iconMap);
@@ -60,7 +62,7 @@ export async function getChart({ content, strict }) {
   } catch (err) {
     if (strict) throw err;
 
-    console.error("Failed to generate D2 chart. Content:", content, "Error:", err);
+    console.error("Failed to generate D2 diagram. Content:", content, "Error:", err);
     return null;
   } finally {
     d2.worker.terminate();
@@ -73,8 +75,6 @@ export async function saveAssets({ markdown, docsDir }) {
     return markdown;
   }
 
-  const codeBlockRegex = /```d2.*\n([\s\S]*?)```/g;
-
   const { replaced } = await runIterator({
     input: markdown,
     regexp: codeBlockRegex,
@@ -90,7 +90,7 @@ export async function saveAssets({ markdown, docsDir }) {
         debug("Found assets cache, skipping generation", svgPath);
       } else {
         try {
-          debug("start generate d2 chart", svgPath);
+          debug("start generate d2 diagram", svgPath);
           if (debug.enabled) {
             const d2FileName = `${getContentHash(d2Content)}.d2`;
             const d2Path = path.join(assetDir, d2FileName);
@@ -102,7 +102,7 @@ export async function saveAssets({ markdown, docsDir }) {
             await fs.writeFile(svgPath, svg, { encoding: "utf8" });
           }
         } catch (error) {
-          debug("Failed to generate D2 chart. Content:", d2Content, "Error:", error);
+          debug("Failed to generate D2 diagram. Content:", d2Content, "Error:", error);
           return _code;
         }
       }
@@ -156,7 +156,12 @@ async function runIterator({ input, regexp, fn = () => {}, options, replace = fa
   };
 }
 
-export async function checkContent({ content }) {
+export async function checkContent({ content: _content }) {
+  const matches = Array.from(_content.matchAll(codeBlockRegex));
+  let content = _content;
+  if (matches.length > 0) {
+    content = matches[0][1];
+  }
   await ensureTmpDir();
   const assetDir = path.join(DOC_SMITH_DIR, TMP_DIR, TMP_ASSETS_DIR, "d2");
   await fs.ensureDir(assetDir);

package/utils/deploy.mjs

@@ -2,10 +2,11 @@ import { BrokerClient, STEPS } from "@blocklet/payment-broker-client/node";
 import chalk from "chalk";
 import open from "open";
 import { getOfficialAccessToken } from "./auth-utils.mjs";
+import { DEFAULT_APP_URL } from "./constants/index.mjs";
 import { saveValueToConfig } from "./utils.mjs";
 
 // ==================== Configuration ====================
-const BASE_URL = process.env.DOC_SMITH_BASE_URL || "";
+const BASE_URL = process.env.DOC_SMITH_BASE_URL || DEFAULT_APP_URL;
 const SUCCESS_MESSAGE = {
   en: "Congratulations! Your website has been successfully installed. You can return to the command-line tool to continue the next steps.",
   zh: "恭喜您，你的网站已安装成功！可以返回命令行工具继续后续操作！",
@@ -24,26 +25,14 @@ export async function deploy(id, cachedUrl) {
     throw new Error("Failed to get official access token");
   }
 
-  const client = new BrokerClient({
-    baseUrl: BASE_URL,
-    authToken,
-    paymentLinkKey: "PAYMENT_LINK_ID",
-    timeout: 300000,
-    polling: {
-      interval: 3000,
-      maxAttempts: 100,
-      backoffStrategy: "linear",
-    },
-  });
+  const client = new BrokerClient({ baseUrl: BASE_URL, authToken });
 
   console.log(`🚀 Starting deployment...`);
 
   const result = await client.deploy({
     cachedCheckoutId: id,
     cachedPaymentUrl: cachedUrl,
-    pageInfo: {
-      successMessage: SUCCESS_MESSAGE,
-    },
+    pageInfo: { successMessage: SUCCESS_MESSAGE },
     hooks: {
       [STEPS.PAYMENT_PENDING]: async ({ sessionId, paymentUrl, isResuming }) => {
         console.log(`⏳ Step 1/4: Waiting for payment...`);
@@ -86,11 +75,6 @@ export async function deploy(id, cachedUrl) {
         }
       },
     },
-
-    onError: (error, step) => {
-      console.error(`${chalk.red("❌")} Deployment failed at ${step || "unknown step"}:`);
-      console.error(`   ${error.message}`);
-    },
   });
 
   const { appUrl, homeUrl, subscriptionUrl, dashboardUrl, vendors } = result;

package/utils/kroki-utils.mjs

@@ -8,6 +8,7 @@ import { joinURL } from "ufo";
 
 import {
   D2_CONFIG,
+  DOC_SMITH_DIR,
   FILE_CONCURRENCY,
   KROKI_CONCURRENCY,
   TMP_ASSETS_DIR,
@@ -76,7 +77,7 @@ export async function saveD2Assets({ markdown, docsDir }) {
         debug("Found assets cache, skipping generation", svgPath);
       } else {
         try {
-          debug("start generate d2 chart", svgPath);
+          debug("Start generate d2 diagram", svgPath);
           if (debug.enabled) {
             const d2FileName = `${getContentHash(d2Content)}.d2`;
             const d2Path = path.join(assetDir, d2FileName);
@@ -88,7 +89,7 @@ export async function saveD2Assets({ markdown, docsDir }) {
             await fs.writeFile(svgPath, svg, { encoding: "utf8" });
           }
         } catch (error) {
-          debug("Failed to generate D2 chart:", error);
+          debug("Failed to generate D2 diagram:", error);
           return _code;
         }
       }
@@ -144,7 +145,7 @@ async function runIterator({ input, regexp, fn = () => {}, options, replace = fa
 
 export async function checkD2Content({ content }) {
   await ensureTmpDir();
-  const assetDir = path.join(".aigne", "doc-smith", TMP_DIR, TMP_ASSETS_DIR, "d2");
+  const assetDir = path.join(DOC_SMITH_DIR, TMP_DIR, TMP_ASSETS_DIR, "d2");
   await fs.ensureDir(assetDir);
   const d2Content = [D2_CONFIG, content].join("\n");
   const fileName = `${getContentHash(d2Content)}.svg`;
@@ -166,7 +167,7 @@ export async function checkD2Content({ content }) {
 }
 
 export async function ensureTmpDir() {
-  const tmpDir = path.join(".aigne", "doc-smith", TMP_DIR);
+  const tmpDir = path.join(DOC_SMITH_DIR, TMP_DIR);
   if (!(await fs.pathExists(path.join(tmpDir, ".gitignore")))) {
     await fs.ensureDir(tmpDir);
     await fs.writeFile(path.join(tmpDir, ".gitignore"), "**/*", { encoding: "utf8" });

package/utils/markdown-checker.mjs

@@ -1,14 +1,11 @@
 import fs from "node:fs";
 import path from "node:path";
-import pMap from "p-map";
 import remarkGfm from "remark-gfm";
 import remarkLint from "remark-lint";
 import remarkParse from "remark-parse";
 import { unified } from "unified";
 import { visit } from "unist-util-visit";
 import { VFile } from "vfile";
-import { KROKI_CONCURRENCY } from "./constants/index.mjs";
-import { checkContent, isValidCode } from "./d2-utils.mjs";
 import { validateMermaidSyntax } from "./mermaid-validator.mjs";
 
 /**
@@ -378,7 +375,6 @@ export async function checkMarkdown(markdown, source = "content", options = {})
 
     // Check mermaid code blocks and other custom validations
     const mermaidChecks = [];
-    const d2ChecksList = [];
     visit(ast, "code", (node) => {
       if (node.lang) {
         const line = node.position?.start?.line || "unknown";
@@ -467,13 +463,6 @@ export async function checkMarkdown(markdown, source = "content", options = {})
             specialCharMatch = nodeWithSpecialCharsRegex.exec(mermaidContent);
           }
         }
-        if (isValidCode(node.lang)) {
-          d2ChecksList.push({
-            content: node.value,
-            line,
-          });
-        }
-        // TODO: @zhanghan need to check correctness of every code language
       }
     });
 
@@ -524,15 +513,6 @@ export async function checkMarkdown(markdown, source = "content", options = {})
 
     // Wait for all mermaid checks to complete
     await Promise.all(mermaidChecks);
-    await pMap(
-      d2ChecksList,
-      async ({ content, line }) =>
-        checkContent({ content }).catch((err) => {
-          const errorMessage = err?.message || String(err) || "Unknown d2 syntax error";
-          errorMessages.push(`Found D2 syntax error in ${source} at line ${line}: ${errorMessage}`);
-        }),
-      { concurrency: KROKI_CONCURRENCY },
-    );
 
     // Run markdown linting rules
     await processor.run(ast, file);