@compresr/sdk 1.3.2 → 1.5.1

package/dist/cjs/integrations/langchain/middleware.d.ts

@@ -0,0 +1,94 @@
+/**
+ * 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 { ToolMessage } from '@langchain/core/messages';
+import type { CompressionClient } from '../../clients/compression.js';
+import { type ErrorPolicy } from '../_shared/index.js';
+type ToolMessageType = ToolMessage;
+export interface ToolCall {
+    id?: string;
+    name?: string;
+    args?: Record<string, unknown>;
+}
+export interface ToolCallRequest {
+    toolCall: ToolCall;
+    messages?: unknown[];
+}
+export type ToolHandler = (request: ToolCallRequest) => Promise<ToolMessageType> | ToolMessageType;
+export type ToolQueryExtractor = (toolCall: ToolCall, messages: unknown[]) => string | undefined;
+export type SummaryQueryExtractor = (messages: unknown[]) => string | undefined;
+export interface CompresrMiddlewareCommonOptions {
+    apiKey?: string;
+    client?: CompressionClient;
+    baseUrl?: string;
+    compressionModel?: string;
+    targetCompressionRatio?: number;
+    coarse?: boolean;
+    onError?: ErrorPolicy;
+}
+export interface CompresrToolMiddlewareOptions extends CompresrMiddlewareCommonOptions {
+    minTokens?: number;
+    allowTools?: Iterable<string>;
+    ignoreTools?: Iterable<string>;
+    query?: string;
+    queryExtractor?: ToolQueryExtractor;
+    queryArg?: string;
+}
+export interface CompresrSummarizationMiddlewareOptions extends CompresrMiddlewareCommonOptions {
+    /** Compression fires when the message-state token count crosses this. */
+    maxTokensBeforeSummary?: number;
+    /** Number of recent messages to preserve verbatim. */
+    messagesToKeep?: number;
+    /** Alias for `maxTokensBeforeSummary` — matches LangChain.js `SummarizationMiddleware`. */
+    trigger?: number;
+    /** Alias for `messagesToKeep` — matches LangChain.js `SummarizationMiddleware`. */
+    keep?: number;
+    /** Custom token counter; defaults to the chars/4 estimator. */
+    tokenCounter?: (s: string) => number;
+    query?: string;
+    queryExtractor?: SummaryQueryExtractor;
+}
+export interface ModelRequest {
+    messages: unknown[];
+    [k: string]: unknown;
+}
+export type ModelHandler<T = unknown> = (request: ModelRequest) => Promise<T> | T;
+export interface CompresrMiddleware {
+    name: string;
+    wrapToolCall?: (request: ToolCallRequest, handler: ToolHandler) => Promise<ToolMessageType>;
+    beforeModel?: (state: {
+        messages?: unknown[];
+    }) => Promise<{
+        messages?: unknown[];
+        llm_input_messages?: unknown[];
+    } | undefined>;
+    wrapModelCall?: <T = unknown>(request: ModelRequest, handler: ModelHandler<T>) => Promise<T>;
+}
+export declare function compresrToolMiddleware(options: CompresrToolMiddlewareOptions): CompresrMiddleware;
+export declare function compresrSummarizationMiddleware(options: CompresrSummarizationMiddlewareOptions): CompresrMiddleware;
+export interface CompresrPromptMiddlewareOptions extends CompresrMiddlewareCommonOptions {
+    /** Hard ceiling for the outbound prompt (sum of message content tokens). */
+    maxTokens: number;
+    /** Skip compression on individual messages shorter than this. */
+    minTokens?: number;
+    tokenCounter?: (s: string) => number;
+    query?: string;
+    queryExtractor?: SummaryQueryExtractor;
+}
+export declare function compresrPromptMiddleware(options: CompresrPromptMiddlewareOptions): CompresrMiddleware;
+export {};
+//# sourceMappingURL=middleware.d.ts.map

package/dist/cjs/integrations/langchain/middleware.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"middleware.d.ts","sourceRoot":"","sources":["../../../../src/integrations/langchain/middleware.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;GAgBG;AACH,OAAO,EAGL,WAAW,EAEZ,MAAM,0BAA0B,CAAC;AAElC,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,8BAA8B,CAAC;AACtE,OAAO,EAOL,KAAK,WAAW,EAIjB,MAAM,qBAAqB,CAAC;AAI7B,KAAK,eAAe,GAAG,WAAW,CAAC;AAEnC,MAAM,WAAW,QAAQ;IACvB,EAAE,CAAC,EAAE,MAAM,CAAC;IACZ,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CAChC;AAED,MAAM,WAAW,eAAe;IAC9B,QAAQ,EAAE,QAAQ,CAAC;IACnB,QAAQ,CAAC,EAAE,OAAO,EAAE,CAAC;CACtB;AAED,MAAM,MAAM,WAAW,GAAG,CACxB,OAAO,EAAE,eAAe,KACrB,OAAO,CAAC,eAAe,CAAC,GAAG,eAAe,CAAC;AAEhD,MAAM,MAAM,kBAAkB,GAAG,CAC/B,QAAQ,EAAE,QAAQ,EAClB,QAAQ,EAAE,OAAO,EAAE,KAChB,MAAM,GAAG,SAAS,CAAC;AAExB,MAAM,MAAM,qBAAqB,GAAG,CAAC,QAAQ,EAAE,OAAO,EAAE,KAAK,MAAM,GAAG,SAAS,CAAC;AAEhF,MAAM,WAAW,+BAA+B;IAC9C,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,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB,OAAO,CAAC,EAAE,WAAW,CAAC;CACvB;AAED,MAAM,WAAW,6BACf,SAAQ,+BAA+B;IACvC,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,UAAU,CAAC,EAAE,QAAQ,CAAC,MAAM,CAAC,CAAC;IAC9B,WAAW,CAAC,EAAE,QAAQ,CAAC,MAAM,CAAC,CAAC;IAC/B,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,cAAc,CAAC,EAAE,kBAAkB,CAAC;IACpC,QAAQ,CAAC,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,sCACf,SAAQ,+BAA+B;IACvC,yEAAyE;IACzE,sBAAsB,CAAC,EAAE,MAAM,CAAC;IAChC,sDAAsD;IACtD,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,2FAA2F;IAC3F,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,mFAAmF;IACnF,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,+DAA+D;IAC/D,YAAY,CAAC,EAAE,CAAC,CAAC,EAAE,MAAM,KAAK,MAAM,CAAC;IACrC,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,cAAc,CAAC,EAAE,qBAAqB,CAAC;CACxC;AAED,MAAM,WAAW,YAAY;IAC3B,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,CAAC,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC;CACtB;AAED,MAAM,MAAM,YAAY,CAAC,CAAC,GAAG,OAAO,IAAI,CAAC,OAAO,EAAE,YAAY,KAAK,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;AAElF,MAAM,WAAW,kBAAkB;IACjC,IAAI,EAAE,MAAM,CAAC;IACb,YAAY,CAAC,EAAE,CACb,OAAO,EAAE,eAAe,EACxB,OAAO,EAAE,WAAW,KACjB,OAAO,CAAC,eAAe,CAAC,CAAC;IAC9B,WAAW,CAAC,EAAE,CAAC,KAAK,EAAE;QACpB,QAAQ,CAAC,EAAE,OAAO,EAAE,CAAC;KACtB,KAAK,OAAO,CAAC;QAAE,QAAQ,CAAC,EAAE,OAAO,EAAE,CAAC;QAAC,kBAAkB,CAAC,EAAE,OAAO,EAAE,CAAA;KAAE,GAAG,SAAS,CAAC,CAAC;IACpF,aAAa,CAAC,EAAE,CAAC,CAAC,GAAG,OAAO,EAC1B,OAAO,EAAE,YAAY,EACrB,OAAO,EAAE,YAAY,CAAC,CAAC,CAAC,KACrB,OAAO,CAAC,CAAC,CAAC,CAAC;CACjB;AA2BD,wBAAgB,sBAAsB,CACpC,OAAO,EAAE,6BAA6B,GACrC,kBAAkB,CA4DpB;AAoBD,wBAAgB,+BAA+B,CAC7C,OAAO,EAAE,sCAAsC,GAC9C,kBAAkB,CAuFpB;AAMD,MAAM,WAAW,+BAAgC,SAAQ,+BAA+B;IACtF,4EAA4E;IAC5E,SAAS,EAAE,MAAM,CAAC;IAClB,iEAAiE;IACjE,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,YAAY,CAAC,EAAE,CAAC,CAAC,EAAE,MAAM,KAAK,MAAM,CAAC;IACrC,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,cAAc,CAAC,EAAE,qBAAqB,CAAC;CACxC;AAmBD,wBAAgB,wBAAwB,CACtC,OAAO,EAAE,+BAA+B,GACvC,kBAAkB,CA+FpB"}

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

@@ -0,0 +1,273 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.compresrToolMiddleware = compresrToolMiddleware;
+exports.compresrSummarizationMiddleware = compresrSummarizationMiddleware;
+exports.compresrPromptMiddleware = compresrPromptMiddleware;
+/**
+ * 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
+ */
+const messages_1 = require("@langchain/core/messages");
+const index_js_1 = require("../_shared/index.js");
+const REMOVE_ALL_MESSAGES = '__remove_all__';
+function isToolMessage(value) {
+    return value instanceof messages_1.ToolMessage;
+}
+function buildToolMessage(content, original) {
+    const orig = original;
+    return new messages_1.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);
+}
+function compresrToolMiddleware(options) {
+    const { apiKey, client, baseUrl, compressionModel = index_js_1.DEFAULT_MODEL, targetCompressionRatio = index_js_1.DEFAULT_RATIO, minTokens = index_js_1.DEFAULT_MIN_TOKENS, coarse, allowTools, ignoreTools, onError = index_js_1.DEFAULT_POLICY, query, queryExtractor, queryArg, } = options;
+    const eligible = (0, index_js_1.makeFilter)({ allow: allowTools, ignore: ignoreTools });
+    const compresr = client ?? (0, index_js_1.buildClient)({ apiKey, baseUrl, caller: 'compresrToolMiddleware' });
+    function resolveTheQuery(toolCall, messages) {
+        return (0, index_js_1.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 (0, index_js_1.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 messages_1.HumanMessage))
+        return false;
+    const c = m.content;
+    return typeof c === 'string' && c.startsWith(SUMMARY_PREFIX);
+}
+function compresrSummarizationMiddleware(options) {
+    const { apiKey, client, baseUrl, compressionModel = index_js_1.DEFAULT_MODEL, targetCompressionRatio = index_js_1.DEFAULT_RATIO, coarse, onError = index_js_1.DEFAULT_POLICY, maxTokensBeforeSummary = 4_000, messagesToKeep = 20, trigger, keep: keepAlias, tokenCounter, query, queryExtractor, } = options;
+    const compresr = client ?? (0, index_js_1.buildClient)({ apiKey, baseUrl, caller: 'compresrSummarizationMiddleware' });
+    const maxTokens = Math.max(1, trigger ?? maxTokensBeforeSummary);
+    const keep = Math.max(1, keepAlias ?? messagesToKeep);
+    const countTokens = tokenCounter ?? index_js_1.estimateTokens;
+    function resolveTheQuery(messages) {
+        return (0, index_js_1.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 (0, index_js_1.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 messages_1.RemoveMessage({ id: REMOVE_ALL_MESSAGES }),
+                    new messages_1.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;
+    }
+}
+function compresrPromptMiddleware(options) {
+    const { apiKey, client, baseUrl, compressionModel = index_js_1.DEFAULT_MODEL, coarse, onError = index_js_1.DEFAULT_POLICY, maxTokens, minTokens = index_js_1.DEFAULT_MIN_TOKENS, tokenCounter, query, queryExtractor, } = options;
+    const compresr = client ?? (0, index_js_1.buildClient)({ apiKey, baseUrl, caller: 'compresrPromptMiddleware' });
+    const budget = Math.max(1, maxTokens);
+    const floor = Math.max(1, minTokens);
+    const countTokens = tokenCounter ?? index_js_1.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 (0, index_js_1.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 (0, index_js_1.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/cjs/integrations/langchain/middleware.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"middleware.js","sourceRoot":"","sources":["../../../../src/integrations/langchain/middleware.ts"],"names":[],"mappings":";;AAmJA,wDA8DC;AAoBD,0EAyFC;AAiCD,4DAiGC;AAhcD;;;;;;;;;;;;;;;;GAgBG;AACH,uDAKkC;AAGlC,kDAW6B;AAE7B,MAAM,mBAAmB,GAAG,gBAAgB,CAAC;AAoF7C,SAAS,aAAa,CAAC,KAAc;IACnC,OAAO,KAAK,YAAY,sBAAW,CAAC;AACtC,CAAC;AAED,SAAS,gBAAgB,CAAC,OAAe,EAAE,QAAyB;IAClE,MAAM,IAAI,GAAG,QAAoD,CAAC;IAClE,OAAO,IAAI,sBAAW,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,SAAgB,sBAAsB,CACpC,OAAsC;IAEtC,MAAM,EACJ,MAAM,EACN,MAAM,EACN,OAAO,EACP,gBAAgB,GAAG,wBAAa,EAChC,sBAAsB,GAAG,wBAAa,EACtC,SAAS,GAAG,6BAAkB,EAC9B,MAAM,EACN,UAAU,EACV,WAAW,EACX,OAAO,GAAG,yBAAc,EACxB,KAAK,EACL,cAAc,EACd,QAAQ,GACT,GAAG,OAAO,CAAC;IAEZ,MAAM,QAAQ,GAAG,IAAA,qBAAU,EAAC,EAAE,KAAK,EAAE,UAAU,EAAE,MAAM,EAAE,WAAW,EAAE,CAAC,CAAC;IACxE,MAAM,QAAQ,GACZ,MAAM,IAAI,IAAA,sBAAW,EAAC,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,wBAAwB,EAAE,CAAC,CAAC;IAE/E,SAAS,eAAe,CAAC,QAAkB,EAAE,QAAmB;QAC9D,OAAO,IAAA,uBAAY,EAAC;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,IAAA,uBAAY,EAAC,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,uBAAY,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,SAAgB,+BAA+B,CAC7C,OAA+C;IAE/C,MAAM,EACJ,MAAM,EACN,MAAM,EACN,OAAO,EACP,gBAAgB,GAAG,wBAAa,EAChC,sBAAsB,GAAG,wBAAa,EACtC,MAAM,EACN,OAAO,GAAG,yBAAc,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,IAAA,sBAAW,EAAC,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,yBAAc,CAAC;IAEnD,SAAS,eAAe,CAAC,QAAmB;QAC1C,OAAO,IAAA,uBAAY,EAAC;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,IAAA,uBAAY,EAAC,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,wBAAa,CAAC,EAAE,EAAE,EAAE,mBAAmB,EAAE,CAAC;oBAC9C,IAAI,uBAAY,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,SAAgB,wBAAwB,CACtC,OAAwC;IAExC,MAAM,EACJ,MAAM,EACN,MAAM,EACN,OAAO,EACP,gBAAgB,GAAG,wBAAa,EAChC,MAAM,EACN,OAAO,GAAG,yBAAc,EACxB,SAAS,EACT,SAAS,GAAG,6BAAkB,EAC9B,YAAY,EACZ,KAAK,EACL,cAAc,GACf,GAAG,OAAO,CAAC;IAEZ,MAAM,QAAQ,GACZ,MAAM,IAAI,IAAA,sBAAW,EAAC,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,yBAAc,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,IAAA,uBAAY,EAAC;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,IAAA,uBAAY,EAAC,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/cjs/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/cjs/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/cjs/integrations/langchain/retriever.js

@@ -0,0 +1,118 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.CompresrExtractor = void 0;
+/**
+ * `BaseDocumentCompressor` implementation for `ContextualCompressionRetriever`.
+ *
+ * Mirrors Python `CompresrExtractor`. One Compresr batch call
+ * replaces N LLM extraction calls.
+ */
+const documents_1 = require("@langchain/core/documents");
+const logger_js_1 = require("../../logger.js");
+const index_js_1 = require("../_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 }),
+ * });
+ * ```
+ */
+class CompresrExtractor {
+    compresr;
+    compressionModel;
+    targetCompressionRatio;
+    minTokens;
+    coarse;
+    onError;
+    dropBelowMin;
+    constructor(options) {
+        const { apiKey, client, baseUrl, compressionModel = index_js_1.DEFAULT_MODEL, targetCompressionRatio = index_js_1.DEFAULT_RATIO, minTokens = index_js_1.DEFAULT_MIN_TOKENS, coarse, onError = index_js_1.DEFAULT_POLICY, dropBelowMin = false, } = options;
+        this.compresr =
+            client ?? (0, index_js_1.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' && (0, index_js_1.estimateTokens)(pc) >= this.minTokens) {
+                indices.push(i);
+                eligible.push(doc);
+            }
+        });
+        return { indices, eligible };
+    }
+    emit(doc, content) {
+        const md = doc.metadata ?? {};
+        return new documents_1.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 += index_js_1.BATCH_LIMIT) {
+            const chunk = eligible.slice(start, start + index_js_1.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;
+                (0, logger_js_1.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;
+    }
+}
+exports.CompresrExtractor = CompresrExtractor;
+function stringifyError(exc) {
+    return exc instanceof Error ? exc.message : String(exc);
+}
+//# sourceMappingURL=retriever.js.map

package/dist/cjs/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,yDAAqD;AAGrD,+CAA4C;AAC5C,kDAS6B;AAiB7B;;;;;;;;;;;;;;;;GAgBG;AACH,MAAa,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,wBAAa,EAChC,sBAAsB,GAAG,wBAAa,EACtC,SAAS,GAAG,6BAAkB,EAC9B,MAAM,EACN,OAAO,GAAG,yBAAc,EACxB,YAAY,GAAG,KAAK,GACrB,GAAG,OAAO,CAAC;QAEZ,IAAI,CAAC,QAAQ;YACX,MAAM,IAAI,IAAA,sBAAW,EAAC,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,IAAA,yBAAc,EAAC,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,oBAAQ,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,sBAAW,EAAE,CAAC;YAClE,MAAM,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC,KAAK,EAAE,KAAK,GAAG,sBAAW,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,IAAA,qBAAS,GAAE,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;AA1GD,8CA0GC;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/cjs/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/cjs/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"}