coze_lab 0.1.0

package/scripts/openclaw/dist/span-manager.js

@@ -0,0 +1,77 @@
+function generateId(length = 16) {
+    const chars = "0123456789abcdef";
+    let result = "";
+    for (let i = 0; i < length; i++) {
+        result += chars[Math.floor(Math.random() * chars.length)];
+    }
+    return result;
+}
+export class SpanManager {
+    activeSpans = new Map();
+    sessionTraceMap = new Map();
+    turnSpanMap = new Map();
+    generateTraceId() {
+        return generateId(32);
+    }
+    generateSpanId() {
+        return generateId(16);
+    }
+    getOrCreateTraceId(sessionId) {
+        let traceId = this.sessionTraceMap.get(sessionId);
+        if (!traceId) {
+            traceId = this.generateTraceId();
+            this.sessionTraceMap.set(sessionId, traceId);
+        }
+        return traceId;
+    }
+    startSpan(sessionId, name, type, attributes = {}, input, parentSpanId) {
+        const traceId = this.getOrCreateTraceId(sessionId);
+        const spanId = this.generateSpanId();
+        const span = {
+            name,
+            type,
+            startTime: Date.now(),
+            attributes: {
+                ...attributes,
+                "session.id": sessionId,
+            },
+            input,
+            parentSpanId,
+            traceId,
+            spanId,
+        };
+        this.activeSpans.set(spanId, span);
+        return span;
+    }
+    endSpan(spanId, output, additionalAttributes) {
+        const span = this.activeSpans.get(spanId);
+        if (!span)
+            return undefined;
+        span.endTime = Date.now();
+        span.output = output;
+        if (additionalAttributes) {
+            Object.assign(span.attributes, additionalAttributes);
+        }
+        this.activeSpans.delete(spanId);
+        return span;
+    }
+    getSpan(spanId) {
+        return this.activeSpans.get(spanId);
+    }
+    setTurnSpan(turnId, spanId) {
+        this.turnSpanMap.set(turnId, spanId);
+    }
+    getTurnSpanId(turnId) {
+        return this.turnSpanMap.get(turnId);
+    }
+    clearSession(sessionId) {
+        this.sessionTraceMap.delete(sessionId);
+        for (const [turnId, spanId] of this.turnSpanMap.entries()) {
+            const span = this.activeSpans.get(spanId);
+            if (span && span.attributes["session.id"] === sessionId) {
+                this.turnSpanMap.delete(turnId);
+                this.activeSpans.delete(spanId);
+            }
+        }
+    }
+}

package/scripts/openclaw/dist/types.js

@@ -0,0 +1 @@
+export {};

package/scripts/openclaw/openclaw.plugin.json

@@ -0,0 +1,55 @@
+{
+  "id": "openclaw-cozeloop-trace",
+  "name": "OpenClaw CozeLoop Trace",
+  "version": "0.1.12",
+  "description": "Report OpenClaw execution traces to CozeLoop via OpenTelemetry",
+  "type": "plugin",
+  "entry": "./dist/index.js",
+  "configSchema": {
+    "type": "object",
+    "properties": {
+      "endpoint": {
+        "type": "string",
+        "default": "https://api.coze.cn/v1/loop/opentelemetry",
+        "description": "CozeLoop OTLP endpoint URL"
+      },
+      "authorization": {
+        "type": "string",
+        "default": "",
+        "description": "Authorization header value"
+      },
+      "workspaceId": {
+        "type": "string",
+        "default": "",
+        "description": "Cozeloop workspace ID"
+      },
+      "serviceName": {
+        "type": "string",
+        "default": "openclaw-agent",
+        "description": "Service name for traces"
+      },
+      "debug": {
+        "type": "boolean",
+        "default": false,
+        "description": "Enable debug logging"
+      },
+      "batchSize": {
+        "type": "number",
+        "default": 10,
+        "description": "Number of spans to buffer before sending"
+      },
+      "batchInterval": {
+        "type": "number",
+        "default": 5000,
+        "description": "Maximum time (ms) to wait before sending buffered spans"
+      },
+      "enabledHooks": {
+        "type": "array",
+        "items": {
+          "type": "string"
+        },
+        "description": "List of hooks to enable (if not set, all hooks are enabled)"
+      }
+    }
+  }
+}

package/scripts/openclaw/package.json

@@ -0,0 +1,45 @@
+{
+  "name": "@cozeloop/openclaw-cozeloop-trace",
+  "version": "0.1.12",
+  "description": "OpenClaw Plugin for reporting traces to CozeLoop via OpenTelemetry",
+  "type": "module",
+  "main": "dist/index.js",
+  "exports": {
+    ".": "./dist/index.js"
+  },
+  "openclaw": {
+    "extensions": [
+      "./dist/index.js"
+    ]
+  },
+  "scripts": {
+    "build": "tsc",
+    "dev": "tsc --watch",
+    "clean": "rm -rf dist",
+    "prepublishOnly": "npm run build"
+  },
+  "files": [
+    "dist",
+    "openclaw.plugin.json"
+  ],
+  "keywords": [
+    "openclaw",
+    "plugin",
+    "cozeloop",
+    "tracing",
+    "opentelemetry"
+  ],
+  "author": "",
+  "license": "MIT",
+  "devDependencies": {
+    "@types/node": "^20.0.0",
+    "typescript": "^5.0.0"
+  },
+  "dependencies": {
+    "@opentelemetry/api": "^1.7.0",
+    "@opentelemetry/exporter-trace-otlp-proto": "^0.48.0",
+    "@opentelemetry/resources": "^1.22.0",
+    "@opentelemetry/sdk-trace-base": "^1.22.0",
+    "@opentelemetry/semantic-conventions": "^1.22.0"
+  }
+}

package/scripts/shared/cozeloop_refresh.py

@@ -0,0 +1,53 @@
+#!/usr/bin/env python3
+"""
+CozeLoop token refresh hook — runs at SessionStart.
+Checks credentials.json and refreshes the token if expiring soon.
+Writes the fresh token back so subsequent Stop hooks pick it up.
+"""
+import json, os, sys, time, urllib.request
+from pathlib import Path
+
+CLIENT_ID = "56089404009908161803155625287505.app.coze"
+COZE_API  = "https://api.coze.cn"
+THRESHOLD = 10 * 60  # 10 minutes
+CREDS     = Path.home() / ".cozeloop" / "credentials.json"
+
+def load():
+    try: return json.loads(CREDS.read_text())
+    except: return None
+
+def save(c):
+    CREDS.parent.mkdir(parents=True, exist_ok=True)
+    CREDS.write_text(json.dumps(c, indent=2))
+    os.chmod(CREDS, 0o600)
+
+def refresh(rt):
+    try:
+        body = json.dumps({"grant_type":"refresh_token","client_id":CLIENT_ID,"refresh_token":rt}).encode()
+        req = urllib.request.Request(f"{COZE_API}/api/permission/oauth2/token",
+            data=body, headers={"Content-Type":"application/json"})
+        with urllib.request.urlopen(req, timeout=10) as r:
+            d = json.loads(r.read())
+        if d.get("access_token"):
+            save({"access_token":d["access_token"],
+                  "refresh_token":d.get("refresh_token",rt),
+                  "expires_at":d.get("expires_in",0)*1000})
+            return True
+    except Exception as e:
+        print(f"[cozeloop_refresh] refresh failed: {e}", file=sys.stderr)
+    return False
+
+def main():
+    creds = load()
+    if not creds: return
+    remaining = (creds.get("expires_at",0)/1000) - time.time()
+    if remaining > THRESHOLD: return  # still fresh
+    print(f"[cozeloop_refresh] token expiring in {int(remaining)}s, refreshing...", file=sys.stderr)
+    if creds.get("refresh_token"):
+        if refresh(creds["refresh_token"]):
+            print("[cozeloop_refresh] token refreshed OK", file=sys.stderr)
+        else:
+            print("[cozeloop_refresh] refresh failed, token may expire soon", file=sys.stderr)
+
+main()
+