@cicctencent/agent-midway 0.1.1

package/dist/controller/agent.controller.js

@@ -0,0 +1,414 @@
+"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 __param = (this && this.__param) || function (paramIndex, decorator) {
+    return function (target, key) { decorator(target, key, paramIndex); }
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AgentController = void 0;
+const core_1 = require("@midwayjs/core");
+const midwayjs_base_1 = require("@cicctencent/midwayjs-base");
+const agent_dto_1 = require("../dto/agent.dto");
+const agent_server_service_1 = require("../service/agent-server.service");
+const agent_task_queue_service_1 = require("../service/agent-task-queue.service");
+const application_service_1 = require("../service/application.service");
+/**
+ * Agent 运行 Controller
+ *
+ * 提供 Agent 运行接口，支持流式和非流式响应（按 Application 隔离）
+ */
+let AgentController = class AgentController extends midwayjs_base_1.BaseController {
+    /** 获取当前应用ID */
+    async getApplicationId() {
+        const app = await this.applicationService.getDefault();
+        return app?.id || 0;
+    }
+    /**
+     * 运行 Agent（流式响应）
+     */
+    async runAgent(body) {
+        const { prompt, threadId, profileId, maxSteps, timeout } = body;
+        // 设置 SSE 响应头
+        this.ctx.set('Cache-Control', 'no-cache');
+        this.ctx.set('Connection', 'keep-alive');
+        this.ctx.set('X-Accel-Buffering', 'no');
+        const userId = this.getUserId();
+        const applicationId = await this.getApplicationId();
+        const stream = body.stream !== false;
+        try {
+            // 运行 Agent
+            const generator = this.agentServerService.runAgentStream(prompt, {
+                userId,
+                applicationId,
+                threadId,
+                profileId,
+                stream,
+                maxSteps,
+                timeout,
+            });
+            // 流式输出
+            for await (const event of generator) {
+                this.writeSSE(event);
+            }
+            // 结束响应
+            this.ctx.res.end();
+        }
+        catch (error) {
+            // 发送错误事件
+            this.writeSSE({
+                type: 'error',
+                data: { message: error.message },
+                timestamp: Date.now(),
+            });
+            this.ctx.res.end();
+        }
+    }
+    /**
+     * 运行 Agent（非流式响应）
+     */
+    async runAgentSync(body) {
+        try {
+            const { prompt, threadId, profileId, maxSteps, timeout } = body;
+            const userId = this.getUserId();
+            const applicationId = await this.getApplicationId();
+            const result = await this.agentServerService.runAgent(prompt, {
+                userId,
+                applicationId,
+                threadId,
+                profileId,
+                stream: false,
+                maxSteps,
+                timeout,
+            });
+            return { ret: 0, data: result, msg: 'success' };
+        }
+        catch (error) {
+            return { ret: 1, msg: error.message };
+        }
+    }
+    /**
+     * 提交异步任务
+     */
+    async submitTask(body) {
+        try {
+            const { prompt, threadId, profileId, maxSteps, timeout } = body;
+            const userId = this.getUserId();
+            const applicationId = await this.getApplicationId();
+            const taskId = await this.taskQueueService.submit(this.agentServerService, prompt, userId, {
+                userId,
+                applicationId,
+                threadId,
+                profileId,
+                maxSteps,
+                timeout,
+            });
+            return { ret: 0, data: { taskId }, msg: 'success' };
+        }
+        catch (error) {
+            return { ret: 1, msg: error.message };
+        }
+    }
+    /**
+     * 获取任务状态
+     */
+    async getTaskStatus(taskId) {
+        try {
+            const task = this.taskQueueService.getTask(taskId);
+            if (!task) {
+                return { ret: 1, msg: '任务不存在' };
+            }
+            return { ret: 0, data: task, msg: 'success' };
+        }
+        catch (error) {
+            return { ret: 1, msg: error.message };
+        }
+    }
+    /**
+     * 取消任务
+     */
+    async cancelTask(taskId) {
+        try {
+            const cancelled = this.taskQueueService.cancelTask(taskId);
+            return {
+                ret: 0,
+                data: { cancelled },
+                msg: cancelled ? '任务已取消' : '无法取消任务',
+            };
+        }
+        catch (error) {
+            return { ret: 1, msg: error.message };
+        }
+    }
+    /**
+     * 获取队列状态
+     */
+    async getQueueStatus() {
+        try {
+            return {
+                ret: 0,
+                data: this.taskQueueService.getStatus(),
+                msg: 'success',
+            };
+        }
+        catch (error) {
+            return { ret: 1, msg: error.message };
+        }
+    }
+    /**
+     * 获取用户任务列表
+     */
+    async getUserTasks() {
+        try {
+            const userId = this.getUserId();
+            const tasks = this.taskQueueService.getUserTasks(userId);
+            return { ret: 0, data: tasks, msg: 'success' };
+        }
+        catch (error) {
+            return { ret: 1, msg: error.message };
+        }
+    }
+    // ========== Agent 高级操作 ==========
+    /**
+     * 中止 Agent 执行
+     */
+    async abortAgent(threadId) {
+        try {
+            const userId = this.getUserId();
+            await this.agentServerService.abortAgent(threadId, userId);
+            return { ret: 0, data: null, msg: 'Agent 已中止' };
+        }
+        catch (error) {
+            return { ret: 1, msg: error.message };
+        }
+    }
+    /**
+     * SSE 断线重连
+     */
+    async reconnectAgent(threadId) {
+        this.ctx.set('Cache-Control', 'no-cache');
+        this.ctx.set('Connection', 'keep-alive');
+        this.ctx.set('X-Accel-Buffering', 'no');
+        try {
+            const userId = this.getUserId();
+            const generator = this.agentServerService.reconnectStream(threadId, userId);
+            for await (const event of generator) {
+                this.writeSSE(event);
+            }
+            this.ctx.res.end();
+        }
+        catch (error) {
+            this.writeSSE({
+                type: 'error',
+                data: { message: error.message },
+                timestamp: Date.now(),
+            });
+            this.ctx.res.end();
+        }
+    }
+    /**
+     * 编辑消息后重新生成
+     */
+    async editMessage(body) {
+        try {
+            const userId = this.getUserId();
+            const applicationId = await this.getApplicationId();
+            const result = await this.agentServerService.editAndRegenerate(body.threadId, body.messageId, body.content, { userId, applicationId });
+            return { ret: 0, data: result, msg: 'success' };
+        }
+        catch (error) {
+            return { ret: 1, msg: error.message };
+        }
+    }
+    /**
+     * 重新生成最后一条 AI 回复
+     */
+    async regenerate(body) {
+        try {
+            const userId = this.getUserId();
+            const applicationId = await this.getApplicationId();
+            const result = await this.agentServerService.regenerate(body.threadId, { userId, applicationId });
+            return { ret: 0, data: result, msg: 'success' };
+        }
+        catch (error) {
+            return { ret: 1, msg: error.message };
+        }
+    }
+    /**
+     * 从指定位置分叉创建新对话
+     */
+    async forkThread(body) {
+        try {
+            const userId = this.getUserId();
+            const applicationId = await this.getApplicationId();
+            const result = await this.agentServerService.forkThread(body.threadId, body.messageId, { userId, applicationId });
+            return { ret: 0, data: result, msg: 'success' };
+        }
+        catch (error) {
+            return { ret: 1, msg: error.message };
+        }
+    }
+    /**
+     * 引擎工具列表诊断
+     */
+    async diagnose() {
+        try {
+            const tools = this.agentServerService.getTools();
+            return {
+                ret: 0,
+                data: {
+                    toolCount: tools.length,
+                    tools: tools.map(t => ({
+                        name: t.name,
+                        description: t.description,
+                    })),
+                    available: this.agentServerService.isAvailable(),
+                },
+                msg: 'success',
+            };
+        }
+        catch (error) {
+            return { ret: 1, msg: error.message };
+        }
+    }
+    // ========== 私有方法 ==========
+    /**
+     * 写入 SSE 事件
+     */
+    writeSSE(event) {
+        const data = `data: ${JSON.stringify(event)}\n\n`;
+        this.ctx.res.write(data);
+    }
+    /**
+     * 获取当前用户 ID
+     */
+    getUserId() {
+        return String(this.ctx?.currentSession?.userId || 'anonymous');
+    }
+};
+exports.AgentController = AgentController;
+__decorate([
+    (0, core_1.Inject)(),
+    __metadata("design:type", agent_server_service_1.AgentServerService)
+], AgentController.prototype, "agentServerService", void 0);
+__decorate([
+    (0, core_1.Inject)(),
+    __metadata("design:type", agent_task_queue_service_1.AgentTaskQueueService)
+], AgentController.prototype, "taskQueueService", void 0);
+__decorate([
+    (0, core_1.Inject)(),
+    __metadata("design:type", application_service_1.ApplicationService)
+], AgentController.prototype, "applicationService", void 0);
+__decorate([
+    (0, core_1.Post)('/run'),
+    (0, core_1.ContentType)('text/event-stream'),
+    midwayjs_base_1.decorators.checkLogin(true),
+    __param(0, (0, core_1.Body)()),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", [agent_dto_1.RunAgentDTO]),
+    __metadata("design:returntype", Promise)
+], AgentController.prototype, "runAgent", null);
+__decorate([
+    (0, core_1.Post)('/run/sync'),
+    midwayjs_base_1.decorators.checkLogin(true),
+    __param(0, (0, core_1.Body)()),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", [agent_dto_1.RunAgentDTO]),
+    __metadata("design:returntype", Promise)
+], AgentController.prototype, "runAgentSync", null);
+__decorate([
+    (0, core_1.Post)('/task'),
+    midwayjs_base_1.decorators.checkLogin(true),
+    __param(0, (0, core_1.Body)()),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", [Object]),
+    __metadata("design:returntype", Promise)
+], AgentController.prototype, "submitTask", null);
+__decorate([
+    (0, core_1.Get)('/task/:taskId'),
+    midwayjs_base_1.decorators.checkLogin(true),
+    __param(0, (0, core_1.Param)('taskId')),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", [String]),
+    __metadata("design:returntype", Promise)
+], AgentController.prototype, "getTaskStatus", null);
+__decorate([
+    (0, core_1.Post)('/task/:taskId/cancel'),
+    midwayjs_base_1.decorators.checkLogin(true),
+    __param(0, (0, core_1.Param)('taskId')),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", [String]),
+    __metadata("design:returntype", Promise)
+], AgentController.prototype, "cancelTask", null);
+__decorate([
+    (0, core_1.Get)('/queue/status'),
+    midwayjs_base_1.decorators.checkLogin(true),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", []),
+    __metadata("design:returntype", Promise)
+], AgentController.prototype, "getQueueStatus", null);
+__decorate([
+    (0, core_1.Get)('/tasks'),
+    midwayjs_base_1.decorators.checkLogin(true),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", []),
+    __metadata("design:returntype", Promise)
+], AgentController.prototype, "getUserTasks", null);
+__decorate([
+    (0, core_1.Post)('/abort/:threadId'),
+    midwayjs_base_1.decorators.checkLogin(true),
+    __param(0, (0, core_1.Param)('threadId')),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", [Number]),
+    __metadata("design:returntype", Promise)
+], AgentController.prototype, "abortAgent", null);
+__decorate([
+    (0, core_1.Get)('/reconnect/:threadId'),
+    (0, core_1.ContentType)('text/event-stream'),
+    midwayjs_base_1.decorators.checkLogin(true),
+    __param(0, (0, core_1.Param)('threadId')),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", [Number]),
+    __metadata("design:returntype", Promise)
+], AgentController.prototype, "reconnectAgent", null);
+__decorate([
+    (0, core_1.Post)('/edit'),
+    midwayjs_base_1.decorators.checkLogin(true),
+    __param(0, (0, core_1.Body)()),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", [Object]),
+    __metadata("design:returntype", Promise)
+], AgentController.prototype, "editMessage", null);
+__decorate([
+    (0, core_1.Post)('/regenerate'),
+    midwayjs_base_1.decorators.checkLogin(true),
+    __param(0, (0, core_1.Body)()),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", [Object]),
+    __metadata("design:returntype", Promise)
+], AgentController.prototype, "regenerate", null);
+__decorate([
+    (0, core_1.Post)('/fork'),
+    midwayjs_base_1.decorators.checkLogin(true),
+    __param(0, (0, core_1.Body)()),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", [Object]),
+    __metadata("design:returntype", Promise)
+], AgentController.prototype, "forkThread", null);
+__decorate([
+    (0, core_1.Get)('/diagnose'),
+    midwayjs_base_1.decorators.checkLogin(true),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", []),
+    __metadata("design:returntype", Promise)
+], AgentController.prototype, "diagnose", null);
+exports.AgentController = AgentController = __decorate([
+    (0, core_1.Provide)(),
+    (0, core_1.Controller)('/api/agent')
+], AgentController);

package/dist/controller/application.controller.d.ts

@@ -0,0 +1,113 @@
+import { BaseController } from '@cicctencent/midwayjs-base';
+import { ApplicationService } from '../service/application.service';
+import { CreateApplicationDTO } from '../dto/ai-config.dto';
+/**
+ * AI 应用管理 Controller
+ */
+export declare class ApplicationController extends BaseController {
+    appService: ApplicationService;
+    /** 获取应用列表 */
+    list(): Promise<{
+        ret: number;
+        data: import("..").AIApplicationEntity[];
+        msg: string;
+    } | {
+        ret: number;
+        msg: any;
+        data?: undefined;
+    }>;
+    /** 获取单个应用 */
+    detail(id: string): Promise<{
+        ret: number;
+        data: import("..").AIApplicationEntity;
+        msg: string;
+    } | {
+        ret: number;
+        msg: any;
+        data?: undefined;
+    }>;
+    /** 获取应用完整配置 */
+    fullConfig(id: string): Promise<{
+        ret: number;
+        data: {
+            application: import("..").AIApplicationEntity;
+            agentProfiles: any[];
+            defaultProfile: any;
+        };
+        msg: string;
+    } | {
+        ret: number;
+        msg: any;
+        data?: undefined;
+    }>;
+    /** 获取默认应用 */
+    getDefault(): Promise<{
+        ret: number;
+        data: import("..").AIApplicationEntity;
+        msg: string;
+    } | {
+        ret: number;
+        msg: any;
+        data?: undefined;
+    }>;
+    /** 创建应用 */
+    create(body: any): Promise<{
+        ret: number;
+        data: import("..").AIApplicationEntity;
+        msg: string;
+    } | {
+        ret: number;
+        msg: any;
+        data?: undefined;
+    }>;
+    /** 更新应用 */
+    update(id: number, body: Partial<CreateApplicationDTO>): Promise<{
+        ret: number;
+        data: import("..").AIApplicationEntity;
+        msg: string;
+    } | {
+        ret: number;
+        msg: any;
+        data?: undefined;
+    }>;
+    /** 删除应用 */
+    remove(id: number): Promise<{
+        ret: number;
+        data: any;
+        msg: string;
+    } | {
+        ret: number;
+        msg: any;
+        data?: undefined;
+    }>;
+    /** 设置为默认 */
+    setDefault(id: number): Promise<{
+        ret: number;
+        data: import("..").AIApplicationEntity;
+        msg: string;
+    } | {
+        ret: number;
+        msg: any;
+        data?: undefined;
+    }>;
+    /** 归档应用 */
+    archive(id: number): Promise<{
+        ret: number;
+        data: import("..").AIApplicationEntity;
+        msg: string;
+    } | {
+        ret: number;
+        msg: any;
+        data?: undefined;
+    }>;
+    /** 激活应用 */
+    activate(id: number): Promise<{
+        ret: number;
+        data: import("..").AIApplicationEntity;
+        msg: string;
+    } | {
+        ret: number;
+        msg: any;
+        data?: undefined;
+    }>;
+}