@cicctencent/agent-midway 0.1.1

package/dist/controller/chat.controller.js

@@ -0,0 +1,375 @@
+"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.ChatController = void 0;
+const core_1 = require("@midwayjs/core");
+const midwayjs_base_1 = require("@cicctencent/midwayjs-base");
+const chat_dto_1 = require("../dto/chat.dto");
+const ai_chat_service_1 = require("../service/ai-chat.service");
+const application_service_1 = require("../service/application.service");
+/**
+ * 聊天 Controller
+ *
+ * 管理工作空间、线程和消息（按 Application 隔离）
+ */
+let ChatController = class ChatController extends midwayjs_base_1.BaseController {
+    /** 获取当前应用ID */
+    async getApplicationId() {
+        const app = await this.applicationService.getDefault();
+        return app?.id || 0;
+    }
+    // ========== Workspace ==========
+    /** 获取工作空间列表 */
+    async getWorkspaces() {
+        try {
+            const userId = this.getUserId();
+            const applicationId = await this.getApplicationId();
+            const workspaces = await this.chatService.getWorkspaces(userId, applicationId);
+            return { ret: 0, data: workspaces, msg: 'success' };
+        }
+        catch (error) {
+            return { ret: 1, msg: error.message };
+        }
+    }
+    /** 获取单个工作空间 */
+    async getWorkspace(id) {
+        try {
+            const userId = this.getUserId();
+            const applicationId = await this.getApplicationId();
+            const workspace = await this.chatService.getWorkspace(id, userId, applicationId);
+            if (!workspace) {
+                return { ret: 1, msg: '工作空间不存在' };
+            }
+            return { ret: 0, data: workspace, msg: 'success' };
+        }
+        catch (error) {
+            return { ret: 1, msg: error.message };
+        }
+    }
+    /** 创建工作空间 */
+    async createWorkspace(body) {
+        try {
+            const userId = this.getUserId();
+            const applicationId = await this.getApplicationId();
+            const workspace = await this.chatService.createWorkspace(userId, applicationId, body);
+            return { ret: 0, data: workspace, msg: 'success' };
+        }
+        catch (error) {
+            return { ret: 1, msg: error.message };
+        }
+    }
+    /** 更新工作空间 */
+    async updateWorkspace(id, body) {
+        try {
+            const userId = this.getUserId();
+            const applicationId = await this.getApplicationId();
+            const workspace = await this.chatService.updateWorkspace(id, userId, applicationId, body);
+            return { ret: 0, data: workspace, msg: 'success' };
+        }
+        catch (error) {
+            return { ret: 1, msg: error.message };
+        }
+    }
+    /** 删除工作空间 */
+    async deleteWorkspace(id) {
+        try {
+            const userId = this.getUserId();
+            const applicationId = await this.getApplicationId();
+            await this.chatService.deleteWorkspace(id, userId, applicationId);
+            return { ret: 0, data: null, msg: 'success' };
+        }
+        catch (error) {
+            return { ret: 1, msg: error.message };
+        }
+    }
+    // ========== Thread ==========
+    /** 获取线程列表 */
+    async getThreads(workspaceId) {
+        try {
+            const userId = this.getUserId();
+            const applicationId = await this.getApplicationId();
+            const threads = await this.chatService.getThreads(workspaceId, userId, applicationId);
+            return { ret: 0, data: threads, msg: 'success' };
+        }
+        catch (error) {
+            return { ret: 1, msg: error.message };
+        }
+    }
+    /** 获取单个线程 */
+    async getThread(id) {
+        try {
+            const userId = this.getUserId();
+            const applicationId = await this.getApplicationId();
+            const thread = await this.chatService.getThread(id, userId, applicationId);
+            if (!thread) {
+                return { ret: 1, msg: '线程不存在' };
+            }
+            return { ret: 0, data: thread, msg: 'success' };
+        }
+        catch (error) {
+            return { ret: 1, msg: error.message };
+        }
+    }
+    /** 创建线程 */
+    async createThread(workspaceId, body) {
+        try {
+            const userId = this.getUserId();
+            const applicationId = await this.getApplicationId();
+            const thread = await this.chatService.createThread(userId, applicationId, {
+                workspaceId,
+                title: body.title,
+            });
+            return { ret: 0, data: thread, msg: 'success' };
+        }
+        catch (error) {
+            return { ret: 1, msg: error.message };
+        }
+    }
+    /** 更新线程 */
+    async updateThread(id, body) {
+        try {
+            const userId = this.getUserId();
+            const applicationId = await this.getApplicationId();
+            const thread = await this.chatService.updateThread(id, userId, applicationId, body);
+            return { ret: 0, data: thread, msg: 'success' };
+        }
+        catch (error) {
+            return { ret: 1, msg: error.message };
+        }
+    }
+    /** 删除线程 */
+    async deleteThread(id) {
+        try {
+            const userId = this.getUserId();
+            const applicationId = await this.getApplicationId();
+            await this.chatService.deleteThread(id, userId, applicationId);
+            return { ret: 0, data: null, msg: 'success' };
+        }
+        catch (error) {
+            return { ret: 1, msg: error.message };
+        }
+    }
+    // ========== Message ==========
+    /** 获取消息列表 */
+    async getMessages(threadId, body = {}) {
+        try {
+            const userId = this.getUserId();
+            const applicationId = await this.getApplicationId();
+            const result = await this.chatService.getMessages(threadId, userId, applicationId, body);
+            return { ret: 0, data: result, msg: 'success' };
+        }
+        catch (error) {
+            return { ret: 1, msg: error.message };
+        }
+    }
+    /** 添加消息 */
+    async addMessage(threadId, body) {
+        try {
+            const userId = this.getUserId();
+            const applicationId = await this.getApplicationId();
+            const message = await this.chatService.addMessage(userId, applicationId, {
+                threadId,
+                role: body.role,
+                content: body.content,
+                type: body.type,
+                meta: body.meta,
+            });
+            return { ret: 0, data: message, msg: 'success' };
+        }
+        catch (error) {
+            return { ret: 1, msg: error.message };
+        }
+    }
+    /** 更新消息 */
+    async updateMessage(id, body) {
+        try {
+            const userId = this.getUserId();
+            const applicationId = await this.getApplicationId();
+            const message = await this.chatService.updateMessage(id, userId, applicationId, body);
+            return { ret: 0, data: message, msg: 'success' };
+        }
+        catch (error) {
+            return { ret: 1, msg: error.message };
+        }
+    }
+    /** 删除消息 */
+    async deleteMessage(id) {
+        try {
+            const userId = this.getUserId();
+            const applicationId = await this.getApplicationId();
+            await this.chatService.deleteMessage(id, userId, applicationId);
+            return { ret: 0, data: null, msg: 'success' };
+        }
+        catch (error) {
+            return { ret: 1, msg: error.message };
+        }
+    }
+    /** 清空线程消息 */
+    async clearThreadMessages(threadId) {
+        try {
+            const userId = this.getUserId();
+            const applicationId = await this.getApplicationId();
+            await this.chatService.clearThreadMessages(threadId, userId, applicationId);
+            return { ret: 0, data: null, msg: 'success' };
+        }
+        catch (error) {
+            return { ret: 1, msg: error.message };
+        }
+    }
+    // ========== 私有方法 ==========
+    /**
+     * 获取当前用户 ID
+     */
+    getUserId() {
+        return String(this.ctx?.currentSession?.userId || 'anonymous');
+    }
+};
+exports.ChatController = ChatController;
+__decorate([
+    (0, core_1.Inject)(),
+    __metadata("design:type", ai_chat_service_1.AIChatService)
+], ChatController.prototype, "chatService", void 0);
+__decorate([
+    (0, core_1.Inject)(),
+    __metadata("design:type", application_service_1.ApplicationService)
+], ChatController.prototype, "applicationService", void 0);
+__decorate([
+    (0, core_1.Get)('/workspaces'),
+    midwayjs_base_1.decorators.checkLogin(true),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", []),
+    __metadata("design:returntype", Promise)
+], ChatController.prototype, "getWorkspaces", null);
+__decorate([
+    (0, core_1.Get)('/workspaces/: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)
+], ChatController.prototype, "getWorkspace", null);
+__decorate([
+    (0, core_1.Post)('/workspaces'),
+    midwayjs_base_1.decorators.checkLogin(true),
+    __param(0, (0, core_1.Body)()),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", [chat_dto_1.CreateWorkspaceDTO]),
+    __metadata("design:returntype", Promise)
+], ChatController.prototype, "createWorkspace", null);
+__decorate([
+    (0, core_1.Put)('/workspaces/: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, chat_dto_1.UpdateWorkspaceDTO]),
+    __metadata("design:returntype", Promise)
+], ChatController.prototype, "updateWorkspace", null);
+__decorate([
+    (0, core_1.Del)('/workspaces/: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)
+], ChatController.prototype, "deleteWorkspace", null);
+__decorate([
+    (0, core_1.Get)('/workspaces/:workspaceId/threads'),
+    midwayjs_base_1.decorators.checkLogin(true),
+    __param(0, (0, core_1.Param)('workspaceId')),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", [Number]),
+    __metadata("design:returntype", Promise)
+], ChatController.prototype, "getThreads", null);
+__decorate([
+    (0, core_1.Get)('/threads/: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)
+], ChatController.prototype, "getThread", null);
+__decorate([
+    (0, core_1.Post)('/workspaces/:workspaceId/threads'),
+    midwayjs_base_1.decorators.checkLogin(true),
+    __param(0, (0, core_1.Param)('workspaceId')),
+    __param(1, (0, core_1.Body)()),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", [Number, Object]),
+    __metadata("design:returntype", Promise)
+], ChatController.prototype, "createThread", null);
+__decorate([
+    (0, core_1.Put)('/threads/: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)
+], ChatController.prototype, "updateThread", null);
+__decorate([
+    (0, core_1.Del)('/threads/: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)
+], ChatController.prototype, "deleteThread", null);
+__decorate([
+    (0, core_1.Get)('/threads/:threadId/messages'),
+    midwayjs_base_1.decorators.checkLogin(true),
+    __param(0, (0, core_1.Param)('threadId')),
+    __param(1, (0, core_1.Body)()),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", [Number, Object]),
+    __metadata("design:returntype", Promise)
+], ChatController.prototype, "getMessages", null);
+__decorate([
+    (0, core_1.Post)('/threads/:threadId/messages'),
+    midwayjs_base_1.decorators.checkLogin(true),
+    __param(0, (0, core_1.Param)('threadId')),
+    __param(1, (0, core_1.Body)()),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", [Number, Object]),
+    __metadata("design:returntype", Promise)
+], ChatController.prototype, "addMessage", null);
+__decorate([
+    (0, core_1.Put)('/messages/: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)
+], ChatController.prototype, "updateMessage", null);
+__decorate([
+    (0, core_1.Del)('/messages/: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)
+], ChatController.prototype, "deleteMessage", null);
+__decorate([
+    (0, core_1.Del)('/threads/:threadId/messages'),
+    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)
+], ChatController.prototype, "clearThreadMessages", null);
+exports.ChatController = ChatController = __decorate([
+    (0, core_1.Provide)(),
+    (0, core_1.Controller)('/api/chat')
+], ChatController);

package/dist/controller/connector.controller.d.ts

@@ -0,0 +1,134 @@
+import { BaseController } from '@cicctencent/midwayjs-base';
+import { ConnectorService } from '../service/connector.service';
+import type { DataSourceConfig } from '../connector/types';
+/**
+ * 连接器管理 Controller
+ *
+ * 提供连接器的 CRUD、连接测试、工具发现、清单生成等接口。
+ */
+export declare class ConnectorController extends BaseController {
+    connectorService: ConnectorService;
+    /** 获取连接器列表 */
+    list(): Promise<{
+        ret: number;
+        data: import("../service/connector.service").ConnectorRecord[];
+        msg: string;
+    } | {
+        ret: number;
+        msg: any;
+        data?: undefined;
+    }>;
+    /** 获取单个连接器 */
+    detail(id: string): Promise<{
+        ret: number;
+        data: import("../service/connector.service").ConnectorRecord;
+        msg: string;
+    } | {
+        ret: number;
+        msg: any;
+        data?: undefined;
+    }>;
+    /** 创建连接器 */
+    create(body: DataSourceConfig): Promise<{
+        ret: number;
+        data: import("../service/connector.service").ConnectorRecord;
+        msg: string;
+    } | {
+        ret: number;
+        msg: any;
+        data?: undefined;
+    }>;
+    /** 更新连接器 */
+    update(id: string, body: Partial<DataSourceConfig>): Promise<{
+        ret: number;
+        data: import("../service/connector.service").ConnectorRecord;
+        msg: string;
+    } | {
+        ret: number;
+        msg: any;
+        data?: undefined;
+    }>;
+    /** 删除连接器 */
+    remove(id: string): Promise<{
+        ret: number;
+        data: any;
+        msg: string;
+    } | {
+        ret: number;
+        msg: any;
+        data?: undefined;
+    }>;
+    /** 测试连接（已保存的连接器） */
+    testConnection(id: string): Promise<{
+        ret: number;
+        data: import("../connector/types").ConnectionTestResult;
+        msg: string;
+    } | {
+        ret: number;
+        msg: any;
+        data?: undefined;
+    }>;
+    /** 测试配置（未保存的配置） */
+    testConfig(body: DataSourceConfig): Promise<{
+        ret: number;
+        data: import("../connector/types").ConnectionTestResult;
+        msg: string;
+    } | {
+        ret: number;
+        msg: any;
+        data?: undefined;
+    }>;
+    /** 发现工具 */
+    discoverTools(id: string): Promise<{
+        ret: number;
+        data: import("../connector/types").DiscoveredTool[];
+        msg: string;
+    } | {
+        ret: number;
+        msg: any;
+        data?: undefined;
+    }>;
+    /** 生成清单 */
+    generateManifest(id: string): Promise<{
+        ret: number;
+        data: {
+            tools: import("../connector/types").DiscoveredTool[];
+            metadata: Record<string, any>;
+        };
+        msg: string;
+    } | {
+        ret: number;
+        msg: any;
+        data?: undefined;
+    }>;
+    /** 生成 MCP 服务器配置 */
+    generateMCPServerConfigs(id: string): Promise<{
+        ret: number;
+        data: import("../connector/types").MCPServerConfig[];
+        msg: string;
+    } | {
+        ret: number;
+        msg: any;
+        data?: undefined;
+    }>;
+    /** 获取所有数据源类型 */
+    getTypes(): Promise<{
+        ret: number;
+        data: import("../connector/types").DataSourceType[];
+        msg: string;
+    } | {
+        ret: number;
+        msg: any;
+        data?: undefined;
+    }>;
+    /** 获取数据源类型信息（前端向导展示用） */
+    getTypeInfos(): Promise<{
+        ret: number;
+        data: import("../connector/types").DataSourceTypeInfo[];
+        msg: string;
+    } | {
+        ret: number;
+        msg: any;
+        data?: undefined;
+    }>;
+}