@usagetap/sdk 1.0.0 → 1.1.0

package/dist/index.cjs

@@ -117,6 +117,13 @@ async function runWithRetry(operation, options, shouldRetry, onSchedule, signal)
 
 // src/prompt-compression.ts
 var DEFAULT_TTC_ENDPOINT = "https://api.thetokencompany.com/v1/compress";
+var DEFAULT_TTC_MODEL = "bear-2";
+var DEFAULT_TTC_AGGRESSIVENESS = 0.2;
+var PROTECTED_TEXT_PATTERN = /<ttc_safe>[\s\S]*?<\/ttc_safe>|<usagetap_safe>[\s\S]*?<\/usagetap_safe>/g;
+function protectPromptText(text) {
+  return `<ttc_safe>${text}</ttc_safe>`;
+}
+var protect = protectPromptText;
 async function compressPrompt(options) {
   try {
     if (options.provider === "thetokencompany" || options.tokenCompanyApiKey) {
@@ -179,6 +186,12 @@ async function compressWithTheTokenCompany(options) {
   }
   const original = stableStringifyInput(options.input);
   const heuristic = compressPromptHeuristic(options.input);
+  const input = typeof heuristic.compressedInput === "string" ? heuristic.compressedInput : stableStringifyInput(heuristic.compressedInput);
+  const model = options.tokenCompanyModel ?? DEFAULT_TTC_MODEL;
+  const aggressiveness = options.tokenCompanyAggressiveness ?? DEFAULT_TTC_AGGRESSIVENESS;
+  if (typeof aggressiveness !== "number" || !Number.isFinite(aggressiveness) || aggressiveness < 0 || aggressiveness > 1) {
+    throw new Error("tokenCompanyAggressiveness must be between 0.0 and 1.0");
+  }
   const response = await fetchCandidate(
     options.tokenCompanyEndpoint ?? DEFAULT_TTC_ENDPOINT,
     {
@@ -187,7 +200,12 @@ async function compressWithTheTokenCompany(options) {
         authorization: `Bearer ${options.tokenCompanyApiKey}`,
         "content-type": "application/json"
       },
-      body: JSON.stringify({ input: heuristic.compressedInput }),
+      body: JSON.stringify({
+        model,
+        input,
+        compression_settings: { aggressiveness },
+        ...options.tokenCompanyAppId ? { app_id: options.tokenCompanyAppId } : {}
+      }),
       signal: options.signal
     }
   );
@@ -295,6 +313,20 @@ function compressValue(value, techniques, options) {
   return value;
 }
 function compressText(value, techniques, options) {
+  const protectedSpans = [];
+  const text = value.replace(PROTECTED_TEXT_PATTERN, (match) => {
+    const placeholder = `__USAGETAP_PROTECTED_${protectedSpans.length}__`;
+    protectedSpans.push(match);
+    techniques.add("protected-text");
+    return placeholder;
+  });
+  const compressed = compressTextWithoutProtection(text, techniques, options);
+  return protectedSpans.reduce(
+    (output, span, index) => output.replace(`__USAGETAP_PROTECTED_${index}__`, span),
+    compressed
+  );
+}
+function compressTextWithoutProtection(value, techniques, options) {
   const fencePattern = /```([\w-]+)?\n([\s\S]*?)```/g;
   const parts = [];
   let cursor = 0;
@@ -588,7 +620,7 @@ var IDEMPOTENCY_HEADER = "idempotency-key";
 var SDK_HEADER = "x-usage-sdk";
 var USER_AGENT = "UsageTapClient";
 var CANONICAL_MEDIA_TYPE = "application/vnd.usagetap.v1+json";
-var SDK_VERSION = "1.0.0" ;
+var SDK_VERSION = "1.1.0" ;
 var HAS_WINDOW = typeof globalThis !== "undefined" && typeof globalThis.window !== "undefined";
 var UsageTapClient = class {
   apiKey;
@@ -605,6 +637,9 @@ var UsageTapClient = class {
   autoIdempotency;
   tokenCompanyApiKey;
   tokenCompanyEndpoint;
+  tokenCompanyModel;
+  tokenCompanyAggressiveness;
+  tokenCompanyAppId;
   constructor(options) {
     if (!options) {
       throw new UsageTapError(
@@ -652,6 +687,9 @@ var UsageTapClient = class {
     this.autoIdempotency = options.autoIdempotency ?? true;
     this.tokenCompanyApiKey = options.tokenCompanyApiKey;
     this.tokenCompanyEndpoint = options.tokenCompanyEndpoint;
+    this.tokenCompanyModel = options.tokenCompanyModel;
+    this.tokenCompanyAggressiveness = options.tokenCompanyAggressiveness;
+    this.tokenCompanyAppId = options.tokenCompanyAppId;
   }
   async beginCall(request, options = {}) {
     const idempotencyKey = request.idempotencyKey ?? request.idempotency ?? (this.autoIdempotency ? this.idempotencyGenerator() : void 0);
@@ -681,17 +719,15 @@ var UsageTapClient = class {
         "promptCompress requires callId"
       );
     }
-    const result = await compressPrompt({
-      input: request.input,
+    const result = await this.compressPromptInput(request.input, {
       provider: request.provider,
-      tokenCompanyApiKey: this.tokenCompanyApiKey,
-      tokenCompanyEndpoint: this.tokenCompanyEndpoint,
-      fetchImpl: this.fetchImpl,
+      tokenCompanyModel: request.tokenCompanyModel,
+      tokenCompanyAggressiveness: request.tokenCompanyAggressiveness,
+      tokenCompanyAppId: request.tokenCompanyAppId,
       signal: options.signal
     });
     try {
-      await this.request(
-        COMPRESS_PROMPT_PATH,
+      await this.recordPromptCompression(
         {
           callId: request.callId,
           promptCompression: this.toPromptCompressionTelemetry(result)
@@ -710,6 +746,36 @@ var UsageTapClient = class {
       };
     }
   }
+  async compressPromptInput(input, options = {}) {
+    return compressPrompt({
+      input,
+      provider: options.provider,
+      tokenCompanyApiKey: this.tokenCompanyApiKey,
+      tokenCompanyEndpoint: this.tokenCompanyEndpoint,
+      tokenCompanyModel: options.tokenCompanyModel ?? this.tokenCompanyModel,
+      tokenCompanyAggressiveness: options.tokenCompanyAggressiveness ?? this.tokenCompanyAggressiveness,
+      tokenCompanyAppId: options.tokenCompanyAppId ?? this.tokenCompanyAppId,
+      fetchImpl: this.fetchImpl,
+      signal: options.signal,
+      failOpen: options.failOpen
+    });
+  }
+  async recordPromptCompression(request, options = {}) {
+    if (!request?.callId) {
+      throw new UsageTapError(
+        "USAGETAP_BAD_REQUEST",
+        "recordPromptCompression requires callId"
+      );
+    }
+    return this.request(
+      COMPRESS_PROMPT_PATH,
+      {
+        callId: request.callId,
+        promptCompression: request.promptCompression
+      },
+      options
+    );
+  }
   async endCall(request, options = {}) {
     if (!request?.callId) {
       throw new UsageTapError(
@@ -1543,6 +1609,8 @@ exports.compressPromptToon = compressPromptToon;
 exports.createIdempotencyKey = createIdempotencyKey;
 exports.estimatePromptTokens = estimatePromptTokens;
 exports.isUsageTapError = isUsageTapError;
+exports.protect = protect;
+exports.protectPromptText = protectPromptText;
 exports.wrapFetch = wrapFetch;
 //# sourceMappingURL=index.cjs.map
 //# sourceMappingURL=index.cjs.map