@usagetap/sdk 1.1.0 → 1.2.0

package/README.md

@@ -178,7 +178,31 @@ await anthropic.messages.create({
 });
 ```
 
-When using The Token Company, configure `tokenCompanyApiKey` on `UsageTapClient` and set `provider: "thetokencompany"`. Optional `tokenCompanyModel`, `tokenCompanyAggressiveness`, and `tokenCompanyAppId` are supported at the client, manual `promptCompress`, and wrapper levels. Use `protectPromptText()` for text that must be passed through unchanged by compression-compatible providers.
+Use `provider: "usagetap"` to compress with UsageTap's hosted endpoints. Manual `promptCompress()` and `compressPromptInput()` use the single-text The Token Company-compatible endpoint at `https://compress.usagetap.com/v1/compress`, where `aggressiveness` is a single number from `0.0` to `1.0`. `wrapOpenAI()` and `wrapAnthropic()` use the message/request endpoint at `https://compress.usagetap.com/v1/messages/compress`, where `aggressiveness` may be a per-role object:
+
+```ts
+const result = await usageTap.promptCompress({
+	callId: begin.data.callId,
+	text: "Your text here",
+	provider: "usagetap",
+	model: "bear-2",
+	aggressiveness: 0.5,
+});
+```
+
+```ts
+const ai = wrapOpenAI(openai, usageTap, {
+		defaultContext: { customerId: "cust_123", feature: "chat.send" },
+		promptCompression: {
+			provider: "usagetap",
+			aggressiveness: { user: 0.5, system: 0.5, tool: 0.5 },
+		},
+	});
+```
+
+`UsageTapClient` sends your UsageTap API key by default. Override `usageTapCompressionEndpoint` for single-text compression or `usageTapCompressionMessagesEndpoint` for wrapper message compression.
+
+When using The Token Company, configure `tokenCompanyApiKey` on `UsageTapClient` and set `provider: "thetokencompany"`. Optional `tokenCompanyModel`, `aggressiveness`, and `tokenCompanyAppId` are supported at the client, manual `promptCompress`, and wrapper levels. Use `protectPromptText()` for text that must be passed through unchanged by compression-compatible providers.
 
 For advanced custom flows, `compressPromptInput(input, options?)` returns compression results without recording telemetry, and `recordPromptCompression({ callId, promptCompression })` records precomputed savings metadata against a call.
 

package/dist/adapters/anthropic.cjs

@@ -268,6 +268,18 @@ async function compressAnthropicParams(params, usageTap, compression, signal) {
   }
   const source = cloneRecord(params);
   const 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 }]
+    };
+  }
   if (typeof source.system === "string") {
     const compressed = await compressTextForRole(
       source.system,
@@ -441,8 +453,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);
@@ -501,16 +516,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;