coze_lab 0.1.4 → 0.1.5

package/scripts/openclaw/dist/index.js

@@ -446,6 +446,32 @@ function resolveChannelId(ctx, eventFrom, defaultValue = "system/unknown") {
 let lastUserChannelId;
 let lastUserTraceContext;
 let lastOpenclawSessionId;
+// Session-level coze-context cache. In a Coze session only the FIRST user
+// message carries the <coze-context> block; /loop follow-ups and later turns
+// do not. We cache the most recently seen context (per openclaw session id, plus
+// a global last-seen fallback) so that ensureRootSpan can still attach the
+// trace to the right Coze session on those later turns instead of dropping it.
+const cozeContextBySession = new Map();
+let lastCozeContext;
+function rememberCozeContext(input, sessionId) {
+    const parsed = parseCozeContext(input);
+    if (Object.keys(parsed).length === 0)
+        return parsed;
+    lastCozeContext = parsed;
+    if (sessionId)
+        cozeContextBySession.set(sessionId, parsed);
+    return parsed;
+}
+function resolveCozeContext(input, sessionId) {
+    // Prefer context parsed from this turn's input; fall back to the per-session
+    // cache, then the global last-seen context.
+    const fromInput = parseCozeContext(input);
+    if (Object.keys(fromInput).length > 0)
+        return fromInput;
+    if (sessionId && cozeContextBySession.has(sessionId))
+        return cozeContextBySession.get(sessionId);
+    return lastCozeContext || {};
+}
 // Active agent context: set in before_agent_start, cleared in agent_end.
 // All hooks between these two (llm_input, llm_output, tool calls, messages)
 // use this to ensure every span lands in the same Trace.
@@ -814,6 +840,9 @@ const cozeloopTracePlugin = {
                         lastUserTraceContext = ctx;
                         ctx.userInput = event.content;
                         lastUserInput = event.content;
+                        // Cache any coze-context as soon as we see the user input,
+                        // so later /loop turns (which carry none) can inherit it.
+                        rememberCozeContext(event.content, ctx.openclawSessionId || lastOpenclawSessionId);
                         if (config.debug) {
                             api.logger.info(`[CozeloopTrace] Saved user context: channelId=${channelId}, traceId=${ctx.traceId}`);
                         }
@@ -895,6 +924,9 @@ const cozeloopTracePlugin = {
                         lastUserInput = event.prompt;
                     }
                 }
+                // Cache coze-context from the prompt too (covers sessions where no
+                // message_received fired), so /loop follow-ups can inherit it.
+                rememberCozeContext(event.prompt, ctx.openclawSessionId || lastOpenclawSessionId);
                 // Fallback: ensure root + agent spans exist in case before_agent_start
                 // was not fired (older OpenClaw versions or resumed sessions).
                 const channelIdForSpans = activeAgentChannelId || rawChannelId;
@@ -1166,9 +1198,20 @@ const cozeloopTracePlugin = {
         // 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) => {
-            // Only trace sessions that carry coze-context — no context means this
-            // is not a Coze-originated session and we skip it entirely.
-            if (!Object.keys(parseCozeContext(ctx.userInput)).length) {
+            // 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-
+            // session cache (and global last-seen) instead of dropping the trace.
+            // Only skip when NO context has ever been seen for this session.
+            const ocSessionId = ctx.openclawSessionId || lastOpenclawSessionId;
+            rememberCozeContext(ctx.userInput, ocSessionId);
+            let cozeCtx = resolveCozeContext(ctx.userInput, ocSessionId);
+            if (Object.keys(cozeCtx).length === 0) {
+                // Try the fallback user inputs too before giving up.
+                const fallbackInput = lastUserTraceContext?.userInput || lastUserInput;
+                cozeCtx = resolveCozeContext(fallbackInput, ocSessionId);
+            }
+            if (Object.keys(cozeCtx).length === 0) {
                 return;
             }
             // Check both: rootSpanStartTime indicates we created a root span before,
@@ -1207,7 +1250,7 @@ const cozeloopTracePlugin = {
                     "run.id": ctx.runId,
                     "turn.id": ctx.turnId,
                     "openclaw.channel_id": channelId,
-                    ...parseCozeContext(ctx.userInput),
+                    ...cozeCtx,
                 },
                 input: ctx.userInput,
                 traceId: ctx.traceId,
@@ -1237,7 +1280,7 @@ const cozeloopTracePlugin = {
                     "run.id": ctx.runId,
                     "turn.id": ctx.turnId,
                     "openclaw.channel_id": channelId,
-                    ...parseCozeContext(ctx.userInput),
+                    ...resolveCozeContext(ctx.userInput, ctx.openclawSessionId || lastOpenclawSessionId),
                 },
                 traceId: ctx.traceId,
                 spanId: ctx.agentSpanId,