@acosmi/sdk-ts 2.17.0 → 2.19.0

package/dist/node/index.cjs

@@ -2319,6 +2319,22 @@ function notifyUpstreamActivity(cb) {
   } catch {
   }
 }
+var GATEWAY_REQUEST_ID_HEADER = "X-Acosmi-Request-Id";
+function readGatewayRequestID(headers) {
+  const raw = headers.get(GATEWAY_REQUEST_ID_HEADER);
+  if (!raw) return void 0;
+  const trimmed = raw.trim();
+  return trimmed === "" ? void 0 : trimmed;
+}
+function notifyGatewayRequestID(cb, headers) {
+  if (!cb) return;
+  const id = readGatewayRequestID(headers);
+  if (id === void 0) return;
+  try {
+    cb(id);
+  } catch {
+  }
+}
 var DEFAULT_API_TIMEOUT_MS = 6e4;
 function newDeferred() {
   let resolve;
@@ -3026,7 +3042,7 @@ var Client = class _Client {
    * 响应的 tokenRemaining / callRemaining 字段来自服务端 Header, 反映结算后余额
    * v0.5.0: 根据 provider 自动路由到 /anthropic 或 /chat 端点
    */
-  async chat(modelID, req, signal) {
+  async chat(modelID, req, signal, onGatewayRequestID) {
     const r = { ...req, stream: false };
     const ctl = withRequestTimeout(CHAT_REQUEST_TIMEOUT_MS, signal);
     try {
@@ -3039,6 +3055,7 @@ var Client = class _Client {
         ctl.signal,
         CHAT_REQUEST_TIMEOUT_MS
       );
+      notifyGatewayRequestID(onGatewayRequestID, headers);
       const resp = adapter.parseResponse(result);
       const v1 = headers.get("X-Token-Remaining");
       if (v1) {
@@ -3159,28 +3176,29 @@ var Client = class _Client {
    *   Anthropic → chatMessagesAnthropic (现有路径, POST /anthropic)
    *   其他厂商 → chatMessagesOpenAI (POST /chat, 响应转换为 AnthropicResponse)
    */
-  async chatMessages(modelID, req, signal) {
+  async chatMessages(modelID, req, signal, onGatewayRequestID) {
     const m = await this.ensureModelCached(modelID, signal);
     const adapter = getAdapterForModel(m);
     if (adapter.format() === 0 /* Anthropic */) {
-      return this.chatMessagesAnthropic(modelID, req, adapter, signal);
+      return this.chatMessagesAnthropic(modelID, req, adapter, signal, onGatewayRequestID);
     }
-    return this.chatMessagesOpenAI(modelID, req, adapter, signal);
+    return this.chatMessagesOpenAI(modelID, req, adapter, signal, onGatewayRequestID);
   }
-  async chatMessagesAnthropic(modelID, req, adapter, signal) {
+  async chatMessagesAnthropic(modelID, req, adapter, signal, onGatewayRequestID) {
     const r = { ...req, stream: false };
     const ctl = withRequestTimeout(CHAT_REQUEST_TIMEOUT_MS, signal);
     try {
       const caps = this.getCachedCapabilities(modelID) ?? zeroModelCapabilities();
       const body = adapter.buildRequestBody(caps, r);
       const data = JSON.stringify(body);
-      const { result } = await this.doJSONFullRaw(
+      const { result, headers } = await this.doJSONFullRaw(
         "POST",
         `/managed-models/${encodeURIComponent(modelID)}/anthropic`,
         data,
         ctl.signal,
         CHAT_REQUEST_TIMEOUT_MS
       );
+      notifyGatewayRequestID(onGatewayRequestID, headers);
       const rawStr = new TextDecoder().decode(result);
       try {
         const wrapper = JSON.parse(rawStr);
@@ -3205,7 +3223,7 @@ var Client = class _Client {
       ctl.dispose();
     }
   }
-  async chatMessagesOpenAI(modelID, req, adapter, signal) {
+  async chatMessagesOpenAI(modelID, req, adapter, signal, onGatewayRequestID) {
     const r = { ...req, stream: false };
     const ctl = withRequestTimeout(CHAT_REQUEST_TIMEOUT_MS, signal);
     try {
@@ -3213,13 +3231,14 @@ var Client = class _Client {
       const body = adapter.buildRequestBody(caps, r);
       const data = JSON.stringify(body);
       const endpoint = `/managed-models/${encodeURIComponent(modelID)}${adapter.endpointSuffix()}`;
-      const { result } = await this.doJSONFullRaw(
+      const { result, headers } = await this.doJSONFullRaw(
         "POST",
         endpoint,
         data,
         ctl.signal,
         CHAT_REQUEST_TIMEOUT_MS
       );
+      notifyGatewayRequestID(onGatewayRequestID, headers);
       const { parseOpenAIResponseToAnthropic: parseOpenAIResponseToAnthropic2 } = await Promise.resolve().then(() => (init_openai(), openai_exports));
       return parseOpenAIResponseToAnthropic2(result);
     } finally {
@@ -3231,10 +3250,11 @@ var Client = class _Client {
    * v0.5.0: 根据 adapter 路由端点
    *
    * @param onUpstreamActivity 见 {@link UpstreamActivityCallback}
+   * @param onGatewayRequestID 见 {@link GatewayRequestIDCallback}
    */
-  chatStream(modelID, req, signal, onUpstreamActivity) {
+  chatStream(modelID, req, signal, onUpstreamActivity, onGatewayRequestID) {
     return {
-      [Symbol.asyncIterator]: () => this.chatStreamGen(modelID, req, signal, false, onUpstreamActivity)
+      [Symbol.asyncIterator]: () => this.chatStreamGen(modelID, req, signal, false, onUpstreamActivity, onGatewayRequestID)
     };
   }
   /**
@@ -3243,13 +3263,14 @@ var Client = class _Client {
    * 无 started/settled/failed 自定义事件, 无 data: [DONE], message_stop 为自然结束
    *
    * @param onUpstreamActivity 见 {@link UpstreamActivityCallback}
+   * @param onGatewayRequestID 见 {@link GatewayRequestIDCallback}
    */
-  chatMessagesStream(modelID, req, signal, onUpstreamActivity) {
+  chatMessagesStream(modelID, req, signal, onUpstreamActivity, onGatewayRequestID) {
     return {
-      [Symbol.asyncIterator]: () => this.chatMessagesStreamGen(modelID, req, signal, false, onUpstreamActivity)
+      [Symbol.asyncIterator]: () => this.chatMessagesStreamGen(modelID, req, signal, false, onUpstreamActivity, onGatewayRequestID)
     };
   }
-  async *chatStreamGen(modelID, req, signal, retried, onUpstreamActivity) {
+  async *chatStreamGen(modelID, req, signal, retried, onUpstreamActivity, onGatewayRequestID) {
     const r = { ...req, stream: true };
     const { body, adapter } = await this.buildChatRequest(modelID, r, signal);
     const token = await this.ensureToken(signal);
@@ -3282,9 +3303,10 @@ var Client = class _Client {
           `stream: unauthorized and refresh failed: ${refreshErr instanceof Error ? refreshErr.message : String(refreshErr)}`
         );
       }
-      yield* this.chatStreamGen(modelID, req, signal, true, onUpstreamActivity);
+      yield* this.chatStreamGen(modelID, req, signal, true, onUpstreamActivity, onGatewayRequestID);
       return;
     }
+    notifyGatewayRequestID(onGatewayRequestID, resp.headers);
     if (!resp.ok) {
       const bodyBytes = await readLimited(resp.body, maxErrorBodySize);
       throw parseHTTPErrorWithHeader(resp.status, bodyBytes, resp.headers);
@@ -3319,7 +3341,7 @@ var Client = class _Client {
       }
     }
   }
-  async *chatMessagesStreamGen(modelID, req, signal, retried, onUpstreamActivity) {
+  async *chatMessagesStreamGen(modelID, req, signal, retried, onUpstreamActivity, onGatewayRequestID) {
     const r = { ...req, stream: true };
     const { body, adapter } = await this.buildChatRequest(modelID, r, signal);
     const token = await this.ensureToken(signal);
@@ -3352,9 +3374,10 @@ var Client = class _Client {
           `messages stream: unauthorized and refresh failed: ${refreshErr instanceof Error ? refreshErr.message : String(refreshErr)}`
         );
       }
-      yield* this.chatMessagesStreamGen(modelID, req, signal, true, onUpstreamActivity);
+      yield* this.chatMessagesStreamGen(modelID, req, signal, true, onUpstreamActivity, onGatewayRequestID);
       return;
     }
+    notifyGatewayRequestID(onGatewayRequestID, resp.headers);
     if (!resp.ok) {
       const bodyBytes = await readLimited(resp.body, maxErrorBodySize);
       throw parseHTTPErrorWithHeader(resp.status, bodyBytes, resp.headers);
@@ -7871,6 +7894,7 @@ exports.FilterStatusFallbackTkdistSkew = FilterStatusFallbackTkdistSkew;
 exports.FilterStatusInternalBypass = FilterStatusInternalBypass;
 exports.FilterStatusOK = FilterStatusOK;
 exports.FilterStatusUnknown = FilterStatusUnknown;
+exports.GATEWAY_REQUEST_ID_HEADER = GATEWAY_REQUEST_ID_HEADER;
 exports.IdempotencyKeyHeader = IdempotencyKeyHeader;
 exports.InMemoryTokenStore = InMemoryTokenStore;
 exports.LocalStorageTokenStore = LocalStorageTokenStore;