headroom-ai 0.1.0 → 0.22.4

package/dist/adapters/gemini.js

@@ -0,0 +1,39 @@
+import { compress } from '../chunk-7AJIWIPP.js';
+
+// src/adapters/gemini.ts
+function withHeadroom(model, options = {}) {
+  const originalGenerate = model.generateContent.bind(model);
+  const originalStream = model.generateContentStream?.bind(model);
+  const compressedGenerate = async (params) => {
+    const contents = params.contents ?? params;
+    const modelName = options.model ?? "gemini-2.0-flash";
+    const result = await compress(
+      Array.isArray(contents) ? contents : [contents],
+      { stack: "adapter_ts_gemini", ...options, model: modelName }
+    );
+    const newParams = Array.isArray(params) ? result.messages : { ...params, contents: result.messages };
+    return originalGenerate(newParams);
+  };
+  const compressedStream = originalStream ? async (params) => {
+    const contents = params.contents ?? params;
+    const modelName = options.model ?? "gemini-2.0-flash";
+    const result = await compress(
+      Array.isArray(contents) ? contents : [contents],
+      { stack: "adapter_ts_gemini", ...options, model: modelName }
+    );
+    const newParams = Array.isArray(params) ? result.messages : { ...params, contents: result.messages };
+    return originalStream(newParams);
+  } : void 0;
+  return new Proxy(model, {
+    get(target, prop) {
+      if (prop === "generateContent") return compressedGenerate;
+      if (prop === "generateContentStream" && compressedStream)
+        return compressedStream;
+      return target[prop];
+    }
+  });
+}
+
+export { withHeadroom };
+//# sourceMappingURL=gemini.js.map
+//# sourceMappingURL=gemini.js.map

package/dist/adapters/gemini.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/adapters/gemini.ts"],"names":[],"mappings":";;;AAgCO,SAAS,YAAA,CACd,KAAA,EACA,OAAA,GAA2B,EAAC,EACzB;AACH,EAAA,MAAM,gBAAA,GAAmB,KAAA,CAAM,eAAA,CAAgB,IAAA,CAAK,KAAK,CAAA;AACzD,EAAA,MAAM,cAAA,GAAiB,KAAA,CAAM,qBAAA,EAAuB,IAAA,CAAK,KAAK,CAAA;AAE9D,EAAA,MAAM,kBAAA,GAAqB,OAAO,MAAA,KAAgB;AAChD,IAAA,MAAM,QAAA,GAAW,OAAO,QAAA,IAAY,MAAA;AACpC,IAAA,MAAM,SAAA,GAAY,QAAQ,KAAA,IAAS,kBAAA;AAGnC,IAAA,MAAM,SAAS,MAAM,QAAA;AAAA,MACnB,MAAM,OAAA,CAAQ,QAAQ,CAAA,GAAI,QAAA,GAAW,CAAC,QAAQ,CAAA;AAAA,MAC9C,EAAE,KAAA,EAAO,mBAAA,EAAqB,GAAG,OAAA,EAAS,OAAO,SAAA;AAAU,KAC7D;AAEA,IAAA,MAAM,SAAA,GAAY,KAAA,CAAM,OAAA,CAAQ,MAAM,CAAA,GAClC,MAAA,CAAO,QAAA,GACP,EAAE,GAAG,MAAA,EAAQ,QAAA,EAAU,MAAA,CAAO,QAAA,EAAS;AAE3C,IAAA,OAAO,iBAAiB,SAAS,CAAA;AAAA,EACnC,CAAA;AAEA,EAAA,MAAM,gBAAA,GAAmB,cAAA,GACrB,OAAO,MAAA,KAAgB;AACrB,IAAA,MAAM,QAAA,GAAW,OAAO,QAAA,IAAY,MAAA;AACpC,IAAA,MAAM,SAAA,GAAY,QAAQ,KAAA,IAAS,kBAAA;AAEnC,IAAA,MAAM,SAAS,MAAM,QAAA;AAAA,MACnB,MAAM,OAAA,CAAQ,QAAQ,CAAA,GAAI,QAAA,GAAW,CAAC,QAAQ,CAAA;AAAA,MAC9C,EAAE,KAAA,EAAO,mBAAA,EAAqB,GAAG,OAAA,EAAS,OAAO,SAAA;AAAU,KAC7D;AAEA,IAAA,MAAM,SAAA,GAAY,KAAA,CAAM,OAAA,CAAQ,MAAM,CAAA,GAClC,MAAA,CAAO,QAAA,GACP,EAAE,GAAG,MAAA,EAAQ,QAAA,EAAU,MAAA,CAAO,QAAA,EAAS;AAE3C,IAAA,OAAO,eAAe,SAAS,CAAA;AAAA,EACjC,CAAA,GACA,MAAA;AAEJ,EAAA,OAAO,IAAI,MAAM,KAAA,EAAO;AAAA,IACtB,GAAA,CAAI,QAAQ,IAAA,EAAM;AAChB,MAAA,IAAI,IAAA,KAAS,mBAAmB,OAAO,kBAAA;AACvC,MAAA,IAAI,SAAS,uBAAA,IAA2B,gBAAA;AACtC,QAAA,OAAO,gBAAA;AACT,MAAA,OAAQ,OAAe,IAAI,CAAA;AAAA,IAC7B;AAAA,GACD,CAAA;AACH","file":"gemini.js","sourcesContent":["/**\n * Gemini adapter — wraps Google Generative AI model with auto-compression.\n * Matches the pattern of openai.ts and anthropic.ts adapters.\n */\n\nimport { compress } from \"../compress.js\";\nimport type { CompressOptions } from \"../types.js\";\n\n/* eslint-disable @typescript-eslint/no-explicit-any */\n\ninterface GeminiModelLike {\n  generateContent: (params: any) => any;\n  generateContentStream?: (params: any) => any;\n  [key: string]: any;\n}\n\n/**\n * Wrap a Google Generative AI model to auto-compress before each request.\n *\n * Intercepts `model.generateContent()` and `model.generateContentStream()`.\n *\n * @example\n * ```typescript\n * import { withHeadroom } from 'headroom-ai/gemini';\n * import { GoogleGenerativeAI } from '@google/generative-ai';\n *\n * const genAI = new GoogleGenerativeAI(process.env.GEMINI_API_KEY!);\n * const model = withHeadroom(genAI.getGenerativeModel({ model: 'gemini-2.0-flash' }));\n *\n * const result = await model.generateContent({ contents: longConversation });\n * ```\n */\nexport function withHeadroom<T extends GeminiModelLike>(\n  model: T,\n  options: CompressOptions = {},\n): T {\n  const originalGenerate = model.generateContent.bind(model);\n  const originalStream = model.generateContentStream?.bind(model);\n\n  const compressedGenerate = async (params: any) => {\n    const contents = params.contents ?? params;\n    const modelName = options.model ?? \"gemini-2.0-flash\";\n\n    // compress() auto-detects Gemini format\n    const result = await compress(\n      Array.isArray(contents) ? contents : [contents],\n      { stack: \"adapter_ts_gemini\", ...options, model: modelName },\n    );\n\n    const newParams = Array.isArray(params)\n      ? result.messages\n      : { ...params, contents: result.messages };\n\n    return originalGenerate(newParams);\n  };\n\n  const compressedStream = originalStream\n    ? async (params: any) => {\n        const contents = params.contents ?? params;\n        const modelName = options.model ?? \"gemini-2.0-flash\";\n\n        const result = await compress(\n          Array.isArray(contents) ? contents : [contents],\n          { stack: \"adapter_ts_gemini\", ...options, model: modelName },\n        );\n\n        const newParams = Array.isArray(params)\n          ? result.messages\n          : { ...params, contents: result.messages };\n\n        return originalStream(newParams);\n      }\n    : undefined;\n\n  return new Proxy(model, {\n    get(target, prop) {\n      if (prop === \"generateContent\") return compressedGenerate;\n      if (prop === \"generateContentStream\" && compressedStream)\n        return compressedStream;\n      return (target as any)[prop];\n    },\n  }) as T;\n}\n"]}

package/dist/adapters/openai.cjs

@@ -1,6 +1,6 @@
 'use strict';
 
-var chunk3N25JEWK_cjs = require('../chunk-3N25JEWK.cjs');
+var chunkR6FB52AD_cjs = require('../chunk-R6FB52AD.cjs');
 
 // src/adapters/openai.ts
 function withHeadroom(client, options = {}) {
@@ -10,7 +10,11 @@ function withHeadroom(client, options = {}) {
   const compressedCreate = async (params) => {
     const messages = params.messages;
     const model = options.model ?? params.model ?? "gpt-4o";
-    const result = await chunk3N25JEWK_cjs.compress(messages, { ...options, model });
+    const result = await chunkR6FB52AD_cjs.compress(messages, {
+      stack: "adapter_ts_openai",
+      ...options,
+      model
+    });
     return originalCreate({
       ...params,
       messages: result.messages

package/dist/adapters/openai.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/adapters/openai.ts"],"names":["compress"],"mappings":";;;;;AAgCO,SAAS,YAAA,CACd,MAAA,EACA,OAAA,GAA2B,EAAC,EACzB;AACH,EAAA,MAAM,cAAA,GAAiB,MAAA,CAAO,IAAA,CAAK,WAAA,CAAY,MAAA,CAAO,IAAA;AAAA,IACpD,OAAO,IAAA,CAAK;AAAA,GACd;AAEA,EAAA,MAAM,gBAAA,GAAmB,OAAO,MAAA,KAAgB;AAC9C,IAAA,MAAM,WAA4B,MAAA,CAAO,QAAA;AACzC,IAAA,MAAM,KAAA,GAAQ,OAAA,CAAQ,KAAA,IAAS,MAAA,CAAO,KAAA,IAAS,QAAA;AAE/C,IAAA,MAAM,MAAA,GAAS,MAAMA,0BAAA,CAAS,QAAA,EAAU,EAAE,GAAG,OAAA,EAAS,OAAO,CAAA;AAE7D,IAAA,OAAO,cAAA,CAAe;AAAA,MACpB,GAAG,MAAA;AAAA,MACH,UAAU,MAAA,CAAO;AAAA,KAClB,CAAA;AAAA,EACH,CAAA;AAEA,EAAA,MAAM,gBAAA,GAAmB,IAAI,KAAA,CAAM,MAAA,CAAO,KAAK,WAAA,EAAa;AAAA,IAC1D,GAAA,CAAI,QAAQ,IAAA,EAAM;AAChB,MAAA,IAAI,IAAA,KAAS,UAAU,OAAO,gBAAA;AAC9B,MAAA,OAAQ,OAAe,IAAI,CAAA;AAAA,IAC7B;AAAA,GACD,CAAA;AAED,EAAA,MAAM,SAAA,GAAY,IAAI,KAAA,CAAM,MAAA,CAAO,IAAA,EAAM;AAAA,IACvC,GAAA,CAAI,QAAQ,IAAA,EAAM;AAChB,MAAA,IAAI,IAAA,KAAS,eAAe,OAAO,gBAAA;AACnC,MAAA,OAAQ,OAAe,IAAI,CAAA;AAAA,IAC7B;AAAA,GACD,CAAA;AAED,EAAA,OAAO,IAAI,MAAM,MAAA,EAAQ;AAAA,IACvB,GAAA,CAAI,QAAQ,IAAA,EAAM;AAChB,MAAA,IAAI,IAAA,KAAS,QAAQ,OAAO,SAAA;AAC5B,MAAA,OAAQ,OAAe,IAAI,CAAA;AAAA,IAC7B;AAAA,GACD,CAAA;AACH","file":"openai.cjs","sourcesContent":["import { compress } from \"../compress.js\";\nimport type { CompressOptions, OpenAIMessage } from \"../types.js\";\n\n/* eslint-disable @typescript-eslint/no-explicit-any */\n\ninterface OpenAILike {\n  chat: {\n    completions: {\n      create: (params: any) => any;\n    };\n  };\n  [key: string]: any;\n}\n\n/**\n * Wrap an OpenAI client to auto-compress messages before each request.\n *\n * Intercepts `client.chat.completions.create()` only. All other methods\n * (embeddings, images, audio, etc.) pass through unchanged.\n *\n * @example\n * ```typescript\n * import { withHeadroom } from 'headroom-ai/openai';\n * import OpenAI from 'openai';\n *\n * const client = withHeadroom(new OpenAI());\n * const response = await client.chat.completions.create({\n *   model: 'gpt-4o',\n *   messages: longConversation,\n * });\n * ```\n */\nexport function withHeadroom<T extends OpenAILike>(\n  client: T,\n  options: CompressOptions = {},\n): T {\n  const originalCreate = client.chat.completions.create.bind(\n    client.chat.completions,\n  );\n\n  const compressedCreate = async (params: any) => {\n    const messages: OpenAIMessage[] = params.messages;\n    const model = options.model ?? params.model ?? \"gpt-4o\";\n\n    const result = await compress(messages, { ...options, model });\n\n    return originalCreate({\n      ...params,\n      messages: result.messages,\n    });\n  };\n\n  const completionsProxy = new Proxy(client.chat.completions, {\n    get(target, prop) {\n      if (prop === \"create\") return compressedCreate;\n      return (target as any)[prop];\n    },\n  });\n\n  const chatProxy = new Proxy(client.chat, {\n    get(target, prop) {\n      if (prop === \"completions\") return completionsProxy;\n      return (target as any)[prop];\n    },\n  });\n\n  return new Proxy(client, {\n    get(target, prop) {\n      if (prop === \"chat\") return chatProxy;\n      return (target as any)[prop];\n    },\n  }) as T;\n}\n"]}
+{"version":3,"sources":["../../src/adapters/openai.ts"],"names":["compress"],"mappings":";;;;;AAgCO,SAAS,YAAA,CACd,MAAA,EACA,OAAA,GAA2B,EAAC,EACzB;AACH,EAAA,MAAM,cAAA,GAAiB,MAAA,CAAO,IAAA,CAAK,WAAA,CAAY,MAAA,CAAO,IAAA;AAAA,IACpD,OAAO,IAAA,CAAK;AAAA,GACd;AAEA,EAAA,MAAM,gBAAA,GAAmB,OAAO,MAAA,KAAgB;AAC9C,IAAA,MAAM,WAA4B,MAAA,CAAO,QAAA;AACzC,IAAA,MAAM,KAAA,GAAQ,OAAA,CAAQ,KAAA,IAAS,MAAA,CAAO,KAAA,IAAS,QAAA;AAE/C,IAAA,MAAM,MAAA,GAAS,MAAMA,0BAAA,CAAS,QAAA,EAAU;AAAA,MACtC,KAAA,EAAO,mBAAA;AAAA,MACP,GAAG,OAAA;AAAA,MACH;AAAA,KACD,CAAA;AAED,IAAA,OAAO,cAAA,CAAe;AAAA,MACpB,GAAG,MAAA;AAAA,MACH,UAAU,MAAA,CAAO;AAAA,KAClB,CAAA;AAAA,EACH,CAAA;AAEA,EAAA,MAAM,gBAAA,GAAmB,IAAI,KAAA,CAAM,MAAA,CAAO,KAAK,WAAA,EAAa;AAAA,IAC1D,GAAA,CAAI,QAAQ,IAAA,EAAM;AAChB,MAAA,IAAI,IAAA,KAAS,UAAU,OAAO,gBAAA;AAC9B,MAAA,OAAQ,OAAe,IAAI,CAAA;AAAA,IAC7B;AAAA,GACD,CAAA;AAED,EAAA,MAAM,SAAA,GAAY,IAAI,KAAA,CAAM,MAAA,CAAO,IAAA,EAAM;AAAA,IACvC,GAAA,CAAI,QAAQ,IAAA,EAAM;AAChB,MAAA,IAAI,IAAA,KAAS,eAAe,OAAO,gBAAA;AACnC,MAAA,OAAQ,OAAe,IAAI,CAAA;AAAA,IAC7B;AAAA,GACD,CAAA;AAED,EAAA,OAAO,IAAI,MAAM,MAAA,EAAQ;AAAA,IACvB,GAAA,CAAI,QAAQ,IAAA,EAAM;AAChB,MAAA,IAAI,IAAA,KAAS,QAAQ,OAAO,SAAA;AAC5B,MAAA,OAAQ,OAAe,IAAI,CAAA;AAAA,IAC7B;AAAA,GACD,CAAA;AACH","file":"openai.cjs","sourcesContent":["import { compress } from \"../compress.js\";\nimport type { CompressOptions, OpenAIMessage } from \"../types.js\";\n\n/* eslint-disable @typescript-eslint/no-explicit-any */\n\ninterface OpenAILike {\n  chat: {\n    completions: {\n      create: (params: any) => any;\n    };\n  };\n  [key: string]: any;\n}\n\n/**\n * Wrap an OpenAI client to auto-compress messages before each request.\n *\n * Intercepts `client.chat.completions.create()` only. All other methods\n * (embeddings, images, audio, etc.) pass through unchanged.\n *\n * @example\n * ```typescript\n * import { withHeadroom } from 'headroom-ai/openai';\n * import OpenAI from 'openai';\n *\n * const client = withHeadroom(new OpenAI());\n * const response = await client.chat.completions.create({\n *   model: 'gpt-4o',\n *   messages: longConversation,\n * });\n * ```\n */\nexport function withHeadroom<T extends OpenAILike>(\n  client: T,\n  options: CompressOptions = {},\n): T {\n  const originalCreate = client.chat.completions.create.bind(\n    client.chat.completions,\n  );\n\n  const compressedCreate = async (params: any) => {\n    const messages: OpenAIMessage[] = params.messages;\n    const model = options.model ?? params.model ?? \"gpt-4o\";\n\n    const result = await compress(messages, {\n      stack: \"adapter_ts_openai\",\n      ...options,\n      model,\n    });\n\n    return originalCreate({\n      ...params,\n      messages: result.messages,\n    });\n  };\n\n  const completionsProxy = new Proxy(client.chat.completions, {\n    get(target, prop) {\n      if (prop === \"create\") return compressedCreate;\n      return (target as any)[prop];\n    },\n  });\n\n  const chatProxy = new Proxy(client.chat, {\n    get(target, prop) {\n      if (prop === \"completions\") return completionsProxy;\n      return (target as any)[prop];\n    },\n  });\n\n  return new Proxy(client, {\n    get(target, prop) {\n      if (prop === \"chat\") return chatProxy;\n      return (target as any)[prop];\n    },\n  }) as T;\n}\n"]}

package/dist/adapters/openai.d.cts

@@ -1,4 +1,4 @@
-import { C as CompressOptions } from '../types-DQtcLXq3.cjs';
+import { C as CompressOptions } from '../types-BTrX7__W.cjs';
 
 interface OpenAILike {
     chat: {

package/dist/adapters/openai.d.ts

@@ -1,4 +1,4 @@
-import { C as CompressOptions } from '../types-DQtcLXq3.js';
+import { C as CompressOptions } from '../types-BTrX7__W.js';
 
 interface OpenAILike {
     chat: {

package/dist/adapters/openai.js

@@ -1,4 +1,4 @@
-import { compress } from '../chunk-YTTW7S2Q.js';
+import { compress } from '../chunk-7AJIWIPP.js';
 
 // src/adapters/openai.ts
 function withHeadroom(client, options = {}) {
@@ -8,7 +8,11 @@ function withHeadroom(client, options = {}) {
   const compressedCreate = async (params) => {
     const messages = params.messages;
     const model = options.model ?? params.model ?? "gpt-4o";
-    const result = await compress(messages, { ...options, model });
+    const result = await compress(messages, {
+      stack: "adapter_ts_openai",
+      ...options,
+      model
+    });
     return originalCreate({
       ...params,
       messages: result.messages

package/dist/adapters/openai.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/adapters/openai.ts"],"names":[],"mappings":";;;AAgCO,SAAS,YAAA,CACd,MAAA,EACA,OAAA,GAA2B,EAAC,EACzB;AACH,EAAA,MAAM,cAAA,GAAiB,MAAA,CAAO,IAAA,CAAK,WAAA,CAAY,MAAA,CAAO,IAAA;AAAA,IACpD,OAAO,IAAA,CAAK;AAAA,GACd;AAEA,EAAA,MAAM,gBAAA,GAAmB,OAAO,MAAA,KAAgB;AAC9C,IAAA,MAAM,WAA4B,MAAA,CAAO,QAAA;AACzC,IAAA,MAAM,KAAA,GAAQ,OAAA,CAAQ,KAAA,IAAS,MAAA,CAAO,KAAA,IAAS,QAAA;AAE/C,IAAA,MAAM,MAAA,GAAS,MAAM,QAAA,CAAS,QAAA,EAAU,EAAE,GAAG,OAAA,EAAS,OAAO,CAAA;AAE7D,IAAA,OAAO,cAAA,CAAe;AAAA,MACpB,GAAG,MAAA;AAAA,MACH,UAAU,MAAA,CAAO;AAAA,KAClB,CAAA;AAAA,EACH,CAAA;AAEA,EAAA,MAAM,gBAAA,GAAmB,IAAI,KAAA,CAAM,MAAA,CAAO,KAAK,WAAA,EAAa;AAAA,IAC1D,GAAA,CAAI,QAAQ,IAAA,EAAM;AAChB,MAAA,IAAI,IAAA,KAAS,UAAU,OAAO,gBAAA;AAC9B,MAAA,OAAQ,OAAe,IAAI,CAAA;AAAA,IAC7B;AAAA,GACD,CAAA;AAED,EAAA,MAAM,SAAA,GAAY,IAAI,KAAA,CAAM,MAAA,CAAO,IAAA,EAAM;AAAA,IACvC,GAAA,CAAI,QAAQ,IAAA,EAAM;AAChB,MAAA,IAAI,IAAA,KAAS,eAAe,OAAO,gBAAA;AACnC,MAAA,OAAQ,OAAe,IAAI,CAAA;AAAA,IAC7B;AAAA,GACD,CAAA;AAED,EAAA,OAAO,IAAI,MAAM,MAAA,EAAQ;AAAA,IACvB,GAAA,CAAI,QAAQ,IAAA,EAAM;AAChB,MAAA,IAAI,IAAA,KAAS,QAAQ,OAAO,SAAA;AAC5B,MAAA,OAAQ,OAAe,IAAI,CAAA;AAAA,IAC7B;AAAA,GACD,CAAA;AACH","file":"openai.js","sourcesContent":["import { compress } from \"../compress.js\";\nimport type { CompressOptions, OpenAIMessage } from \"../types.js\";\n\n/* eslint-disable @typescript-eslint/no-explicit-any */\n\ninterface OpenAILike {\n  chat: {\n    completions: {\n      create: (params: any) => any;\n    };\n  };\n  [key: string]: any;\n}\n\n/**\n * Wrap an OpenAI client to auto-compress messages before each request.\n *\n * Intercepts `client.chat.completions.create()` only. All other methods\n * (embeddings, images, audio, etc.) pass through unchanged.\n *\n * @example\n * ```typescript\n * import { withHeadroom } from 'headroom-ai/openai';\n * import OpenAI from 'openai';\n *\n * const client = withHeadroom(new OpenAI());\n * const response = await client.chat.completions.create({\n *   model: 'gpt-4o',\n *   messages: longConversation,\n * });\n * ```\n */\nexport function withHeadroom<T extends OpenAILike>(\n  client: T,\n  options: CompressOptions = {},\n): T {\n  const originalCreate = client.chat.completions.create.bind(\n    client.chat.completions,\n  );\n\n  const compressedCreate = async (params: any) => {\n    const messages: OpenAIMessage[] = params.messages;\n    const model = options.model ?? params.model ?? \"gpt-4o\";\n\n    const result = await compress(messages, { ...options, model });\n\n    return originalCreate({\n      ...params,\n      messages: result.messages,\n    });\n  };\n\n  const completionsProxy = new Proxy(client.chat.completions, {\n    get(target, prop) {\n      if (prop === \"create\") return compressedCreate;\n      return (target as any)[prop];\n    },\n  });\n\n  const chatProxy = new Proxy(client.chat, {\n    get(target, prop) {\n      if (prop === \"completions\") return completionsProxy;\n      return (target as any)[prop];\n    },\n  });\n\n  return new Proxy(client, {\n    get(target, prop) {\n      if (prop === \"chat\") return chatProxy;\n      return (target as any)[prop];\n    },\n  }) as T;\n}\n"]}
+{"version":3,"sources":["../../src/adapters/openai.ts"],"names":[],"mappings":";;;AAgCO,SAAS,YAAA,CACd,MAAA,EACA,OAAA,GAA2B,EAAC,EACzB;AACH,EAAA,MAAM,cAAA,GAAiB,MAAA,CAAO,IAAA,CAAK,WAAA,CAAY,MAAA,CAAO,IAAA;AAAA,IACpD,OAAO,IAAA,CAAK;AAAA,GACd;AAEA,EAAA,MAAM,gBAAA,GAAmB,OAAO,MAAA,KAAgB;AAC9C,IAAA,MAAM,WAA4B,MAAA,CAAO,QAAA;AACzC,IAAA,MAAM,KAAA,GAAQ,OAAA,CAAQ,KAAA,IAAS,MAAA,CAAO,KAAA,IAAS,QAAA;AAE/C,IAAA,MAAM,MAAA,GAAS,MAAM,QAAA,CAAS,QAAA,EAAU;AAAA,MACtC,KAAA,EAAO,mBAAA;AAAA,MACP,GAAG,OAAA;AAAA,MACH;AAAA,KACD,CAAA;AAED,IAAA,OAAO,cAAA,CAAe;AAAA,MACpB,GAAG,MAAA;AAAA,MACH,UAAU,MAAA,CAAO;AAAA,KAClB,CAAA;AAAA,EACH,CAAA;AAEA,EAAA,MAAM,gBAAA,GAAmB,IAAI,KAAA,CAAM,MAAA,CAAO,KAAK,WAAA,EAAa;AAAA,IAC1D,GAAA,CAAI,QAAQ,IAAA,EAAM;AAChB,MAAA,IAAI,IAAA,KAAS,UAAU,OAAO,gBAAA;AAC9B,MAAA,OAAQ,OAAe,IAAI,CAAA;AAAA,IAC7B;AAAA,GACD,CAAA;AAED,EAAA,MAAM,SAAA,GAAY,IAAI,KAAA,CAAM,MAAA,CAAO,IAAA,EAAM;AAAA,IACvC,GAAA,CAAI,QAAQ,IAAA,EAAM;AAChB,MAAA,IAAI,IAAA,KAAS,eAAe,OAAO,gBAAA;AACnC,MAAA,OAAQ,OAAe,IAAI,CAAA;AAAA,IAC7B;AAAA,GACD,CAAA;AAED,EAAA,OAAO,IAAI,MAAM,MAAA,EAAQ;AAAA,IACvB,GAAA,CAAI,QAAQ,IAAA,EAAM;AAChB,MAAA,IAAI,IAAA,KAAS,QAAQ,OAAO,SAAA;AAC5B,MAAA,OAAQ,OAAe,IAAI,CAAA;AAAA,IAC7B;AAAA,GACD,CAAA;AACH","file":"openai.js","sourcesContent":["import { compress } from \"../compress.js\";\nimport type { CompressOptions, OpenAIMessage } from \"../types.js\";\n\n/* eslint-disable @typescript-eslint/no-explicit-any */\n\ninterface OpenAILike {\n  chat: {\n    completions: {\n      create: (params: any) => any;\n    };\n  };\n  [key: string]: any;\n}\n\n/**\n * Wrap an OpenAI client to auto-compress messages before each request.\n *\n * Intercepts `client.chat.completions.create()` only. All other methods\n * (embeddings, images, audio, etc.) pass through unchanged.\n *\n * @example\n * ```typescript\n * import { withHeadroom } from 'headroom-ai/openai';\n * import OpenAI from 'openai';\n *\n * const client = withHeadroom(new OpenAI());\n * const response = await client.chat.completions.create({\n *   model: 'gpt-4o',\n *   messages: longConversation,\n * });\n * ```\n */\nexport function withHeadroom<T extends OpenAILike>(\n  client: T,\n  options: CompressOptions = {},\n): T {\n  const originalCreate = client.chat.completions.create.bind(\n    client.chat.completions,\n  );\n\n  const compressedCreate = async (params: any) => {\n    const messages: OpenAIMessage[] = params.messages;\n    const model = options.model ?? params.model ?? \"gpt-4o\";\n\n    const result = await compress(messages, {\n      stack: \"adapter_ts_openai\",\n      ...options,\n      model,\n    });\n\n    return originalCreate({\n      ...params,\n      messages: result.messages,\n    });\n  };\n\n  const completionsProxy = new Proxy(client.chat.completions, {\n    get(target, prop) {\n      if (prop === \"create\") return compressedCreate;\n      return (target as any)[prop];\n    },\n  });\n\n  const chatProxy = new Proxy(client.chat, {\n    get(target, prop) {\n      if (prop === \"completions\") return completionsProxy;\n      return (target as any)[prop];\n    },\n  });\n\n  return new Proxy(client, {\n    get(target, prop) {\n      if (prop === \"chat\") return chatProxy;\n      return (target as any)[prop];\n    },\n  }) as T;\n}\n"]}

package/dist/adapters/vercel-ai.cjs

@@ -1,8 +1,9 @@
 'use strict';
 
-var chunk3N25JEWK_cjs = require('../chunk-3N25JEWK.cjs');
+var chunkR6FB52AD_cjs = require('../chunk-R6FB52AD.cjs');
+var module$1 = require('module');
 
-// src/adapters/vercel-ai.ts
+var _documentCurrentScript = typeof document !== 'undefined' ? document.currentScript : null;
 function headroomMiddleware(options = {}) {
   return {
     transformParams: async ({
@@ -11,25 +12,49 @@ function headroomMiddleware(options = {}) {
       const prompt = params.prompt;
       if (!prompt || prompt.length === 0) return params;
       const model = options.model ?? params.modelId ?? "gpt-4o";
-      const openaiMessages = chunk3N25JEWK_cjs.vercelToOpenAI(prompt);
-      const result = await chunk3N25JEWK_cjs.compress(openaiMessages, { ...options, model });
+      const openaiMessages = chunkR6FB52AD_cjs.vercelToOpenAI(prompt);
+      const result = await chunkR6FB52AD_cjs.compress(openaiMessages, {
+        stack: "adapter_ts_vercel_ai",
+        ...options,
+        model
+      });
       if (!result.compressed) return params;
-      const compressedPrompt = chunk3N25JEWK_cjs.openAIToVercel(result.messages);
+      const compressedPrompt = chunkR6FB52AD_cjs.openAIToVercel(result.messages);
       return { ...params, prompt: compressedPrompt };
     }
   };
 }
 async function compressVercelMessages(messages, options = {}) {
-  const openaiMessages = chunk3N25JEWK_cjs.vercelToOpenAI(messages);
-  const result = await chunk3N25JEWK_cjs.compress(openaiMessages, options);
-  const vercelMessages = chunk3N25JEWK_cjs.openAIToVercel(result.messages);
+  const openaiMessages = chunkR6FB52AD_cjs.vercelToOpenAI(messages);
+  const result = await chunkR6FB52AD_cjs.compress(openaiMessages, {
+    stack: "adapter_ts_vercel_ai",
+    ...options
+  });
+  const vercelMessages = chunkR6FB52AD_cjs.openAIToVercel(result.messages);
   return {
     ...result,
     messages: vercelMessages
   };
 }
+function withHeadroom(model, options = {}) {
+  let wrapLanguageModel;
+  try {
+    const require2 = module$1.createRequire((typeof document === 'undefined' ? require('u' + 'rl').pathToFileURL(__filename).href : (_documentCurrentScript && _documentCurrentScript.tagName.toUpperCase() === 'SCRIPT' && _documentCurrentScript.src || new URL('vercel-ai.cjs', document.baseURI).href)));
+    const ai = require2("ai");
+    wrapLanguageModel = ai.wrapLanguageModel;
+  } catch {
+    throw new Error(
+      'withHeadroom() requires the "ai" package. Install it with: npm install ai'
+    );
+  }
+  return wrapLanguageModel({
+    model,
+    middleware: headroomMiddleware(options)
+  });
+}
 
 exports.compressVercelMessages = compressVercelMessages;
 exports.headroomMiddleware = headroomMiddleware;
+exports.withHeadroom = withHeadroom;
 //# sourceMappingURL=vercel-ai.cjs.map
 //# sourceMappingURL=vercel-ai.cjs.map

package/dist/adapters/vercel-ai.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/adapters/vercel-ai.ts"],"names":["vercelToOpenAI","compress","openAIToVercel"],"mappings":";;;;;AAuBO,SAAS,kBAAA,CAAmB,OAAA,GAA2B,EAAC,EAAG;AAChE,EAAA,OAAO;AAAA,IACL,iBAAiB,OAAO;AAAA,MACtB;AAAA,KACF,KAIM;AACJ,MAAA,MAAM,SAA0B,MAAA,CAAO,MAAA;AACvC,MAAA,IAAI,CAAC,MAAA,IAAU,MAAA,CAAO,MAAA,KAAW,GAAG,OAAO,MAAA;AAE3C,MAAA,MAAM,KAAA,GAAQ,OAAA,CAAQ,KAAA,IAAS,MAAA,CAAO,OAAA,IAAW,QAAA;AAGjD,MAAA,MAAM,cAAA,GAAiBA,iCAAe,MAAM,CAAA;AAG5C,MAAA,MAAM,MAAA,GAAS,MAAMC,0BAAA,CAAS,cAAA,EAAgB,EAAE,GAAG,OAAA,EAAS,OAAO,CAAA;AAEnE,MAAA,IAAI,CAAC,MAAA,CAAO,UAAA,EAAY,OAAO,MAAA;AAG/B,MAAA,MAAM,gBAAA,GAAmBC,gCAAA,CAAe,MAAA,CAAO,QAAQ,CAAA;AAEvD,MAAA,OAAO,EAAE,GAAG,MAAA,EAAQ,MAAA,EAAQ,gBAAA,EAAiB;AAAA,IAC/C;AAAA,GACF;AACF;AAMA,eAAsB,sBAAA,CACpB,QAAA,EACA,OAAA,GAA2B,EAAC,EAC6B;AACzD,EAAA,MAAM,cAAA,GAAiBF,iCAAe,QAAQ,CAAA;AAC9C,EAAA,MAAM,MAAA,GAAS,MAAMC,0BAAA,CAAS,cAAA,EAAgB,OAAO,CAAA;AACrD,EAAA,MAAM,cAAA,GAAiBC,gCAAA,CAAe,MAAA,CAAO,QAAQ,CAAA;AAErD,EAAA,OAAO;AAAA,IACL,GAAG,MAAA;AAAA,IACH,QAAA,EAAU;AAAA,GACZ;AACF","file":"vercel-ai.cjs","sourcesContent":["import { compress } from \"../compress.js\";\nimport type { CompressOptions, CompressResult } from \"../types.js\";\nimport { vercelToOpenAI, openAIToVercel } from \"../utils/format.js\";\n\n/* eslint-disable @typescript-eslint/no-explicit-any */\ntype VercelMessage = any;\n\n/**\n * Vercel AI SDK LanguageModelV3Middleware that compresses messages\n * before they reach the LLM.\n *\n * @example\n * ```typescript\n * import { headroomMiddleware } from 'headroom-ai/vercel-ai';\n * import { wrapLanguageModel } from 'ai';\n * import { openai } from '@ai-sdk/openai';\n *\n * const model = wrapLanguageModel({\n *   model: openai('gpt-4o'),\n *   middleware: headroomMiddleware(),\n * });\n * ```\n */\nexport function headroomMiddleware(options: CompressOptions = {}) {\n  return {\n    transformParams: async ({\n      params,\n    }: {\n      params: any;\n      model: any;\n      type: string;\n    }) => {\n      const prompt: VercelMessage[] = params.prompt;\n      if (!prompt || prompt.length === 0) return params;\n\n      const model = options.model ?? params.modelId ?? \"gpt-4o\";\n\n      // Convert Vercel format → OpenAI format\n      const openaiMessages = vercelToOpenAI(prompt);\n\n      // Compress via Headroom\n      const result = await compress(openaiMessages, { ...options, model });\n\n      if (!result.compressed) return params;\n\n      // Convert back to Vercel format\n      const compressedPrompt = openAIToVercel(result.messages);\n\n      return { ...params, prompt: compressedPrompt };\n    },\n  };\n}\n\n/**\n * Standalone: compress Vercel AI SDK ModelMessage[] directly.\n * Returns compressed messages in Vercel format + compression stats.\n */\nexport async function compressVercelMessages(\n  messages: VercelMessage[],\n  options: CompressOptions = {},\n): Promise<CompressResult & { messages: VercelMessage[] }> {\n  const openaiMessages = vercelToOpenAI(messages);\n  const result = await compress(openaiMessages, options);\n  const vercelMessages = openAIToVercel(result.messages);\n\n  return {\n    ...result,\n    messages: vercelMessages,\n  };\n}\n"]}
+{"version":3,"sources":["../../src/adapters/vercel-ai.ts"],"names":["vercelToOpenAI","compress","openAIToVercel","require","createRequire"],"mappings":";;;;;;AAsCO,SAAS,kBAAA,CAAmB,OAAA,GAA2B,EAAC,EAAG;AAChE,EAAA,OAAO;AAAA,IACL,iBAAiB,OAAO;AAAA,MACtB;AAAA,KACF,KAIM;AACJ,MAAA,MAAM,SAA0B,MAAA,CAAO,MAAA;AACvC,MAAA,IAAI,CAAC,MAAA,IAAU,MAAA,CAAO,MAAA,KAAW,GAAG,OAAO,MAAA;AAE3C,MAAA,MAAM,KAAA,GAAQ,OAAA,CAAQ,KAAA,IAAS,MAAA,CAAO,OAAA,IAAW,QAAA;AAGjD,MAAA,MAAM,cAAA,GAAiBA,iCAAe,MAAM,CAAA;AAG5C,MAAA,MAAM,MAAA,GAAS,MAAMC,0BAAA,CAAS,cAAA,EAAgB;AAAA,QAC5C,KAAA,EAAO,sBAAA;AAAA,QACP,GAAG,OAAA;AAAA,QACH;AAAA,OACD,CAAA;AAED,MAAA,IAAI,CAAC,MAAA,CAAO,UAAA,EAAY,OAAO,MAAA;AAG/B,MAAA,MAAM,gBAAA,GAAmBC,gCAAA,CAAe,MAAA,CAAO,QAAQ,CAAA;AAEvD,MAAA,OAAO,EAAE,GAAG,MAAA,EAAQ,MAAA,EAAQ,gBAAA,EAAiB;AAAA,IAC/C;AAAA,GACF;AACF;AAMA,eAAsB,sBAAA,CACpB,QAAA,EACA,OAAA,GAA2B,EAAC,EAC6B;AACzD,EAAA,MAAM,cAAA,GAAiBF,iCAAe,QAAQ,CAAA;AAC9C,EAAA,MAAM,MAAA,GAAS,MAAMC,0BAAA,CAAS,cAAA,EAAgB;AAAA,IAC5C,KAAA,EAAO,sBAAA;AAAA,IACP,GAAG;AAAA,GACJ,CAAA;AACD,EAAA,MAAM,cAAA,GAAiBC,gCAAA,CAAe,MAAA,CAAO,QAAQ,CAAA;AAErD,EAAA,OAAO;AAAA,IACL,GAAG,MAAA;AAAA,IACH,QAAA,EAAU;AAAA,GACZ;AACF;AAgBO,SAAS,YAAA,CACd,KAAA,EACA,OAAA,GAA2B,EAAC,EACzB;AACH,EAAA,IAAI,iBAAA;AAKJ,EAAA,IAAI;AACF,IAAA,MAAMC,QAAAA,GAAUC,sBAAA,CAAc,+PAAe,CAAA;AAC7C,IAAA,MAAM,EAAA,GAAKD,SAAQ,IAAI,CAAA;AACvB,IAAA,iBAAA,GAAoB,EAAA,CAAG,iBAAA;AAAA,EACzB,CAAA,CAAA,MAAQ;AACN,IAAA,MAAM,IAAI,KAAA;AAAA,MACR;AAAA,KACF;AAAA,EACF;AAEA,EAAA,OAAO,iBAAA,CAAkB;AAAA,IACvB,KAAA;AAAA,IACA,UAAA,EAAY,mBAAmB,OAAO;AAAA,GACvC,CAAA;AACH","file":"vercel-ai.cjs","sourcesContent":["import { createRequire } from \"node:module\";\nimport { compress } from \"../compress.js\";\nimport type { CompressOptions, CompressResult } from \"../types.js\";\nimport { vercelToOpenAI, openAIToVercel } from \"../utils/format.js\";\n\n/**\n * Minimal structural type for Vercel AI SDK language models.\n * Compatible with both LanguageModelV1 (@ai-sdk/provider <=1.x)\n * and LanguageModelV3 (@ai-sdk/provider >=2.x).\n */\ninterface LanguageModel {\n  readonly specificationVersion: string;\n  readonly provider: string;\n  readonly modelId: string;\n  doGenerate: (...args: any[]) => any;\n  doStream: (...args: any[]) => any;\n  [key: string]: any;\n}\n\n/* eslint-disable @typescript-eslint/no-explicit-any */\ntype VercelMessage = any;\n\n/**\n * Vercel AI SDK LanguageModelV3Middleware that compresses messages\n * before they reach the LLM.\n *\n * @example\n * ```typescript\n * import { headroomMiddleware } from 'headroom-ai/vercel-ai';\n * import { wrapLanguageModel } from 'ai';\n * import { openai } from '@ai-sdk/openai';\n *\n * const model = wrapLanguageModel({\n *   model: openai('gpt-4o'),\n *   middleware: headroomMiddleware(),\n * });\n * ```\n */\nexport function headroomMiddleware(options: CompressOptions = {}) {\n  return {\n    transformParams: async ({\n      params,\n    }: {\n      params: any;\n      model: any;\n      type: string;\n    }) => {\n      const prompt: VercelMessage[] = params.prompt;\n      if (!prompt || prompt.length === 0) return params;\n\n      const model = options.model ?? params.modelId ?? \"gpt-4o\";\n\n      // Convert Vercel format → OpenAI format\n      const openaiMessages = vercelToOpenAI(prompt);\n\n      // Compress via Headroom\n      const result = await compress(openaiMessages, {\n        stack: \"adapter_ts_vercel_ai\",\n        ...options,\n        model,\n      });\n\n      if (!result.compressed) return params;\n\n      // Convert back to Vercel format\n      const compressedPrompt = openAIToVercel(result.messages);\n\n      return { ...params, prompt: compressedPrompt };\n    },\n  };\n}\n\n/**\n * Standalone: compress Vercel AI SDK ModelMessage[] directly.\n * Returns compressed messages in Vercel format + compression stats.\n */\nexport async function compressVercelMessages(\n  messages: VercelMessage[],\n  options: CompressOptions = {},\n): Promise<CompressResult & { messages: VercelMessage[] }> {\n  const openaiMessages = vercelToOpenAI(messages);\n  const result = await compress(openaiMessages, {\n    stack: \"adapter_ts_vercel_ai\",\n    ...options,\n  });\n  const vercelMessages = openAIToVercel(result.messages);\n\n  return {\n    ...result,\n    messages: vercelMessages,\n  };\n}\n\n/**\n * Wrap a Vercel AI SDK language model with Headroom compression.\n * Convenience wrapper around `wrapLanguageModel` + `headroomMiddleware`.\n *\n * @example\n * ```typescript\n * import { withHeadroom } from 'headroom-ai/vercel-ai';\n * import { openai } from '@ai-sdk/openai';\n * import { generateText } from 'ai';\n *\n * const model = withHeadroom(openai('gpt-4o'));\n * const { text } = await generateText({ model, messages });\n * ```\n */\nexport function withHeadroom<T extends LanguageModel>(\n  model: T,\n  options: CompressOptions = {},\n): T {\n  let wrapLanguageModel: (opts: {\n    model: T;\n    middleware: any;\n  }) => T;\n\n  try {\n    const require = createRequire(import.meta.url);\n    const ai = require(\"ai\");\n    wrapLanguageModel = ai.wrapLanguageModel;\n  } catch {\n    throw new Error(\n      'withHeadroom() requires the \"ai\" package. Install it with: npm install ai',\n    );\n  }\n\n  return wrapLanguageModel({\n    model,\n    middleware: headroomMiddleware(options),\n  });\n}\n"]}

package/dist/adapters/vercel-ai.d.cts

@@ -1,5 +1,18 @@
-import { C as CompressOptions, a as CompressResult } from '../types-DQtcLXq3.cjs';
+import { C as CompressOptions, a as CompressResult } from '../types-BTrX7__W.cjs';
 
+/**
+ * Minimal structural type for Vercel AI SDK language models.
+ * Compatible with both LanguageModelV1 (@ai-sdk/provider <=1.x)
+ * and LanguageModelV3 (@ai-sdk/provider >=2.x).
+ */
+interface LanguageModel {
+    readonly specificationVersion: string;
+    readonly provider: string;
+    readonly modelId: string;
+    doGenerate: (...args: any[]) => any;
+    doStream: (...args: any[]) => any;
+    [key: string]: any;
+}
 type VercelMessage = any;
 /**
  * Vercel AI SDK LanguageModelV3Middleware that compresses messages
@@ -31,5 +44,20 @@ declare function headroomMiddleware(options?: CompressOptions): {
 declare function compressVercelMessages(messages: VercelMessage[], options?: CompressOptions): Promise<CompressResult & {
     messages: VercelMessage[];
 }>;
+/**
+ * Wrap a Vercel AI SDK language model with Headroom compression.
+ * Convenience wrapper around `wrapLanguageModel` + `headroomMiddleware`.
+ *
+ * @example
+ * ```typescript
+ * import { withHeadroom } from 'headroom-ai/vercel-ai';
+ * import { openai } from '@ai-sdk/openai';
+ * import { generateText } from 'ai';
+ *
+ * const model = withHeadroom(openai('gpt-4o'));
+ * const { text } = await generateText({ model, messages });
+ * ```
+ */
+declare function withHeadroom<T extends LanguageModel>(model: T, options?: CompressOptions): T;
 
-export { compressVercelMessages, headroomMiddleware };
+export { compressVercelMessages, headroomMiddleware, withHeadroom };

package/dist/adapters/vercel-ai.d.ts

@@ -1,5 +1,18 @@
-import { C as CompressOptions, a as CompressResult } from '../types-DQtcLXq3.js';
+import { C as CompressOptions, a as CompressResult } from '../types-BTrX7__W.js';
 
+/**
+ * Minimal structural type for Vercel AI SDK language models.
+ * Compatible with both LanguageModelV1 (@ai-sdk/provider <=1.x)
+ * and LanguageModelV3 (@ai-sdk/provider >=2.x).
+ */
+interface LanguageModel {
+    readonly specificationVersion: string;
+    readonly provider: string;
+    readonly modelId: string;
+    doGenerate: (...args: any[]) => any;
+    doStream: (...args: any[]) => any;
+    [key: string]: any;
+}
 type VercelMessage = any;
 /**
  * Vercel AI SDK LanguageModelV3Middleware that compresses messages
@@ -31,5 +44,20 @@ declare function headroomMiddleware(options?: CompressOptions): {
 declare function compressVercelMessages(messages: VercelMessage[], options?: CompressOptions): Promise<CompressResult & {
     messages: VercelMessage[];
 }>;
+/**
+ * Wrap a Vercel AI SDK language model with Headroom compression.
+ * Convenience wrapper around `wrapLanguageModel` + `headroomMiddleware`.
+ *
+ * @example
+ * ```typescript
+ * import { withHeadroom } from 'headroom-ai/vercel-ai';
+ * import { openai } from '@ai-sdk/openai';
+ * import { generateText } from 'ai';
+ *
+ * const model = withHeadroom(openai('gpt-4o'));
+ * const { text } = await generateText({ model, messages });
+ * ```
+ */
+declare function withHeadroom<T extends LanguageModel>(model: T, options?: CompressOptions): T;
 
-export { compressVercelMessages, headroomMiddleware };
+export { compressVercelMessages, headroomMiddleware, withHeadroom };

package/dist/adapters/vercel-ai.js

@@ -1,6 +1,6 @@
-import { vercelToOpenAI, compress, openAIToVercel } from '../chunk-YTTW7S2Q.js';
+import { vercelToOpenAI, compress, openAIToVercel } from '../chunk-7AJIWIPP.js';
+import { createRequire } from 'module';
 
-// src/adapters/vercel-ai.ts
 function headroomMiddleware(options = {}) {
   return {
     transformParams: async ({
@@ -10,7 +10,11 @@ function headroomMiddleware(options = {}) {
       if (!prompt || prompt.length === 0) return params;
       const model = options.model ?? params.modelId ?? "gpt-4o";
       const openaiMessages = vercelToOpenAI(prompt);
-      const result = await compress(openaiMessages, { ...options, model });
+      const result = await compress(openaiMessages, {
+        stack: "adapter_ts_vercel_ai",
+        ...options,
+        model
+      });
       if (!result.compressed) return params;
       const compressedPrompt = openAIToVercel(result.messages);
       return { ...params, prompt: compressedPrompt };
@@ -19,14 +23,33 @@ function headroomMiddleware(options = {}) {
 }
 async function compressVercelMessages(messages, options = {}) {
   const openaiMessages = vercelToOpenAI(messages);
-  const result = await compress(openaiMessages, options);
+  const result = await compress(openaiMessages, {
+    stack: "adapter_ts_vercel_ai",
+    ...options
+  });
   const vercelMessages = openAIToVercel(result.messages);
   return {
     ...result,
     messages: vercelMessages
   };
 }
+function withHeadroom(model, options = {}) {
+  let wrapLanguageModel;
+  try {
+    const require2 = createRequire(import.meta.url);
+    const ai = require2("ai");
+    wrapLanguageModel = ai.wrapLanguageModel;
+  } catch {
+    throw new Error(
+      'withHeadroom() requires the "ai" package. Install it with: npm install ai'
+    );
+  }
+  return wrapLanguageModel({
+    model,
+    middleware: headroomMiddleware(options)
+  });
+}
 
-export { compressVercelMessages, headroomMiddleware };
+export { compressVercelMessages, headroomMiddleware, withHeadroom };
 //# sourceMappingURL=vercel-ai.js.map
 //# sourceMappingURL=vercel-ai.js.map

package/dist/adapters/vercel-ai.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/adapters/vercel-ai.ts"],"names":[],"mappings":";;;AAuBO,SAAS,kBAAA,CAAmB,OAAA,GAA2B,EAAC,EAAG;AAChE,EAAA,OAAO;AAAA,IACL,iBAAiB,OAAO;AAAA,MACtB;AAAA,KACF,KAIM;AACJ,MAAA,MAAM,SAA0B,MAAA,CAAO,MAAA;AACvC,MAAA,IAAI,CAAC,MAAA,IAAU,MAAA,CAAO,MAAA,KAAW,GAAG,OAAO,MAAA;AAE3C,MAAA,MAAM,KAAA,GAAQ,OAAA,CAAQ,KAAA,IAAS,MAAA,CAAO,OAAA,IAAW,QAAA;AAGjD,MAAA,MAAM,cAAA,GAAiB,eAAe,MAAM,CAAA;AAG5C,MAAA,MAAM,MAAA,GAAS,MAAM,QAAA,CAAS,cAAA,EAAgB,EAAE,GAAG,OAAA,EAAS,OAAO,CAAA;AAEnE,MAAA,IAAI,CAAC,MAAA,CAAO,UAAA,EAAY,OAAO,MAAA;AAG/B,MAAA,MAAM,gBAAA,GAAmB,cAAA,CAAe,MAAA,CAAO,QAAQ,CAAA;AAEvD,MAAA,OAAO,EAAE,GAAG,MAAA,EAAQ,MAAA,EAAQ,gBAAA,EAAiB;AAAA,IAC/C;AAAA,GACF;AACF;AAMA,eAAsB,sBAAA,CACpB,QAAA,EACA,OAAA,GAA2B,EAAC,EAC6B;AACzD,EAAA,MAAM,cAAA,GAAiB,eAAe,QAAQ,CAAA;AAC9C,EAAA,MAAM,MAAA,GAAS,MAAM,QAAA,CAAS,cAAA,EAAgB,OAAO,CAAA;AACrD,EAAA,MAAM,cAAA,GAAiB,cAAA,CAAe,MAAA,CAAO,QAAQ,CAAA;AAErD,EAAA,OAAO;AAAA,IACL,GAAG,MAAA;AAAA,IACH,QAAA,EAAU;AAAA,GACZ;AACF","file":"vercel-ai.js","sourcesContent":["import { compress } from \"../compress.js\";\nimport type { CompressOptions, CompressResult } from \"../types.js\";\nimport { vercelToOpenAI, openAIToVercel } from \"../utils/format.js\";\n\n/* eslint-disable @typescript-eslint/no-explicit-any */\ntype VercelMessage = any;\n\n/**\n * Vercel AI SDK LanguageModelV3Middleware that compresses messages\n * before they reach the LLM.\n *\n * @example\n * ```typescript\n * import { headroomMiddleware } from 'headroom-ai/vercel-ai';\n * import { wrapLanguageModel } from 'ai';\n * import { openai } from '@ai-sdk/openai';\n *\n * const model = wrapLanguageModel({\n *   model: openai('gpt-4o'),\n *   middleware: headroomMiddleware(),\n * });\n * ```\n */\nexport function headroomMiddleware(options: CompressOptions = {}) {\n  return {\n    transformParams: async ({\n      params,\n    }: {\n      params: any;\n      model: any;\n      type: string;\n    }) => {\n      const prompt: VercelMessage[] = params.prompt;\n      if (!prompt || prompt.length === 0) return params;\n\n      const model = options.model ?? params.modelId ?? \"gpt-4o\";\n\n      // Convert Vercel format → OpenAI format\n      const openaiMessages = vercelToOpenAI(prompt);\n\n      // Compress via Headroom\n      const result = await compress(openaiMessages, { ...options, model });\n\n      if (!result.compressed) return params;\n\n      // Convert back to Vercel format\n      const compressedPrompt = openAIToVercel(result.messages);\n\n      return { ...params, prompt: compressedPrompt };\n    },\n  };\n}\n\n/**\n * Standalone: compress Vercel AI SDK ModelMessage[] directly.\n * Returns compressed messages in Vercel format + compression stats.\n */\nexport async function compressVercelMessages(\n  messages: VercelMessage[],\n  options: CompressOptions = {},\n): Promise<CompressResult & { messages: VercelMessage[] }> {\n  const openaiMessages = vercelToOpenAI(messages);\n  const result = await compress(openaiMessages, options);\n  const vercelMessages = openAIToVercel(result.messages);\n\n  return {\n    ...result,\n    messages: vercelMessages,\n  };\n}\n"]}
+{"version":3,"sources":["../../src/adapters/vercel-ai.ts"],"names":["require"],"mappings":";;;AAsCO,SAAS,kBAAA,CAAmB,OAAA,GAA2B,EAAC,EAAG;AAChE,EAAA,OAAO;AAAA,IACL,iBAAiB,OAAO;AAAA,MACtB;AAAA,KACF,KAIM;AACJ,MAAA,MAAM,SAA0B,MAAA,CAAO,MAAA;AACvC,MAAA,IAAI,CAAC,MAAA,IAAU,MAAA,CAAO,MAAA,KAAW,GAAG,OAAO,MAAA;AAE3C,MAAA,MAAM,KAAA,GAAQ,OAAA,CAAQ,KAAA,IAAS,MAAA,CAAO,OAAA,IAAW,QAAA;AAGjD,MAAA,MAAM,cAAA,GAAiB,eAAe,MAAM,CAAA;AAG5C,MAAA,MAAM,MAAA,GAAS,MAAM,QAAA,CAAS,cAAA,EAAgB;AAAA,QAC5C,KAAA,EAAO,sBAAA;AAAA,QACP,GAAG,OAAA;AAAA,QACH;AAAA,OACD,CAAA;AAED,MAAA,IAAI,CAAC,MAAA,CAAO,UAAA,EAAY,OAAO,MAAA;AAG/B,MAAA,MAAM,gBAAA,GAAmB,cAAA,CAAe,MAAA,CAAO,QAAQ,CAAA;AAEvD,MAAA,OAAO,EAAE,GAAG,MAAA,EAAQ,MAAA,EAAQ,gBAAA,EAAiB;AAAA,IAC/C;AAAA,GACF;AACF;AAMA,eAAsB,sBAAA,CACpB,QAAA,EACA,OAAA,GAA2B,EAAC,EAC6B;AACzD,EAAA,MAAM,cAAA,GAAiB,eAAe,QAAQ,CAAA;AAC9C,EAAA,MAAM,MAAA,GAAS,MAAM,QAAA,CAAS,cAAA,EAAgB;AAAA,IAC5C,KAAA,EAAO,sBAAA;AAAA,IACP,GAAG;AAAA,GACJ,CAAA;AACD,EAAA,MAAM,cAAA,GAAiB,cAAA,CAAe,MAAA,CAAO,QAAQ,CAAA;AAErD,EAAA,OAAO;AAAA,IACL,GAAG,MAAA;AAAA,IACH,QAAA,EAAU;AAAA,GACZ;AACF;AAgBO,SAAS,YAAA,CACd,KAAA,EACA,OAAA,GAA2B,EAAC,EACzB;AACH,EAAA,IAAI,iBAAA;AAKJ,EAAA,IAAI;AACF,IAAA,MAAMA,QAAAA,GAAU,aAAA,CAAc,MAAA,CAAA,IAAA,CAAY,GAAG,CAAA;AAC7C,IAAA,MAAM,EAAA,GAAKA,SAAQ,IAAI,CAAA;AACvB,IAAA,iBAAA,GAAoB,EAAA,CAAG,iBAAA;AAAA,EACzB,CAAA,CAAA,MAAQ;AACN,IAAA,MAAM,IAAI,KAAA;AAAA,MACR;AAAA,KACF;AAAA,EACF;AAEA,EAAA,OAAO,iBAAA,CAAkB;AAAA,IACvB,KAAA;AAAA,IACA,UAAA,EAAY,mBAAmB,OAAO;AAAA,GACvC,CAAA;AACH","file":"vercel-ai.js","sourcesContent":["import { createRequire } from \"node:module\";\nimport { compress } from \"../compress.js\";\nimport type { CompressOptions, CompressResult } from \"../types.js\";\nimport { vercelToOpenAI, openAIToVercel } from \"../utils/format.js\";\n\n/**\n * Minimal structural type for Vercel AI SDK language models.\n * Compatible with both LanguageModelV1 (@ai-sdk/provider <=1.x)\n * and LanguageModelV3 (@ai-sdk/provider >=2.x).\n */\ninterface LanguageModel {\n  readonly specificationVersion: string;\n  readonly provider: string;\n  readonly modelId: string;\n  doGenerate: (...args: any[]) => any;\n  doStream: (...args: any[]) => any;\n  [key: string]: any;\n}\n\n/* eslint-disable @typescript-eslint/no-explicit-any */\ntype VercelMessage = any;\n\n/**\n * Vercel AI SDK LanguageModelV3Middleware that compresses messages\n * before they reach the LLM.\n *\n * @example\n * ```typescript\n * import { headroomMiddleware } from 'headroom-ai/vercel-ai';\n * import { wrapLanguageModel } from 'ai';\n * import { openai } from '@ai-sdk/openai';\n *\n * const model = wrapLanguageModel({\n *   model: openai('gpt-4o'),\n *   middleware: headroomMiddleware(),\n * });\n * ```\n */\nexport function headroomMiddleware(options: CompressOptions = {}) {\n  return {\n    transformParams: async ({\n      params,\n    }: {\n      params: any;\n      model: any;\n      type: string;\n    }) => {\n      const prompt: VercelMessage[] = params.prompt;\n      if (!prompt || prompt.length === 0) return params;\n\n      const model = options.model ?? params.modelId ?? \"gpt-4o\";\n\n      // Convert Vercel format → OpenAI format\n      const openaiMessages = vercelToOpenAI(prompt);\n\n      // Compress via Headroom\n      const result = await compress(openaiMessages, {\n        stack: \"adapter_ts_vercel_ai\",\n        ...options,\n        model,\n      });\n\n      if (!result.compressed) return params;\n\n      // Convert back to Vercel format\n      const compressedPrompt = openAIToVercel(result.messages);\n\n      return { ...params, prompt: compressedPrompt };\n    },\n  };\n}\n\n/**\n * Standalone: compress Vercel AI SDK ModelMessage[] directly.\n * Returns compressed messages in Vercel format + compression stats.\n */\nexport async function compressVercelMessages(\n  messages: VercelMessage[],\n  options: CompressOptions = {},\n): Promise<CompressResult & { messages: VercelMessage[] }> {\n  const openaiMessages = vercelToOpenAI(messages);\n  const result = await compress(openaiMessages, {\n    stack: \"adapter_ts_vercel_ai\",\n    ...options,\n  });\n  const vercelMessages = openAIToVercel(result.messages);\n\n  return {\n    ...result,\n    messages: vercelMessages,\n  };\n}\n\n/**\n * Wrap a Vercel AI SDK language model with Headroom compression.\n * Convenience wrapper around `wrapLanguageModel` + `headroomMiddleware`.\n *\n * @example\n * ```typescript\n * import { withHeadroom } from 'headroom-ai/vercel-ai';\n * import { openai } from '@ai-sdk/openai';\n * import { generateText } from 'ai';\n *\n * const model = withHeadroom(openai('gpt-4o'));\n * const { text } = await generateText({ model, messages });\n * ```\n */\nexport function withHeadroom<T extends LanguageModel>(\n  model: T,\n  options: CompressOptions = {},\n): T {\n  let wrapLanguageModel: (opts: {\n    model: T;\n    middleware: any;\n  }) => T;\n\n  try {\n    const require = createRequire(import.meta.url);\n    const ai = require(\"ai\");\n    wrapLanguageModel = ai.wrapLanguageModel;\n  } catch {\n    throw new Error(\n      'withHeadroom() requires the \"ai\" package. Install it with: npm install ai',\n    );\n  }\n\n  return wrapLanguageModel({\n    model,\n    middleware: headroomMiddleware(options),\n  });\n}\n"]}