cartethyia 2.5.4 → 2.5.6

package/src/providers/integrations/alibabacp/alibabacp.ts

@@ -6,7 +6,7 @@
  * plans are served by different hosts and their keys are not interchangeable,
  * and which plan a key belongs to is per-account state stored with it. A spec
  * declares one `base_url` at construction, so the host is resolved from the
- * credential at dispatch time and a stored region that no longer resolves is a
+ * credential at dispatch time and a stored region that does not resolve is a
  * dispatch-time error rather than a silent fallback to the default host.
  */
 import { createApiKeyAdapter } from "../../configured-provider";

package/src/providers/integrations/antigravity/antigravity-quota.ts

@@ -33,10 +33,9 @@ interface AntigravityQuotaFamily {
  * resulting id, so a quota row exists for exactly the models the operator can
  * select.
  *
- * This replaced a hand-kept list plus a permissive id pattern, which let a
- * deployment the catalog no longer serves (`gemini-2.5-pro`) report a quota row
- * for a model that cannot be chosen. Deriving the set means removing a catalog
- * model removes its quota row in the same change.
+ * Deriving the set means removing a catalog model removes its quota row in the
+ * same change — a hand-kept list would let a deployment the catalog does not
+ * serve report a quota row for a model that cannot be chosen.
  */
 const CATALOG_MODEL_IDS: ReadonlySet<string> = new Set(
   ANTIGRAVITY_MODELS.filter((model) => model.serviceKind !== "websearch").map((model) => model.modelId),
@@ -83,9 +82,8 @@ const SUMMARY_WINDOW_LABELS: Readonly<Record<SummaryWindowKind, string>> = Objec
  *
  * The upstream names the window in `window` (`5h`, `daily`, `weekly`) and
  * repeats it in the display name. Daily must be tested before the 5-hour
- * session: an older spelling put `daily` on the session bucket, but a bucket
- * whose own text says "daily" is the daily allowance, and treating it as the
- * session is what dropped the daily row.
+ * session: a bucket whose own text says "daily" is the daily allowance, and
+ * treating it as the session is what drops the daily row.
  */
 function summaryWindowKind(bucket: Record<string, unknown>): SummaryWindowKind | undefined {
   const source = `${text(bucket.window) ?? ""} ${text(bucket.bucketId) ?? ""} ${text(bucket.displayName) ?? ""}`.toLowerCase();

package/src/providers/integrations/buddy/buddy-chat-shared.ts

@@ -822,9 +822,30 @@ export function dropEmptyBuddyMessages(messages: Array<Record<string, unknown>>)
     if (Array.isArray(message["tool_calls"]) && message["tool_calls"].length > 0) return true;
     if (typeof message["tool_call_id"] === "string" && message["tool_call_id"].trim().length > 0)
       return true;
-    if (typeof message["reasoning_content"] === "string") return true;
     const content = message["content"];
-    if (typeof content === "string") return content.trim().length > 0;
+    if (typeof content === "string") {
+      if (content.trim().length > 0) return true;
+      // `content === ""` on a bare assistant turn (no tool_calls, no
+      // tool_call_id) is unverifiable by the receiving model regardless of
+      // what `reasoning_content` carries. The upstream rejects it with
+      // code `11151 "history not found"` for both shapes observed in the
+      // wild:
+      //   (a) `reasoning_content: ""` — the Chat encoder's output for a
+      //       canonical `reasoning` part with no `.summary` and a non-string
+      //       `.payload` (signature-less thinking block already stripped on
+      //       an earlier hop), and
+      //   (b) `reasoning_content: "reasoning unavailable"` — the OMP client's
+      //       sentinel in a Responses `reasoning` input item of shape
+      //       `{summary: [], content: [{type: "reasoning_text",
+      //       text: "reasoning unavailable"}]}`, emitted when the client
+      //       strips the previous turn's trace on a model switch.
+      // Both are unverifiable: the receiving model never issued that
+      // signature, so nothing is round-trippable. Verifiable signatures
+      // round-trip on `tool_calls` turns (kept by the guard above) via the
+      // `reasoning_content` side channel; a signature on a bare turn with
+      // no tool_calls is not a shape any upstream in this family accepts.
+      return false;
+    }
     if (Array.isArray(content)) return content.length > 0;
     // Null content with no tool/reasoning load is a bare turn the upstream
     // rejects — drop it rather than dispatch a known-invalid payload.

package/src/providers/integrations/buddy/buddy-checkin.ts

@@ -19,6 +19,7 @@
 // claim short-circuits the day, so a restart cannot double-post.
 
 import { providerBaseUrl } from "../../provider-metadata";
+import { sleep } from "../../../shared/sleep";
 import { WORKBUDDY_DOMAIN } from "./workbuddy/shared";
 import { BUDDY_PROVIDER_IDS } from "../../provider-metadata";
 import {
@@ -368,10 +369,6 @@ function isRetryableClaim(error: unknown, status: number | null): boolean {
   return !(error instanceof DOMException && error.name === "AbortError");
 }
 
-function sleep(ms: number): Promise<void> {
-  return new Promise((resolve) => setTimeout(resolve, ms));
-}
-
 function asRecord(value: unknown): Record<string, unknown> | null {
   return value !== null && typeof value === "object" && !Array.isArray(value)
     ? (value as Record<string, unknown>)

package/src/providers/integrations/buddy/buddy-cli-headers.ts

@@ -69,7 +69,14 @@ export function buddyCliTraceHeaders(
  * conversation family, because those vary by provider and realm: only the
  * headers that are the same for every CLI dispatch are built here.
  */
-export function buddyCliHeaders(args: { readonly userId: string }): Record<string, string> {
+export function buddyCliHeaders(args: {
+  readonly userId: string;
+  readonly enterpriseId?: string | undefined;
+}): Record<string, string> {
+  const enterpriseId =
+    args.enterpriseId !== undefined && args.enterpriseId.trim().length > 0
+      ? args.enterpriseId.trim()
+      : undefined;
   return {
     "X-Agent-Intent": "craft",
     "X-Agent-Type": "main",
@@ -78,11 +85,21 @@ export function buddyCliHeaders(args: { readonly userId: string }): Record<strin
     // of the three, so "other" is the honest value rather than a guess.
     "X-Agent-Runtime": "other",
     "X-User-Id": args.userId,
-    // Personal accounts are not an enterprise and the reference sends the flag
-    // either way (`true` only for an enterprise id), so the non-enterprise
-    // default goes out here. Step 3 replaces this with the account's real
-    // enterprise id once discovery carries one.
-    "X-Private-Data": "false",
+    // The reference sends the flag either way, `true` only for an enterprise
+    // id. A personal account stays `false`; an enterprise account flips to
+    // `true` and names the enterprise, because the two travel together — a
+    // `true` with no id, or an id under `false`, is a combination no real
+    // client produces.
+    "X-Private-Data": enterpriseId === undefined ? "false" : "true",
+    ...(enterpriseId === undefined
+      ? {}
+      : {
+          "X-Enterprise-Id": enterpriseId,
+          // The tenant falls back to the enterprise: the two are the same value
+          // for an account that belongs to one enterprise, and the upstream
+          // reads both.
+          "X-Tenant-Id": enterpriseId,
+        }),
     ...buddyStainlessHeaders(),
   };
 }

package/src/providers/integrations/buddy/buddy-conversation-headers.ts

@@ -91,7 +91,6 @@ export function buddyConversationHeaders(
   };
   // Always sent, invented when the caller gave none.
   //
-  // This used to be omitted unless the client supplied one. That was wrong:
   // `x-conversation-id` is mandatory, and a chat request without it is refused
   // with `400 · 11-128 "request illegal"` — an error that names neither the
   // header nor the channel, and which reads like a block rather than a missing

package/src/providers/integrations/buddy/buddy-identity.ts

@@ -1,4 +1,5 @@
 import type { ProviderDispatchContext } from "../../provider-contracts";
+import { decodeJwtPayload } from "../../authentication/oauth-flow-store";
 
 /**
  * Which upstream client a buddy provider presents itself as.
@@ -44,3 +45,54 @@ export function buddyUserId(context?: ProviderDispatchContext): string {
   const accountId = context?.credential.account_id;
   return accountId !== undefined && accountId.length > 0 ? accountId : "anonymous";
 }
+
+/**
+ * Claim names a buddy access token may carry the enterprise id under.
+ *
+ * A list of candidates rather than one name because the upstream's token shape
+ * is not pinned anywhere this codebase can verify, and guessing one name would
+ * silently report "personal" for every account whose token spells it another
+ * way. Every candidate is a plausible spelling of the same fact; the first that
+ * holds a non-empty string wins, and a token carrying none yields no enterprise.
+ */
+const ENTERPRISE_CLAIMS: readonly string[] = [
+  "enterprise_id",
+  "enterpriseId",
+  "enterprise",
+  "tenant_id",
+  "tenantId",
+  "org_id",
+  "organization_id",
+];
+
+/**
+ * The enterprise this buddy account belongs to, or `undefined` for a personal
+ * one.
+ *
+ * Two sources, in this order:
+ *
+ * 1. **`auth_state.enterpriseId`** — written when the account was authorized, or
+ *    set by an operator for an account whose token does not carry the fact.
+ *    Wins because it is the one a human can correct.
+ * 2. **A claim on the access token** — for accounts authorized before this
+ *    existed, whose `auth_state` was never populated.
+ *
+ * Deliberately conservative: no enterprise id means *personal*, and personal is
+ * what the gateway already sends. An unknown account is therefore unchanged
+ * rather than being given an enterprise it may not have — claiming one the
+ * upstream does not recognise is the kind of mismatch that draws a channel
+ * check.
+ */
+export function buddyEnterprise(context?: ProviderDispatchContext): string | undefined {
+  const stored = context?.credential.auth_state?.["enterpriseId"];
+  if (typeof stored === "string" && stored.trim().length > 0) return stored.trim();
+  const token = context?.credential.secret;
+  if (token === undefined || token.length === 0) return undefined;
+  const claims = decodeJwtPayload(new TextDecoder().decode(token).trim());
+  if (claims === undefined) return undefined;
+  for (const claim of ENTERPRISE_CLAIMS) {
+    const value = claims[claim];
+    if (typeof value === "string" && value.trim().length > 0) return value.trim();
+  }
+  return undefined;
+}

package/src/providers/integrations/buddy/buddy-media.ts

@@ -15,7 +15,7 @@
  *
  * ⚠️ The provider's actual polling route has not been verified. The sibling
  *   URL is an implementation inference, not an observed contract. See
- *   `documentation/features/buddy-media-generation.md` before changing it.
+ *   `documentation/features/07-buddy-media-generation.md` before changing it.
  */
 import { OpenAICompatibleAdapter, authHeaders, type OpenAICompatibleAdapterConfig } from "../../adapters/compatible-adapter";
 import { GatewayError } from "../../../protocol/gateway-error";
@@ -305,7 +305,7 @@ export function buddyErrorOf(payload: unknown): string | undefined {
  * inferred sibling task URL until it reports a finished clip.
  *
  * ⚠️ The status path is not provider-verified; see the module header and
- * `documentation/features/buddy-media-generation.md`.
+ * `documentation/features/07-buddy-media-generation.md`.
  */
 export async function buddyGenerateVideo(
   config: BuddyMediaConfig,

package/src/providers/integrations/buddy/buddy-usage-store.ts

@@ -92,10 +92,9 @@ const DEFAULT_LIMIT_PER_ACCOUNT = 30;
 /**
  * Reads each account's stored usage, newest day first.
  *
- * Bounded on both axes, because it used to be bounded on neither: with no day
- * filter and no `LIMIT`, an account fetched every bucket it had ever stored
- * (~1500 rows for a 30-day window) to render 30 of them, on the route the
- * dashboard polls.
+ * Bounded on both axes: with no day filter and no `LIMIT`, an account would
+ * fetch every bucket it had ever stored (~1500 rows for a 30-day window) to
+ * render 30 of them, on the route the dashboard polls.
  *
  * - The day window is semantic, not a truncation: {@link pruneBuddyUsageSnapshots}
  *   already drops everything older, so no row it excludes could have been

package/src/providers/integrations/buddy/codebuddy/quota.ts

@@ -24,9 +24,9 @@ export const CODEBUDDY_CN_USAGE_URL =
   `${providerBaseUrl("cbcn")}/billing/meter/get-user-resource`;
 export const CODEBUDDY_INTL_USAGE_URL =
   `${providerBaseUrl("cb")}/billing/meter/get-user-resource`;
-export const CODEBUDDY_CN_REQUEST_USAGE_URL =
+const CODEBUDDY_CN_REQUEST_USAGE_URL =
   `${providerBaseUrl("cbcn")}/billing/meter/get-user-request-usage`;
-export const CODEBUDDY_INTL_REQUEST_USAGE_URL =
+const CODEBUDDY_INTL_REQUEST_USAGE_URL =
   `${providerBaseUrl("cb")}/billing/meter/get-user-request-usage`;
 
 type CodeBuddyProviderId = "cb" | "cbcn";
@@ -152,14 +152,3 @@ export async function fetchCodeBuddyRequestUsage(
   });
 }
 
-export const fetchCodeBuddyCnRequestUsage = (
-  credential: string,
-  fetcher: FetchLike,
-  days: number = 30,
-): Promise<BuddyUsageResult> => fetchCodeBuddyRequestUsage(credential, fetcher, "cbcn", days);
-
-export const fetchCodeBuddyIntlRequestUsage = (
-  credential: string,
-  fetcher: FetchLike,
-  days: number = 30,
-): Promise<BuddyUsageResult> => fetchCodeBuddyRequestUsage(credential, fetcher, "cb", days);

package/src/providers/integrations/buddy/codebuddy/shared.ts

@@ -1,115 +1,118 @@
-import { withBearerAuthentication, type OpenAICompatibleAdapterConfig } from "../../../adapters/compatible-adapter";
-import type { CanonicalRequest } from "../../../../protocol/canonical-model";
-import type { ProviderDispatchContext, ProviderDispatchTarget, ProviderId } from "../../../provider-contracts";
-import { resolveInboundSessionId } from "../../../operations/session-resolution";
-import { buddyConversationHeaders } from "../buddy-conversation-headers";
-import { buddyUserId, resolveBuddyIdentity } from "../buddy-identity";
-import { buildCodeBuddyUserAgent, getCodeBuddyCnVersion, getCodeBuddyVersion } from "../../../operations/client-versions";
-import { readBuddyDeviceToken } from "../workbuddy/shared";
-import { buddyCliHeaders, buddyCliTraceHeaders } from "../buddy-cli-headers";
-
-/** Client identity variant behind the per-request CodeBuddy headers. */
-export type CodeBuddyVariant = "IDE" | "CLI";
-
-export function codebuddyDomain(variant: CodeBuddyVariant): string {
-  return variant === "IDE" ? "www.codebuddy.ai" : "copilot.tencent.com";
-}
-
-
-/**
- * Per-request CodeBuddy headers — exact provider contract shared by the
- * international (IDE identity) and CN (CLI identity). The stable conversation
- * header is preserved from the client when available; request IDs rotate.
- *
- * `cliIdentity` selects the real CLI contract rather than today's hybrid: the
- * reference CLI is issued by the OpenAI Node SDK, so it carries no browser
- * headers and no device token, and it adds the agent envelope. Left unset it
- * keeps the current behaviour for every provider.
- */
-export async function codebuddyHeaders(
-  variant: CodeBuddyVariant,
-  context?: ProviderDispatchContext,
-  request?: CanonicalRequest,
-  cliIdentity?: boolean,
-): Promise<Record<string, string>> {
-  const identity = variant === "IDE" ? "IDE" : "CLI";
-  const ua = buildCodeBuddyUserAgent(identity);
-  const deviceToken = readBuddyDeviceToken(context);
-  const conversationId = resolveInboundSessionId(context, request);
-  const conversation = buddyConversationHeaders({
-    ...(conversationId === undefined ? {} : { conversationId }),
-  });
-  if (resolveBuddyIdentity(context, cliIdentity) === "cli") {
-    return {
-      // The CLI sends `Accept: application/json` and no device token. The
-      // browser-shaped pair (`X-Requested-With`, `x-codebuddy-request`) stays
-      // off this branch: a request carrying both halves is a hybrid no real
-      // client produces, and that is a fidelity choice, not a requirement —
-      // the live API answers either way.
-      accept: "application/json",
-      "User-Agent": `CLI/${getCodeBuddyVersion()} CodeBuddy/${getCodeBuddyVersion()}`,
-      "X-Product": "SaaS",
-      "X-IDE-Type": "CLI",
-      "X-IDE-Name": "CLI",
-      "X-IDE-Version": getCodeBuddyVersion(),
-      "X-Domain": codebuddyDomain(variant),
-      // The conversation purpose the reference client states. Not required —
-      // verified against the live API that a request succeeds without it — but
-      // it is part of the client's real request, and matching the client is the
-      // defence against a channel check tightening later.
-      "X-Agent-Purpose": "conversation",
-      ...conversation,
-      ...buddyCliTraceHeaders(conversation),
-      ...buddyCliHeaders({ userId: buddyUserId(context) }),
-    };
-  }
-  return {
-    accept: "text/event-stream",
-    "User-Agent": ua,
-    "X-Product": "SaaS",
-    "X-IDE-Type": identity,
-    "X-IDE-Name": identity,
-    // Every reference branch sends this, in both the desktop and the CLI
-    // product; its absence here is what made a CodeBuddy dispatch the only
-    // buddy caller with no client version on the wire.
-    "X-IDE-Version": identity === "CLI" ? getCodeBuddyCnVersion() : getCodeBuddyVersion(),
-    "X-Domain": codebuddyDomain(variant),
-    "x-requested-with": "XMLHttpRequest",
-    "x-codebuddy-request": "1",
-    // Same value the CLI branch sends, for the same reason: part of the
-    // client's real request, harmless when the channel check does not read it.
-    "X-Agent-Purpose": "conversation",
-    // Same family WorkBuddy sends: the upstream's channel check reads the
-    // turn-level aggregation key, and a chat request without it is refused for a
-    // missing conversation id.
-    ...conversation,
-    ...(deviceToken === undefined ? {} : { "X-Device-Token": deviceToken }),
-  };
-}
-
-/**
- * Shared OpenAI-compatible chat-adapter config for both CodeBuddy variants:
- * bearer auth, chat-only wire contract, per-request identity headers, and the
- * variant's payload hook.
- */
-export function codebuddyAdapterConfig(args: {
-  providerId: ProviderId;
-  baseUrl: string;
-  variant: CodeBuddyVariant;
-  prePayload: (
-    payload: Record<string, unknown>,
-    request: CanonicalRequest,
-    candidate: ProviderDispatchTarget,
-  ) => void;
-  fetchImpl?: typeof fetch;
-  cliIdentity?: boolean;
-}): OpenAICompatibleAdapterConfig {
-  return withBearerAuthentication({
-    provider_id: args.providerId,
-    base_url: args.baseUrl,
-    buildExtraHeaders: (context, request) =>
-      codebuddyHeaders(args.variant, context, request, args.cliIdentity),
-    prePayload: args.prePayload,
-    ...(args.fetchImpl ? { fetchImpl: args.fetchImpl } : {}),
-  });
-}
+import { withBearerAuthentication, type OpenAICompatibleAdapterConfig } from "../../../adapters/compatible-adapter";
+import type { CanonicalRequest } from "../../../../protocol/canonical-model";
+import type { ProviderDispatchContext, ProviderDispatchTarget, ProviderId } from "../../../provider-contracts";
+import { resolveInboundSessionId } from "../../../operations/session-resolution";
+import { buddyConversationHeaders } from "../buddy-conversation-headers";
+import { buddyEnterprise, buddyUserId, resolveBuddyIdentity } from "../buddy-identity";
+import { buildCodeBuddyUserAgent, getCodeBuddyCnVersion, getCodeBuddyVersion } from "../../../operations/client-versions";
+import { readBuddyDeviceToken } from "../workbuddy/shared";
+import { buddyCliHeaders, buddyCliTraceHeaders } from "../buddy-cli-headers";
+
+/** Client identity variant behind the per-request CodeBuddy headers. */
+export type CodeBuddyVariant = "IDE" | "CLI";
+
+export function codebuddyDomain(variant: CodeBuddyVariant): string {
+  return variant === "IDE" ? "www.codebuddy.ai" : "copilot.tencent.com";
+}
+
+
+/**
+ * Per-request CodeBuddy headers — exact provider contract shared by the
+ * international (IDE identity) and CN (CLI identity). The stable conversation
+ * header is preserved from the client when available; request IDs rotate.
+ *
+ * `cliIdentity` selects the real CLI contract rather than today's hybrid: the
+ * reference CLI is issued by the OpenAI Node SDK, so it carries no browser
+ * headers and no device token, and it adds the agent envelope. Left unset it
+ * keeps the current behaviour for every provider.
+ */
+export async function codebuddyHeaders(
+  variant: CodeBuddyVariant,
+  context?: ProviderDispatchContext,
+  request?: CanonicalRequest,
+  cliIdentity?: boolean,
+): Promise<Record<string, string>> {
+  const identity = variant === "IDE" ? "IDE" : "CLI";
+  const ua = buildCodeBuddyUserAgent(identity);
+  const deviceToken = readBuddyDeviceToken(context);
+  const conversationId = resolveInboundSessionId(context, request);
+  const conversation = buddyConversationHeaders({
+    ...(conversationId === undefined ? {} : { conversationId }),
+  });
+  if (resolveBuddyIdentity(context, cliIdentity) === "cli") {
+    return {
+      // The CLI sends `Accept: application/json` and no device token. The
+      // browser-shaped pair (`X-Requested-With`, `x-codebuddy-request`) stays
+      // off this branch: a request carrying both halves is a hybrid no real
+      // client produces, and that is a fidelity choice, not a requirement —
+      // the live API answers either way.
+      accept: "application/json",
+      "User-Agent": `CLI/${getCodeBuddyVersion()} CodeBuddy/${getCodeBuddyVersion()}`,
+      "X-Product": "SaaS",
+      "X-IDE-Type": "CLI",
+      "X-IDE-Name": "CLI",
+      "X-IDE-Version": getCodeBuddyVersion(),
+      "X-Domain": codebuddyDomain(variant),
+      // The conversation purpose the reference client states. Not required —
+      // verified against the live API that a request succeeds without it — but
+      // it is part of the client's real request, and matching the client is the
+      // defence against a channel check tightening later.
+      "X-Agent-Purpose": "conversation",
+      ...conversation,
+      ...buddyCliTraceHeaders(conversation),
+      ...buddyCliHeaders({
+        userId: buddyUserId(context),
+        enterpriseId: buddyEnterprise(context),
+      }),
+    };
+  }
+  return {
+    accept: "text/event-stream",
+    "User-Agent": ua,
+    "X-Product": "SaaS",
+    "X-IDE-Type": identity,
+    "X-IDE-Name": identity,
+    // Every reference branch sends this, in both the desktop and the CLI
+    // product; its absence here is what made a CodeBuddy dispatch the only
+    // buddy caller with no client version on the wire.
+    "X-IDE-Version": identity === "CLI" ? getCodeBuddyCnVersion() : getCodeBuddyVersion(),
+    "X-Domain": codebuddyDomain(variant),
+    "x-requested-with": "XMLHttpRequest",
+    "x-codebuddy-request": "1",
+    // Same value the CLI branch sends, for the same reason: part of the
+    // client's real request, harmless when the channel check does not read it.
+    "X-Agent-Purpose": "conversation",
+    // Same family WorkBuddy sends: the upstream's channel check reads the
+    // turn-level aggregation key, and a chat request without it is refused for a
+    // missing conversation id.
+    ...conversation,
+    ...(deviceToken === undefined ? {} : { "X-Device-Token": deviceToken }),
+  };
+}
+
+/**
+ * Shared OpenAI-compatible chat-adapter config for both CodeBuddy variants:
+ * bearer auth, chat-only wire contract, per-request identity headers, and the
+ * variant's payload hook.
+ */
+export function codebuddyAdapterConfig(args: {
+  providerId: ProviderId;
+  baseUrl: string;
+  variant: CodeBuddyVariant;
+  prePayload: (
+    payload: Record<string, unknown>,
+    request: CanonicalRequest,
+    candidate: ProviderDispatchTarget,
+  ) => void;
+  fetchImpl?: typeof fetch;
+  cliIdentity?: boolean;
+}): OpenAICompatibleAdapterConfig {
+  return withBearerAuthentication({
+    provider_id: args.providerId,
+    base_url: args.baseUrl,
+    buildExtraHeaders: (context, request) =>
+      codebuddyHeaders(args.variant, context, request, args.cliIdentity),
+    prePayload: args.prePayload,
+    ...(args.fetchImpl ? { fetchImpl: args.fetchImpl } : {}),
+  });
+}

package/src/providers/integrations/buddy/workbuddy/quota.ts

@@ -38,11 +38,6 @@ const WORKBUDDY_CN_BILLING_BASE_URL = "https://www.codebuddy.cn";
 export const WORKBUDDY_USAGE_URL =
   `${providerBaseUrl("workbuddy")}/v2/billing/meter/get-user-resource`;
 
-export const WORKBUDDY_REQUEST_USAGE_URL =
-  `${providerBaseUrl("workbuddy")}/billing/meter/get-user-request-usage`;
-export const WORKBUDDY_CN_REQUEST_USAGE_URL =
-  `${providerBaseUrl("workbuddycn")}/billing/meter/get-user-request-usage`;
-
 async function billingHeaders(
   domain: string,
   clientVersion: string,
@@ -164,9 +159,3 @@ export async function fetchWorkBuddyRequestUsage(
   });
 }
 
-export const fetchWorkBuddyCnRequestUsage = (
-  credential: string,
-  fetcher: FetchLike,
-  days: number = 30,
-): Promise<BuddyUsageResult> =>
-  fetchWorkBuddyRequestUsage(credential, fetcher, { providerId: "workbuddycn", days });

package/src/providers/integrations/buddy/workbuddy/shared.ts

@@ -8,7 +8,7 @@ import type { ProviderDispatchContext, ProviderDispatchTarget, ProviderId } from
 import { resolveInboundSessionId } from "../../../operations/session-resolution";
 import { buddyConversationHeaders } from "../buddy-conversation-headers";
 import { buddyCliHeaders, buddyCliTraceHeaders } from "../buddy-cli-headers";
-import { buddyUserId, resolveBuddyIdentity } from "../buddy-identity";
+import { buddyEnterprise, buddyUserId, resolveBuddyIdentity } from "../buddy-identity";
 import {
   buildWorkBuddyUserAgent,
   getWorkBuddyClientVersion,
@@ -91,7 +91,10 @@ export async function workbuddyHeaders(
       "x-no-enterprise-id": "1",
       ...conversation,
       ...buddyCliTraceHeaders(conversation),
-      ...buddyCliHeaders({ userId: buddyUserId(context) }),
+      ...buddyCliHeaders({
+        userId: buddyUserId(context),
+        enterpriseId: buddyEnterprise(context),
+      }),
     };
   }
   const headers: Record<string, string> = {