@cicctencent/agent-midway 0.1.1

package/dist/controller/a2a.controller.js

@@ -0,0 +1,150 @@
+"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.A2AController = void 0;
+/**
+ * A2A 协议控制器
+ *
+ * 路由说明：
+ * - GET  /.well-known/agent.json  — Agent Card 发现（不需要登录）
+ * - POST /a2a                     — JSON-RPC 分发入口（不需要登录，支持 SSE 流式）
+ * - POST /api/a2a/discover         — 发现远程 Agent（需要登录）
+ *
+ * 注意：`/.well-known/agent.json` 和 `/a2a` 挂载在根路径
+ */
+const core_1 = require("@midwayjs/core");
+const midwayjs_base_1 = require("@cicctencent/midwayjs-base");
+const a2a_service_1 = require("../service/a2a.service");
+let A2AController = class A2AController extends midwayjs_base_1.BaseController {
+    /**
+     * Agent Card 发现端点
+     * GET /.well-known/agent.json
+     */
+    async getAgentCard() {
+        const baseUrl = `${this.ctx.protocol}://${this.ctx.host}`;
+        const card = this.a2aService.generateAgentCard(baseUrl);
+        return card;
+    }
+    /**
+     * JSON-RPC 分发入口
+     * POST /a2a
+     * 支持 SSE 流式响应（当 method 为 tasks/sendSubscribe 时）
+     */
+    async handleJsonRpc(body) {
+        // 检查是否为流式请求
+        const acceptHeader = this.ctx.headers['accept'] || '';
+        const isStreaming = body?.method === 'tasks/sendSubscribe' ||
+            acceptHeader.includes('text/event-stream');
+        if (isStreaming) {
+            return this.handleStreaming(body);
+        }
+        // 非流式：直接返回 JSON-RPC 响应
+        const response = await this.a2aService.handleJsonRpc(body);
+        return response;
+    }
+    /**
+     * 发现远程 Agent
+     * POST /api/a2a/discover
+     */
+    async discoverAgent(body) {
+        try {
+            if (!body?.url) {
+                return { ret: 1, msg: 'URL is required' };
+            }
+            const card = await this.a2aService.discoverRemoteAgent(body.url, body.auth);
+            return { ret: 0, data: card, msg: 'success' };
+        }
+        catch (error) {
+            return { ret: 1, msg: error.message };
+        }
+    }
+    /**
+     * 处理流式 JSON-RPC 请求（SSE）
+     */
+    async handleStreaming(body) {
+        this.ctx.set('Cache-Control', 'no-cache');
+        this.ctx.set('Connection', 'keep-alive');
+        this.ctx.set('X-Accel-Buffering', 'no');
+        try {
+            const params = body?.params || {};
+            const generator = this.a2aService.handleSendTaskStreaming(params);
+            for await (const event of generator) {
+                const sseData = this.formatSSEEvent(event);
+                this.ctx.res.write(sseData);
+            }
+        }
+        catch (error) {
+            const errorEvent = this.formatSSEEvent({
+                type: 'error',
+                data: { message: error.message },
+            });
+            this.ctx.res.write(errorEvent);
+        }
+        finally {
+            this.ctx.res.end();
+        }
+    }
+    /**
+     * 格式化 SSE 事件
+     */
+    formatSSEEvent(event) {
+        const jsonRpcResponse = {
+            jsonrpc: '2.0',
+            id: 0,
+            result: {
+                type: event.type,
+                data: event.data,
+            },
+        };
+        return `data: ${JSON.stringify(jsonRpcResponse)}\n\n`;
+    }
+};
+exports.A2AController = A2AController;
+__decorate([
+    (0, core_1.Inject)(),
+    __metadata("design:type", a2a_service_1.A2AService)
+], A2AController.prototype, "a2aService", void 0);
+__decorate([
+    (0, core_1.Get)('.well-known/agent.json'),
+    midwayjs_base_1.decorators.checkLogin(false),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", []),
+    __metadata("design:returntype", Promise)
+], A2AController.prototype, "getAgentCard", null);
+__decorate([
+    (0, core_1.Post)('a2a'),
+    midwayjs_base_1.decorators.checkLogin(false),
+    __param(0, (0, core_1.Body)()),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", [Object]),
+    __metadata("design:returntype", Promise)
+], A2AController.prototype, "handleJsonRpc", null);
+__decorate([
+    (0, core_1.Post)('api/a2a/discover'),
+    midwayjs_base_1.decorators.checkLogin(true),
+    __param(0, (0, core_1.Body)()),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", [Object]),
+    __metadata("design:returntype", Promise)
+], A2AController.prototype, "discoverAgent", null);
+__decorate([
+    (0, core_1.ContentType)('text/event-stream'),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", [Object]),
+    __metadata("design:returntype", Promise)
+], A2AController.prototype, "handleStreaming", null);
+exports.A2AController = A2AController = __decorate([
+    (0, core_1.Provide)(),
+    (0, core_1.Controller)('/')
+], A2AController);

package/dist/controller/agent-profile.controller.d.ts

@@ -0,0 +1,97 @@
+import { BaseController } from '@cicctencent/midwayjs-base';
+import { CreateAgentProfileDTO } from '../dto/ai-config.dto';
+import { AgentProfileService } from '../service/agent-profile.service';
+import { ApplicationService } from '../service/application.service';
+/**
+ * Agent Profile 管理 Controller
+ */
+export declare class AgentProfileController extends BaseController {
+    profileService: AgentProfileService;
+    applicationService: ApplicationService;
+    /** 获取当前应用ID */
+    private getApplicationId;
+    /** 获取 Profile 列表 */
+    list(): Promise<{
+        ret: number;
+        data: import("..").AIAgentProfileEntity[];
+        msg: string;
+    } | {
+        ret: number;
+        msg: any;
+        data?: undefined;
+    }>;
+    /** 获取单个 Profile */
+    detail(id: number): Promise<{
+        ret: number;
+        data: import("..").AIAgentProfileEntity;
+        msg: string;
+    } | {
+        ret: number;
+        msg: any;
+        data?: undefined;
+    }>;
+    /** 获取 Profile 完整配置 */
+    fullConfig(id: number): Promise<{
+        ret: number;
+        data: {
+            profile: import("..").AIAgentProfileEntity;
+            mcpServers: any[];
+            skills: any[];
+        };
+        msg: string;
+    } | {
+        ret: number;
+        msg: any;
+        data?: undefined;
+    }>;
+    /** 获取默认 Profile */
+    getDefault(): Promise<{
+        ret: number;
+        data: import("..").AIAgentProfileEntity;
+        msg: string;
+    } | {
+        ret: number;
+        msg: any;
+        data?: undefined;
+    }>;
+    /** 创建 Profile */
+    create(body: CreateAgentProfileDTO): Promise<{
+        ret: number;
+        data: import("..").AIAgentProfileEntity;
+        msg: string;
+    } | {
+        ret: number;
+        msg: any;
+        data?: undefined;
+    }>;
+    /** 更新 Profile */
+    update(id: number, body: any): Promise<{
+        ret: number;
+        data: import("..").AIAgentProfileEntity;
+        msg: string;
+    } | {
+        ret: number;
+        msg: any;
+        data?: undefined;
+    }>;
+    /** 删除 Profile */
+    remove(id: number): Promise<{
+        ret: number;
+        data: any;
+        msg: string;
+    } | {
+        ret: number;
+        msg: any;
+        data?: undefined;
+    }>;
+    /** 设置为默认 */
+    setDefault(id: number): Promise<{
+        ret: number;
+        data: import("..").AIAgentProfileEntity;
+        msg: string;
+    } | {
+        ret: number;
+        msg: any;
+        data?: undefined;
+    }>;
+}

package/dist/controller/agent-profile.controller.js

@@ -0,0 +1,200 @@
+"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.AgentProfileController = void 0;
+const core_1 = require("@midwayjs/core");
+const midwayjs_base_1 = require("@cicctencent/midwayjs-base");
+const ai_config_dto_1 = require("../dto/ai-config.dto");
+const agent_profile_service_1 = require("../service/agent-profile.service");
+const application_service_1 = require("../service/application.service");
+/**
+ * Agent Profile 管理 Controller
+ */
+let AgentProfileController = class AgentProfileController extends midwayjs_base_1.BaseController {
+    /** 获取当前应用ID */
+    async getApplicationId() {
+        const app = await this.applicationService.getDefault();
+        return app?.id || 0;
+    }
+    /** 获取 Profile 列表 */
+    async list() {
+        try {
+            const applicationId = await this.getApplicationId();
+            const profiles = await this.profileService.list(applicationId);
+            return { ret: 0, data: profiles, msg: 'success' };
+        }
+        catch (error) {
+            return { ret: 1, msg: error.message };
+        }
+    }
+    /** 获取单个 Profile */
+    async detail(id) {
+        try {
+            const applicationId = await this.getApplicationId();
+            const profile = await this.profileService.get(Number(id), applicationId);
+            if (!profile) {
+                return { ret: 1, msg: 'Profile 不存在' };
+            }
+            return { ret: 0, data: profile, msg: 'success' };
+        }
+        catch (error) {
+            return { ret: 1, msg: error.message };
+        }
+    }
+    /** 获取 Profile 完整配置 */
+    async fullConfig(id) {
+        try {
+            const applicationId = await this.getApplicationId();
+            const config = await this.profileService.getFullConfig(Number(id), applicationId);
+            if (!config) {
+                return { ret: 1, msg: 'Profile 不存在' };
+            }
+            return { ret: 0, data: config, msg: 'success' };
+        }
+        catch (error) {
+            return { ret: 1, msg: error.message };
+        }
+    }
+    /** 获取默认 Profile */
+    async getDefault() {
+        try {
+            const applicationId = await this.getApplicationId();
+            const profile = await this.profileService.getDefault(applicationId);
+            return { ret: 0, data: profile, msg: 'success' };
+        }
+        catch (error) {
+            return { ret: 1, msg: error.message };
+        }
+    }
+    /** 创建 Profile */
+    async create(body) {
+        try {
+            const applicationId = await this.getApplicationId();
+            const profile = await this.profileService.create(applicationId, body);
+            return { ret: 0, data: profile, msg: 'success' };
+        }
+        catch (error) {
+            return { ret: 1, msg: error.message };
+        }
+    }
+    /** 更新 Profile */
+    async update(id, body) {
+        try {
+            const applicationId = await this.getApplicationId();
+            const profile = await this.profileService.update(Number(id), applicationId, body);
+            return { ret: 0, data: profile, msg: 'success' };
+        }
+        catch (error) {
+            return { ret: 1, msg: error.message };
+        }
+    }
+    /** 删除 Profile */
+    async remove(id) {
+        try {
+            const applicationId = await this.getApplicationId();
+            await this.profileService.delete(Number(id), applicationId);
+            return { ret: 0, data: null, msg: 'success' };
+        }
+        catch (error) {
+            return { ret: 1, msg: error.message };
+        }
+    }
+    /** 设置为默认 */
+    async setDefault(id) {
+        try {
+            const applicationId = await this.getApplicationId();
+            const profile = await this.profileService.setDefault(Number(id), applicationId);
+            return { ret: 0, data: profile, msg: 'success' };
+        }
+        catch (error) {
+            return { ret: 1, msg: error.message };
+        }
+    }
+};
+exports.AgentProfileController = AgentProfileController;
+__decorate([
+    (0, core_1.Inject)(),
+    __metadata("design:type", agent_profile_service_1.AgentProfileService)
+], AgentProfileController.prototype, "profileService", void 0);
+__decorate([
+    (0, core_1.Inject)(),
+    __metadata("design:type", application_service_1.ApplicationService)
+], AgentProfileController.prototype, "applicationService", void 0);
+__decorate([
+    (0, core_1.Get)('/'),
+    midwayjs_base_1.decorators.checkLogin(true),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", []),
+    __metadata("design:returntype", Promise)
+], AgentProfileController.prototype, "list", null);
+__decorate([
+    (0, core_1.Get)('/:id'),
+    midwayjs_base_1.decorators.checkLogin(true),
+    __param(0, (0, core_1.Param)('id')),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", [Number]),
+    __metadata("design:returntype", Promise)
+], AgentProfileController.prototype, "detail", null);
+__decorate([
+    (0, core_1.Get)('/:id/full'),
+    midwayjs_base_1.decorators.checkLogin(true),
+    __param(0, (0, core_1.Param)('id')),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", [Number]),
+    __metadata("design:returntype", Promise)
+], AgentProfileController.prototype, "fullConfig", null);
+__decorate([
+    (0, core_1.Get)('/default'),
+    midwayjs_base_1.decorators.checkLogin(true),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", []),
+    __metadata("design:returntype", Promise)
+], AgentProfileController.prototype, "getDefault", null);
+__decorate([
+    (0, core_1.Post)('/'),
+    midwayjs_base_1.decorators.checkLogin(true),
+    __param(0, (0, core_1.Body)()),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", [ai_config_dto_1.CreateAgentProfileDTO]),
+    __metadata("design:returntype", Promise)
+], AgentProfileController.prototype, "create", null);
+__decorate([
+    (0, core_1.Put)('/:id'),
+    midwayjs_base_1.decorators.checkLogin(true),
+    __param(0, (0, core_1.Param)('id')),
+    __param(1, (0, core_1.Body)()),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", [Number, Object]),
+    __metadata("design:returntype", Promise)
+], AgentProfileController.prototype, "update", null);
+__decorate([
+    (0, core_1.Del)('/:id'),
+    midwayjs_base_1.decorators.checkLogin(true),
+    __param(0, (0, core_1.Param)('id')),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", [Number]),
+    __metadata("design:returntype", Promise)
+], AgentProfileController.prototype, "remove", null);
+__decorate([
+    (0, core_1.Post)('/:id/default'),
+    midwayjs_base_1.decorators.checkLogin(true),
+    __param(0, (0, core_1.Param)('id')),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", [Number]),
+    __metadata("design:returntype", Promise)
+], AgentProfileController.prototype, "setDefault", null);
+exports.AgentProfileController = AgentProfileController = __decorate([
+    (0, core_1.Provide)(),
+    (0, core_1.Controller)('/api/agent-profiles')
+], AgentProfileController);

package/dist/controller/agent.controller.d.ts

@@ -0,0 +1,199 @@
+import { BaseController } from '@cicctencent/midwayjs-base';
+import { RunAgentDTO } from '../dto/agent.dto';
+import { AgentServerService } from '../service/agent-server.service';
+import { AgentTaskQueueService } from '../service/agent-task-queue.service';
+import { ApplicationService } from '../service/application.service';
+/**
+ * Agent 运行 Controller
+ *
+ * 提供 Agent 运行接口，支持流式和非流式响应（按 Application 隔离）
+ */
+export declare class AgentController extends BaseController {
+    agentServerService: AgentServerService;
+    taskQueueService: AgentTaskQueueService;
+    applicationService: ApplicationService;
+    /** 获取当前应用ID */
+    private getApplicationId;
+    /**
+     * 运行 Agent（流式响应）
+     */
+    runAgent(body: RunAgentDTO): Promise<void>;
+    /**
+     * 运行 Agent（非流式响应）
+     */
+    runAgentSync(body: RunAgentDTO): Promise<{
+        ret: number;
+        data: any;
+        msg: string;
+    } | {
+        ret: number;
+        msg: any;
+        data?: undefined;
+    }>;
+    /**
+     * 提交异步任务
+     */
+    submitTask(body: {
+        prompt: string;
+        threadId?: number;
+        profileId?: number;
+        maxSteps?: number;
+        timeout?: number;
+    }): Promise<{
+        ret: number;
+        data: {
+            taskId: string;
+        };
+        msg: string;
+    } | {
+        ret: number;
+        msg: any;
+        data?: undefined;
+    }>;
+    /**
+     * 获取任务状态
+     */
+    getTaskStatus(taskId: string): Promise<{
+        ret: number;
+        data: import("../service/agent-task-queue.service").AgentTask;
+        msg: string;
+    } | {
+        ret: number;
+        msg: any;
+        data?: undefined;
+    }>;
+    /**
+     * 取消任务
+     */
+    cancelTask(taskId: string): Promise<{
+        ret: number;
+        data: {
+            cancelled: boolean;
+        };
+        msg: string;
+    } | {
+        ret: number;
+        msg: any;
+        data?: undefined;
+    }>;
+    /**
+     * 获取队列状态
+     */
+    getQueueStatus(): Promise<{
+        ret: number;
+        data: {
+            total: number;
+            pending: number;
+            running: number;
+            completed: number;
+            failed: number;
+            maxConcurrency: number;
+        };
+        msg: string;
+    } | {
+        ret: number;
+        msg: any;
+        data?: undefined;
+    }>;
+    /**
+     * 获取用户任务列表
+     */
+    getUserTasks(): Promise<{
+        ret: number;
+        data: import("../service/agent-task-queue.service").AgentTask[];
+        msg: string;
+    } | {
+        ret: number;
+        msg: any;
+        data?: undefined;
+    }>;
+    /**
+     * 中止 Agent 执行
+     */
+    abortAgent(threadId: number): Promise<{
+        ret: number;
+        data: any;
+        msg: string;
+    } | {
+        ret: number;
+        msg: any;
+        data?: undefined;
+    }>;
+    /**
+     * SSE 断线重连
+     */
+    reconnectAgent(threadId: number): Promise<void>;
+    /**
+     * 编辑消息后重新生成
+     */
+    editMessage(body: {
+        threadId: number;
+        messageId: number;
+        content: string;
+    }): Promise<{
+        ret: number;
+        data: any;
+        msg: string;
+    } | {
+        ret: number;
+        msg: any;
+        data?: undefined;
+    }>;
+    /**
+     * 重新生成最后一条 AI 回复
+     */
+    regenerate(body: {
+        threadId: number;
+    }): Promise<{
+        ret: number;
+        data: any;
+        msg: string;
+    } | {
+        ret: number;
+        msg: any;
+        data?: undefined;
+    }>;
+    /**
+     * 从指定位置分叉创建新对话
+     */
+    forkThread(body: {
+        threadId: number;
+        messageId: number;
+        title?: string;
+    }): Promise<{
+        ret: number;
+        data: any;
+        msg: string;
+    } | {
+        ret: number;
+        msg: any;
+        data?: undefined;
+    }>;
+    /**
+     * 引擎工具列表诊断
+     */
+    diagnose(): Promise<{
+        ret: number;
+        data: {
+            toolCount: number;
+            tools: {
+                name: any;
+                description: any;
+            }[];
+            available: boolean;
+        };
+        msg: string;
+    } | {
+        ret: number;
+        msg: any;
+        data?: undefined;
+    }>;
+    /**
+     * 写入 SSE 事件
+     */
+    private writeSSE;
+    /**
+     * 获取当前用户 ID
+     */
+    private getUserId;
+}