@cicctencent/agent-midway 0.1.1

package/dist/service/automation.service.js

@@ -0,0 +1,196 @@
+"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.AutomationService = void 0;
+/**
+ * 自动化定时任务调度服务
+ *
+ * 基于 node-cron 的定时调度，触发时创建独立 Agent 会话执行任务。
+ * 使用动态 require 处理 node-cron 依赖，避免硬依赖。
+ */
+const core_1 = require("@midwayjs/core");
+const midwayjs_base_1 = require("@cicctencent/midwayjs-base");
+const typeorm_1 = require("@midwayjs/typeorm");
+const agent_server_service_1 = require("./agent-server.service");
+// 动态加载 node-cron（可选依赖）
+let cron;
+try {
+    cron = require('node-cron');
+}
+catch {
+    console.warn('[AutomationService] node-cron not installed, scheduler disabled');
+}
+/** 活跃的 cron 任务映射（id → ScheduledTask） */
+const activeJobs = new Map();
+let AutomationService = class AutomationService extends midwayjs_base_1.BaseService {
+    constructor() {
+        super(...arguments);
+        this.initialized = false;
+    }
+    /** 初始化自动化调度器：加载所有已启用任务并注册 cron */
+    async initAutomationScheduler() {
+        if (this.initialized)
+            return;
+        if (!cron) {
+            console.warn('[AutomationService] node-cron not available, skipping scheduler init');
+            return;
+        }
+        try {
+            const enabled = await this.dataSource.query('SELECT * FROM t_ai_automation WHERE is_deleted = 0 AND status = ?', ['active']);
+            for (const automation of enabled) {
+                this.scheduleJob(automation);
+            }
+            this.initialized = true;
+            console.log(`[AutomationService] Initialized with ${enabled.length} active job(s)`);
+        }
+        catch (err) {
+            console.error('[AutomationService] Init failed:', err?.message);
+        }
+    }
+    /** 确保调度器已初始化 */
+    async ensureInitialized() {
+        if (!this.initialized) {
+            await this.initAutomationScheduler();
+        }
+    }
+    /** 为单个任务注册 cron 调度 */
+    scheduleJob(automation) {
+        if (!cron)
+            return;
+        // 先注销已有的
+        this.unscheduleJob(automation.id);
+        const schedule = automation.config?.cron || automation.config?.schedule || automation.schedule;
+        if (!schedule) {
+            console.warn(`[AutomationService] No schedule for task ${automation.id}`);
+            return;
+        }
+        if (!cron.validate(schedule)) {
+            console.warn(`[AutomationService] Invalid cron expression for task ${automation.id}: ${schedule}`);
+            return;
+        }
+        const task = cron.schedule(schedule, () => {
+            this.executeAutomation(automation.id).catch(err => {
+                console.error(`[AutomationService] Task ${automation.id} failed:`, err);
+            });
+        });
+        activeJobs.set(automation.id, task);
+        console.log(`[AutomationService] Scheduled task ${automation.id} (${automation.name}) with cron: ${schedule}`);
+    }
+    /** 注销调度 */
+    unscheduleJob(id) {
+        const existing = activeJobs.get(id);
+        if (existing) {
+            existing.stop();
+            activeJobs.delete(id);
+        }
+    }
+    /** 实际执行自动化任务：创建 Agent 会话 + 运行 Agent + 记录结果 */
+    async executeAutomation(id, triggerType = 'schedule') {
+        // 查询任务
+        const automations = await this.dataSource.query('SELECT * FROM t_ai_automation WHERE id = ? AND is_deleted = 0', [id]);
+        if (!automations || automations.length === 0) {
+            throw new Error(`Automation ${id} not found`);
+        }
+        const automation = automations[0];
+        // 解析 config
+        let config = {};
+        try {
+            config = typeof automation.config === 'string' ? JSON.parse(automation.config) : automation.config;
+        }
+        catch { /* ignore */ }
+        const prompt = config.prompt || automation.prompt || automation.name;
+        const agentProfileId = config.agentProfileId || config.agent_profile_id || automation.agent_profile_id;
+        const startedAt = new Date();
+        const startTime = Date.now();
+        // 标记运行中
+        try {
+            await this.dataSource.query('UPDATE t_ai_automation SET last_status = ?, last_run_at = NOW() WHERE id = ?', ['running', id]);
+        }
+        catch { /* ignore */ }
+        let status = 'success';
+        let result = '';
+        const errors = [];
+        try {
+            // 通过 AgentServerService 运行 Agent
+            const runResult = await this.agentServerService.runAgent(prompt, {
+                profileId: agentProfileId,
+                maxSteps: config.maxSteps || 20,
+                timeout: config.timeout || 120000,
+            });
+            result = runResult?.content || runResult?.finalContent || String(runResult || 'Completed');
+        }
+        catch (err) {
+            errors.push(err.message || 'Automation execution failed');
+            status = 'error';
+            result = errors.join('\n');
+        }
+        const finishedAt = new Date();
+        const duration = Date.now() - startTime;
+        // 更新任务状态
+        try {
+            await this.dataSource.query('UPDATE t_ai_automation SET last_status = ?, last_result = ? WHERE id = ?', [status, result.slice(0, 2000), id]);
+        }
+        catch { /* ignore */ }
+        // 保存执行记录
+        try {
+            await this.dataSource.query(`INSERT INTO t_ai_automation_run (automation_id, status, trigger_type, prompt, result, errors, started_at, finished_at, duration) 
+                 VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)`, [id, status, triggerType, prompt, result.slice(0, 5000), JSON.stringify(errors), startedAt, finishedAt, duration]);
+        }
+        catch { /* ignore */ }
+        return result;
+    }
+    /** 刷新所有调度 */
+    async refreshScheduler() {
+        // 停止所有现有任务
+        for (const [, task] of activeJobs) {
+            task.stop();
+        }
+        activeJobs.clear();
+        // 重新加载启用的任务
+        if (!cron)
+            return;
+        try {
+            const enabled = await this.dataSource.query('SELECT * FROM t_ai_automation WHERE is_deleted = 0 AND status = ?', ['active']);
+            for (const automation of enabled) {
+                this.scheduleJob(automation);
+            }
+        }
+        catch (err) {
+            console.error('[AutomationService] Refresh failed:', err?.message);
+        }
+    }
+    /** 获取活跃任务数 */
+    getActiveJobCount() {
+        return activeJobs.size;
+    }
+    /** 停止所有调度任务（服务关闭时调用） */
+    stopAll() {
+        for (const [, task] of activeJobs) {
+            task.stop();
+        }
+        activeJobs.clear();
+    }
+};
+exports.AutomationService = AutomationService;
+__decorate([
+    (0, typeorm_1.InjectDataSource)('default'),
+    __metadata("design:type", Function)
+], AutomationService.prototype, "dataSource", void 0);
+__decorate([
+    (0, core_1.Inject)(),
+    __metadata("design:type", agent_server_service_1.AgentServerService)
+], AutomationService.prototype, "agentServerService", void 0);
+exports.AutomationService = AutomationService = __decorate([
+    (0, core_1.Provide)(),
+    (0, core_1.Scope)(core_1.ScopeEnum.Singleton),
+    (0, core_1.Provide)(),
+    (0, core_1.Scope)(core_1.ScopeEnum.Singleton)
+], AutomationService);

package/dist/service/connector.service.d.ts

@@ -0,0 +1,136 @@
+import { BaseService } from '@cicctencent/midwayjs-base';
+import type { DataSourceConfig, DataSourceType, ConnectionTestResult, DiscoveredTool, MCPServerConfig, DataSourceTypeInfo } from '../connector/types';
+/** 连接器配置（带元数据） */
+export interface ConnectorRecord {
+    id: string;
+    config: DataSourceConfig;
+    createdAt: number;
+    updatedAt: number;
+}
+/** 连接器 manifest 类型 */
+export interface ConnectorManifest {
+    name: string;
+    description?: string;
+    category?: string;
+    version?: string;
+    author?: string;
+    icon?: string;
+    color?: string;
+    enabled?: boolean;
+    dataSourceType?: string;
+    dataSourceConfig?: Record<string, unknown>;
+    apiRoot?: string;
+    requestBuilderId?: string;
+    urlTemplate?: string;
+    relay?: {
+        url: string;
+        headers?: Record<string, string>;
+    };
+    auth?: {
+        type: 'bearer' | 'basic' | 'apiKey' | 'custom';
+        value: string;
+        name?: string;
+    };
+    response?: {
+        extract?: string;
+        errorField?: string;
+        errorMessage?: string;
+    };
+    skillsEnabled?: Record<string, boolean>;
+}
+/** 目录式连接器实体 */
+export interface ConnectorEntity {
+    id: string;
+    name: string;
+    description?: string;
+    icon?: string;
+    color?: string;
+    category: string;
+    version?: string;
+    author?: string;
+    mcpServerIds: (number | string)[];
+    skillIds: string[];
+    enabled: boolean;
+    status: string;
+    dataSourceType?: string;
+    dataSourceConfig?: Record<string, unknown>;
+    createdAt: string;
+    updatedAt: string;
+}
+export declare class ConnectorService extends BaseService {
+    private initialized;
+    init(): Promise<void>;
+    /** 列出所有连接器 */
+    list(): Promise<ConnectorRecord[]>;
+    /** 获取单个连接器 */
+    get(id: string): Promise<ConnectorRecord | null>;
+    /** 创建连接器 */
+    create(config: DataSourceConfig): Promise<ConnectorRecord>;
+    /** 更新连接器 */
+    update(id: string, config: Partial<DataSourceConfig>): Promise<ConnectorRecord | null>;
+    /** 删除连接器 */
+    delete(id: string): Promise<boolean>;
+    /** 测试连接 */
+    testConnection(id: string): Promise<ConnectionTestResult>;
+    /** 测试配置（未保存的配置） */
+    testConfig(config: DataSourceConfig): Promise<ConnectionTestResult>;
+    /** 发现工具 */
+    discoverTools(id: string): Promise<DiscoveredTool[]>;
+    /** 生成清单 */
+    generateManifest(id: string): Promise<{
+        tools: DiscoveredTool[];
+        metadata: Record<string, any>;
+    }>;
+    /** 生成 MCP 服务器配置 */
+    generateMCPServerConfigs(id: string): Promise<MCPServerConfig[]>;
+    /** 获取所有数据源类型 */
+    getDataSourceTypes(): DataSourceType[];
+    /** 获取所有适配器类型 */
+    getAdapterTypes(): DataSourceType[];
+    /** 获取数据源类型信息（前端向导展示用） */
+    getDataSourceTypeInfos(): DataSourceTypeInfo[];
+    /** 获取指定类型的信息 */
+    getDataSourceTypeInfo(type: DataSourceType): DataSourceTypeInfo;
+    /** 自动发现连接器：扫描 data/connectors/ 目录，为每个有 manifest.json 的目录构建连接器实体 */
+    autoDiscoverConnectors(): Promise<void>;
+    /** 从 manifest 构建连接器实体 */
+    private buildConnectorFromManifest;
+    /** 列出所有目录式连接器实体 */
+    listConnectorEntities(): ConnectorEntity[];
+    /** 获取单个目录式连接器实体 */
+    getConnectorEntity(id: string): ConnectorEntity | undefined;
+    /** 获取连接器包目录（校验存在性） */
+    private getConnectorPackageDir;
+    /** 读取 manifest.json */
+    getManifest(connectorId: string): ConnectorManifest;
+    /** 更新 manifest.json */
+    updateManifest(connectorId: string, updates: Partial<ConnectorManifest>): ConnectorManifest;
+    /** 启用/禁用连接器，回写 manifest.json */
+    toggleConnector(id: string, enabled: boolean): ConnectorEntity | undefined;
+    /** 重新生成连接器：重新调用适配器发现工具，清空并重写 mcp-servers/*.json */
+    regenerateConnector(connectorId: string): Promise<{
+        toolCount: number;
+        connector: ConnectorEntity;
+    }>;
+    /** 新增 MCP 服务器配置文件 */
+    addConnectorMCPServer(connectorId: string, config: Record<string, unknown>): Promise<{
+        serverName: string;
+    }>;
+    /** 更新 MCP 服务器配置文件 */
+    updateConnectorMCPServer(connectorId: string, serverName: string, config: Record<string, unknown>): Promise<void>;
+    /** 删除 MCP 服务器配置文件 */
+    removeConnectorMCPServer(connectorId: string, serverName: string): Promise<void>;
+    /** 读取 Skill .md 文件完整内容 */
+    getConnectorSkillFile(connectorId: string, skillId: string): {
+        content: string;
+    };
+    /** 新建 Skill .md 文件 */
+    addConnectorSkill(connectorId: string, skillId: string, content: string): Promise<void>;
+    /** 更新 Skill .md 文件 */
+    updateConnectorSkill(connectorId: string, skillId: string, content: string): Promise<void>;
+    /** 删除 Skill .md 文件 */
+    removeConnectorSkill(connectorId: string, skillId: string): Promise<void>;
+    private getFilePath;
+    private generateId;
+    private ensureDir;
+}