@aigne/doc-smith 0.8.12-beta.9 → 0.8.13-beta

package/agents/init/index.mjs

@@ -21,14 +21,13 @@ import {
   validatePath,
 } from "../../utils/utils.mjs";
 
-// UI constants
 const _PRESS_ENTER_TO_FINISH = "Press Enter to finish";
 
 /**
- * Guide users through multi-turn dialogue to collect information and generate YAML configuration
+ * Guides the user through a multi-turn dialog to generate a YAML configuration file.
  * @param {Object} params
- * @param {string} params.outputPath - Output file path
- * @param {string} params.fileName - File name
+ * @param {string} params.outputPath - The path to the output file.
+ * @param {string} params.fileName - The name of the file.
  * @returns {Promise<Object>}
  */
 export default async function init(
@@ -47,7 +46,7 @@ export default async function init(
     const configContent = await readFile(filePath, "utf8").catch(() => null);
 
     if (!configContent || configContent.trim() === "") {
-      console.log(`⚠️ No configuration found.`);
+      console.log("⚠️ No configuration file found.");
       console.log(
         `🚀 Run ${chalk.cyan("aigne doc init")} to set up your documentation configuration.`,
       );
@@ -71,14 +70,12 @@ export default async function init(
   console.log("🚀 Welcome to AIGNE DocSmith!");
   console.log("Let's set up your documentation preferences.\n");
 
-  // Collect user information
   const input = {};
 
-  // 1. Primary purpose - what's the main outcome you want readers to achieve?
   const purposeChoices = await options.prompts.checkbox({
     message: "📝 [1/9]: What should your documentation help readers achieve?",
     choices: Object.entries(DOCUMENT_STYLES)
-      .filter(([key]) => key !== "custom") // Remove custom option for multiselect
+      .filter(([key]) => key !== "custom")
       .map(([key, style]) => ({
         name: `${style.name}`,
         description: style.description,
@@ -86,19 +83,18 @@ export default async function init(
       })),
     validate: (input) => {
       if (input.length === 0) {
-        return "Please choose at least one goal for your documentation.";
+        return "You must choose at least one goal for your documentation.";
       }
       return true;
     },
   });
 
-  // Follow-up logic: If ONLY mixedPurpose selected, ask for priority ranking
   let prioritizedPurposes = purposeChoices;
   if (purposeChoices.length === 1 && purposeChoices.includes("mixedPurpose")) {
     const topPriorities = await options.prompts.checkbox({
-      message: "🎯 Which is most important? (Select top 2 priorities)",
+      message: "🎯 Which is most important? (Select up to 2 priorities)",
       choices: Object.entries(DOCUMENT_STYLES)
-        .filter(([key]) => key !== "custom" && key !== "mixedPurpose") // Filter out custom and mixedPurpose
+        .filter(([key]) => key !== "custom" && key !== "mixedPurpose")
         .map(([key, style]) => ({
           name: `${style.name}`,
           description: style.description,
@@ -106,27 +102,24 @@ export default async function init(
         })),
       validate: (input) => {
         if (input.length === 0) {
-          return "Please choose at least one priority.";
+          return "You must choose at least one priority.";
         }
         if (input.length > 2) {
-          return "Please choose maximum 2 priorities.";
+          return "You may only choose up to 2 priorities.";
         }
         return true;
       },
     });
 
-    // Replace mixedPurpose with selected priorities
     prioritizedPurposes = topPriorities;
   }
 
-  // Save document purpose choices as keys
   input.documentPurpose = prioritizedPurposes;
 
-  // 2. Target audience - who will be reading this most often?
   const audienceChoices = await options.prompts.checkbox({
     message: "👥 [2/9]: Who will be reading your documentation?",
     choices: Object.entries(TARGET_AUDIENCES)
-      .filter(([key]) => key !== "custom") // Remove custom option for multiselect
+      .filter(([key]) => key !== "custom")
       .map(([key, audience]) => ({
         name: `${audience.name}`,
         description: audience.description,
@@ -134,23 +127,19 @@ export default async function init(
       })),
     validate: (input) => {
       if (input.length === 0) {
-        return "Please choose at least one audience.";
+        return "You must choose at least one audience.";
       }
       return true;
     },
   });
 
-  // Save target audience choices as keys
   input.targetAudienceTypes = audienceChoices;
 
-  // 3. 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],
   );
   const defaultKnowledge = mappedPurpose ? PURPOSE_TO_KNOWLEDGE_MAPPING[mappedPurpose] : null;
 
-  // Filter knowledge level options based on previous selections
   const { filteredOptions: filteredKnowledgeOptions } = getFilteredOptions(
     "readerKnowledgeLevel",
     { documentPurpose: prioritizedPurposes, targetAudienceTypes: audienceChoices },
@@ -158,7 +147,7 @@ export default async function init(
   );
 
   const knowledgeChoice = await options.prompts.select({
-    message: "🧠 [3/9]: How much do readers already know about your project?",
+    message: "🧠 [3/9]: How much do your readers already know about your project?",
     choices: Object.entries(filteredKnowledgeOptions).map(([key, level]) => ({
       name: `${level.name}`,
       description: level.description,
@@ -221,7 +210,7 @@ export default async function init(
 
   // Let user select primary language from supported list
   const primaryLanguageChoice = await options.prompts.select({
-    message: "🌐 [5/9]: What's your main documentation language?",
+    message: "🌐 [5/9]: What is the main language of your documentation?",
     choices: SUPPORTED_LANGUAGES.map((lang) => ({
       name: `${lang.label} - ${lang.sample}`,
       value: lang.code,
@@ -238,7 +227,7 @@ export default async function init(
   );
 
   const translateLanguageChoices = await options.prompts.checkbox({
-    message: "🔄 [6/9]: Which languages should we translate to?",
+    message: "🔄 [6/9]: What languages should we translate to?",
     choices: availableTranslationLanguages.map((lang) => ({
       name: `${lang.label} - ${lang.sample}`,
       value: lang.code,
@@ -257,15 +246,17 @@ export default async function init(
   // 8. Content sources
   console.log("\n🔍 [8/9]: Content Sources");
   console.log(
-    "What folders/files should we analyze for documentation? (e.g., ./src, ./docs, ./README.md)",
+    "Please specify the folders and files we should analyze to generate your documentation (e.g., ./src, ./docs, ./README.md).",
   );
-  console.log("💡 Advanced: Use patterns like src/**/*.js or docs/**/*.md for specific files");
-  console.log("💡 Leave empty to scan everything");
+  console.log(
+    "💡 You can also use glob patterns like src/**/*.js or docs/**/*.md for more specific file matching.",
+  );
+  console.log("💡 If you leave this empty, we will scan the entire directory.");
 
   const sourcePaths = [];
   while (true) {
     const selectedPath = await options.prompts.search({
-      message: "File/folder path or pattern:",
+      message: "Please enter a file or folder path, or a glob pattern:",
       source: async (input) => {
         if (!input || input.trim() === "") {
           return [
@@ -292,7 +283,7 @@ export default async function init(
           options.push({
             name: searchTerm,
             value: searchTerm,
-            description: "This input will be used as a glob pattern for file matching",
+            description: "Use this glob pattern for file matching.",
           });
         }
 
@@ -342,7 +333,7 @@ export default async function init(
   // 9. Custom rules - any specific requirements for the documentation?
   const rulesInput = await options.prompts.input({
     message:
-      "📋 [9/9]: Any custom rules or requirements for your documentation? (Optional, press Enter to skip)",
+      "📋 [9/9]: Do you have any custom rules or requirements for your documentation? (Optional, press Enter to skip)",
     default: "",
   });
   input.rules = rulesInput.trim();
@@ -365,22 +356,24 @@ export default async function init(
     await mkdir(dirPath, { recursive: true });
 
     await writeFile(filePath, yamlContent, "utf8");
-    console.log(`\n✅ Setup complete! Configuration saved to: ${chalk.cyan(filePath)}`);
-    // Print YAML content for user review
+    console.log(
+      `\n✅ Setup complete! Your configuration has been saved to: ${chalk.cyan(filePath)}`,
+    );
     console.log(chalk.cyan("---"));
     console.log(chalk.cyan(yamlContent));
     console.log(chalk.cyan("---"));
-    console.log("💡 You can edit the configuration file anytime to modify settings.\n");
-    console.log(`🚀 Ready to generate docs? Run: ${chalk.cyan("aigne doc generate")}\n`);
+    console.log("💡 You can edit this file at any time to change your settings.\n");
+    console.log(`🚀 To generate your documentation, run: ${chalk.cyan("aigne doc generate")}\n`);
 
-    // load config from file
     if (skipIfExists) {
       return loadConfig({ config: filePath, appUrl });
     }
 
     return {};
   } catch (error) {
-    console.error(`❌ Failed to save configuration file: ${error.message}`);
+    console.error(
+      `❌ Sorry, I encountered an error while saving your configuration file: ${error.message}`,
+    );
     return {
       inputGeneratorStatus: false,
       inputGeneratorError: error.message,
@@ -502,14 +495,15 @@ export function generateYAML(input) {
   yaml += `${rulesSection.replace(/rules: ''/, "rules: |\n  ")}\n\n`;
 
   // Target Audience Description
-  yaml += "# Target Audience: Describe your specific target audience and their characteristics\n";
+  yaml += "# Target Audience: Describe your specific target audience and their characteristics.\n";
   const targetAudienceSection = yamlStringify({ targetAudience: config.targetAudience }).trim();
   // Use literal style for multiline strings
   yaml += `${targetAudienceSection.replace(/targetAudience: ''/, "targetAudience: |\n  ")}\n\n`;
 
   // Glossary Configuration
-  yaml += "# Glossary: Define project-specific terms and definitions\n";
-  yaml += '# glossary: "@glossary.md"  # Path to markdown file containing glossary definitions\n\n';
+  yaml += "# Glossary: Define project-specific terms and definitions.\n";
+  yaml +=
+    '# glossary: "@glossary.md"  # Path to a markdown file containing glossary definitions.\n\n';
 
   // Language settings - safely serialize
   const localeSection = yamlStringify({ locale: config.locale }).trim();
@@ -522,25 +516,25 @@ export function generateYAML(input) {
     }).trim();
     yaml += `${translateLanguagesSection.replace(/^translateLanguages:/, "translateLanguages:")}\n`;
   } else {
-    yaml += "# translateLanguages:  # List of languages to translate the documentation to\n";
+    yaml += "# translateLanguages:  # A list of languages to translate the documentation to.\n";
     yaml += "#   - zh  # Example: Chinese translation\n";
     yaml += "#   - en  # Example: English translation\n";
   }
 
   // Directory and source path configurations - safely serialize
   const docsDirSection = yamlStringify({ docsDir: config.docsDir }).trim();
-  yaml += `${docsDirSection.replace(/^docsDir:/, "docsDir:")}  # Directory to save generated documentation\n`;
+  yaml += `${docsDirSection.replace(/^docsDir:/, "docsDir:")}  # The directory where the generated documentation will be saved.\n`;
 
   const sourcesPathSection = yamlStringify({ sourcesPath: config.sourcesPath }).trim();
-  yaml += `${sourcesPathSection.replace(/^sourcesPath:/, "sourcesPath:  # Source code paths to analyze")}\n`;
+  yaml += `${sourcesPathSection.replace(/^sourcesPath:/, "sourcesPath:  # The source code paths to analyze.")}\n`;
 
   // Image filtering settings
   const mediaInfoSection = yamlStringify({
     media: config.media,
   }).trim();
-  yaml += `# minImageWidth: Only images wider than this value (in pixels) will be used in page generation\n${mediaInfoSection}\n`;
+  yaml += `# minImageWidth: Only images wider than this value (in pixels) will be used in the page generation.\n${mediaInfoSection}\n`;
 
   return yaml;
 }
 
-init.description = "Generate a configuration file for the documentation generation process";
+init.description = "Create a configuration file for the documentation generation process.";

package/agents/publish/index.yaml

@@ -9,6 +9,7 @@ skills:
     default_input:
       skipIfExists: true
       checkOnly: true
+  - url: ../init/check.mjs
   - translate-meta.mjs
   - publish-docs.mjs
 input_schema:

package/agents/publish/publish-docs.mjs

@@ -1,12 +1,10 @@
-import { basename, extname, join, relative } from "node:path";
-import slugify from "slugify";
+import { basename, extname, join } from "node:path";
 import { publishDocs as publishDocsFn } from "@aigne/publish-docs";
 import { BrokerClient } from "@blocklet/payment-broker-client/node";
 import chalk from "chalk";
 import fs from "fs-extra";
+import slugify from "slugify";
 
-import { getExtnameFromContentType } from "../../utils/file-utils.mjs";
-import { isHttp } from "../../utils/utils.mjs";
 import { getAccessToken, getOfficialAccessToken } from "../../utils/auth-utils.mjs";
 import {
   CLOUD_SERVICE_URL_PROD,
@@ -17,7 +15,14 @@ import {
 } 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";
+import { getExtnameFromContentType } from "../../utils/file-utils.mjs";
+import { uploadFiles } from "../../utils/upload-files.mjs";
+import {
+  getGithubRepoUrl,
+  isHttp,
+  loadConfigFromFile,
+  saveValueToConfig,
+} from "../../utils/utils.mjs";
 import updateBranding from "../utils/update-branding.mjs";
 
 const BASE_URL = process.env.DOC_SMITH_BASE_URL || CLOUD_SERVICE_URL_PROD;
@@ -86,16 +91,8 @@ export default async function publishDocs(
       }
 
       const choice = await options.prompts.select({
-        message: "Select platform to publish your documents:",
+        message: "Please select a platform to publish your documents:",
         choices: [
-          {
-            name: `${chalk.blue("DocSmith Cloud (docsmith.aigne.io)")} – ${chalk.green("Free")} hosting. Your documents will be publicly accessible. Best for open-source projects or community sharing.`,
-            value: "default",
-          },
-          {
-            name: `${chalk.blue("Your existing website")} - Integrate and publish directly on your current site (setup required)`,
-            value: "custom",
-          },
           ...(sessionId
             ? [
                 {
@@ -104,6 +101,14 @@ export default async function publishDocs(
                 },
               ]
             : []),
+          {
+            name: `${chalk.blue("DocSmith Cloud (docsmith.aigne.io)")} – ${chalk.green("Free")} hosting. Your documents will be publicly accessible. Best for open-source projects or community sharing.`,
+            value: "default",
+          },
+          {
+            name: `${chalk.blue("Your existing website")} - Integrate and publish directly on your current site (setup required)`,
+            value: "custom",
+          },
           {
             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",
@@ -117,7 +122,7 @@ export default async function publishDocs(
             `Start here to run your own website:\n${chalk.cyan(DISCUSS_KIT_STORE_URL)}\n`,
         );
         const userInput = await options.prompts.input({
-          message: "Please enter your website URL:",
+          message: "Please enter the URL of your website:",
           validate: (input) => {
             try {
               // Check if input contains protocol, if not, prepend https://
@@ -146,7 +151,7 @@ export default async function publishDocs(
           paymentUrl = paymentLink;
           console.log(`\nResuming your previous website setup...`);
         } else {
-          console.log(`\nCreating new website for your documentation...`);
+          console.log(`\nCreating a new website for your documentation...`);
         }
         const { appUrl: homeUrl, token: ltToken } = (await deploy(id, paymentUrl)) || {};
 
@@ -155,7 +160,7 @@ export default async function publishDocs(
       }
     }
 
-    console.log(`\nPublishing docs to ${chalk.cyan(appUrl)}\n`);
+    console.log(`\nPublishing your documentation to ${chalk.cyan(appUrl)}\n`);
 
     const accessToken = await getAccessToken(appUrl, token);
 
@@ -192,8 +197,16 @@ export default async function publishDocs(
           const finalPath = ext ? `${tempFilePath}.${ext}` : tempFilePath;
           fs.writeFileSync(finalPath, buffer);
 
-          // Update to relative path
-          projectInfo.icon = relative(join(process.cwd(), DOC_SMITH_DIR), finalPath);
+          const filePath = join(process.cwd(), finalPath);
+          const { results: uploadResults } = await uploadFiles({
+            appUrl,
+            filePaths: [filePath],
+            accessToken,
+            concurrency: 1,
+          });
+
+          // FIXME: 暂时保持使用绝对路径，需要修改为相对路径 @pengfei
+          projectInfo.icon = uploadResults?.[0]?.url || projectInfo.icon;
         }
       } catch (error) {
         console.warn(`Failed to download project logo from ${projectInfo.icon}: ${error.message}`);
@@ -253,21 +266,21 @@ export default async function publishDocs(
     } else {
       // If the error is 401 or 403, it means the access token is invalid
       if (error?.includes("401") || error?.includes("403")) {
-        message = `❌ Publishing failed: you don't have valid authorization.\n   Run ${chalk.cyan("aigne doc clear")} to reset it, then publish again.`;
+        message = `❌ Publishing failed due to an authorization error. Please run ${chalk.cyan("aigne doc clear")} to reset your credentials and try again.`;
       }
     }
 
     // clean up tmp work dir
     await fs.rm(docsDir, { recursive: true, force: true });
   } catch (error) {
-    message = `❌ Failed to publish docs: ${error.message}`;
+    message = `❌ Sorry, I encountered an error while publishing your documentation: ${error.message}`;
 
     // clean up tmp work dir in case of error
     try {
       const docsDir = join(DOC_SMITH_DIR, TMP_DIR, TMP_DOCS_DIR);
       await fs.rm(docsDir, { recursive: true, force: true });
     } catch {
-      // ignore cleanup errors
+      // Ignore cleanup errors
     }
   }
 
@@ -279,34 +292,34 @@ publishDocs.input_schema = {
   properties: {
     docsDir: {
       type: "string",
-      description: "The directory of the docs",
+      description: "The directory of the documentation.",
     },
     appUrl: {
       type: "string",
-      description: "The url of the app",
+      description: "The URL of the app.",
       default: CLOUD_SERVICE_URL_PROD,
     },
     boardId: {
       type: "string",
-      description: "The id of the board",
+      description: "The ID of the board.",
     },
     "with-branding": {
       type: "boolean",
-      description: "Update your website branding (title, description, logo)",
+      description: "Update the website branding (title, description, and logo).",
     },
     projectName: {
       type: "string",
-      description: "The name of the project",
+      description: "The name of the project.",
     },
     projectDesc: {
       type: "string",
-      description: "The description of the project",
+      description: "A description of the project.",
     },
     projectLogo: {
       type: "string",
-      description: "The logo/icon of the project",
+      description: "The logo or icon of the project.",
     },
   },
 };
 
-publishDocs.description = "Publish the documentation to website";
+publishDocs.description = "Publish the documentation to a website";

package/agents/translate/choose-language.mjs

@@ -2,12 +2,12 @@ import { SUPPORTED_LANGUAGES } from "../../utils/constants/index.mjs";
 import { loadConfigFromFile, saveValueToConfig } from "../../utils/utils.mjs";
 
 /**
- * Interactive language selector for translation from configured languages
+ * Interactively selects languages for translation.
  * @param {Object} params
- * @param {Array<string>} [params.langs] - Pre-selected languages
- * @param {string} params.locale - Primary language code
- * @param {Object} options - Options object with prompts
- * @returns {Promise<Object>} Selected languages
+ * @param {Array<string>} [params.langs] - Pre-selected languages.
+ * @param {string} params.locale - The primary language code.
+ * @param {Object} options - Options object with prompts.
+ * @returns {Promise<Object>} The selected languages.
  */
 export default async function chooseLanguage({ langs, locale, selectedDocs }, options) {
   let selectedLanguages = [];
@@ -24,7 +24,7 @@ export default async function chooseLanguage({ langs, locale, selectedDocs }, op
     (lang) => lang.code !== primaryLanguage,
   );
 
-  // If languages are provided as parameter, validate against available languages
+  // If languages are provided as a parameter, validate them
   if (langs && Array.isArray(langs) && langs.length > 0) {
     const validLanguages = langs.filter((lang) =>
       availableTranslationLanguages.some((availableLang) => availableLang.code === lang),
@@ -35,8 +35,7 @@ export default async function chooseLanguage({ langs, locale, selectedDocs }, op
     } else {
       console.log(`⚠️  Invalid languages provided: ${langs.join(", ")}`);
       console.log(
-        "Available translation languages:",
-        availableTranslationLanguages.map((l) => l.code).join(", "),
+        `Available languages: ${availableTranslationLanguages.map((l) => l.code).join(", ")}`,
       );
     }
   }
@@ -51,11 +50,11 @@ export default async function chooseLanguage({ langs, locale, selectedDocs }, op
     }));
 
     selectedLanguages = await options.prompts.checkbox({
-      message: "Select translation languages:",
+      message: "Please select the languages you'd like to translate to:",
       choices: choices,
       validate: (answer) => {
         if (answer.length === 0) {
-          return "Please select at least one language";
+          return "You must select at least one language.";
         }
         return true;
       },
@@ -63,7 +62,7 @@ export default async function chooseLanguage({ langs, locale, selectedDocs }, op
   }
 
   if (selectedLanguages.length === 0) {
-    throw new Error("No languages selected for translation");
+    throw new Error("You must select at least one language to continue.");
   }
 
   // Find new languages that need to be added
@@ -98,12 +97,13 @@ chooseLanguage.input_schema = {
       items: {
         type: "string",
       },
-      description: "Pre-selected languages for translation",
+      description: "A list of pre-selected languages for translation.",
     },
 
     locale: {
       type: "string",
-      description: "Primary language code (will be excluded from translation options)",
+      description:
+        "The primary language code, which will be excluded from the translation options.",
     },
   },
   required: [],

package/agents/translate/index.yaml

@@ -7,6 +7,7 @@ skills:
       skipIfExists: true
       checkOnly: true
   - ../utils/load-sources.mjs
+  - url: ../init/check.mjs
   - type: transform
     task_render_mode: hide
     jsonata: |

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

@@ -1,10 +1,10 @@
 export default async function checkUpdateIsSingle({ selectedDocs, ...rest }, options) {
   if (!selectedDocs || !Array.isArray(selectedDocs)) {
-    throw new Error("selectedDocs must be provided as an array");
+    throw new Error("A list of documents to update must be provided.");
   }
 
   if (selectedDocs.length === 0) {
-    throw new Error("selectedDocs cannot be empty");
+    throw new Error("You must select at least one document to update.");
   }
 
   let targetAgent;
@@ -20,7 +20,7 @@ export default async function checkUpdateIsSingle({ selectedDocs, ...rest }, opt
   }
 
   if (!targetAgent) {
-    throw new Error(`Agent "${agentName}" is not available`);
+    throw new Error(`Sorry, I can't seem to find the "${agentName}" agent.`);
   }
 
   try {
@@ -31,7 +31,10 @@ export default async function checkUpdateIsSingle({ selectedDocs, ...rest }, opt
 
     return result;
   } catch (error) {
-    console.error(`Error invoking ${agentName}:`, error.message);
+    console.error(
+      `Sorry, I encountered an error while trying to run the ${agentName} agent:`,
+      error.message,
+    );
     throw error;
   }
 }

package/agents/update/index.yaml

@@ -8,6 +8,7 @@ skills:
     default_input:
       skipIfExists: true
       checkOnly: true
+  - url: ../init/check.mjs
   - ../utils/load-sources.mjs
   - type: transform
     task_render_mode: hide

package/agents/utils/choose-docs.mjs

@@ -1,3 +1,4 @@
+import chalk from "chalk";
 import {
   addFeedbackToItems,
   findItemByPath,
@@ -35,7 +36,7 @@ export default async function chooseDocs(
 
       if (mainLanguageFiles.length === 0) {
         throw new Error(
-          "No documents found in the docs directory. Please run `aigne docs generate` to generate the documents",
+          `No documents found in the docs directory. Please run ${chalk.yellow("`aigne doc generate`")} to generate the documents`,
         );
       }
 
@@ -80,9 +81,10 @@ export default async function chooseDocs(
       // Process selected files and convert to found items
       foundItems = await processSelectedFiles(selectedFiles, documentExecutionStructure, docsDir);
     } catch (error) {
-      throw new Error(
+      console.log(
         getActionText(isTranslate, `\nFailed to select documents to {action}: ${error.message}`),
       );
+      process.exit(0);
     }
   } else {
     // Process the provided docs array

package/agents/utils/update-branding.mjs

@@ -17,12 +17,13 @@ export default async function updateBranding({ appUrl, projectInfo, accessToken
   try {
     const origin = new URL(appUrl).origin;
     if ([CLOUD_SERVICE_URL_PROD, CLOUD_SERVICE_URL_STAGING].includes(origin)) {
-      console.log("Skipped updating branding for official service\n");
+      console.log("ℹ️ Skipped updating branding for official service\n");
       return;
     }
 
     console.log(`🔄 Updating branding for ${chalk.cyan(origin)}`);
 
+    // Get component information and mount point
     const componentInfo = await getComponentInfoWithMountPoint(origin, DISCUSS_KIT_DID);
     const mountPoint = componentInfo.mountPoint || "/";
 
@@ -56,7 +57,7 @@ export default async function updateBranding({ appUrl, projectInfo, accessToken
             endpoint: `${origin}/.well-known/service/blocklet/logo/upload/square/${componentInfo.did}`,
           });
         }
-        console.log("✅ Updated branding successfully!\n");
+        console.log("✅ Branding has been successfully updated\n");
       } catch (error) {
         console.warn(`⚠️ Just failed to update logo: ${error.message}\n`);
       }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aigne/doc-smith",
-  "version": "0.8.12-beta.9",
+  "version": "0.8.13-beta",
   "description": "AI-driven documentation generation tool built on the AIGNE Framework",
   "publishConfig": {
     "access": "public"