@context-chef/ai-sdk-middleware 1.0.7 → 1.1.0

package/README.md

@@ -100,35 +100,29 @@ const model = withContextChef(openai('gpt-4o'), {
 
 The middleware automatically extracts token usage from `generateText` and `streamText` responses and feeds it back to the compression engine. No manual `reportTokenUsage()` calls needed.
 
-### Compact (Mechanical Clearing)
+### Compact (Mechanical Pruning)
 
-Zero-LLM-cost content clearing for thinking blocks and tool results:
+Zero-LLM-cost message pruning via AI SDK's `pruneMessages` — removes reasoning, tool calls, and empty messages:
 
 ```typescript
 const model = withContextChef(openai('gpt-4o'), {
   contextWindow: 128_000,
   compact: {
-    clear: ['thinking', { target: 'tool-result', keepRecent: 5 }],
+    reasoning: 'all',                          // Remove all reasoning
+    toolCalls: 'before-last-message',          // Keep tools only in the last message
   },
 });
 ```
 
-> **Important: compact + compress interaction**
->
-> When using `compact` together with `compress`, only clear `thinking` in compact:
->
-> ```typescript
-> const model = withContextChef(openai('gpt-4o'), {
->   contextWindow: 128_000,
->   compact: { clear: ['thinking'] },                // thinking only
->   compress: { model: openai('gpt-4o-mini') },
-> });
-> ```
->
-> Clearing `tool-result` before compression causes the compression model to receive
-> empty placeholders instead of actual tool outputs, producing low-quality summaries.
-> Compression's turn-based splitting already manages history length — use `compact`
-> for `tool-result` clearing only when `compress` is **not** configured.
+Per-tool control is also supported:
+
+```typescript
+compact: {
+  toolCalls: [
+    { type: 'before-last-message', tools: ['search', 'calculator'] },
+  ],
+}
+```
 
 ## API
 
@@ -155,7 +149,7 @@ const wrappedModel = withContextChef(model, options);
 | `truncate.headChars` | `number` | No | Characters to preserve from start (default: `0`) |
 | `truncate.tailChars` | `number` | No | Characters to preserve from end (default: `1000`) |
 | `truncate.storage` | `VFSStorageAdapter` | No | Storage adapter to persist original content before truncation |
-| `compact` | `CompactConfig` | No | Mechanical content clearing (thinking, tool-result). When combined with `compress`, use `clear: ['thinking']` only |
+| `compact` | `CompactConfig` | No | Mechanical message pruning (reasoning, tool calls). Delegates to AI SDK's `pruneMessages` |
 | `tokenizer` | `(msgs) => number` | No | Custom tokenizer for precise counting |
 | `onCompress` | `(summary, count) => void` | No | Hook called after compression |
 

package/README.zh-CN.md

@@ -102,34 +102,29 @@ const model = withContextChef(openai('gpt-4o'), {
 
 中间件自动从 `generateText` 和 `streamText` 响应中提取 token 用量，并回传给压缩引擎。无需手动调用 `reportTokenUsage()`。
 
-### Compact（机械清理）
+### Compact（机械裁剪）
 
-零 LLM 成本的 thinking 块和工具结果清理：
+零 LLM 成本的消息裁剪，基于 AI SDK 的 `pruneMessages` — 移除 reasoning、工具调用和空消息：
 
 ```typescript
 const model = withContextChef(openai('gpt-4o'), {
   contextWindow: 128_000,
   compact: {
-    clear: ['thinking', { target: 'tool-result', keepRecent: 5 }],
+    reasoning: 'all',                          // 移除所有 reasoning
+    toolCalls: 'before-last-message',          // 仅保留最后一条消息中的工具调用
   },
 });
 ```
 
-> **注意：compact + compress 交互**
->
-> 同时使用 `compact` 和 `compress` 时，compact 中仅清理 `thinking`：
->
-> ```typescript
-> const model = withContextChef(openai('gpt-4o'), {
->   contextWindow: 128_000,
->   compact: { clear: ['thinking'] },                // 仅 thinking
->   compress: { model: openai('gpt-4o-mini') },
-> });
-> ```
->
-> 在压缩之前清理 `tool-result` 会导致压缩模型收到空占位符而非实际工具输出，
-> 产生低质量摘要。压缩的 turn 分组机制已经管理了历史长度 — 仅在**未配置**
-> `compress` 时使用 `compact` 清理 `tool-result`。
+也支持按工具名称精细控制：
+
+```typescript
+compact: {
+  toolCalls: [
+    { type: 'before-last-message', tools: ['search', 'calculator'] },
+  ],
+}
+```
 
 ## API
 
@@ -156,7 +151,7 @@ const wrappedModel = withContextChef(model, options);
 | `truncate.headChars` | `number` | 否 | 保留开头的字符数（默认：`0`） |
 | `truncate.tailChars` | `number` | 否 | 保留结尾的字符数（默认：`1000`） |
 | `truncate.storage` | `VFSStorageAdapter` | 否 | 截断前持久化原始内容的存储适配器 |
-| `compact` | `CompactConfig` | 否 | 机械内容清理（thinking、tool-result）。与 `compress` 同时使用时，仅使用 `clear: ['thinking']` |
+| `compact` | `CompactConfig` | 否 | 机械消息裁剪（reasoning、工具调用）。委托给 AI SDK 的 `pruneMessages` |
 | `tokenizer` | `(msgs) => number` | 否 | 自定义分词器用于精确计数 |
 | `onCompress` | `(summary, count) => void` | 否 | 压缩完成后的回调 |
 

package/dist/index.cjs

@@ -269,23 +269,11 @@ function createMiddleware(options) {
 		transformParams: async ({ params }) => {
 			let { prompt } = params;
 			if (options.truncate) prompt = await truncateToolResults(prompt, options.truncate);
+			if (options.compact) prompt = compactPrompt(prompt, options.compact);
 			const allIR = fromAISDK(prompt);
-			let systemMessages = allIR.filter((m) => m.role === "system");
+			const systemMessages = allIR.filter((m) => m.role === "system");
 			let conversation = allIR.filter((m) => m.role !== "system");
-			if (options.compact) {
-				const preCompact = conversation;
-				conversation = janitor.compact(conversation, options.compact);
-				if (options.compact.clear.includes("thinking")) {
-					for (let i = 0; i < conversation.length; i++) if (preCompact[i].thinking && !conversation[i].thinking) delete conversation[i]._assistantContent;
-				}
-			}
 			conversation = await janitor.compress(conversation);
-			if (options.compact) {
-				if (options.compact.clear.some((t) => t === "tool-result" || typeof t === "object" && t.target === "tool-result")) systemMessages = [...systemMessages, {
-					role: "system",
-					content: _context_chef_core.Prompts.TOOL_RESULT_CLEARED_INSTRUCTION
-				}];
-			}
 			prompt = toAISDK([...systemMessages, ...conversation]);
 			if (options.dynamicState) prompt = await injectDynamicState(prompt, options.dynamicState);
 			if (options.transformContext) prompt = await options.transformContext(prompt);
@@ -323,6 +311,29 @@ function createMiddleware(options) {
 	};
 }
 /**
+* Prunes a LanguageModelV3Prompt via AI SDK's pruneMessages.
+*
+* LanguageModelV3Message (from @ai-sdk/provider) and ModelMessage
+* (from @ai-sdk/provider-utils) share identical runtime structure but
+* differ at the TypeScript level (e.g. ImagePart, FilePart.data).
+* Since pruneMessages only filters — never transforms — every content
+* part in the output is an original V3 part, making the casts safe.
+*/
+function compactPrompt(prompt, config) {
+	return (0, ai.pruneMessages)({
+		messages: prompt.map((msg) => ({
+			role: msg.role,
+			content: msg.content,
+			providerOptions: msg.providerOptions
+		})),
+		...config
+	}).map((msg) => ({
+		role: msg.role,
+		content: msg.content,
+		providerOptions: msg.providerOptions
+	}));
+}
+/**
 * Injects dynamic state XML into the AI SDK prompt.
 *
 * - `last_user`: Appends to the last user message's content parts.

package/dist/index.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"index.cjs","names":["Offloader","Janitor","Prompts","XmlGenerator"],"sources":["../src/adapter.ts","../src/truncator.ts","../src/middleware.ts","../src/index.ts"],"sourcesContent":["import type {\n  LanguageModelV3Message,\n  LanguageModelV3Prompt,\n  LanguageModelV3ToolResultOutput,\n  LanguageModelV3ToolResultPart,\n  SharedV3ProviderOptions,\n} from '@ai-sdk/provider';\nimport type { Message, ToolCall } from '@context-chef/core';\n\n/** Content types for each AI SDK message role */\ntype UserContent = Extract<LanguageModelV3Message, { role: 'user' }>['content'];\ntype AssistantContent = Extract<LanguageModelV3Message, { role: 'assistant' }>['content'];\ntype ToolContent = Extract<LanguageModelV3Message, { role: 'tool' }>['content'];\n\n/**\n * Extended IR message with typed pass-through fields for lossless AI SDK round-trip.\n * Per-role content fields avoid union types, so no `as` casts are needed in `toAISDK`.\n */\nexport interface AISDKMessage extends Message {\n  _userContent?: UserContent;\n  _assistantContent?: AssistantContent;\n  _toolContent?: ToolContent;\n  _originalText?: string;\n  _providerOptions?: SharedV3ProviderOptions;\n  _toolName?: string;\n}\n\n/**\n * Converts an AI SDK V3 prompt to context-chef IR messages.\n *\n * Original AI SDK content is stored in per-role fields for lossless round-trip.\n * `_originalText` caches the extracted text so `toAISDK` can detect Janitor modifications.\n * `_providerOptions` preserves message-level provider options (e.g. Anthropic cache control).\n */\nexport function fromAISDK(prompt: LanguageModelV3Prompt): AISDKMessage[] {\n  const messages: AISDKMessage[] = [];\n\n  for (const msg of prompt) {\n    if (msg.role === 'system') {\n      messages.push({\n        role: 'system',\n        content: msg.content,\n        ...(msg.providerOptions ? { _providerOptions: msg.providerOptions } : {}),\n      });\n      continue;\n    }\n\n    if (msg.role === 'user') {\n      const text = msg.content\n        .filter((p) => p.type === 'text')\n        .map((p) => p.text)\n        .join('\\n');\n      messages.push({\n        role: 'user',\n        content: text,\n        _userContent: msg.content,\n        _originalText: text,\n        ...(msg.providerOptions ? { _providerOptions: msg.providerOptions } : {}),\n      });\n      continue;\n    }\n\n    if (msg.role === 'assistant') {\n      const text: string[] = [];\n      const toolCalls: ToolCall[] = [];\n      let thinking: { thinking: string } | undefined;\n\n      for (const part of msg.content) {\n        if (part.type === 'text') text.push(part.text);\n        else if (part.type === 'tool-call') {\n          toolCalls.push({\n            id: part.toolCallId,\n            type: 'function',\n            function: {\n              name: part.toolName,\n              arguments: typeof part.input === 'string' ? part.input : JSON.stringify(part.input),\n            },\n          });\n        } else if (part.type === 'reasoning') {\n          thinking = { thinking: part.text };\n        }\n      }\n\n      const joinedText = text.join('\\n');\n      const m: AISDKMessage = {\n        role: 'assistant',\n        content: joinedText,\n        _assistantContent: msg.content,\n        _originalText: joinedText,\n        ...(msg.providerOptions ? { _providerOptions: msg.providerOptions } : {}),\n      };\n      if (toolCalls.length > 0) m.tool_calls = toolCalls;\n      if (thinking) m.thinking = thinking;\n      messages.push(m);\n      continue;\n    }\n\n    if (msg.role === 'tool') {\n      for (const part of msg.content) {\n        if (part.type === 'tool-result') {\n          const text = stringifyToolOutput(part.output);\n          messages.push({\n            role: 'tool',\n            content: text,\n            tool_call_id: part.toolCallId,\n            _toolContent: [part],\n            _originalText: text,\n            _toolName: part.toolName,\n          });\n        }\n      }\n    }\n  }\n\n  return messages;\n}\n\n/**\n * Narrows a generic Message to AISDKMessage for typed access to pass-through fields.\n */\nfunction asAISDK(msg: Message): AISDKMessage {\n  return msg;\n}\n\n/**\n * Converts context-chef IR messages back to AI SDK V3 prompt format.\n *\n * Uses per-role original content when unmodified (detected via `_originalText`).\n * Falls back to constructing from IR fields when content was modified by Janitor\n * (e.g. compact() cleared tool results) or for new messages (e.g. compression summaries).\n */\nexport function toAISDK(messages: Message[]): LanguageModelV3Prompt {\n  const prompt: LanguageModelV3Prompt = [];\n\n  let i = 0;\n  while (i < messages.length) {\n    const msg = asAISDK(messages[i]);\n    const contentModified = msg._originalText !== undefined && msg._originalText !== msg.content;\n\n    if (msg.role === 'system') {\n      prompt.push({\n        role: 'system',\n        content: msg.content,\n        ...(msg._providerOptions ? { providerOptions: msg._providerOptions } : {}),\n      });\n      i++;\n      continue;\n    }\n\n    if (msg.role === 'user') {\n      prompt.push({\n        role: 'user',\n        content:\n          !contentModified && msg._userContent\n            ? msg._userContent\n            : [{ type: 'text', text: msg.content }],\n        ...(msg._providerOptions ? { providerOptions: msg._providerOptions } : {}),\n      });\n      i++;\n      continue;\n    }\n\n    if (msg.role === 'assistant') {\n      prompt.push({\n        role: 'assistant',\n        content:\n          !contentModified && msg._assistantContent\n            ? msg._assistantContent\n            : [{ type: 'text', text: msg.content }],\n        ...(msg._providerOptions ? { providerOptions: msg._providerOptions } : {}),\n      });\n      i++;\n      continue;\n    }\n\n    if (msg.role === 'tool') {\n      const toolResults: LanguageModelV3ToolResultPart[] = [];\n      while (i < messages.length && messages[i].role === 'tool') {\n        const toolMsg = asAISDK(messages[i]);\n        const toolModified =\n          toolMsg._originalText !== undefined && toolMsg._originalText !== toolMsg.content;\n\n        if (!toolModified && toolMsg._toolContent) {\n          for (const part of toolMsg._toolContent) {\n            if (part.type === 'tool-result') {\n              toolResults.push(part);\n            }\n          }\n        } else {\n          toolResults.push({\n            type: 'tool-result',\n            toolCallId: toolMsg.tool_call_id ?? '',\n            toolName: toolMsg._toolName ?? 'unknown',\n            output: { type: 'text', value: toolMsg.content },\n          });\n        }\n        i++;\n      }\n      prompt.push({ role: 'tool', content: toolResults });\n      continue;\n    }\n\n    i++;\n  }\n\n  return prompt;\n}\n\nfunction stringifyToolOutput(output: LanguageModelV3ToolResultOutput): string {\n  switch (output.type) {\n    case 'text':\n    case 'error-text':\n      return output.value;\n    case 'json':\n    case 'error-json':\n      return JSON.stringify(output.value);\n    case 'content':\n      return output.value\n        .map((v) => (v.type === 'text' ? v.text : ''))\n        .filter(Boolean)\n        .join('\\n');\n    default:\n      return JSON.stringify(output);\n  }\n}\n","import type {\n  LanguageModelV3Prompt,\n  LanguageModelV3ToolResultOutput,\n  LanguageModelV3ToolResultPart,\n} from '@ai-sdk/provider';\nimport { Offloader } from '@context-chef/core';\nimport type { TruncateOptions } from './types';\n\n/**\n * Truncates tool-result content within an AI SDK prompt when it exceeds the configured threshold.\n * When a storage adapter is provided, original content is persisted and a URI is included in the output.\n */\nexport async function truncateToolResults(\n  prompt: LanguageModelV3Prompt,\n  options: TruncateOptions,\n): Promise<LanguageModelV3Prompt> {\n  const { threshold, headChars = 0, tailChars = 1000, storage } = options;\n\n  const offloader = storage ? new Offloader({ threshold, adapter: storage, storageDir: '' }) : null;\n\n  const result: LanguageModelV3Prompt = [];\n\n  for (const msg of prompt) {\n    if (msg.role !== 'tool') {\n      result.push(msg);\n      continue;\n    }\n\n    const newContent: typeof msg.content = [];\n\n    for (const part of msg.content) {\n      if (part.type !== 'tool-result') {\n        newContent.push(part);\n        continue;\n      }\n\n      const text = extractText(part.output);\n      if (text.length <= threshold || headChars + tailChars >= text.length) {\n        newContent.push(part);\n        continue;\n      }\n\n      // With storage: use Offloader to persist original and get a URI-annotated truncation\n      if (offloader) {\n        try {\n          const vfsResult = await offloader.offloadAsync(text, { threshold, headChars, tailChars });\n          newContent.push({\n            ...part,\n            output: {\n              type: 'text',\n              value: vfsResult.content,\n            } satisfies LanguageModelV3ToolResultOutput,\n          } satisfies LanguageModelV3ToolResultPart);\n          continue;\n        } catch (error) {\n          console.warn(\n            `[context-chef] Storage adapter write failed for tool result (${part.toolCallId}). ` +\n              `Falling back to simple truncation. Error: ${error instanceof Error ? error.message : String(error)}`,\n          );\n          // Fall through to simple truncation below\n        }\n      }\n\n      // Without storage: simple truncation, original is discarded\n      const head = text.slice(0, headChars);\n      const tail = text.slice(text.length - tailChars);\n      const totalLines = text.split('\\n').length;\n\n      const truncated = [\n        head,\n        `\\n--- truncated (${totalLines} lines, ${text.length} chars total) ---\\n`,\n        tail,\n      ]\n        .filter(Boolean)\n        .join('')\n        .trim();\n\n      newContent.push({\n        ...part,\n        output: { type: 'text', value: truncated } satisfies LanguageModelV3ToolResultOutput,\n      } satisfies LanguageModelV3ToolResultPart);\n    }\n\n    result.push({ ...msg, content: newContent });\n  }\n\n  return result;\n}\n\nfunction extractText(output: LanguageModelV3ToolResultOutput): string {\n  switch (output.type) {\n    case 'text':\n    case 'error-text':\n      return output.value;\n    case 'json':\n    case 'error-json':\n      return JSON.stringify(output.value);\n    case 'content':\n      return output.value\n        .map((v: { type: string; text?: string }) => (v.type === 'text' ? (v.text ?? '') : ''))\n        .filter(Boolean)\n        .join('\\n');\n    default:\n      return '';\n  }\n}\n","import type {\n  LanguageModelV3,\n  LanguageModelV3Prompt,\n  LanguageModelV3StreamPart,\n} from '@ai-sdk/provider';\nimport { Janitor, type Message, Prompts, XmlGenerator } from '@context-chef/core';\nimport { generateText, type LanguageModelMiddleware } from 'ai';\n\nimport { fromAISDK, toAISDK } from './adapter';\nimport { truncateToolResults } from './truncator';\nimport type { ContextChefOptions, DynamicStateConfig } from './types';\n\ntype CompressRole = 'system' | 'user' | 'assistant';\n\n/**\n * Creates a LanguageModelMiddleware that transparently applies\n * context-chef compression and truncation to AI SDK model calls.\n *\n * The middleware holds a stateful Janitor instance that tracks\n * token usage across calls for compression decisions.\n */\nexport function createMiddleware(options: ContextChefOptions): LanguageModelMiddleware {\n  let usageWarned = false;\n\n  const janitor = new Janitor({\n    contextWindow: options.contextWindow,\n    tokenizer: options.tokenizer ? (msgs: Message[]) => options.tokenizer?.(msgs) ?? 0 : undefined,\n    preserveRatio: options.compress?.preserveRatio ?? 0.8,\n    compressionModel: options.compress?.model\n      ? createCompressionAdapter(options.compress.model)\n      : undefined,\n    onCompress: options.onCompress\n      ? (summary, count) => options.onCompress?.(summary.content, count)\n      : undefined,\n    onBeforeCompress: options.onBeforeCompress ?? options.onBudgetExceeded,\n  });\n\n  return {\n    specificationVersion: 'v3',\n\n    transformParams: async ({ params }) => {\n      let { prompt } = params;\n\n      // 1. Truncate large tool results\n      if (options.truncate) {\n        prompt = await truncateToolResults(prompt, options.truncate);\n      }\n\n      // 2. Convert to IR and separate system messages from conversation.\n      // System messages are standing instructions — they must not be\n      // compressed away. Only conversation history goes through compact/compress.\n      const allIR = fromAISDK(prompt);\n      let systemMessages = allIR.filter((m) => m.role === 'system');\n      let conversation = allIR.filter((m) => m.role !== 'system');\n\n      // 3. Compact (mechanical, zero LLM cost) before compression\n      if (options.compact) {\n        const preCompact = conversation;\n        conversation = janitor.compact(conversation, options.compact);\n\n        // When thinking is stripped, invalidate adapter pass-through\n        // so toAISDK reconstructs from IR fields (without reasoning)\n        if (options.compact.clear.includes('thinking')) {\n          for (let i = 0; i < conversation.length; i++) {\n            if (preCompact[i].thinking && !conversation[i].thinking) {\n              delete conversation[i]._assistantContent;\n            }\n          }\n        }\n      }\n\n      // 4. Compress conversation history if over token budget\n      conversation = await janitor.compress(conversation);\n\n      // 5. Rebuild system layer: original system messages + middleware instructions.\n      // Appended (not prepended) to preserve the user's system prompt ordering.\n      if (options.compact) {\n        const hasToolResultTarget = options.compact.clear.some(\n          (t) => t === 'tool-result' || (typeof t === 'object' && t.target === 'tool-result'),\n        );\n        if (hasToolResultTarget) {\n          systemMessages = [\n            ...systemMessages,\n            { role: 'system', content: Prompts.TOOL_RESULT_CLEARED_INSTRUCTION },\n          ];\n        }\n      }\n\n      // 6. Reassemble sandwich: system + conversation\n      const irMessages = [...systemMessages, ...conversation];\n\n      // 7. Convert back to AI SDK format\n      prompt = toAISDK(irMessages);\n\n      // 8. Dynamic state injection\n      if (options.dynamicState) {\n        prompt = await injectDynamicState(prompt, options.dynamicState);\n      }\n\n      // 9. Custom transform hook\n      if (options.transformContext) {\n        prompt = await options.transformContext(prompt);\n      }\n\n      return { ...params, prompt };\n    },\n\n    wrapGenerate: async ({ doGenerate }) => {\n      const result = await doGenerate();\n\n      if (result.usage?.inputTokens?.total != null) {\n        janitor.feedTokenUsage(result.usage.inputTokens.total);\n      } else if (!usageWarned && !options.tokenizer) {\n        usageWarned = true;\n        console.warn(\n          '[context-chef] Model response did not include usage.inputTokens.total. ' +\n            'Token-based compression may not trigger accurately. ' +\n            'Consider providing a tokenizer for precise token counting.',\n        );\n      }\n\n      return result;\n    },\n\n    wrapStream: async ({ doStream }) => {\n      const { stream, ...rest } = await doStream();\n\n      const transform = new TransformStream<LanguageModelV3StreamPart, LanguageModelV3StreamPart>({\n        transform(chunk, controller) {\n          if (chunk.type === 'finish') {\n            if (chunk.usage?.inputTokens?.total != null) {\n              janitor.feedTokenUsage(chunk.usage.inputTokens.total);\n            } else if (!usageWarned && !options.tokenizer) {\n              usageWarned = true;\n              console.warn(\n                '[context-chef] Stream finish did not include usage.inputTokens.total. ' +\n                  'Token-based compression may not trigger accurately. ' +\n                  'Consider providing a tokenizer for precise token counting.',\n              );\n            }\n          }\n          controller.enqueue(chunk);\n        },\n      });\n\n      return { ...rest, stream: stream.pipeThrough(transform) };\n    },\n  };\n}\n\n/**\n * Injects dynamic state XML into the AI SDK prompt.\n *\n * - `last_user`: Appends to the last user message's content parts.\n *   Leverages Recency Bias for maximum LLM attention.\n * - `system`: Adds as a standalone system message at the end.\n */\nasync function injectDynamicState(\n  prompt: LanguageModelV3Prompt,\n  config: DynamicStateConfig,\n): Promise<LanguageModelV3Prompt> {\n  const state = await config.getState();\n  const xml = XmlGenerator.objectToXml(state, 'dynamic_state');\n  const placement = config.placement ?? 'last_user';\n\n  if (placement === 'system') {\n    return [...prompt, { role: 'system', content: `CURRENT TASK STATE:\\n${xml}` }];\n  }\n\n  // last_user: inject into the last user message\n  const result = [...prompt];\n  const stateBlock = `\\n\\n${xml}\\nAbove is the current system state. Use it to guide your next action.`;\n\n  for (let i = result.length - 1; i >= 0; i--) {\n    const msg = result[i];\n    if (msg.role === 'user') {\n      result[i] = {\n        ...msg,\n        content: [...msg.content, { type: 'text', text: stateBlock }],\n      };\n      return result;\n    }\n  }\n\n  // No user message found — append as new user message\n  result.push({\n    role: 'user',\n    content: [{ type: 'text', text: stateBlock.trim() }],\n  });\n  return result;\n}\n\n/**\n * Maps an IR role to a role accepted by generateText.\n * Tool messages are handled separately before this is called.\n */\nfunction toCompressRole(role: string): CompressRole {\n  if (role === 'system' || role === 'user' || role === 'assistant') return role;\n  return 'user';\n}\n\n/**\n * Adapts an AI SDK LanguageModelV3 into the compressionModel callback\n * that Janitor expects: (messages: Message[]) => Promise<string>\n *\n * Tool messages are converted to user messages describing the tool interaction,\n * since generateText only accepts system/user/assistant roles.\n */\nfunction createCompressionAdapter(\n  model: LanguageModelV3,\n): (messages: Message[]) => Promise<string> {\n  return async (messages: Message[]): Promise<string> => {\n    const formatted = messages.map((m): { role: CompressRole; content: string } => {\n      if (m.role === 'tool') {\n        return {\n          role: 'user' satisfies CompressRole,\n          content: `[Tool result${m.tool_call_id ? ` (${m.tool_call_id})` : ''}: ${m.content}]`,\n        };\n      }\n      if (m.role === 'assistant' && m.tool_calls?.length) {\n        const toolCallsDesc = m.tool_calls\n          .map((tc) => `[Called tool: ${tc.function.name}(${tc.function.arguments})]`)\n          .join('\\n');\n        return {\n          role: 'assistant' satisfies CompressRole,\n          content: m.content ? `${m.content}\\n${toolCallsDesc}` : toolCallsDesc,\n        };\n      }\n      return {\n        role: toCompressRole(m.role),\n        content: m.content,\n      };\n    });\n\n    const { text } = await generateText({\n      model,\n      messages: formatted,\n      maxOutputTokens: 2048,\n    });\n\n    return text || '[Compression produced no output]';\n  };\n}\n","import type { LanguageModelV3 } from '@ai-sdk/provider';\nimport { wrapLanguageModel } from 'ai';\n\nimport { createMiddleware } from './middleware';\nimport type { ContextChefOptions } from './types';\n\nexport { type AISDKMessage, fromAISDK, toAISDK } from './adapter';\nexport { createMiddleware } from './middleware';\nexport type {\n  CompactConfig,\n  CompressOptions,\n  ContextChefOptions,\n  DynamicStateConfig,\n  TruncateOptions,\n} from './types';\n\n/**\n * Wraps an AI SDK language model with context-chef middleware for\n * transparent history compression, tool result truncation, and token budget management.\n *\n * @example\n * ```typescript\n * import { withContextChef } from '@context-chef/ai-sdk-middleware';\n * import { openai } from '@ai-sdk/openai';\n * import { generateText } from 'ai';\n *\n * const model = withContextChef(openai('gpt-4o'), {\n *   contextWindow: 128_000,\n *   compress: { model: openai('gpt-4o-mini') },\n *   truncate: { threshold: 5000, headChars: 500, tailChars: 1000 },\n * });\n *\n * // Use exactly like normal — zero other code changes\n * const result = await generateText({ model, messages, tools });\n * ```\n */\nexport function withContextChef(\n  model: LanguageModelV3,\n  options: ContextChefOptions,\n): LanguageModelV3 {\n  const middleware = createMiddleware(options);\n  return wrapLanguageModel({ model, middleware });\n}\n"],"mappings":";;;;;;;;;;;;AAkCA,SAAgB,UAAU,QAA+C;CACvE,MAAM,WAA2B,EAAE;AAEnC,MAAK,MAAM,OAAO,QAAQ;AACxB,MAAI,IAAI,SAAS,UAAU;AACzB,YAAS,KAAK;IACZ,MAAM;IACN,SAAS,IAAI;IACb,GAAI,IAAI,kBAAkB,EAAE,kBAAkB,IAAI,iBAAiB,GAAG,EAAE;IACzE,CAAC;AACF;;AAGF,MAAI,IAAI,SAAS,QAAQ;GACvB,MAAM,OAAO,IAAI,QACd,QAAQ,MAAM,EAAE,SAAS,OAAO,CAChC,KAAK,MAAM,EAAE,KAAK,CAClB,KAAK,KAAK;AACb,YAAS,KAAK;IACZ,MAAM;IACN,SAAS;IACT,cAAc,IAAI;IAClB,eAAe;IACf,GAAI,IAAI,kBAAkB,EAAE,kBAAkB,IAAI,iBAAiB,GAAG,EAAE;IACzE,CAAC;AACF;;AAGF,MAAI,IAAI,SAAS,aAAa;GAC5B,MAAM,OAAiB,EAAE;GACzB,MAAM,YAAwB,EAAE;GAChC,IAAI;AAEJ,QAAK,MAAM,QAAQ,IAAI,QACrB,KAAI,KAAK,SAAS,OAAQ,MAAK,KAAK,KAAK,KAAK;YACrC,KAAK,SAAS,YACrB,WAAU,KAAK;IACb,IAAI,KAAK;IACT,MAAM;IACN,UAAU;KACR,MAAM,KAAK;KACX,WAAW,OAAO,KAAK,UAAU,WAAW,KAAK,QAAQ,KAAK,UAAU,KAAK,MAAM;KACpF;IACF,CAAC;YACO,KAAK,SAAS,YACvB,YAAW,EAAE,UAAU,KAAK,MAAM;GAItC,MAAM,aAAa,KAAK,KAAK,KAAK;GAClC,MAAM,IAAkB;IACtB,MAAM;IACN,SAAS;IACT,mBAAmB,IAAI;IACvB,eAAe;IACf,GAAI,IAAI,kBAAkB,EAAE,kBAAkB,IAAI,iBAAiB,GAAG,EAAE;IACzE;AACD,OAAI,UAAU,SAAS,EAAG,GAAE,aAAa;AACzC,OAAI,SAAU,GAAE,WAAW;AAC3B,YAAS,KAAK,EAAE;AAChB;;AAGF,MAAI,IAAI,SAAS,QACf;QAAK,MAAM,QAAQ,IAAI,QACrB,KAAI,KAAK,SAAS,eAAe;IAC/B,MAAM,OAAO,oBAAoB,KAAK,OAAO;AAC7C,aAAS,KAAK;KACZ,MAAM;KACN,SAAS;KACT,cAAc,KAAK;KACnB,cAAc,CAAC,KAAK;KACpB,eAAe;KACf,WAAW,KAAK;KACjB,CAAC;;;;AAMV,QAAO;;;;;AAMT,SAAS,QAAQ,KAA4B;AAC3C,QAAO;;;;;;;;;AAUT,SAAgB,QAAQ,UAA4C;CAClE,MAAM,SAAgC,EAAE;CAExC,IAAI,IAAI;AACR,QAAO,IAAI,SAAS,QAAQ;EAC1B,MAAM,MAAM,QAAQ,SAAS,GAAG;EAChC,MAAM,kBAAkB,IAAI,kBAAkB,UAAa,IAAI,kBAAkB,IAAI;AAErF,MAAI,IAAI,SAAS,UAAU;AACzB,UAAO,KAAK;IACV,MAAM;IACN,SAAS,IAAI;IACb,GAAI,IAAI,mBAAmB,EAAE,iBAAiB,IAAI,kBAAkB,GAAG,EAAE;IAC1E,CAAC;AACF;AACA;;AAGF,MAAI,IAAI,SAAS,QAAQ;AACvB,UAAO,KAAK;IACV,MAAM;IACN,SACE,CAAC,mBAAmB,IAAI,eACpB,IAAI,eACJ,CAAC;KAAE,MAAM;KAAQ,MAAM,IAAI;KAAS,CAAC;IAC3C,GAAI,IAAI,mBAAmB,EAAE,iBAAiB,IAAI,kBAAkB,GAAG,EAAE;IAC1E,CAAC;AACF;AACA;;AAGF,MAAI,IAAI,SAAS,aAAa;AAC5B,UAAO,KAAK;IACV,MAAM;IACN,SACE,CAAC,mBAAmB,IAAI,oBACpB,IAAI,oBACJ,CAAC;KAAE,MAAM;KAAQ,MAAM,IAAI;KAAS,CAAC;IAC3C,GAAI,IAAI,mBAAmB,EAAE,iBAAiB,IAAI,kBAAkB,GAAG,EAAE;IAC1E,CAAC;AACF;AACA;;AAGF,MAAI,IAAI,SAAS,QAAQ;GACvB,MAAM,cAA+C,EAAE;AACvD,UAAO,IAAI,SAAS,UAAU,SAAS,GAAG,SAAS,QAAQ;IACzD,MAAM,UAAU,QAAQ,SAAS,GAAG;AAIpC,QAAI,EAFF,QAAQ,kBAAkB,UAAa,QAAQ,kBAAkB,QAAQ,YAEtD,QAAQ,cAC3B;UAAK,MAAM,QAAQ,QAAQ,aACzB,KAAI,KAAK,SAAS,cAChB,aAAY,KAAK,KAAK;UAI1B,aAAY,KAAK;KACf,MAAM;KACN,YAAY,QAAQ,gBAAgB;KACpC,UAAU,QAAQ,aAAa;KAC/B,QAAQ;MAAE,MAAM;MAAQ,OAAO,QAAQ;MAAS;KACjD,CAAC;AAEJ;;AAEF,UAAO,KAAK;IAAE,MAAM;IAAQ,SAAS;IAAa,CAAC;AACnD;;AAGF;;AAGF,QAAO;;AAGT,SAAS,oBAAoB,QAAiD;AAC5E,SAAQ,OAAO,MAAf;EACE,KAAK;EACL,KAAK,aACH,QAAO,OAAO;EAChB,KAAK;EACL,KAAK,aACH,QAAO,KAAK,UAAU,OAAO,MAAM;EACrC,KAAK,UACH,QAAO,OAAO,MACX,KAAK,MAAO,EAAE,SAAS,SAAS,EAAE,OAAO,GAAI,CAC7C,OAAO,QAAQ,CACf,KAAK,KAAK;EACf,QACE,QAAO,KAAK,UAAU,OAAO;;;;;;;;;;AClNnC,eAAsB,oBACpB,QACA,SACgC;CAChC,MAAM,EAAE,WAAW,YAAY,GAAG,YAAY,KAAM,YAAY;CAEhE,MAAM,YAAY,UAAU,IAAIA,6BAAU;EAAE;EAAW,SAAS;EAAS,YAAY;EAAI,CAAC,GAAG;CAE7F,MAAM,SAAgC,EAAE;AAExC,MAAK,MAAM,OAAO,QAAQ;AACxB,MAAI,IAAI,SAAS,QAAQ;AACvB,UAAO,KAAK,IAAI;AAChB;;EAGF,MAAM,aAAiC,EAAE;AAEzC,OAAK,MAAM,QAAQ,IAAI,SAAS;AAC9B,OAAI,KAAK,SAAS,eAAe;AAC/B,eAAW,KAAK,KAAK;AACrB;;GAGF,MAAM,OAAO,YAAY,KAAK,OAAO;AACrC,OAAI,KAAK,UAAU,aAAa,YAAY,aAAa,KAAK,QAAQ;AACpE,eAAW,KAAK,KAAK;AACrB;;AAIF,OAAI,UACF,KAAI;IACF,MAAM,YAAY,MAAM,UAAU,aAAa,MAAM;KAAE;KAAW;KAAW;KAAW,CAAC;AACzF,eAAW,KAAK;KACd,GAAG;KACH,QAAQ;MACN,MAAM;MACN,OAAO,UAAU;MAClB;KACF,CAAyC;AAC1C;YACO,OAAO;AACd,YAAQ,KACN,gEAAgE,KAAK,WAAW,+CACjC,iBAAiB,QAAQ,MAAM,UAAU,OAAO,MAAM,GACtG;;GAML,MAAM,OAAO,KAAK,MAAM,GAAG,UAAU;GACrC,MAAM,OAAO,KAAK,MAAM,KAAK,SAAS,UAAU;GAGhD,MAAM,YAAY;IAChB;IACA,oBAJiB,KAAK,MAAM,KAAK,CAAC,OAIH,UAAU,KAAK,OAAO;IACrD;IACD,CACE,OAAO,QAAQ,CACf,KAAK,GAAG,CACR,MAAM;AAET,cAAW,KAAK;IACd,GAAG;IACH,QAAQ;KAAE,MAAM;KAAQ,OAAO;KAAW;IAC3C,CAAyC;;AAG5C,SAAO,KAAK;GAAE,GAAG;GAAK,SAAS;GAAY,CAAC;;AAG9C,QAAO;;AAGT,SAAS,YAAY,QAAiD;AACpE,SAAQ,OAAO,MAAf;EACE,KAAK;EACL,KAAK,aACH,QAAO,OAAO;EAChB,KAAK;EACL,KAAK,aACH,QAAO,KAAK,UAAU,OAAO,MAAM;EACrC,KAAK,UACH,QAAO,OAAO,MACX,KAAK,MAAwC,EAAE,SAAS,SAAU,EAAE,QAAQ,KAAM,GAAI,CACtF,OAAO,QAAQ,CACf,KAAK,KAAK;EACf,QACE,QAAO;;;;;;;;;;;;;AClFb,SAAgB,iBAAiB,SAAsD;CACrF,IAAI,cAAc;CAElB,MAAM,UAAU,IAAIC,2BAAQ;EAC1B,eAAe,QAAQ;EACvB,WAAW,QAAQ,aAAa,SAAoB,QAAQ,YAAY,KAAK,IAAI,IAAI;EACrF,eAAe,QAAQ,UAAU,iBAAiB;EAClD,kBAAkB,QAAQ,UAAU,QAChC,yBAAyB,QAAQ,SAAS,MAAM,GAChD;EACJ,YAAY,QAAQ,cACf,SAAS,UAAU,QAAQ,aAAa,QAAQ,SAAS,MAAM,GAChE;EACJ,kBAAkB,QAAQ,oBAAoB,QAAQ;EACvD,CAAC;AAEF,QAAO;EACL,sBAAsB;EAEtB,iBAAiB,OAAO,EAAE,aAAa;GACrC,IAAI,EAAE,WAAW;AAGjB,OAAI,QAAQ,SACV,UAAS,MAAM,oBAAoB,QAAQ,QAAQ,SAAS;GAM9D,MAAM,QAAQ,UAAU,OAAO;GAC/B,IAAI,iBAAiB,MAAM,QAAQ,MAAM,EAAE,SAAS,SAAS;GAC7D,IAAI,eAAe,MAAM,QAAQ,MAAM,EAAE,SAAS,SAAS;AAG3D,OAAI,QAAQ,SAAS;IACnB,MAAM,aAAa;AACnB,mBAAe,QAAQ,QAAQ,cAAc,QAAQ,QAAQ;AAI7D,QAAI,QAAQ,QAAQ,MAAM,SAAS,WAAW,EAC5C;UAAK,IAAI,IAAI,GAAG,IAAI,aAAa,QAAQ,IACvC,KAAI,WAAW,GAAG,YAAY,CAAC,aAAa,GAAG,SAC7C,QAAO,aAAa,GAAG;;;AAO/B,kBAAe,MAAM,QAAQ,SAAS,aAAa;AAInD,OAAI,QAAQ,SAIV;QAH4B,QAAQ,QAAQ,MAAM,MAC/C,MAAM,MAAM,iBAAkB,OAAO,MAAM,YAAY,EAAE,WAAW,cACtE,CAEC,kBAAiB,CACf,GAAG,gBACH;KAAE,MAAM;KAAU,SAASC,2BAAQ;KAAiC,CACrE;;AAQL,YAAS,QAHU,CAAC,GAAG,gBAAgB,GAAG,aAAa,CAG3B;AAG5B,OAAI,QAAQ,aACV,UAAS,MAAM,mBAAmB,QAAQ,QAAQ,aAAa;AAIjE,OAAI,QAAQ,iBACV,UAAS,MAAM,QAAQ,iBAAiB,OAAO;AAGjD,UAAO;IAAE,GAAG;IAAQ;IAAQ;;EAG9B,cAAc,OAAO,EAAE,iBAAiB;GACtC,MAAM,SAAS,MAAM,YAAY;AAEjC,OAAI,OAAO,OAAO,aAAa,SAAS,KACtC,SAAQ,eAAe,OAAO,MAAM,YAAY,MAAM;YAC7C,CAAC,eAAe,CAAC,QAAQ,WAAW;AAC7C,kBAAc;AACd,YAAQ,KACN,wLAGD;;AAGH,UAAO;;EAGT,YAAY,OAAO,EAAE,eAAe;GAClC,MAAM,EAAE,QAAQ,GAAG,SAAS,MAAM,UAAU;GAE5C,MAAM,YAAY,IAAI,gBAAsE,EAC1F,UAAU,OAAO,YAAY;AAC3B,QAAI,MAAM,SAAS,UACjB;SAAI,MAAM,OAAO,aAAa,SAAS,KACrC,SAAQ,eAAe,MAAM,MAAM,YAAY,MAAM;cAC5C,CAAC,eAAe,CAAC,QAAQ,WAAW;AAC7C,oBAAc;AACd,cAAQ,KACN,uLAGD;;;AAGL,eAAW,QAAQ,MAAM;MAE5B,CAAC;AAEF,UAAO;IAAE,GAAG;IAAM,QAAQ,OAAO,YAAY,UAAU;IAAE;;EAE5D;;;;;;;;;AAUH,eAAe,mBACb,QACA,QACgC;CAChC,MAAM,QAAQ,MAAM,OAAO,UAAU;CACrC,MAAM,MAAMC,gCAAa,YAAY,OAAO,gBAAgB;AAG5D,MAFkB,OAAO,aAAa,iBAEpB,SAChB,QAAO,CAAC,GAAG,QAAQ;EAAE,MAAM;EAAU,SAAS,wBAAwB;EAAO,CAAC;CAIhF,MAAM,SAAS,CAAC,GAAG,OAAO;CAC1B,MAAM,aAAa,OAAO,IAAI;AAE9B,MAAK,IAAI,IAAI,OAAO,SAAS,GAAG,KAAK,GAAG,KAAK;EAC3C,MAAM,MAAM,OAAO;AACnB,MAAI,IAAI,SAAS,QAAQ;AACvB,UAAO,KAAK;IACV,GAAG;IACH,SAAS,CAAC,GAAG,IAAI,SAAS;KAAE,MAAM;KAAQ,MAAM;KAAY,CAAC;IAC9D;AACD,UAAO;;;AAKX,QAAO,KAAK;EACV,MAAM;EACN,SAAS,CAAC;GAAE,MAAM;GAAQ,MAAM,WAAW,MAAM;GAAE,CAAC;EACrD,CAAC;AACF,QAAO;;;;;;AAOT,SAAS,eAAe,MAA4B;AAClD,KAAI,SAAS,YAAY,SAAS,UAAU,SAAS,YAAa,QAAO;AACzE,QAAO;;;;;;;;;AAUT,SAAS,yBACP,OAC0C;AAC1C,QAAO,OAAO,aAAyC;EAuBrD,MAAM,EAAE,SAAS,2BAAmB;GAClC;GACA,UAxBgB,SAAS,KAAK,MAA+C;AAC7E,QAAI,EAAE,SAAS,OACb,QAAO;KACL,MAAM;KACN,SAAS,eAAe,EAAE,eAAe,KAAK,EAAE,aAAa,KAAK,GAAG,IAAI,EAAE,QAAQ;KACpF;AAEH,QAAI,EAAE,SAAS,eAAe,EAAE,YAAY,QAAQ;KAClD,MAAM,gBAAgB,EAAE,WACrB,KAAK,OAAO,iBAAiB,GAAG,SAAS,KAAK,GAAG,GAAG,SAAS,UAAU,IAAI,CAC3E,KAAK,KAAK;AACb,YAAO;MACL,MAAM;MACN,SAAS,EAAE,UAAU,GAAG,EAAE,QAAQ,IAAI,kBAAkB;MACzD;;AAEH,WAAO;KACL,MAAM,eAAe,EAAE,KAAK;KAC5B,SAAS,EAAE;KACZ;KACD;GAKA,iBAAiB;GAClB,CAAC;AAEF,SAAO,QAAQ;;;;;;;;;;;;;;;;;;;;;;;;;;AC5MnB,SAAgB,gBACd,OACA,SACiB;AAEjB,kCAAyB;EAAE;EAAO,YADf,iBAAiB,QAAQ;EACE,CAAC"}
+{"version":3,"file":"index.cjs","names":["Offloader","Janitor","XmlGenerator"],"sources":["../src/adapter.ts","../src/truncator.ts","../src/middleware.ts","../src/index.ts"],"sourcesContent":["import type {\n  LanguageModelV3Message,\n  LanguageModelV3Prompt,\n  LanguageModelV3ToolResultOutput,\n  LanguageModelV3ToolResultPart,\n  SharedV3ProviderOptions,\n} from '@ai-sdk/provider';\nimport type { Message, ToolCall } from '@context-chef/core';\n\n/** Content types for each AI SDK message role */\ntype UserContent = Extract<LanguageModelV3Message, { role: 'user' }>['content'];\ntype AssistantContent = Extract<LanguageModelV3Message, { role: 'assistant' }>['content'];\ntype ToolContent = Extract<LanguageModelV3Message, { role: 'tool' }>['content'];\n\n/**\n * Extended IR message with typed pass-through fields for lossless AI SDK round-trip.\n * Per-role content fields avoid union types, so no `as` casts are needed in `toAISDK`.\n */\nexport interface AISDKMessage extends Message {\n  _userContent?: UserContent;\n  _assistantContent?: AssistantContent;\n  _toolContent?: ToolContent;\n  _originalText?: string;\n  _providerOptions?: SharedV3ProviderOptions;\n  _toolName?: string;\n}\n\n/**\n * Converts an AI SDK V3 prompt to context-chef IR messages.\n *\n * Original AI SDK content is stored in per-role fields for lossless round-trip.\n * `_originalText` caches the extracted text so `toAISDK` can detect Janitor modifications.\n * `_providerOptions` preserves message-level provider options (e.g. Anthropic cache control).\n */\nexport function fromAISDK(prompt: LanguageModelV3Prompt): AISDKMessage[] {\n  const messages: AISDKMessage[] = [];\n\n  for (const msg of prompt) {\n    if (msg.role === 'system') {\n      messages.push({\n        role: 'system',\n        content: msg.content,\n        ...(msg.providerOptions ? { _providerOptions: msg.providerOptions } : {}),\n      });\n      continue;\n    }\n\n    if (msg.role === 'user') {\n      const text = msg.content\n        .filter((p) => p.type === 'text')\n        .map((p) => p.text)\n        .join('\\n');\n      messages.push({\n        role: 'user',\n        content: text,\n        _userContent: msg.content,\n        _originalText: text,\n        ...(msg.providerOptions ? { _providerOptions: msg.providerOptions } : {}),\n      });\n      continue;\n    }\n\n    if (msg.role === 'assistant') {\n      const text: string[] = [];\n      const toolCalls: ToolCall[] = [];\n      let thinking: { thinking: string } | undefined;\n\n      for (const part of msg.content) {\n        if (part.type === 'text') text.push(part.text);\n        else if (part.type === 'tool-call') {\n          toolCalls.push({\n            id: part.toolCallId,\n            type: 'function',\n            function: {\n              name: part.toolName,\n              arguments: typeof part.input === 'string' ? part.input : JSON.stringify(part.input),\n            },\n          });\n        } else if (part.type === 'reasoning') {\n          thinking = { thinking: part.text };\n        }\n      }\n\n      const joinedText = text.join('\\n');\n      const m: AISDKMessage = {\n        role: 'assistant',\n        content: joinedText,\n        _assistantContent: msg.content,\n        _originalText: joinedText,\n        ...(msg.providerOptions ? { _providerOptions: msg.providerOptions } : {}),\n      };\n      if (toolCalls.length > 0) m.tool_calls = toolCalls;\n      if (thinking) m.thinking = thinking;\n      messages.push(m);\n      continue;\n    }\n\n    if (msg.role === 'tool') {\n      for (const part of msg.content) {\n        if (part.type === 'tool-result') {\n          const text = stringifyToolOutput(part.output);\n          messages.push({\n            role: 'tool',\n            content: text,\n            tool_call_id: part.toolCallId,\n            _toolContent: [part],\n            _originalText: text,\n            _toolName: part.toolName,\n          });\n        }\n      }\n    }\n  }\n\n  return messages;\n}\n\n/**\n * Narrows a generic Message to AISDKMessage for typed access to pass-through fields.\n */\nfunction asAISDK(msg: Message): AISDKMessage {\n  return msg;\n}\n\n/**\n * Converts context-chef IR messages back to AI SDK V3 prompt format.\n *\n * Uses per-role original content when unmodified (detected via `_originalText`).\n * Falls back to constructing from IR fields when content was modified by Janitor\n * (e.g. compact() cleared tool results) or for new messages (e.g. compression summaries).\n */\nexport function toAISDK(messages: Message[]): LanguageModelV3Prompt {\n  const prompt: LanguageModelV3Prompt = [];\n\n  let i = 0;\n  while (i < messages.length) {\n    const msg = asAISDK(messages[i]);\n    const contentModified = msg._originalText !== undefined && msg._originalText !== msg.content;\n\n    if (msg.role === 'system') {\n      prompt.push({\n        role: 'system',\n        content: msg.content,\n        ...(msg._providerOptions ? { providerOptions: msg._providerOptions } : {}),\n      });\n      i++;\n      continue;\n    }\n\n    if (msg.role === 'user') {\n      prompt.push({\n        role: 'user',\n        content:\n          !contentModified && msg._userContent\n            ? msg._userContent\n            : [{ type: 'text', text: msg.content }],\n        ...(msg._providerOptions ? { providerOptions: msg._providerOptions } : {}),\n      });\n      i++;\n      continue;\n    }\n\n    if (msg.role === 'assistant') {\n      prompt.push({\n        role: 'assistant',\n        content:\n          !contentModified && msg._assistantContent\n            ? msg._assistantContent\n            : [{ type: 'text', text: msg.content }],\n        ...(msg._providerOptions ? { providerOptions: msg._providerOptions } : {}),\n      });\n      i++;\n      continue;\n    }\n\n    if (msg.role === 'tool') {\n      const toolResults: LanguageModelV3ToolResultPart[] = [];\n      while (i < messages.length && messages[i].role === 'tool') {\n        const toolMsg = asAISDK(messages[i]);\n        const toolModified =\n          toolMsg._originalText !== undefined && toolMsg._originalText !== toolMsg.content;\n\n        if (!toolModified && toolMsg._toolContent) {\n          for (const part of toolMsg._toolContent) {\n            if (part.type === 'tool-result') {\n              toolResults.push(part);\n            }\n          }\n        } else {\n          toolResults.push({\n            type: 'tool-result',\n            toolCallId: toolMsg.tool_call_id ?? '',\n            toolName: toolMsg._toolName ?? 'unknown',\n            output: { type: 'text', value: toolMsg.content },\n          });\n        }\n        i++;\n      }\n      prompt.push({ role: 'tool', content: toolResults });\n      continue;\n    }\n\n    i++;\n  }\n\n  return prompt;\n}\n\nfunction stringifyToolOutput(output: LanguageModelV3ToolResultOutput): string {\n  switch (output.type) {\n    case 'text':\n    case 'error-text':\n      return output.value;\n    case 'json':\n    case 'error-json':\n      return JSON.stringify(output.value);\n    case 'content':\n      return output.value\n        .map((v) => (v.type === 'text' ? v.text : ''))\n        .filter(Boolean)\n        .join('\\n');\n    default:\n      return JSON.stringify(output);\n  }\n}\n","import type {\n  LanguageModelV3Prompt,\n  LanguageModelV3ToolResultOutput,\n  LanguageModelV3ToolResultPart,\n} from '@ai-sdk/provider';\nimport { Offloader } from '@context-chef/core';\nimport type { TruncateOptions } from './types';\n\n/**\n * Truncates tool-result content within an AI SDK prompt when it exceeds the configured threshold.\n * When a storage adapter is provided, original content is persisted and a URI is included in the output.\n */\nexport async function truncateToolResults(\n  prompt: LanguageModelV3Prompt,\n  options: TruncateOptions,\n): Promise<LanguageModelV3Prompt> {\n  const { threshold, headChars = 0, tailChars = 1000, storage } = options;\n\n  const offloader = storage ? new Offloader({ threshold, adapter: storage, storageDir: '' }) : null;\n\n  const result: LanguageModelV3Prompt = [];\n\n  for (const msg of prompt) {\n    if (msg.role !== 'tool') {\n      result.push(msg);\n      continue;\n    }\n\n    const newContent: typeof msg.content = [];\n\n    for (const part of msg.content) {\n      if (part.type !== 'tool-result') {\n        newContent.push(part);\n        continue;\n      }\n\n      const text = extractText(part.output);\n      if (text.length <= threshold || headChars + tailChars >= text.length) {\n        newContent.push(part);\n        continue;\n      }\n\n      // With storage: use Offloader to persist original and get a URI-annotated truncation\n      if (offloader) {\n        try {\n          const vfsResult = await offloader.offloadAsync(text, { threshold, headChars, tailChars });\n          newContent.push({\n            ...part,\n            output: {\n              type: 'text',\n              value: vfsResult.content,\n            } satisfies LanguageModelV3ToolResultOutput,\n          } satisfies LanguageModelV3ToolResultPart);\n          continue;\n        } catch (error) {\n          console.warn(\n            `[context-chef] Storage adapter write failed for tool result (${part.toolCallId}). ` +\n              `Falling back to simple truncation. Error: ${error instanceof Error ? error.message : String(error)}`,\n          );\n          // Fall through to simple truncation below\n        }\n      }\n\n      // Without storage: simple truncation, original is discarded\n      const head = text.slice(0, headChars);\n      const tail = text.slice(text.length - tailChars);\n      const totalLines = text.split('\\n').length;\n\n      const truncated = [\n        head,\n        `\\n--- truncated (${totalLines} lines, ${text.length} chars total) ---\\n`,\n        tail,\n      ]\n        .filter(Boolean)\n        .join('')\n        .trim();\n\n      newContent.push({\n        ...part,\n        output: { type: 'text', value: truncated } satisfies LanguageModelV3ToolResultOutput,\n      } satisfies LanguageModelV3ToolResultPart);\n    }\n\n    result.push({ ...msg, content: newContent });\n  }\n\n  return result;\n}\n\nfunction extractText(output: LanguageModelV3ToolResultOutput): string {\n  switch (output.type) {\n    case 'text':\n    case 'error-text':\n      return output.value;\n    case 'json':\n    case 'error-json':\n      return JSON.stringify(output.value);\n    case 'content':\n      return output.value\n        .map((v: { type: string; text?: string }) => (v.type === 'text' ? (v.text ?? '') : ''))\n        .filter(Boolean)\n        .join('\\n');\n    default:\n      return '';\n  }\n}\n","import type {\n  LanguageModelV3,\n  LanguageModelV3Message,\n  LanguageModelV3Prompt,\n  LanguageModelV3StreamPart,\n} from '@ai-sdk/provider';\nimport { Janitor, type Message, XmlGenerator } from '@context-chef/core';\nimport { generateText, type LanguageModelMiddleware, type ModelMessage, pruneMessages } from 'ai';\n\nimport { fromAISDK, toAISDK } from './adapter';\nimport { truncateToolResults } from './truncator';\nimport type { ContextChefOptions, DynamicStateConfig } from './types';\n\ntype CompressRole = 'system' | 'user' | 'assistant';\n\n/**\n * Creates a LanguageModelMiddleware that transparently applies\n * context-chef compression and truncation to AI SDK model calls.\n *\n * The middleware holds a stateful Janitor instance that tracks\n * token usage across calls for compression decisions.\n */\nexport function createMiddleware(options: ContextChefOptions): LanguageModelMiddleware {\n  let usageWarned = false;\n\n  const janitor = new Janitor({\n    contextWindow: options.contextWindow,\n    tokenizer: options.tokenizer ? (msgs: Message[]) => options.tokenizer?.(msgs) ?? 0 : undefined,\n    preserveRatio: options.compress?.preserveRatio ?? 0.8,\n    compressionModel: options.compress?.model\n      ? createCompressionAdapter(options.compress.model)\n      : undefined,\n    onCompress: options.onCompress\n      ? (summary, count) => options.onCompress?.(summary.content, count)\n      : undefined,\n    onBeforeCompress: options.onBeforeCompress ?? options.onBudgetExceeded,\n  });\n\n  return {\n    specificationVersion: 'v3',\n\n    transformParams: async ({ params }) => {\n      let { prompt } = params;\n\n      // 1. Truncate large tool results\n      if (options.truncate) {\n        prompt = await truncateToolResults(prompt, options.truncate);\n      }\n\n      // 2. Compact (mechanical, zero LLM cost) via pruneMessages\n      if (options.compact) {\n        prompt = compactPrompt(prompt, options.compact);\n      }\n\n      // 3. Convert to IR and separate system messages from conversation.\n      // System messages are standing instructions — they must not be\n      // compressed away. Only conversation history goes through compact/compress.\n      const allIR = fromAISDK(prompt);\n      const systemMessages = allIR.filter((m) => m.role === 'system');\n      let conversation = allIR.filter((m) => m.role !== 'system');\n\n      // 4. Compress conversation history if over token budget\n      conversation = await janitor.compress(conversation);\n\n      // 5. Reassemble sandwich: system + conversation\n      const irMessages = [...systemMessages, ...conversation];\n\n      // 6. Convert back to AI SDK format\n      prompt = toAISDK(irMessages);\n\n      // 7. Dynamic state injection\n      if (options.dynamicState) {\n        prompt = await injectDynamicState(prompt, options.dynamicState);\n      }\n\n      // 8. Custom transform hook\n      if (options.transformContext) {\n        prompt = await options.transformContext(prompt);\n      }\n\n      return { ...params, prompt };\n    },\n\n    wrapGenerate: async ({ doGenerate }) => {\n      const result = await doGenerate();\n\n      if (result.usage?.inputTokens?.total != null) {\n        janitor.feedTokenUsage(result.usage.inputTokens.total);\n      } else if (!usageWarned && !options.tokenizer) {\n        usageWarned = true;\n        console.warn(\n          '[context-chef] Model response did not include usage.inputTokens.total. ' +\n            'Token-based compression may not trigger accurately. ' +\n            'Consider providing a tokenizer for precise token counting.',\n        );\n      }\n\n      return result;\n    },\n\n    wrapStream: async ({ doStream }) => {\n      const { stream, ...rest } = await doStream();\n\n      const transform = new TransformStream<LanguageModelV3StreamPart, LanguageModelV3StreamPart>({\n        transform(chunk, controller) {\n          if (chunk.type === 'finish') {\n            if (chunk.usage?.inputTokens?.total != null) {\n              janitor.feedTokenUsage(chunk.usage.inputTokens.total);\n            } else if (!usageWarned && !options.tokenizer) {\n              usageWarned = true;\n              console.warn(\n                '[context-chef] Stream finish did not include usage.inputTokens.total. ' +\n                  'Token-based compression may not trigger accurately. ' +\n                  'Consider providing a tokenizer for precise token counting.',\n              );\n            }\n          }\n          controller.enqueue(chunk);\n        },\n      });\n\n      return { ...rest, stream: stream.pipeThrough(transform) };\n    },\n  };\n}\n\n/**\n * Prunes a LanguageModelV3Prompt via AI SDK's pruneMessages.\n *\n * LanguageModelV3Message (from @ai-sdk/provider) and ModelMessage\n * (from @ai-sdk/provider-utils) share identical runtime structure but\n * differ at the TypeScript level (e.g. ImagePart, FilePart.data).\n * Since pruneMessages only filters — never transforms — every content\n * part in the output is an original V3 part, making the casts safe.\n */\nfunction compactPrompt(\n  prompt: LanguageModelV3Prompt,\n  config: Omit<Parameters<typeof pruneMessages>[0], 'messages'>,\n): LanguageModelV3Prompt {\n  const messages = prompt.map(\n    (msg) =>\n      ({\n        role: msg.role,\n        content: msg.content,\n        providerOptions: msg.providerOptions,\n      }) as ModelMessage,\n  );\n  const pruned = pruneMessages({ messages, ...config });\n  return pruned.map(\n    (msg) =>\n      ({\n        role: msg.role,\n        content: msg.content,\n        providerOptions: msg.providerOptions,\n      }) as LanguageModelV3Message,\n  );\n}\n\n/**\n * Injects dynamic state XML into the AI SDK prompt.\n *\n * - `last_user`: Appends to the last user message's content parts.\n *   Leverages Recency Bias for maximum LLM attention.\n * - `system`: Adds as a standalone system message at the end.\n */\nasync function injectDynamicState(\n  prompt: LanguageModelV3Prompt,\n  config: DynamicStateConfig,\n): Promise<LanguageModelV3Prompt> {\n  const state = await config.getState();\n  const xml = XmlGenerator.objectToXml(state, 'dynamic_state');\n  const placement = config.placement ?? 'last_user';\n\n  if (placement === 'system') {\n    return [...prompt, { role: 'system', content: `CURRENT TASK STATE:\\n${xml}` }];\n  }\n\n  // last_user: inject into the last user message\n  const result = [...prompt];\n  const stateBlock = `\\n\\n${xml}\\nAbove is the current system state. Use it to guide your next action.`;\n\n  for (let i = result.length - 1; i >= 0; i--) {\n    const msg = result[i];\n    if (msg.role === 'user') {\n      result[i] = {\n        ...msg,\n        content: [...msg.content, { type: 'text', text: stateBlock }],\n      };\n      return result;\n    }\n  }\n\n  // No user message found — append as new user message\n  result.push({\n    role: 'user',\n    content: [{ type: 'text', text: stateBlock.trim() }],\n  });\n  return result;\n}\n\n/**\n * Maps an IR role to a role accepted by generateText.\n * Tool messages are handled separately before this is called.\n */\nfunction toCompressRole(role: string): CompressRole {\n  if (role === 'system' || role === 'user' || role === 'assistant') return role;\n  return 'user';\n}\n\n/**\n * Adapts an AI SDK LanguageModelV3 into the compressionModel callback\n * that Janitor expects: (messages: Message[]) => Promise<string>\n *\n * Tool messages are converted to user messages describing the tool interaction,\n * since generateText only accepts system/user/assistant roles.\n */\nfunction createCompressionAdapter(\n  model: LanguageModelV3,\n): (messages: Message[]) => Promise<string> {\n  return async (messages: Message[]): Promise<string> => {\n    const formatted = messages.map((m): { role: CompressRole; content: string } => {\n      if (m.role === 'tool') {\n        return {\n          role: 'user' satisfies CompressRole,\n          content: `[Tool result${m.tool_call_id ? ` (${m.tool_call_id})` : ''}: ${m.content}]`,\n        };\n      }\n      if (m.role === 'assistant' && m.tool_calls?.length) {\n        const toolCallsDesc = m.tool_calls\n          .map((tc) => `[Called tool: ${tc.function.name}(${tc.function.arguments})]`)\n          .join('\\n');\n        return {\n          role: 'assistant' satisfies CompressRole,\n          content: m.content ? `${m.content}\\n${toolCallsDesc}` : toolCallsDesc,\n        };\n      }\n      return {\n        role: toCompressRole(m.role),\n        content: m.content,\n      };\n    });\n\n    const { text } = await generateText({\n      model,\n      messages: formatted,\n      maxOutputTokens: 2048,\n    });\n\n    return text || '[Compression produced no output]';\n  };\n}\n","import type { LanguageModelV3 } from '@ai-sdk/provider';\nimport { wrapLanguageModel } from 'ai';\n\nimport { createMiddleware } from './middleware';\nimport type { ContextChefOptions } from './types';\n\nexport { type AISDKMessage, fromAISDK, toAISDK } from './adapter';\nexport { createMiddleware } from './middleware';\nexport type {\n  CompactConfig,\n  CompressOptions,\n  ContextChefOptions,\n  DynamicStateConfig,\n  TruncateOptions,\n} from './types';\n\n/**\n * Wraps an AI SDK language model with context-chef middleware for\n * transparent history compression, tool result truncation, and token budget management.\n *\n * @example\n * ```typescript\n * import { withContextChef } from '@context-chef/ai-sdk-middleware';\n * import { openai } from '@ai-sdk/openai';\n * import { generateText } from 'ai';\n *\n * const model = withContextChef(openai('gpt-4o'), {\n *   contextWindow: 128_000,\n *   compress: { model: openai('gpt-4o-mini') },\n *   truncate: { threshold: 5000, headChars: 500, tailChars: 1000 },\n * });\n *\n * // Use exactly like normal — zero other code changes\n * const result = await generateText({ model, messages, tools });\n * ```\n */\nexport function withContextChef(\n  model: LanguageModelV3,\n  options: ContextChefOptions,\n): LanguageModelV3 {\n  const middleware = createMiddleware(options);\n  return wrapLanguageModel({ model, middleware });\n}\n"],"mappings":";;;;;;;;;;;;AAkCA,SAAgB,UAAU,QAA+C;CACvE,MAAM,WAA2B,EAAE;AAEnC,MAAK,MAAM,OAAO,QAAQ;AACxB,MAAI,IAAI,SAAS,UAAU;AACzB,YAAS,KAAK;IACZ,MAAM;IACN,SAAS,IAAI;IACb,GAAI,IAAI,kBAAkB,EAAE,kBAAkB,IAAI,iBAAiB,GAAG,EAAE;IACzE,CAAC;AACF;;AAGF,MAAI,IAAI,SAAS,QAAQ;GACvB,MAAM,OAAO,IAAI,QACd,QAAQ,MAAM,EAAE,SAAS,OAAO,CAChC,KAAK,MAAM,EAAE,KAAK,CAClB,KAAK,KAAK;AACb,YAAS,KAAK;IACZ,MAAM;IACN,SAAS;IACT,cAAc,IAAI;IAClB,eAAe;IACf,GAAI,IAAI,kBAAkB,EAAE,kBAAkB,IAAI,iBAAiB,GAAG,EAAE;IACzE,CAAC;AACF;;AAGF,MAAI,IAAI,SAAS,aAAa;GAC5B,MAAM,OAAiB,EAAE;GACzB,MAAM,YAAwB,EAAE;GAChC,IAAI;AAEJ,QAAK,MAAM,QAAQ,IAAI,QACrB,KAAI,KAAK,SAAS,OAAQ,MAAK,KAAK,KAAK,KAAK;YACrC,KAAK,SAAS,YACrB,WAAU,KAAK;IACb,IAAI,KAAK;IACT,MAAM;IACN,UAAU;KACR,MAAM,KAAK;KACX,WAAW,OAAO,KAAK,UAAU,WAAW,KAAK,QAAQ,KAAK,UAAU,KAAK,MAAM;KACpF;IACF,CAAC;YACO,KAAK,SAAS,YACvB,YAAW,EAAE,UAAU,KAAK,MAAM;GAItC,MAAM,aAAa,KAAK,KAAK,KAAK;GAClC,MAAM,IAAkB;IACtB,MAAM;IACN,SAAS;IACT,mBAAmB,IAAI;IACvB,eAAe;IACf,GAAI,IAAI,kBAAkB,EAAE,kBAAkB,IAAI,iBAAiB,GAAG,EAAE;IACzE;AACD,OAAI,UAAU,SAAS,EAAG,GAAE,aAAa;AACzC,OAAI,SAAU,GAAE,WAAW;AAC3B,YAAS,KAAK,EAAE;AAChB;;AAGF,MAAI,IAAI,SAAS,QACf;QAAK,MAAM,QAAQ,IAAI,QACrB,KAAI,KAAK,SAAS,eAAe;IAC/B,MAAM,OAAO,oBAAoB,KAAK,OAAO;AAC7C,aAAS,KAAK;KACZ,MAAM;KACN,SAAS;KACT,cAAc,KAAK;KACnB,cAAc,CAAC,KAAK;KACpB,eAAe;KACf,WAAW,KAAK;KACjB,CAAC;;;;AAMV,QAAO;;;;;AAMT,SAAS,QAAQ,KAA4B;AAC3C,QAAO;;;;;;;;;AAUT,SAAgB,QAAQ,UAA4C;CAClE,MAAM,SAAgC,EAAE;CAExC,IAAI,IAAI;AACR,QAAO,IAAI,SAAS,QAAQ;EAC1B,MAAM,MAAM,QAAQ,SAAS,GAAG;EAChC,MAAM,kBAAkB,IAAI,kBAAkB,UAAa,IAAI,kBAAkB,IAAI;AAErF,MAAI,IAAI,SAAS,UAAU;AACzB,UAAO,KAAK;IACV,MAAM;IACN,SAAS,IAAI;IACb,GAAI,IAAI,mBAAmB,EAAE,iBAAiB,IAAI,kBAAkB,GAAG,EAAE;IAC1E,CAAC;AACF;AACA;;AAGF,MAAI,IAAI,SAAS,QAAQ;AACvB,UAAO,KAAK;IACV,MAAM;IACN,SACE,CAAC,mBAAmB,IAAI,eACpB,IAAI,eACJ,CAAC;KAAE,MAAM;KAAQ,MAAM,IAAI;KAAS,CAAC;IAC3C,GAAI,IAAI,mBAAmB,EAAE,iBAAiB,IAAI,kBAAkB,GAAG,EAAE;IAC1E,CAAC;AACF;AACA;;AAGF,MAAI,IAAI,SAAS,aAAa;AAC5B,UAAO,KAAK;IACV,MAAM;IACN,SACE,CAAC,mBAAmB,IAAI,oBACpB,IAAI,oBACJ,CAAC;KAAE,MAAM;KAAQ,MAAM,IAAI;KAAS,CAAC;IAC3C,GAAI,IAAI,mBAAmB,EAAE,iBAAiB,IAAI,kBAAkB,GAAG,EAAE;IAC1E,CAAC;AACF;AACA;;AAGF,MAAI,IAAI,SAAS,QAAQ;GACvB,MAAM,cAA+C,EAAE;AACvD,UAAO,IAAI,SAAS,UAAU,SAAS,GAAG,SAAS,QAAQ;IACzD,MAAM,UAAU,QAAQ,SAAS,GAAG;AAIpC,QAAI,EAFF,QAAQ,kBAAkB,UAAa,QAAQ,kBAAkB,QAAQ,YAEtD,QAAQ,cAC3B;UAAK,MAAM,QAAQ,QAAQ,aACzB,KAAI,KAAK,SAAS,cAChB,aAAY,KAAK,KAAK;UAI1B,aAAY,KAAK;KACf,MAAM;KACN,YAAY,QAAQ,gBAAgB;KACpC,UAAU,QAAQ,aAAa;KAC/B,QAAQ;MAAE,MAAM;MAAQ,OAAO,QAAQ;MAAS;KACjD,CAAC;AAEJ;;AAEF,UAAO,KAAK;IAAE,MAAM;IAAQ,SAAS;IAAa,CAAC;AACnD;;AAGF;;AAGF,QAAO;;AAGT,SAAS,oBAAoB,QAAiD;AAC5E,SAAQ,OAAO,MAAf;EACE,KAAK;EACL,KAAK,aACH,QAAO,OAAO;EAChB,KAAK;EACL,KAAK,aACH,QAAO,KAAK,UAAU,OAAO,MAAM;EACrC,KAAK,UACH,QAAO,OAAO,MACX,KAAK,MAAO,EAAE,SAAS,SAAS,EAAE,OAAO,GAAI,CAC7C,OAAO,QAAQ,CACf,KAAK,KAAK;EACf,QACE,QAAO,KAAK,UAAU,OAAO;;;;;;;;;;AClNnC,eAAsB,oBACpB,QACA,SACgC;CAChC,MAAM,EAAE,WAAW,YAAY,GAAG,YAAY,KAAM,YAAY;CAEhE,MAAM,YAAY,UAAU,IAAIA,6BAAU;EAAE;EAAW,SAAS;EAAS,YAAY;EAAI,CAAC,GAAG;CAE7F,MAAM,SAAgC,EAAE;AAExC,MAAK,MAAM,OAAO,QAAQ;AACxB,MAAI,IAAI,SAAS,QAAQ;AACvB,UAAO,KAAK,IAAI;AAChB;;EAGF,MAAM,aAAiC,EAAE;AAEzC,OAAK,MAAM,QAAQ,IAAI,SAAS;AAC9B,OAAI,KAAK,SAAS,eAAe;AAC/B,eAAW,KAAK,KAAK;AACrB;;GAGF,MAAM,OAAO,YAAY,KAAK,OAAO;AACrC,OAAI,KAAK,UAAU,aAAa,YAAY,aAAa,KAAK,QAAQ;AACpE,eAAW,KAAK,KAAK;AACrB;;AAIF,OAAI,UACF,KAAI;IACF,MAAM,YAAY,MAAM,UAAU,aAAa,MAAM;KAAE;KAAW;KAAW;KAAW,CAAC;AACzF,eAAW,KAAK;KACd,GAAG;KACH,QAAQ;MACN,MAAM;MACN,OAAO,UAAU;MAClB;KACF,CAAyC;AAC1C;YACO,OAAO;AACd,YAAQ,KACN,gEAAgE,KAAK,WAAW,+CACjC,iBAAiB,QAAQ,MAAM,UAAU,OAAO,MAAM,GACtG;;GAML,MAAM,OAAO,KAAK,MAAM,GAAG,UAAU;GACrC,MAAM,OAAO,KAAK,MAAM,KAAK,SAAS,UAAU;GAGhD,MAAM,YAAY;IAChB;IACA,oBAJiB,KAAK,MAAM,KAAK,CAAC,OAIH,UAAU,KAAK,OAAO;IACrD;IACD,CACE,OAAO,QAAQ,CACf,KAAK,GAAG,CACR,MAAM;AAET,cAAW,KAAK;IACd,GAAG;IACH,QAAQ;KAAE,MAAM;KAAQ,OAAO;KAAW;IAC3C,CAAyC;;AAG5C,SAAO,KAAK;GAAE,GAAG;GAAK,SAAS;GAAY,CAAC;;AAG9C,QAAO;;AAGT,SAAS,YAAY,QAAiD;AACpE,SAAQ,OAAO,MAAf;EACE,KAAK;EACL,KAAK,aACH,QAAO,OAAO;EAChB,KAAK;EACL,KAAK,aACH,QAAO,KAAK,UAAU,OAAO,MAAM;EACrC,KAAK,UACH,QAAO,OAAO,MACX,KAAK,MAAwC,EAAE,SAAS,SAAU,EAAE,QAAQ,KAAM,GAAI,CACtF,OAAO,QAAQ,CACf,KAAK,KAAK;EACf,QACE,QAAO;;;;;;;;;;;;;ACjFb,SAAgB,iBAAiB,SAAsD;CACrF,IAAI,cAAc;CAElB,MAAM,UAAU,IAAIC,2BAAQ;EAC1B,eAAe,QAAQ;EACvB,WAAW,QAAQ,aAAa,SAAoB,QAAQ,YAAY,KAAK,IAAI,IAAI;EACrF,eAAe,QAAQ,UAAU,iBAAiB;EAClD,kBAAkB,QAAQ,UAAU,QAChC,yBAAyB,QAAQ,SAAS,MAAM,GAChD;EACJ,YAAY,QAAQ,cACf,SAAS,UAAU,QAAQ,aAAa,QAAQ,SAAS,MAAM,GAChE;EACJ,kBAAkB,QAAQ,oBAAoB,QAAQ;EACvD,CAAC;AAEF,QAAO;EACL,sBAAsB;EAEtB,iBAAiB,OAAO,EAAE,aAAa;GACrC,IAAI,EAAE,WAAW;AAGjB,OAAI,QAAQ,SACV,UAAS,MAAM,oBAAoB,QAAQ,QAAQ,SAAS;AAI9D,OAAI,QAAQ,QACV,UAAS,cAAc,QAAQ,QAAQ,QAAQ;GAMjD,MAAM,QAAQ,UAAU,OAAO;GAC/B,MAAM,iBAAiB,MAAM,QAAQ,MAAM,EAAE,SAAS,SAAS;GAC/D,IAAI,eAAe,MAAM,QAAQ,MAAM,EAAE,SAAS,SAAS;AAG3D,kBAAe,MAAM,QAAQ,SAAS,aAAa;AAMnD,YAAS,QAHU,CAAC,GAAG,gBAAgB,GAAG,aAAa,CAG3B;AAG5B,OAAI,QAAQ,aACV,UAAS,MAAM,mBAAmB,QAAQ,QAAQ,aAAa;AAIjE,OAAI,QAAQ,iBACV,UAAS,MAAM,QAAQ,iBAAiB,OAAO;AAGjD,UAAO;IAAE,GAAG;IAAQ;IAAQ;;EAG9B,cAAc,OAAO,EAAE,iBAAiB;GACtC,MAAM,SAAS,MAAM,YAAY;AAEjC,OAAI,OAAO,OAAO,aAAa,SAAS,KACtC,SAAQ,eAAe,OAAO,MAAM,YAAY,MAAM;YAC7C,CAAC,eAAe,CAAC,QAAQ,WAAW;AAC7C,kBAAc;AACd,YAAQ,KACN,wLAGD;;AAGH,UAAO;;EAGT,YAAY,OAAO,EAAE,eAAe;GAClC,MAAM,EAAE,QAAQ,GAAG,SAAS,MAAM,UAAU;GAE5C,MAAM,YAAY,IAAI,gBAAsE,EAC1F,UAAU,OAAO,YAAY;AAC3B,QAAI,MAAM,SAAS,UACjB;SAAI,MAAM,OAAO,aAAa,SAAS,KACrC,SAAQ,eAAe,MAAM,MAAM,YAAY,MAAM;cAC5C,CAAC,eAAe,CAAC,QAAQ,WAAW;AAC7C,oBAAc;AACd,cAAQ,KACN,uLAGD;;;AAGL,eAAW,QAAQ,MAAM;MAE5B,CAAC;AAEF,UAAO;IAAE,GAAG;IAAM,QAAQ,OAAO,YAAY,UAAU;IAAE;;EAE5D;;;;;;;;;;;AAYH,SAAS,cACP,QACA,QACuB;AAUvB,8BAD6B;EAAE,UARd,OAAO,KACrB,SACE;GACC,MAAM,IAAI;GACV,SAAS,IAAI;GACb,iBAAiB,IAAI;GACtB,EACJ;EACwC,GAAG;EAAQ,CAAC,CACvC,KACX,SACE;EACC,MAAM,IAAI;EACV,SAAS,IAAI;EACb,iBAAiB,IAAI;EACtB,EACJ;;;;;;;;;AAUH,eAAe,mBACb,QACA,QACgC;CAChC,MAAM,QAAQ,MAAM,OAAO,UAAU;CACrC,MAAM,MAAMC,gCAAa,YAAY,OAAO,gBAAgB;AAG5D,MAFkB,OAAO,aAAa,iBAEpB,SAChB,QAAO,CAAC,GAAG,QAAQ;EAAE,MAAM;EAAU,SAAS,wBAAwB;EAAO,CAAC;CAIhF,MAAM,SAAS,CAAC,GAAG,OAAO;CAC1B,MAAM,aAAa,OAAO,IAAI;AAE9B,MAAK,IAAI,IAAI,OAAO,SAAS,GAAG,KAAK,GAAG,KAAK;EAC3C,MAAM,MAAM,OAAO;AACnB,MAAI,IAAI,SAAS,QAAQ;AACvB,UAAO,KAAK;IACV,GAAG;IACH,SAAS,CAAC,GAAG,IAAI,SAAS;KAAE,MAAM;KAAQ,MAAM;KAAY,CAAC;IAC9D;AACD,UAAO;;;AAKX,QAAO,KAAK;EACV,MAAM;EACN,SAAS,CAAC;GAAE,MAAM;GAAQ,MAAM,WAAW,MAAM;GAAE,CAAC;EACrD,CAAC;AACF,QAAO;;;;;;AAOT,SAAS,eAAe,MAA4B;AAClD,KAAI,SAAS,YAAY,SAAS,UAAU,SAAS,YAAa,QAAO;AACzE,QAAO;;;;;;;;;AAUT,SAAS,yBACP,OAC0C;AAC1C,QAAO,OAAO,aAAyC;EAuBrD,MAAM,EAAE,SAAS,2BAAmB;GAClC;GACA,UAxBgB,SAAS,KAAK,MAA+C;AAC7E,QAAI,EAAE,SAAS,OACb,QAAO;KACL,MAAM;KACN,SAAS,eAAe,EAAE,eAAe,KAAK,EAAE,aAAa,KAAK,GAAG,IAAI,EAAE,QAAQ;KACpF;AAEH,QAAI,EAAE,SAAS,eAAe,EAAE,YAAY,QAAQ;KAClD,MAAM,gBAAgB,EAAE,WACrB,KAAK,OAAO,iBAAiB,GAAG,SAAS,KAAK,GAAG,GAAG,SAAS,UAAU,IAAI,CAC3E,KAAK,KAAK;AACb,YAAO;MACL,MAAM;MACN,SAAS,EAAE,UAAU,GAAG,EAAE,QAAQ,IAAI,kBAAkB;MACzD;;AAEH,WAAO;KACL,MAAM,eAAe,EAAE,KAAK;KAC5B,SAAS,EAAE;KACZ;KACD;GAKA,iBAAiB;GAClB,CAAC;AAEF,SAAO,QAAQ;;;;;;;;;;;;;;;;;;;;;;;;;;ACpNnB,SAAgB,gBACd,OACA,SACiB;AAEjB,kCAAyB;EAAE;EAAO,YADf,iBAAiB,QAAQ;EACE,CAAC"}

package/dist/index.d.cts

@@ -1,5 +1,5 @@
 import { LanguageModelV3, LanguageModelV3Message, LanguageModelV3Prompt, SharedV3ProviderOptions } from "@ai-sdk/provider";
-import { ClearTarget, Message, VFSStorageAdapter } from "@context-chef/core";
+import { Message, VFSStorageAdapter } from "@context-chef/core";
 import { LanguageModelMiddleware } from "ai";
 
 //#region src/types.d.ts
@@ -26,17 +26,35 @@ interface CompressOptions {
 }
 /**
  * Mechanical compaction options — zero LLM cost.
+ * Delegates to AI SDK's `pruneMessages` before IR conversion.
  * Runs before LLM-based compression to reduce token usage at no cost.
- *
- * **Important:** When using together with `compress`, only clear `thinking`.
- * Clearing `tool-result` before compression causes the compression model to
- * receive empty placeholders instead of actual tool outputs, producing
- * low-quality summaries. Leave tool-result management to compression's
- * turn-based splitting.
  */
 interface CompactConfig {
-  /** Which content types to clear from history. */
-  clear: ClearTarget[];
+  /**
+   * Controls removal of reasoning content from assistant messages.
+   * - `'all'`: Remove reasoning from all messages.
+   * - `'before-last-message'`: Keep reasoning only in the final message.
+   * - `'none'` (default): Keep all reasoning.
+   */
+  reasoning?: 'all' | 'before-last-message' | 'none';
+  /**
+   * Controls removal of tool-call, tool-result, and tool-approval chunks.
+   * - `'all'`: Remove all tool-related chunks.
+   * - `'before-last-message'`: Keep tool chunks only in the final message.
+   * - `'before-last-${N}-messages'`: Keep tool chunks in the last N messages.
+   * - `'none'`: Keep all tool chunks.
+   * - Array form allows per-tool control.
+   */
+  toolCalls?: 'all' | 'before-last-message' | `before-last-${number}-messages` | 'none' | Array<{
+    type: 'all' | 'before-last-message' | `before-last-${number}-messages`;
+    tools?: string[];
+  }>;
+  /**
+   * Whether to retain messages with no content after pruning.
+   * - `'remove'` (default): Exclude empty messages.
+   * - `'keep'`: Retain them.
+   */
+  emptyMessages?: 'keep' | 'remove';
 }
 /**
  * Dynamic state injection config.
@@ -64,10 +82,9 @@ interface ContextChefOptions {
   /** Enable tool result truncation. Omit for no truncation. */
   truncate?: TruncateOptions;
   /**
-   * Mechanical compaction before LLM compression.
-   * Clears specified content types (tool-result, thinking) at zero LLM cost.
+   * Mechanical compaction via AI SDK's `pruneMessages`.
+   * Prunes reasoning, tool calls, and empty messages at zero LLM cost.
    *
-   * When combined with `compress`, use `clear: ['thinking']` only.
    * See CompactConfig for details.
    */
   compact?: CompactConfig;

package/dist/index.d.mts

@@ -1,5 +1,5 @@
 import { LanguageModelMiddleware } from "ai";
-import { ClearTarget, Message, VFSStorageAdapter } from "@context-chef/core";
+import { Message, VFSStorageAdapter } from "@context-chef/core";
 import { LanguageModelV3, LanguageModelV3Message, LanguageModelV3Prompt, SharedV3ProviderOptions } from "@ai-sdk/provider";
 
 //#region src/types.d.ts
@@ -26,17 +26,35 @@ interface CompressOptions {
 }
 /**
  * Mechanical compaction options — zero LLM cost.
+ * Delegates to AI SDK's `pruneMessages` before IR conversion.
  * Runs before LLM-based compression to reduce token usage at no cost.
- *
- * **Important:** When using together with `compress`, only clear `thinking`.
- * Clearing `tool-result` before compression causes the compression model to
- * receive empty placeholders instead of actual tool outputs, producing
- * low-quality summaries. Leave tool-result management to compression's
- * turn-based splitting.
  */
 interface CompactConfig {
-  /** Which content types to clear from history. */
-  clear: ClearTarget[];
+  /**
+   * Controls removal of reasoning content from assistant messages.
+   * - `'all'`: Remove reasoning from all messages.
+   * - `'before-last-message'`: Keep reasoning only in the final message.
+   * - `'none'` (default): Keep all reasoning.
+   */
+  reasoning?: 'all' | 'before-last-message' | 'none';
+  /**
+   * Controls removal of tool-call, tool-result, and tool-approval chunks.
+   * - `'all'`: Remove all tool-related chunks.
+   * - `'before-last-message'`: Keep tool chunks only in the final message.
+   * - `'before-last-${N}-messages'`: Keep tool chunks in the last N messages.
+   * - `'none'`: Keep all tool chunks.
+   * - Array form allows per-tool control.
+   */
+  toolCalls?: 'all' | 'before-last-message' | `before-last-${number}-messages` | 'none' | Array<{
+    type: 'all' | 'before-last-message' | `before-last-${number}-messages`;
+    tools?: string[];
+  }>;
+  /**
+   * Whether to retain messages with no content after pruning.
+   * - `'remove'` (default): Exclude empty messages.
+   * - `'keep'`: Retain them.
+   */
+  emptyMessages?: 'keep' | 'remove';
 }
 /**
  * Dynamic state injection config.
@@ -64,10 +82,9 @@ interface ContextChefOptions {
   /** Enable tool result truncation. Omit for no truncation. */
   truncate?: TruncateOptions;
   /**
-   * Mechanical compaction before LLM compression.
-   * Clears specified content types (tool-result, thinking) at zero LLM cost.
+   * Mechanical compaction via AI SDK's `pruneMessages`.
+   * Prunes reasoning, tool calls, and empty messages at zero LLM cost.
    *
-   * When combined with `compress`, use `clear: ['thinking']` only.
    * See CompactConfig for details.
    */
   compact?: CompactConfig;

package/dist/index.mjs

@@ -1,5 +1,5 @@
-import { generateText, wrapLanguageModel } from "ai";
-import { Janitor, Offloader, Prompts, XmlGenerator } from "@context-chef/core";
+import { generateText, pruneMessages, wrapLanguageModel } from "ai";
+import { Janitor, Offloader, XmlGenerator } from "@context-chef/core";
 
 //#region src/adapter.ts
 /**
@@ -268,23 +268,11 @@ function createMiddleware(options) {
 		transformParams: async ({ params }) => {
 			let { prompt } = params;
 			if (options.truncate) prompt = await truncateToolResults(prompt, options.truncate);
+			if (options.compact) prompt = compactPrompt(prompt, options.compact);
 			const allIR = fromAISDK(prompt);
-			let systemMessages = allIR.filter((m) => m.role === "system");
+			const systemMessages = allIR.filter((m) => m.role === "system");
 			let conversation = allIR.filter((m) => m.role !== "system");
-			if (options.compact) {
-				const preCompact = conversation;
-				conversation = janitor.compact(conversation, options.compact);
-				if (options.compact.clear.includes("thinking")) {
-					for (let i = 0; i < conversation.length; i++) if (preCompact[i].thinking && !conversation[i].thinking) delete conversation[i]._assistantContent;
-				}
-			}
 			conversation = await janitor.compress(conversation);
-			if (options.compact) {
-				if (options.compact.clear.some((t) => t === "tool-result" || typeof t === "object" && t.target === "tool-result")) systemMessages = [...systemMessages, {
-					role: "system",
-					content: Prompts.TOOL_RESULT_CLEARED_INSTRUCTION
-				}];
-			}
 			prompt = toAISDK([...systemMessages, ...conversation]);
 			if (options.dynamicState) prompt = await injectDynamicState(prompt, options.dynamicState);
 			if (options.transformContext) prompt = await options.transformContext(prompt);
@@ -322,6 +310,29 @@ function createMiddleware(options) {
 	};
 }
 /**
+* Prunes a LanguageModelV3Prompt via AI SDK's pruneMessages.
+*
+* LanguageModelV3Message (from @ai-sdk/provider) and ModelMessage
+* (from @ai-sdk/provider-utils) share identical runtime structure but
+* differ at the TypeScript level (e.g. ImagePart, FilePart.data).
+* Since pruneMessages only filters — never transforms — every content
+* part in the output is an original V3 part, making the casts safe.
+*/
+function compactPrompt(prompt, config) {
+	return pruneMessages({
+		messages: prompt.map((msg) => ({
+			role: msg.role,
+			content: msg.content,
+			providerOptions: msg.providerOptions
+		})),
+		...config
+	}).map((msg) => ({
+		role: msg.role,
+		content: msg.content,
+		providerOptions: msg.providerOptions
+	}));
+}
+/**
 * Injects dynamic state XML into the AI SDK prompt.
 *
 * - `last_user`: Appends to the last user message's content parts.

package/dist/index.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"index.mjs","names":[],"sources":["../src/adapter.ts","../src/truncator.ts","../src/middleware.ts","../src/index.ts"],"sourcesContent":["import type {\n  LanguageModelV3Message,\n  LanguageModelV3Prompt,\n  LanguageModelV3ToolResultOutput,\n  LanguageModelV3ToolResultPart,\n  SharedV3ProviderOptions,\n} from '@ai-sdk/provider';\nimport type { Message, ToolCall } from '@context-chef/core';\n\n/** Content types for each AI SDK message role */\ntype UserContent = Extract<LanguageModelV3Message, { role: 'user' }>['content'];\ntype AssistantContent = Extract<LanguageModelV3Message, { role: 'assistant' }>['content'];\ntype ToolContent = Extract<LanguageModelV3Message, { role: 'tool' }>['content'];\n\n/**\n * Extended IR message with typed pass-through fields for lossless AI SDK round-trip.\n * Per-role content fields avoid union types, so no `as` casts are needed in `toAISDK`.\n */\nexport interface AISDKMessage extends Message {\n  _userContent?: UserContent;\n  _assistantContent?: AssistantContent;\n  _toolContent?: ToolContent;\n  _originalText?: string;\n  _providerOptions?: SharedV3ProviderOptions;\n  _toolName?: string;\n}\n\n/**\n * Converts an AI SDK V3 prompt to context-chef IR messages.\n *\n * Original AI SDK content is stored in per-role fields for lossless round-trip.\n * `_originalText` caches the extracted text so `toAISDK` can detect Janitor modifications.\n * `_providerOptions` preserves message-level provider options (e.g. Anthropic cache control).\n */\nexport function fromAISDK(prompt: LanguageModelV3Prompt): AISDKMessage[] {\n  const messages: AISDKMessage[] = [];\n\n  for (const msg of prompt) {\n    if (msg.role === 'system') {\n      messages.push({\n        role: 'system',\n        content: msg.content,\n        ...(msg.providerOptions ? { _providerOptions: msg.providerOptions } : {}),\n      });\n      continue;\n    }\n\n    if (msg.role === 'user') {\n      const text = msg.content\n        .filter((p) => p.type === 'text')\n        .map((p) => p.text)\n        .join('\\n');\n      messages.push({\n        role: 'user',\n        content: text,\n        _userContent: msg.content,\n        _originalText: text,\n        ...(msg.providerOptions ? { _providerOptions: msg.providerOptions } : {}),\n      });\n      continue;\n    }\n\n    if (msg.role === 'assistant') {\n      const text: string[] = [];\n      const toolCalls: ToolCall[] = [];\n      let thinking: { thinking: string } | undefined;\n\n      for (const part of msg.content) {\n        if (part.type === 'text') text.push(part.text);\n        else if (part.type === 'tool-call') {\n          toolCalls.push({\n            id: part.toolCallId,\n            type: 'function',\n            function: {\n              name: part.toolName,\n              arguments: typeof part.input === 'string' ? part.input : JSON.stringify(part.input),\n            },\n          });\n        } else if (part.type === 'reasoning') {\n          thinking = { thinking: part.text };\n        }\n      }\n\n      const joinedText = text.join('\\n');\n      const m: AISDKMessage = {\n        role: 'assistant',\n        content: joinedText,\n        _assistantContent: msg.content,\n        _originalText: joinedText,\n        ...(msg.providerOptions ? { _providerOptions: msg.providerOptions } : {}),\n      };\n      if (toolCalls.length > 0) m.tool_calls = toolCalls;\n      if (thinking) m.thinking = thinking;\n      messages.push(m);\n      continue;\n    }\n\n    if (msg.role === 'tool') {\n      for (const part of msg.content) {\n        if (part.type === 'tool-result') {\n          const text = stringifyToolOutput(part.output);\n          messages.push({\n            role: 'tool',\n            content: text,\n            tool_call_id: part.toolCallId,\n            _toolContent: [part],\n            _originalText: text,\n            _toolName: part.toolName,\n          });\n        }\n      }\n    }\n  }\n\n  return messages;\n}\n\n/**\n * Narrows a generic Message to AISDKMessage for typed access to pass-through fields.\n */\nfunction asAISDK(msg: Message): AISDKMessage {\n  return msg;\n}\n\n/**\n * Converts context-chef IR messages back to AI SDK V3 prompt format.\n *\n * Uses per-role original content when unmodified (detected via `_originalText`).\n * Falls back to constructing from IR fields when content was modified by Janitor\n * (e.g. compact() cleared tool results) or for new messages (e.g. compression summaries).\n */\nexport function toAISDK(messages: Message[]): LanguageModelV3Prompt {\n  const prompt: LanguageModelV3Prompt = [];\n\n  let i = 0;\n  while (i < messages.length) {\n    const msg = asAISDK(messages[i]);\n    const contentModified = msg._originalText !== undefined && msg._originalText !== msg.content;\n\n    if (msg.role === 'system') {\n      prompt.push({\n        role: 'system',\n        content: msg.content,\n        ...(msg._providerOptions ? { providerOptions: msg._providerOptions } : {}),\n      });\n      i++;\n      continue;\n    }\n\n    if (msg.role === 'user') {\n      prompt.push({\n        role: 'user',\n        content:\n          !contentModified && msg._userContent\n            ? msg._userContent\n            : [{ type: 'text', text: msg.content }],\n        ...(msg._providerOptions ? { providerOptions: msg._providerOptions } : {}),\n      });\n      i++;\n      continue;\n    }\n\n    if (msg.role === 'assistant') {\n      prompt.push({\n        role: 'assistant',\n        content:\n          !contentModified && msg._assistantContent\n            ? msg._assistantContent\n            : [{ type: 'text', text: msg.content }],\n        ...(msg._providerOptions ? { providerOptions: msg._providerOptions } : {}),\n      });\n      i++;\n      continue;\n    }\n\n    if (msg.role === 'tool') {\n      const toolResults: LanguageModelV3ToolResultPart[] = [];\n      while (i < messages.length && messages[i].role === 'tool') {\n        const toolMsg = asAISDK(messages[i]);\n        const toolModified =\n          toolMsg._originalText !== undefined && toolMsg._originalText !== toolMsg.content;\n\n        if (!toolModified && toolMsg._toolContent) {\n          for (const part of toolMsg._toolContent) {\n            if (part.type === 'tool-result') {\n              toolResults.push(part);\n            }\n          }\n        } else {\n          toolResults.push({\n            type: 'tool-result',\n            toolCallId: toolMsg.tool_call_id ?? '',\n            toolName: toolMsg._toolName ?? 'unknown',\n            output: { type: 'text', value: toolMsg.content },\n          });\n        }\n        i++;\n      }\n      prompt.push({ role: 'tool', content: toolResults });\n      continue;\n    }\n\n    i++;\n  }\n\n  return prompt;\n}\n\nfunction stringifyToolOutput(output: LanguageModelV3ToolResultOutput): string {\n  switch (output.type) {\n    case 'text':\n    case 'error-text':\n      return output.value;\n    case 'json':\n    case 'error-json':\n      return JSON.stringify(output.value);\n    case 'content':\n      return output.value\n        .map((v) => (v.type === 'text' ? v.text : ''))\n        .filter(Boolean)\n        .join('\\n');\n    default:\n      return JSON.stringify(output);\n  }\n}\n","import type {\n  LanguageModelV3Prompt,\n  LanguageModelV3ToolResultOutput,\n  LanguageModelV3ToolResultPart,\n} from '@ai-sdk/provider';\nimport { Offloader } from '@context-chef/core';\nimport type { TruncateOptions } from './types';\n\n/**\n * Truncates tool-result content within an AI SDK prompt when it exceeds the configured threshold.\n * When a storage adapter is provided, original content is persisted and a URI is included in the output.\n */\nexport async function truncateToolResults(\n  prompt: LanguageModelV3Prompt,\n  options: TruncateOptions,\n): Promise<LanguageModelV3Prompt> {\n  const { threshold, headChars = 0, tailChars = 1000, storage } = options;\n\n  const offloader = storage ? new Offloader({ threshold, adapter: storage, storageDir: '' }) : null;\n\n  const result: LanguageModelV3Prompt = [];\n\n  for (const msg of prompt) {\n    if (msg.role !== 'tool') {\n      result.push(msg);\n      continue;\n    }\n\n    const newContent: typeof msg.content = [];\n\n    for (const part of msg.content) {\n      if (part.type !== 'tool-result') {\n        newContent.push(part);\n        continue;\n      }\n\n      const text = extractText(part.output);\n      if (text.length <= threshold || headChars + tailChars >= text.length) {\n        newContent.push(part);\n        continue;\n      }\n\n      // With storage: use Offloader to persist original and get a URI-annotated truncation\n      if (offloader) {\n        try {\n          const vfsResult = await offloader.offloadAsync(text, { threshold, headChars, tailChars });\n          newContent.push({\n            ...part,\n            output: {\n              type: 'text',\n              value: vfsResult.content,\n            } satisfies LanguageModelV3ToolResultOutput,\n          } satisfies LanguageModelV3ToolResultPart);\n          continue;\n        } catch (error) {\n          console.warn(\n            `[context-chef] Storage adapter write failed for tool result (${part.toolCallId}). ` +\n              `Falling back to simple truncation. Error: ${error instanceof Error ? error.message : String(error)}`,\n          );\n          // Fall through to simple truncation below\n        }\n      }\n\n      // Without storage: simple truncation, original is discarded\n      const head = text.slice(0, headChars);\n      const tail = text.slice(text.length - tailChars);\n      const totalLines = text.split('\\n').length;\n\n      const truncated = [\n        head,\n        `\\n--- truncated (${totalLines} lines, ${text.length} chars total) ---\\n`,\n        tail,\n      ]\n        .filter(Boolean)\n        .join('')\n        .trim();\n\n      newContent.push({\n        ...part,\n        output: { type: 'text', value: truncated } satisfies LanguageModelV3ToolResultOutput,\n      } satisfies LanguageModelV3ToolResultPart);\n    }\n\n    result.push({ ...msg, content: newContent });\n  }\n\n  return result;\n}\n\nfunction extractText(output: LanguageModelV3ToolResultOutput): string {\n  switch (output.type) {\n    case 'text':\n    case 'error-text':\n      return output.value;\n    case 'json':\n    case 'error-json':\n      return JSON.stringify(output.value);\n    case 'content':\n      return output.value\n        .map((v: { type: string; text?: string }) => (v.type === 'text' ? (v.text ?? '') : ''))\n        .filter(Boolean)\n        .join('\\n');\n    default:\n      return '';\n  }\n}\n","import type {\n  LanguageModelV3,\n  LanguageModelV3Prompt,\n  LanguageModelV3StreamPart,\n} from '@ai-sdk/provider';\nimport { Janitor, type Message, Prompts, XmlGenerator } from '@context-chef/core';\nimport { generateText, type LanguageModelMiddleware } from 'ai';\n\nimport { fromAISDK, toAISDK } from './adapter';\nimport { truncateToolResults } from './truncator';\nimport type { ContextChefOptions, DynamicStateConfig } from './types';\n\ntype CompressRole = 'system' | 'user' | 'assistant';\n\n/**\n * Creates a LanguageModelMiddleware that transparently applies\n * context-chef compression and truncation to AI SDK model calls.\n *\n * The middleware holds a stateful Janitor instance that tracks\n * token usage across calls for compression decisions.\n */\nexport function createMiddleware(options: ContextChefOptions): LanguageModelMiddleware {\n  let usageWarned = false;\n\n  const janitor = new Janitor({\n    contextWindow: options.contextWindow,\n    tokenizer: options.tokenizer ? (msgs: Message[]) => options.tokenizer?.(msgs) ?? 0 : undefined,\n    preserveRatio: options.compress?.preserveRatio ?? 0.8,\n    compressionModel: options.compress?.model\n      ? createCompressionAdapter(options.compress.model)\n      : undefined,\n    onCompress: options.onCompress\n      ? (summary, count) => options.onCompress?.(summary.content, count)\n      : undefined,\n    onBeforeCompress: options.onBeforeCompress ?? options.onBudgetExceeded,\n  });\n\n  return {\n    specificationVersion: 'v3',\n\n    transformParams: async ({ params }) => {\n      let { prompt } = params;\n\n      // 1. Truncate large tool results\n      if (options.truncate) {\n        prompt = await truncateToolResults(prompt, options.truncate);\n      }\n\n      // 2. Convert to IR and separate system messages from conversation.\n      // System messages are standing instructions — they must not be\n      // compressed away. Only conversation history goes through compact/compress.\n      const allIR = fromAISDK(prompt);\n      let systemMessages = allIR.filter((m) => m.role === 'system');\n      let conversation = allIR.filter((m) => m.role !== 'system');\n\n      // 3. Compact (mechanical, zero LLM cost) before compression\n      if (options.compact) {\n        const preCompact = conversation;\n        conversation = janitor.compact(conversation, options.compact);\n\n        // When thinking is stripped, invalidate adapter pass-through\n        // so toAISDK reconstructs from IR fields (without reasoning)\n        if (options.compact.clear.includes('thinking')) {\n          for (let i = 0; i < conversation.length; i++) {\n            if (preCompact[i].thinking && !conversation[i].thinking) {\n              delete conversation[i]._assistantContent;\n            }\n          }\n        }\n      }\n\n      // 4. Compress conversation history if over token budget\n      conversation = await janitor.compress(conversation);\n\n      // 5. Rebuild system layer: original system messages + middleware instructions.\n      // Appended (not prepended) to preserve the user's system prompt ordering.\n      if (options.compact) {\n        const hasToolResultTarget = options.compact.clear.some(\n          (t) => t === 'tool-result' || (typeof t === 'object' && t.target === 'tool-result'),\n        );\n        if (hasToolResultTarget) {\n          systemMessages = [\n            ...systemMessages,\n            { role: 'system', content: Prompts.TOOL_RESULT_CLEARED_INSTRUCTION },\n          ];\n        }\n      }\n\n      // 6. Reassemble sandwich: system + conversation\n      const irMessages = [...systemMessages, ...conversation];\n\n      // 7. Convert back to AI SDK format\n      prompt = toAISDK(irMessages);\n\n      // 8. Dynamic state injection\n      if (options.dynamicState) {\n        prompt = await injectDynamicState(prompt, options.dynamicState);\n      }\n\n      // 9. Custom transform hook\n      if (options.transformContext) {\n        prompt = await options.transformContext(prompt);\n      }\n\n      return { ...params, prompt };\n    },\n\n    wrapGenerate: async ({ doGenerate }) => {\n      const result = await doGenerate();\n\n      if (result.usage?.inputTokens?.total != null) {\n        janitor.feedTokenUsage(result.usage.inputTokens.total);\n      } else if (!usageWarned && !options.tokenizer) {\n        usageWarned = true;\n        console.warn(\n          '[context-chef] Model response did not include usage.inputTokens.total. ' +\n            'Token-based compression may not trigger accurately. ' +\n            'Consider providing a tokenizer for precise token counting.',\n        );\n      }\n\n      return result;\n    },\n\n    wrapStream: async ({ doStream }) => {\n      const { stream, ...rest } = await doStream();\n\n      const transform = new TransformStream<LanguageModelV3StreamPart, LanguageModelV3StreamPart>({\n        transform(chunk, controller) {\n          if (chunk.type === 'finish') {\n            if (chunk.usage?.inputTokens?.total != null) {\n              janitor.feedTokenUsage(chunk.usage.inputTokens.total);\n            } else if (!usageWarned && !options.tokenizer) {\n              usageWarned = true;\n              console.warn(\n                '[context-chef] Stream finish did not include usage.inputTokens.total. ' +\n                  'Token-based compression may not trigger accurately. ' +\n                  'Consider providing a tokenizer for precise token counting.',\n              );\n            }\n          }\n          controller.enqueue(chunk);\n        },\n      });\n\n      return { ...rest, stream: stream.pipeThrough(transform) };\n    },\n  };\n}\n\n/**\n * Injects dynamic state XML into the AI SDK prompt.\n *\n * - `last_user`: Appends to the last user message's content parts.\n *   Leverages Recency Bias for maximum LLM attention.\n * - `system`: Adds as a standalone system message at the end.\n */\nasync function injectDynamicState(\n  prompt: LanguageModelV3Prompt,\n  config: DynamicStateConfig,\n): Promise<LanguageModelV3Prompt> {\n  const state = await config.getState();\n  const xml = XmlGenerator.objectToXml(state, 'dynamic_state');\n  const placement = config.placement ?? 'last_user';\n\n  if (placement === 'system') {\n    return [...prompt, { role: 'system', content: `CURRENT TASK STATE:\\n${xml}` }];\n  }\n\n  // last_user: inject into the last user message\n  const result = [...prompt];\n  const stateBlock = `\\n\\n${xml}\\nAbove is the current system state. Use it to guide your next action.`;\n\n  for (let i = result.length - 1; i >= 0; i--) {\n    const msg = result[i];\n    if (msg.role === 'user') {\n      result[i] = {\n        ...msg,\n        content: [...msg.content, { type: 'text', text: stateBlock }],\n      };\n      return result;\n    }\n  }\n\n  // No user message found — append as new user message\n  result.push({\n    role: 'user',\n    content: [{ type: 'text', text: stateBlock.trim() }],\n  });\n  return result;\n}\n\n/**\n * Maps an IR role to a role accepted by generateText.\n * Tool messages are handled separately before this is called.\n */\nfunction toCompressRole(role: string): CompressRole {\n  if (role === 'system' || role === 'user' || role === 'assistant') return role;\n  return 'user';\n}\n\n/**\n * Adapts an AI SDK LanguageModelV3 into the compressionModel callback\n * that Janitor expects: (messages: Message[]) => Promise<string>\n *\n * Tool messages are converted to user messages describing the tool interaction,\n * since generateText only accepts system/user/assistant roles.\n */\nfunction createCompressionAdapter(\n  model: LanguageModelV3,\n): (messages: Message[]) => Promise<string> {\n  return async (messages: Message[]): Promise<string> => {\n    const formatted = messages.map((m): { role: CompressRole; content: string } => {\n      if (m.role === 'tool') {\n        return {\n          role: 'user' satisfies CompressRole,\n          content: `[Tool result${m.tool_call_id ? ` (${m.tool_call_id})` : ''}: ${m.content}]`,\n        };\n      }\n      if (m.role === 'assistant' && m.tool_calls?.length) {\n        const toolCallsDesc = m.tool_calls\n          .map((tc) => `[Called tool: ${tc.function.name}(${tc.function.arguments})]`)\n          .join('\\n');\n        return {\n          role: 'assistant' satisfies CompressRole,\n          content: m.content ? `${m.content}\\n${toolCallsDesc}` : toolCallsDesc,\n        };\n      }\n      return {\n        role: toCompressRole(m.role),\n        content: m.content,\n      };\n    });\n\n    const { text } = await generateText({\n      model,\n      messages: formatted,\n      maxOutputTokens: 2048,\n    });\n\n    return text || '[Compression produced no output]';\n  };\n}\n","import type { LanguageModelV3 } from '@ai-sdk/provider';\nimport { wrapLanguageModel } from 'ai';\n\nimport { createMiddleware } from './middleware';\nimport type { ContextChefOptions } from './types';\n\nexport { type AISDKMessage, fromAISDK, toAISDK } from './adapter';\nexport { createMiddleware } from './middleware';\nexport type {\n  CompactConfig,\n  CompressOptions,\n  ContextChefOptions,\n  DynamicStateConfig,\n  TruncateOptions,\n} from './types';\n\n/**\n * Wraps an AI SDK language model with context-chef middleware for\n * transparent history compression, tool result truncation, and token budget management.\n *\n * @example\n * ```typescript\n * import { withContextChef } from '@context-chef/ai-sdk-middleware';\n * import { openai } from '@ai-sdk/openai';\n * import { generateText } from 'ai';\n *\n * const model = withContextChef(openai('gpt-4o'), {\n *   contextWindow: 128_000,\n *   compress: { model: openai('gpt-4o-mini') },\n *   truncate: { threshold: 5000, headChars: 500, tailChars: 1000 },\n * });\n *\n * // Use exactly like normal — zero other code changes\n * const result = await generateText({ model, messages, tools });\n * ```\n */\nexport function withContextChef(\n  model: LanguageModelV3,\n  options: ContextChefOptions,\n): LanguageModelV3 {\n  const middleware = createMiddleware(options);\n  return wrapLanguageModel({ model, middleware });\n}\n"],"mappings":";;;;;;;;;;;AAkCA,SAAgB,UAAU,QAA+C;CACvE,MAAM,WAA2B,EAAE;AAEnC,MAAK,MAAM,OAAO,QAAQ;AACxB,MAAI,IAAI,SAAS,UAAU;AACzB,YAAS,KAAK;IACZ,MAAM;IACN,SAAS,IAAI;IACb,GAAI,IAAI,kBAAkB,EAAE,kBAAkB,IAAI,iBAAiB,GAAG,EAAE;IACzE,CAAC;AACF;;AAGF,MAAI,IAAI,SAAS,QAAQ;GACvB,MAAM,OAAO,IAAI,QACd,QAAQ,MAAM,EAAE,SAAS,OAAO,CAChC,KAAK,MAAM,EAAE,KAAK,CAClB,KAAK,KAAK;AACb,YAAS,KAAK;IACZ,MAAM;IACN,SAAS;IACT,cAAc,IAAI;IAClB,eAAe;IACf,GAAI,IAAI,kBAAkB,EAAE,kBAAkB,IAAI,iBAAiB,GAAG,EAAE;IACzE,CAAC;AACF;;AAGF,MAAI,IAAI,SAAS,aAAa;GAC5B,MAAM,OAAiB,EAAE;GACzB,MAAM,YAAwB,EAAE;GAChC,IAAI;AAEJ,QAAK,MAAM,QAAQ,IAAI,QACrB,KAAI,KAAK,SAAS,OAAQ,MAAK,KAAK,KAAK,KAAK;YACrC,KAAK,SAAS,YACrB,WAAU,KAAK;IACb,IAAI,KAAK;IACT,MAAM;IACN,UAAU;KACR,MAAM,KAAK;KACX,WAAW,OAAO,KAAK,UAAU,WAAW,KAAK,QAAQ,KAAK,UAAU,KAAK,MAAM;KACpF;IACF,CAAC;YACO,KAAK,SAAS,YACvB,YAAW,EAAE,UAAU,KAAK,MAAM;GAItC,MAAM,aAAa,KAAK,KAAK,KAAK;GAClC,MAAM,IAAkB;IACtB,MAAM;IACN,SAAS;IACT,mBAAmB,IAAI;IACvB,eAAe;IACf,GAAI,IAAI,kBAAkB,EAAE,kBAAkB,IAAI,iBAAiB,GAAG,EAAE;IACzE;AACD,OAAI,UAAU,SAAS,EAAG,GAAE,aAAa;AACzC,OAAI,SAAU,GAAE,WAAW;AAC3B,YAAS,KAAK,EAAE;AAChB;;AAGF,MAAI,IAAI,SAAS,QACf;QAAK,MAAM,QAAQ,IAAI,QACrB,KAAI,KAAK,SAAS,eAAe;IAC/B,MAAM,OAAO,oBAAoB,KAAK,OAAO;AAC7C,aAAS,KAAK;KACZ,MAAM;KACN,SAAS;KACT,cAAc,KAAK;KACnB,cAAc,CAAC,KAAK;KACpB,eAAe;KACf,WAAW,KAAK;KACjB,CAAC;;;;AAMV,QAAO;;;;;AAMT,SAAS,QAAQ,KAA4B;AAC3C,QAAO;;;;;;;;;AAUT,SAAgB,QAAQ,UAA4C;CAClE,MAAM,SAAgC,EAAE;CAExC,IAAI,IAAI;AACR,QAAO,IAAI,SAAS,QAAQ;EAC1B,MAAM,MAAM,QAAQ,SAAS,GAAG;EAChC,MAAM,kBAAkB,IAAI,kBAAkB,UAAa,IAAI,kBAAkB,IAAI;AAErF,MAAI,IAAI,SAAS,UAAU;AACzB,UAAO,KAAK;IACV,MAAM;IACN,SAAS,IAAI;IACb,GAAI,IAAI,mBAAmB,EAAE,iBAAiB,IAAI,kBAAkB,GAAG,EAAE;IAC1E,CAAC;AACF;AACA;;AAGF,MAAI,IAAI,SAAS,QAAQ;AACvB,UAAO,KAAK;IACV,MAAM;IACN,SACE,CAAC,mBAAmB,IAAI,eACpB,IAAI,eACJ,CAAC;KAAE,MAAM;KAAQ,MAAM,IAAI;KAAS,CAAC;IAC3C,GAAI,IAAI,mBAAmB,EAAE,iBAAiB,IAAI,kBAAkB,GAAG,EAAE;IAC1E,CAAC;AACF;AACA;;AAGF,MAAI,IAAI,SAAS,aAAa;AAC5B,UAAO,KAAK;IACV,MAAM;IACN,SACE,CAAC,mBAAmB,IAAI,oBACpB,IAAI,oBACJ,CAAC;KAAE,MAAM;KAAQ,MAAM,IAAI;KAAS,CAAC;IAC3C,GAAI,IAAI,mBAAmB,EAAE,iBAAiB,IAAI,kBAAkB,GAAG,EAAE;IAC1E,CAAC;AACF;AACA;;AAGF,MAAI,IAAI,SAAS,QAAQ;GACvB,MAAM,cAA+C,EAAE;AACvD,UAAO,IAAI,SAAS,UAAU,SAAS,GAAG,SAAS,QAAQ;IACzD,MAAM,UAAU,QAAQ,SAAS,GAAG;AAIpC,QAAI,EAFF,QAAQ,kBAAkB,UAAa,QAAQ,kBAAkB,QAAQ,YAEtD,QAAQ,cAC3B;UAAK,MAAM,QAAQ,QAAQ,aACzB,KAAI,KAAK,SAAS,cAChB,aAAY,KAAK,KAAK;UAI1B,aAAY,KAAK;KACf,MAAM;KACN,YAAY,QAAQ,gBAAgB;KACpC,UAAU,QAAQ,aAAa;KAC/B,QAAQ;MAAE,MAAM;MAAQ,OAAO,QAAQ;MAAS;KACjD,CAAC;AAEJ;;AAEF,UAAO,KAAK;IAAE,MAAM;IAAQ,SAAS;IAAa,CAAC;AACnD;;AAGF;;AAGF,QAAO;;AAGT,SAAS,oBAAoB,QAAiD;AAC5E,SAAQ,OAAO,MAAf;EACE,KAAK;EACL,KAAK,aACH,QAAO,OAAO;EAChB,KAAK;EACL,KAAK,aACH,QAAO,KAAK,UAAU,OAAO,MAAM;EACrC,KAAK,UACH,QAAO,OAAO,MACX,KAAK,MAAO,EAAE,SAAS,SAAS,EAAE,OAAO,GAAI,CAC7C,OAAO,QAAQ,CACf,KAAK,KAAK;EACf,QACE,QAAO,KAAK,UAAU,OAAO;;;;;;;;;;AClNnC,eAAsB,oBACpB,QACA,SACgC;CAChC,MAAM,EAAE,WAAW,YAAY,GAAG,YAAY,KAAM,YAAY;CAEhE,MAAM,YAAY,UAAU,IAAI,UAAU;EAAE;EAAW,SAAS;EAAS,YAAY;EAAI,CAAC,GAAG;CAE7F,MAAM,SAAgC,EAAE;AAExC,MAAK,MAAM,OAAO,QAAQ;AACxB,MAAI,IAAI,SAAS,QAAQ;AACvB,UAAO,KAAK,IAAI;AAChB;;EAGF,MAAM,aAAiC,EAAE;AAEzC,OAAK,MAAM,QAAQ,IAAI,SAAS;AAC9B,OAAI,KAAK,SAAS,eAAe;AAC/B,eAAW,KAAK,KAAK;AACrB;;GAGF,MAAM,OAAO,YAAY,KAAK,OAAO;AACrC,OAAI,KAAK,UAAU,aAAa,YAAY,aAAa,KAAK,QAAQ;AACpE,eAAW,KAAK,KAAK;AACrB;;AAIF,OAAI,UACF,KAAI;IACF,MAAM,YAAY,MAAM,UAAU,aAAa,MAAM;KAAE;KAAW;KAAW;KAAW,CAAC;AACzF,eAAW,KAAK;KACd,GAAG;KACH,QAAQ;MACN,MAAM;MACN,OAAO,UAAU;MAClB;KACF,CAAyC;AAC1C;YACO,OAAO;AACd,YAAQ,KACN,gEAAgE,KAAK,WAAW,+CACjC,iBAAiB,QAAQ,MAAM,UAAU,OAAO,MAAM,GACtG;;GAML,MAAM,OAAO,KAAK,MAAM,GAAG,UAAU;GACrC,MAAM,OAAO,KAAK,MAAM,KAAK,SAAS,UAAU;GAGhD,MAAM,YAAY;IAChB;IACA,oBAJiB,KAAK,MAAM,KAAK,CAAC,OAIH,UAAU,KAAK,OAAO;IACrD;IACD,CACE,OAAO,QAAQ,CACf,KAAK,GAAG,CACR,MAAM;AAET,cAAW,KAAK;IACd,GAAG;IACH,QAAQ;KAAE,MAAM;KAAQ,OAAO;KAAW;IAC3C,CAAyC;;AAG5C,SAAO,KAAK;GAAE,GAAG;GAAK,SAAS;GAAY,CAAC;;AAG9C,QAAO;;AAGT,SAAS,YAAY,QAAiD;AACpE,SAAQ,OAAO,MAAf;EACE,KAAK;EACL,KAAK,aACH,QAAO,OAAO;EAChB,KAAK;EACL,KAAK,aACH,QAAO,KAAK,UAAU,OAAO,MAAM;EACrC,KAAK,UACH,QAAO,OAAO,MACX,KAAK,MAAwC,EAAE,SAAS,SAAU,EAAE,QAAQ,KAAM,GAAI,CACtF,OAAO,QAAQ,CACf,KAAK,KAAK;EACf,QACE,QAAO;;;;;;;;;;;;;AClFb,SAAgB,iBAAiB,SAAsD;CACrF,IAAI,cAAc;CAElB,MAAM,UAAU,IAAI,QAAQ;EAC1B,eAAe,QAAQ;EACvB,WAAW,QAAQ,aAAa,SAAoB,QAAQ,YAAY,KAAK,IAAI,IAAI;EACrF,eAAe,QAAQ,UAAU,iBAAiB;EAClD,kBAAkB,QAAQ,UAAU,QAChC,yBAAyB,QAAQ,SAAS,MAAM,GAChD;EACJ,YAAY,QAAQ,cACf,SAAS,UAAU,QAAQ,aAAa,QAAQ,SAAS,MAAM,GAChE;EACJ,kBAAkB,QAAQ,oBAAoB,QAAQ;EACvD,CAAC;AAEF,QAAO;EACL,sBAAsB;EAEtB,iBAAiB,OAAO,EAAE,aAAa;GACrC,IAAI,EAAE,WAAW;AAGjB,OAAI,QAAQ,SACV,UAAS,MAAM,oBAAoB,QAAQ,QAAQ,SAAS;GAM9D,MAAM,QAAQ,UAAU,OAAO;GAC/B,IAAI,iBAAiB,MAAM,QAAQ,MAAM,EAAE,SAAS,SAAS;GAC7D,IAAI,eAAe,MAAM,QAAQ,MAAM,EAAE,SAAS,SAAS;AAG3D,OAAI,QAAQ,SAAS;IACnB,MAAM,aAAa;AACnB,mBAAe,QAAQ,QAAQ,cAAc,QAAQ,QAAQ;AAI7D,QAAI,QAAQ,QAAQ,MAAM,SAAS,WAAW,EAC5C;UAAK,IAAI,IAAI,GAAG,IAAI,aAAa,QAAQ,IACvC,KAAI,WAAW,GAAG,YAAY,CAAC,aAAa,GAAG,SAC7C,QAAO,aAAa,GAAG;;;AAO/B,kBAAe,MAAM,QAAQ,SAAS,aAAa;AAInD,OAAI,QAAQ,SAIV;QAH4B,QAAQ,QAAQ,MAAM,MAC/C,MAAM,MAAM,iBAAkB,OAAO,MAAM,YAAY,EAAE,WAAW,cACtE,CAEC,kBAAiB,CACf,GAAG,gBACH;KAAE,MAAM;KAAU,SAAS,QAAQ;KAAiC,CACrE;;AAQL,YAAS,QAHU,CAAC,GAAG,gBAAgB,GAAG,aAAa,CAG3B;AAG5B,OAAI,QAAQ,aACV,UAAS,MAAM,mBAAmB,QAAQ,QAAQ,aAAa;AAIjE,OAAI,QAAQ,iBACV,UAAS,MAAM,QAAQ,iBAAiB,OAAO;AAGjD,UAAO;IAAE,GAAG;IAAQ;IAAQ;;EAG9B,cAAc,OAAO,EAAE,iBAAiB;GACtC,MAAM,SAAS,MAAM,YAAY;AAEjC,OAAI,OAAO,OAAO,aAAa,SAAS,KACtC,SAAQ,eAAe,OAAO,MAAM,YAAY,MAAM;YAC7C,CAAC,eAAe,CAAC,QAAQ,WAAW;AAC7C,kBAAc;AACd,YAAQ,KACN,wLAGD;;AAGH,UAAO;;EAGT,YAAY,OAAO,EAAE,eAAe;GAClC,MAAM,EAAE,QAAQ,GAAG,SAAS,MAAM,UAAU;GAE5C,MAAM,YAAY,IAAI,gBAAsE,EAC1F,UAAU,OAAO,YAAY;AAC3B,QAAI,MAAM,SAAS,UACjB;SAAI,MAAM,OAAO,aAAa,SAAS,KACrC,SAAQ,eAAe,MAAM,MAAM,YAAY,MAAM;cAC5C,CAAC,eAAe,CAAC,QAAQ,WAAW;AAC7C,oBAAc;AACd,cAAQ,KACN,uLAGD;;;AAGL,eAAW,QAAQ,MAAM;MAE5B,CAAC;AAEF,UAAO;IAAE,GAAG;IAAM,QAAQ,OAAO,YAAY,UAAU;IAAE;;EAE5D;;;;;;;;;AAUH,eAAe,mBACb,QACA,QACgC;CAChC,MAAM,QAAQ,MAAM,OAAO,UAAU;CACrC,MAAM,MAAM,aAAa,YAAY,OAAO,gBAAgB;AAG5D,MAFkB,OAAO,aAAa,iBAEpB,SAChB,QAAO,CAAC,GAAG,QAAQ;EAAE,MAAM;EAAU,SAAS,wBAAwB;EAAO,CAAC;CAIhF,MAAM,SAAS,CAAC,GAAG,OAAO;CAC1B,MAAM,aAAa,OAAO,IAAI;AAE9B,MAAK,IAAI,IAAI,OAAO,SAAS,GAAG,KAAK,GAAG,KAAK;EAC3C,MAAM,MAAM,OAAO;AACnB,MAAI,IAAI,SAAS,QAAQ;AACvB,UAAO,KAAK;IACV,GAAG;IACH,SAAS,CAAC,GAAG,IAAI,SAAS;KAAE,MAAM;KAAQ,MAAM;KAAY,CAAC;IAC9D;AACD,UAAO;;;AAKX,QAAO,KAAK;EACV,MAAM;EACN,SAAS,CAAC;GAAE,MAAM;GAAQ,MAAM,WAAW,MAAM;GAAE,CAAC;EACrD,CAAC;AACF,QAAO;;;;;;AAOT,SAAS,eAAe,MAA4B;AAClD,KAAI,SAAS,YAAY,SAAS,UAAU,SAAS,YAAa,QAAO;AACzE,QAAO;;;;;;;;;AAUT,SAAS,yBACP,OAC0C;AAC1C,QAAO,OAAO,aAAyC;EAuBrD,MAAM,EAAE,SAAS,MAAM,aAAa;GAClC;GACA,UAxBgB,SAAS,KAAK,MAA+C;AAC7E,QAAI,EAAE,SAAS,OACb,QAAO;KACL,MAAM;KACN,SAAS,eAAe,EAAE,eAAe,KAAK,EAAE,aAAa,KAAK,GAAG,IAAI,EAAE,QAAQ;KACpF;AAEH,QAAI,EAAE,SAAS,eAAe,EAAE,YAAY,QAAQ;KAClD,MAAM,gBAAgB,EAAE,WACrB,KAAK,OAAO,iBAAiB,GAAG,SAAS,KAAK,GAAG,GAAG,SAAS,UAAU,IAAI,CAC3E,KAAK,KAAK;AACb,YAAO;MACL,MAAM;MACN,SAAS,EAAE,UAAU,GAAG,EAAE,QAAQ,IAAI,kBAAkB;MACzD;;AAEH,WAAO;KACL,MAAM,eAAe,EAAE,KAAK;KAC5B,SAAS,EAAE;KACZ;KACD;GAKA,iBAAiB;GAClB,CAAC;AAEF,SAAO,QAAQ;;;;;;;;;;;;;;;;;;;;;;;;;;AC5MnB,SAAgB,gBACd,OACA,SACiB;AAEjB,QAAO,kBAAkB;EAAE;EAAO,YADf,iBAAiB,QAAQ;EACE,CAAC"}
+{"version":3,"file":"index.mjs","names":[],"sources":["../src/adapter.ts","../src/truncator.ts","../src/middleware.ts","../src/index.ts"],"sourcesContent":["import type {\n  LanguageModelV3Message,\n  LanguageModelV3Prompt,\n  LanguageModelV3ToolResultOutput,\n  LanguageModelV3ToolResultPart,\n  SharedV3ProviderOptions,\n} from '@ai-sdk/provider';\nimport type { Message, ToolCall } from '@context-chef/core';\n\n/** Content types for each AI SDK message role */\ntype UserContent = Extract<LanguageModelV3Message, { role: 'user' }>['content'];\ntype AssistantContent = Extract<LanguageModelV3Message, { role: 'assistant' }>['content'];\ntype ToolContent = Extract<LanguageModelV3Message, { role: 'tool' }>['content'];\n\n/**\n * Extended IR message with typed pass-through fields for lossless AI SDK round-trip.\n * Per-role content fields avoid union types, so no `as` casts are needed in `toAISDK`.\n */\nexport interface AISDKMessage extends Message {\n  _userContent?: UserContent;\n  _assistantContent?: AssistantContent;\n  _toolContent?: ToolContent;\n  _originalText?: string;\n  _providerOptions?: SharedV3ProviderOptions;\n  _toolName?: string;\n}\n\n/**\n * Converts an AI SDK V3 prompt to context-chef IR messages.\n *\n * Original AI SDK content is stored in per-role fields for lossless round-trip.\n * `_originalText` caches the extracted text so `toAISDK` can detect Janitor modifications.\n * `_providerOptions` preserves message-level provider options (e.g. Anthropic cache control).\n */\nexport function fromAISDK(prompt: LanguageModelV3Prompt): AISDKMessage[] {\n  const messages: AISDKMessage[] = [];\n\n  for (const msg of prompt) {\n    if (msg.role === 'system') {\n      messages.push({\n        role: 'system',\n        content: msg.content,\n        ...(msg.providerOptions ? { _providerOptions: msg.providerOptions } : {}),\n      });\n      continue;\n    }\n\n    if (msg.role === 'user') {\n      const text = msg.content\n        .filter((p) => p.type === 'text')\n        .map((p) => p.text)\n        .join('\\n');\n      messages.push({\n        role: 'user',\n        content: text,\n        _userContent: msg.content,\n        _originalText: text,\n        ...(msg.providerOptions ? { _providerOptions: msg.providerOptions } : {}),\n      });\n      continue;\n    }\n\n    if (msg.role === 'assistant') {\n      const text: string[] = [];\n      const toolCalls: ToolCall[] = [];\n      let thinking: { thinking: string } | undefined;\n\n      for (const part of msg.content) {\n        if (part.type === 'text') text.push(part.text);\n        else if (part.type === 'tool-call') {\n          toolCalls.push({\n            id: part.toolCallId,\n            type: 'function',\n            function: {\n              name: part.toolName,\n              arguments: typeof part.input === 'string' ? part.input : JSON.stringify(part.input),\n            },\n          });\n        } else if (part.type === 'reasoning') {\n          thinking = { thinking: part.text };\n        }\n      }\n\n      const joinedText = text.join('\\n');\n      const m: AISDKMessage = {\n        role: 'assistant',\n        content: joinedText,\n        _assistantContent: msg.content,\n        _originalText: joinedText,\n        ...(msg.providerOptions ? { _providerOptions: msg.providerOptions } : {}),\n      };\n      if (toolCalls.length > 0) m.tool_calls = toolCalls;\n      if (thinking) m.thinking = thinking;\n      messages.push(m);\n      continue;\n    }\n\n    if (msg.role === 'tool') {\n      for (const part of msg.content) {\n        if (part.type === 'tool-result') {\n          const text = stringifyToolOutput(part.output);\n          messages.push({\n            role: 'tool',\n            content: text,\n            tool_call_id: part.toolCallId,\n            _toolContent: [part],\n            _originalText: text,\n            _toolName: part.toolName,\n          });\n        }\n      }\n    }\n  }\n\n  return messages;\n}\n\n/**\n * Narrows a generic Message to AISDKMessage for typed access to pass-through fields.\n */\nfunction asAISDK(msg: Message): AISDKMessage {\n  return msg;\n}\n\n/**\n * Converts context-chef IR messages back to AI SDK V3 prompt format.\n *\n * Uses per-role original content when unmodified (detected via `_originalText`).\n * Falls back to constructing from IR fields when content was modified by Janitor\n * (e.g. compact() cleared tool results) or for new messages (e.g. compression summaries).\n */\nexport function toAISDK(messages: Message[]): LanguageModelV3Prompt {\n  const prompt: LanguageModelV3Prompt = [];\n\n  let i = 0;\n  while (i < messages.length) {\n    const msg = asAISDK(messages[i]);\n    const contentModified = msg._originalText !== undefined && msg._originalText !== msg.content;\n\n    if (msg.role === 'system') {\n      prompt.push({\n        role: 'system',\n        content: msg.content,\n        ...(msg._providerOptions ? { providerOptions: msg._providerOptions } : {}),\n      });\n      i++;\n      continue;\n    }\n\n    if (msg.role === 'user') {\n      prompt.push({\n        role: 'user',\n        content:\n          !contentModified && msg._userContent\n            ? msg._userContent\n            : [{ type: 'text', text: msg.content }],\n        ...(msg._providerOptions ? { providerOptions: msg._providerOptions } : {}),\n      });\n      i++;\n      continue;\n    }\n\n    if (msg.role === 'assistant') {\n      prompt.push({\n        role: 'assistant',\n        content:\n          !contentModified && msg._assistantContent\n            ? msg._assistantContent\n            : [{ type: 'text', text: msg.content }],\n        ...(msg._providerOptions ? { providerOptions: msg._providerOptions } : {}),\n      });\n      i++;\n      continue;\n    }\n\n    if (msg.role === 'tool') {\n      const toolResults: LanguageModelV3ToolResultPart[] = [];\n      while (i < messages.length && messages[i].role === 'tool') {\n        const toolMsg = asAISDK(messages[i]);\n        const toolModified =\n          toolMsg._originalText !== undefined && toolMsg._originalText !== toolMsg.content;\n\n        if (!toolModified && toolMsg._toolContent) {\n          for (const part of toolMsg._toolContent) {\n            if (part.type === 'tool-result') {\n              toolResults.push(part);\n            }\n          }\n        } else {\n          toolResults.push({\n            type: 'tool-result',\n            toolCallId: toolMsg.tool_call_id ?? '',\n            toolName: toolMsg._toolName ?? 'unknown',\n            output: { type: 'text', value: toolMsg.content },\n          });\n        }\n        i++;\n      }\n      prompt.push({ role: 'tool', content: toolResults });\n      continue;\n    }\n\n    i++;\n  }\n\n  return prompt;\n}\n\nfunction stringifyToolOutput(output: LanguageModelV3ToolResultOutput): string {\n  switch (output.type) {\n    case 'text':\n    case 'error-text':\n      return output.value;\n    case 'json':\n    case 'error-json':\n      return JSON.stringify(output.value);\n    case 'content':\n      return output.value\n        .map((v) => (v.type === 'text' ? v.text : ''))\n        .filter(Boolean)\n        .join('\\n');\n    default:\n      return JSON.stringify(output);\n  }\n}\n","import type {\n  LanguageModelV3Prompt,\n  LanguageModelV3ToolResultOutput,\n  LanguageModelV3ToolResultPart,\n} from '@ai-sdk/provider';\nimport { Offloader } from '@context-chef/core';\nimport type { TruncateOptions } from './types';\n\n/**\n * Truncates tool-result content within an AI SDK prompt when it exceeds the configured threshold.\n * When a storage adapter is provided, original content is persisted and a URI is included in the output.\n */\nexport async function truncateToolResults(\n  prompt: LanguageModelV3Prompt,\n  options: TruncateOptions,\n): Promise<LanguageModelV3Prompt> {\n  const { threshold, headChars = 0, tailChars = 1000, storage } = options;\n\n  const offloader = storage ? new Offloader({ threshold, adapter: storage, storageDir: '' }) : null;\n\n  const result: LanguageModelV3Prompt = [];\n\n  for (const msg of prompt) {\n    if (msg.role !== 'tool') {\n      result.push(msg);\n      continue;\n    }\n\n    const newContent: typeof msg.content = [];\n\n    for (const part of msg.content) {\n      if (part.type !== 'tool-result') {\n        newContent.push(part);\n        continue;\n      }\n\n      const text = extractText(part.output);\n      if (text.length <= threshold || headChars + tailChars >= text.length) {\n        newContent.push(part);\n        continue;\n      }\n\n      // With storage: use Offloader to persist original and get a URI-annotated truncation\n      if (offloader) {\n        try {\n          const vfsResult = await offloader.offloadAsync(text, { threshold, headChars, tailChars });\n          newContent.push({\n            ...part,\n            output: {\n              type: 'text',\n              value: vfsResult.content,\n            } satisfies LanguageModelV3ToolResultOutput,\n          } satisfies LanguageModelV3ToolResultPart);\n          continue;\n        } catch (error) {\n          console.warn(\n            `[context-chef] Storage adapter write failed for tool result (${part.toolCallId}). ` +\n              `Falling back to simple truncation. Error: ${error instanceof Error ? error.message : String(error)}`,\n          );\n          // Fall through to simple truncation below\n        }\n      }\n\n      // Without storage: simple truncation, original is discarded\n      const head = text.slice(0, headChars);\n      const tail = text.slice(text.length - tailChars);\n      const totalLines = text.split('\\n').length;\n\n      const truncated = [\n        head,\n        `\\n--- truncated (${totalLines} lines, ${text.length} chars total) ---\\n`,\n        tail,\n      ]\n        .filter(Boolean)\n        .join('')\n        .trim();\n\n      newContent.push({\n        ...part,\n        output: { type: 'text', value: truncated } satisfies LanguageModelV3ToolResultOutput,\n      } satisfies LanguageModelV3ToolResultPart);\n    }\n\n    result.push({ ...msg, content: newContent });\n  }\n\n  return result;\n}\n\nfunction extractText(output: LanguageModelV3ToolResultOutput): string {\n  switch (output.type) {\n    case 'text':\n    case 'error-text':\n      return output.value;\n    case 'json':\n    case 'error-json':\n      return JSON.stringify(output.value);\n    case 'content':\n      return output.value\n        .map((v: { type: string; text?: string }) => (v.type === 'text' ? (v.text ?? '') : ''))\n        .filter(Boolean)\n        .join('\\n');\n    default:\n      return '';\n  }\n}\n","import type {\n  LanguageModelV3,\n  LanguageModelV3Message,\n  LanguageModelV3Prompt,\n  LanguageModelV3StreamPart,\n} from '@ai-sdk/provider';\nimport { Janitor, type Message, XmlGenerator } from '@context-chef/core';\nimport { generateText, type LanguageModelMiddleware, type ModelMessage, pruneMessages } from 'ai';\n\nimport { fromAISDK, toAISDK } from './adapter';\nimport { truncateToolResults } from './truncator';\nimport type { ContextChefOptions, DynamicStateConfig } from './types';\n\ntype CompressRole = 'system' | 'user' | 'assistant';\n\n/**\n * Creates a LanguageModelMiddleware that transparently applies\n * context-chef compression and truncation to AI SDK model calls.\n *\n * The middleware holds a stateful Janitor instance that tracks\n * token usage across calls for compression decisions.\n */\nexport function createMiddleware(options: ContextChefOptions): LanguageModelMiddleware {\n  let usageWarned = false;\n\n  const janitor = new Janitor({\n    contextWindow: options.contextWindow,\n    tokenizer: options.tokenizer ? (msgs: Message[]) => options.tokenizer?.(msgs) ?? 0 : undefined,\n    preserveRatio: options.compress?.preserveRatio ?? 0.8,\n    compressionModel: options.compress?.model\n      ? createCompressionAdapter(options.compress.model)\n      : undefined,\n    onCompress: options.onCompress\n      ? (summary, count) => options.onCompress?.(summary.content, count)\n      : undefined,\n    onBeforeCompress: options.onBeforeCompress ?? options.onBudgetExceeded,\n  });\n\n  return {\n    specificationVersion: 'v3',\n\n    transformParams: async ({ params }) => {\n      let { prompt } = params;\n\n      // 1. Truncate large tool results\n      if (options.truncate) {\n        prompt = await truncateToolResults(prompt, options.truncate);\n      }\n\n      // 2. Compact (mechanical, zero LLM cost) via pruneMessages\n      if (options.compact) {\n        prompt = compactPrompt(prompt, options.compact);\n      }\n\n      // 3. Convert to IR and separate system messages from conversation.\n      // System messages are standing instructions — they must not be\n      // compressed away. Only conversation history goes through compact/compress.\n      const allIR = fromAISDK(prompt);\n      const systemMessages = allIR.filter((m) => m.role === 'system');\n      let conversation = allIR.filter((m) => m.role !== 'system');\n\n      // 4. Compress conversation history if over token budget\n      conversation = await janitor.compress(conversation);\n\n      // 5. Reassemble sandwich: system + conversation\n      const irMessages = [...systemMessages, ...conversation];\n\n      // 6. Convert back to AI SDK format\n      prompt = toAISDK(irMessages);\n\n      // 7. Dynamic state injection\n      if (options.dynamicState) {\n        prompt = await injectDynamicState(prompt, options.dynamicState);\n      }\n\n      // 8. Custom transform hook\n      if (options.transformContext) {\n        prompt = await options.transformContext(prompt);\n      }\n\n      return { ...params, prompt };\n    },\n\n    wrapGenerate: async ({ doGenerate }) => {\n      const result = await doGenerate();\n\n      if (result.usage?.inputTokens?.total != null) {\n        janitor.feedTokenUsage(result.usage.inputTokens.total);\n      } else if (!usageWarned && !options.tokenizer) {\n        usageWarned = true;\n        console.warn(\n          '[context-chef] Model response did not include usage.inputTokens.total. ' +\n            'Token-based compression may not trigger accurately. ' +\n            'Consider providing a tokenizer for precise token counting.',\n        );\n      }\n\n      return result;\n    },\n\n    wrapStream: async ({ doStream }) => {\n      const { stream, ...rest } = await doStream();\n\n      const transform = new TransformStream<LanguageModelV3StreamPart, LanguageModelV3StreamPart>({\n        transform(chunk, controller) {\n          if (chunk.type === 'finish') {\n            if (chunk.usage?.inputTokens?.total != null) {\n              janitor.feedTokenUsage(chunk.usage.inputTokens.total);\n            } else if (!usageWarned && !options.tokenizer) {\n              usageWarned = true;\n              console.warn(\n                '[context-chef] Stream finish did not include usage.inputTokens.total. ' +\n                  'Token-based compression may not trigger accurately. ' +\n                  'Consider providing a tokenizer for precise token counting.',\n              );\n            }\n          }\n          controller.enqueue(chunk);\n        },\n      });\n\n      return { ...rest, stream: stream.pipeThrough(transform) };\n    },\n  };\n}\n\n/**\n * Prunes a LanguageModelV3Prompt via AI SDK's pruneMessages.\n *\n * LanguageModelV3Message (from @ai-sdk/provider) and ModelMessage\n * (from @ai-sdk/provider-utils) share identical runtime structure but\n * differ at the TypeScript level (e.g. ImagePart, FilePart.data).\n * Since pruneMessages only filters — never transforms — every content\n * part in the output is an original V3 part, making the casts safe.\n */\nfunction compactPrompt(\n  prompt: LanguageModelV3Prompt,\n  config: Omit<Parameters<typeof pruneMessages>[0], 'messages'>,\n): LanguageModelV3Prompt {\n  const messages = prompt.map(\n    (msg) =>\n      ({\n        role: msg.role,\n        content: msg.content,\n        providerOptions: msg.providerOptions,\n      }) as ModelMessage,\n  );\n  const pruned = pruneMessages({ messages, ...config });\n  return pruned.map(\n    (msg) =>\n      ({\n        role: msg.role,\n        content: msg.content,\n        providerOptions: msg.providerOptions,\n      }) as LanguageModelV3Message,\n  );\n}\n\n/**\n * Injects dynamic state XML into the AI SDK prompt.\n *\n * - `last_user`: Appends to the last user message's content parts.\n *   Leverages Recency Bias for maximum LLM attention.\n * - `system`: Adds as a standalone system message at the end.\n */\nasync function injectDynamicState(\n  prompt: LanguageModelV3Prompt,\n  config: DynamicStateConfig,\n): Promise<LanguageModelV3Prompt> {\n  const state = await config.getState();\n  const xml = XmlGenerator.objectToXml(state, 'dynamic_state');\n  const placement = config.placement ?? 'last_user';\n\n  if (placement === 'system') {\n    return [...prompt, { role: 'system', content: `CURRENT TASK STATE:\\n${xml}` }];\n  }\n\n  // last_user: inject into the last user message\n  const result = [...prompt];\n  const stateBlock = `\\n\\n${xml}\\nAbove is the current system state. Use it to guide your next action.`;\n\n  for (let i = result.length - 1; i >= 0; i--) {\n    const msg = result[i];\n    if (msg.role === 'user') {\n      result[i] = {\n        ...msg,\n        content: [...msg.content, { type: 'text', text: stateBlock }],\n      };\n      return result;\n    }\n  }\n\n  // No user message found — append as new user message\n  result.push({\n    role: 'user',\n    content: [{ type: 'text', text: stateBlock.trim() }],\n  });\n  return result;\n}\n\n/**\n * Maps an IR role to a role accepted by generateText.\n * Tool messages are handled separately before this is called.\n */\nfunction toCompressRole(role: string): CompressRole {\n  if (role === 'system' || role === 'user' || role === 'assistant') return role;\n  return 'user';\n}\n\n/**\n * Adapts an AI SDK LanguageModelV3 into the compressionModel callback\n * that Janitor expects: (messages: Message[]) => Promise<string>\n *\n * Tool messages are converted to user messages describing the tool interaction,\n * since generateText only accepts system/user/assistant roles.\n */\nfunction createCompressionAdapter(\n  model: LanguageModelV3,\n): (messages: Message[]) => Promise<string> {\n  return async (messages: Message[]): Promise<string> => {\n    const formatted = messages.map((m): { role: CompressRole; content: string } => {\n      if (m.role === 'tool') {\n        return {\n          role: 'user' satisfies CompressRole,\n          content: `[Tool result${m.tool_call_id ? ` (${m.tool_call_id})` : ''}: ${m.content}]`,\n        };\n      }\n      if (m.role === 'assistant' && m.tool_calls?.length) {\n        const toolCallsDesc = m.tool_calls\n          .map((tc) => `[Called tool: ${tc.function.name}(${tc.function.arguments})]`)\n          .join('\\n');\n        return {\n          role: 'assistant' satisfies CompressRole,\n          content: m.content ? `${m.content}\\n${toolCallsDesc}` : toolCallsDesc,\n        };\n      }\n      return {\n        role: toCompressRole(m.role),\n        content: m.content,\n      };\n    });\n\n    const { text } = await generateText({\n      model,\n      messages: formatted,\n      maxOutputTokens: 2048,\n    });\n\n    return text || '[Compression produced no output]';\n  };\n}\n","import type { LanguageModelV3 } from '@ai-sdk/provider';\nimport { wrapLanguageModel } from 'ai';\n\nimport { createMiddleware } from './middleware';\nimport type { ContextChefOptions } from './types';\n\nexport { type AISDKMessage, fromAISDK, toAISDK } from './adapter';\nexport { createMiddleware } from './middleware';\nexport type {\n  CompactConfig,\n  CompressOptions,\n  ContextChefOptions,\n  DynamicStateConfig,\n  TruncateOptions,\n} from './types';\n\n/**\n * Wraps an AI SDK language model with context-chef middleware for\n * transparent history compression, tool result truncation, and token budget management.\n *\n * @example\n * ```typescript\n * import { withContextChef } from '@context-chef/ai-sdk-middleware';\n * import { openai } from '@ai-sdk/openai';\n * import { generateText } from 'ai';\n *\n * const model = withContextChef(openai('gpt-4o'), {\n *   contextWindow: 128_000,\n *   compress: { model: openai('gpt-4o-mini') },\n *   truncate: { threshold: 5000, headChars: 500, tailChars: 1000 },\n * });\n *\n * // Use exactly like normal — zero other code changes\n * const result = await generateText({ model, messages, tools });\n * ```\n */\nexport function withContextChef(\n  model: LanguageModelV3,\n  options: ContextChefOptions,\n): LanguageModelV3 {\n  const middleware = createMiddleware(options);\n  return wrapLanguageModel({ model, middleware });\n}\n"],"mappings":";;;;;;;;;;;AAkCA,SAAgB,UAAU,QAA+C;CACvE,MAAM,WAA2B,EAAE;AAEnC,MAAK,MAAM,OAAO,QAAQ;AACxB,MAAI,IAAI,SAAS,UAAU;AACzB,YAAS,KAAK;IACZ,MAAM;IACN,SAAS,IAAI;IACb,GAAI,IAAI,kBAAkB,EAAE,kBAAkB,IAAI,iBAAiB,GAAG,EAAE;IACzE,CAAC;AACF;;AAGF,MAAI,IAAI,SAAS,QAAQ;GACvB,MAAM,OAAO,IAAI,QACd,QAAQ,MAAM,EAAE,SAAS,OAAO,CAChC,KAAK,MAAM,EAAE,KAAK,CAClB,KAAK,KAAK;AACb,YAAS,KAAK;IACZ,MAAM;IACN,SAAS;IACT,cAAc,IAAI;IAClB,eAAe;IACf,GAAI,IAAI,kBAAkB,EAAE,kBAAkB,IAAI,iBAAiB,GAAG,EAAE;IACzE,CAAC;AACF;;AAGF,MAAI,IAAI,SAAS,aAAa;GAC5B,MAAM,OAAiB,EAAE;GACzB,MAAM,YAAwB,EAAE;GAChC,IAAI;AAEJ,QAAK,MAAM,QAAQ,IAAI,QACrB,KAAI,KAAK,SAAS,OAAQ,MAAK,KAAK,KAAK,KAAK;YACrC,KAAK,SAAS,YACrB,WAAU,KAAK;IACb,IAAI,KAAK;IACT,MAAM;IACN,UAAU;KACR,MAAM,KAAK;KACX,WAAW,OAAO,KAAK,UAAU,WAAW,KAAK,QAAQ,KAAK,UAAU,KAAK,MAAM;KACpF;IACF,CAAC;YACO,KAAK,SAAS,YACvB,YAAW,EAAE,UAAU,KAAK,MAAM;GAItC,MAAM,aAAa,KAAK,KAAK,KAAK;GAClC,MAAM,IAAkB;IACtB,MAAM;IACN,SAAS;IACT,mBAAmB,IAAI;IACvB,eAAe;IACf,GAAI,IAAI,kBAAkB,EAAE,kBAAkB,IAAI,iBAAiB,GAAG,EAAE;IACzE;AACD,OAAI,UAAU,SAAS,EAAG,GAAE,aAAa;AACzC,OAAI,SAAU,GAAE,WAAW;AAC3B,YAAS,KAAK,EAAE;AAChB;;AAGF,MAAI,IAAI,SAAS,QACf;QAAK,MAAM,QAAQ,IAAI,QACrB,KAAI,KAAK,SAAS,eAAe;IAC/B,MAAM,OAAO,oBAAoB,KAAK,OAAO;AAC7C,aAAS,KAAK;KACZ,MAAM;KACN,SAAS;KACT,cAAc,KAAK;KACnB,cAAc,CAAC,KAAK;KACpB,eAAe;KACf,WAAW,KAAK;KACjB,CAAC;;;;AAMV,QAAO;;;;;AAMT,SAAS,QAAQ,KAA4B;AAC3C,QAAO;;;;;;;;;AAUT,SAAgB,QAAQ,UAA4C;CAClE,MAAM,SAAgC,EAAE;CAExC,IAAI,IAAI;AACR,QAAO,IAAI,SAAS,QAAQ;EAC1B,MAAM,MAAM,QAAQ,SAAS,GAAG;EAChC,MAAM,kBAAkB,IAAI,kBAAkB,UAAa,IAAI,kBAAkB,IAAI;AAErF,MAAI,IAAI,SAAS,UAAU;AACzB,UAAO,KAAK;IACV,MAAM;IACN,SAAS,IAAI;IACb,GAAI,IAAI,mBAAmB,EAAE,iBAAiB,IAAI,kBAAkB,GAAG,EAAE;IAC1E,CAAC;AACF;AACA;;AAGF,MAAI,IAAI,SAAS,QAAQ;AACvB,UAAO,KAAK;IACV,MAAM;IACN,SACE,CAAC,mBAAmB,IAAI,eACpB,IAAI,eACJ,CAAC;KAAE,MAAM;KAAQ,MAAM,IAAI;KAAS,CAAC;IAC3C,GAAI,IAAI,mBAAmB,EAAE,iBAAiB,IAAI,kBAAkB,GAAG,EAAE;IAC1E,CAAC;AACF;AACA;;AAGF,MAAI,IAAI,SAAS,aAAa;AAC5B,UAAO,KAAK;IACV,MAAM;IACN,SACE,CAAC,mBAAmB,IAAI,oBACpB,IAAI,oBACJ,CAAC;KAAE,MAAM;KAAQ,MAAM,IAAI;KAAS,CAAC;IAC3C,GAAI,IAAI,mBAAmB,EAAE,iBAAiB,IAAI,kBAAkB,GAAG,EAAE;IAC1E,CAAC;AACF;AACA;;AAGF,MAAI,IAAI,SAAS,QAAQ;GACvB,MAAM,cAA+C,EAAE;AACvD,UAAO,IAAI,SAAS,UAAU,SAAS,GAAG,SAAS,QAAQ;IACzD,MAAM,UAAU,QAAQ,SAAS,GAAG;AAIpC,QAAI,EAFF,QAAQ,kBAAkB,UAAa,QAAQ,kBAAkB,QAAQ,YAEtD,QAAQ,cAC3B;UAAK,MAAM,QAAQ,QAAQ,aACzB,KAAI,KAAK,SAAS,cAChB,aAAY,KAAK,KAAK;UAI1B,aAAY,KAAK;KACf,MAAM;KACN,YAAY,QAAQ,gBAAgB;KACpC,UAAU,QAAQ,aAAa;KAC/B,QAAQ;MAAE,MAAM;MAAQ,OAAO,QAAQ;MAAS;KACjD,CAAC;AAEJ;;AAEF,UAAO,KAAK;IAAE,MAAM;IAAQ,SAAS;IAAa,CAAC;AACnD;;AAGF;;AAGF,QAAO;;AAGT,SAAS,oBAAoB,QAAiD;AAC5E,SAAQ,OAAO,MAAf;EACE,KAAK;EACL,KAAK,aACH,QAAO,OAAO;EAChB,KAAK;EACL,KAAK,aACH,QAAO,KAAK,UAAU,OAAO,MAAM;EACrC,KAAK,UACH,QAAO,OAAO,MACX,KAAK,MAAO,EAAE,SAAS,SAAS,EAAE,OAAO,GAAI,CAC7C,OAAO,QAAQ,CACf,KAAK,KAAK;EACf,QACE,QAAO,KAAK,UAAU,OAAO;;;;;;;;;;AClNnC,eAAsB,oBACpB,QACA,SACgC;CAChC,MAAM,EAAE,WAAW,YAAY,GAAG,YAAY,KAAM,YAAY;CAEhE,MAAM,YAAY,UAAU,IAAI,UAAU;EAAE;EAAW,SAAS;EAAS,YAAY;EAAI,CAAC,GAAG;CAE7F,MAAM,SAAgC,EAAE;AAExC,MAAK,MAAM,OAAO,QAAQ;AACxB,MAAI,IAAI,SAAS,QAAQ;AACvB,UAAO,KAAK,IAAI;AAChB;;EAGF,MAAM,aAAiC,EAAE;AAEzC,OAAK,MAAM,QAAQ,IAAI,SAAS;AAC9B,OAAI,KAAK,SAAS,eAAe;AAC/B,eAAW,KAAK,KAAK;AACrB;;GAGF,MAAM,OAAO,YAAY,KAAK,OAAO;AACrC,OAAI,KAAK,UAAU,aAAa,YAAY,aAAa,KAAK,QAAQ;AACpE,eAAW,KAAK,KAAK;AACrB;;AAIF,OAAI,UACF,KAAI;IACF,MAAM,YAAY,MAAM,UAAU,aAAa,MAAM;KAAE;KAAW;KAAW;KAAW,CAAC;AACzF,eAAW,KAAK;KACd,GAAG;KACH,QAAQ;MACN,MAAM;MACN,OAAO,UAAU;MAClB;KACF,CAAyC;AAC1C;YACO,OAAO;AACd,YAAQ,KACN,gEAAgE,KAAK,WAAW,+CACjC,iBAAiB,QAAQ,MAAM,UAAU,OAAO,MAAM,GACtG;;GAML,MAAM,OAAO,KAAK,MAAM,GAAG,UAAU;GACrC,MAAM,OAAO,KAAK,MAAM,KAAK,SAAS,UAAU;GAGhD,MAAM,YAAY;IAChB;IACA,oBAJiB,KAAK,MAAM,KAAK,CAAC,OAIH,UAAU,KAAK,OAAO;IACrD;IACD,CACE,OAAO,QAAQ,CACf,KAAK,GAAG,CACR,MAAM;AAET,cAAW,KAAK;IACd,GAAG;IACH,QAAQ;KAAE,MAAM;KAAQ,OAAO;KAAW;IAC3C,CAAyC;;AAG5C,SAAO,KAAK;GAAE,GAAG;GAAK,SAAS;GAAY,CAAC;;AAG9C,QAAO;;AAGT,SAAS,YAAY,QAAiD;AACpE,SAAQ,OAAO,MAAf;EACE,KAAK;EACL,KAAK,aACH,QAAO,OAAO;EAChB,KAAK;EACL,KAAK,aACH,QAAO,KAAK,UAAU,OAAO,MAAM;EACrC,KAAK,UACH,QAAO,OAAO,MACX,KAAK,MAAwC,EAAE,SAAS,SAAU,EAAE,QAAQ,KAAM,GAAI,CACtF,OAAO,QAAQ,CACf,KAAK,KAAK;EACf,QACE,QAAO;;;;;;;;;;;;;ACjFb,SAAgB,iBAAiB,SAAsD;CACrF,IAAI,cAAc;CAElB,MAAM,UAAU,IAAI,QAAQ;EAC1B,eAAe,QAAQ;EACvB,WAAW,QAAQ,aAAa,SAAoB,QAAQ,YAAY,KAAK,IAAI,IAAI;EACrF,eAAe,QAAQ,UAAU,iBAAiB;EAClD,kBAAkB,QAAQ,UAAU,QAChC,yBAAyB,QAAQ,SAAS,MAAM,GAChD;EACJ,YAAY,QAAQ,cACf,SAAS,UAAU,QAAQ,aAAa,QAAQ,SAAS,MAAM,GAChE;EACJ,kBAAkB,QAAQ,oBAAoB,QAAQ;EACvD,CAAC;AAEF,QAAO;EACL,sBAAsB;EAEtB,iBAAiB,OAAO,EAAE,aAAa;GACrC,IAAI,EAAE,WAAW;AAGjB,OAAI,QAAQ,SACV,UAAS,MAAM,oBAAoB,QAAQ,QAAQ,SAAS;AAI9D,OAAI,QAAQ,QACV,UAAS,cAAc,QAAQ,QAAQ,QAAQ;GAMjD,MAAM,QAAQ,UAAU,OAAO;GAC/B,MAAM,iBAAiB,MAAM,QAAQ,MAAM,EAAE,SAAS,SAAS;GAC/D,IAAI,eAAe,MAAM,QAAQ,MAAM,EAAE,SAAS,SAAS;AAG3D,kBAAe,MAAM,QAAQ,SAAS,aAAa;AAMnD,YAAS,QAHU,CAAC,GAAG,gBAAgB,GAAG,aAAa,CAG3B;AAG5B,OAAI,QAAQ,aACV,UAAS,MAAM,mBAAmB,QAAQ,QAAQ,aAAa;AAIjE,OAAI,QAAQ,iBACV,UAAS,MAAM,QAAQ,iBAAiB,OAAO;AAGjD,UAAO;IAAE,GAAG;IAAQ;IAAQ;;EAG9B,cAAc,OAAO,EAAE,iBAAiB;GACtC,MAAM,SAAS,MAAM,YAAY;AAEjC,OAAI,OAAO,OAAO,aAAa,SAAS,KACtC,SAAQ,eAAe,OAAO,MAAM,YAAY,MAAM;YAC7C,CAAC,eAAe,CAAC,QAAQ,WAAW;AAC7C,kBAAc;AACd,YAAQ,KACN,wLAGD;;AAGH,UAAO;;EAGT,YAAY,OAAO,EAAE,eAAe;GAClC,MAAM,EAAE,QAAQ,GAAG,SAAS,MAAM,UAAU;GAE5C,MAAM,YAAY,IAAI,gBAAsE,EAC1F,UAAU,OAAO,YAAY;AAC3B,QAAI,MAAM,SAAS,UACjB;SAAI,MAAM,OAAO,aAAa,SAAS,KACrC,SAAQ,eAAe,MAAM,MAAM,YAAY,MAAM;cAC5C,CAAC,eAAe,CAAC,QAAQ,WAAW;AAC7C,oBAAc;AACd,cAAQ,KACN,uLAGD;;;AAGL,eAAW,QAAQ,MAAM;MAE5B,CAAC;AAEF,UAAO;IAAE,GAAG;IAAM,QAAQ,OAAO,YAAY,UAAU;IAAE;;EAE5D;;;;;;;;;;;AAYH,SAAS,cACP,QACA,QACuB;AAUvB,QADe,cAAc;EAAE,UARd,OAAO,KACrB,SACE;GACC,MAAM,IAAI;GACV,SAAS,IAAI;GACb,iBAAiB,IAAI;GACtB,EACJ;EACwC,GAAG;EAAQ,CAAC,CACvC,KACX,SACE;EACC,MAAM,IAAI;EACV,SAAS,IAAI;EACb,iBAAiB,IAAI;EACtB,EACJ;;;;;;;;;AAUH,eAAe,mBACb,QACA,QACgC;CAChC,MAAM,QAAQ,MAAM,OAAO,UAAU;CACrC,MAAM,MAAM,aAAa,YAAY,OAAO,gBAAgB;AAG5D,MAFkB,OAAO,aAAa,iBAEpB,SAChB,QAAO,CAAC,GAAG,QAAQ;EAAE,MAAM;EAAU,SAAS,wBAAwB;EAAO,CAAC;CAIhF,MAAM,SAAS,CAAC,GAAG,OAAO;CAC1B,MAAM,aAAa,OAAO,IAAI;AAE9B,MAAK,IAAI,IAAI,OAAO,SAAS,GAAG,KAAK,GAAG,KAAK;EAC3C,MAAM,MAAM,OAAO;AACnB,MAAI,IAAI,SAAS,QAAQ;AACvB,UAAO,KAAK;IACV,GAAG;IACH,SAAS,CAAC,GAAG,IAAI,SAAS;KAAE,MAAM;KAAQ,MAAM;KAAY,CAAC;IAC9D;AACD,UAAO;;;AAKX,QAAO,KAAK;EACV,MAAM;EACN,SAAS,CAAC;GAAE,MAAM;GAAQ,MAAM,WAAW,MAAM;GAAE,CAAC;EACrD,CAAC;AACF,QAAO;;;;;;AAOT,SAAS,eAAe,MAA4B;AAClD,KAAI,SAAS,YAAY,SAAS,UAAU,SAAS,YAAa,QAAO;AACzE,QAAO;;;;;;;;;AAUT,SAAS,yBACP,OAC0C;AAC1C,QAAO,OAAO,aAAyC;EAuBrD,MAAM,EAAE,SAAS,MAAM,aAAa;GAClC;GACA,UAxBgB,SAAS,KAAK,MAA+C;AAC7E,QAAI,EAAE,SAAS,OACb,QAAO;KACL,MAAM;KACN,SAAS,eAAe,EAAE,eAAe,KAAK,EAAE,aAAa,KAAK,GAAG,IAAI,EAAE,QAAQ;KACpF;AAEH,QAAI,EAAE,SAAS,eAAe,EAAE,YAAY,QAAQ;KAClD,MAAM,gBAAgB,EAAE,WACrB,KAAK,OAAO,iBAAiB,GAAG,SAAS,KAAK,GAAG,GAAG,SAAS,UAAU,IAAI,CAC3E,KAAK,KAAK;AACb,YAAO;MACL,MAAM;MACN,SAAS,EAAE,UAAU,GAAG,EAAE,QAAQ,IAAI,kBAAkB;MACzD;;AAEH,WAAO;KACL,MAAM,eAAe,EAAE,KAAK;KAC5B,SAAS,EAAE;KACZ;KACD;GAKA,iBAAiB;GAClB,CAAC;AAEF,SAAO,QAAQ;;;;;;;;;;;;;;;;;;;;;;;;;;ACpNnB,SAAgB,gBACd,OACA,SACiB;AAEjB,QAAO,kBAAkB;EAAE;EAAO,YADf,iBAAiB,QAAQ;EACE,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@context-chef/ai-sdk-middleware",
-  "version": "1.0.7",
+  "version": "1.1.0",
   "type": "module",
   "main": "./dist/index.cjs",
   "module": "./dist/index.mjs",