@cicctencent/agent-midway 0.1.1

package/dist/service/observability.service.js

@@ -0,0 +1,281 @@
+"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 ObservabilityService_1;
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ObservabilityService = void 0;
+const core_1 = require("@midwayjs/core");
+const midwayjs_base_1 = require("@cicctencent/midwayjs-base");
+/**
+ * 可观测性服务
+ * 提供链路追踪、指标记录、仪表板和系统健康监控
+ */
+let ObservabilityService = ObservabilityService_1 = class ObservabilityService extends midwayjs_base_1.BaseService {
+    constructor() {
+        super(...arguments);
+        /** 追踪记录环形缓冲区 */
+        this.traceBuffer = [];
+        this.traceIndex = 0;
+        /** 指标记录环形缓冲区 */
+        this.metricBuffer = [];
+        this.metricIndex = 0;
+        /** 自增 ID 计数器 */
+        this.traceIdCounter = 0;
+        this.metricIdCounter = 0;
+        /** 服务启动时间 */
+        this.startTime = Date.now();
+    }
+    // ===================== 链路追踪 =====================
+    /**
+     * 记录链路追踪数据
+     */
+    recordTrace(trace) {
+        const record = {
+            ...trace,
+            id: `trace_${++this.traceIdCounter}`,
+            timestamp: trace.timestamp || new Date().toISOString(),
+        };
+        if (this.traceBuffer.length < ObservabilityService_1.MAX_TRACES) {
+            this.traceBuffer.push(record);
+        }
+        else {
+            this.traceBuffer[this.traceIndex] = record;
+        }
+        this.traceIndex =
+            (this.traceIndex + 1) % ObservabilityService_1.MAX_TRACES;
+        return record;
+    }
+    /**
+     * 查询追踪记录
+     */
+    getTraces(options = {}) {
+        let results = [...this.traceBuffer];
+        if (options.traceId) {
+            results = results.filter(t => t.traceId === options.traceId);
+        }
+        if (options.threadId !== undefined) {
+            results = results.filter(t => t.threadId === options.threadId);
+        }
+        if (options.action) {
+            results = results.filter(t => t.action === options.action);
+        }
+        if (options.status) {
+            results = results.filter(t => t.status === options.status);
+        }
+        if (options.startTime) {
+            results = results.filter(t => t.timestamp >= options.startTime);
+        }
+        if (options.endTime) {
+            results = results.filter(t => t.timestamp <= options.endTime);
+        }
+        // 按时间倒序
+        results.sort((a, b) => b.timestamp.localeCompare(a.timestamp));
+        const total = results.length;
+        const page = options.page || 1;
+        const pageSize = options.pageSize || 20;
+        const start = (page - 1) * pageSize;
+        const list = results.slice(start, start + pageSize);
+        return { list, total, page, pageSize };
+    }
+    /**
+     * 获取单个追踪详情
+     */
+    getTraceById(traceId) {
+        return this.traceBuffer.find(t => t.traceId === traceId) || null;
+    }
+    // ===================== 指标记录 =====================
+    /**
+     * 记录指标数据
+     */
+    recordMetric(metric) {
+        const record = {
+            ...metric,
+            id: `metric_${++this.metricIdCounter}`,
+            timestamp: metric.timestamp || new Date().toISOString(),
+        };
+        if (this.metricBuffer.length < ObservabilityService_1.MAX_METRICS) {
+            this.metricBuffer.push(record);
+        }
+        else {
+            this.metricBuffer[this.metricIndex] = record;
+        }
+        this.metricIndex =
+            (this.metricIndex + 1) % ObservabilityService_1.MAX_METRICS;
+        return record;
+    }
+    /**
+     * 查询指标
+     */
+    getMetrics(options = {}) {
+        let results = [...this.metricBuffer];
+        if (options.name) {
+            results = results.filter(m => m.name === options.name);
+        }
+        if (options.startTime) {
+            results = results.filter(m => m.timestamp >= options.startTime);
+        }
+        if (options.endTime) {
+            results = results.filter(m => m.timestamp <= options.endTime);
+        }
+        results.sort((a, b) => b.timestamp.localeCompare(a.timestamp));
+        const total = results.length;
+        const page = options.page || 1;
+        const pageSize = options.pageSize || 20;
+        const start = (page - 1) * pageSize;
+        const list = results.slice(start, start + pageSize);
+        return { list, total, page, pageSize };
+    }
+    // ===================== 仪表板 =====================
+    /**
+     * 获取仪表板数据
+     */
+    getDashboard() {
+        const traces = this.traceBuffer;
+        const totalRequests = traces.length;
+        const successCount = traces.filter(t => t.status === 'success').length;
+        const errorCount = traces.filter(t => t.status === 'error').length;
+        const timeoutCount = traces.filter(t => t.status === 'timeout').length;
+        const successRate = totalRequests > 0
+            ? Math.round((successCount / totalRequests) * 10000) / 100
+            : 100;
+        const totalDuration = traces.reduce((sum, t) => sum + t.duration, 0);
+        const avgDuration = totalRequests > 0 ? Math.round(totalDuration / totalRequests) : 0;
+        // 活跃线程数（最近 5 分钟有活动的 threadId）
+        const fiveMinAgo = new Date(Date.now() - 5 * 60 * 1000).toISOString();
+        const activeThreadIds = new Set();
+        for (const t of traces) {
+            if (t.threadId !== undefined && t.timestamp >= fiveMinAgo) {
+                activeThreadIds.add(t.threadId);
+            }
+        }
+        // 工具调用统计
+        const toolCallStats = {};
+        for (const t of traces) {
+            if (t.action === 'tool_call') {
+                const toolName = t.input?.tool || t.input?.name || 'unknown';
+                if (!toolCallStats[toolName]) {
+                    toolCallStats[toolName] = {
+                        total: 0,
+                        success: 0,
+                        error: 0,
+                    };
+                }
+                toolCallStats[toolName].total++;
+                if (t.status === 'success') {
+                    toolCallStats[toolName].success++;
+                }
+                else {
+                    toolCallStats[toolName].error++;
+                }
+            }
+        }
+        // 最近的追踪记录
+        const recentTraces = [...traces]
+            .sort((a, b) => b.timestamp.localeCompare(a.timestamp))
+            .slice(0, 10);
+        return {
+            totalRequests,
+            successCount,
+            errorCount,
+            timeoutCount,
+            successRate,
+            avgDuration,
+            activeThreads: activeThreadIds.size,
+            toolCallStats,
+            recentTraces,
+        };
+    }
+    // ===================== 系统健康 =====================
+    /**
+     * 获取系统健康状态
+     */
+    getSystemHealth() {
+        const memUsage = process.memoryUsage();
+        const cpuUsage = process.cpuUsage();
+        // 判断健康状态
+        const heapUsedMB = memUsage.heapUsed / (1024 * 1024);
+        const heapTotalMB = memUsage.heapTotal / (1024 * 1024);
+        const heapUsageRatio = heapTotalMB > 0 ? heapUsedMB / heapTotalMB : 0;
+        let status = 'healthy';
+        if (heapUsageRatio > 0.9 || heapUsedMB > 1500) {
+            status = 'critical';
+        }
+        else if (heapUsageRatio > 0.7 || heapUsedMB > 800) {
+            status = 'warning';
+        }
+        return {
+            status,
+            uptime: Date.now() - this.startTime,
+            memory: {
+                rss: memUsage.rss,
+                heapUsed: memUsage.heapUsed,
+                heapTotal: memUsage.heapTotal,
+                external: memUsage.external,
+            },
+            cpu: {
+                user: cpuUsage.user,
+                system: cpuUsage.system,
+            },
+            activeConnections: this.traceBuffer.filter(t => Date.now() - new Date(t.timestamp).getTime() < 60 * 1000).length,
+            traceCount: this.traceBuffer.length,
+            metricCount: this.metricBuffer.length,
+        };
+    }
+    // ===================== 导出 =====================
+    /**
+     * 导出追踪数据
+     */
+    exportTraces(format = 'json') {
+        const traces = [...this.traceBuffer].sort((a, b) => a.timestamp.localeCompare(b.timestamp));
+        if (format === 'csv') {
+            const headers = [
+                'id',
+                'traceId',
+                'threadId',
+                'action',
+                'status',
+                'duration',
+                'error',
+                'timestamp',
+                'userId',
+                'applicationId',
+            ];
+            const rows = traces.map(t => [
+                t.id,
+                t.traceId,
+                t.threadId ?? '',
+                t.action,
+                t.status,
+                t.duration,
+                t.error ? `"${t.error.replace(/"/g, '""')}"` : '',
+                t.timestamp,
+                t.userId ?? '',
+                t.applicationId ?? '',
+            ].join(','));
+            return [headers.join(','), ...rows].join('\n');
+        }
+        return JSON.stringify(traces, null, 2);
+    }
+    // ===================== 清理 =====================
+    /**
+     * 清空所有数据
+     */
+    clear() {
+        this.traceBuffer = [];
+        this.traceIndex = 0;
+        this.metricBuffer = [];
+        this.metricIndex = 0;
+    }
+};
+exports.ObservabilityService = ObservabilityService;
+/** 环形缓冲区最大容量 */
+ObservabilityService.MAX_TRACES = 10000;
+ObservabilityService.MAX_METRICS = 10000;
+exports.ObservabilityService = ObservabilityService = ObservabilityService_1 = __decorate([
+    (0, core_1.Provide)(),
+    (0, core_1.Scope)(core_1.ScopeEnum.Singleton)
+], ObservabilityService);

package/dist/service/openai.service.d.ts

@@ -0,0 +1,88 @@
+import { Context } from '@midwayjs/koa';
+/** OpenAI 兼容 API 的配置接口 */
+export interface OpenAIConfig {
+    baseUrl: string;
+    apiKey: string;
+    model: string;
+    fallbackModels?: string[];
+    maxTokens: number;
+    temperature: number;
+}
+/**
+ * OpenAI 基础服务
+ * 封装 OpenAI API 调用，提供通用对话、流式输出、模型容错降级等基础能力
+ */
+export declare class OpenAIService {
+    ctx: Context;
+    openaiConfig: OpenAIConfig;
+    protected client: any;
+    init(): Promise<void>;
+    isAvailable(): boolean;
+    /**
+     * 通用 AI 对话流式输出
+     * 支持多轮对话上下文，返回 SSE 事件流
+     */
+    chatStream(messages: Array<{
+        role: string;
+        content: string;
+    }>, systemPrompt?: string): AsyncGenerator<{
+        event: string;
+        data: any;
+    }>;
+    /**
+     * 通用 AI 对话完整调用（非流式）
+     */
+    chatCompletion(systemPrompt: string, userMessage: string, modelOverride?: string): Promise<{
+        content: string;
+        reasoningContent?: string;
+    }>;
+    /**
+     * 带工具调用的单轮流式对话
+     * 子类可传入自定义 tools 参数覆盖默认行为
+     */
+    streamToolRound(messages: any[], modelOverride?: string, tools?: any[]): AsyncGenerator<{
+        kind: 'reasoning' | 'content';
+        text: string;
+    } | {
+        kind: 'result';
+        message: {
+            role: 'assistant';
+            content: string;
+            tool_calls?: any[];
+            reasoning_content?: string;
+        };
+    }>;
+    /**
+     * 从消息列表中流式输出对话内容
+     */
+    streamCompletionFromMessages(messages: any[], modelOverride?: string): AsyncGenerator<{
+        kind: 'reasoning' | 'content';
+        text: string;
+    }>;
+    normalizeTextPayload(payload: any): string;
+    extractReasoningText(message: any): string;
+    extractContentText(message: any): string;
+    parseToolArguments(rawArgs?: string): Record<string, any>;
+    mergeToolCallDelta(target: any[], deltaToolCall: any): void;
+    normalizeToolCalls(toolCalls: any[]): {
+        id: any;
+        type: any;
+        function: {
+            name: any;
+            arguments: any;
+        };
+    }[];
+    isModelEngineError(error: any): boolean;
+    /**
+     * 判断是否为网络连接错误（ECONNRESET / ETIMEDOUT 等）
+     * 这类错误通常可以通过重试或降级恢复
+     */
+    isNetworkError(error: any): boolean;
+    /**
+     * 判断是否为可恢复错误（模型引擎错误 + 网络错误）
+     * 可恢复错误会触发降级/重试，而非直接抛出
+     */
+    isRecoverableError(error: any): boolean;
+    logModelError(scope: string, error: any): void;
+    getFallbackModels(): string[];
+}