coze_lab 0.1.18 → 0.1.19

package/index.js

@@ -4464,6 +4464,10 @@ function mergeJson(filepath, mergeFn) {
   return mergeFn(existing);
 }
 
+function shellEnvLine(key, value) {
+  return `${key}='${String(value).replace(/'/g, `'\\''`)}'`;
+}
+
 // writeClaudeCodeHook 配置 Claude Code 的 hook。
 // configBaseDir 缺省 process.cwd()（全局/项目级）；传入 agent 的 workspace 则 per-agent：
 // settings.json + 凭证写进 <configBaseDir>/.claude，仅该 agent（以此为 cwd 启动）生效。
@@ -4565,24 +4569,29 @@ function writeCodexHook(token, workspaceId, pythonCmd, codexHome, cloud) {
   writeHookScript(refreshScript, readScript('shared/cozeloop_refresh.py'));
 
   // 2. Write env file with chmod 600
-  //    云端（cloud）：不落明文 token，hook 运行时从环境变量 COZE_API_TOKEN 读取。
   const envLines = [
-    `COZELOOP_WORKSPACE_ID=${workspaceId}`,
+    shellEnvLine('COZELOOP_WORKSPACE_ID', workspaceId),
   ];
-  if (!cloud) {
-    envLines.push(`COZELOOP_API_TOKEN=${token}`);
+  if (cloud) {
+    if (process.env.COZELOOP_API_TOKEN) {
+      envLines.push(shellEnvLine('COZELOOP_API_TOKEN', process.env.COZELOOP_API_TOKEN));
+    } else if (process.env.COZE_API_TOKEN) {
+      envLines.push(shellEnvLine('COZE_API_TOKEN', process.env.COZE_API_TOKEN));
+    }
+  } else {
+    envLines.push(shellEnvLine('COZELOOP_API_TOKEN', token));
   }
-  envLines.push(`CODEX_HOME=${home}`);
-  envLines.push(`COZELOOP_HOOK_LOG=${logFile}`);
+  envLines.push(shellEnvLine('CODEX_HOME', home));
+  envLines.push(shellEnvLine('COZELOOP_HOOK_LOG', logFile));
   envLines.push('TRACE_TO_COZELOOP=true');
   if (process.env.COZELOOP_API_BASE_URL) {
-    envLines.push(`COZELOOP_API_BASE_URL=${process.env.COZELOOP_API_BASE_URL}`);
+    envLines.push(shellEnvLine('COZELOOP_API_BASE_URL', process.env.COZELOOP_API_BASE_URL));
   } else if (process.env.OTEL_ENDPOINT) {
-    envLines.push(`OTEL_ENDPOINT=${process.env.OTEL_ENDPOINT}`);
+    envLines.push(shellEnvLine('OTEL_ENDPOINT', process.env.OTEL_ENDPOINT));
   }
   // PPE 泳道：cozeloop SDK 读这两个环境变量，自动注入 x-tt-env / x-use-ppe header
-  envLines.push(`x_tt_env=${PPE_TT_ENV}`);
-  envLines.push(`x_use_ppe=${PPE_USE_PPE}`);
+  envLines.push(shellEnvLine('x_tt_env', PPE_TT_ENV));
+  envLines.push(shellEnvLine('x_use_ppe', PPE_USE_PPE));
   const envContent = envLines.join('\n') + '\n';
   try {
     fs.writeFileSync(envFile, envContent, { mode: 0o600 });
@@ -4666,6 +4675,11 @@ function getOpenClawEndpoint(cloud) {
     : 'https://api.coze.cn/v1/loop/opentelemetry';
 }
 
+function getCloudCozeloopBaseUrl() {
+  const endpoint = process.env.COZELOOP_API_BASE_URL || process.env.OTEL_ENDPOINT || '';
+  return endpoint.replace(/\/v1\/loop\/opentelemetry\/?$/, '').replace(/\/+$/, '') || undefined;
+}
+
 function applyOpenClawPluginConfig(existing, token, workspaceId, agentId, cloud) {
   if (!existing.plugins)         existing.plugins = {};
   if (!existing.plugins.allow)   existing.plugins.allow = [];
@@ -4820,7 +4834,7 @@ function httpsPost(url, body, extraHeaders) {
     const data = JSON.stringify(body);
     const u = new URL(url);
     const req = https.request(
-      { hostname: u.hostname, path: u.pathname + u.search, method: 'POST',
+      { hostname: u.hostname, port: u.port || undefined, path: u.pathname + u.search, method: 'POST',
         headers: { 'Content-Type': 'application/json', 'Content-Length': Buffer.byteLength(data),
                    'x-tt-env': PPE_TT_ENV, 'x-use-ppe': PPE_USE_PPE,
                    ...(extraHeaders || {}) } },
@@ -5230,7 +5244,7 @@ async function main() {
     info('验证 trace 上报链路...');
     const token = await getValidToken();   // 无凭证会自动走登录/刷新
     console.log('');
-    const result = await verifyTraceReport(token, WORKSPACE_ID, args.pairCode, process.env.COZELOOP_API_BASE_URL || undefined);
+    const result = await verifyTraceReport(token, WORKSPACE_ID, args.pairCode, getCloudCozeloopBaseUrl());
     process.exit(result.success ? 0 : 1);
   }
 
@@ -5352,7 +5366,7 @@ async function main() {
 
   // Step 5: Verify trace reporting end-to-end
   info('Step 5/5: 验证 trace 上报链路...');
-  const verifyResult = await verifyTraceReport(token, WORKSPACE_ID, args.pairCode, args.cloud ? process.env.COZELOOP_API_BASE_URL : undefined);
+  const verifyResult = await verifyTraceReport(token, WORKSPACE_ID, args.pairCode, args.cloud ? getCloudCozeloopBaseUrl() : undefined);
   if (verifyResult.success) {
     cloudResult.verify = 'ok';
   } else if (CLOUD_MODE) {

package/package.json

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

package/scripts/codex/cozeloop_hook.py

@@ -850,7 +850,12 @@ def send_turns_to_cozeloop(turns: List[Dict[str, Any]], session_id: str, model_n
         hook_log(f"token resolved prefix={token[:12]}...")
         print(f"[CozeLoop] Token 获取成功 ({token[:12]}...)", file=sys.stderr)
     else:
-        hook_log("token missing")
+        hook_log(
+            "token missing "
+            f"has_cozeloop_token={bool(os.environ.get('COZELOOP_API_TOKEN'))} "
+            f"has_coze_token={bool(os.environ.get('COZE_API_TOKEN'))} "
+            f"api_base_url={bool(get_api_base_url())}"
+        )
         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