the-token-company 0.2.2 → 0.2.5

package/dist/ai-sdk.js

@@ -17,7 +17,7 @@ import { CompressionStats } from "./types.js";
  */
 export function compressionMiddleware(options) {
     const stats = new CompressionStats();
-    const compressor = new StatsTTC(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 = {

package/dist/anthropic.js

@@ -24,7 +24,7 @@ import { CompressionStats } from "./types.js";
  */
 export function withCompression(client, options) {
     const stats = new CompressionStats();
-    const compressor = new StatsTTC(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"];

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/openai.js

@@ -22,7 +22,7 @@ import { CompressionStats } from "./types.js";
  */
 export function withCompression(client, options) {
     const stats = new CompressionStats();
-    const compressor = new StatsTTC(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);

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.2",
+  "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",