coze_lab 0.1.13 → 0.1.15

package/README.md

@@ -8,6 +8,9 @@ Configure local AI agents (Claude Code, Codex, OpenClaw) to report traces to Coz
 # First-time setup — triggers browser OAuth authorization
 npx coze_lab --agent=<type>
 
+# Cloud setup for a managed agent
+npx coze_lab --cloud --agent-id=<agentId>
+
 # Auth-only commands (no agent configuration)
 npx coze_lab --login      # Device Code login only
 npx coze_lab --status     # Show current authorization status
@@ -20,6 +23,9 @@ npx coze_lab --logout     # Clear cached credentials
 | Parameter | Required | Values / Effect |
 |-----------|----------|-----------------|
 | `--agent` | ✓ (for setup) | `claude-code`, `codex`, `openclaw` |
+| `--agent-id` | — | Resolve `~/.coze/agents/<agentId>/config.json` and write per-agent config |
+| `--cloud` | — | Cloud mode: read token from env and emit `COZE_LAB_RESULT=...` |
+| `--codex-home` | — | Override Codex config home for non-cloud/custom runs |
 | `--login` | — | Run the Device Code login flow only |
 | `--status` | — | Print local token status (valid / expiring / expired) |
 | `--refresh` | — | Force-refresh the access token via `refresh_token` |
@@ -50,6 +56,11 @@ npx coze_lab --logout     # Clear cached credentials
 | `codex` | `~/.codex/hooks/cozeloop_hook.py` | `~/.codex/hooks.json` | `~/.codex/hooks/cozeloop.env` |
 | `openclaw` | — (Node.js plugin) | `~/.openclaw/openclaw.json` | inline in config |
 
+For cloud Codex with `--cloud --agent-id=<agentId>`, Codex hooks are written to
+`~/.coze/agents/<agentId>/codex-home` by default. The directory is created if it
+does not already exist, so callers do not need to pass `--codex-home` for the
+standard coze-bridge layout.
+
 ## Token lifecycle
 
 OAuth tokens are stored in `~/.cozeloop/credentials.json` (mode 600).

package/index.js

@@ -4618,23 +4618,118 @@ function writeCodexHook(token, workspaceId, pythonCmd, codexHome, cloud) {
   return { hookScript, envFile, hooksJson };
 }
 
+function resolveCodexHome(args) {
+  if (args.cloud && args.agentId) {
+    return path.join(os.homedir(), '.coze', 'agents', args.agentId, 'codex-home');
+  }
+  return args['codex-home'] || process.env.CODEX_HOME || undefined;
+}
+
 // writeOpenClawHook 配置 OpenClaw 的 cozeloop-trace 插件（全局装在 ~/.openclaw）。
 // agentId 非空时并入 plugins.entries[...].config.traceAgentIds allowlist —— 插件运行时
 // 用 resolveAgentIdFromHookCtx 取当前 agentId，仅 allowlist 内的 agent 才上报 trace。
 // allowlist 为空（本地全局模式）= 全部放行，向后兼容。
+// resolveHomeDir 解析 home 目录。云端 SandboxShellExec 执行环境的 $HOME 可能缺失/不一致，
+// 导致 os.homedir() 解析不到真实 home（如 /root）。云端模式下做兜底探测。
+function resolveHomeDir(cloud) {
+  const h = os.homedir();
+  if (!cloud) return h;
+  // 优先 $HOME，其次 os.homedir()，再回退云端常见 root home。
+  const candidates = [process.env.HOME, h, '/root'].filter(Boolean);
+  for (const c of candidates) {
+    try {
+      if (fs.existsSync(path.join(c, '.coze')) || fs.existsSync(path.join(c, '.openclaw'))) {
+        return c;
+      }
+    } catch { /* ignore */ }
+  }
+  return h || '/root';
+}
+
+function normalizeTraceAgentIds(ids) {
+  return (Array.isArray(ids) ? ids : [])
+    .map((s) => String(s).trim().toLowerCase())
+    .filter(Boolean);
+}
+
+function getOpenClawEndpoint(cloud) {
+  return (cloud && process.env.COZELOOP_API_BASE_URL)
+    ? process.env.COZELOOP_API_BASE_URL.replace(/\/+$/, '') + '/v1/loop/opentelemetry'
+    : 'https://api.coze.cn/v1/loop/opentelemetry';
+}
+
+function applyOpenClawPluginConfig(existing, token, workspaceId, agentId, cloud) {
+  if (!existing.plugins)         existing.plugins = {};
+  if (!existing.plugins.allow)   existing.plugins.allow = [];
+  if (!existing.plugins.entries) existing.plugins.entries = {};
+
+  const PLUGIN = 'openclaw-cozeloop-trace';
+  if (!existing.plugins.allow.includes(PLUGIN)) {
+    existing.plugins.allow.push(PLUGIN);
+  }
+  // Preserve existing entry structure, only update config.
+  if (!existing.plugins.entries[PLUGIN]) {
+    existing.plugins.entries[PLUGIN] = { enabled: true };
+  }
+  existing.plugins.entries[PLUGIN].enabled = true;
+  // hooks.allowConversationAccess required for 2026.5+ to access session content.
+  existing.plugins.entries[PLUGIN].hooks = { allowConversationAccess: true };
+  if (!existing.plugins.entries[PLUGIN].config) existing.plugins.entries[PLUGIN].config = {};
+  const pcfg = existing.plugins.entries[PLUGIN].config;
+  pcfg.authorization = `Bearer ${token}`;
+  pcfg.endpoint      = getOpenClawEndpoint(cloud);
+  pcfg.workspaceId   = workspaceId;
+  pcfg.debug         = true;
+  // per-agent trace 放行：把当前 agentId 并入 traceAgentIds（去重、归一为小写，
+  // 与插件侧 resolveAgentIdFromHookCtx 的归一一致）。无 agentId（全局模式）则不动
+  // allowlist —— 空 allowlist 表示全部放行。
+  if (agentId) {
+    const norm = String(agentId).trim().toLowerCase();
+    const list = normalizeTraceAgentIds(pcfg.traceAgentIds);
+    if (norm && !list.includes(norm)) list.push(norm);
+    pcfg.traceAgentIds = list;
+  }
+  return existing;
+}
+
+function isOpenClawAlreadyInjected(configPath, pluginDir, token, workspaceId, agentId, cloud) {
+  if (!fs.existsSync(pluginDir)) return false;
+  let existing;
+  try {
+    existing = JSON.parse(fs.readFileSync(configPath, 'utf8'));
+  } catch {
+    return false;
+  }
+  const desired = applyOpenClawPluginConfig(
+    JSON.parse(JSON.stringify(existing)),
+    token,
+    workspaceId,
+    agentId,
+    cloud,
+  );
+  return JSON.stringify(existing) === JSON.stringify(desired);
+}
+
 function writeOpenClawHook(token, workspaceId, agentId, cloud) {
-  const configPath = path.join(os.homedir(), '.openclaw', 'openclaw.json');
-  const pluginDir  = path.join(os.homedir(), '.cozeloop', 'openclaw-plugin');
+  const home       = resolveHomeDir(cloud);
+  const configPath = path.join(home, '.openclaw', 'openclaw.json');
+  const pluginDir  = path.join(home, '.cozeloop', 'openclaw-plugin');
 
   if (!fs.existsSync(configPath)) {
     errorBox([
-      'ERROR: ~/.openclaw/openclaw.json not found',
+      `ERROR: openclaw.json not found at ${configPath}`,
       '',
       'Make sure OpenClaw is installed and has been run at least once.',
-      'Install: npm install -g openclaw',
+      `(home=${home}, $HOME=${process.env.HOME || 'unset'})`,
     ]);
   }
 
+  if (isOpenClawAlreadyInjected(configPath, pluginDir, token, workspaceId, agentId, cloud)) {
+    ok(`OpenClaw plugin already configured in ${configPath}`);
+    info('OpenClaw gateway restart skipped (configuration unchanged).');
+    return { configPath, pluginDir, unchanged: true };
+  }
+
   // 1. Write plugin files to ~/.cozeloop/openclaw-plugin/
   ensureDir(pluginDir);
   ensureDir(path.join(pluginDir, 'dist'));
@@ -4685,46 +4780,7 @@ function writeOpenClawHook(token, workspaceId, agentId, cloud) {
 
   // 4. Update openclaw.json with token and workspace
   const config = mergeJson(configPath, (existing) => {
-    if (!existing.plugins)         existing.plugins = {};
-    if (!existing.plugins.allow)   existing.plugins.allow = [];
-    if (!existing.plugins.entries) existing.plugins.entries = {};
-
-    const PLUGIN = 'openclaw-cozeloop-trace';
-    if (!existing.plugins.allow.includes(PLUGIN)) {
-      existing.plugins.allow.push(PLUGIN);
-    }
-    // Preserve existing entry structure, only update config
-    if (!existing.plugins.entries[PLUGIN]) {
-      existing.plugins.entries[PLUGIN] = { enabled: true };
-    }
-    existing.plugins.entries[PLUGIN].enabled = true;
-    // hooks.allowConversationAccess required for 2026.5+ to access session content
-    existing.plugins.entries[PLUGIN].hooks = { allowConversationAccess: true };
-    if (!existing.plugins.entries[PLUGIN].config) existing.plugins.entries[PLUGIN].config = {};
-    const pcfg = existing.plugins.entries[PLUGIN].config;
-    pcfg.authorization = `Bearer ${token}`;
-    // 云端：endpoint 走 sandbox 注入的 COZELOOP_API_BASE_URL 代理（token 经它鉴权）；
-    // 缺省回退 api.coze.cn 直连。插件 exporter 会在此基础上拼 /v1/traces。
-    const ocBase = (cloud && process.env.COZELOOP_API_BASE_URL)
-      ? process.env.COZELOOP_API_BASE_URL.replace(/\/+$/, '') + '/v1/loop/opentelemetry'
-      : 'https://api.coze.cn/v1/loop/opentelemetry';
-    pcfg.endpoint      = ocBase;
-    pcfg.workspaceId   = workspaceId;
-    pcfg.debug         = true;
-    // per-agent trace 放行：把当前 agentId 并入 traceAgentIds（去重、归一为小写，
-    // 与插件侧 resolveAgentIdFromHookCtx 的归一一致）。无 agentId（全局模式）则不动
-    // allowlist —— 空 allowlist 表示全部放行。
-    if (agentId) {
-      const norm = String(agentId).trim().toLowerCase();
-      // 读 existing 时一并归一（小写、去空），与插件侧 resolveAgentIdFromHookCtx 一致，
-      // 防手工编辑混入大写条目导致去重失效。
-      const list = (Array.isArray(pcfg.traceAgentIds) ? pcfg.traceAgentIds : [])
-        .map((s) => String(s).trim().toLowerCase())
-        .filter(Boolean);
-      if (norm && !list.includes(norm)) list.push(norm);
-      pcfg.traceAgentIds = list;
-    }
-    return existing;
+    return applyOpenClawPluginConfig(existing, token, workspaceId, agentId, cloud);
   });
 
   try {
@@ -5241,8 +5297,11 @@ async function main() {
     }
     written = writeClaudeCodeHook(token, WORKSPACE_ID, pythonCmd, args.agentId ? agentWorkspace : undefined, args.cloud);
   } else if (agent === 'codex') {
-    // CODEX_HOME 来源优先级：--codex-home= > 环境变量 CODEX_HOME > ~/.codex（缺省）。
-    const codexHome = args['codex-home'] || process.env.CODEX_HOME || undefined;
+    const codexHome = resolveCodexHome(args);
+    if (args.cloud && args.agentId && codexHome && !fs.existsSync(codexHome)) {
+      ensureDir(codexHome);
+      info(`已创建云端 Codex 配置目录: ${codexHome}`);
+    }
     if (codexHome) info(`Codex 配置目录: ${codexHome}`);
     written = writeCodexHook(token, WORKSPACE_ID, pythonCmd, codexHome, args.cloud);
   } else {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "coze_lab",
-  "version": "0.1.13",
+  "version": "0.1.15",
   "description": "Configure local AI agents (Claude Code, Codex, OpenClaw) to report traces to CozeLoop",
   "keywords": [
     "cozeloop",
@@ -16,7 +16,12 @@
   },
   "files": [
     "index.js",
-    "scripts/"
+    "scripts/claude-code/cozeloop_hook.py",
+    "scripts/codex/cozeloop_hook.py",
+    "scripts/shared/cozeloop_refresh.py",
+    "scripts/openclaw/dist/",
+    "scripts/openclaw/openclaw.plugin.json",
+    "scripts/openclaw/package.json"
   ],
   "engines": {
     "node": ">=18"

package/scripts/claude-code/cozeloop_hook.py

@@ -29,15 +29,53 @@ from pathlib import Path
 from typing import Optional, List, Dict, Any
 
 # --- SDK Import ---
-try:
+def _ensure_cozeloop_sdk():
+    try:
+        import cozeloop  # noqa: F401
+        return True
+    except ImportError:
+        pass
+    import subprocess
+    import importlib
+    import site
+    attempts = (
+        ["--quiet", "--disable-pip-version-check", "cozeloop"],
+        ["--quiet", "--disable-pip-version-check", "--break-system-packages", "cozeloop"],
+        ["--quiet", "--disable-pip-version-check", "--break-system-packages", "--user", "cozeloop"],
+    )
+    for extra in attempts:
+        try:
+            subprocess.run(
+                [sys.executable, "-m", "pip", "install", *extra],
+                timeout=180, check=True,
+                stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL,
+            )
+        except Exception:
+            continue
+        try:
+            importlib.reload(site)
+            user_site = site.getusersitepackages()
+            for p in ([user_site] if isinstance(user_site, str) else list(user_site)):
+                if p and p not in sys.path:
+                    sys.path.insert(0, p)
+            importlib.invalidate_caches()
+            import cozeloop  # noqa: F401
+            print("[CozeLoop] cozeloop SDK auto-installed at runtime.", file=sys.stderr)
+            return True
+        except ImportError:
+            continue
+    return False
+
+
+if _ensure_cozeloop_sdk():
     import cozeloop
     from cozeloop.spec.tracespec import (
         Runtime, ModelInput, ModelMessage, ModelToolChoice,
         ModelOutput, ModelChoice, ModelToolCall, ModelToolCallFunction,
         ModelMessagePart, ModelMessagePartType
     )
-except ImportError:
-    print("Error: cozeloop SDK not found. Please install it with: pip install cozeloop", file=sys.stderr)
+else:
+    print("Error: cozeloop SDK not found and auto-install failed. Try: pip install cozeloop", file=sys.stderr)
     sys.exit(1)
 
 # --- Configuration ---
@@ -1412,4 +1450,3 @@ if __name__ == "__main__":
     main()
 
 
-

package/scripts/codex/cozeloop_hook.py

@@ -202,15 +202,53 @@ def get_fresh_token():
 # -------------------------------------------------------------------------
 
 # --- SDK Import ---
-try:
+def _ensure_cozeloop_sdk():
+    try:
+        import cozeloop  # noqa: F401
+        return True
+    except ImportError:
+        pass
+    import subprocess
+    import importlib
+    import site
+    attempts = (
+        ["--quiet", "--disable-pip-version-check", "cozeloop"],
+        ["--quiet", "--disable-pip-version-check", "--break-system-packages", "cozeloop"],
+        ["--quiet", "--disable-pip-version-check", "--break-system-packages", "--user", "cozeloop"],
+    )
+    for extra in attempts:
+        try:
+            subprocess.run(
+                [sys.executable, "-m", "pip", "install", *extra],
+                timeout=180, check=True,
+                stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL,
+            )
+        except Exception:
+            continue
+        try:
+            importlib.reload(site)
+            user_site = site.getusersitepackages()
+            for p in ([user_site] if isinstance(user_site, str) else list(user_site)):
+                if p and p not in sys.path:
+                    sys.path.insert(0, p)
+            importlib.invalidate_caches()
+            import cozeloop  # noqa: F401
+            print("[CozeLoop] cozeloop SDK auto-installed at runtime.", file=sys.stderr)
+            return True
+        except ImportError:
+            continue
+    return False
+
+
+if _ensure_cozeloop_sdk():
     import cozeloop
     from cozeloop.spec.tracespec import (
         Runtime, ModelInput, ModelMessage, ModelToolChoice,
         ModelOutput, ModelChoice, ModelToolCall, ModelToolCallFunction,
         ModelMessagePart, ModelMessagePartType
     )
-except ImportError:
-    print("Error: cozeloop SDK not found. Please install it with: pip install cozeloop", file=sys.stderr)
+else:
+    print("Error: cozeloop SDK not found and auto-install failed. Try: pip install cozeloop", file=sys.stderr)
     sys.exit(1)
 
 # --- Configuration ---
@@ -1120,4 +1158,3 @@ def main():
 
 if __name__ == "__main__":
     main()
-