@kweaver-ai/kweaver-sdk 0.8.1 → 0.8.2

package/dist/commands/agent.js

@@ -1,6 +1,7 @@
 import { ensureValidToken, formatHttpError, with401RefreshRetry } from "../auth/oauth.js";
 import { runAgentChatCommand } from "./agent-chat.js";
 import { runAgentSkillCommand } from "./agent-members.js";
+import { AGENT_MODE_HELP, applyAgentModeToConfig, normalizeAgentConfigInput, parseAgentMode, } from "./agent/mode.js";
 import { listAgents, getAgent, getAgentByKey, createAgent, updateAgent, deleteAgent, publishAgent, unpublishAgent, listPersonalAgents, listPublishedAgentTemplates, getPublishedAgentTemplate, listAgentCategories, } from "../api/agent-list.js";
 import { listConversations, listMessages, getTracesByConversation } from "../api/conversations.js";
 import { fetchAgentInfo } from "../api/agent-chat.js";
@@ -47,21 +48,21 @@ export async function resolveLlmName(options) {
     }
 }
 /**
- * 生成带时间戳的文件路径
- * @param path 用户提供的路径
- * @returns 带时间戳的文件路径
+ * Generate a timestamped file path.
+ * @param path User-provided output path.
+ * @returns The timestamped file path.
  */
 function generateTimestampedPath(path) {
     const timestamp = new Date().toISOString().replace(/[:.]/g, "-").slice(0, 19);
-    // 如果路径以 / 结尾，视为目录，在目录下生成文件
+    // Treat paths ending in / as directories.
     if (path.endsWith("/")) {
         return join(path, `agent-config-${timestamp}.json`);
     }
-    // 在文件名中插入时间戳：config.json -> config-2025-01-15T12-30-45.json
+    // Insert the timestamp before the extension: config.json -> config-2025-01-15T12-30-45.json.
     const ext = extname(path);
     const base = basename(path, ext);
     const dir = dirname(path);
-    // 如果 dir 是 "."，说明没有目录前缀，直接返回带时间戳的文件名
+    // No directory prefix was provided, so return only the timestamped file name.
     if (dir === ".") {
         return `${base}-${timestamp}${ext}`;
     }
@@ -549,7 +550,7 @@ Subcommands:
   get-by-key <key>                   Get agent by key
   create --name <n> --profile <p>    Create a new agent
        [--key <key>] [--product-key <pk>] [--system-prompt <sp>]
-       [--llm-id <id>] [--llm-max-tokens <n>]
+       [--llm-id <id>] [--llm-max-tokens <n>] [--mode <mode>]
   update <agent_id> [options]        Update an existing agent
   delete <agent_id> [-y]             Delete an agent
   publish <agent_id>                 Publish an agent
@@ -653,7 +654,8 @@ Options:
   --profile <text>          Agent description (max 500)
   --system-prompt <text>    System prompt
   --knowledge-network-id <id>  Business knowledge network ID to configure
-  --config-path <path>      Path to config file (read from file instead of API)`);
+  --config-path <path>      Path to config object file, or full agent JSON with config
+${AGENT_MODE_HELP}`);
             return 0;
         }
     }
@@ -891,12 +893,12 @@ Options:
             businessDomain: options.businessDomain,
         });
         if (body) {
-            // 如果指定了 --save-config，保存 config 到文件（带时间戳）
+            // Save only the agent config object when --save-config is provided.
             if (options.saveConfig) {
                 const parsed = JSON.parse(body);
                 const config = parsed.config ?? {};
                 const timestampedPath = generateTimestampedPath(options.saveConfig);
-                // 确保目录存在
+                // Ensure the output directory exists.
                 const dir = dirname(timestampedPath);
                 await fs.mkdir(dir, { recursive: true });
                 await fs.writeFile(timestampedPath, JSON.stringify(config, null, 2), "utf-8");
@@ -1146,6 +1148,7 @@ async function runAgentCreateCommand(args) {
     let llmMaxTokens = 4096;
     let businessDomain = "";
     let configStr = "";
+    let explicitMode;
     for (let i = 0; i < args.length; i += 1) {
         const arg = args[i];
         if (arg === "--help" || arg === "-h") {
@@ -1163,6 +1166,7 @@ Optional:
   --system-prompt <text>   System prompt
   --llm-id <id>            LLM model ID (required for public API)
   --llm-max-tokens <n>     LLM max tokens (default: 4096)
+${AGENT_MODE_HELP}
   --config <json|path>     Full config object as JSON string or file path (overrides individual config options)
   -bd, --biz-domain <val>  Business domain (default: bd_public)`);
             return 0;
@@ -1195,6 +1199,17 @@ Optional:
             llmMaxTokens = parseInt(args[++i] ?? "4096", 10);
             continue;
         }
+        if (arg === "--mode") {
+            const value = args[++i] ?? "";
+            try {
+                explicitMode = parseAgentMode(value);
+            }
+            catch (error) {
+                console.error(error instanceof Error ? error.message : String(error));
+                return 1;
+            }
+            continue;
+        }
         if (arg === "--config") {
             configStr = args[++i] ?? "";
             continue;
@@ -1241,13 +1256,19 @@ Optional:
             system_prompt: systemPrompt,
         };
     }
+    try {
+        applyAgentModeToConfig(config, explicitMode);
+    }
+    catch (error) {
+        console.error(error instanceof Error ? error.message : String(error));
+        return 1;
+    }
     const payload = {
         name,
         profile,
         avatar_type: 1,
         avatar: "icon-dip-agent-default",
         product_key: productKey,
-        product_name: "DIP",
         config,
     };
     if (key)
@@ -1276,41 +1297,77 @@ Optional:
 // ── Update ───────────────────────────────────────────────────────────────────
 async function runAgentUpdateCommand(args) {
     const agentId = args[0];
+    if (agentId === "--help" || agentId === "-h") {
+        console.log(`kweaver agent update <agent_id> [options]
+
+Update an existing agent.
+
+Options:
+  --name <text>                Agent name
+  --profile <text>             Agent description
+  --system-prompt <text>       System prompt
+  --knowledge-network-id <id>  Knowledge network ID
+  --config-path <path>         Read config object from file, or full agent JSON with config
+${AGENT_MODE_HELP}`);
+        return 0;
+    }
     if (!agentId || agentId.startsWith("-")) {
-        console.error("Usage: kweaver agent update <agent_id> [--name <n>] [--profile <p>] [--system-prompt <sp>] [--knowledge-network-id <id> [--config-path <path>]]");
+        console.error("Usage: kweaver agent update <agent_id> [--name <n>] [--profile <p>] [--system-prompt <sp>] [--mode <mode>] [--knowledge-network-id <id> [--config-path <path>]]");
         return 1;
     }
     let knowledgeNetworkId = null;
     let configPath = null;
-    try {
-        const token = await ensureValidToken();
-        let current;
-        let configFromFile = null;
-        // 如果指定了 --config-path，从文件读取配置
-        if (args.includes("--config-path")) {
-            const configPathIndex = args.indexOf("--config-path");
-            configPath = args[configPathIndex + 1] ?? "";
+    let explicitMode;
+    for (let i = 1; i < args.length; i += 1) {
+        const arg = args[i];
+        if (arg === "--mode") {
+            const value = args[++i] ?? "";
+            try {
+                explicitMode = parseAgentMode(value);
+            }
+            catch (error) {
+                console.error(error instanceof Error ? error.message : String(error));
+                return 1;
+            }
+            continue;
+        }
+        if (arg === "--config-path") {
+            configPath = args[++i] ?? "";
             if (!configPath || configPath.startsWith("-")) {
                 console.error("Missing value for --config-path flag");
                 return 1;
             }
+            continue;
+        }
+    }
+    try {
+        let current;
+        let configFromFile = null;
+        // Read config from file before fetching the current agent so invalid local config fails early.
+        if (configPath) {
             try {
                 const fileContent = await fs.readFile(configPath, "utf-8");
-                configFromFile = JSON.parse(fileContent);
+                configFromFile = normalizeAgentConfigInput(JSON.parse(fileContent));
+                if (!explicitMode) {
+                    applyAgentModeToConfig(configFromFile);
+                }
             }
             catch (error) {
-                console.error(`Failed to read config from ${configPath}: ${error}`);
+                console.error(error instanceof Error && error.message.startsWith("config.mode")
+                    ? error.message
+                    : `Failed to read config from ${configPath}: ${error}`);
                 return 1;
             }
         }
-        // 从API获取当前 agent 配置
+        const token = await ensureValidToken();
+        // Fetch the current agent for read-modify-write updates.
         const currentRaw = await getAgent({
             baseUrl: token.baseUrl,
             accessToken: token.accessToken,
             agentId,
         });
         current = JSON.parse(currentRaw);
-        // 如果从文件读取了 config，合并到 current 中
+        // Replace only current.config when a config file was provided.
         if (configFromFile) {
             current.config = configFromFile;
         }
@@ -1324,6 +1381,14 @@ async function runAgentUpdateCommand(args) {
                 current.profile = args[++i] ?? current.profile;
                 continue;
             }
+            if (arg === "--mode") {
+                i += 1;
+                continue;
+            }
+            if (arg === "--config-path") {
+                i += 1;
+                continue;
+            }
             if (arg === "--system-prompt") {
                 const config = (current.config ?? {});
                 config.system_prompt = args[++i] ?? "";
@@ -1339,21 +1404,28 @@ async function runAgentUpdateCommand(args) {
                 continue;
             }
         }
-        // 如果指定了 --knowledge-network-id，更新 data_source.knowledge_network
+        // Update config.data_source.knowledge_network when requested.
         if (knowledgeNetworkId) {
             const config = (current.config ?? {});
             const dataSource = (config.data_source ?? {});
-            // 获取知识网络名称（如果需要的话，可以查询BKN获取）
             const knowledgeNetwork = [
                 {
                     knowledge_network_id: knowledgeNetworkId,
-                    knowledge_network_name: "", // 可选：通过BKN API获取名称
                 },
             ];
             dataSource.knowledge_network = knowledgeNetwork;
             config.data_source = dataSource;
             current.config = config;
         }
+        const config = (current.config ?? {});
+        try {
+            applyAgentModeToConfig(config, explicitMode);
+        }
+        catch (error) {
+            console.error(error instanceof Error ? error.message : String(error));
+            return 1;
+        }
+        current.config = config;
         const body = await updateAgent({
             baseUrl: token.baseUrl,
             accessToken: token.accessToken,
@@ -1601,12 +1673,12 @@ Options:
             businessDomain: options.businessDomain,
         });
         if (body) {
-            // 如果指定了 --save-config，保存 config 到文件（带时间戳）
+            // Save only the template config object when --save-config is provided.
             if (options.saveConfig) {
                 const parsed = JSON.parse(body);
                 const config = parsed.config ?? {};
                 const timestampedPath = generateTimestampedPath(options.saveConfig);
-                // 确保目录存在
+                // Ensure the output directory exists.
                 const dir = dirname(timestampedPath);
                 await fs.mkdir(dir, { recursive: true });
                 await fs.writeFile(timestampedPath, JSON.stringify(config, null, 2), "utf-8");