coze_lab 0.1.24 → 0.1.25

package/README.md

@@ -74,6 +74,8 @@ In cloud mode, trace verification and hook uploads prefer
 `COZELOOP_API_TOKEN` and fall back to `COZE_API_TOKEN`. The selfcheck result is
 authoritative: if the token does not have trace ingest permission, onboard still
 writes the hook configuration but reports `verify=fail` with `token_source`.
+For Python SDK uploads, `OTEL_ENDPOINT` is not used as the SDK base URL; set
+`COZELOOP_API_BASE_URL` only when the SDK ingest endpoint should be overridden.
 
 **At hook execution time** (Claude Code / Codex), the Python hook script automatically:
 1. Reads `~/.cozeloop/credentials.json`

package/index.js

@@ -4706,6 +4706,10 @@ function normalizeTraceAgentIds(ids) {
 
 function getCloudCozeloopApiBaseUrl() {
   const raw = process.env.COZELOOP_API_BASE_URL || process.env.OTEL_ENDPOINT || '';
+  return normalizeCozeloopApiBaseUrl(raw);
+}
+
+function normalizeCozeloopApiBaseUrl(raw) {
   const base = raw.trim().replace(/\/+$/, '');
   if (!base) return '';
   if (base.endsWith('/v1/loop/opentelemetry/v1/traces')) {
@@ -4730,6 +4734,14 @@ function getCozeloopApiBaseUrl(cloud) {
   return cloud ? (getCloudCozeloopApiBaseUrl() || COZE_API) : COZE_API;
 }
 
+function getCloudCozeloopSdkApiBaseUrl() {
+  return normalizeCozeloopApiBaseUrl(process.env.COZELOOP_API_BASE_URL || '');
+}
+
+function getCozeloopSdkApiBaseUrl(cloud) {
+  return cloud ? (getCloudCozeloopSdkApiBaseUrl() || COZE_API) : COZE_API;
+}
+
 function getOtelEndpointBase(cloud) {
   return `${getCozeloopApiBaseUrl(cloud).replace(/\/+$/, '')}/v1/loop/opentelemetry`;
 }
@@ -4935,11 +4947,13 @@ function runCommand(input) {
 
 async function verifyTraceReportViaSdk(token, workspaceId, pairCode, pythonCmd, tokenSource) {
   const pair = pairCode || crypto.randomBytes(6).toString('hex');
-  const apiBase = getCozeloopApiBaseUrl(true);
+  const apiBase = getCozeloopSdkApiBaseUrl(true);
   const script = `
 import json
 import os
 import sys
+import urllib.error
+import urllib.request
 
 events = []
 
@@ -4947,6 +4961,51 @@ def finish_event(info):
     if getattr(info, "is_event_fail", False):
         events.append(getattr(info, "detail_msg", "") or "trace export failed")
 
+def extract_logid(text):
+    if not text:
+        return ""
+    marker = "logid="
+    idx = text.find(marker)
+    if idx >= 0:
+        out = []
+        for ch in text[idx + len(marker):]:
+            if ch.isalnum():
+                out.append(ch)
+            else:
+                break
+        return "".join(out)
+    for marker in ('"logid":"', '"log_id":"', '"Logid":"'):
+        idx = text.find(marker)
+        if idx >= 0:
+            rest = text[idx + len(marker):]
+            return rest.split('"', 1)[0]
+    return ""
+
+def http_diag():
+    base = os.environ.get("COZELOOP_API_BASE_URL", "").strip().rstrip("/") or "https://api.coze.cn"
+    url = base + "/v1/loop/traces/ingest"
+    body = json.dumps({"spans": []}).encode()
+    req = urllib.request.Request(
+        url,
+        data=body,
+        headers={
+            "Content-Type": "application/json",
+            "Authorization": "Bearer " + os.environ.get("COZELOOP_API_TOKEN", ""),
+            "x-tt-env": os.environ.get("x_tt_env", ""),
+            "x-use-ppe": os.environ.get("x_use_ppe", ""),
+        },
+        method="POST",
+    )
+    try:
+        with urllib.request.urlopen(req, timeout=10) as resp:
+            text = resp.read().decode("utf-8", "replace")
+            return {"url": url, "status": resp.status, "logid": resp.headers.get("x-tt-logid", "") or extract_logid(text), "body": text[:300]}
+    except urllib.error.HTTPError as e:
+        text = e.read().decode("utf-8", "replace")
+        return {"url": url, "status": e.code, "logid": e.headers.get("x-tt-logid", "") or extract_logid(text), "body": text[:300]}
+    except Exception as e:
+        return {"url": url, "status": 0, "logid": "", "body": type(e).__name__ + ": " + str(e)}
+
 try:
     import cozeloop
     kwargs = {
@@ -4966,11 +5025,15 @@ try:
     client.flush()
     client.close()
     if events:
-        print(json.dumps({"success": False, "body": "\\n".join(events), "api_base_url": os.environ.get("COZELOOP_API_BASE_URL", ""), "token_source": os.environ.get("COZELOOP_TOKEN_SOURCE", "")}, ensure_ascii=False))
+        diag = http_diag()
+        body = "\\n".join(events) + "\\nhttp_diag=" + json.dumps(diag, ensure_ascii=False)
+        print(json.dumps({"success": False, "body": body, "logid": diag.get("logid", ""), "api_base_url": os.environ.get("COZELOOP_API_BASE_URL", ""), "token_source": os.environ.get("COZELOOP_TOKEN_SOURCE", "")}, ensure_ascii=False))
         sys.exit(1)
-    print(json.dumps({"success": True, "body": "", "api_base_url": os.environ.get("COZELOOP_API_BASE_URL", ""), "token_source": os.environ.get("COZELOOP_TOKEN_SOURCE", "")}, ensure_ascii=False))
+    print(json.dumps({"success": True, "body": "", "logid": "", "api_base_url": os.environ.get("COZELOOP_API_BASE_URL", ""), "token_source": os.environ.get("COZELOOP_TOKEN_SOURCE", "")}, ensure_ascii=False))
 except Exception as e:
-    print(json.dumps({"success": False, "body": str(e), "api_base_url": os.environ.get("COZELOOP_API_BASE_URL", ""), "token_source": os.environ.get("COZELOOP_TOKEN_SOURCE", "")}, ensure_ascii=False))
+    diag = http_diag()
+    body = str(e) + "\\nhttp_diag=" + json.dumps(diag, ensure_ascii=False)
+    print(json.dumps({"success": False, "body": body, "logid": diag.get("logid", ""), "api_base_url": os.environ.get("COZELOOP_API_BASE_URL", ""), "token_source": os.environ.get("COZELOOP_TOKEN_SOURCE", "")}, ensure_ascii=False))
     sys.exit(1)
 `;
   const env = {
@@ -5005,7 +5068,7 @@ except Exception as e:
     const snippet = String(body || '').slice(0, 300);
     if (snippet) console.log(snippet);
   }
-  return { success, status: result.code || 0, body, traceId: '', pairCode: pair, apiBaseUrl: apiBase, tokenSource };
+  return { success, status: result.code || 0, body, traceId: '', pairCode: pair, apiBaseUrl: apiBase, tokenSource, logid: parsed?.logid || '' };
 }
 
 // 真实发一条最小 OTLP trace 到 CozeLoop，验证上报链路是否打通。
@@ -5538,7 +5601,7 @@ async function main() {
   } else if (CLOUD_MODE) {
     // 云端：注入已成功，验证失败不阻断（放行），记录结果供后台弹 warning。
     cloudResult.verify = 'fail';
-    cloudResult.logid = extractLogid(verifyResult.body) || cloudResult.logid;
+    cloudResult.logid = verifyResult.logid || extractLogid(verifyResult.body) || cloudResult.logid;
     cloudResult.message = `trace 上报自检失败 HTTP ${verifyResult.status}: ${(verifyResult.body || '').slice(0, 200)}`;
     warn('trace 上报自检失败，但 hook 配置已写入（云端放行）。');
     console.log('');

package/package.json

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

package/scripts/claude-code/cozeloop_hook.py

@@ -264,7 +264,7 @@ def _normalize_api_base_url(url: str) -> str:
 
 def get_api_base_url() -> str:
     return _normalize_api_base_url(
-        os.environ.get("COZELOOP_API_BASE_URL", "") or os.environ.get("OTEL_ENDPOINT", "")
+        os.environ.get("COZELOOP_API_BASE_URL", "")
     )
 
 def get_fresh_token() -> Optional[str]:

package/scripts/codex/cozeloop_hook.py

@@ -225,7 +225,7 @@ def _normalize_api_base_url(url: str) -> str:
 
 def get_api_base_url() -> str:
     return _normalize_api_base_url(
-        os.environ.get("COZELOOP_API_BASE_URL", "") or os.environ.get("OTEL_ENDPOINT", "")
+        os.environ.get("COZELOOP_API_BASE_URL", "")
     )