coze_lab 0.1.14 → 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,6 +4618,13 @@ 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。
@@ -5290,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.14",
+  "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()
-