@cicctencent/agent-midway 0.1.1

package/dist/service/mcp-server.service.js

@@ -0,0 +1,143 @@
+"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 };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.MCPServerService = 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_mcp_server_entity_1 = __importDefault(require("../model/ai-mcp-server.entity"));
+/**
+ * MCP Server 服务
+ * 管理 MCP Server 配置、状态、工具定义等
+ */
+let MCPServerService = class MCPServerService extends midwayjs_base_1.BaseService {
+    get model() {
+        if (this.repo)
+            return this.repo;
+        return (this.repo = this.dataSource.getRepository(ai_mcp_server_entity_1.default));
+    }
+    /** 获取所有 MCP Server */
+    async list(applicationId, options) {
+        const where = { applicationId };
+        if (options?.enabled !== undefined) {
+            where.enabled = options.enabled ? 1 : 0;
+        }
+        if (options?.status) {
+            where.status = options.status;
+        }
+        return this.model.find({
+            where,
+            order: { createTime: 'DESC' },
+        });
+    }
+    /** 获取单个 MCP Server */
+    async get(id, applicationId) {
+        const where = { id };
+        if (applicationId)
+            where.applicationId = applicationId;
+        return this.model.findOne({ where });
+    }
+    /** 获取启用的 MCP Server */
+    async getEnabled(applicationId) {
+        return this.model.find({
+            where: { applicationId, enabled: 1 },
+            order: { name: 'ASC' },
+        });
+    }
+    /** 创建 MCP Server */
+    async create(applicationId, data) {
+        const server = this.model.create({
+            ...data,
+            applicationId,
+            status: 'disconnected',
+            toolCount: 0,
+            lastError: '',
+        });
+        return this.model.save(server);
+    }
+    /** 更新 MCP Server */
+    async update(id, applicationId, data) {
+        await this.model.update({ id, applicationId }, data);
+        return this.get(id, applicationId);
+    }
+    /** 删除 MCP Server */
+    async delete(id, applicationId) {
+        await this.model.delete({ id, applicationId });
+    }
+    /** 启用 MCP Server */
+    async enable(id, applicationId) {
+        await this.model.update({ id, applicationId }, { enabled: 1 });
+        return this.get(id, applicationId);
+    }
+    /** 禁用 MCP Server */
+    async disable(id, applicationId) {
+        await this.model.update({ id, applicationId }, { enabled: 0, status: 'disconnected' });
+        return this.get(id, applicationId);
+    }
+    /** 更新连接状态 */
+    async updateStatus(id, applicationId, status, error) {
+        const updateData = { status };
+        if (error) {
+            updateData.lastError = error;
+        }
+        await this.model.update({ id, applicationId }, updateData);
+        return this.get(id, applicationId);
+    }
+    /** 更新工具定义 */
+    async updateToolDefinitions(id, applicationId, tools) {
+        await this.model.update({ id, applicationId }, {
+            toolDefinitions: tools,
+            toolCount: tools.length,
+            status: 'connected',
+            lastError: '',
+        });
+        return this.get(id, applicationId);
+    }
+    /** 获取 MCP Server 的工具列表 */
+    async getTools(id, applicationId) {
+        const server = await this.get(id, applicationId);
+        if (!server || !server.toolDefinitions) {
+            return [];
+        }
+        return server.toolDefinitions;
+    }
+    /** 批量获取工具定义 */
+    async getToolsByServerIds(applicationId, ids) {
+        if (!ids || ids.length === 0)
+            return [];
+        const servers = await this.model.find({
+            where: ids.map(id => ({ id: Number(id), applicationId })),
+        });
+        const tools = [];
+        for (const server of servers) {
+            if (server.toolDefinitions) {
+                tools.push(...server.toolDefinitions.map(t => ({
+                    ...t,
+                    serverId: server.id,
+                    serverName: server.name,
+                })));
+            }
+        }
+        return tools;
+    }
+};
+exports.MCPServerService = MCPServerService;
+__decorate([
+    (0, typeorm_1.InjectDataSource)('default'),
+    __metadata("design:type", typeorm_2.DataSource)
+], MCPServerService.prototype, "dataSource", void 0);
+exports.MCPServerService = MCPServerService = __decorate([
+    (0, core_1.Provide)()
+], MCPServerService);

package/dist/service/model-config.service.d.ts

@@ -0,0 +1,57 @@
+import { BaseService } from '@cicctencent/midwayjs-base';
+import { DataSource, Repository } from 'typeorm';
+import AIModelConfigEntity from '../model/ai-model-config.entity';
+/**
+ * 模型配置服务
+ * 管理多模型供应商配置、分层策略和降级链
+ */
+export declare class ModelConfigService extends BaseService {
+    dataSource: DataSource;
+    private repo?;
+    get model(): Repository<AIModelConfigEntity>;
+    /** 获取所有模型配置 */
+    list(applicationId: number, options?: {
+        enabled?: boolean;
+        tier?: string;
+        role?: string;
+    }): Promise<AIModelConfigEntity[]>;
+    /** 获取单个模型配置 */
+    get(id: number, applicationId?: number): Promise<AIModelConfigEntity>;
+    /** 获取默认模型配置 */
+    getDefault(applicationId: number): Promise<AIModelConfigEntity>;
+    /** 获取主模型配置 */
+    getPrimary(applicationId: number): Promise<AIModelConfigEntity>;
+    /** 获取降级链（按优先级排序） */
+    getFallbackChain(applicationId: number, role?: string): Promise<{
+        primary: AIModelConfigEntity;
+        fallbacks: AIModelConfigEntity[];
+        roleSpecific: AIModelConfigEntity[];
+    }>;
+    /** 创建模型配置 */
+    create(applicationId: number, data: Partial<AIModelConfigEntity>): Promise<AIModelConfigEntity>;
+    /** 更新模型配置 */
+    update(id: number, applicationId: number, data: Partial<AIModelConfigEntity>): Promise<AIModelConfigEntity>;
+    /** 删除模型配置 */
+    delete(id: number, applicationId: number): Promise<void>;
+    /** 设置为默认 */
+    setDefault(id: number, applicationId: number): Promise<AIModelConfigEntity>;
+    /** 启用模型配置 */
+    enable(id: number, applicationId: number): Promise<AIModelConfigEntity>;
+    /** 禁用模型配置 */
+    disable(id: number, applicationId: number): Promise<AIModelConfigEntity>;
+    /** 获取指定角色的模型配置 */
+    getByRole(applicationId: number, role: string): Promise<AIModelConfigEntity[]>;
+    /** 获取可用的模型配置（用于 Agent 运行时） */
+    getAvailableConfig(applicationId: number, role?: string): Promise<AIModelConfigEntity>;
+    /** 转换为 agent-core 兼容的模型配置格式 */
+    toAgentModelConfig(applicationId: number, id?: number, role?: string): Promise<{
+        provider: string;
+        model: string;
+        apiKey: string;
+        baseUrl: string;
+        temperature: number;
+        maxTokens: number;
+        timeout: number;
+        extraConfig: Record<string, any>;
+    }>;
+}

package/dist/service/model-config.service.js

@@ -0,0 +1,168 @@
+"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 };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ModelConfigService = 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_model_config_entity_1 = __importDefault(require("../model/ai-model-config.entity"));
+/**
+ * 模型配置服务
+ * 管理多模型供应商配置、分层策略和降级链
+ */
+let ModelConfigService = class ModelConfigService extends midwayjs_base_1.BaseService {
+    get model() {
+        if (this.repo)
+            return this.repo;
+        return (this.repo = this.dataSource.getRepository(ai_model_config_entity_1.default));
+    }
+    /** 获取所有模型配置 */
+    async list(applicationId, options) {
+        const where = { applicationId };
+        if (options?.enabled !== undefined) {
+            where.enabled = options.enabled ? 1 : 0;
+        }
+        if (options?.tier) {
+            where.tier = options.tier;
+        }
+        if (options?.role) {
+            where.role = options.role;
+        }
+        return this.model.find({
+            where,
+            order: { tier: 'ASC', priority: 'DESC', createTime: 'ASC' },
+        });
+    }
+    /** 获取单个模型配置 */
+    async get(id, applicationId) {
+        const where = { id };
+        if (applicationId)
+            where.applicationId = applicationId;
+        return this.model.findOne({ where });
+    }
+    /** 获取默认模型配置 */
+    async getDefault(applicationId) {
+        return this.model.findOne({
+            where: { applicationId, isDefault: 1, enabled: 1 },
+        });
+    }
+    /** 获取主模型配置 */
+    async getPrimary(applicationId) {
+        return this.model.findOne({
+            where: { applicationId, tier: 'primary', enabled: 1 },
+            order: { priority: 'DESC' },
+        });
+    }
+    /** 获取降级链（按优先级排序） */
+    async getFallbackChain(applicationId, role) {
+        const where = { applicationId, enabled: 1 };
+        if (role) {
+            where.role = role;
+        }
+        const configs = await this.model.find({
+            where,
+            order: { tier: 'ASC', priority: 'DESC' },
+        });
+        // 按 tier 分组
+        const primary = configs.filter(c => c.tier === 'primary');
+        const fallback = configs.filter(c => c.tier === 'fallback');
+        const roleSpecific = configs.filter(c => c.tier === 'role_specific');
+        return {
+            primary: primary[0] || null,
+            fallbacks: fallback,
+            roleSpecific: roleSpecific.filter(c => !role || c.role === role),
+        };
+    }
+    /** 创建模型配置 */
+    async create(applicationId, data) {
+        // 如果设置为默认，先清除其他默认
+        if (data.isDefault === 1) {
+            await this.model.update({ applicationId, isDefault: 1 }, { isDefault: 0 });
+        }
+        const config = this.model.create({ ...data, applicationId });
+        return this.model.save(config);
+    }
+    /** 更新模型配置 */
+    async update(id, applicationId, data) {
+        // 如果设置为默认，先清除其他默认
+        if (data.isDefault === 1) {
+            await this.model.update({ applicationId, isDefault: 1 }, { isDefault: 0 });
+        }
+        await this.model.update({ id, applicationId }, data);
+        return this.get(id, applicationId);
+    }
+    /** 删除模型配置 */
+    async delete(id, applicationId) {
+        await this.model.delete({ id, applicationId });
+    }
+    /** 设置为默认 */
+    async setDefault(id, applicationId) {
+        await this.model.update({ applicationId, isDefault: 1 }, { isDefault: 0 });
+        await this.model.update({ id, applicationId }, { isDefault: 1 });
+        return this.get(id, applicationId);
+    }
+    /** 启用模型配置 */
+    async enable(id, applicationId) {
+        await this.model.update({ id, applicationId }, { enabled: 1 });
+        return this.get(id, applicationId);
+    }
+    /** 禁用模型配置 */
+    async disable(id, applicationId) {
+        await this.model.update({ id, applicationId }, { enabled: 0 });
+        return this.get(id, applicationId);
+    }
+    /** 获取指定角色的模型配置 */
+    async getByRole(applicationId, role) {
+        return this.model.find({
+            where: { applicationId, role, enabled: 1 },
+            order: { priority: 'DESC' },
+        });
+    }
+    /** 获取可用的模型配置（用于 Agent 运行时） */
+    async getAvailableConfig(applicationId, role) {
+        const chain = await this.getFallbackChain(applicationId, role);
+        return chain.primary || chain.fallbacks[0] || null;
+    }
+    /** 转换为 agent-core 兼容的模型配置格式 */
+    async toAgentModelConfig(applicationId, id, role) {
+        let config = null;
+        if (id) {
+            config = await this.get(id, applicationId);
+        }
+        else {
+            config = await this.getAvailableConfig(applicationId, role);
+        }
+        if (!config)
+            return null;
+        return {
+            provider: config.provider,
+            model: config.model,
+            apiKey: config.apiKey,
+            baseUrl: config.baseUrl,
+            temperature: config.temperature,
+            maxTokens: config.maxTokens,
+            timeout: config.timeout,
+            extraConfig: config.extraConfig,
+        };
+    }
+};
+exports.ModelConfigService = ModelConfigService;
+__decorate([
+    (0, typeorm_1.InjectDataSource)('default'),
+    __metadata("design:type", typeorm_2.DataSource)
+], ModelConfigService.prototype, "dataSource", void 0);
+exports.ModelConfigService = ModelConfigService = __decorate([
+    (0, core_1.Provide)()
+], ModelConfigService);

package/dist/service/observability.service.d.ts

@@ -0,0 +1,145 @@
+import { BaseService } from '@cicctencent/midwayjs-base';
+/**
+ * 链路追踪记录
+ */
+export interface TraceRecord {
+    id: string;
+    traceId: string;
+    threadId?: number;
+    action: string;
+    status: 'success' | 'error' | 'timeout';
+    duration: number;
+    input?: any;
+    output?: any;
+    error?: string;
+    timestamp: string;
+    userId?: string;
+    applicationId?: number;
+}
+/**
+ * 指标记录
+ */
+export interface MetricRecord {
+    id: string;
+    name: string;
+    value: number;
+    unit: string;
+    labels: Record<string, string>;
+    timestamp: string;
+}
+/**
+ * 查询选项
+ */
+export interface TraceQueryOptions {
+    traceId?: string;
+    threadId?: number;
+    action?: string;
+    status?: string;
+    startTime?: string;
+    endTime?: string;
+    page?: number;
+    pageSize?: number;
+}
+export interface MetricQueryOptions {
+    name?: string;
+    startTime?: string;
+    endTime?: string;
+    page?: number;
+    pageSize?: number;
+}
+/**
+ * 可观测性服务
+ * 提供链路追踪、指标记录、仪表板和系统健康监控
+ */
+export declare class ObservabilityService extends BaseService {
+    /** 环形缓冲区最大容量 */
+    private static readonly MAX_TRACES;
+    private static readonly MAX_METRICS;
+    /** 追踪记录环形缓冲区 */
+    private traceBuffer;
+    private traceIndex;
+    /** 指标记录环形缓冲区 */
+    private metricBuffer;
+    private metricIndex;
+    /** 自增 ID 计数器 */
+    private traceIdCounter;
+    private metricIdCounter;
+    /** 服务启动时间 */
+    private readonly startTime;
+    /**
+     * 记录链路追踪数据
+     */
+    recordTrace(trace: Omit<TraceRecord, 'id' | 'timestamp'> & Partial<Pick<TraceRecord, 'timestamp'>>): TraceRecord;
+    /**
+     * 查询追踪记录
+     */
+    getTraces(options?: TraceQueryOptions): {
+        list: TraceRecord[];
+        total: number;
+        page: number;
+        pageSize: number;
+    };
+    /**
+     * 获取单个追踪详情
+     */
+    getTraceById(traceId: string): TraceRecord | null;
+    /**
+     * 记录指标数据
+     */
+    recordMetric(metric: Omit<MetricRecord, 'id' | 'timestamp'> & Partial<Pick<MetricRecord, 'timestamp'>>): MetricRecord;
+    /**
+     * 查询指标
+     */
+    getMetrics(options?: MetricQueryOptions): {
+        list: MetricRecord[];
+        total: number;
+        page: number;
+        pageSize: number;
+    };
+    /**
+     * 获取仪表板数据
+     */
+    getDashboard(): {
+        totalRequests: number;
+        successCount: number;
+        errorCount: number;
+        timeoutCount: number;
+        successRate: number;
+        avgDuration: number;
+        activeThreads: number;
+        toolCallStats: Record<string, {
+            total: number;
+            success: number;
+            error: number;
+        }>;
+        recentTraces: TraceRecord[];
+    };
+    /**
+     * 获取系统健康状态
+     */
+    getSystemHealth(): {
+        status: 'healthy' | 'warning' | 'critical';
+        uptime: number;
+        memory: {
+            rss: number;
+            heapUsed: number;
+            heapTotal: number;
+            external: number;
+        };
+        cpu: {
+            user: number;
+            system: number;
+        };
+        activeConnections: number;
+        traceCount: number;
+        metricCount: number;
+    };
+    /**
+     * 导出追踪数据
+     */
+    exportTraces(format?: 'json' | 'csv'): string;
+    /**
+     * 清空所有数据
+     */
+    clear(): void;
+}