@code4bug/jarvis-agent 1.0.3 → 1.1.4

package/dist/tools/subscribeMessage.js

@@ -0,0 +1,59 @@
+import { getBus } from '../core/busAccess.js';
+export const subscribeMessage = {
+    name: 'subscribe_message',
+    description: [
+        '订阅指定频道，等待其他 Agent 发布消息后返回内容。',
+        '适用场景：',
+        '  - SubAgent B 等待 SubAgent A 完成并发布结果后再继续执行',
+        '  - 实现 Agent 间的同步协调（生产者-消费者模式）',
+        '  - 读取频道最新历史消息（设置 read_latest=true 时不阻塞）',
+        '注意：默认超时 30 秒，超时后返回 null，Agent 应处理超时情况。',
+    ].join('\n'),
+    parameters: {
+        channel: {
+            type: 'string',
+            description: '要订阅的频道名称',
+            required: true,
+        },
+        read_latest: {
+            type: 'string',
+            description: '可选。设为 "true" 时，若频道已有历史消息则立即返回最新一条，不阻塞等待。默认 false',
+            required: false,
+        },
+        from_offset: {
+            type: 'string',
+            description: '可选。从指定位置（0-based）开始消费。若该位置已有消息则立即返回，否则阻塞等待。用于避免错过在订阅前已发布的消息',
+            required: false,
+        },
+        timeout_seconds: {
+            type: 'string',
+            description: '可选。等待超时秒数，默认 30 秒。超时返回空结果',
+            required: false,
+        },
+    },
+    execute: async (args) => {
+        const channel = (args.channel || '').trim();
+        if (!channel)
+            throw new Error('缺少必填参数: channel');
+        const readLatest = (args.read_latest || '').toLowerCase() === 'true';
+        const timeoutSec = parseInt(args.timeout_seconds || '30', 10);
+        const timeoutMs = Math.max(1000, Math.min(timeoutSec * 1000, 300_000));
+        const fromOffsetRaw = (args.from_offset || '').trim();
+        const fromOffset = fromOffsetRaw !== '' ? parseInt(fromOffsetRaw, 10) : undefined;
+        const bus = await getBus();
+        if (readLatest) {
+            const history = await bus.getHistory(channel, 1);
+            if (history.length === 0)
+                return `频道 "${channel}" 暂无历史消息`;
+            return formatMessage(history[0]);
+        }
+        const msg = await bus.subscribe(channel, timeoutMs, fromOffset);
+        if (!msg)
+            return `订阅频道 "${channel}" 超时（${timeoutSec}s），未收到消息`;
+        return formatMessage(msg);
+    },
+};
+function formatMessage(msg) {
+    const time = new Date(msg.timestamp).toISOString();
+    return `[频道: ${msg.channel}] [来自: ${msg.from}] [时间: ${time}]\n${msg.payload}`;
+}

package/dist/tools/writeFile.d.ts

@@ -1,2 +1,2 @@
-import { Tool } from '../types/index';
+import { Tool } from '../types/index.js';
 export declare const writeFile: Tool;

package/dist/tools/writeFile.js

@@ -1,6 +1,6 @@
 import fs from 'fs';
 import path from 'path';
-import { detectSensitiveContent } from '../core/safeguard';
+import { detectSensitiveContent } from '../core/safeguard.js';
 /**
  * 将 unified diff 补丁应用到原始文本上。
  * 支持标准 unified diff 格式（@@ -a,b +c,d @@ 开头的 hunk）。

package/dist/types/index.d.ts

@@ -24,6 +24,8 @@ export interface Message {
     abortHint?: string;
     /** 并行执行组 ID，同组工具同时运行 */
     parallelGroupId?: string;
+    /** 来源 SubAgent 标识，格式如 "ResearchAgent"，用于 UI 前缀展示 */
+    subAgentId?: string;
 }
 export type ContentBlock = {
     type: 'text';
@@ -45,11 +47,18 @@ export interface ToolParameter {
     description: string;
     required?: boolean;
 }
+/** 工具执行时可选的额外回调，用于特殊工具（如 dispatch_subagent）向主线程推送实时事件 */
+export interface ToolCallbacks {
+    /** SubAgent 产生新消息时推送到主线程 UI */
+    onSubAgentMessage?: (msg: Message) => void;
+    /** SubAgent 更新已有消息 */
+    onSubAgentUpdateMessage?: (id: string, updates: Partial<Message>) => void;
+}
 export interface Tool {
     name: string;
     description: string;
     parameters: Record<string, ToolParameter>;
-    execute: (args: Record<string, unknown>, abortSignal?: AbortSignal) => Promise<string>;
+    execute: (args: Record<string, unknown>, abortSignal?: AbortSignal, toolCallbacks?: ToolCallbacks) => Promise<string>;
 }
 export interface Session {
     id: string;
@@ -93,3 +102,42 @@ export interface TranscriptMessage {
 export interface LLMService {
     streamMessage: (transcript: TranscriptMessage[], tools: Tool[], callbacks: StreamCallbacks, abortSignal?: AbortSignal) => Promise<void>;
 }
+/** SubAgent 状态 */
+export type SubAgentStatus = 'idle' | 'running' | 'done' | 'error' | 'aborted';
+/** SubAgent 任务描述 */
+export interface SubAgentTask {
+    /** 任务唯一 ID */
+    taskId: string;
+    /** 任务描述（发给 SubAgent 的指令） */
+    instruction: string;
+    /** 可选：限制 SubAgent 可用的工具名列表（为空则继承全部工具） */
+    allowedTools?: string[];
+    /** 可选：SubAgent 角色描述，注入 system prompt */
+    role?: string;
+    /** 可选：初始上下文 transcript */
+    contextTranscript?: TranscriptMessage[];
+    /** 可选：直接指定 SubAgent 的 system prompt，跳过主 Agent 的 agent 文件加载 */
+    systemPrompt?: string;
+}
+/** SubAgent 执行结果 */
+export interface SubAgentResult {
+    taskId: string;
+    status: 'done' | 'error' | 'aborted';
+    /** 最终输出文本 */
+    output: string;
+    /** 执行过程中产生的消息列表（用于主 Agent 展示） */
+    messages: Message[];
+    /** 更新后的 transcript */
+    transcript: TranscriptMessage[];
+    /** 错误信息（status=error 时） */
+    error?: string;
+}
+/** AgentManager 向外暴露的任务派发回调 */
+export interface AgentManagerCallbacks {
+    /** SubAgent 产生新消息时（用于 UI 展示） */
+    onSubAgentMessage: (taskId: string, msg: Message) => void;
+    /** SubAgent 状态变更 */
+    onSubAgentStatusChange: (taskId: string, status: SubAgentStatus) => void;
+    /** 所有 SubAgent 完成后汇总回调 */
+    onAllDone: (results: SubAgentResult[]) => void;
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@code4bug/jarvis-agent",
-  "version": "1.0.3",
+  "version": "1.1.4",
   "description": "基于 React + TypeScript + Ink 构建的命令行智能体交互界面",
   "type": "module",
   "main": "dist/index.js",