coze_lab 0.1.14 → 0.1.16

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,16 @@ 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.
+
+Codex hook diagnostics are appended to `hooks/cozeloop.log` under the same
+Codex home. For cloud Codex, check
+`~/.coze/agents/<agentId>/codex-home/hooks/cozeloop.log`. If that file is not
+created after a new Codex turn, Codex did not load or execute the hook.
+
 ## Token lifecycle
 
 OAuth tokens are stored in `~/.cozeloop/credentials.json` (mode 600).

package/index.js

@@ -4554,6 +4554,7 @@ function writeCodexHook(token, workspaceId, pythonCmd, codexHome, cloud) {
   const hooksDir   = path.join(home, 'hooks');
   const hookScript = path.join(hooksDir, 'cozeloop_hook.py');
   const envFile    = path.join(hooksDir, 'cozeloop.env');
+  const logFile    = path.join(hooksDir, 'cozeloop.log');
   const hooksJson  = path.join(home, 'hooks.json');
 
   // 1. Write Python hook scripts (trace + refresh)
@@ -4571,6 +4572,7 @@ function writeCodexHook(token, workspaceId, pythonCmd, codexHome, cloud) {
   if (!cloud) {
     envLines.push(`COZELOOP_API_TOKEN=${token}`);
   }
+  envLines.push(`COZELOOP_HOOK_LOG=${logFile}`);
   envLines.push('TRACE_TO_COZELOOP=true');
   // PPE 泳道：cozeloop SDK 读这两个环境变量，自动注入 x-tt-env / x-use-ppe header
   envLines.push(`x_tt_env=${PPE_TT_ENV}`);
@@ -4615,7 +4617,14 @@ function writeCodexHook(token, workspaceId, pythonCmd, codexHome, cloud) {
   ok(`Hook registered in ${hooksJson}`);
   warn('Codex hook trust: 首次启动 Codex 会提示 "Hooks need review"。在该提示按 t（Trust all and continue），或在会话内运行 /hooks 后按 t，即可一次性信任全部 hook 启用 trace 上报。');
 
-  return { hookScript, envFile, hooksJson };
+  return { hookScript, envFile, hooksJson, logFile };
+}
+
+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）。
@@ -5290,8 +5299,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 {
@@ -5315,6 +5327,7 @@ async function main() {
     summaryLines.push(`Hook script: ${written.hookScript || '~/.codex/hooks/cozeloop_hook.py'}`);
     summaryLines.push(`Config:      ${written.hooksJson || '~/.codex/hooks.json'}`);
     summaryLines.push(`Credentials: ${written.envFile || '~/.codex/hooks/cozeloop.env'}`);
+    if (written.logFile) summaryLines.push(`Hook log:    ${written.logFile}`);
   } else {
     summaryLines.push(`Config:      ~/.openclaw/openclaw.json`);
   }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "coze_lab",
-  "version": "0.1.14",
+  "version": "0.1.16",
   "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

@@ -130,12 +130,15 @@ def _make_finish_event_processor():
     def _processor(info):
         try:
             if not getattr(info, "is_event_fail", False):
+                hook_log("upload success")
                 return
             detail = getattr(info, "detail_msg", "") or ""
             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
@@ -202,23 +205,80 @@ 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 ---
 DEBUG = os.environ.get("CC_COZELOOP_DEBUG", "").lower() == "true"
 
 
+def _log_file_path() -> str:
+    return os.environ.get("COZELOOP_HOOK_LOG", "").strip()
+
+
+def hook_log(message: str):
+    """Append one diagnostic line to the hook log, if configured."""
+    log_path = _log_file_path()
+    if not log_path:
+        return
+    try:
+        p = Path(log_path).expanduser()
+        p.parent.mkdir(parents=True, exist_ok=True)
+        with p.open("a", encoding="utf-8") as f:
+            f.write(f"{datetime.now().isoformat()} {message}\n")
+    except Exception:
+        pass
+
+
 def debug_log(message: str):
     """Print debug message if debug mode is enabled."""
+    hook_log(f"DEBUG {message}")
     if DEBUG:
         print(f"[COZELOOP_HOOK_DEBUG] {datetime.now().isoformat()} - {message}", file=sys.stderr)
 
@@ -652,12 +712,15 @@ def send_turns_to_cozeloop(turns: List[Dict[str, Any]], session_id: str, model_n
     token = get_fresh_token()
     if token:
         os.environ["COZELOOP_API_TOKEN"] = token
+        hook_log(f"token resolved prefix={token[:12]}...")
         print(f"[CozeLoop] Token 获取成功 ({token[:12]}...)", file=sys.stderr)
     else:
+        hook_log("token missing")
         print("[CozeLoop] 警告: 未找到有效 Token，上报可能失败", file=sys.stderr)
     creds = _load_credentials()
     workspace_id = (creds or {}).get("workspace_id") or os.environ.get("COZELOOP_WORKSPACE_ID", "") or _DEFAULT_WORKSPACE_ID
     os.environ["COZELOOP_WORKSPACE_ID"] = workspace_id
+    hook_log(f"workspace_id={workspace_id}")
     client_kwargs = {
         "ultra_large_report": True,
         "upload_timeout": 120,
@@ -980,13 +1043,16 @@ def send_turns_to_cozeloop(turns: List[Dict[str, Any]], session_id: str, model_n
                     debug_log(f"Error processing turn {i}: {e}")
                     continue
 
+        hook_log(f"processed turns={len(turns)} session_id={session_id}")
         debug_log(f"Successfully processed {len(turns)} turn(s) for session {session_id}")
 
     except Exception as e:
+        hook_log(f"send exception={repr(e)}")
         debug_log(f"An error occurred while sending traces to CozeLoop: {e}")
         return None
     finally:
         client.close()
+        hook_log("client closed")
         debug_log("CozeLoop client closed.")
 
     return ctx
@@ -997,10 +1063,12 @@ def send_turns_to_cozeloop(turns: List[Dict[str, Any]], session_id: str, model_n
 def main():
     """Main entry point for the Codex CozeLoop hook."""
     print("[CozeLoop] Hook triggered (Codex).", file=sys.stderr)
+    hook_log("hook triggered")
     debug_log("Codex CozeLoop hook started.")
 
     # Check if tracing is enabled
     if os.environ.get("TRACE_TO_COZELOOP", "").lower() == "false":
+        hook_log("skip trace disabled")
         debug_log("TRACE_TO_COZELOOP is set to 'false', skipping")
         return
 
@@ -1008,10 +1076,12 @@ def main():
     try:
         raw_input = sys.stdin.read().strip()
         if not raw_input:
+            hook_log("skip no stdin")
             debug_log("No input received from stdin")
             return
         hook_input = json.loads(raw_input)
     except Exception as e:
+        hook_log(f"skip stdin parse error={repr(e)}")
         debug_log(f"Error reading hook input from stdin: {e}")
         return
 
@@ -1020,10 +1090,12 @@ def main():
     # Get transcript path
     transcript_path = hook_input.get("transcript_path")
     if not transcript_path:
+        hook_log("skip missing transcript_path")
         debug_log("No transcript_path in hook input")
         return
 
     if not os.path.exists(transcript_path):
+        hook_log(f"skip transcript not found path={transcript_path}")
         debug_log(f"Transcript file not found: {transcript_path}")
         return
 
@@ -1035,9 +1107,11 @@ def main():
     entries = read_rollout_messages(transcript_path, state["last_processed_line"])
 
     if not entries:
+        hook_log(f"skip no new entries transcript={transcript_path}")
         debug_log("No new entries to process")
         return
 
+    hook_log(f"read entries={len(entries)} from_line={state['last_processed_line']} transcript={transcript_path}")
     debug_log(f"Read {len(entries)} new entries from line {state['last_processed_line']}")
 
     # Parse session identity
@@ -1087,6 +1161,7 @@ def main():
         last_line = max(e.get("_line_number", 0) for e in entries) + 1
         state["last_processed_line"] = last_line
         save_state(state_file, state)
+        hook_log(f"subagent saved session_id={session_id} turns={len(turns[-1:])} last_line={last_line}")
         debug_log("Subagent data saved, hook completed")
         return
 
@@ -1097,6 +1172,7 @@ def main():
             for t in turns
         )
         if not has_coze_ctx:
+            hook_log(f"skip no coze-context turns={len(turns)} session_id={session_id}")
             debug_log("No coze-context found in any turn, skipping upload.")
             return
         history_context = state.get("conversation_history", [])
@@ -1109,15 +1185,18 @@ def main():
             state["last_processed_line"] = last_line
             state["conversation_history"] = updated_history
             save_state(state_file, state)
+            hook_log(f"state advanced last_line={last_line} session_id={session_id}")
             debug_log(f"State updated, last processed line: {last_line}")
         else:
+            hook_log(f"send failed state not advanced session_id={session_id}")
             debug_log("Send failed, state not advanced")
     else:
+        hook_log(f"skip no turns session_id={session_id}")
         debug_log("No turns to send")
 
+    hook_log("hook completed")
     debug_log("Codex CozeLoop hook completed.")
 
 
 if __name__ == "__main__":
     main()
-