promptopskit 0.2.6 → 0.3.0

package/dist/usagetap/index.cjs

@@ -0,0 +1,425 @@
+"use strict";
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+
+// src/usagetap/index.ts
+var usagetap_exports = {};
+__export(usagetap_exports, {
+  applyUsageTapEntitlements: () => applyUsageTapEntitlements,
+  beginUsageTapCall: () => beginUsageTapCall,
+  createUsageTapClient: () => createUsageTapClient,
+  defaultUsageTapErrorMapper: () => defaultUsageTapErrorMapper,
+  endUsageTapCall: () => endUsageTapCall,
+  extractAnthropicUsage: () => extractAnthropicUsage,
+  extractGeminiUsage: () => extractGeminiUsage,
+  extractOpenAIUsage: () => extractOpenAIUsage,
+  runAnthropicWithUsageTap: () => runAnthropicWithUsageTap,
+  runGeminiWithUsageTap: () => runGeminiWithUsageTap,
+  runOpenAIWithUsageTap: () => runOpenAIWithUsageTap,
+  runOpenRouterWithUsageTap: () => runOpenRouterWithUsageTap,
+  withUsageTapCall: () => withUsageTapCall
+});
+module.exports = __toCommonJS(usagetap_exports);
+
+// src/usagetap/client.ts
+var DEFAULT_BASE_URL = "https://api.usagetap.com";
+var ACCEPT_HEADER = "application/vnd.usagetap.v1+json";
+function createUsageTapHttpError(status, body) {
+  const error = new Error(`UsageTap request failed with status ${status}`);
+  error.statusCode = status;
+  error.body = body;
+  return error;
+}
+function createUsageTapClient(config) {
+  const fetchImpl = config.fetch ?? globalThis.fetch;
+  if (!fetchImpl) {
+    throw new Error("Fetch API is not available. Provide config.fetch when creating the UsageTap client.");
+  }
+  const baseUrl = (config.baseUrl ?? DEFAULT_BASE_URL).replace(/\/$/, "");
+  async function request(path, body, init = {}) {
+    const response = await fetchImpl(`${baseUrl}${path}`, {
+      method: "POST",
+      headers: {
+        Authorization: `Bearer ${config.apiKey}`,
+        Accept: ACCEPT_HEADER,
+        "Content-Type": "application/json"
+      },
+      body: JSON.stringify(body),
+      signal: init.signal
+    });
+    const text = await response.text();
+    const parsed = text ? JSON.parse(text) : {};
+    if (!response.ok) {
+      throw createUsageTapHttpError(response.status, parsed);
+    }
+    return parsed;
+  }
+  return {
+    request,
+    beginCall(begin, init) {
+      return request("/call_begin", begin, init);
+    },
+    endCall(end, init) {
+      return request("/call_end", end, init);
+    }
+  };
+}
+
+// src/usagetap/lifecycle.ts
+function isObject(value) {
+  return typeof value === "object" && value !== null;
+}
+function isInvokeResult(value) {
+  return isObject(value) && "result" in value && "usage" in value;
+}
+function getErrorStatusCode(error) {
+  if (!isObject(error)) {
+    return void 0;
+  }
+  if (typeof error.statusCode === "number") {
+    return error.statusCode;
+  }
+  if (typeof error.status === "number") {
+    return error.status;
+  }
+  if (isObject(error.response) && typeof error.response.status === "number") {
+    return error.response.status;
+  }
+  return void 0;
+}
+function attachEndErrorCause(thrownError, endError) {
+  if (thrownError instanceof Error) {
+    if (thrownError.cause === void 0) {
+      Object.defineProperty(thrownError, "cause", {
+        value: endError,
+        configurable: true,
+        writable: true
+      });
+    }
+    throw thrownError;
+  }
+  throw new AggregateError([thrownError, endError], "Vendor call failed and UsageTap call_end also failed.");
+}
+function defaultUsageTapErrorMapper(error) {
+  return {
+    code: "VENDOR_ERROR",
+    message: String(error)
+  };
+}
+function beginUsageTapCall(client, begin, init) {
+  return client.beginCall(begin, init);
+}
+function endUsageTapCall(client, end, init) {
+  return client.endCall(end, init);
+}
+async function withUsageTapCall(client, options) {
+  const begin = await beginUsageTapCall(client, options.begin, { signal: options.signal });
+  const onError = options.onError ?? defaultUsageTapErrorMapper;
+  let result;
+  let usage;
+  let thrownError;
+  const setUsage = (nextUsage) => {
+    usage = { ...usage, ...nextUsage };
+  };
+  try {
+    const invoked = await options.invoke({
+      begin,
+      setUsage,
+      signal: options.signal
+    });
+    if (isInvokeResult(invoked)) {
+      result = invoked.result;
+      setUsage(invoked.usage);
+    } else {
+      result = invoked;
+    }
+  } catch (error) {
+    thrownError = error;
+  }
+  const effectiveUsage = {
+    ...usage
+  };
+  if (thrownError) {
+    effectiveUsage.error = effectiveUsage.error ?? onError(thrownError);
+    effectiveUsage.responseStatusCode = effectiveUsage.responseStatusCode ?? getErrorStatusCode(thrownError);
+  } else {
+    effectiveUsage.responseStatusCode = effectiveUsage.responseStatusCode ?? 200;
+  }
+  let end;
+  try {
+    end = await endUsageTapCall(client, {
+      callId: begin.data.callId,
+      ...effectiveUsage
+    }, { signal: options.signal });
+  } catch (endError) {
+    if (thrownError) {
+      attachEndErrorCause(thrownError, endError);
+    }
+    throw endError;
+  }
+  if (thrownError) {
+    throw thrownError;
+  }
+  return {
+    result,
+    begin,
+    end,
+    effectiveUsage
+  };
+}
+
+// src/usagetap/providers.ts
+function cloneRequest(request) {
+  return structuredClone(request);
+}
+function isObject2(value) {
+  return typeof value === "object" && value !== null;
+}
+function isRecordArray(value) {
+  return Array.isArray(value) && value.every((item) => isObject2(item));
+}
+function isInvokeResult2(value) {
+  return isObject2(value) && "result" in value && "usage" in value;
+}
+function capabilityAllowed(allowed, capability) {
+  return allowed[capability] !== false;
+}
+function reasoningLevelOrder(level) {
+  switch (level) {
+    case "HIGH":
+      return 3;
+    case "MEDIUM":
+      return 2;
+    case "LOW":
+      return 1;
+    default:
+      return 0;
+  }
+}
+function capOpenAIReasoning(body, allowed) {
+  if (typeof body.reasoning_effort !== "string") {
+    return;
+  }
+  const current = String(body.reasoning_effort).toUpperCase();
+  const allowedLevel = allowed.reasoningLevel ?? "HIGH";
+  if (allowedLevel === "NONE") {
+    delete body.reasoning_effort;
+    return;
+  }
+  if (reasoningLevelOrder(current) > reasoningLevelOrder(allowedLevel)) {
+    body.reasoning_effort = allowedLevel.toLowerCase();
+  }
+}
+function capGeminiReasoning(body, allowed) {
+  if (allowed.reasoningLevel === "NONE") {
+    delete body.thinkingConfig;
+    return;
+  }
+  const budgets = {
+    LOW: 1024,
+    MEDIUM: 4096,
+    HIGH: 8192
+  };
+  const allowedBudget = allowed.reasoningLevel ? budgets[allowed.reasoningLevel] : void 0;
+  if (allowedBudget === void 0) {
+    return;
+  }
+  if (!isObject2(body.thinkingConfig)) {
+    body.thinkingConfig = { thinkingBudget: allowedBudget };
+    return;
+  }
+  const thinkingConfig = body.thinkingConfig;
+  if (typeof thinkingConfig.thinkingBudget === "number") {
+    thinkingConfig.thinkingBudget = Math.min(thinkingConfig.thinkingBudget, allowedBudget);
+    return;
+  }
+  thinkingConfig.thinkingBudget = allowedBudget;
+}
+function applyModelTier(request, allowed, options) {
+  const nextModel = allowed.premium && options.modelTiers?.premium ? options.modelTiers.premium : allowed.standard && options.modelTiers?.standard ? options.modelTiers.standard : void 0;
+  if (!nextModel) {
+    return;
+  }
+  request.model = nextModel;
+  if (isObject2(request.body) && "model" in request.body) {
+    request.body.model = nextModel;
+  }
+}
+function filterOpenAITools(tools, allowed, options) {
+  return tools.filter((tool) => {
+    if (tool.type === "web_search" && allowed.search === false) {
+      return false;
+    }
+    const functionDef = isObject2(tool.function) ? tool.function : void 0;
+    const name = typeof functionDef?.name === "string" ? functionDef.name : typeof tool.name === "string" ? tool.name : typeof tool.type === "string" ? tool.type : void 0;
+    if (!name) {
+      return true;
+    }
+    const capability = options.toolEntitlements?.[name];
+    return capability ? capabilityAllowed(allowed, capability) : true;
+  });
+}
+function filterAnthropicTools(tools, allowed, options) {
+  return tools.filter((tool) => {
+    const name = typeof tool.name === "string" ? tool.name : void 0;
+    if (!name) {
+      return true;
+    }
+    const capability = options.toolEntitlements?.[name];
+    return capability ? capabilityAllowed(allowed, capability) : true;
+  });
+}
+function filterGeminiTools(tools, allowed, options) {
+  return tools.map((tool) => {
+    const declarations = Array.isArray(tool.functionDeclarations) ? tool.functionDeclarations.filter((declaration) => {
+      if (!isObject2(declaration) || typeof declaration.name !== "string") {
+        return true;
+      }
+      const capability = options.toolEntitlements?.[declaration.name];
+      return capability ? capabilityAllowed(allowed, capability) : true;
+    }) : tool.functionDeclarations;
+    return {
+      ...tool,
+      functionDeclarations: declarations
+    };
+  }).filter((tool) => {
+    if (!Array.isArray(tool.functionDeclarations)) {
+      return true;
+    }
+    return tool.functionDeclarations.length > 0;
+  });
+}
+function applyUsageTapEntitlements(request, begin, options = {}) {
+  const nextRequest = cloneRequest(request);
+  const allowed = begin.data.allowed ?? {};
+  applyModelTier(nextRequest, allowed, options);
+  if (!isObject2(nextRequest.body)) {
+    return nextRequest;
+  }
+  if (nextRequest.provider === "openai" || nextRequest.provider === "openrouter") {
+    capOpenAIReasoning(nextRequest.body, allowed);
+    if (isRecordArray(nextRequest.body.tools)) {
+      nextRequest.body.tools = filterOpenAITools(nextRequest.body.tools, allowed, options);
+    }
+    return nextRequest;
+  }
+  if (nextRequest.provider === "gemini") {
+    capGeminiReasoning(nextRequest.body, allowed);
+    if (isRecordArray(nextRequest.body.tools)) {
+      nextRequest.body.tools = filterGeminiTools(nextRequest.body.tools, allowed, options);
+    }
+    return nextRequest;
+  }
+  if (nextRequest.provider === "anthropic" && isRecordArray(nextRequest.body.tools)) {
+    nextRequest.body.tools = filterAnthropicTools(nextRequest.body.tools, allowed, options);
+  }
+  return nextRequest;
+}
+function extractOpenAIUsage(response, meta = {}) {
+  const usage = isObject2(response) && isObject2(response.usage) ? response.usage : {};
+  const promptDetails = isObject2(usage.prompt_tokens_details) ? usage.prompt_tokens_details : {};
+  const completionDetails = isObject2(usage.completion_tokens_details) ? usage.completion_tokens_details : {};
+  return {
+    modelUsed: meta.modelUsed,
+    inputTokens: typeof usage.prompt_tokens === "number" ? usage.prompt_tokens : 0,
+    responseTokens: typeof usage.completion_tokens === "number" ? usage.completion_tokens : 0,
+    cachedInputTokens: typeof promptDetails.cached_tokens === "number" ? promptDetails.cached_tokens : void 0,
+    reasoningTokens: typeof completionDetails.reasoning_tokens === "number" ? completionDetails.reasoning_tokens : void 0,
+    responseStatusCode: meta.responseStatusCode ?? 200
+  };
+}
+function extractAnthropicUsage(response, meta = {}) {
+  const usage = isObject2(response) && isObject2(response.usage) ? response.usage : {};
+  return {
+    modelUsed: meta.modelUsed,
+    inputTokens: typeof usage.input_tokens === "number" ? usage.input_tokens : 0,
+    responseTokens: typeof usage.output_tokens === "number" ? usage.output_tokens : 0,
+    cachedInputTokens: typeof usage.cache_read_input_tokens === "number" ? usage.cache_read_input_tokens : void 0,
+    responseStatusCode: meta.responseStatusCode ?? 200
+  };
+}
+function extractGeminiUsage(response, meta = {}) {
+  const usage = isObject2(response) && isObject2(response.usageMetadata) ? response.usageMetadata : {};
+  return {
+    modelUsed: meta.modelUsed,
+    inputTokens: typeof usage.promptTokenCount === "number" ? usage.promptTokenCount : 0,
+    responseTokens: typeof usage.candidatesTokenCount === "number" ? usage.candidatesTokenCount : 0,
+    cachedInputTokens: typeof usage.cachedContentTokenCount === "number" ? usage.cachedContentTokenCount : void 0,
+    reasoningTokens: typeof usage.thoughtsTokenCount === "number" ? usage.thoughtsTokenCount : void 0,
+    responseStatusCode: meta.responseStatusCode ?? 200
+  };
+}
+async function runProviderWithUsageTap(client, options, defaultExtractor) {
+  let requestUsed = cloneRequest(options.request);
+  const lifecycle = await withUsageTapCall(client, {
+    begin: options.begin,
+    signal: options.signal,
+    onError: options.onError ?? defaultUsageTapErrorMapper,
+    invoke: async ({ begin, setUsage }) => {
+      requestUsed = options.entitlementMode === "apply" ? applyUsageTapEntitlements(options.request, begin, options) : cloneRequest(options.request);
+      const invoked = await options.invoke(requestUsed);
+      if (isInvokeResult2(invoked)) {
+        return invoked;
+      }
+      const response = invoked;
+      const usage = (options.extractUsage ?? defaultExtractor)(response, {
+        modelUsed: requestUsed.model,
+        responseStatusCode: 200
+      });
+      setUsage(usage);
+      return response;
+    }
+  });
+  return {
+    response: lifecycle.result,
+    begin: lifecycle.begin,
+    end: lifecycle.end,
+    requestUsed,
+    effectiveUsage: lifecycle.effectiveUsage,
+    allowed: lifecycle.begin.data.allowed ?? {}
+  };
+}
+function runOpenAIWithUsageTap(client, options) {
+  return runProviderWithUsageTap(client, options, extractOpenAIUsage);
+}
+function runOpenRouterWithUsageTap(client, options) {
+  return runProviderWithUsageTap(client, options, extractOpenAIUsage);
+}
+function runAnthropicWithUsageTap(client, options) {
+  return runProviderWithUsageTap(client, options, extractAnthropicUsage);
+}
+function runGeminiWithUsageTap(client, options) {
+  return runProviderWithUsageTap(client, options, extractGeminiUsage);
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  applyUsageTapEntitlements,
+  beginUsageTapCall,
+  createUsageTapClient,
+  defaultUsageTapErrorMapper,
+  endUsageTapCall,
+  extractAnthropicUsage,
+  extractGeminiUsage,
+  extractOpenAIUsage,
+  runAnthropicWithUsageTap,
+  runGeminiWithUsageTap,
+  runOpenAIWithUsageTap,
+  runOpenRouterWithUsageTap,
+  withUsageTapCall
+});
+//# sourceMappingURL=index.cjs.map

package/dist/usagetap/index.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/usagetap/index.ts","../../src/usagetap/client.ts","../../src/usagetap/lifecycle.ts","../../src/usagetap/providers.ts"],"sourcesContent":["export type {\n  UsageTapAllowed,\n  UsageTapAllowedCapability,\n  UsageTapBeginRequest,\n  UsageTapBeginResponse,\n  UsageTapCallOptions,\n  UsageTapCallResult,\n  UsageTapClient,\n  UsageTapClientConfig,\n  UsageTapEndRequest,\n  UsageTapEndResponse,\n  UsageTapEndUsage,\n  UsageTapEntitlementMode,\n  UsageTapEntitlementOptions,\n  UsageTapErrorPayload,\n  UsageTapInvokeContext,\n  UsageTapInvokeResult,\n  UsageTapProviderRunOptions,\n  UsageTapProviderRunResult,\n  UsageTapReasoningLevel,\n} from './types.js';\nexport { createUsageTapClient } from './client.js';\nexport { beginUsageTapCall, defaultUsageTapErrorMapper, endUsageTapCall, withUsageTapCall } from './lifecycle.js';\nexport {\n  applyUsageTapEntitlements,\n  extractAnthropicUsage,\n  extractGeminiUsage,\n  extractOpenAIUsage,\n  runAnthropicWithUsageTap,\n  runGeminiWithUsageTap,\n  runOpenAIWithUsageTap,\n  runOpenRouterWithUsageTap,\n} from './providers.js';\n","import type {\n  UsageTapBeginRequest,\n  UsageTapBeginResponse,\n  UsageTapClient,\n  UsageTapClientConfig,\n  UsageTapEndRequest,\n  UsageTapEndResponse,\n  UsageTapRequestInit,\n} from './types.js';\n\nconst DEFAULT_BASE_URL = 'https://api.usagetap.com';\nconst ACCEPT_HEADER = 'application/vnd.usagetap.v1+json';\n\nfunction createUsageTapHttpError(status: number, body: unknown): Error & { statusCode: number; body: unknown } {\n  const error = new Error(`UsageTap request failed with status ${status}`) as Error & {\n    statusCode: number;\n    body: unknown;\n  };\n  error.statusCode = status;\n  error.body = body;\n  return error;\n}\n\nexport function createUsageTapClient(config: UsageTapClientConfig): UsageTapClient {\n  const fetchImpl = config.fetch ?? globalThis.fetch;\n\n  if (!fetchImpl) {\n    throw new Error('Fetch API is not available. Provide config.fetch when creating the UsageTap client.');\n  }\n\n  const baseUrl = (config.baseUrl ?? DEFAULT_BASE_URL).replace(/\\/$/, '');\n\n  async function request<T>(path: string, body: unknown, init: UsageTapRequestInit = {}): Promise<T> {\n    const response = await fetchImpl(`${baseUrl}${path}`, {\n      method: 'POST',\n      headers: {\n        Authorization: `Bearer ${config.apiKey}`,\n        Accept: ACCEPT_HEADER,\n        'Content-Type': 'application/json',\n      },\n      body: JSON.stringify(body),\n      signal: init.signal,\n    });\n\n    const text = await response.text();\n    const parsed = text ? JSON.parse(text) as unknown : {};\n\n    if (!response.ok) {\n      throw createUsageTapHttpError(response.status, parsed);\n    }\n\n    return parsed as T;\n  }\n\n  return {\n    request,\n    beginCall(begin: UsageTapBeginRequest, init?: UsageTapRequestInit): Promise<UsageTapBeginResponse> {\n      return request<UsageTapBeginResponse>('/call_begin', begin, init);\n    },\n    endCall(end: UsageTapEndRequest, init?: UsageTapRequestInit): Promise<UsageTapEndResponse> {\n      return request<UsageTapEndResponse>('/call_end', end, init);\n    },\n  };\n}\n","import type {\n  UsageTapCallOptions,\n  UsageTapCallResult,\n  UsageTapClient,\n  UsageTapEndRequest,\n  UsageTapEndUsage,\n  UsageTapErrorPayload,\n  UsageTapInvokeResult,\n  UsageTapRequestInit,\n} from './types.js';\n\nfunction isObject(value: unknown): value is Record<string, unknown> {\n  return typeof value === 'object' && value !== null;\n}\n\nfunction isInvokeResult<TResult>(value: TResult | UsageTapInvokeResult<TResult>): value is UsageTapInvokeResult<TResult> {\n  return isObject(value) && 'result' in value && 'usage' in value;\n}\n\nfunction getErrorStatusCode(error: unknown): number | undefined {\n  if (!isObject(error)) {\n    return undefined;\n  }\n\n  if (typeof error.statusCode === 'number') {\n    return error.statusCode;\n  }\n\n  if (typeof error.status === 'number') {\n    return error.status;\n  }\n\n  if (isObject(error.response) && typeof error.response.status === 'number') {\n    return error.response.status;\n  }\n\n  return undefined;\n}\n\nfunction attachEndErrorCause(thrownError: unknown, endError: unknown): never {\n  if (thrownError instanceof Error) {\n    if (thrownError.cause === undefined) {\n      Object.defineProperty(thrownError, 'cause', {\n        value: endError,\n        configurable: true,\n        writable: true,\n      });\n    }\n\n    throw thrownError;\n  }\n\n  throw new AggregateError([thrownError, endError], 'Vendor call failed and UsageTap call_end also failed.');\n}\n\nexport function defaultUsageTapErrorMapper(error: unknown): UsageTapErrorPayload {\n  return {\n    code: 'VENDOR_ERROR',\n    message: String(error),\n  };\n}\n\nexport function beginUsageTapCall(client: UsageTapClient, begin: Parameters<UsageTapClient['beginCall']>[0], init?: UsageTapRequestInit) {\n  return client.beginCall(begin, init);\n}\n\nexport function endUsageTapCall(client: UsageTapClient, end: UsageTapEndRequest, init?: UsageTapRequestInit) {\n  return client.endCall(end, init);\n}\n\nexport async function withUsageTapCall<TResult>(\n  client: UsageTapClient,\n  options: UsageTapCallOptions<TResult>,\n): Promise<UsageTapCallResult<TResult>> {\n  const begin = await beginUsageTapCall(client, options.begin, { signal: options.signal });\n  const onError = options.onError ?? defaultUsageTapErrorMapper;\n\n  let result: TResult | undefined;\n  let usage: UsageTapEndUsage | undefined;\n  let thrownError: unknown;\n\n  const setUsage = (nextUsage: UsageTapEndUsage): void => {\n    usage = { ...usage, ...nextUsage };\n  };\n\n  try {\n    const invoked = await options.invoke({\n      begin,\n      setUsage,\n      signal: options.signal,\n    });\n\n    if (isInvokeResult(invoked)) {\n      result = invoked.result;\n      setUsage(invoked.usage);\n    } else {\n      result = invoked;\n    }\n  } catch (error) {\n    thrownError = error;\n  }\n\n  const effectiveUsage: UsageTapEndUsage = {\n    ...usage,\n  };\n\n  if (thrownError) {\n    effectiveUsage.error = effectiveUsage.error ?? onError(thrownError);\n    effectiveUsage.responseStatusCode = effectiveUsage.responseStatusCode ?? getErrorStatusCode(thrownError);\n  } else {\n    effectiveUsage.responseStatusCode = effectiveUsage.responseStatusCode ?? 200;\n  }\n\n  let end;\n  try {\n    end = await endUsageTapCall(client, {\n      callId: begin.data.callId,\n      ...effectiveUsage,\n    }, { signal: options.signal });\n  } catch (endError) {\n    if (thrownError) {\n      attachEndErrorCause(thrownError, endError);\n    }\n    throw endError;\n  }\n\n  if (thrownError) {\n    throw thrownError;\n  }\n\n  return {\n    result: result as TResult,\n    begin,\n    end,\n    effectiveUsage,\n  };\n}\n","import type { ProviderRequest } from '../providers/types.js';\nimport { defaultUsageTapErrorMapper, withUsageTapCall } from './lifecycle.js';\nimport type {\n  UsageTapAllowed,\n  UsageTapAllowedCapability,\n  UsageTapBeginResponse,\n  UsageTapClient,\n  UsageTapEndUsage,\n  UsageTapEntitlementOptions,\n  UsageTapProviderRunOptions,\n  UsageTapProviderRunResult,\n  UsageTapReasoningLevel,\n} from './types.js';\n\nfunction cloneRequest(request: ProviderRequest): ProviderRequest {\n  return structuredClone(request);\n}\n\nfunction isObject(value: unknown): value is Record<string, unknown> {\n  return typeof value === 'object' && value !== null;\n}\n\nfunction isRecordArray(value: unknown): value is Array<Record<string, unknown>> {\n  return Array.isArray(value) && value.every((item) => isObject(item));\n}\n\nfunction isInvokeResult<TResult>(value: unknown): value is { result: TResult; usage: UsageTapEndUsage } {\n  return isObject(value) && 'result' in value && 'usage' in value;\n}\n\nfunction capabilityAllowed(allowed: UsageTapAllowed, capability: UsageTapAllowedCapability): boolean {\n  return allowed[capability] !== false;\n}\n\nfunction reasoningLevelOrder(level: UsageTapReasoningLevel | undefined): number {\n  switch (level) {\n    case 'HIGH':\n      return 3;\n    case 'MEDIUM':\n      return 2;\n    case 'LOW':\n      return 1;\n    default:\n      return 0;\n  }\n}\n\nfunction capOpenAIReasoning(body: Record<string, unknown>, allowed: UsageTapAllowed): void {\n  if (typeof body.reasoning_effort !== 'string') {\n    return;\n  }\n\n  const current = String(body.reasoning_effort).toUpperCase() as UsageTapReasoningLevel;\n  const allowedLevel = allowed.reasoningLevel ?? 'HIGH';\n\n  if (allowedLevel === 'NONE') {\n    delete body.reasoning_effort;\n    return;\n  }\n\n  if (reasoningLevelOrder(current) > reasoningLevelOrder(allowedLevel)) {\n    body.reasoning_effort = allowedLevel.toLowerCase();\n  }\n}\n\nfunction capGeminiReasoning(body: Record<string, unknown>, allowed: UsageTapAllowed): void {\n  if (allowed.reasoningLevel === 'NONE') {\n    delete body.thinkingConfig;\n    return;\n  }\n\n  const budgets: Record<Exclude<UsageTapReasoningLevel, 'NONE'>, number> = {\n    LOW: 1024,\n    MEDIUM: 4096,\n    HIGH: 8192,\n  };\n\n  const allowedBudget = allowed.reasoningLevel ? budgets[allowed.reasoningLevel as Exclude<UsageTapReasoningLevel, 'NONE'>] : undefined;\n\n  if (allowedBudget === undefined) {\n    return;\n  }\n\n  if (!isObject(body.thinkingConfig)) {\n    body.thinkingConfig = { thinkingBudget: allowedBudget };\n    return;\n  }\n\n  const thinkingConfig = body.thinkingConfig;\n\n  if (typeof thinkingConfig.thinkingBudget === 'number') {\n    thinkingConfig.thinkingBudget = Math.min(thinkingConfig.thinkingBudget, allowedBudget);\n    return;\n  }\n\n  thinkingConfig.thinkingBudget = allowedBudget;\n}\n\nfunction applyModelTier(request: ProviderRequest, allowed: UsageTapAllowed, options: UsageTapEntitlementOptions): void {\n  const nextModel = allowed.premium && options.modelTiers?.premium\n    ? options.modelTiers.premium\n    : allowed.standard && options.modelTiers?.standard\n      ? options.modelTiers.standard\n      : undefined;\n\n  if (!nextModel) {\n    return;\n  }\n\n  request.model = nextModel;\n  if (isObject(request.body) && 'model' in request.body) {\n    request.body.model = nextModel;\n  }\n}\n\nfunction filterOpenAITools(tools: Array<Record<string, unknown>>, allowed: UsageTapAllowed, options: UsageTapEntitlementOptions): Array<Record<string, unknown>> {\n  return tools.filter((tool) => {\n    if (tool.type === 'web_search' && allowed.search === false) {\n      return false;\n    }\n\n    const functionDef = isObject(tool.function) ? tool.function : undefined;\n    const name = typeof functionDef?.name === 'string'\n      ? functionDef.name\n      : typeof tool.name === 'string'\n        ? tool.name\n        : typeof tool.type === 'string'\n          ? tool.type\n          : undefined;\n\n    if (!name) {\n      return true;\n    }\n\n    const capability = options.toolEntitlements?.[name];\n    return capability ? capabilityAllowed(allowed, capability) : true;\n  });\n}\n\nfunction filterAnthropicTools(tools: Array<Record<string, unknown>>, allowed: UsageTapAllowed, options: UsageTapEntitlementOptions): Array<Record<string, unknown>> {\n  return tools.filter((tool) => {\n    const name = typeof tool.name === 'string' ? tool.name : undefined;\n    if (!name) {\n      return true;\n    }\n    const capability = options.toolEntitlements?.[name];\n    return capability ? capabilityAllowed(allowed, capability) : true;\n  });\n}\n\nfunction filterGeminiTools(tools: Array<Record<string, unknown>>, allowed: UsageTapAllowed, options: UsageTapEntitlementOptions): Array<Record<string, unknown>> {\n  return tools\n    .map((tool) => {\n      const declarations = Array.isArray(tool.functionDeclarations)\n        ? tool.functionDeclarations.filter((declaration) => {\n          if (!isObject(declaration) || typeof declaration.name !== 'string') {\n            return true;\n          }\n          const capability = options.toolEntitlements?.[declaration.name];\n          return capability ? capabilityAllowed(allowed, capability) : true;\n        })\n        : tool.functionDeclarations;\n\n      return {\n        ...tool,\n        functionDeclarations: declarations,\n      };\n    })\n    .filter((tool) => {\n      if (!Array.isArray(tool.functionDeclarations)) {\n        return true;\n      }\n      return tool.functionDeclarations.length > 0;\n    });\n}\n\nexport function applyUsageTapEntitlements(\n  request: ProviderRequest,\n  begin: UsageTapBeginResponse,\n  options: UsageTapEntitlementOptions = {},\n): ProviderRequest {\n  const nextRequest = cloneRequest(request);\n  const allowed = begin.data.allowed ?? {};\n\n  applyModelTier(nextRequest, allowed, options);\n\n  if (!isObject(nextRequest.body)) {\n    return nextRequest;\n  }\n\n  if (nextRequest.provider === 'openai' || nextRequest.provider === 'openrouter') {\n    capOpenAIReasoning(nextRequest.body, allowed);\n    if (isRecordArray(nextRequest.body.tools)) {\n      nextRequest.body.tools = filterOpenAITools(nextRequest.body.tools, allowed, options);\n    }\n    return nextRequest;\n  }\n\n  if (nextRequest.provider === 'gemini') {\n    capGeminiReasoning(nextRequest.body, allowed);\n    if (isRecordArray(nextRequest.body.tools)) {\n      nextRequest.body.tools = filterGeminiTools(nextRequest.body.tools, allowed, options);\n    }\n    return nextRequest;\n  }\n\n  if (nextRequest.provider === 'anthropic' && isRecordArray(nextRequest.body.tools)) {\n    nextRequest.body.tools = filterAnthropicTools(nextRequest.body.tools, allowed, options);\n  }\n\n  return nextRequest;\n}\n\ninterface UsageMeta {\n  modelUsed?: string;\n  responseStatusCode?: number;\n}\n\nexport function extractOpenAIUsage(response: unknown, meta: UsageMeta = {}): UsageTapEndUsage {\n  const usage = isObject(response) && isObject(response.usage) ? response.usage : {};\n  const promptDetails = isObject(usage.prompt_tokens_details) ? usage.prompt_tokens_details : {};\n  const completionDetails = isObject(usage.completion_tokens_details) ? usage.completion_tokens_details : {};\n\n  return {\n    modelUsed: meta.modelUsed,\n    inputTokens: typeof usage.prompt_tokens === 'number' ? usage.prompt_tokens : 0,\n    responseTokens: typeof usage.completion_tokens === 'number' ? usage.completion_tokens : 0,\n    cachedInputTokens: typeof promptDetails.cached_tokens === 'number' ? promptDetails.cached_tokens : undefined,\n    reasoningTokens: typeof completionDetails.reasoning_tokens === 'number' ? completionDetails.reasoning_tokens : undefined,\n    responseStatusCode: meta.responseStatusCode ?? 200,\n  };\n}\n\nexport function extractAnthropicUsage(response: unknown, meta: UsageMeta = {}): UsageTapEndUsage {\n  const usage = isObject(response) && isObject(response.usage) ? response.usage : {};\n\n  return {\n    modelUsed: meta.modelUsed,\n    inputTokens: typeof usage.input_tokens === 'number' ? usage.input_tokens : 0,\n    responseTokens: typeof usage.output_tokens === 'number' ? usage.output_tokens : 0,\n    cachedInputTokens: typeof usage.cache_read_input_tokens === 'number' ? usage.cache_read_input_tokens : undefined,\n    responseStatusCode: meta.responseStatusCode ?? 200,\n  };\n}\n\nexport function extractGeminiUsage(response: unknown, meta: UsageMeta = {}): UsageTapEndUsage {\n  const usage = isObject(response) && isObject(response.usageMetadata) ? response.usageMetadata : {};\n\n  return {\n    modelUsed: meta.modelUsed,\n    inputTokens: typeof usage.promptTokenCount === 'number' ? usage.promptTokenCount : 0,\n    responseTokens: typeof usage.candidatesTokenCount === 'number' ? usage.candidatesTokenCount : 0,\n    cachedInputTokens: typeof usage.cachedContentTokenCount === 'number' ? usage.cachedContentTokenCount : undefined,\n    reasoningTokens: typeof usage.thoughtsTokenCount === 'number' ? usage.thoughtsTokenCount : undefined,\n    responseStatusCode: meta.responseStatusCode ?? 200,\n  };\n}\n\nasync function runProviderWithUsageTap<TResult>(\n  client: UsageTapClient,\n  options: UsageTapProviderRunOptions<TResult>,\n  defaultExtractor: (response: TResult, meta: UsageMeta) => UsageTapEndUsage,\n): Promise<UsageTapProviderRunResult<TResult>> {\n  let requestUsed = cloneRequest(options.request);\n\n  const lifecycle = await withUsageTapCall(client, {\n    begin: options.begin,\n    signal: options.signal,\n    onError: options.onError ?? defaultUsageTapErrorMapper,\n    invoke: async ({ begin, setUsage }) => {\n      requestUsed = options.entitlementMode === 'apply'\n        ? applyUsageTapEntitlements(options.request, begin, options)\n        : cloneRequest(options.request);\n\n      const invoked = await options.invoke(requestUsed);\n\n      if (isInvokeResult<TResult>(invoked)) {\n        return invoked;\n      }\n\n      const response = invoked as TResult;\n      const usage = (options.extractUsage ?? defaultExtractor)(response, {\n        modelUsed: requestUsed.model,\n        responseStatusCode: 200,\n      });\n      setUsage(usage);\n      return response;\n    },\n  });\n\n  return {\n    response: lifecycle.result,\n    begin: lifecycle.begin,\n    end: lifecycle.end,\n    requestUsed,\n    effectiveUsage: lifecycle.effectiveUsage,\n    allowed: lifecycle.begin.data.allowed ?? {},\n  };\n}\n\nexport function runOpenAIWithUsageTap<TResult>(client: UsageTapClient, options: UsageTapProviderRunOptions<TResult>) {\n  return runProviderWithUsageTap(client, options, extractOpenAIUsage);\n}\n\nexport function runOpenRouterWithUsageTap<TResult>(client: UsageTapClient, options: UsageTapProviderRunOptions<TResult>) {\n  return runProviderWithUsageTap(client, options, extractOpenAIUsage);\n}\n\nexport function runAnthropicWithUsageTap<TResult>(client: UsageTapClient, options: UsageTapProviderRunOptions<TResult>) {\n  return runProviderWithUsageTap(client, options, extractAnthropicUsage);\n}\n\nexport function runGeminiWithUsageTap<TResult>(client: UsageTapClient, options: UsageTapProviderRunOptions<TResult>) {\n  return runProviderWithUsageTap(client, options, extractGeminiUsage);\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACUA,IAAM,mBAAmB;AACzB,IAAM,gBAAgB;AAEtB,SAAS,wBAAwB,QAAgB,MAA8D;AAC7G,QAAM,QAAQ,IAAI,MAAM,uCAAuC,MAAM,EAAE;AAIvE,QAAM,aAAa;AACnB,QAAM,OAAO;AACb,SAAO;AACT;AAEO,SAAS,qBAAqB,QAA8C;AACjF,QAAM,YAAY,OAAO,SAAS,WAAW;AAE7C,MAAI,CAAC,WAAW;AACd,UAAM,IAAI,MAAM,qFAAqF;AAAA,EACvG;AAEA,QAAM,WAAW,OAAO,WAAW,kBAAkB,QAAQ,OAAO,EAAE;AAEtE,iBAAe,QAAW,MAAc,MAAe,OAA4B,CAAC,GAAe;AACjG,UAAM,WAAW,MAAM,UAAU,GAAG,OAAO,GAAG,IAAI,IAAI;AAAA,MACpD,QAAQ;AAAA,MACR,SAAS;AAAA,QACP,eAAe,UAAU,OAAO,MAAM;AAAA,QACtC,QAAQ;AAAA,QACR,gBAAgB;AAAA,MAClB;AAAA,MACA,MAAM,KAAK,UAAU,IAAI;AAAA,MACzB,QAAQ,KAAK;AAAA,IACf,CAAC;AAED,UAAM,OAAO,MAAM,SAAS,KAAK;AACjC,UAAM,SAAS,OAAO,KAAK,MAAM,IAAI,IAAe,CAAC;AAErD,QAAI,CAAC,SAAS,IAAI;AAChB,YAAM,wBAAwB,SAAS,QAAQ,MAAM;AAAA,IACvD;AAEA,WAAO;AAAA,EACT;AAEA,SAAO;AAAA,IACL;AAAA,IACA,UAAU,OAA6B,MAA4D;AACjG,aAAO,QAA+B,eAAe,OAAO,IAAI;AAAA,IAClE;AAAA,IACA,QAAQ,KAAyB,MAA0D;AACzF,aAAO,QAA6B,aAAa,KAAK,IAAI;AAAA,IAC5D;AAAA,EACF;AACF;;;ACpDA,SAAS,SAAS,OAAkD;AAClE,SAAO,OAAO,UAAU,YAAY,UAAU;AAChD;AAEA,SAAS,eAAwB,OAAwF;AACvH,SAAO,SAAS,KAAK,KAAK,YAAY,SAAS,WAAW;AAC5D;AAEA,SAAS,mBAAmB,OAAoC;AAC9D,MAAI,CAAC,SAAS,KAAK,GAAG;AACpB,WAAO;AAAA,EACT;AAEA,MAAI,OAAO,MAAM,eAAe,UAAU;AACxC,WAAO,MAAM;AAAA,EACf;AAEA,MAAI,OAAO,MAAM,WAAW,UAAU;AACpC,WAAO,MAAM;AAAA,EACf;AAEA,MAAI,SAAS,MAAM,QAAQ,KAAK,OAAO,MAAM,SAAS,WAAW,UAAU;AACzE,WAAO,MAAM,SAAS;AAAA,EACxB;AAEA,SAAO;AACT;AAEA,SAAS,oBAAoB,aAAsB,UAA0B;AAC3E,MAAI,uBAAuB,OAAO;AAChC,QAAI,YAAY,UAAU,QAAW;AACnC,aAAO,eAAe,aAAa,SAAS;AAAA,QAC1C,OAAO;AAAA,QACP,cAAc;AAAA,QACd,UAAU;AAAA,MACZ,CAAC;AAAA,IACH;AAEA,UAAM;AAAA,EACR;AAEA,QAAM,IAAI,eAAe,CAAC,aAAa,QAAQ,GAAG,uDAAuD;AAC3G;AAEO,SAAS,2BAA2B,OAAsC;AAC/E,SAAO;AAAA,IACL,MAAM;AAAA,IACN,SAAS,OAAO,KAAK;AAAA,EACvB;AACF;AAEO,SAAS,kBAAkB,QAAwB,OAAmD,MAA4B;AACvI,SAAO,OAAO,UAAU,OAAO,IAAI;AACrC;AAEO,SAAS,gBAAgB,QAAwB,KAAyB,MAA4B;AAC3G,SAAO,OAAO,QAAQ,KAAK,IAAI;AACjC;AAEA,eAAsB,iBACpB,QACA,SACsC;AACtC,QAAM,QAAQ,MAAM,kBAAkB,QAAQ,QAAQ,OAAO,EAAE,QAAQ,QAAQ,OAAO,CAAC;AACvF,QAAM,UAAU,QAAQ,WAAW;AAEnC,MAAI;AACJ,MAAI;AACJ,MAAI;AAEJ,QAAM,WAAW,CAAC,cAAsC;AACtD,YAAQ,EAAE,GAAG,OAAO,GAAG,UAAU;AAAA,EACnC;AAEA,MAAI;AACF,UAAM,UAAU,MAAM,QAAQ,OAAO;AAAA,MACnC;AAAA,MACA;AAAA,MACA,QAAQ,QAAQ;AAAA,IAClB,CAAC;AAED,QAAI,eAAe,OAAO,GAAG;AAC3B,eAAS,QAAQ;AACjB,eAAS,QAAQ,KAAK;AAAA,IACxB,OAAO;AACL,eAAS;AAAA,IACX;AAAA,EACF,SAAS,OAAO;AACd,kBAAc;AAAA,EAChB;AAEA,QAAM,iBAAmC;AAAA,IACvC,GAAG;AAAA,EACL;AAEA,MAAI,aAAa;AACf,mBAAe,QAAQ,eAAe,SAAS,QAAQ,WAAW;AAClE,mBAAe,qBAAqB,eAAe,sBAAsB,mBAAmB,WAAW;AAAA,EACzG,OAAO;AACL,mBAAe,qBAAqB,eAAe,sBAAsB;AAAA,EAC3E;AAEA,MAAI;AACJ,MAAI;AACF,UAAM,MAAM,gBAAgB,QAAQ;AAAA,MAClC,QAAQ,MAAM,KAAK;AAAA,MACnB,GAAG;AAAA,IACL,GAAG,EAAE,QAAQ,QAAQ,OAAO,CAAC;AAAA,EAC/B,SAAS,UAAU;AACjB,QAAI,aAAa;AACf,0BAAoB,aAAa,QAAQ;AAAA,IAC3C;AACA,UAAM;AAAA,EACR;AAEA,MAAI,aAAa;AACf,UAAM;AAAA,EACR;AAEA,SAAO;AAAA,IACL;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF;AACF;;;AC1HA,SAAS,aAAa,SAA2C;AAC/D,SAAO,gBAAgB,OAAO;AAChC;AAEA,SAASA,UAAS,OAAkD;AAClE,SAAO,OAAO,UAAU,YAAY,UAAU;AAChD;AAEA,SAAS,cAAc,OAAyD;AAC9E,SAAO,MAAM,QAAQ,KAAK,KAAK,MAAM,MAAM,CAAC,SAASA,UAAS,IAAI,CAAC;AACrE;AAEA,SAASC,gBAAwB,OAAuE;AACtG,SAAOD,UAAS,KAAK,KAAK,YAAY,SAAS,WAAW;AAC5D;AAEA,SAAS,kBAAkB,SAA0B,YAAgD;AACnG,SAAO,QAAQ,UAAU,MAAM;AACjC;AAEA,SAAS,oBAAoB,OAAmD;AAC9E,UAAQ,OAAO;AAAA,IACb,KAAK;AACH,aAAO;AAAA,IACT,KAAK;AACH,aAAO;AAAA,IACT,KAAK;AACH,aAAO;AAAA,IACT;AACE,aAAO;AAAA,EACX;AACF;AAEA,SAAS,mBAAmB,MAA+B,SAAgC;AACzF,MAAI,OAAO,KAAK,qBAAqB,UAAU;AAC7C;AAAA,EACF;AAEA,QAAM,UAAU,OAAO,KAAK,gBAAgB,EAAE,YAAY;AAC1D,QAAM,eAAe,QAAQ,kBAAkB;AAE/C,MAAI,iBAAiB,QAAQ;AAC3B,WAAO,KAAK;AACZ;AAAA,EACF;AAEA,MAAI,oBAAoB,OAAO,IAAI,oBAAoB,YAAY,GAAG;AACpE,SAAK,mBAAmB,aAAa,YAAY;AAAA,EACnD;AACF;AAEA,SAAS,mBAAmB,MAA+B,SAAgC;AACzF,MAAI,QAAQ,mBAAmB,QAAQ;AACrC,WAAO,KAAK;AACZ;AAAA,EACF;AAEA,QAAM,UAAmE;AAAA,IACvE,KAAK;AAAA,IACL,QAAQ;AAAA,IACR,MAAM;AAAA,EACR;AAEA,QAAM,gBAAgB,QAAQ,iBAAiB,QAAQ,QAAQ,cAAyD,IAAI;AAE5H,MAAI,kBAAkB,QAAW;AAC/B;AAAA,EACF;AAEA,MAAI,CAACA,UAAS,KAAK,cAAc,GAAG;AAClC,SAAK,iBAAiB,EAAE,gBAAgB,cAAc;AACtD;AAAA,EACF;AAEA,QAAM,iBAAiB,KAAK;AAE5B,MAAI,OAAO,eAAe,mBAAmB,UAAU;AACrD,mBAAe,iBAAiB,KAAK,IAAI,eAAe,gBAAgB,aAAa;AACrF;AAAA,EACF;AAEA,iBAAe,iBAAiB;AAClC;AAEA,SAAS,eAAe,SAA0B,SAA0B,SAA2C;AACrH,QAAM,YAAY,QAAQ,WAAW,QAAQ,YAAY,UACrD,QAAQ,WAAW,UACnB,QAAQ,YAAY,QAAQ,YAAY,WACtC,QAAQ,WAAW,WACnB;AAEN,MAAI,CAAC,WAAW;AACd;AAAA,EACF;AAEA,UAAQ,QAAQ;AAChB,MAAIA,UAAS,QAAQ,IAAI,KAAK,WAAW,QAAQ,MAAM;AACrD,YAAQ,KAAK,QAAQ;AAAA,EACvB;AACF;AAEA,SAAS,kBAAkB,OAAuC,SAA0B,SAAqE;AAC/J,SAAO,MAAM,OAAO,CAAC,SAAS;AAC5B,QAAI,KAAK,SAAS,gBAAgB,QAAQ,WAAW,OAAO;AAC1D,aAAO;AAAA,IACT;AAEA,UAAM,cAAcA,UAAS,KAAK,QAAQ,IAAI,KAAK,WAAW;AAC9D,UAAM,OAAO,OAAO,aAAa,SAAS,WACtC,YAAY,OACZ,OAAO,KAAK,SAAS,WACnB,KAAK,OACL,OAAO,KAAK,SAAS,WACnB,KAAK,OACL;AAER,QAAI,CAAC,MAAM;AACT,aAAO;AAAA,IACT;AAEA,UAAM,aAAa,QAAQ,mBAAmB,IAAI;AAClD,WAAO,aAAa,kBAAkB,SAAS,UAAU,IAAI;AAAA,EAC/D,CAAC;AACH;AAEA,SAAS,qBAAqB,OAAuC,SAA0B,SAAqE;AAClK,SAAO,MAAM,OAAO,CAAC,SAAS;AAC5B,UAAM,OAAO,OAAO,KAAK,SAAS,WAAW,KAAK,OAAO;AACzD,QAAI,CAAC,MAAM;AACT,aAAO;AAAA,IACT;AACA,UAAM,aAAa,QAAQ,mBAAmB,IAAI;AAClD,WAAO,aAAa,kBAAkB,SAAS,UAAU,IAAI;AAAA,EAC/D,CAAC;AACH;AAEA,SAAS,kBAAkB,OAAuC,SAA0B,SAAqE;AAC/J,SAAO,MACJ,IAAI,CAAC,SAAS;AACb,UAAM,eAAe,MAAM,QAAQ,KAAK,oBAAoB,IACxD,KAAK,qBAAqB,OAAO,CAAC,gBAAgB;AAClD,UAAI,CAACA,UAAS,WAAW,KAAK,OAAO,YAAY,SAAS,UAAU;AAClE,eAAO;AAAA,MACT;AACA,YAAM,aAAa,QAAQ,mBAAmB,YAAY,IAAI;AAC9D,aAAO,aAAa,kBAAkB,SAAS,UAAU,IAAI;AAAA,IAC/D,CAAC,IACC,KAAK;AAET,WAAO;AAAA,MACL,GAAG;AAAA,MACH,sBAAsB;AAAA,IACxB;AAAA,EACF,CAAC,EACA,OAAO,CAAC,SAAS;AAChB,QAAI,CAAC,MAAM,QAAQ,KAAK,oBAAoB,GAAG;AAC7C,aAAO;AAAA,IACT;AACA,WAAO,KAAK,qBAAqB,SAAS;AAAA,EAC5C,CAAC;AACL;AAEO,SAAS,0BACd,SACA,OACA,UAAsC,CAAC,GACtB;AACjB,QAAM,cAAc,aAAa,OAAO;AACxC,QAAM,UAAU,MAAM,KAAK,WAAW,CAAC;AAEvC,iBAAe,aAAa,SAAS,OAAO;AAE5C,MAAI,CAACA,UAAS,YAAY,IAAI,GAAG;AAC/B,WAAO;AAAA,EACT;AAEA,MAAI,YAAY,aAAa,YAAY,YAAY,aAAa,cAAc;AAC9E,uBAAmB,YAAY,MAAM,OAAO;AAC5C,QAAI,cAAc,YAAY,KAAK,KAAK,GAAG;AACzC,kBAAY,KAAK,QAAQ,kBAAkB,YAAY,KAAK,OAAO,SAAS,OAAO;AAAA,IACrF;AACA,WAAO;AAAA,EACT;AAEA,MAAI,YAAY,aAAa,UAAU;AACrC,uBAAmB,YAAY,MAAM,OAAO;AAC5C,QAAI,cAAc,YAAY,KAAK,KAAK,GAAG;AACzC,kBAAY,KAAK,QAAQ,kBAAkB,YAAY,KAAK,OAAO,SAAS,OAAO;AAAA,IACrF;AACA,WAAO;AAAA,EACT;AAEA,MAAI,YAAY,aAAa,eAAe,cAAc,YAAY,KAAK,KAAK,GAAG;AACjF,gBAAY,KAAK,QAAQ,qBAAqB,YAAY,KAAK,OAAO,SAAS,OAAO;AAAA,EACxF;AAEA,SAAO;AACT;AAOO,SAAS,mBAAmB,UAAmB,OAAkB,CAAC,GAAqB;AAC5F,QAAM,QAAQA,UAAS,QAAQ,KAAKA,UAAS,SAAS,KAAK,IAAI,SAAS,QAAQ,CAAC;AACjF,QAAM,gBAAgBA,UAAS,MAAM,qBAAqB,IAAI,MAAM,wBAAwB,CAAC;AAC7F,QAAM,oBAAoBA,UAAS,MAAM,yBAAyB,IAAI,MAAM,4BAA4B,CAAC;AAEzG,SAAO;AAAA,IACL,WAAW,KAAK;AAAA,IAChB,aAAa,OAAO,MAAM,kBAAkB,WAAW,MAAM,gBAAgB;AAAA,IAC7E,gBAAgB,OAAO,MAAM,sBAAsB,WAAW,MAAM,oBAAoB;AAAA,IACxF,mBAAmB,OAAO,cAAc,kBAAkB,WAAW,cAAc,gBAAgB;AAAA,IACnG,iBAAiB,OAAO,kBAAkB,qBAAqB,WAAW,kBAAkB,mBAAmB;AAAA,IAC/G,oBAAoB,KAAK,sBAAsB;AAAA,EACjD;AACF;AAEO,SAAS,sBAAsB,UAAmB,OAAkB,CAAC,GAAqB;AAC/F,QAAM,QAAQA,UAAS,QAAQ,KAAKA,UAAS,SAAS,KAAK,IAAI,SAAS,QAAQ,CAAC;AAEjF,SAAO;AAAA,IACL,WAAW,KAAK;AAAA,IAChB,aAAa,OAAO,MAAM,iBAAiB,WAAW,MAAM,eAAe;AAAA,IAC3E,gBAAgB,OAAO,MAAM,kBAAkB,WAAW,MAAM,gBAAgB;AAAA,IAChF,mBAAmB,OAAO,MAAM,4BAA4B,WAAW,MAAM,0BAA0B;AAAA,IACvG,oBAAoB,KAAK,sBAAsB;AAAA,EACjD;AACF;AAEO,SAAS,mBAAmB,UAAmB,OAAkB,CAAC,GAAqB;AAC5F,QAAM,QAAQA,UAAS,QAAQ,KAAKA,UAAS,SAAS,aAAa,IAAI,SAAS,gBAAgB,CAAC;AAEjG,SAAO;AAAA,IACL,WAAW,KAAK;AAAA,IAChB,aAAa,OAAO,MAAM,qBAAqB,WAAW,MAAM,mBAAmB;AAAA,IACnF,gBAAgB,OAAO,MAAM,yBAAyB,WAAW,MAAM,uBAAuB;AAAA,IAC9F,mBAAmB,OAAO,MAAM,4BAA4B,WAAW,MAAM,0BAA0B;AAAA,IACvG,iBAAiB,OAAO,MAAM,uBAAuB,WAAW,MAAM,qBAAqB;AAAA,IAC3F,oBAAoB,KAAK,sBAAsB;AAAA,EACjD;AACF;AAEA,eAAe,wBACb,QACA,SACA,kBAC6C;AAC7C,MAAI,cAAc,aAAa,QAAQ,OAAO;AAE9C,QAAM,YAAY,MAAM,iBAAiB,QAAQ;AAAA,IAC/C,OAAO,QAAQ;AAAA,IACf,QAAQ,QAAQ;AAAA,IAChB,SAAS,QAAQ,WAAW;AAAA,IAC5B,QAAQ,OAAO,EAAE,OAAO,SAAS,MAAM;AACrC,oBAAc,QAAQ,oBAAoB,UACtC,0BAA0B,QAAQ,SAAS,OAAO,OAAO,IACzD,aAAa,QAAQ,OAAO;AAEhC,YAAM,UAAU,MAAM,QAAQ,OAAO,WAAW;AAEhD,UAAIC,gBAAwB,OAAO,GAAG;AACpC,eAAO;AAAA,MACT;AAEA,YAAM,WAAW;AACjB,YAAM,SAAS,QAAQ,gBAAgB,kBAAkB,UAAU;AAAA,QACjE,WAAW,YAAY;AAAA,QACvB,oBAAoB;AAAA,MACtB,CAAC;AACD,eAAS,KAAK;AACd,aAAO;AAAA,IACT;AAAA,EACF,CAAC;AAED,SAAO;AAAA,IACL,UAAU,UAAU;AAAA,IACpB,OAAO,UAAU;AAAA,IACjB,KAAK,UAAU;AAAA,IACf;AAAA,IACA,gBAAgB,UAAU;AAAA,IAC1B,SAAS,UAAU,MAAM,KAAK,WAAW,CAAC;AAAA,EAC5C;AACF;AAEO,SAAS,sBAA+B,QAAwB,SAA8C;AACnH,SAAO,wBAAwB,QAAQ,SAAS,kBAAkB;AACpE;AAEO,SAAS,0BAAmC,QAAwB,SAA8C;AACvH,SAAO,wBAAwB,QAAQ,SAAS,kBAAkB;AACpE;AAEO,SAAS,yBAAkC,QAAwB,SAA8C;AACtH,SAAO,wBAAwB,QAAQ,SAAS,qBAAqB;AACvE;AAEO,SAAS,sBAA+B,QAAwB,SAA8C;AACnH,SAAO,wBAAwB,QAAQ,SAAS,kBAAkB;AACpE;","names":["isObject","isInvokeResult"]}

package/dist/usagetap/index.d.cts

@@ -0,0 +1,147 @@
+import { a as ProviderRequest } from '../types-1zDDD-Ij.cjs';
+import '../schema-DIOA4OiO.cjs';
+import 'zod';
+
+type UsageTapReasoningLevel = 'NONE' | 'LOW' | 'MEDIUM' | 'HIGH';
+type UsageTapEntitlementMode = 'off' | 'apply';
+type UsageTapAllowedCapability = 'standard' | 'premium' | 'audio' | 'image' | 'search';
+interface UsageTapAllowed {
+    standard?: boolean;
+    premium?: boolean;
+    audio?: boolean;
+    image?: boolean;
+    search?: boolean;
+    reasoningLevel?: UsageTapReasoningLevel;
+}
+interface UsageTapBeginRequest {
+    customerId: string;
+    feature?: string;
+    tags?: string[];
+    customerName?: string;
+    customerEmail?: string;
+    stripeCustomerId?: string;
+    requested?: Record<string, boolean | string>;
+    idempotencyKey?: string;
+    batch?: boolean;
+    pricingMode?: 'batch' | 'standard';
+}
+interface UsageTapBeginResponse {
+    data: {
+        callId: string;
+        allowed: UsageTapAllowed;
+        entitlementHints?: Record<string, unknown>;
+        subscription?: Record<string, unknown>;
+        plan?: Record<string, unknown>;
+        balances?: Record<string, unknown>;
+        stripeCustomerId?: string | null;
+        [key: string]: unknown;
+    };
+    [key: string]: unknown;
+}
+interface UsageTapErrorPayload {
+    code: string;
+    message: string;
+}
+interface UsageTapEndUsage {
+    modelUsed?: string;
+    inputTokens?: number;
+    cachedInputTokens?: number;
+    responseTokens?: number;
+    reasoningTokens?: number;
+    searches?: number;
+    audio?: number;
+    audioSeconds?: number;
+    isPremium?: boolean;
+    stripeCustomerId?: string;
+    responseStatusCode?: number;
+    batch?: boolean;
+    pricingMode?: 'batch' | 'standard';
+    error?: UsageTapErrorPayload;
+}
+interface UsageTapEndRequest extends UsageTapEndUsage {
+    callId: string;
+}
+interface UsageTapEndResponse {
+    data?: Record<string, unknown>;
+    [key: string]: unknown;
+}
+interface UsageTapClientConfig {
+    apiKey: string;
+    baseUrl?: string;
+    fetch?: typeof globalThis.fetch;
+}
+interface UsageTapRequestInit {
+    signal?: AbortSignal;
+}
+interface UsageTapClient {
+    request<T>(path: string, body: unknown, init?: UsageTapRequestInit): Promise<T>;
+    beginCall(begin: UsageTapBeginRequest, init?: UsageTapRequestInit): Promise<UsageTapBeginResponse>;
+    endCall(end: UsageTapEndRequest, init?: UsageTapRequestInit): Promise<UsageTapEndResponse>;
+}
+interface UsageTapInvokeContext {
+    begin: UsageTapBeginResponse;
+    setUsage: (usage: UsageTapEndUsage) => void;
+    signal?: AbortSignal;
+}
+interface UsageTapInvokeResult<TResult> {
+    result: TResult;
+    usage: UsageTapEndUsage;
+}
+interface UsageTapCallOptions<TResult> {
+    begin: UsageTapBeginRequest;
+    invoke: (context: UsageTapInvokeContext) => Promise<TResult | UsageTapInvokeResult<TResult>>;
+    onError?: (error: unknown) => UsageTapErrorPayload;
+    signal?: AbortSignal;
+}
+interface UsageTapCallResult<TResult> {
+    result: TResult;
+    begin: UsageTapBeginResponse;
+    end: UsageTapEndResponse;
+    effectiveUsage: UsageTapEndUsage;
+}
+interface UsageTapEntitlementOptions {
+    modelTiers?: {
+        standard?: string;
+        premium?: string;
+    };
+    toolEntitlements?: Record<string, UsageTapAllowedCapability>;
+}
+interface UsageTapProviderRunOptions<TResult> extends UsageTapEntitlementOptions {
+    request: ProviderRequest;
+    begin: UsageTapBeginRequest;
+    invoke: (request: ProviderRequest) => Promise<TResult | UsageTapInvokeResult<TResult>>;
+    extractUsage?: (response: TResult) => UsageTapEndUsage;
+    entitlementMode?: UsageTapEntitlementMode;
+    signal?: AbortSignal;
+    onError?: (error: unknown) => UsageTapErrorPayload;
+}
+interface UsageTapProviderRunResult<TResult> {
+    response: TResult;
+    begin: UsageTapBeginResponse;
+    end: UsageTapEndResponse;
+    requestUsed: ProviderRequest;
+    effectiveUsage: UsageTapEndUsage;
+    allowed: UsageTapAllowed;
+}
+
+declare function createUsageTapClient(config: UsageTapClientConfig): UsageTapClient;
+
+declare function defaultUsageTapErrorMapper(error: unknown): UsageTapErrorPayload;
+declare function beginUsageTapCall(client: UsageTapClient, begin: Parameters<UsageTapClient['beginCall']>[0], init?: UsageTapRequestInit): Promise<UsageTapBeginResponse>;
+declare function endUsageTapCall(client: UsageTapClient, end: UsageTapEndRequest, init?: UsageTapRequestInit): Promise<UsageTapEndResponse>;
+declare function withUsageTapCall<TResult>(client: UsageTapClient, options: UsageTapCallOptions<TResult>): Promise<UsageTapCallResult<TResult>>;
+
+declare function applyUsageTapEntitlements(request: ProviderRequest, begin: UsageTapBeginResponse, options?: UsageTapEntitlementOptions): ProviderRequest;
+interface UsageMeta {
+    modelUsed?: string;
+    responseStatusCode?: number;
+}
+declare function extractOpenAIUsage(response: unknown, meta?: UsageMeta): UsageTapEndUsage;
+declare function extractAnthropicUsage(response: unknown, meta?: UsageMeta): UsageTapEndUsage;
+declare function extractGeminiUsage(response: unknown, meta?: UsageMeta): UsageTapEndUsage;
+declare function runOpenAIWithUsageTap<TResult>(client: UsageTapClient, options: UsageTapProviderRunOptions<TResult>): Promise<UsageTapProviderRunResult<TResult>>;
+declare function runOpenRouterWithUsageTap<TResult>(client: UsageTapClient, options: UsageTapProviderRunOptions<TResult>): Promise<UsageTapProviderRunResult<TResult>>;
+declare function runAnthropicWithUsageTap<TResult>(client: UsageTapClient, options: UsageTapProviderRunOptions<TResult>): Promise<UsageTapProviderRunResult<TResult>>;
+declare function runGeminiWithUsageTap<TResult>(client: UsageTapClient, options: UsageTapProviderRunOptions<TResult>): Promise<UsageTapProviderRunResult<TResult>>;
+
+export { type UsageTapAllowed, type UsageTapAllowedCapability, type UsageTapBeginRequest, type UsageTapBeginResponse, type UsageTapCallOptions, type UsageTapCallResult, type UsageTapClient, type UsageTapClientConfig, type UsageTapEndRequest, type UsageTapEndResponse, type UsageTapEndUsage, type UsageTapEntitlementMode, type UsageTapEntitlementOptions, type UsageTapErrorPayload, type UsageTapInvokeContext, type UsageTapInvokeResult, type UsageTapProviderRunOptions, type UsageTapProviderRunResult, type UsageTapReasoningLevel, applyUsageTapEntitlements, beginUsageTapCall, createUsageTapClient, defaultUsageTapErrorMapper, endUsageTapCall, extractAnthropicUsage, extractGeminiUsage, extractOpenAIUsage, runAnthropicWithUsageTap, runGeminiWithUsageTap, runOpenAIWithUsageTap, runOpenRouterWithUsageTap, withUsageTapCall };