@usagetap/sdk 1.1.0 → 1.2.0

package/dist/adapters/openai.d.cts

@@ -1,5 +1,5 @@
 import OpenAI from 'openai';
-import { B as BeginCallRequest, P as PromptCompressionProvider, U as UsageTapClient, W as WithUsageOptions, a as PromptCompressionTelemetry, V as VendorHints, b as UsageTapSuccessResponse, c as BeginCallResponseBody, E as EndCallRequest } from '../client-BA-QlnRq.cjs';
+import { B as BeginCallRequest, P as PromptCompressionProvider, U as UsageTapClient, W as WithUsageOptions, a as PromptCompressionTelemetry, V as VendorHints, b as UsageTapSuccessResponse, c as BeginCallResponseBody, E as EndCallRequest } from '../client-EMXt9_fA.cjs';
 
 interface OpenAIAdapterInit {
     client: OpenAI;
@@ -46,7 +46,11 @@ interface OpenAIPromptCompressionRoleOptions {
     enabled?: boolean;
     provider?: PromptCompressionProvider;
     minTokens?: number;
+    aggressiveness?: number;
+    /** @deprecated Use aggressiveness instead. */
     tokenCompanyAggressiveness?: number;
+    /** @deprecated Use aggressiveness instead. */
+    usageTapCompressionAggressiveness?: number;
 }
 type OpenAIPromptCompressionRoleSetting = boolean | OpenAIPromptCompressionRoleOptions;
 interface OpenAIPromptCompressionOptions {
@@ -56,8 +60,13 @@ interface OpenAIPromptCompressionOptions {
     minTokens?: number;
     failOpen?: boolean;
     tokenCompanyModel?: string;
+    aggressiveness?: number | Partial<Record<OpenAIPromptCompressionRole, number>>;
+    /** @deprecated Use aggressiveness instead. */
     tokenCompanyAggressiveness?: number | Partial<Record<OpenAIPromptCompressionRole, number>>;
     tokenCompanyAppId?: string;
+    usageTapCompressionModel?: string;
+    /** @deprecated Use aggressiveness instead. */
+    usageTapCompressionAggressiveness?: number | Partial<Record<OpenAIPromptCompressionRole, number>>;
 }
 interface OpenAIPromptCompressionTurnStats extends PromptCompressionTelemetry {
     callId: string;

package/dist/adapters/openai.d.ts

@@ -1,5 +1,5 @@
 import OpenAI from 'openai';
-import { B as BeginCallRequest, P as PromptCompressionProvider, U as UsageTapClient, W as WithUsageOptions, a as PromptCompressionTelemetry, V as VendorHints, b as UsageTapSuccessResponse, c as BeginCallResponseBody, E as EndCallRequest } from '../client-BA-QlnRq.js';
+import { B as BeginCallRequest, P as PromptCompressionProvider, U as UsageTapClient, W as WithUsageOptions, a as PromptCompressionTelemetry, V as VendorHints, b as UsageTapSuccessResponse, c as BeginCallResponseBody, E as EndCallRequest } from '../client-EMXt9_fA.js';
 
 interface OpenAIAdapterInit {
     client: OpenAI;
@@ -46,7 +46,11 @@ interface OpenAIPromptCompressionRoleOptions {
     enabled?: boolean;
     provider?: PromptCompressionProvider;
     minTokens?: number;
+    aggressiveness?: number;
+    /** @deprecated Use aggressiveness instead. */
     tokenCompanyAggressiveness?: number;
+    /** @deprecated Use aggressiveness instead. */
+    usageTapCompressionAggressiveness?: number;
 }
 type OpenAIPromptCompressionRoleSetting = boolean | OpenAIPromptCompressionRoleOptions;
 interface OpenAIPromptCompressionOptions {
@@ -56,8 +60,13 @@ interface OpenAIPromptCompressionOptions {
     minTokens?: number;
     failOpen?: boolean;
     tokenCompanyModel?: string;
+    aggressiveness?: number | Partial<Record<OpenAIPromptCompressionRole, number>>;
+    /** @deprecated Use aggressiveness instead. */
     tokenCompanyAggressiveness?: number | Partial<Record<OpenAIPromptCompressionRole, number>>;
     tokenCompanyAppId?: string;
+    usageTapCompressionModel?: string;
+    /** @deprecated Use aggressiveness instead. */
+    usageTapCompressionAggressiveness?: number | Partial<Record<OpenAIPromptCompressionRole, number>>;
 }
 interface OpenAIPromptCompressionTurnStats extends PromptCompressionTelemetry {
     callId: string;

package/dist/adapters/openai.mjs

@@ -603,6 +603,18 @@ async function compressChatParams(params, usageTap, compression, signal) {
   if (!messages) {
     return { params, segments: [] };
   }
+  if (shouldUseUsageTapMessageEndpoint(compression)) {
+    const result = await usageTap.compressPromptMessages(source, {
+      provider: "usagetap",
+      failOpen: compression.failOpen,
+      aggressiveness: resolveMessageEndpointAggressiveness(compression),
+      signal
+    });
+    return {
+      params: result.compressedInput,
+      segments: [{ role: "user", result }]
+    };
+  }
   const messageResults = await Promise.all(
     messages.map(
       (message) => compressOpenAIMessage(message, usageTap, compression, signal)
@@ -826,8 +838,11 @@ async function compressTextForRole(text, role, usageTap, compression, signal) {
     provider: roleOptions.provider,
     failOpen: roleOptions.failOpen,
     tokenCompanyModel: roleOptions.tokenCompanyModel,
+    aggressiveness: roleOptions.aggressiveness,
     tokenCompanyAggressiveness: roleOptions.tokenCompanyAggressiveness,
     tokenCompanyAppId: roleOptions.tokenCompanyAppId,
+    usageTapCompressionModel: roleOptions.usageTapCompressionModel,
+    usageTapCompressionAggressiveness: roleOptions.usageTapCompressionAggressiveness,
     signal
   });
   const compressedText = typeof result.compressedInput === "string" ? result.compressedInput : String(result.compressedInput);
@@ -886,16 +901,67 @@ function resolveRoleCompressionOptions(compression, role) {
     minTokens: roleOptions?.minTokens ?? compression.minTokens,
     failOpen: compression.failOpen,
     tokenCompanyModel: compression.tokenCompanyModel,
+    aggressiveness: roleOptions?.aggressiveness ?? resolveAggressiveness(compression, role),
     tokenCompanyAggressiveness: roleOptions?.tokenCompanyAggressiveness ?? resolveTokenCompanyAggressiveness(compression, role),
-    tokenCompanyAppId: compression.tokenCompanyAppId
+    tokenCompanyAppId: compression.tokenCompanyAppId,
+    usageTapCompressionModel: compression.usageTapCompressionModel,
+    usageTapCompressionAggressiveness: roleOptions?.usageTapCompressionAggressiveness ?? resolveUsageTapCompressionAggressiveness(compression, role)
   };
 }
+function resolveAggressiveness(compression, role) {
+  if (typeof compression.aggressiveness === "number") {
+    return compression.aggressiveness;
+  }
+  return compression.aggressiveness?.[role];
+}
 function resolveTokenCompanyAggressiveness(compression, role) {
   if (typeof compression.tokenCompanyAggressiveness === "number") {
     return compression.tokenCompanyAggressiveness;
   }
   return compression.tokenCompanyAggressiveness?.[role];
 }
+function resolveUsageTapCompressionAggressiveness(compression, role) {
+  if (typeof compression.usageTapCompressionAggressiveness === "number") {
+    return compression.usageTapCompressionAggressiveness;
+  }
+  return compression.usageTapCompressionAggressiveness?.[role];
+}
+function shouldUseUsageTapMessageEndpoint(compression) {
+  if (compression.provider !== "usagetap") {
+    return false;
+  }
+  return Object.values(compression.roles ?? {}).every((setting) => {
+    if (typeof setting !== "object" || setting === null) {
+      return true;
+    }
+    return setting.provider === void 0 || setting.provider === "usagetap";
+  });
+}
+function resolveMessageEndpointAggressiveness(compression) {
+  const base = compression.aggressiveness ?? compression.usageTapCompressionAggressiveness ?? compression.tokenCompanyAggressiveness;
+  if (typeof base === "number" || base === void 0) {
+    return hasExplicitEnabledRoles(compression) ? buildRoleAggressiveness(compression, base) : base;
+  }
+  return buildRoleAggressiveness(compression, void 0, base);
+}
+function hasExplicitEnabledRoles(compression) {
+  return Object.values(compression.roles ?? {}).some((setting) => setting !== false);
+}
+function buildRoleAggressiveness(compression, fallback, base = {}) {
+  const roles = ["system", "user", "tool", "assistant"];
+  const result = {};
+  for (const role of roles) {
+    const roleOptions = resolveRoleCompressionOptions(compression, role);
+    if (!roleOptions) {
+      continue;
+    }
+    const roleAggressiveness = roleOptions.aggressiveness ?? roleOptions.usageTapCompressionAggressiveness ?? roleOptions.tokenCompanyAggressiveness ?? base[role] ?? fallback;
+    if (roleAggressiveness !== void 0) {
+      result[role] = roleAggressiveness;
+    }
+  }
+  return result;
+}
 function buildPromptCompressionTelemetry(segments) {
   if (!segments.length) {
     return void 0;