the-token-company 0.2.1 → 0.2.5

package/dist/ai-sdk.js

@@ -1,6 +1,6 @@
 import { wrapLanguageModel } from "ai";
 import { TheTokenCompany } from "./client.js";
-import { AnalyticsTTC, compressAISDKPrompt, resolveAggressiveness } from "./compress.js";
+import { StatsTTC, compressAISDKPrompt, resolveAggressiveness } from "./compress.js";
 import { CompressionStats } from "./types.js";
 /**
  * Create a Vercel AI SDK middleware that auto-compresses non-assistant messages.
@@ -17,7 +17,7 @@ import { CompressionStats } from "./types.js";
  */
 export function compressionMiddleware(options) {
     const stats = new CompressionStats();
-    const analytics = new AnalyticsTTC(new TheTokenCompany({ apiKey: options.compressionApiKey, appId: options.appId }), stats);
+    const compressor = new StatsTTC(new TheTokenCompany({ apiKey: options.compressionApiKey, appId: options.appId, fetch: options.fetch }), stats);
     const compressionModel = options.model ?? "bear-2";
     const roleAggr = resolveAggressiveness(options.aggressiveness ?? 0.2);
     const middleware = {
@@ -26,7 +26,7 @@ export function compressionMiddleware(options) {
         transformParams: async ({ params }) => {
             if (params.prompt) {
                 stats._startTurn();
-                const compressed = await compressAISDKPrompt(analytics, params.prompt, compressionModel, roleAggr);
+                const compressed = await compressAISDKPrompt(compressor, params.prompt, compressionModel, roleAggr);
                 stats._endTurn();
                 return { ...params, prompt: compressed };
             }

package/dist/anthropic.js

@@ -1,5 +1,5 @@
 import { TheTokenCompany } from "./client.js";
-import { AnalyticsTTC, compressAnthropicMessages, resolveAggressiveness } from "./compress.js";
+import { StatsTTC, compressAnthropicMessages, resolveAggressiveness } from "./compress.js";
 import { CompressionStats } from "./types.js";
 /**
  * Wrap an Anthropic client to auto-compress non-assistant messages.
@@ -24,7 +24,7 @@ import { CompressionStats } from "./types.js";
  */
 export function withCompression(client, options) {
     const stats = new CompressionStats();
-    const analytics = new AnalyticsTTC(new TheTokenCompany({ apiKey: options.compressionApiKey, appId: options.appId }), stats);
+    const compressor = new StatsTTC(new TheTokenCompany({ apiKey: options.compressionApiKey, appId: options.appId, fetch: options.fetch }), stats);
     const model = options.model ?? "bear-2";
     const roleAggr = resolveAggressiveness(options.aggressiveness ?? 0.2);
     const systemAggr = roleAggr["system"];
@@ -34,11 +34,11 @@ export function withCompression(client, options) {
         if (params?.messages) {
             params = {
                 ...params,
-                messages: await compressAnthropicMessages(analytics, params.messages, model, roleAggr),
+                messages: await compressAnthropicMessages(compressor, params.messages, model, roleAggr),
             };
         }
         if (systemAggr != null && typeof params?.system === "string" && params.system.trim()) {
-            const result = await analytics.compress(params.system, { model, aggressiveness: systemAggr });
+            const result = await compressor.compress(params.system, { model, aggressiveness: systemAggr });
             params = { ...params, system: result.output };
         }
         stats._endTurn();

package/dist/client.d.ts

@@ -9,6 +9,7 @@ export declare class TheTokenCompany {
     private readonly timeout;
     private readonly gzip;
     private readonly appId?;
+    private readonly fetchFn;
     constructor(options: TheTokenCompanyOptions);
     compress(text: string, options?: {
         model?: string;

package/dist/client.js

@@ -14,12 +14,14 @@ export class TheTokenCompany {
     timeout;
     gzip;
     appId;
+    fetchFn;
     constructor(options) {
         this.apiKey = options.apiKey;
         this.baseUrl = (options.baseUrl ?? BASE_URL).replace(/\/$/, "");
         this.timeout = options.timeout ?? DEFAULT_TIMEOUT;
         this.gzip = options.gzip ?? true;
         this.appId = options.appId;
+        this.fetchFn = options.fetch ?? globalThis.fetch;
     }
     async compress(text, options = {}) {
         const { model = "bear-2", aggressiveness = 0.2 } = options;
@@ -49,7 +51,7 @@ export class TheTokenCompany {
         else {
             body = jsonBody;
         }
-        const response = await fetch(`${this.baseUrl}/v1/compress`, {
+        const response = await this.fetchFn(`${this.baseUrl}/v1/compress`, {
             method: "POST",
             headers,
             body: body,

package/dist/compress.d.ts

@@ -7,7 +7,7 @@ export interface Compressor {
         aggressiveness?: number;
     }): Promise<CompressResult>;
 }
-export declare class AnalyticsTTC implements Compressor {
+export declare class StatsTTC implements Compressor {
     private readonly inner;
     readonly stats: CompressionStats;
     constructor(inner: TheTokenCompany, stats: CompressionStats);

package/dist/compress.js

@@ -1,5 +1,5 @@
 const DEFAULT_ROLES = ["user", "system", "tool"];
-export class AnalyticsTTC {
+export class StatsTTC {
     inner;
     stats;
     constructor(inner, stats) {

package/dist/openai.js

@@ -1,5 +1,5 @@
 import { TheTokenCompany } from "./client.js";
-import { AnalyticsTTC, compressOpenAIMessages, resolveAggressiveness } from "./compress.js";
+import { StatsTTC, compressOpenAIMessages, resolveAggressiveness } from "./compress.js";
 import { CompressionStats } from "./types.js";
 /**
  * Wrap an OpenAI-compatible client to auto-compress non-assistant messages.
@@ -22,7 +22,7 @@ import { CompressionStats } from "./types.js";
  */
 export function withCompression(client, options) {
     const stats = new CompressionStats();
-    const analytics = new AnalyticsTTC(new TheTokenCompany({ apiKey: options.compressionApiKey, appId: options.appId }), stats);
+    const compressor = new StatsTTC(new TheTokenCompany({ apiKey: options.compressionApiKey, appId: options.appId, fetch: options.fetch }), stats);
     const model = options.model ?? "bear-2";
     const roleAggr = resolveAggressiveness(options.aggressiveness ?? 0.2);
     const originalCreate = client.chat.completions.create.bind(client.chat.completions);
@@ -31,7 +31,7 @@ export function withCompression(client, options) {
             stats._startTurn();
             params = {
                 ...params,
-                messages: await compressOpenAIMessages(analytics, params.messages, model, roleAggr),
+                messages: await compressOpenAIMessages(compressor, params.messages, model, roleAggr),
             };
             stats._endTurn();
         }

package/dist/types.d.ts

@@ -24,6 +24,7 @@ export interface TheTokenCompanyOptions {
     timeout?: number;
     gzip?: boolean;
     appId?: string;
+    fetch?: typeof globalThis.fetch;
 }
 export type Aggressiveness = number | Record<string, number>;
 export interface WithCompressionOptions {
@@ -31,6 +32,7 @@ export interface WithCompressionOptions {
     model?: string;
     aggressiveness?: Aggressiveness;
     appId?: string;
+    fetch?: typeof globalThis.fetch;
 }
 export interface TurnStats {
     inputTokens: number;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "the-token-company",
-  "version": "0.2.1",
+  "version": "0.2.5",
   "description": "Node.js SDK for The Token Company — compress LLM prompts to reduce costs and latency",
   "type": "module",
   "main": "./dist/index.js",