@ai-setting/roy-agent-cli 1.3.14 → 1.3.16

package/dist/cli/src/commands/shared/event-handler.js

@@ -0,0 +1,167 @@
+/**
+ * @fileoverview Event Handler
+ *
+ * 管理环境事件的订阅、队列和处理
+ */
+import { EventMessageFormatter } from "./event-message-formatter.js";
+/**
+ * 事件处理器
+ *
+ * 订阅环境事件，格式化后通过队列管理，agent 空闲时触发处理
+ */
+export class EventHandler {
+    env;
+    eventTypes;
+    onEvent;
+    formatter;
+    isIdle;
+    idleCheckInterval;
+    queue = [];
+    isProcessing = false;
+    isStopped = false;
+    unsubscribe = null;
+    constructor(options) {
+        this.env = options.env;
+        this.eventTypes = options.eventTypes;
+        this.onEvent = options.onEvent;
+        this.formatter = options.formatter ?? new EventMessageFormatter();
+        this.isIdle = options.isIdle ?? (() => true); // 默认认为总是空闲
+        this.idleCheckInterval = options.idleCheckInterval ?? 100;
+    }
+    /**
+     * 启动事件处理
+     * @returns 取消订阅函数
+     */
+    start() {
+        this.isStopped = false;
+        // 使用 subscribeAll 接收所有事件，然后在 handler 中过滤类型
+        // 因为 subscribeTo 不支持通配符模式
+        this.unsubscribe = this.env.subscribeAll((event) => {
+            // 检查事件类型是否匹配
+            if (this.matchesEventType(event.type)) {
+                this.handleEvent(event);
+            }
+        });
+        return () => this.stop();
+    }
+    /**
+     * 检查事件类型是否匹配
+     */
+    matchesEventType(eventType) {
+        for (const pattern of this.eventTypes) {
+            if (this.matchPattern(eventType, pattern)) {
+                return true;
+            }
+        }
+        return false;
+    }
+    /**
+     * 通配符模式匹配
+     * 支持:
+     * - "task.*" 匹配 "task.started", "task.completed" 等
+     * - "task.started" 精确匹配
+     */
+    matchPattern(eventType, pattern) {
+        if (pattern === "*")
+            return true;
+        if (pattern.endsWith(".*")) {
+            const prefix = pattern.slice(0, -2);
+            return eventType === prefix || eventType.startsWith(prefix + ".");
+        }
+        return eventType === pattern;
+    }
+    /**
+     * 停止事件处理
+     */
+    stop() {
+        this.isStopped = true;
+        if (this.unsubscribe) {
+            this.unsubscribe();
+            this.unsubscribe = null;
+        }
+        // 清空队列
+        this.queue = [];
+        this.isProcessing = false;
+    }
+    /**
+     * 处理收到的事件
+     */
+    handleEvent(event) {
+        // 如果已停止，忽略事件
+        if (this.isStopped) {
+            return;
+        }
+        // 格式化事件为用户消息
+        const message = this.formatter.format(event);
+        // 加入队列
+        this.enqueue(message);
+    }
+    /**
+     * 将消息加入队列
+     */
+    enqueue(message) {
+        // 如果已停止，忽略
+        if (this.isStopped) {
+            return;
+        }
+        this.queue.push(message);
+        // 如果没有在处理，触发处理
+        if (!this.isProcessing) {
+            this.processQueue();
+        }
+    }
+    /**
+     * 获取队列长度
+     */
+    getQueueLength() {
+        return this.queue.length;
+    }
+    /**
+     * 检查队列是否为空
+     */
+    isQueueEmpty() {
+        return this.queue.length === 0;
+    }
+    /**
+     * 处理队列中的消息（使用 while 循环避免递归栈溢出）
+     */
+    async processQueue() {
+        this.isProcessing = true;
+        while (!this.isStopped && this.queue.length > 0) {
+            // 等待 agent 空闲
+            await this.waitForIdle();
+            // 如果已停止，退出循环
+            if (this.isStopped) {
+                break;
+            }
+            // 取出队首消息
+            const message = this.queue.shift();
+            if (!message) {
+                break;
+            }
+            try {
+                // 触发事件处理
+                await this.onEvent(message);
+            }
+            catch (error) {
+                console.error(`[EventHandler] 处理事件失败:`, error);
+            }
+        }
+        this.isProcessing = false;
+    }
+    /**
+     * 等待 agent 空闲
+     */
+    async waitForIdle() {
+        while (!this.isStopped && !this.isIdle()) {
+            await this.sleep(this.idleCheckInterval);
+        }
+    }
+    /**
+     * 睡眠工具函数
+     */
+    sleep(ms) {
+        return new Promise(resolve => setTimeout(resolve, ms));
+    }
+}
+//# sourceMappingURL=event-handler.js.map

package/dist/cli/src/commands/shared/event-message-formatter.d.ts

@@ -0,0 +1,150 @@
+/**
+ * @fileoverview Event Message Formatter
+ *
+ * 将 EnvEvent 格式化为用户可读的消息
+ */
+import type { EnvEvent } from "@ai-setting/roy-agent-core";
+/**
+ * 任务事件类型
+ */
+export declare const TaskEventTypes: {
+    readonly TASK_STARTED: "task.started";
+    readonly TASK_PROGRESS: "task.progress";
+    readonly TASK_COMPLETED: "task.completed";
+    readonly TASK_FAILED: "task.failed";
+    readonly TASK_TIMEOUT: "task.timeout";
+    readonly TASK_STOPPED: "task.stopped";
+};
+/**
+ * EventSource 事件类型
+ */
+export declare const EventSourceEventTypes: {
+    readonly EVENT_SOURCE_STARTED: "event-source.started";
+    readonly EVENT_SOURCE_STOPPED: "event-source.stopped";
+    readonly EVENT_SOURCE_EVENT_PREFIX: "event-source.event.";
+};
+/**
+ * 事件元信息接口
+ */
+export interface EventMetadata {
+    eventType?: string;
+    appId?: string;
+    chatId?: string;
+    chatType?: string;
+    messageId?: string;
+    messageType?: string;
+    senderId?: string;
+    tenantKey?: string;
+    [key: string]: unknown;
+}
+/**
+ * 推荐动作接口
+ */
+export interface RecommendedAction {
+    action: string;
+    replyTo?: {
+        appId?: string;
+        chatId?: string;
+        messageId?: string;
+    };
+    rawCommand?: string;
+}
+/**
+ * EventSource 事件负载接口
+ */
+export interface EventSourceEventPayload {
+    sourceId: string;
+    sourceType: string;
+    rawEvent: unknown;
+    message: string;
+    metadata: EventMetadata;
+    recommendedAction?: RecommendedAction;
+    timestamp: number;
+}
+/**
+ * 任务事件负载接口
+ */
+export interface TaskEventPayload {
+    taskId: string;
+    subSessionId: string;
+    parentSessionId: string;
+    description: string;
+    subagentType: string;
+    status?: string;
+    progress?: number;
+    progressMessage?: string;
+    result?: string;
+    error?: string;
+    executionTimeMs?: number;
+    associatedTaskId?: number;
+}
+/**
+ * 格式化选项
+ */
+export interface EventMessageFormatterOptions {
+    /** 消息前缀 */
+    prefix?: string;
+    /** 是否显示详细时间 */
+    showTimestamp?: boolean;
+}
+/**
+ * 事件消息格式化器
+ *
+ * 将 EnvEvent 转换为用户可读的文本消息
+ */
+export declare class EventMessageFormatter {
+    private prefix;
+    private showTimestamp;
+    constructor(options?: EventMessageFormatterOptions);
+    /**
+     * 格式化事件为用户消息
+     */
+    format(event: EnvEvent): string;
+    /**
+     * 格式化任务开始事件
+     */
+    private formatTaskStarted;
+    /**
+     * 格式化任务进度事件
+     */
+    private formatTaskProgress;
+    /**
+     * 格式化任务完成事件
+     */
+    private formatTaskCompleted;
+    /**
+     * 格式化任务失败事件
+     */
+    private formatTaskFailed;
+    /**
+     * 格式化任务超时事件
+     */
+    private formatTaskTimeout;
+    /**
+     * 格式化任务停止事件
+     */
+    private formatTaskStopped;
+    /**
+     * 格式化通用事件
+     */
+    private formatGeneric;
+    /**
+     * 格式化事件源启动事件
+     */
+    private formatEventSourceStarted;
+    /**
+     * 格式化事件源停止事件
+     */
+    private formatEventSourceStopped;
+    /**
+     * 格式化事件源收到的事件（如飞书消息）
+     *
+     * 泛化处理：直接序列化事件为可读文本，不依赖特定字段
+     */
+    private formatEventSourceEvent;
+    /**
+     * 格式化时长
+     */
+    private formatDuration;
+}
+//# sourceMappingURL=event-message-formatter.d.ts.map

package/dist/cli/src/commands/shared/event-message-formatter.js

@@ -0,0 +1,198 @@
+/**
+ * @fileoverview Event Message Formatter
+ *
+ * 将 EnvEvent 格式化为用户可读的消息
+ */
+// ============================================================================
+// Event Type Mapping
+// ============================================================================
+/**
+ * 任务事件类型
+ */
+export const TaskEventTypes = {
+    TASK_STARTED: "task.started",
+    TASK_PROGRESS: "task.progress",
+    TASK_COMPLETED: "task.completed",
+    TASK_FAILED: "task.failed",
+    TASK_TIMEOUT: "task.timeout",
+    TASK_STOPPED: "task.stopped",
+};
+/**
+ * EventSource 事件类型
+ */
+export const EventSourceEventTypes = {
+    EVENT_SOURCE_STARTED: "event-source.started",
+    EVENT_SOURCE_STOPPED: "event-source.stopped",
+    EVENT_SOURCE_EVENT_PREFIX: "event-source.event.",
+};
+// ============================================================================
+// EventMessageFormatter
+// ============================================================================
+/**
+ * 事件消息格式化器
+ *
+ * 将 EnvEvent 转换为用户可读的文本消息
+ */
+export class EventMessageFormatter {
+    prefix;
+    showTimestamp;
+    constructor(options = {}) {
+        this.prefix = options.prefix ?? "[通知]";
+        this.showTimestamp = options.showTimestamp ?? false;
+    }
+    /**
+     * 格式化事件为用户消息
+     */
+    format(event) {
+        const { type, payload } = event;
+        // 根据事件类型格式化
+        switch (type) {
+            case TaskEventTypes.TASK_STARTED:
+                return this.formatTaskStarted(payload);
+            case TaskEventTypes.TASK_PROGRESS:
+                return this.formatTaskProgress(payload);
+            case TaskEventTypes.TASK_COMPLETED:
+                return this.formatTaskCompleted(payload);
+            case TaskEventTypes.TASK_FAILED:
+                return this.formatTaskFailed(payload);
+            case TaskEventTypes.TASK_TIMEOUT:
+                return this.formatTaskTimeout(payload);
+            case TaskEventTypes.TASK_STOPPED:
+                return this.formatTaskStopped(payload);
+            case EventSourceEventTypes.EVENT_SOURCE_STARTED:
+                return this.formatEventSourceStarted(payload);
+            case EventSourceEventTypes.EVENT_SOURCE_STOPPED:
+                return this.formatEventSourceStopped(payload);
+            default:
+                // 处理 event-source.event.* 类型的事件
+                if (type.startsWith(EventSourceEventTypes.EVENT_SOURCE_EVENT_PREFIX)) {
+                    return this.formatEventSourceEvent(event);
+                }
+                return this.formatGeneric(event);
+        }
+    }
+    /**
+     * 格式化任务开始事件
+     */
+    formatTaskStarted(payload) {
+        const desc = payload.description || "未命名任务";
+        return `${this.prefix} 📢 后台任务「${desc}」已启动`;
+    }
+    /**
+     * 格式化任务进度事件
+     */
+    formatTaskProgress(payload) {
+        const desc = payload.description || "未命名任务";
+        const progress = payload.progress !== undefined ? `${payload.progress}%` : "";
+        const message = payload.progressMessage || "";
+        let result = `${this.prefix} 🔄 后台任务「${desc}」进度更新`;
+        if (progress)
+            result += `: ${progress}`;
+        if (message)
+            result += ` - ${message}`;
+        return result;
+    }
+    /**
+     * 格式化任务完成事件
+     */
+    formatTaskCompleted(payload) {
+        const desc = payload.description || "未命名任务";
+        const result = payload.result;
+        const time = payload.executionTimeMs ? `(${this.formatDuration(payload.executionTimeMs)})` : "";
+        let message = `${this.prefix} ✅ 后台任务「${desc}」已完成 ${time}`;
+        if (result) {
+            // 截断过长结果
+            const truncatedResult = result.length > 100
+                ? result.substring(0, 100) + "..."
+                : result;
+            message += `\n结果: ${truncatedResult}`;
+        }
+        return message;
+    }
+    /**
+     * 格式化任务失败事件
+     */
+    formatTaskFailed(payload) {
+        const desc = payload.description || "未命名任务";
+        const error = payload.error || "未知错误";
+        const time = payload.executionTimeMs ? `(${this.formatDuration(payload.executionTimeMs)})` : "";
+        let message = `${this.prefix} ❌ 后台任务「${desc}」失败 ${time}`;
+        if (error) {
+            // 截断过长错误信息
+            const truncatedError = error.length > 100
+                ? error.substring(0, 100) + "..."
+                : error;
+            message += `\n错误: ${truncatedError}`;
+        }
+        return message;
+    }
+    /**
+     * 格式化任务超时事件
+     */
+    formatTaskTimeout(payload) {
+        const desc = payload.description || "未命名任务";
+        const time = payload.executionTimeMs ? `(${this.formatDuration(payload.executionTimeMs)})` : "";
+        return `${this.prefix} ⏱️ 后台任务「${desc}」超时 ${time}`;
+    }
+    /**
+     * 格式化任务停止事件
+     */
+    formatTaskStopped(payload) {
+        const desc = payload.description || "未命名任务";
+        const time = payload.executionTimeMs ? `(${this.formatDuration(payload.executionTimeMs)})` : "";
+        return `${this.prefix} 🛑 后台任务「${desc}」已停止 ${time}`;
+    }
+    /**
+     * 格式化通用事件
+     */
+    formatGeneric(event) {
+        return `${this.prefix} 收到事件: ${event.type}`;
+    }
+    // ============================================================================
+    // EventSource 事件格式化
+    // ============================================================================
+    /**
+     * 格式化事件源启动事件
+     */
+    formatEventSourceStarted(payload) {
+        return `${this.prefix} 📡 事件源「${payload.sourceId}」(类型: ${payload.sourceType}) 已启动`;
+    }
+    /**
+     * 格式化事件源停止事件
+     */
+    formatEventSourceStopped(payload) {
+        return `${this.prefix} 📡 事件源「${payload.sourceId}」(类型: ${payload.sourceType}) 已停止`;
+    }
+    /**
+     * 格式化事件源收到的事件（如飞书消息）
+     *
+     * 泛化处理：直接序列化事件为可读文本，不依赖特定字段
+     */
+    formatEventSourceEvent(event) {
+        const payload = event.payload;
+        const sourceType = payload?.sourceType || "unknown";
+        const message = payload?.message || "";
+        // 直接使用 EventSourceComponent 格式化好的 message
+        if (message) {
+            return `[${sourceType}] ${message}`;
+        }
+        // 如果没有预格式化的 message，整个事件序列化
+        return `[${sourceType}] ${JSON.stringify(payload, null, 2)}`;
+    }
+    // ============================================================================
+    // 辅助方法
+    // ============================================================================
+    /**
+     * 格式化时长
+     */
+    formatDuration(ms) {
+        if (ms < 1000)
+            return `${ms}ms`;
+        if (ms < 60000)
+            return `${(ms / 1000).toFixed(1)}秒`;
+        const minutes = Math.floor(ms / 60000);
+        const seconds = Math.floor((ms % 60000) / 1000);
+        return `${minutes}分${seconds}秒`;
+    }
+}
+//# sourceMappingURL=event-message-formatter.js.map

package/dist/cli/src/commands/shared/index.d.ts

@@ -0,0 +1,10 @@
+/**
+ * @fileoverview Shared modules for CLI commands
+ *
+ * 导出共享模块供 act、interactive 等命令复用
+ */
+export { SessionManager } from "./session-manager";
+export type { SessionInitResult, SessionManagerOptions } from "./session-manager";
+export { QueryExecutor } from "./query-executor";
+export type { QueryExecutorOptions, StreamOptions } from "./query-executor";
+//# sourceMappingURL=index.d.ts.map

package/dist/cli/src/commands/shared/index.js

@@ -0,0 +1,8 @@
+/**
+ * @fileoverview Shared modules for CLI commands
+ *
+ * 导出共享模块供 act、interactive 等命令复用
+ */
+export { SessionManager } from "./session-manager.js";
+export { QueryExecutor } from "./query-executor.js";
+//# sourceMappingURL=index.js.map

package/dist/cli/src/commands/shared/query-executor.d.ts

@@ -0,0 +1,81 @@
+/**
+ * @fileoverview Query Executor - 查询执行共享逻辑
+ *
+ * 从 act.ts 抽取的查询执行逻辑，供 interactive 复用
+ *
+ * 功能：
+ * - 初始化 SummaryAgent
+ * - 订阅 LLM/上下文事件
+ * - 流式输出
+ * - ContextHandler 执行
+ * - 支持 abort 流式输出
+ */
+import { OutputService } from "../../services/output.service";
+import type { BaseEnvironment, SessionComponent, LLMComponent, PromptComponent } from "@ai-setting/roy-agent-core";
+export interface QueryExecutorOptions {
+    env: BaseEnvironment;
+    sessionComponent: SessionComponent;
+    output: OutputService;
+    quiet: boolean;
+}
+export interface StreamOptions {
+    showReasoning: boolean;
+    showToolCalls: boolean;
+    showToolResults: boolean;
+}
+/**
+ * QueryExecutor - 查询执行器
+ *
+ * 封装查询执行的完整流程
+ */
+export declare class QueryExecutor {
+    private env;
+    private sessionComponent;
+    private output;
+    private quiet;
+    private streamService;
+    private unsubscribe;
+    private streamOptions;
+    private _pendingContextConfig;
+    constructor(options: QueryExecutorOptions);
+    /**
+     * 初始化 SummaryAgent（用于 compact）
+     */
+    initSummaryAgent(llmComponent: LLMComponent, promptComponent: PromptComponent): void;
+    /**
+     * 初始化流式输出配置（设置 context window 信息）
+     */
+    initStreamOutput(llmComponent: LLMComponent | null, providerId?: string): void;
+    /**
+     * 订阅事件（用于流式输出）
+     */
+    subscribeToEvents(options: StreamOptions): void;
+    /**
+     * 处理事件
+     */
+    protected handleEvent(event: any): void;
+    /**
+     * 执行查询
+     */
+    execute(message: string, sessionId: string, streamOptions?: Partial<StreamOptions>, traceId?: string): Promise<string>;
+    /**
+     * 清理资源
+     */
+    dispose(): void;
+    /**
+     * 获取流式输出服务的完整文本
+     */
+    getFullText(): string;
+    /**
+     * 手动压缩会话上下文
+     *
+     * 复用 ContextHandlerService.compactSession 逻辑
+     */
+    compact(sessionId: string): Promise<{
+        success: boolean;
+        checkpoint?: any;
+        deletedMessageCount?: number;
+        error?: string;
+    }>;
+}
+//# sourceMappingURL=query-executor.d.ts.map