@cicctencent/agent-midway 0.1.1

package/dist/service/ai-chat.service.js

@@ -0,0 +1,431 @@
+"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);
+};
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+var AIChatService_1;
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AIChatService = void 0;
+const core_1 = require("@midwayjs/core");
+const midwayjs_base_1 = require("@cicctencent/midwayjs-base");
+const typeorm_1 = require("@midwayjs/typeorm");
+const typeorm_2 = require("typeorm");
+const ai_chat_workspace_entity_1 = __importDefault(require("../model/ai-chat-workspace.entity"));
+const ai_chat_thread_entity_1 = __importDefault(require("../model/ai-chat-thread.entity"));
+const ai_chat_message_entity_1 = __importDefault(require("../model/ai-chat-message.entity"));
+let AIChatService = AIChatService_1 = class AIChatService extends midwayjs_base_1.BaseService {
+    get workspaceModel() {
+        if (this.wsRepo)
+            return this.wsRepo;
+        return (this.wsRepo = this.dataSource.getRepository(ai_chat_workspace_entity_1.default));
+    }
+    get threadModel() {
+        if (this.thRepo)
+            return this.thRepo;
+        return (this.thRepo =
+            this.dataSource.getRepository(ai_chat_thread_entity_1.default));
+    }
+    get messageModel() {
+        if (this.msgRepo)
+            return this.msgRepo;
+        return (this.msgRepo =
+            this.dataSource.getRepository(ai_chat_message_entity_1.default));
+    }
+    // ========== Workspace ==========
+    async getWorkspaces(userId, applicationId) {
+        return this.workspaceModel.find({
+            where: { userId, applicationId, isDeleted: 0 },
+            order: { createTime: 'ASC' },
+        });
+    }
+    async getWorkspace(id, userId, applicationId) {
+        return this.workspaceModel.findOne({
+            where: { id, userId, applicationId, isDeleted: 0 },
+        });
+    }
+    async createWorkspace(userId, applicationId, data) {
+        const ws = this.workspaceModel.create({
+            ...data,
+            userId,
+            applicationId,
+            threadCount: 0,
+            status: 'active',
+            isDeleted: 0,
+        });
+        return this.workspaceModel.save(ws);
+    }
+    async updateWorkspace(id, userId, applicationId, data) {
+        await this.workspaceModel.update({ id, userId, applicationId, isDeleted: 0 }, data);
+        return this.getWorkspace(id, userId, applicationId);
+    }
+    async deleteWorkspace(id, userId, applicationId) {
+        // 软删除空间下的所有线程和消息
+        const threads = await this.threadModel.find({
+            where: { workspaceId: id, userId, applicationId, isDeleted: 0 },
+        });
+        for (const th of threads) {
+            await this.threadModel.update({ id: th.id }, { isDeleted: 1 });
+            await this.messageModel.update({ threadId: th.id, userId, applicationId, isDeleted: 0 }, { isDeleted: 1 });
+        }
+        await this.workspaceModel.update({ id, userId, applicationId, isDeleted: 0 }, { isDeleted: 1 });
+    }
+    // ========== Thread ==========
+    async getThreads(workspaceId, userId, applicationId) {
+        return this.threadModel.find({
+            where: { workspaceId, userId, applicationId, isDeleted: 0 },
+            order: { createTime: 'DESC' },
+        });
+    }
+    async getThread(id, userId, applicationId) {
+        return this.threadModel.findOne({
+            where: { id, userId, applicationId, isDeleted: 0 },
+        });
+    }
+    async createThread(userId, applicationId, data) {
+        const th = this.threadModel.create({
+            userId,
+            applicationId,
+            workspaceId: data.workspaceId,
+            title: data.title || '新对话',
+            messageCount: 0,
+            status: 'active',
+            isDeleted: 0,
+        });
+        const saved = await this.threadModel.save(th);
+        // 更新工作空间的线程计数
+        await this.workspaceModel.increment({ id: data.workspaceId, userId, applicationId, isDeleted: 0 }, 'threadCount', 1);
+        return saved;
+    }
+    async updateThread(id, userId, applicationId, data) {
+        await this.threadModel.update({ id, userId, applicationId, isDeleted: 0 }, data);
+        return this.getThread(id, userId, applicationId);
+    }
+    async deleteThread(id, userId, applicationId) {
+        const thread = await this.getThread(id, userId, applicationId);
+        if (!thread)
+            return;
+        // 软删除该线程的所有消息
+        await this.messageModel.update({ threadId: id, userId, applicationId, isDeleted: 0 }, { isDeleted: 1 });
+        // 软删除线程
+        await this.threadModel.update({ id, userId, applicationId, isDeleted: 0 }, { isDeleted: 1 });
+        // 减少工作空间的线程计数
+        await this.workspaceModel.decrement({ id: thread.workspaceId, userId, applicationId, isDeleted: 0 }, 'threadCount', 1);
+    }
+    // ========== Message ==========
+    async getMessages(threadId, userId, applicationId, options) {
+        const limit = options?.limit || 50;
+        const offset = options?.offset || 0;
+        const list = await this.messageModel.find({
+            where: { threadId, userId, applicationId, isDeleted: 0 },
+            order: { createTime: 'DESC' },
+            take: limit,
+            skip: offset,
+        });
+        const total = await this.messageModel.count({
+            where: { threadId, userId, applicationId, isDeleted: 0 },
+        });
+        return {
+            items: list.reverse(),
+            total,
+            hasMore: offset + list.length < total,
+        };
+    }
+    async addMessage(userId, applicationId, data) {
+        const msg = this.messageModel.create({
+            userId,
+            applicationId,
+            threadId: data.threadId,
+            role: data.role,
+            content: data.content,
+            type: data.type || 'text',
+            meta: data.meta || null,
+            isDeleted: 0,
+        });
+        const saved = await this.messageModel.save(msg);
+        // 更新线程的消息计数和时间
+        await this.threadModel.increment({ id: data.threadId, userId, applicationId, isDeleted: 0 }, 'messageCount', 1);
+        await this.threadModel.update({ id: data.threadId, userId, applicationId, isDeleted: 0 }, { updater: userId });
+        return saved;
+    }
+    async updateMessage(id, userId, applicationId, data) {
+        await this.messageModel.update({ id, userId, applicationId, isDeleted: 0 }, data);
+        return this.messageModel.findOne({
+            where: { id, userId, applicationId, isDeleted: 0 },
+        });
+    }
+    async deleteMessage(id, userId, applicationId) {
+        const msg = await this.messageModel.findOne({
+            where: { id, userId, applicationId, isDeleted: 0 },
+        });
+        if (!msg)
+            return;
+        await this.messageModel.update({ id, userId, applicationId, isDeleted: 0 }, { isDeleted: 1 });
+        // 减少线程的消息计数
+        await this.threadModel.decrement({ id: msg.threadId, userId, applicationId, isDeleted: 0 }, 'messageCount', 1);
+    }
+    async clearThreadMessages(threadId, userId, applicationId) {
+        await this.messageModel.update({ threadId, userId, applicationId, isDeleted: 0 }, { isDeleted: 1 });
+        await this.threadModel.update({ id: threadId, userId, applicationId, isDeleted: 0 }, { messageCount: 0 });
+    }
+    /** 初始化默认工作空间（新用户首次使用时，可由业务层覆盖） */
+    async ensureDefaultWorkspaces(userId, applicationId) {
+        const count = await this.workspaceModel.count({
+            where: { userId, applicationId, isDeleted: 0 },
+        });
+        if (count > 0)
+            return;
+        // 创建一个通用默认工作空间，业务层可覆盖此方法创建业务特定的工作空间
+        await this.createWorkspace(userId, applicationId, {
+            name: '默认工作空间',
+            description: '默认工作空间',
+            color: '#2d9f6f',
+            icon: 'folder',
+        });
+    }
+    // ========== 搜索 ==========
+    /**
+     * 全局搜索对话消息
+     * @param query 搜索关键词
+     * @param options.workspaceId 限定工作空间
+     * @param options.limit 结果上限
+     */
+    async searchMessages(query, options) {
+        if (!query.trim())
+            return [];
+        const limit = options?.limit || 50;
+        // 查找匹配的消息（使用 LIKE 搜索）
+        const messages = await this.messageModel.find({
+            where: {
+                content: (0, typeorm_2.Like)(`%${query}%`),
+                isDeleted: 0,
+                ...(options?.workspaceId
+                    ? {}
+                    : {}),
+            },
+            order: { createTime: 'DESC' },
+            take: limit,
+        });
+        // 如果指定了 workspaceId，需要过滤
+        let filtered = messages;
+        if (options?.workspaceId) {
+            const threadIds = [...new Set(messages.map(m => m.threadId))];
+            const threads = await this.threadModel.find({
+                where: threadIds.map(id => ({ id, isDeleted: 0 })),
+            });
+            const validThreadIds = new Set(threads
+                .filter(t => t.workspaceId === options.workspaceId)
+                .map(t => t.id));
+            filtered = messages.filter(m => validThreadIds.has(m.threadId));
+        }
+        // 获取线程标题
+        const threadIds = [...new Set(filtered.map(m => m.threadId))];
+        const threads = await this.threadModel.find({
+            where: threadIds.map(id => ({ id, isDeleted: 0 })),
+        });
+        const threadMap = new Map(threads.map(t => [t.id, t]));
+        const queryLower = query.toLowerCase();
+        return filtered.map(msg => {
+            const thread = threadMap.get(msg.threadId);
+            const content = msg.content || '';
+            const idx = content.toLowerCase().indexOf(queryLower);
+            const snippetStart = Math.max(0, idx - 30);
+            const snippetEnd = Math.min(content.length, idx + query.length + 50);
+            const prefix = snippetStart > 0 ? '...' : '';
+            const suffix = snippetEnd < content.length ? '...' : '';
+            return {
+                threadId: msg.threadId,
+                threadTitle: thread?.title || '未命名对话',
+                messageId: msg.id,
+                role: msg.role,
+                content,
+                createTime: msg.createTime,
+                snippet: prefix + content.slice(snippetStart, snippetEnd) + suffix,
+            };
+        });
+    }
+    // ========== 导出 ==========
+    /**
+     * 导出对话为指定格式
+     * @param threadId 线程 ID
+     * @param format 导出格式：markdown / json / text
+     * @param userId 用户 ID
+     * @param applicationId 应用 ID
+     */
+    async exportThread(threadId, format, userId, applicationId) {
+        const thread = await this.getThread(threadId, userId, applicationId);
+        if (!thread)
+            throw new Error('Thread not found');
+        const messages = await this.messageModel.find({
+            where: { threadId, userId, applicationId, isDeleted: 0 },
+            order: { createTime: 'ASC' },
+        });
+        switch (format) {
+            case 'markdown':
+                return this.exportAsMarkdown(thread.title || '未命名对话', messages);
+            case 'json':
+                return JSON.stringify({ thread, messages }, null, 2);
+            case 'text':
+                return this.exportAsText(thread.title || '未命名对话', messages);
+            default:
+                throw new Error(`Unsupported format: ${format}`);
+        }
+    }
+    exportAsMarkdown(title, messages) {
+        const lines = [];
+        lines.push(`# ${title}`);
+        lines.push('');
+        lines.push(`> 导出时间: ${new Date().toLocaleString('zh-CN')}`);
+        lines.push(`> 消息数: ${messages.length}`);
+        lines.push('');
+        lines.push('---');
+        lines.push('');
+        for (const msg of messages) {
+            const time = new Date(msg.createTime).toLocaleString('zh-CN');
+            if (msg.role === 'user') {
+                lines.push(`### 用户 (${time})`);
+                lines.push('');
+                lines.push(msg.content || '');
+            }
+            else if (msg.role === 'assistant') {
+                lines.push(`### 助手 (${time})`);
+                lines.push('');
+                lines.push(msg.content || '');
+                const toolCalls = msg.meta?.toolCalls;
+                if (Array.isArray(toolCalls) && toolCalls.length > 0) {
+                    lines.push('');
+                    lines.push('<details><summary>工具调用</summary>');
+                    lines.push('');
+                    for (const tc of toolCalls) {
+                        lines.push(`- **${tc.displayName || tc.toolName}**: ${tc.status} (${tc.duration || 0}ms)`);
+                    }
+                    lines.push('');
+                    lines.push('</details>');
+                }
+            }
+            lines.push('');
+            lines.push('---');
+            lines.push('');
+        }
+        return lines.join('\n');
+    }
+    exportAsText(title, messages) {
+        const lines = [];
+        lines.push(`=== ${title} ===`);
+        lines.push(`导出时间: ${new Date().toLocaleString('zh-CN')}`);
+        lines.push('');
+        for (const msg of messages) {
+            const time = new Date(msg.createTime).toLocaleString('zh-CN');
+            const role = msg.role === 'user' ? '用户' : '助手';
+            lines.push(`[${role} ${time}]`);
+            lines.push(msg.content || '');
+            lines.push('');
+        }
+        return lines.join('\n');
+    }
+    /**
+     * 判断 assistant 消息是否应排除出历史（精简 token）
+     */
+    shouldSkipMessage(msg) {
+        const toolCalls = msg.meta?.toolCalls;
+        const hasToolCalls = toolCalls && toolCalls.length > 0;
+        const content = (msg.content || '').trim();
+        if (!hasToolCalls && content.length < 5)
+            return true;
+        if (hasToolCalls && toolCalls.every(tc => tc.status === 'error'))
+            return true;
+        if (msg.meta?.isError || msg.meta?.error)
+            return true;
+        return false;
+    }
+    /**
+     * 从已持久化的消息记录重建 LLM 格式的对话历史（用于 AgentEngine）
+     *
+     * @param threadId 线程 ID
+     * @param userId 用户 ID
+     * @param applicationId 应用 ID
+     * @param limit 最大 DB 消息数，默认 20
+     * @returns Message[] 格式的对话历史
+     */
+    async buildThreadHistory(threadId, userId, applicationId, limit = 20) {
+        const { items } = await this.getMessages(threadId, userId, applicationId, { limit });
+        if (items.length === 0)
+            return [];
+        const result = [];
+        for (const msg of items) {
+            if (msg.role === 'user') {
+                const content = (msg.content || '').trim();
+                if (content.length < 3)
+                    continue;
+                result.push({ role: 'user', content: msg.content });
+            }
+            else if (msg.role === 'assistant') {
+                if (this.shouldSkipMessage(msg))
+                    continue;
+                let content = msg.content || '';
+                if (msg.meta?.error && content) {
+                    content = `${content}\n\n[执行出错: ${msg.meta.error}]`;
+                }
+                else if (msg.meta?.error) {
+                    content = `[执行出错: ${msg.meta.error}]`;
+                }
+                const toolCalls = msg.meta?.toolCalls;
+                if (toolCalls && toolCalls.length > 0) {
+                    // assistant 消息带 tool_calls
+                    result.push({ role: 'assistant', content });
+                    for (const tc of toolCalls) {
+                        let output;
+                        if (tc.output == null || tc.output === '') {
+                            output = '';
+                        }
+                        else if (typeof tc.output === 'string') {
+                            output = tc.output;
+                        }
+                        else {
+                            output = JSON.stringify(tc.output);
+                        }
+                        if (tc.status === 'error') {
+                            output = `[工具调用失败] ${tc.error || output || 'Unknown error'}`;
+                        }
+                        const truncated = output.length > AIChatService_1.MAX_TOOL_OUTPUT_LEN
+                            ? output.slice(0, AIChatService_1.MAX_TOOL_OUTPUT_LEN) + `\n... [truncated]`
+                            : output;
+                        result.push({ role: 'tool', content: truncated });
+                    }
+                }
+                else {
+                    result.push({ role: 'assistant', content });
+                }
+            }
+        }
+        // 硬上限
+        if (result.length > AIChatService_1.MAX_RESULT_MESSAGES) {
+            const trimmed = result.slice(result.length - AIChatService_1.MAX_RESULT_MESSAGES);
+            while (trimmed.length > 0 && trimmed[0].role === 'tool') {
+                trimmed.shift();
+            }
+            return trimmed;
+        }
+        return result;
+    }
+};
+exports.AIChatService = AIChatService;
+// ========== 对话历史重建 ==========
+/** 单条工具输出截断长度 */
+AIChatService.MAX_TOOL_OUTPUT_LEN = 4000;
+/** 最终 LLM 消息数硬上限 */
+AIChatService.MAX_RESULT_MESSAGES = 60;
+__decorate([
+    (0, typeorm_1.InjectDataSource)('default'),
+    __metadata("design:type", typeorm_2.DataSource)
+], AIChatService.prototype, "dataSource", void 0);
+exports.AIChatService = AIChatService = AIChatService_1 = __decorate([
+    (0, core_1.Provide)()
+], AIChatService);

package/dist/service/ai-skill.service.d.ts

@@ -0,0 +1,116 @@
+import { Application } from '@midwayjs/koa';
+import type { ToolDefinition } from '../types';
+/**
+ * 从 Markdown 文件加载的 Skill 数据结构
+ */
+export interface FileSkill {
+    id: number;
+    name: string;
+    description: string;
+    category: string;
+    contentType: string;
+    content: string;
+    agentTypes: string;
+    injectionMode: string;
+    sort: number;
+    isActive: number;
+    isDeleted: number;
+    toolSchema?: any;
+    /** 源文件路径（相对于 skills 目录） */
+    filePath: string;
+}
+/**
+ * AI Skill 管理 Service（本地文件版）
+ *
+ * 从 data/skills/ 目录递归加载 Markdown 文件作为 Skills。
+ * 每个 .md 文件使用 YAML frontmatter 定义元数据，正文为 Skill 内容。
+ *
+ * 支持两种注入方式：
+ * - prompt: 将 skill 内容拼接到 system prompt
+ * - tool: 将 skill 注册为 Tool，AI 主动调用获取内容
+ * - both: 两者兼具
+ *
+ * 缓存策略：
+ * - 文件加载结果缓存在内存中，TTL 5 分钟
+ * - 文件变更后自动刷新缓存
+ */
+export declare class AISkillService {
+    app: Application;
+    /** skills 根目录 */
+    private skillsDir;
+    /** 加载的 Skills */
+    private skills;
+    /** 自增 ID 计数器 */
+    private nextId;
+    /** 缓存 */
+    private static cache;
+    private static readonly CACHE_TTL;
+    /** 上次扫描时间 */
+    private _lastScanTime;
+    get lastScanTime(): number;
+    init(): Promise<void>;
+    /** 重新扫描并加载所有 Skills */
+    reloadSkills(): Promise<void>;
+    /** 递归扫描目录获取所有 .md 文件 */
+    private scanDirectory;
+    /** 解析单个 Skill 文件 */
+    private parseSkillFile;
+    /** 解析 YAML frontmatter */
+    private parseFrontmatter;
+    private clearCache;
+    private getFromCache;
+    private setCache;
+    /** 查询所有 Skills */
+    getSkills(options?: {
+        category?: string;
+        agentType?: string;
+        activeOnly?: boolean;
+    }): Promise<FileSkill[]>;
+    /** 获取单个 Skill */
+    getSkill(id: number): Promise<FileSkill | null>;
+    /** 根据 name 获取 Skill */
+    getSkillByName(name: string): Promise<FileSkill | null>;
+    /** 创建 Skill（写入新文件） */
+    createSkill(data: Partial<FileSkill>): Promise<FileSkill>;
+    /** 更新 Skill（覆写文件） */
+    updateSkill(id: number, data: Partial<FileSkill>): Promise<FileSkill>;
+    /** 删除 Skill（删除文件） */
+    deleteSkill(id: number): Promise<void>;
+    /** 生成 Skill 文件内容 */
+    private buildSkillFileContent;
+    /** 确保 Skills 已加载 */
+    private ensureLoaded;
+    /**
+     * 获取指定 Agent 类型的 Skills
+     */
+    getSkillsForAgent(agentType: string, injectionMode?: 'prompt' | 'tool' | 'both'): Promise<FileSkill[]>;
+    /**
+     * 构建 Skill Context（用于注入到 system prompt）
+     */
+    buildSkillContext(agentType: string): Promise<string>;
+    /**
+     * 将 Skills 转换为 ToolDefinition（供 Agent 自动调用）
+     */
+    getSkillsAsTools(agentType: string): Promise<ToolDefinition[]>;
+    /**
+     * 预览指定 Agent 的 Skills Context
+     */
+    previewSkillContext(agentType: string): Promise<{
+        agentType: string;
+        total: number;
+        promptSkills: number;
+        toolSkills: number;
+        context: string;
+        tools: Array<{
+            id: string;
+            name: string;
+            description: string;
+        }>;
+    }>;
+    /** 判断 skill 是否匹配指定 agent 类型 */
+    private matchesAgentType;
+    /** 按 category 分组 */
+    private groupByCategory;
+    /** 将 Skill 转换为 ToolDefinition */
+    private skillToTool;
+}