@cicctencent/agent-midway 0.1.1

package/dist/service/prompt-builder.service.js

@@ -0,0 +1,246 @@
+"use strict";
+var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
+    var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
+    if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
+    else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
+    return c > 3 && r && Object.defineProperty(target, key, r), r;
+};
+var __metadata = (this && this.__metadata) || function (k, v) {
+    if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.PromptBuilderService = void 0;
+const core_1 = require("@midwayjs/core");
+const agent_profile_service_1 = require("./agent-profile.service");
+const mcp_server_service_1 = require("./mcp-server.service");
+const ai_skill_service_1 = require("./ai-skill.service");
+const agent_server_1 = require("@cicctencent/agent-server");
+/**
+ * Prompt 构建服务
+ *
+ * 负责构建 Agent 的系统提示词，包括：
+ * - 基础 Prompt（来自 Agent Profile）
+ * - 工具说明注入
+ * - 记忆上下文
+ * - Skill 上下文
+ */
+let PromptBuilderService = class PromptBuilderService {
+    /**
+     * 构建完整的系统提示词
+     */
+    async buildSystemPrompt(applicationId, profileId, options) {
+        const parts = [];
+        // 1. 获取 Agent Profile
+        let profile = null;
+        if (profileId) {
+            profile = await this.agentProfileService.get(profileId, applicationId);
+        }
+        else {
+            profile = await this.agentProfileService.getDefault(applicationId);
+        }
+        // 2. 基础 Prompt
+        if (profile?.basePrompt) {
+            parts.push(profile.basePrompt);
+        }
+        else {
+            parts.push(this.getDefaultPrompt(profile?.agentType));
+        }
+        // 3. 工具说明
+        if (options?.includeTools !== false && profile) {
+            const toolSection = await this.buildToolSection(profile, applicationId);
+            if (toolSection) {
+                parts.push(toolSection);
+            }
+        }
+        // 4. Skill 上下文
+        if (options?.includeSkills !== false && profile) {
+            const skillSection = await this.buildSkillSection(profile);
+            if (skillSection) {
+                parts.push(skillSection);
+            }
+        }
+        // 5. 记忆上下文
+        if (options?.includeMemory && options?.memoryContext) {
+            parts.push(`\n## 记忆上下文\n\n${options.memoryContext}`);
+        }
+        // 6. 领域上下文
+        if (profile?.domain) {
+            parts.push(`\n## 领域\n\n你正在为「${profile.domain}」领域提供服务。`);
+        }
+        return parts.join('\n\n');
+    }
+    /**
+     * 构建工具说明部分（使用 agent-server 共享片段）
+     */
+    async buildToolSection(profile, applicationId) {
+        const tools = [];
+        // 内置工具说明
+        tools.push('- 你可以使用以下内置工具来完成任务');
+        // MCP Server 工具
+        if (profile.mcpServers && profile.mcpServers.length > 0) {
+            for (const serverId of profile.mcpServers) {
+                const serverTools = await this.mcpServerService.getTools(Number(serverId), applicationId);
+                for (const tool of serverTools) {
+                    tools.push(`  - ${tool.name || tool.id}: ${tool.description || ''}`);
+                }
+            }
+        }
+        // 禁用的工具
+        if (profile.disabledTools && profile.disabledTools.length > 0) {
+            tools.push(`\n- 以下工具已被禁用: ${profile.disabledTools.join(', ')}`);
+        }
+        if (tools.length <= 1)
+            return '';
+        // 合并 agent-server 的结果综合指引
+        const synthesisPrompt = agent_server_1.RESULT_SYNTHESIS_PROMPT;
+        return `## 可用工具\n\n${tools.join('\n')}${synthesisPrompt}`;
+    }
+    /**
+     * 构建 Skill 上下文部分
+     */
+    async buildSkillSection(profile) {
+        if (!profile.alwaysInjectSkills ||
+            profile.alwaysInjectSkills.length === 0) {
+            return '';
+        }
+        try {
+            const skillContext = await this.skillService.previewSkillContext(profile.agentType || 'chat');
+            if (skillContext) {
+                return `## 技能上下文\n\n${skillContext}`;
+            }
+        }
+        catch {
+            // Skill 服务不可用时静默跳过
+        }
+        return '';
+    }
+    /**
+     * 构建默认 Agent 的系统提示词（使用 agent-server 共享片段）
+     */
+    getDefaultPrompt(agentType) {
+        const basePrompt = `你是一个智能助手，能够帮助用户完成各种任务。
+
+## 核心能力
+- 理解用户的自然语言指令
+- 使用可用工具获取信息和执行操作
+- 提供准确、有用的回答
+
+## 行为准则
+- 始终保持友好和专业的态度
+- 当不确定时，主动询问用户以获取更多信息
+- 使用工具时，说明你在做什么以及为什么`;
+        // 合并 agent-server 的全局基础规则
+        let prompt = basePrompt + agent_server_1.GLOBAL_BASE_RULES;
+        if (agentType === 'data-analyst') {
+            prompt += `
+
+## 数据分析专家
+你专注于数据分析任务，能够：
+- 查询和分析数据
+- 生成图表和报告
+- 提供数据驱动的洞察和建议`;
+            // 添加结果综合指引
+            prompt += agent_server_1.RESULT_SYNTHESIS_PROMPT;
+        }
+        return prompt;
+    }
+    /**
+     * 构建 Specialist Agent 的系统提示词（使用 agent-server 共享片段）
+     */
+    buildSpecialistPrompt(profile, toolRegistry, skillManager) {
+        const base = profile.basePrompt || '你是一个专业的 AI 助手，能够使用工具完成任务。';
+        // 合并通用规则（来自 agent-server）
+        let prompt = base + agent_server_1.GLOBAL_BASE_RULES;
+        // 始终注入 alwaysInjectSkills 中的 Skill 内容
+        if (profile.alwaysInjectSkills?.length && skillManager) {
+            for (const skillId of profile.alwaysInjectSkills) {
+                const skill = skillManager.get(skillId);
+                if (skill?.content) {
+                    prompt += `\n\n## ${skill.name || skillId}\n\n${skill.content}`;
+                }
+            }
+        }
+        // 工具优先引导（使用 agent-server 共享片段）
+        if (toolRegistry) {
+            const capabilities = (0, agent_server_1.getAgentToolCapabilities)(toolRegistry);
+            if (capabilities.mcpToolNames.length > 0) {
+                const toolList = (0, agent_server_1.buildMCPToolDescList)(toolRegistry, true);
+                prompt += (0, agent_server_1.buildSpecialistMcpPriorityPrompt)(profile.name, toolList);
+                prompt += agent_server_1.RESULT_SYNTHESIS_PROMPT;
+                // 数据可视化引导
+                if (capabilities.hasOutputTools || capabilities.hasShowWidget) {
+                    prompt += (0, agent_server_1.buildDataPresentationPrompt)(capabilities, {
+                        jsonFileExample: 'data.json',
+                        includeHtmlQualityRules: true,
+                    });
+                }
+                prompt += agent_server_1.SPECIALIST_EFFICIENCY_RULES;
+            }
+            // 通用角色边界规则
+            prompt += agent_server_1.OUTPUT_CONCISION_RULES;
+            prompt += agent_server_1.SUB_AGENT_RESULT_SUMMARY_RULES;
+            prompt += (0, agent_server_1.buildSpecialistOutputBoundaryPrompt)(capabilities, {
+                includeDocumentFormatRules: true,
+            });
+        }
+        // 领域上下文
+        if (profile.domain) {
+            prompt += `\n\n## 领域\n\n你正在为「${profile.domain}」领域提供服务。`;
+        }
+        return prompt;
+    }
+    /**
+     * 解析工具显示名（delegate_task → "分发任务 → {agentName}"）
+     */
+    resolveToolDisplayName(toolName, args, toolRegistry) {
+        // delegate_task：显示目标 Agent 名称
+        if (toolName === 'delegate_task') {
+            const agentName = args.agent || '';
+            return agentName ? `分发任务 → ${agentName}` : '分发任务';
+        }
+        // 内置工具
+        if (agent_server_1.BUILTIN_TOOL_LABELS[toolName]) {
+            return agent_server_1.BUILTIN_TOOL_LABELS[toolName];
+        }
+        // MCP 工具
+        if (toolName.startsWith('mcp_') && toolRegistry) {
+            const tool = toolRegistry.get(toolName);
+            const serverName = tool?.metadata?.mcpServer || '';
+            const connectorId = tool?.metadata?.connectorId || '';
+            const originalName = tool?.metadata?.originalName || '';
+            const readable = (originalName || toolName.slice(4))
+                .replace(/_/g, ' ')
+                .replace(/\b\w/g, c => c.toUpperCase());
+            const parts = [];
+            if (serverName)
+                parts.push(serverName);
+            if (connectorId)
+                parts.push(connectorId);
+            return parts.length > 0 ? `${readable} (${parts.join(' · ')})` : readable;
+        }
+        // 其他工具
+        const readable = toolName.replace(/_/g, ' ');
+        return readable.charAt(0).toUpperCase() + readable.slice(1);
+    }
+};
+exports.PromptBuilderService = PromptBuilderService;
+__decorate([
+    (0, core_1.Inject)(),
+    __metadata("design:type", Object)
+], PromptBuilderService.prototype, "ctx", void 0);
+__decorate([
+    (0, core_1.Inject)(),
+    __metadata("design:type", agent_profile_service_1.AgentProfileService)
+], PromptBuilderService.prototype, "agentProfileService", void 0);
+__decorate([
+    (0, core_1.Inject)(),
+    __metadata("design:type", mcp_server_service_1.MCPServerService)
+], PromptBuilderService.prototype, "mcpServerService", void 0);
+__decorate([
+    (0, core_1.Inject)(),
+    __metadata("design:type", ai_skill_service_1.AISkillService)
+], PromptBuilderService.prototype, "skillService", void 0);
+exports.PromptBuilderService = PromptBuilderService = __decorate([
+    (0, core_1.Provide)(),
+    (0, core_1.Scope)(core_1.ScopeEnum.Request, { allowDowngrade: true })
+], PromptBuilderService);

package/dist/tools/code-exec.tool.d.ts

@@ -0,0 +1,37 @@
+/**
+ * 代码执行工具
+ *
+ * 让 Agent 能够在安全沙箱中执行 JavaScript 代码片段，
+ * 用于数据计算、格式转换、统计分析等任务。
+ *
+ * 工具说明：
+ * - execute_code：在沙箱中执行任意 JS 代码，支持标准库
+ * - analyze_data：对 JSON 数组数据执行常见分析操作（摘要/排序/过滤/分组）
+ *
+ * 安全机制：
+ * - 使用 Function 构造器在受限作用域中执行，无法访问外部模块
+ * - 仅暴露安全的标准库（Math、JSON、Date、Array、Object 等）
+ * - 执行超时控制（默认 5s，最大 30s）
+ * - 输出长度限制（最大 10KB）
+ *
+ * 生产环境升级路径：
+ * - 使用 vm2 / isolated-vm / Docker 容器实现更强的隔离
+ * - 支持 Python 执行（通过 Pyodide 或子进程）
+ */
+import type { ToolDefinition } from '../types';
+/** 代码执行工具 */
+export declare const executeCodeTool: ToolDefinition<{
+    code: string;
+    language?: string;
+    timeout?: number;
+}, {
+    result: any;
+    stdout: string;
+    error?: string;
+}>;
+/** 数据分析工具（基于代码执行） */
+export declare const analyzeDataTool: ToolDefinition<{
+    data: any;
+    operation: string;
+    description?: string;
+}, any>;

package/dist/tools/code-exec.tool.js

@@ -0,0 +1,162 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.analyzeDataTool = exports.executeCodeTool = void 0;
+/** 安全沙箱执行代码 */
+async function executeInSandbox(code, timeout) {
+    const logs = [];
+    const maxOutputLength = 10000;
+    // 构造受限的执行上下文
+    const sandbox = {
+        console: {
+            log: (...args) => logs.push(args.map(String).join(' ')),
+            warn: (...args) => logs.push(`[WARN] ${args.map(String).join(' ')}`),
+            error: (...args) => logs.push(`[ERROR] ${args.map(String).join(' ')}`),
+        },
+        Math,
+        JSON,
+        Date,
+        Array,
+        Object,
+        String,
+        Number,
+        Boolean,
+        RegExp,
+        Map,
+        Set,
+        parseInt,
+        parseFloat,
+        isNaN,
+        isFinite,
+        encodeURIComponent,
+        decodeURIComponent,
+    };
+    try {
+        // 使用 Function 构造器在受限作用域中执行
+        const paramNames = Object.keys(sandbox);
+        const paramValues = Object.values(sandbox);
+        const fn = new Function(...paramNames, `"use strict";\n${code}`);
+        // 超时控制
+        const result = await Promise.race([
+            Promise.resolve(fn(...paramValues)),
+            new Promise((_, reject) => setTimeout(() => reject(new Error(`执行超时 (${timeout}ms)`)), timeout)),
+        ]);
+        const stdout = logs.join('\n').slice(0, maxOutputLength);
+        return { result, stdout };
+    }
+    catch (e) {
+        return {
+            result: null,
+            stdout: logs.join('\n').slice(0, maxOutputLength),
+            error: e.message || String(e),
+        };
+    }
+}
+/** 代码执行工具 */
+exports.executeCodeTool = {
+    id: 'execute_code',
+    name: '执行代码',
+    description: '在安全沙箱中执行 JavaScript 代码片段。可用于数据计算、格式转换、统计分析等。支持 Math、JSON、Date、Array、Object 等标准库。使用 console.log 输出中间结果。',
+    category: 'compute',
+    parameters: {
+        type: 'object',
+        properties: {
+            code: {
+                type: 'string',
+                description: 'JavaScript 代码。最后一行表达式或 return 语句的值作为返回结果。',
+            },
+            language: {
+                type: 'string',
+                description: '编程语言（当前仅支持 javascript）',
+                enum: ['javascript'],
+                default: 'javascript',
+            },
+            timeout: {
+                type: 'integer',
+                description: '执行超时时间（毫秒），默认 5000，最大 30000',
+                minimum: 1000,
+                maximum: 30000,
+                default: 5000,
+            },
+        },
+        required: ['code'],
+    },
+    async execute(input) {
+        const timeout = Math.min(Math.max(Number(input.timeout) || 5000, 1000), 30000);
+        return executeInSandbox(input.code, timeout);
+    },
+};
+/** 数据分析工具（基于代码执行） */
+exports.analyzeDataTool = {
+    id: 'analyze_data',
+    name: '数据分析',
+    description: '对输入的数据执行分析操作。支持：统计摘要(sum/avg/min/max/count)、排序、过滤、分组聚合等。',
+    category: 'compute',
+    parameters: {
+        type: 'object',
+        properties: {
+            data: {
+                type: 'string',
+                description: 'JSON 格式的数据（数组或对象）',
+            },
+            operation: {
+                type: 'string',
+                description: '分析操作类型',
+                enum: [
+                    'summary',
+                    'sort',
+                    'filter',
+                    'group',
+                    'transform',
+                    'custom',
+                ],
+            },
+            description: {
+                type: 'string',
+                description: '分析目标的自然语言描述，如"按月份分组计算总营收"',
+            },
+        },
+        required: ['data', 'operation'],
+    },
+    async execute(input) {
+        try {
+            const data = typeof input.data === 'string'
+                ? JSON.parse(input.data)
+                : input.data;
+            if (!Array.isArray(data)) {
+                return { error: 'data 必须是 JSON 数组格式' };
+            }
+            const result = {
+                operation: input.operation,
+                rowCount: data.length,
+            };
+            switch (input.operation) {
+                case 'summary': {
+                    if (!data.length)
+                        return { ...result, summary: '空数据集' };
+                    const numericKeys = Object.keys(data[0]).filter(k => typeof data[0][k] === 'number');
+                    const summary = {};
+                    for (const key of numericKeys) {
+                        const values = data.map(row => Number(row[key] || 0));
+                        summary[key] = {
+                            sum: values.reduce((a, b) => a + b, 0),
+                            avg: values.reduce((a, b) => a + b, 0) /
+                                values.length,
+                            min: Math.min(...values),
+                            max: Math.max(...values),
+                            count: values.length,
+                        };
+                    }
+                    result.summary = summary;
+                    break;
+                }
+                default:
+                    result.hint = `操作 "${input.operation}" 的详细实现待完善，请使用 execute_code 工具自定义分析逻辑。`;
+                    result.sampleData = data.slice(0, 5);
+            }
+            return result;
+        }
+        catch (e) {
+            return { error: `数据分析失败: ${e.message}` };
+        }
+    },
+};

package/dist/tools/datetime.tool.d.ts

@@ -0,0 +1,21 @@
+/**
+ * 日期时间工具
+ *
+ * 让 Agent 能够执行各种日期时间操作：格式化、解析、计算、时区转换等。
+ *
+ * 工具说明：
+ * - datetime_format：日期时间格式化与解析
+ * - datetime_calc：日期时间计算（加减、差值、范围）
+ *
+ * 特性：
+ * - 支持多种输入格式（ISO 8601、时间戳、自然语言描述）
+ * - 支持常见时区转换（UTC、Asia/Shanghai、America/New_York 等）
+ * - 支持日期加减运算（年/月/日/时/分/秒）
+ * - 支持日期差值计算
+ * - 支持工作日/周末判断、季度/周数计算
+ */
+import type { ToolDefinition } from '../types';
+/** 日期时间格式化工具 */
+export declare const datetimeFormatTool: ToolDefinition;
+/** 日期时间计算工具 */
+export declare const datetimeCalcTool: ToolDefinition;