coze_lab 0.1.5 → 0.1.7

package/scripts/openclaw/dist/index.js

@@ -511,6 +511,31 @@ const cozeloopTracePlugin = {
             enabledHooks: pluginConfig.enabledHooks,
         };
         const exporter = new CozeloopExporter(api, config);
+        // per-agent trace 放行：traceAgentIds 为 onboard 写入的 allowlist（小写归一）。
+        // 非空时，仅 allowlist 内的 agentId 上报 trace；为空 / 未配置则全部放行（本地全局模式）。
+        const traceAgentIds = Array.isArray(pluginConfig.traceAgentIds)
+            ? pluginConfig.traceAgentIds.map((s) => String(s).trim().toLowerCase()).filter(Boolean)
+            : [];
+        const isAgentAllowed = (hookCtx) => {
+            if (traceAgentIds.length === 0)
+                return true; // 空 allowlist = 全部放行
+            const agentId = resolveAgentIdFromHookCtx(hookCtx || {});
+            const allowed = traceAgentIds.includes(agentId);
+            if (!allowed && config.debug) {
+                api.logger.info(`[CozeloopTrace] agent "${agentId}" not in traceAgentIds, skip trace`);
+            }
+            return allowed;
+        };
+        // 包裹 api.on：带 hookCtx 的 hook 在 agentId 不在 allowlist 时直接跳过，
+        // 不进入任何上报逻辑。gateway_start/stop 等无 hookCtx 的生命周期 hook 不拦截。
+        const rawOn = api.on.bind(api);
+        const on = (name, handler) => {
+            rawOn(name, async (event, hookCtx) => {
+                if (hookCtx !== undefined && !isAgentAllowed(hookCtx))
+                    return;
+                return handler(event, hookCtx);
+            });
+        };
         const contextByChannelId = new Map();
         const contextByRunId = new Map();
         const shouldHookEnabled = (hookName) => {
@@ -787,6 +812,14 @@ const cozeloopTracePlugin = {
         });
         if (shouldHookEnabled("gateway_start")) {
             api.on("gateway_start", async (event) => {
+                // gateway_start 是进程级事件、无 agent 归属。云端 per-agent 模式
+                // （traceAgentIds 非空）下不上报这条无归属 span，避免越过 allowlist。
+                if (traceAgentIds.length > 0) {
+                    if (config.debug) {
+                        api.logger.info("[CozeloopTrace] traceAgentIds set, skip gateway_start span");
+                    }
+                    return;
+                }
                 const now = Date.now();
                 const { ctx, channelId } = getOrCreateContext("system/gateway", undefined, "gateway_start");
                 const span = createSpan(ctx, channelId, "gateway_start", "gateway", now, now, {
@@ -800,7 +833,7 @@ const cozeloopTracePlugin = {
             });
         }
         if (shouldHookEnabled("session_start")) {
-            api.on("session_start", async (event, hookCtx) => {
+            on("session_start", async (event, hookCtx) => {
                 // Refresh token if expiring soon (< 10 min)
                 await exporter.refreshAuthIfNeeded();
                 const rawChannelId = resolveChannelId(hookCtx, event.sessionId);
@@ -817,7 +850,7 @@ const cozeloopTracePlugin = {
             });
         }
         if (shouldHookEnabled("message_received")) {
-            api.on("message_received", async (event, hookCtx) => {
+            on("message_received", async (event, hookCtx) => {
                 const rawChannelId = resolveChannelId(hookCtx, event.from || event.metadata?.senderId);
                 if (config.debug) {
                     api.logger.info(`[CozeloopTrace] message_received hookCtx: ${JSON.stringify({ channelId: hookCtx.channelId, sessionKey: hookCtx.sessionKey, conversationId: hookCtx.conversationId })}, event.from=${event.from}`);
@@ -858,7 +891,7 @@ const cozeloopTracePlugin = {
             });
         }
         if (shouldHookEnabled("message_sending")) {
-            api.on("message_sending", async (event, hookCtx) => {
+            on("message_sending", async (event, hookCtx) => {
                 if (lastUserTraceContext) {
                     lastUserTraceContext.lastOutput = event.content;
                     if (config.debug) {
@@ -876,7 +909,7 @@ const cozeloopTracePlugin = {
             });
         }
         if (shouldHookEnabled("message_sent")) {
-            api.on("message_sent", async (event, hookCtx) => {
+            on("message_sent", async (event, hookCtx) => {
                 if (event.content && event.success) {
                     if (lastUserTraceContext) {
                         lastUserTraceContext.lastOutput = event.content;
@@ -899,7 +932,7 @@ const cozeloopTracePlugin = {
         let lastLlmStartTime = undefined;
         let lastLlmSpanId = undefined;
         if (shouldHookEnabled("llm_input")) {
-            api.on("llm_input", async (event, hookCtx) => {
+            on("llm_input", async (event, hookCtx) => {
                 const rawChannelId = resolveChannelId(hookCtx);
                 if (config.debug) {
                     api.logger.info(`[CozeloopTrace] llm_input hookCtx: ${JSON.stringify({ channelId: hookCtx.channelId, sessionKey: hookCtx.sessionKey, conversationId: hookCtx.conversationId })}, event.runId=${event.runId}`);
@@ -990,7 +1023,7 @@ const cozeloopTracePlugin = {
             });
         }
         if (shouldHookEnabled("llm_output")) {
-            api.on("llm_output", async (event, hookCtx) => {
+            on("llm_output", async (event, hookCtx) => {
                 const rawChannelId = resolveChannelId(hookCtx);
                 if (config.debug) {
                     api.logger.info(`[CozeloopTrace][DEBUG] llm_output event.usage=${JSON.stringify(event.usage)}`);
@@ -1086,7 +1119,7 @@ const cozeloopTracePlugin = {
             });
         }
         if (shouldHookEnabled("before_tool_call")) {
-            api.on("before_tool_call", async (event, hookCtx) => {
+            on("before_tool_call", async (event, hookCtx) => {
                 const rawChannelId = resolveChannelId(hookCtx);
                 if (config.debug) {
                     api.logger.info(`[CozeloopTrace] before_tool_call hookCtx: ${JSON.stringify({ channelId: hookCtx.channelId, sessionKey: hookCtx.sessionKey, conversationId: hookCtx.conversationId })}, toolName=${event.toolName}`);
@@ -1107,7 +1140,7 @@ const cozeloopTracePlugin = {
             });
         }
         if (shouldHookEnabled("after_tool_call")) {
-            api.on("after_tool_call", async (event, hookCtx) => {
+            on("after_tool_call", async (event, hookCtx) => {
                 if (config.debug) {
                     api.logger.info(`[CozeloopTrace] after_tool_call hookCtx: ${JSON.stringify({ channelId: hookCtx.channelId, sessionKey: hookCtx.sessionKey, conversationId: hookCtx.conversationId })}, toolName=${event.toolName}`);
                 }
@@ -1194,10 +1227,32 @@ const cozeloopTracePlugin = {
                 traceFinalized = false;
             }, 200);
         };
+        // OpenClaw runtime 周期性发送的心跳轮询消息，不是真实对话，整条 trace 丢弃。
+        const HEARTBEAT_INPUT = "[OpenClaw heartbeat poll]";
+        const isHeartbeatInput = (input) => {
+            if (typeof input === "string") {
+                return input.trim() === HEARTBEAT_INPUT;
+            }
+            // content 可能是 [{type:'text', text:'...'}] 形式
+            if (Array.isArray(input)) {
+                return input.some((p) => p && typeof p === "object"
+                    && typeof p.text === "string" && p.text.trim() === HEARTBEAT_INPUT);
+            }
+            return false;
+        };
         // Helper: ensure root openclaw_request span is started for a given context.
         // Must be called before creating the agent span so that the exporter's
         // currentRootContext is set and the agent span becomes a proper child.
         const ensureRootSpan = async (ctx, channelId) => {
+            // 心跳轮询消息：不是真实对话，整条 trace 不上报（沿用下方“无 coze-context 即 return”的同款范式）。
+            const heartbeatInput = ctx.userInput
+                || lastUserTraceContext?.userInput || lastUserInput;
+            if (isHeartbeatInput(heartbeatInput)) {
+                if (config.debug) {
+                    api.logger.info(`[CozeloopTrace] skip heartbeat poll trace, traceId=${ctx.traceId}`);
+                }
+                return;
+            }
             // Resolve coze-context with SESSION-LEVEL inheritance: only the first
             // user message of a Coze session carries the <coze-context> block, so
             // /loop follow-ups and later turns have none. Fall back to the per-
@@ -1295,7 +1350,7 @@ const cozeloopTracePlugin = {
             }
         };
         if (shouldHookEnabled("before_agent_start")) {
-            api.on("before_agent_start", async (event, hookCtx) => {
+            on("before_agent_start", async (event, hookCtx) => {
                 const rawChannelId = resolveChannelId(hookCtx);
                 const agentId = hookCtx.agentId || event.agentId || "main";
                 if (config.debug) {
@@ -1313,7 +1368,7 @@ const cozeloopTracePlugin = {
             });
         }
         if (shouldHookEnabled("agent_end")) {
-            api.on("agent_end", async (event, hookCtx) => {
+            on("agent_end", async (event, hookCtx) => {
                 const rawChannelId = resolveChannelId(hookCtx);
                 if (config.debug) {
                     api.logger.info(`[CozeloopTrace] agent_end hookCtx: ${JSON.stringify({ channelId: hookCtx.channelId, sessionKey: hookCtx.sessionKey, conversationId: hookCtx.conversationId })}`);
@@ -1333,7 +1388,7 @@ const cozeloopTracePlugin = {
         // versions), finalize the trace here so that agent + root spans get ended
         // and exported.
         if (shouldHookEnabled("session_end")) {
-            api.on("session_end", async (event, hookCtx) => {
+            on("session_end", async (event, hookCtx) => {
                 const rawChannelId = resolveChannelId(hookCtx, event.sessionId);
                 if (config.debug) {
                     api.logger.info(`[CozeloopTrace] session_end: ${rawChannelId}`);