coze_lab 0.1.48 → 0.1.49

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "coze_lab",
-  "version": "0.1.48",
+  "version": "0.1.49",
   "description": "Configure local AI agents (Claude Code, Codex, OpenClaw) to report traces to CozeLoop",
   "keywords": [
     "cozeloop",

package/scripts/codex/cozeloop_hook.py

@@ -131,6 +131,127 @@ else:
 
 # --- Configuration ---
 DEBUG = os.environ.get("CC_COZELOOP_DEBUG", "").lower() == "true"
+_DEFAULT_WORKSPACE_ID = "7649231955045072915"  # hardcoded spaceID fallback
+
+# 实时 tool span 上报的批量间隔（秒）：距上次上报不足此值则本次不发，把已结束的 tool span
+# 攒到下次一起发（最多每 5s 一批）。终态(Stop/SubagentStop)不受限，立即收尾。
+REALTIME_BATCH_INTERVAL = float(os.environ.get("COZELOOP_REALTIME_BATCH_INTERVAL", "5"))
+
+
+# --- coze-context parsing -------------------------------------------------
+# User messages may embed a block like:
+#   <coze-context>
+#   account_id: 0
+#   agent_id: 7644920552473395499
+#   session_id: 7644919579054997796
+#   message_id: 04dd5246-...
+#   </coze-context>
+# We parse its key:value pairs and inject them into the trace.
+
+_COZE_CTX_OPEN = "<coze-context>"
+_COZE_CTX_CLOSE = "</coze-context>"
+
+
+def parse_coze_context(text: str) -> Dict[str, str]:
+    """Extract the LAST <coze-context> block's key:value pairs from text.
+
+    Returns {} if no block is present. Tag keys are prefixed with
+    'coze_' by the caller; here we return raw keys as written.
+    """
+    if not text or _COZE_CTX_OPEN not in text:
+        return {}
+    # Take the last occurrence (latest context wins).
+    open_idx = text.rfind(_COZE_CTX_OPEN)
+    close_idx = text.find(_COZE_CTX_CLOSE, open_idx)
+    if close_idx == -1:
+        return {}
+    body = text[open_idx + len(_COZE_CTX_OPEN):close_idx]
+    # The block may arrive with real newlines, OR with literal backslash-n
+    # (e.g. when the whole message is an embedded JSON string that was never
+    # un-escaped). Normalize both forms before splitting into lines.
+    body = body.replace("\\r\\n", "\n").replace("\\n", "\n").replace("\\r", "\n")
+    result: Dict[str, str] = {}
+    for line in body.splitlines():
+        line = line.strip()
+        if not line or ":" not in line:
+            continue
+        key, _, value = line.partition(":")
+        key = key.strip()
+        value = value.strip()
+        if key:
+            result[key] = value
+    return result
+
+
+def turn_coze_context(turn: Dict[str, Any]) -> Dict[str, str]:
+    """Extract coze-context from a grouped turn with fallbacks for Codex rollout shapes."""
+    texts = [turn.get("user_message_text", "")]
+    for msg in turn.get("input_messages", []):
+        if isinstance(msg, dict):
+            content = msg.get("content", "")
+            if isinstance(content, str):
+                texts.append(content)
+            elif isinstance(content, list):
+                texts.append(extract_message_content_text({"content": content}))
+            elif isinstance(content, dict):
+                texts.append(extract_message_content_text({"content": [content]}))
+    user_payload = turn.get("user_message")
+    if isinstance(user_payload, dict):
+        texts.append(extract_message_content_text(user_payload))
+    for text in texts:
+        ctx = parse_coze_context(text)
+        if ctx:
+            return ctx
+    return {}
+
+
+# --- trace upload failure / logid capture ---------------------------------
+def _extract_logid(msg: str) -> str:
+    """Pull the server logid out of an SDK error message, if present.
+
+    SDK failure messages embed it as 'logid=XXXX' (sometimes within brackets).
+    """
+    if not msg:
+        return ""
+    marker = "logid="
+    idx = msg.find(marker)
+    if idx == -1:
+        return ""
+    rest = msg[idx + len(marker):]
+    logid = []
+    for ch in rest:
+        if ch.isalnum():
+            logid.append(ch)
+        else:
+            break
+    return "".join(logid)
+
+
+def _make_finish_event_processor(upload_events: Optional[List[str]] = None):
+    """Return a trace_finish_event_processor that surfaces failures + logid.
+
+    The CozeLoop SDK calls this for each flush event; on failure we print the
+    server logid to stderr so it can be handed to platform support for tracing
+    the root cause (e.g. via `bytedcli log get-logid-log <logid>`).
+    """
+    def _processor(info):
+        try:
+            if not getattr(info, "is_event_fail", False):
+                hook_log("upload success")
+                return
+            detail = getattr(info, "detail_msg", "") or ""
+            if upload_events is not None:
+                upload_events.append(detail or "trace export failed")
+            logid = _extract_logid(detail)
+            if logid:
+                hook_log(f"upload failed logid={logid} detail={detail[:500]}")
+                print(f"[CozeLoop] 上报失败 logid={logid} (可用 bytedcli log get-logid-log {logid} 排查)", file=sys.stderr)
+            else:
+                hook_log(f"upload failed detail={detail[:500]}")
+                print(f"[CozeLoop] 上报失败: {detail[:300]}", file=sys.stderr)
+        except Exception:
+            pass
+    return _processor
 
 
 def _normalize_api_base_url(raw: str) -> str: