the-token-company 0.3.0 → 0.3.1

package/dist/anthropic.js

@@ -101,7 +101,10 @@ export function withCompression(client, options) {
         if (params?.messages) {
             params = {
                 ...params,
-                messages: await compressAnthropicMessages(compressor, params.messages, model, roleAggr, { stripServerToolResults }),
+                messages: await compressAnthropicMessages(compressor, params.messages, model, roleAggr, {
+                    stripServerToolResults,
+                    skipToolName: webSearch ? "ttc_web_search" : undefined,
+                }),
             };
         }
         if (systemAggr != null && typeof params?.system === "string" && params.system.trim()) {

package/dist/compress.d.ts

@@ -30,6 +30,7 @@ export declare function compressOpenAIMessages(ttc: Compressor, messages: OpenAI
 interface AnthropicBlock {
     type: string;
     text?: string;
+    tool_use_id?: string;
     content?: string | Array<{
         type: string;
         text?: string;
@@ -44,6 +45,7 @@ interface AnthropicMessage {
 }
 export declare function compressAnthropicMessages(ttc: Compressor, messages: AnthropicMessage[], model: string, roleAggr: Record<string, number>, options?: {
     stripServerToolResults?: boolean;
+    skipToolName?: string;
 }): Promise<AnthropicMessage[]>;
 interface AISDKTextPart {
     type: "text";

package/dist/compress.js

@@ -47,8 +47,22 @@ export async function compressOpenAIMessages(ttc, messages, model, roleAggr) {
         return msg;
     }));
 }
+function collectToolUseIds(messages, toolName) {
+    const ids = new Set();
+    for (const msg of messages) {
+        if (msg.role !== "assistant" || !Array.isArray(msg.content))
+            continue;
+        for (const block of msg.content) {
+            if (block.type === "tool_use" && block.name === toolName && block.id) {
+                ids.add(block.id);
+            }
+        }
+    }
+    return ids;
+}
 export async function compressAnthropicMessages(ttc, messages, model, roleAggr, options) {
     const stripServerToolResults = options?.stripServerToolResults ?? false;
+    const skipIds = options?.skipToolName ? collectToolUseIds(messages, options.skipToolName) : undefined;
     return Promise.all(messages.map(async (msg) => {
         if (msg.role === "assistant") {
             const assistantAggr = roleAggr["assistant"];
@@ -75,7 +89,7 @@ export async function compressAnthropicMessages(ttc, messages, model, roleAggr,
             return { ...msg, content: result.output };
         }
         if (Array.isArray(msg.content)) {
-            const blocks = await compressAnthropicBlocks(ttc, msg.content, model, userAggr, toolAggr);
+            const blocks = await compressAnthropicBlocks(ttc, msg.content, model, userAggr, toolAggr, skipIds);
             return { ...msg, content: blocks };
         }
         return msg;
@@ -95,13 +109,15 @@ async function compressAssistantBlocks(ttc, blocks, model, assistantAggr, stripS
     const filtered = results.filter((b) => b != null);
     return filtered.length > 0 ? filtered : blocks;
 }
-async function compressAnthropicBlocks(ttc, blocks, model, userAggr, toolAggr) {
+async function compressAnthropicBlocks(ttc, blocks, model, userAggr, toolAggr, skipIds) {
     return Promise.all(blocks.map(async (block) => {
         if (block.type === "text" && userAggr != null && typeof block.text === "string" && block.text.trim()) {
             const result = await ttc.compress(block.text, { model, aggressiveness: userAggr });
             return { ...block, text: result.output };
         }
         if (block.type === "tool_result" && toolAggr != null) {
+            if (block.tool_use_id && skipIds?.has(block.tool_use_id))
+                return block;
             return compressToolResult(ttc, block, model, toolAggr);
         }
         return block;

package/package.json

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