@compresr/sdk 1.3.1 → 1.5.0

package/dist/esm/integrations/langchain/middleware.js

@@ -0,0 +1,268 @@
+/**
+ * Agent middleware for LangChain.js 1.0+ `createAgent`.
+ *
+ * - `compresrToolMiddleware` — compresses each tool output as it returns,
+ *   before it enters agent state.
+ * - `compresrSummarizationMiddleware` — when state grows past a token
+ *   threshold, compresses the older block into a single summary message
+ *   and replaces it in state. Mirrors LangChain's `SummarizationMiddleware`
+ *   but uses Compresr instead of an LLM call. KV-cache-friendly:
+ *   the summary is generated once at trigger time and persists in state.
+ *
+ * Unified Compresr query API across both:
+ *
+ *     query: string                                 // static query
+ *     queryExtractor: (...) => string               // custom extractor
+ *     queryArg: string                              // name of the tool arg
+ */
+import { HumanMessage, RemoveMessage, ToolMessage, } from '@langchain/core/messages';
+import { buildClient, compressSafe, DEFAULT_MIN_TOKENS, DEFAULT_MODEL, DEFAULT_POLICY, DEFAULT_RATIO, estimateTokens, makeFilter, resolveQuery, } from '../_shared/index.js';
+const REMOVE_ALL_MESSAGES = '__remove_all__';
+function isToolMessage(value) {
+    return value instanceof ToolMessage;
+}
+function buildToolMessage(content, original) {
+    const orig = original;
+    return new ToolMessage({
+        content,
+        tool_call_id: orig.tool_call_id ?? '',
+        ...(orig.name !== undefined ? { name: orig.name } : {}),
+    });
+}
+function isEligible(result, toolCall, eligible) {
+    if (!isToolMessage(result))
+        return false;
+    const content = result.content;
+    if (typeof content !== 'string')
+        return false;
+    const name = toolCall.name ?? result.name;
+    return eligible(name);
+}
+export function compresrToolMiddleware(options) {
+    const { apiKey, client, baseUrl, compressionModel = DEFAULT_MODEL, targetCompressionRatio = DEFAULT_RATIO, minTokens = DEFAULT_MIN_TOKENS, coarse, allowTools, ignoreTools, onError = DEFAULT_POLICY, query, queryExtractor, queryArg, } = options;
+    const eligible = makeFilter({ allow: allowTools, ignore: ignoreTools });
+    const compresr = client ?? buildClient({ apiKey, baseUrl, caller: 'compresrToolMiddleware' });
+    function resolveTheQuery(toolCall, messages) {
+        return resolveQuery({
+            staticQuery: query,
+            extractor: queryExtractor
+                ? (arg) => queryExtractor(...arg)
+                : undefined,
+            extractorArg: queryExtractor ? [toolCall, messages] : undefined,
+            args: toolCall.args ?? null,
+            argsKey: queryArg,
+            messages,
+            toolCallId: toolCall.id,
+        });
+    }
+    return {
+        name: 'compresr-tool',
+        async wrapToolCall(request, handler) {
+            const result = await handler(request);
+            const toolCall = request.toolCall ?? {};
+            if (!isEligible(result, toolCall, eligible)) {
+                return result;
+            }
+            const messages = request.messages ?? [];
+            const resolvedQuery = resolveTheQuery(toolCall, messages);
+            const compressed = await compressSafe(compresr, {
+                context: result.content,
+                ...(resolvedQuery !== undefined ? { query: resolvedQuery } : {}),
+                compressionModel,
+                targetCompressionRatio,
+                ...(coarse !== undefined ? { coarse } : {}),
+                minTokens,
+                onError,
+                contextLabel: `tool:${result.name ?? '?'}`,
+            });
+            return compressed !== result.content
+                ? buildToolMessage(compressed, result)
+                : result;
+        },
+    };
+}
+const SUMMARY_PREFIX = '[Earlier conversation summary]\n\n';
+function msgToText(m) {
+    const ctor = m.constructor?.name ?? 'Msg';
+    const role = ctor.replace('Message', '').toLowerCase() || 'msg';
+    const name = m.name;
+    const label = name ? `${role}:${name}` : role;
+    const c = m.content;
+    const text = typeof c === 'string' ? c : JSON.stringify(c);
+    return `${label}: ${text}`;
+}
+function isSummaryMessage(m) {
+    if (!(m instanceof HumanMessage))
+        return false;
+    const c = m.content;
+    return typeof c === 'string' && c.startsWith(SUMMARY_PREFIX);
+}
+export function compresrSummarizationMiddleware(options) {
+    const { apiKey, client, baseUrl, compressionModel = DEFAULT_MODEL, targetCompressionRatio = DEFAULT_RATIO, coarse, onError = DEFAULT_POLICY, maxTokensBeforeSummary = 4_000, messagesToKeep = 20, trigger, keep: keepAlias, tokenCounter, query, queryExtractor, } = options;
+    const compresr = client ?? buildClient({ apiKey, baseUrl, caller: 'compresrSummarizationMiddleware' });
+    const maxTokens = Math.max(1, trigger ?? maxTokensBeforeSummary);
+    const keep = Math.max(1, keepAlias ?? messagesToKeep);
+    const countTokens = tokenCounter ?? estimateTokens;
+    function resolveTheQuery(messages) {
+        return resolveQuery({
+            staticQuery: query,
+            extractor: queryExtractor
+                ? (arg) => queryExtractor(arg)
+                : undefined,
+            extractorArg: queryExtractor ? messages : undefined,
+            messages,
+        });
+    }
+    return {
+        name: 'compresr-summarization',
+        async beforeModel(state) {
+            const messages = [...(state.messages ?? [])];
+            if (messages.length <= keep)
+                return undefined;
+            let total = 0;
+            for (const m of messages) {
+                const c = m.content;
+                if (typeof c === 'string')
+                    total += countTokens(c);
+            }
+            if (total < maxTokens)
+                return undefined;
+            const toSummarize = messages.slice(0, messages.length - keep);
+            const recent = messages.slice(messages.length - keep);
+            if (toSummarize.length === 0)
+                return undefined;
+            let joined;
+            if (toSummarize.length > 0 && isSummaryMessage(toSummarize[0])) {
+                const prev = toSummarize[0].content.slice(SUMMARY_PREFIX.length);
+                const extra = toSummarize.slice(1).map(msgToText).join('\n');
+                joined = prev + '\n\n' + extra;
+            }
+            else {
+                joined = toSummarize.map(msgToText).join('\n');
+            }
+            const compressed = await compressSafe(compresr, {
+                context: joined,
+                ...(resolveTheQuery(messages) !== undefined
+                    ? { query: resolveTheQuery(messages) }
+                    : {}),
+                compressionModel,
+                targetCompressionRatio,
+                ...(coarse !== undefined ? { coarse } : {}),
+                minTokens: 1,
+                onError,
+                contextLabel: 'summary',
+            });
+            if (compressed === joined)
+                return undefined;
+            return {
+                messages: [
+                    new RemoveMessage({ id: REMOVE_ALL_MESSAGES }),
+                    new HumanMessage({ content: `${SUMMARY_PREFIX}${compressed}` }),
+                    ...recent,
+                ],
+            };
+        },
+    };
+}
+function rebuildWithContent(msg, content) {
+    if (msg === null || typeof msg !== 'object')
+        return msg;
+    const ctor = msg.constructor;
+    if (typeof ctor !== 'function')
+        return msg;
+    const init = { content };
+    for (const attr of ['tool_call_id', 'name', 'tool_calls', 'id']) {
+        const v = msg[attr];
+        if (v !== undefined && v !== null)
+            init[attr] = v;
+    }
+    try {
+        return new ctor(init);
+    }
+    catch {
+        // Some message classes need positional or different shape — fall back.
+        return msg;
+    }
+}
+export function compresrPromptMiddleware(options) {
+    const { apiKey, client, baseUrl, compressionModel = DEFAULT_MODEL, coarse, onError = DEFAULT_POLICY, maxTokens, minTokens = DEFAULT_MIN_TOKENS, tokenCounter, query, queryExtractor, } = options;
+    const compresr = client ?? buildClient({ apiKey, baseUrl, caller: 'compresrPromptMiddleware' });
+    const budget = Math.max(1, maxTokens);
+    const floor = Math.max(1, minTokens);
+    const countTokens = tokenCounter ?? estimateTokens;
+    function totalTokens(messages) {
+        let t = 0;
+        for (const m of messages) {
+            const c = m.content;
+            if (typeof c === 'string')
+                t += countTokens(c);
+        }
+        return t;
+    }
+    function resolveTheQuery(messages) {
+        return resolveQuery({
+            staticQuery: query,
+            extractor: queryExtractor ? (arg) => queryExtractor(arg) : undefined,
+            extractorArg: queryExtractor ? messages : undefined,
+            messages,
+        });
+    }
+    async function shrink(messages) {
+        if (totalTokens(messages) <= budget)
+            return messages;
+        const candidates = [];
+        for (let i = 0; i < messages.length; i++) {
+            const m = messages[i];
+            const c = m.content;
+            if (typeof c === 'string') {
+                const t = countTokens(c);
+                if (t >= floor)
+                    candidates.push({ idx: i, msg: m, tokens: t });
+            }
+        }
+        candidates.sort((a, b) => b.tokens - a.tokens);
+        const resolvedQuery = resolveTheQuery(messages);
+        const next = [...messages];
+        for (const { idx, msg, tokens: current } of candidates) {
+            const totalNow = totalTokens(next);
+            if (totalNow <= budget)
+                break;
+            const overshoot = totalNow - budget;
+            const target = Math.max(floor, current - Math.max(1, overshoot));
+            const ratio = Math.max(current / Math.max(1, target), 1.0);
+            const original = msg.content;
+            const compressed = await compressSafe(compresr, {
+                context: original,
+                ...(resolvedQuery !== undefined ? { query: resolvedQuery } : {}),
+                compressionModel,
+                targetCompressionRatio: ratio,
+                ...(coarse !== undefined ? { coarse } : {}),
+                minTokens: floor,
+                onError,
+                contextLabel: 'prompt_budget',
+            });
+            if (compressed !== original) {
+                next[idx] = rebuildWithContent(msg, compressed);
+            }
+        }
+        return next;
+    }
+    return {
+        name: 'compresr-prompt',
+        async wrapModelCall(request, handler) {
+            let next = request;
+            try {
+                const shrunk = await shrink(request.messages);
+                if (shrunk !== request.messages) {
+                    // Parity with Python PY-AM4 — never mutate the caller's request.
+                    next = { ...request, messages: shrunk };
+                }
+            }
+            catch {
+                // Fail open — send the original prompt rather than block the call.
+            }
+            return handler(next);
+        },
+    };
+}
+//# sourceMappingURL=middleware.js.map

package/dist/esm/integrations/langchain/middleware.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"middleware.js","sourceRoot":"","sources":["../../../../src/integrations/langchain/middleware.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;GAgBG;AACH,OAAO,EACL,YAAY,EACZ,aAAa,EACb,WAAW,GAEZ,MAAM,0BAA0B,CAAC;AAGlC,OAAO,EACL,WAAW,EACX,YAAY,EACZ,kBAAkB,EAClB,aAAa,EACb,cAAc,EACd,aAAa,EAEb,cAAc,EACd,UAAU,EACV,YAAY,GACb,MAAM,qBAAqB,CAAC;AAE7B,MAAM,mBAAmB,GAAG,gBAAgB,CAAC;AAoF7C,SAAS,aAAa,CAAC,KAAc;IACnC,OAAO,KAAK,YAAY,WAAW,CAAC;AACtC,CAAC;AAED,SAAS,gBAAgB,CAAC,OAAe,EAAE,QAAyB;IAClE,MAAM,IAAI,GAAG,QAAoD,CAAC;IAClE,OAAO,IAAI,WAAW,CAAC;QACrB,OAAO;QACP,YAAY,EAAE,IAAI,CAAC,YAAY,IAAI,EAAE;QACrC,GAAG,CAAC,IAAI,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KACxD,CAAC,CAAC;AACL,CAAC;AAED,SAAS,UAAU,CACjB,MAAe,EACf,QAAkB,EAClB,QAA2C;IAE3C,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC;QAAE,OAAO,KAAK,CAAC;IACzC,MAAM,OAAO,GAAI,MAAgC,CAAC,OAAO,CAAC;IAC1D,IAAI,OAAO,OAAO,KAAK,QAAQ;QAAE,OAAO,KAAK,CAAC;IAC9C,MAAM,IAAI,GAAG,QAAQ,CAAC,IAAI,IAAK,MAA4B,CAAC,IAAI,CAAC;IACjE,OAAO,QAAQ,CAAC,IAAI,CAAC,CAAC;AACxB,CAAC;AAED,MAAM,UAAU,sBAAsB,CACpC,OAAsC;IAEtC,MAAM,EACJ,MAAM,EACN,MAAM,EACN,OAAO,EACP,gBAAgB,GAAG,aAAa,EAChC,sBAAsB,GAAG,aAAa,EACtC,SAAS,GAAG,kBAAkB,EAC9B,MAAM,EACN,UAAU,EACV,WAAW,EACX,OAAO,GAAG,cAAc,EACxB,KAAK,EACL,cAAc,EACd,QAAQ,GACT,GAAG,OAAO,CAAC;IAEZ,MAAM,QAAQ,GAAG,UAAU,CAAC,EAAE,KAAK,EAAE,UAAU,EAAE,MAAM,EAAE,WAAW,EAAE,CAAC,CAAC;IACxE,MAAM,QAAQ,GACZ,MAAM,IAAI,WAAW,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,wBAAwB,EAAE,CAAC,CAAC;IAE/E,SAAS,eAAe,CAAC,QAAkB,EAAE,QAAmB;QAC9D,OAAO,YAAY,CAAC;YAClB,WAAW,EAAE,KAAK;YAClB,SAAS,EAAE,cAAc;gBACvB,CAAC,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,cAAc,CAAC,GAAI,GAA6B,CAAC;gBAC5D,CAAC,CAAC,SAAS;YACb,YAAY,EAAE,cAAc,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC,CAAC,CAAC,SAAS;YAC/D,IAAI,EAAE,QAAQ,CAAC,IAAI,IAAI,IAAI;YAC3B,OAAO,EAAE,QAAQ;YACjB,QAAQ;YACR,UAAU,EAAE,QAAQ,CAAC,EAAE;SACxB,CAAC,CAAC;IACL,CAAC;IAED,OAAO;QACL,IAAI,EAAE,eAAe;QACrB,KAAK,CAAC,YAAY,CAAC,OAAO,EAAE,OAAO;YACjC,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,OAAO,CAAC,CAAC;YACtC,MAAM,QAAQ,GAAG,OAAO,CAAC,QAAQ,IAAI,EAAE,CAAC;YACxC,IAAI,CAAC,UAAU,CAAC,MAAM,EAAE,QAAQ,EAAE,QAAQ,CAAC,EAAE,CAAC;gBAC5C,OAAO,MAAM,CAAC;YAChB,CAAC;YACD,MAAM,QAAQ,GAAG,OAAO,CAAC,QAAQ,IAAI,EAAE,CAAC;YACxC,MAAM,aAAa,GAAG,eAAe,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;YAC1D,MAAM,UAAU,GAAG,MAAM,YAAY,CAAC,QAAQ,EAAE;gBAC9C,OAAO,EAAG,MAA8B,CAAC,OAAO;gBAChD,GAAG,CAAC,aAAa,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,aAAa,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBAChE,gBAAgB;gBAChB,sBAAsB;gBACtB,GAAG,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBAC3C,SAAS;gBACT,OAAO;gBACP,YAAY,EAAE,QAAS,MAA4B,CAAC,IAAI,IAAI,GAAG,EAAE;aAClE,CAAC,CAAC;YACH,OAAO,UAAU,KAAM,MAA8B,CAAC,OAAO;gBAC3D,CAAC,CAAC,gBAAgB,CAAC,UAAU,EAAE,MAAM,CAAC;gBACtC,CAAC,CAAC,MAAM,CAAC;QACb,CAAC;KACF,CAAC;AACJ,CAAC;AAED,MAAM,cAAc,GAAG,oCAAoC,CAAC;AAE5D,SAAS,SAAS,CAAC,CAAU;IAC3B,MAAM,IAAI,GAAI,CAAyC,CAAC,WAAW,EAAE,IAAI,IAAI,KAAK,CAAC;IACnF,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC,WAAW,EAAE,IAAI,KAAK,CAAC;IAChE,MAAM,IAAI,GAAI,CAAuB,CAAC,IAAI,CAAC;IAC3C,MAAM,KAAK,GAAG,IAAI,CAAC,CAAC,CAAC,GAAG,IAAI,IAAI,IAAI,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;IAC9C,MAAM,CAAC,GAAI,CAA2B,CAAC,OAAO,CAAC;IAC/C,MAAM,IAAI,GAAG,OAAO,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;IAC3D,OAAO,GAAG,KAAK,KAAK,IAAI,EAAE,CAAC;AAC7B,CAAC;AAED,SAAS,gBAAgB,CAAC,CAAU;IAClC,IAAI,CAAC,CAAC,CAAC,YAAY,YAAY,CAAC;QAAE,OAAO,KAAK,CAAC;IAC/C,MAAM,CAAC,GAAI,CAA2B,CAAC,OAAO,CAAC;IAC/C,OAAO,OAAO,CAAC,KAAK,QAAQ,IAAI,CAAC,CAAC,UAAU,CAAC,cAAc,CAAC,CAAC;AAC/D,CAAC;AAED,MAAM,UAAU,+BAA+B,CAC7C,OAA+C;IAE/C,MAAM,EACJ,MAAM,EACN,MAAM,EACN,OAAO,EACP,gBAAgB,GAAG,aAAa,EAChC,sBAAsB,GAAG,aAAa,EACtC,MAAM,EACN,OAAO,GAAG,cAAc,EACxB,sBAAsB,GAAG,KAAK,EAC9B,cAAc,GAAG,EAAE,EACnB,OAAO,EACP,IAAI,EAAE,SAAS,EACf,YAAY,EACZ,KAAK,EACL,cAAc,GACf,GAAG,OAAO,CAAC;IAEZ,MAAM,QAAQ,GACZ,MAAM,IAAI,WAAW,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,iCAAiC,EAAE,CAAC,CAAC;IAExF,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,OAAO,IAAI,sBAAsB,CAAC,CAAC;IACjE,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,SAAS,IAAI,cAAc,CAAC,CAAC;IACtD,MAAM,WAAW,GAAG,YAAY,IAAI,cAAc,CAAC;IAEnD,SAAS,eAAe,CAAC,QAAmB;QAC1C,OAAO,YAAY,CAAC;YAClB,WAAW,EAAE,KAAK;YAClB,SAAS,EAAE,cAAc;gBACvB,CAAC,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,cAAc,CAAC,GAAgB,CAAC;gBAC3C,CAAC,CAAC,SAAS;YACb,YAAY,EAAE,cAAc,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS;YACnD,QAAQ;SACT,CAAC,CAAC;IACL,CAAC;IAED,OAAO;QACL,IAAI,EAAE,wBAAwB;QAC9B,KAAK,CAAC,WAAW,CAAC,KAAK;YACrB,MAAM,QAAQ,GAAG,CAAC,GAAG,CAAC,KAAK,CAAC,QAAQ,IAAI,EAAE,CAAC,CAAC,CAAC;YAC7C,IAAI,QAAQ,CAAC,MAAM,IAAI,IAAI;gBAAE,OAAO,SAAS,CAAC;YAE9C,IAAI,KAAK,GAAG,CAAC,CAAC;YACd,KAAK,MAAM,CAAC,IAAI,QAAQ,EAAE,CAAC;gBACzB,MAAM,CAAC,GAAI,CAA2B,CAAC,OAAO,CAAC;gBAC/C,IAAI,OAAO,CAAC,KAAK,QAAQ;oBAAE,KAAK,IAAI,WAAW,CAAC,CAAC,CAAC,CAAC;YACrD,CAAC;YACD,IAAI,KAAK,GAAG,SAAS;gBAAE,OAAO,SAAS,CAAC;YAExC,MAAM,WAAW,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,QAAQ,CAAC,MAAM,GAAG,IAAI,CAAC,CAAC;YAC9D,MAAM,MAAM,GAAG,QAAQ,CAAC,KAAK,CAAC,QAAQ,CAAC,MAAM,GAAG,IAAI,CAAC,CAAC;YACtD,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC;gBAAE,OAAO,SAAS,CAAC;YAE/C,IAAI,MAAc,CAAC;YACnB,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,IAAI,gBAAgB,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;gBAC/D,MAAM,IAAI,GAAK,WAAW,CAAC,CAAC,CAAiB,CAAC,OAAkB,CAAC,KAAK,CACpE,cAAc,CAAC,MAAM,CACtB,CAAC;gBACF,MAAM,KAAK,GAAG,WAAW,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBAC7D,MAAM,GAAG,IAAI,GAAG,MAAM,GAAG,KAAK,CAAC;YACjC,CAAC;iBAAM,CAAC;gBACN,MAAM,GAAG,WAAW,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACjD,CAAC;YAED,MAAM,UAAU,GAAG,MAAM,YAAY,CAAC,QAAQ,EAAE;gBAC9C,OAAO,EAAE,MAAM;gBACf,GAAG,CAAC,eAAe,CAAC,QAAQ,CAAC,KAAK,SAAS;oBACzC,CAAC,CAAC,EAAE,KAAK,EAAE,eAAe,CAAC,QAAQ,CAAW,EAAE;oBAChD,CAAC,CAAC,EAAE,CAAC;gBACP,gBAAgB;gBAChB,sBAAsB;gBACtB,GAAG,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBAC3C,SAAS,EAAE,CAAC;gBACZ,OAAO;gBACP,YAAY,EAAE,SAAS;aACxB,CAAC,CAAC;YACH,IAAI,UAAU,KAAK,MAAM;gBAAE,OAAO,SAAS,CAAC;YAE5C,OAAO;gBACL,QAAQ,EAAE;oBACR,IAAI,aAAa,CAAC,EAAE,EAAE,EAAE,mBAAmB,EAAE,CAAC;oBAC9C,IAAI,YAAY,CAAC,EAAE,OAAO,EAAE,GAAG,cAAc,GAAG,UAAU,EAAE,EAAE,CAAC;oBAC/D,GAAG,MAAM;iBACV;aACF,CAAC;QACJ,CAAC;KACF,CAAC;AACJ,CAAC;AAgBD,SAAS,kBAAkB,CAAC,GAAY,EAAE,OAAe;IACvD,IAAI,GAAG,KAAK,IAAI,IAAI,OAAO,GAAG,KAAK,QAAQ;QAAE,OAAO,GAAG,CAAC;IACxD,MAAM,IAAI,GAAI,GAAwE,CAAC,WAAW,CAAC;IACnG,IAAI,OAAO,IAAI,KAAK,UAAU;QAAE,OAAO,GAAG,CAAC;IAC3C,MAAM,IAAI,GAA4B,EAAE,OAAO,EAAE,CAAC;IAClD,KAAK,MAAM,IAAI,IAAI,CAAC,cAAc,EAAE,MAAM,EAAE,YAAY,EAAE,IAAI,CAAU,EAAE,CAAC;QACzE,MAAM,CAAC,GAAI,GAA+B,CAAC,IAAI,CAAC,CAAC;QACjD,IAAI,CAAC,KAAK,SAAS,IAAI,CAAC,KAAK,IAAI;YAAE,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IACpD,CAAC;IACD,IAAI,CAAC;QACH,OAAO,IAAI,IAAI,CAAC,IAAI,CAAC,CAAC;IACxB,CAAC;IAAC,MAAM,CAAC;QACP,uEAAuE;QACvE,OAAO,GAAG,CAAC;IACb,CAAC;AACH,CAAC;AAED,MAAM,UAAU,wBAAwB,CACtC,OAAwC;IAExC,MAAM,EACJ,MAAM,EACN,MAAM,EACN,OAAO,EACP,gBAAgB,GAAG,aAAa,EAChC,MAAM,EACN,OAAO,GAAG,cAAc,EACxB,SAAS,EACT,SAAS,GAAG,kBAAkB,EAC9B,YAAY,EACZ,KAAK,EACL,cAAc,GACf,GAAG,OAAO,CAAC;IAEZ,MAAM,QAAQ,GACZ,MAAM,IAAI,WAAW,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,0BAA0B,EAAE,CAAC,CAAC;IACjF,MAAM,MAAM,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC;IACtC,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC;IACrC,MAAM,WAAW,GAAG,YAAY,IAAI,cAAc,CAAC;IAEnD,SAAS,WAAW,CAAC,QAAmB;QACtC,IAAI,CAAC,GAAG,CAAC,CAAC;QACV,KAAK,MAAM,CAAC,IAAI,QAAQ,EAAE,CAAC;YACzB,MAAM,CAAC,GAAI,CAA2B,CAAC,OAAO,CAAC;YAC/C,IAAI,OAAO,CAAC,KAAK,QAAQ;gBAAE,CAAC,IAAI,WAAW,CAAC,CAAC,CAAC,CAAC;QACjD,CAAC;QACD,OAAO,CAAC,CAAC;IACX,CAAC;IAED,SAAS,eAAe,CAAC,QAAmB;QAC1C,OAAO,YAAY,CAAC;YAClB,WAAW,EAAE,KAAK;YAClB,SAAS,EAAE,cAAc,CAAC,CAAC,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,cAAc,CAAC,GAAgB,CAAC,CAAC,CAAC,CAAC,SAAS;YACjF,YAAY,EAAE,cAAc,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS;YACnD,QAAQ;SACT,CAAC,CAAC;IACL,CAAC;IAED,KAAK,UAAU,MAAM,CAAC,QAAmB;QACvC,IAAI,WAAW,CAAC,QAAQ,CAAC,IAAI,MAAM;YAAE,OAAO,QAAQ,CAAC;QAErD,MAAM,UAAU,GAAoD,EAAE,CAAC;QACvE,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,QAAQ,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YACzC,MAAM,CAAC,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC;YACtB,MAAM,CAAC,GAAI,CAA2B,CAAC,OAAO,CAAC;YAC/C,IAAI,OAAO,CAAC,KAAK,QAAQ,EAAE,CAAC;gBAC1B,MAAM,CAAC,GAAG,WAAW,CAAC,CAAC,CAAC,CAAC;gBACzB,IAAI,CAAC,IAAI,KAAK;oBAAE,UAAU,CAAC,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC,EAAE,GAAG,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,CAAC,CAAC;YACjE,CAAC;QACH,CAAC;QACD,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,MAAM,CAAC,CAAC;QAE/C,MAAM,aAAa,GAAG,eAAe,CAAC,QAAQ,CAAC,CAAC;QAChD,MAAM,IAAI,GAAG,CAAC,GAAG,QAAQ,CAAC,CAAC;QAC3B,KAAK,MAAM,EAAE,GAAG,EAAE,GAAG,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,UAAU,EAAE,CAAC;YACvD,MAAM,QAAQ,GAAG,WAAW,CAAC,IAAI,CAAC,CAAC;YACnC,IAAI,QAAQ,IAAI,MAAM;gBAAE,MAAM;YAC9B,MAAM,SAAS,GAAG,QAAQ,GAAG,MAAM,CAAC;YACpC,MAAM,MAAM,GAAG,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,OAAO,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC,CAAC;YACjE,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,OAAO,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,MAAM,CAAC,EAAE,GAAG,CAAC,CAAC;YAC3D,MAAM,QAAQ,GAAI,GAA2B,CAAC,OAAO,CAAC;YACtD,MAAM,UAAU,GAAG,MAAM,YAAY,CAAC,QAAQ,EAAE;gBAC9C,OAAO,EAAE,QAAQ;gBACjB,GAAG,CAAC,aAAa,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,aAAa,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBAChE,gBAAgB;gBAChB,sBAAsB,EAAE,KAAK;gBAC7B,GAAG,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBAC3C,SAAS,EAAE,KAAK;gBAChB,OAAO;gBACP,YAAY,EAAE,eAAe;aAC9B,CAAC,CAAC;YACH,IAAI,UAAU,KAAK,QAAQ,EAAE,CAAC;gBAC5B,IAAI,CAAC,GAAG,CAAC,GAAG,kBAAkB,CAAC,GAAG,EAAE,UAAU,CAAC,CAAC;YAClD,CAAC;QACH,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,OAAO;QACL,IAAI,EAAE,iBAAiB;QACvB,KAAK,CAAC,aAAa,CAAC,OAAO,EAAE,OAAO;YAClC,IAAI,IAAI,GAAG,OAAO,CAAC;YACnB,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;gBAC9C,IAAI,MAAM,KAAK,OAAO,CAAC,QAAQ,EAAE,CAAC;oBAChC,iEAAiE;oBACjE,IAAI,GAAG,EAAE,GAAG,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC;gBAC1C,CAAC;YACH,CAAC;YAAC,MAAM,CAAC;gBACP,mEAAmE;YACrE,CAAC;YACD,OAAO,OAAO,CAAC,IAAI,CAAC,CAAC;QACvB,CAAC;KACF,CAAC;AACJ,CAAC"}

package/dist/esm/integrations/langchain/retriever.d.ts

@@ -0,0 +1,54 @@
+/**
+ * `BaseDocumentCompressor` implementation for `ContextualCompressionRetriever`.
+ *
+ * Mirrors Python `CompresrExtractor`. One Compresr batch call
+ * replaces N LLM extraction calls.
+ */
+import { Document } from '@langchain/core/documents';
+import type { CompressionClient } from '../../clients/compression.js';
+import { type ErrorPolicy } from '../_shared/index.js';
+type DocumentType = Document;
+export interface CompresrExtractorOptions {
+    apiKey?: string;
+    client?: CompressionClient;
+    baseUrl?: string;
+    compressionModel?: string;
+    targetCompressionRatio?: number;
+    minTokens?: number;
+    coarse?: boolean;
+    onError?: ErrorPolicy;
+    /** Drop documents whose compressed content is empty. */
+    dropBelowMin?: boolean;
+}
+/**
+ * Compress retrieved `Document`s with query-aware compression.
+ *
+ * @example
+ * ```ts
+ * import { ContextualCompressionRetriever } from '@langchain/classic/retrievers/contextual_compression';
+ * import { CompresrExtractor } from '@compresr/sdk/integrations/langchain';
+ *
+ * const compressor = new CompresrExtractor({
+ *   apiKey: process.env.COMPRESR_API_KEY!,
+ * });
+ * const retriever = new ContextualCompressionRetriever({
+ *   baseCompressor: compressor,
+ *   baseRetriever: vectorStore.asRetriever({ k: 8 }),
+ * });
+ * ```
+ */
+export declare class CompresrExtractor {
+    private readonly compresr;
+    private readonly compressionModel;
+    private readonly targetCompressionRatio;
+    private readonly minTokens;
+    private readonly coarse;
+    private readonly onError;
+    private readonly dropBelowMin;
+    constructor(options: CompresrExtractorOptions);
+    private partition;
+    private emit;
+    compressDocuments(documents: readonly DocumentType[], query: string): Promise<DocumentType[]>;
+}
+export {};
+//# sourceMappingURL=retriever.d.ts.map

package/dist/esm/integrations/langchain/retriever.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"retriever.d.ts","sourceRoot":"","sources":["../../../../src/integrations/langchain/retriever.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AACH,OAAO,EAAE,QAAQ,EAAE,MAAM,2BAA2B,CAAC;AAErD,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,8BAA8B,CAAC;AAEtE,OAAO,EAOL,KAAK,WAAW,EAEjB,MAAM,qBAAqB,CAAC;AAE7B,KAAK,YAAY,GAAG,QAAQ,CAAC;AAE7B,MAAM,WAAW,wBAAwB;IACvC,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,MAAM,CAAC,EAAE,iBAAiB,CAAC;IAC3B,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,sBAAsB,CAAC,EAAE,MAAM,CAAC;IAChC,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB,OAAO,CAAC,EAAE,WAAW,CAAC;IACtB,wDAAwD;IACxD,YAAY,CAAC,EAAE,OAAO,CAAC;CACxB;AAED;;;;;;;;;;;;;;;;GAgBG;AACH,qBAAa,iBAAiB;IAC5B,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAoB;IAC7C,OAAO,CAAC,QAAQ,CAAC,gBAAgB,CAAS;IAC1C,OAAO,CAAC,QAAQ,CAAC,sBAAsB,CAAS;IAChD,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAS;IACnC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAsB;IAC7C,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAc;IACtC,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAU;gBAE3B,OAAO,EAAE,wBAAwB;IAuB7C,OAAO,CAAC,SAAS;IAejB,OAAO,CAAC,IAAI;IAQN,iBAAiB,CACrB,SAAS,EAAE,SAAS,YAAY,EAAE,EAClC,KAAK,EAAE,MAAM,GACZ,OAAO,CAAC,YAAY,EAAE,CAAC;CAgD3B"}

package/dist/esm/integrations/langchain/retriever.js

@@ -0,0 +1,114 @@
+/**
+ * `BaseDocumentCompressor` implementation for `ContextualCompressionRetriever`.
+ *
+ * Mirrors Python `CompresrExtractor`. One Compresr batch call
+ * replaces N LLM extraction calls.
+ */
+import { Document } from '@langchain/core/documents';
+import { getLogger } from '../../logger.js';
+import { BATCH_LIMIT, buildClient, DEFAULT_MIN_TOKENS, DEFAULT_MODEL, DEFAULT_POLICY, DEFAULT_RATIO, estimateTokens, } from '../_shared/index.js';
+/**
+ * Compress retrieved `Document`s with query-aware compression.
+ *
+ * @example
+ * ```ts
+ * import { ContextualCompressionRetriever } from '@langchain/classic/retrievers/contextual_compression';
+ * import { CompresrExtractor } from '@compresr/sdk/integrations/langchain';
+ *
+ * const compressor = new CompresrExtractor({
+ *   apiKey: process.env.COMPRESR_API_KEY!,
+ * });
+ * const retriever = new ContextualCompressionRetriever({
+ *   baseCompressor: compressor,
+ *   baseRetriever: vectorStore.asRetriever({ k: 8 }),
+ * });
+ * ```
+ */
+export class CompresrExtractor {
+    compresr;
+    compressionModel;
+    targetCompressionRatio;
+    minTokens;
+    coarse;
+    onError;
+    dropBelowMin;
+    constructor(options) {
+        const { apiKey, client, baseUrl, compressionModel = DEFAULT_MODEL, targetCompressionRatio = DEFAULT_RATIO, minTokens = DEFAULT_MIN_TOKENS, coarse, onError = DEFAULT_POLICY, dropBelowMin = false, } = options;
+        this.compresr =
+            client ?? buildClient({ apiKey, baseUrl, caller: 'CompresrExtractor' });
+        this.compressionModel = compressionModel;
+        this.targetCompressionRatio = targetCompressionRatio;
+        this.minTokens = minTokens;
+        this.coarse = coarse;
+        this.onError = onError;
+        this.dropBelowMin = dropBelowMin;
+    }
+    partition(documents) {
+        const indices = [];
+        const eligible = [];
+        documents.forEach((doc, i) => {
+            const pc = doc.pageContent;
+            if (typeof pc === 'string' && estimateTokens(pc) >= this.minTokens) {
+                indices.push(i);
+                eligible.push(doc);
+            }
+        });
+        return { indices, eligible };
+    }
+    emit(doc, content) {
+        const md = doc.metadata ?? {};
+        return new Document({
+            pageContent: content,
+            metadata: { ...md, compresr: true },
+        });
+    }
+    async compressDocuments(documents, query) {
+        if (documents.length === 0)
+            return [];
+        const { indices, eligible } = this.partition(documents);
+        if (eligible.length === 0)
+            return [...documents];
+        const out = [...documents];
+        for (let start = 0; start < eligible.length; start += BATCH_LIMIT) {
+            const chunk = eligible.slice(start, start + BATCH_LIMIT);
+            let results = [];
+            try {
+                const resp = await this.compresr.compressBatch({
+                    contexts: chunk.map((d) => d.pageContent),
+                    queries: query,
+                    compressionModelName: this.compressionModel,
+                    targetCompressionRatio: this.targetCompressionRatio,
+                    ...(this.coarse !== undefined ? { coarse: this.coarse } : {}),
+                });
+                results = (resp.data?.results ?? []);
+            }
+            catch (exc) {
+                if (this.onError === 'raise')
+                    throw exc;
+                getLogger().warn(`batch compress failed (${stringifyError(exc)}); passthrough ${chunk.length} docs.`);
+                continue;
+            }
+            results.slice(0, chunk.length).forEach((item, offset) => {
+                const globalIdx = indices[start + offset];
+                const newText = item?.compressed_context;
+                if (typeof newText === 'string' && newText && globalIdx !== undefined) {
+                    const doc = documents[globalIdx];
+                    if (doc !== undefined) {
+                        out[globalIdx] = this.emit(doc, newText);
+                    }
+                }
+            });
+        }
+        if (this.dropBelowMin) {
+            return out.filter((d) => {
+                const pc = d.pageContent;
+                return typeof pc === 'string' && pc.trim().length > 0;
+            });
+        }
+        return out;
+    }
+}
+function stringifyError(exc) {
+    return exc instanceof Error ? exc.message : String(exc);
+}
+//# sourceMappingURL=retriever.js.map

package/dist/esm/integrations/langchain/retriever.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"retriever.js","sourceRoot":"","sources":["../../../../src/integrations/langchain/retriever.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AACH,OAAO,EAAE,QAAQ,EAAE,MAAM,2BAA2B,CAAC;AAGrD,OAAO,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAC5C,OAAO,EACL,WAAW,EACX,WAAW,EACX,kBAAkB,EAClB,aAAa,EACb,cAAc,EACd,aAAa,EAEb,cAAc,GACf,MAAM,qBAAqB,CAAC;AAiB7B;;;;;;;;;;;;;;;;GAgBG;AACH,MAAM,OAAO,iBAAiB;IACX,QAAQ,CAAoB;IAC5B,gBAAgB,CAAS;IACzB,sBAAsB,CAAS;IAC/B,SAAS,CAAS;IAClB,MAAM,CAAsB;IAC5B,OAAO,CAAc;IACrB,YAAY,CAAU;IAEvC,YAAY,OAAiC;QAC3C,MAAM,EACJ,MAAM,EACN,MAAM,EACN,OAAO,EACP,gBAAgB,GAAG,aAAa,EAChC,sBAAsB,GAAG,aAAa,EACtC,SAAS,GAAG,kBAAkB,EAC9B,MAAM,EACN,OAAO,GAAG,cAAc,EACxB,YAAY,GAAG,KAAK,GACrB,GAAG,OAAO,CAAC;QAEZ,IAAI,CAAC,QAAQ;YACX,MAAM,IAAI,WAAW,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,mBAAmB,EAAE,CAAC,CAAC;QAC1E,IAAI,CAAC,gBAAgB,GAAG,gBAAgB,CAAC;QACzC,IAAI,CAAC,sBAAsB,GAAG,sBAAsB,CAAC;QACrD,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,YAAY,GAAG,YAAY,CAAC;IACnC,CAAC;IAEO,SAAS,CACf,SAAkC;QAElC,MAAM,OAAO,GAAa,EAAE,CAAC;QAC7B,MAAM,QAAQ,GAAmB,EAAE,CAAC;QACpC,SAAS,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE;YAC3B,MAAM,EAAE,GAAI,GAAiC,CAAC,WAAW,CAAC;YAC1D,IAAI,OAAO,EAAE,KAAK,QAAQ,IAAI,cAAc,CAAC,EAAE,CAAC,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;gBACnE,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;gBAChB,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACrB,CAAC;QACH,CAAC,CAAC,CAAC;QACH,OAAO,EAAE,OAAO,EAAE,QAAQ,EAAE,CAAC;IAC/B,CAAC;IAEO,IAAI,CAAC,GAAiB,EAAE,OAAe;QAC7C,MAAM,EAAE,GAAI,GAA8C,CAAC,QAAQ,IAAI,EAAE,CAAC;QAC1E,OAAO,IAAI,QAAQ,CAAC;YAClB,WAAW,EAAE,OAAO;YACpB,QAAQ,EAAE,EAAE,GAAG,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE;SACpC,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,iBAAiB,CACrB,SAAkC,EAClC,KAAa;QAEb,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,EAAE,CAAC;QAEtC,MAAM,EAAE,OAAO,EAAE,QAAQ,EAAE,GAAG,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;QACxD,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,CAAC,GAAG,SAAS,CAAC,CAAC;QAEjD,MAAM,GAAG,GAAmB,CAAC,GAAG,SAAS,CAAC,CAAC;QAE3C,KAAK,IAAI,KAAK,GAAG,CAAC,EAAE,KAAK,GAAG,QAAQ,CAAC,MAAM,EAAE,KAAK,IAAI,WAAW,EAAE,CAAC;YAClE,MAAM,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC,KAAK,EAAE,KAAK,GAAG,WAAW,CAAC,CAAC;YACzD,IAAI,OAAO,GAAqC,EAAE,CAAC;YACnD,IAAI,CAAC;gBACH,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,aAAa,CAAC;oBAC7C,QAAQ,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAE,CAA6B,CAAC,WAAW,CAAC;oBACtE,OAAO,EAAE,KAAK;oBACd,oBAAoB,EAAE,IAAI,CAAC,gBAAgB;oBAC3C,sBAAsB,EAAE,IAAI,CAAC,sBAAsB;oBACnD,GAAG,CAAC,IAAI,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;iBAC9D,CAAC,CAAC;gBACH,OAAO,GAAG,CAAC,IAAI,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE,CAAqC,CAAC;YAC3E,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,IAAI,IAAI,CAAC,OAAO,KAAK,OAAO;oBAAE,MAAM,GAAG,CAAC;gBACxC,SAAS,EAAE,CAAC,IAAI,CACd,0BAA0B,cAAc,CAAC,GAAG,CAAC,kBAAkB,KAAK,CAAC,MAAM,QAAQ,CACpF,CAAC;gBACF,SAAS;YACX,CAAC;YAED,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,MAAM,EAAE,EAAE;gBACtD,MAAM,SAAS,GAAG,OAAO,CAAC,KAAK,GAAG,MAAM,CAAC,CAAC;gBAC1C,MAAM,OAAO,GAAG,IAAI,EAAE,kBAAkB,CAAC;gBACzC,IAAI,OAAO,OAAO,KAAK,QAAQ,IAAI,OAAO,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;oBACtE,MAAM,GAAG,GAAG,SAAS,CAAC,SAAS,CAAC,CAAC;oBACjC,IAAI,GAAG,KAAK,SAAS,EAAE,CAAC;wBACtB,GAAG,CAAC,SAAS,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;oBAC3C,CAAC;gBACH,CAAC;YACH,CAAC,CAAC,CAAC;QACL,CAAC;QAED,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YACtB,OAAO,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE;gBACtB,MAAM,EAAE,GAAI,CAA+B,CAAC,WAAW,CAAC;gBACxD,OAAO,OAAO,EAAE,KAAK,QAAQ,IAAI,EAAE,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC,CAAC;YACxD,CAAC,CAAC,CAAC;QACL,CAAC;QACD,OAAO,GAAG,CAAC;IACb,CAAC;CACF;AAED,SAAS,cAAc,CAAC,GAAY;IAClC,OAAO,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;AAC1D,CAAC"}

package/dist/esm/integrations/langchain/wrappers.d.ts

@@ -0,0 +1,39 @@
+/**
+ * Tool-output wrappers for LangChain.js.
+ *
+ * Mirrors Python `wrap_tool_with_compression`. Works without `createAgent`
+ * middleware — wrap any tool and call it directly in LCEL chains or
+ * LangGraph custom graphs.
+ *
+ * For `latte_v1` the unified Compresr query API applies (see middleware).
+ */
+import { type StructuredTool } from '@langchain/core/tools';
+import type { CompressionClient } from '../../clients/compression.js';
+import { type ErrorPolicy } from '../_shared/index.js';
+type StructuredToolType = StructuredTool;
+export interface WrapToolOptions {
+    apiKey?: string;
+    client?: CompressionClient;
+    baseUrl?: string;
+    compressionModel?: string;
+    targetCompressionRatio?: number;
+    minTokens?: number;
+    coarse?: boolean;
+    query?: string;
+    queryExtractor?: (args: Record<string, unknown>) => string | undefined;
+    queryArg?: string;
+    onError?: ErrorPolicy;
+}
+/**
+ * Wrap a structured tool so its string output is compressed transparently.
+ *
+ * The wrapped tool preserves the original `name`, `description`, and schema.
+ * Accepts any tool that exposes `name` + (`func` | `_call` | `invoke`) — i.e.
+ * what the `tool()` factory and `StructuredTool` subclasses produce.
+ */
+export declare function wrapToolWithCompression(tool: StructuredToolType, options?: WrapToolOptions): StructuredToolType;
+export type ToolDecorator = (tool: StructuredToolType) => StructuredToolType;
+/** Decorator-style helper — for users who prefer function composition. */
+export declare function compressToolOutput(options?: WrapToolOptions): ToolDecorator;
+export {};
+//# sourceMappingURL=wrappers.d.ts.map

package/dist/esm/integrations/langchain/wrappers.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"wrappers.d.ts","sourceRoot":"","sources":["../../../../src/integrations/langchain/wrappers.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AACH,OAAO,EAAuB,KAAK,cAAc,EAAE,MAAM,uBAAuB,CAAC;AAEjF,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,8BAA8B,CAAC;AACtE,OAAO,EAOL,KAAK,WAAW,EAEjB,MAAM,qBAAqB,CAAC;AAE7B,KAAK,kBAAkB,GAAG,cAAc,CAAC;AAEzC,MAAM,WAAW,eAAe;IAC9B,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,MAAM,CAAC,EAAE,iBAAiB,CAAC;IAC3B,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,sBAAsB,CAAC,EAAE,MAAM,CAAC;IAChC,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,cAAc,CAAC,EAAE,CAAC,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,KAAK,MAAM,GAAG,SAAS,CAAC;IACvE,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,OAAO,CAAC,EAAE,WAAW,CAAC;CACvB;AAsBD;;;;;;GAMG;AACH,wBAAgB,uBAAuB,CACrC,IAAI,EAAE,kBAAkB,EACxB,OAAO,GAAE,eAAoB,GAC5B,kBAAkB,CA+DpB;AAYD,MAAM,MAAM,aAAa,GAAG,CAAC,IAAI,EAAE,kBAAkB,KAAK,kBAAkB,CAAC;AAE7E,0EAA0E;AAC1E,wBAAgB,kBAAkB,CAAC,OAAO,GAAE,eAAoB,GAAG,aAAa,CAE/E"}

package/dist/esm/integrations/langchain/wrappers.js

@@ -0,0 +1,84 @@
+/**
+ * Tool-output wrappers for LangChain.js.
+ *
+ * Mirrors Python `wrap_tool_with_compression`. Works without `createAgent`
+ * middleware — wrap any tool and call it directly in LCEL chains or
+ * LangGraph custom graphs.
+ *
+ * For `latte_v1` the unified Compresr query API applies (see middleware).
+ */
+import { tool as toolFactory } from '@langchain/core/tools';
+import { buildClient, compressSafe, DEFAULT_MIN_TOKENS, DEFAULT_MODEL, DEFAULT_POLICY, DEFAULT_RATIO, resolveQuery, } from '../_shared/index.js';
+function isWrappableTool(t) {
+    if (typeof t !== 'object' || t === null)
+        return false;
+    const x = t;
+    return (typeof x.name === 'string' &&
+        (typeof x.func === 'function' ||
+            typeof x._call === 'function' ||
+            typeof x.invoke === 'function'));
+}
+/**
+ * Wrap a structured tool so its string output is compressed transparently.
+ *
+ * The wrapped tool preserves the original `name`, `description`, and schema.
+ * Accepts any tool that exposes `name` + (`func` | `_call` | `invoke`) — i.e.
+ * what the `tool()` factory and `StructuredTool` subclasses produce.
+ */
+export function wrapToolWithCompression(tool, options = {}) {
+    if (!isWrappableTool(tool)) {
+        throw new TypeError('wrapToolWithCompression supports structured tools (with `name` + ' +
+            '`func`/`_call`/`invoke`). Use the `tool()` factory from @langchain/core.');
+    }
+    const { apiKey, client, baseUrl, compressionModel = DEFAULT_MODEL, targetCompressionRatio = DEFAULT_RATIO, minTokens = DEFAULT_MIN_TOKENS, coarse, query, queryExtractor, queryArg, onError = DEFAULT_POLICY, } = options;
+    const compresr = client ?? buildClient({ apiKey, baseUrl, caller: 'wrapToolWithCompression' });
+    function resolveTheQuery(args) {
+        return resolveQuery({
+            staticQuery: query,
+            extractor: queryExtractor
+                ? (arg) => queryExtractor(arg)
+                : undefined,
+            extractorArg: queryExtractor ? args : undefined,
+            args,
+            argsKey: queryArg,
+        });
+    }
+    const t = tool;
+    const innerCall = pickCallable(t, tool);
+    const wrappedFunc = async (args) => {
+        const out = await innerCall(args);
+        if (typeof out !== 'string')
+            return out;
+        const q = resolveTheQuery(args);
+        return compressSafe(compresr, {
+            context: out,
+            ...(q !== undefined ? { query: q } : {}),
+            compressionModel,
+            targetCompressionRatio,
+            ...(coarse !== undefined ? { coarse } : {}),
+            minTokens,
+            onError,
+            contextLabel: `tool:${t.name}`,
+        });
+    };
+    const cfg = { name: t.name };
+    if (t.description !== undefined)
+        cfg.description = t.description;
+    if (t.schema !== undefined)
+        cfg.schema = t.schema;
+    return toolFactory(wrappedFunc, cfg);
+}
+function pickCallable(meta, tool) {
+    if (typeof meta.func === 'function')
+        return meta.func.bind(tool);
+    if (typeof meta._call === 'function')
+        return meta._call.bind(tool);
+    if (typeof meta.invoke === 'function')
+        return meta.invoke.bind(tool);
+    throw new TypeError(`Tool '${meta.name}' has no callable surface.`);
+}
+/** Decorator-style helper — for users who prefer function composition. */
+export function compressToolOutput(options = {}) {
+    return (tool) => wrapToolWithCompression(tool, options);
+}
+//# sourceMappingURL=wrappers.js.map

package/dist/esm/integrations/langchain/wrappers.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"wrappers.js","sourceRoot":"","sources":["../../../../src/integrations/langchain/wrappers.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AACH,OAAO,EAAE,IAAI,IAAI,WAAW,EAAuB,MAAM,uBAAuB,CAAC;AAGjF,OAAO,EACL,WAAW,EACX,YAAY,EACZ,kBAAkB,EAClB,aAAa,EACb,cAAc,EACd,aAAa,EAEb,YAAY,GACb,MAAM,qBAAqB,CAAC;AA2B7B,SAAS,eAAe,CAAC,CAAU;IACjC,IAAI,OAAO,CAAC,KAAK,QAAQ,IAAI,CAAC,KAAK,IAAI;QAAE,OAAO,KAAK,CAAC;IACtD,MAAM,CAAC,GAAG,CAAkB,CAAC;IAC7B,OAAO,CACL,OAAO,CAAC,CAAC,IAAI,KAAK,QAAQ;QAC1B,CAAC,OAAO,CAAC,CAAC,IAAI,KAAK,UAAU;YAC3B,OAAO,CAAC,CAAC,KAAK,KAAK,UAAU;YAC7B,OAAO,CAAC,CAAC,MAAM,KAAK,UAAU,CAAC,CAClC,CAAC;AACJ,CAAC;AAED;;;;;;GAMG;AACH,MAAM,UAAU,uBAAuB,CACrC,IAAwB,EACxB,UAA2B,EAAE;IAE7B,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,EAAE,CAAC;QAC3B,MAAM,IAAI,SAAS,CACjB,mEAAmE;YACjE,0EAA0E,CAC7E,CAAC;IACJ,CAAC;IAED,MAAM,EACJ,MAAM,EACN,MAAM,EACN,OAAO,EACP,gBAAgB,GAAG,aAAa,EAChC,sBAAsB,GAAG,aAAa,EACtC,SAAS,GAAG,kBAAkB,EAC9B,MAAM,EACN,KAAK,EACL,cAAc,EACd,QAAQ,EACR,OAAO,GAAG,cAAc,GACzB,GAAG,OAAO,CAAC;IAEZ,MAAM,QAAQ,GACZ,MAAM,IAAI,WAAW,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,yBAAyB,EAAE,CAAC,CAAC;IAEhF,SAAS,eAAe,CAAC,IAA6B;QACpD,OAAO,YAAY,CAAC;YAClB,WAAW,EAAE,KAAK;YAClB,SAAS,EAAE,cAAc;gBACvB,CAAC,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,cAAc,CAAC,GAA8B,CAAC;gBACzD,CAAC,CAAC,SAAS;YACb,YAAY,EAAE,cAAc,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS;YAC/C,IAAI;YACJ,OAAO,EAAE,QAAQ;SAClB,CAAC,CAAC;IACL,CAAC;IAED,MAAM,CAAC,GAAG,IAAqB,CAAC;IAChC,MAAM,SAAS,GAAG,YAAY,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC;IAExC,MAAM,WAAW,GAAG,KAAK,EAAE,IAA6B,EAAoB,EAAE;QAC5E,MAAM,GAAG,GAAG,MAAM,SAAS,CAAC,IAAI,CAAC,CAAC;QAClC,IAAI,OAAO,GAAG,KAAK,QAAQ;YAAE,OAAO,GAAG,CAAC;QACxC,MAAM,CAAC,GAAG,eAAe,CAAC,IAAI,CAAC,CAAC;QAChC,OAAO,YAAY,CAAC,QAAQ,EAAE;YAC5B,OAAO,EAAE,GAAG;YACZ,GAAG,CAAC,CAAC,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACxC,gBAAgB;YAChB,sBAAsB;YACtB,GAAG,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC3C,SAAS;YACT,OAAO;YACP,YAAY,EAAE,QAAQ,CAAC,CAAC,IAAI,EAAE;SAC/B,CAAC,CAAC;IACL,CAAC,CAAC;IAEF,MAAM,GAAG,GAA4B,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;IACtD,IAAI,CAAC,CAAC,WAAW,KAAK,SAAS;QAAE,GAAG,CAAC,WAAW,GAAG,CAAC,CAAC,WAAW,CAAC;IACjE,IAAI,CAAC,CAAC,MAAM,KAAK,SAAS;QAAE,GAAG,CAAC,MAAM,GAAG,CAAC,CAAC,MAAM,CAAC;IAClD,OAAQ,WAAuG,CAC7G,WAAW,EACX,GAAG,CACJ,CAAC;AACJ,CAAC;AAED,SAAS,YAAY,CACnB,IAAmB,EACnB,IAAa;IAEb,IAAI,OAAO,IAAI,CAAC,IAAI,KAAK,UAAU;QAAE,OAAO,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACjE,IAAI,OAAO,IAAI,CAAC,KAAK,KAAK,UAAU;QAAE,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACnE,IAAI,OAAO,IAAI,CAAC,MAAM,KAAK,UAAU;QAAE,OAAO,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACrE,MAAM,IAAI,SAAS,CAAC,SAAS,IAAI,CAAC,IAAI,4BAA4B,CAAC,CAAC;AACtE,CAAC;AAID,0EAA0E;AAC1E,MAAM,UAAU,kBAAkB,CAAC,UAA2B,EAAE;IAC9D,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC,uBAAuB,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;AAC1D,CAAC"}

package/dist/esm/integrations/langgraph/checkpoint.d.ts

@@ -0,0 +1,49 @@
+/**
+ * `CompresrCheckpointSerializer` — compresses long string fields before
+ * they hit the checkpoint store.
+ *
+ * Lossy by design: there is no decompression — compressed strings persist
+ * in storage and flow back into state on resume. Use `fields` to restrict
+ * compression to known-large keys.
+ */
+import type { CompressionClient } from '../../clients/compression.js';
+import { type ErrorPolicy } from '../_shared/index.js';
+export interface CompresrCheckpointSerializerOptions {
+    apiKey?: string;
+    client?: CompressionClient;
+    baseUrl?: string;
+    compressionModel?: string;
+    /**
+     * Query passed to `latte_v1`. If unset, the field name (e.g. `retrieved_text`)
+     * is used as the query — a reasonable hint for what the value represents.
+     */
+    query?: string;
+    targetCompressionRatio?: number;
+    minTokens?: number;
+    coarse?: boolean;
+    fields?: ReadonlySet<string>;
+    onError?: ErrorPolicy;
+}
+/**
+ * Drop-in for LangGraph.js checkpoint serializers. Calls
+ * `dumpsTyped(obj) → [tag, bytes]`; recursively rewrites long strings
+ * into a `{ __compresr__: true, v: <compressed> }` sentinel before
+ * JSON-encoding. `loadsTyped` returns the payload as-is (sentinels and all).
+ */
+export declare class CompresrCheckpointSerializer {
+    private readonly compresr;
+    private readonly compressionModel;
+    private readonly query;
+    private readonly targetCompressionRatio;
+    private readonly minTokens;
+    private readonly coarse;
+    private readonly fields;
+    private readonly onError;
+    constructor(options: CompresrCheckpointSerializerOptions);
+    dumpsTyped(obj: unknown): Promise<[string, Uint8Array]>;
+    loadsTyped([_tag, bytes]: [string, Uint8Array]): unknown;
+    private walkCompress;
+    private shouldCompress;
+    private wrapCompressed;
+}
+//# sourceMappingURL=checkpoint.d.ts.map

package/dist/esm/integrations/langgraph/checkpoint.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"checkpoint.d.ts","sourceRoot":"","sources":["../../../../src/integrations/langgraph/checkpoint.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AACH,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,8BAA8B,CAAC;AACtE,OAAO,EAML,KAAK,WAAW,EAEjB,MAAM,qBAAqB,CAAC;AAI7B,MAAM,WAAW,mCAAmC;IAClD,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,MAAM,CAAC,EAAE,iBAAiB,CAAC;IAC3B,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B;;;OAGG;IACH,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,sBAAsB,CAAC,EAAE,MAAM,CAAC;IAChC,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB,MAAM,CAAC,EAAE,WAAW,CAAC,MAAM,CAAC,CAAC;IAC7B,OAAO,CAAC,EAAE,WAAW,CAAC;CACvB;AAED;;;;;GAKG;AACH,qBAAa,4BAA4B;IACvC,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAoB;IAC7C,OAAO,CAAC,QAAQ,CAAC,gBAAgB,CAAS;IAC1C,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAqB;IAC3C,OAAO,CAAC,QAAQ,CAAC,sBAAsB,CAAS;IAChD,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAS;IACnC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAsB;IAC7C,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAkC;IACzD,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAc;gBAE1B,OAAO,EAAE,mCAAmC;IAyBlD,UAAU,CAAC,GAAG,EAAE,OAAO,GAAG,OAAO,CAAC,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;IAM7D,UAAU,CAAC,CAAC,IAAI,EAAE,KAAK,CAAC,EAAE,CAAC,MAAM,EAAE,UAAU,CAAC,GAAG,OAAO;YAI1C,YAAY;IAmB1B,OAAO,CAAC,cAAc;YAOR,cAAc;CAc7B"}