@compresr/sdk 1.3.2 → 1.5.1

package/dist/cjs/integrations/langgraph/store.js

@@ -0,0 +1,121 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.CompresrStore = void 0;
+/**
+ * `CompresrStore` — wrap a LangGraph.js `BaseStore` so long string fields
+ * are compressed before being persisted to the underlying store.
+ *
+ * Composition over inheritance: holds an inner store and proxies all
+ * methods, intercepting `put` and `batch` to walk the value object and
+ * rewrite strings above `minTokens` in place. Lossy — there is no
+ * decompression on read.
+ */
+const langgraph_checkpoint_1 = require("@langchain/langgraph-checkpoint");
+const index_js_1 = require("../_shared/index.js");
+function isPutOp(op) {
+    const o = op;
+    return Object.prototype.hasOwnProperty.call(op, 'value') && typeof o.value !== 'undefined';
+}
+/**
+ * Drop-in `BaseStore` wrapper that compresses string fields on write.
+ *
+ * @example
+ * import { InMemoryStore } from '@langchain/langgraph';
+ * import { CompresrStore } from '@compresr/sdk/integrations/langgraph';
+ *
+ * const store = new CompresrStore(new InMemoryStore(), {
+ *   apiKey: process.env.COMPRESR_API_KEY!,
+ *   fields: new Set(['retrievedText']),
+ *   minTokens: 500,
+ * });
+ */
+class CompresrStore extends langgraph_checkpoint_1.BaseStore {
+    inner;
+    compresr;
+    compressionModel;
+    query;
+    targetCompressionRatio;
+    minTokens;
+    coarse;
+    fields;
+    onError;
+    constructor(inner, options = {}) {
+        super();
+        const { apiKey, client, baseUrl, compressionModel = index_js_1.DEFAULT_MODEL, query, targetCompressionRatio = index_js_1.DEFAULT_RATIO, minTokens = 500, coarse, fields, onError = index_js_1.DEFAULT_POLICY, } = options;
+        this.inner = inner;
+        this.compresr = client ?? (0, index_js_1.buildClient)({ apiKey, baseUrl, caller: 'CompresrStore' });
+        this.compressionModel = compressionModel;
+        this.query = query;
+        this.targetCompressionRatio = targetCompressionRatio;
+        this.minTokens = Math.max(1, minTokens);
+        this.coarse = coarse;
+        this.fields = fields;
+        this.onError = onError;
+    }
+    shouldCompress(text, parentKey) {
+        if (this.fields !== undefined && (parentKey === undefined || !this.fields.has(parentKey))) {
+            return false;
+        }
+        return (0, index_js_1.estimateTokens)(text) >= this.minTokens;
+    }
+    async compressValue(text, parentKey) {
+        return await (0, index_js_1.compressSafe)(this.compresr, {
+            context: text,
+            query: this.query ?? parentKey ?? 'stored content',
+            compressionModel: this.compressionModel,
+            targetCompressionRatio: this.targetCompressionRatio,
+            ...(this.coarse !== undefined ? { coarse: this.coarse } : {}),
+            minTokens: 1,
+            onError: this.onError,
+            contextLabel: `store:${parentKey ?? '?'}`,
+        });
+    }
+    async walk(value, parentKey) {
+        if (typeof value === 'string') {
+            if (!this.shouldCompress(value, parentKey))
+                return value;
+            return await this.compressValue(value, parentKey);
+        }
+        if (Array.isArray(value)) {
+            return Promise.all(value.map((v) => this.walk(v, parentKey)));
+        }
+        if (value !== null && typeof value === 'object') {
+            const entries = await Promise.all(Object.entries(value).map(async ([k, v]) => [k, await this.walk(v, k)]));
+            return Object.fromEntries(entries);
+        }
+        return value;
+    }
+    async put(namespace, key, value, index) {
+        const compressed = (await this.walk(value, undefined));
+        return this.inner.put(namespace, key, compressed, index);
+    }
+    async batch(operations) {
+        const rewritten = await Promise.all(operations.map(async (op) => {
+            if (isPutOp(op) && op.value !== null) {
+                return { ...op, value: (await this.walk(op.value, undefined)) };
+            }
+            return op;
+        }));
+        return this.inner.batch(rewritten);
+    }
+    async get(namespace, key) {
+        return this.inner.get(namespace, key);
+    }
+    async search(namespacePrefix, options) {
+        return this.inner.search(namespacePrefix, options);
+    }
+    async delete(namespace, key) {
+        return this.inner.delete(namespace, key);
+    }
+    async listNamespaces(options) {
+        return this.inner.listNamespaces(options);
+    }
+    async start() {
+        await this.inner.start();
+    }
+    async stop() {
+        await this.inner.stop();
+    }
+}
+exports.CompresrStore = CompresrStore;
+//# sourceMappingURL=store.js.map

package/dist/cjs/integrations/langgraph/store.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"store.js","sourceRoot":"","sources":["../../../../src/integrations/langgraph/store.ts"],"names":[],"mappings":";;;AAAA;;;;;;;;GAQG;AACH,0EAMyC;AAGzC,kDAQ6B;AA0B7B,SAAS,OAAO,CAAC,EAAa;IAC5B,MAAM,CAAC,GAAG,EAAyB,CAAC;IACpC,OAAO,MAAM,CAAC,SAAS,CAAC,cAAc,CAAC,IAAI,CAAC,EAAE,EAAE,OAAO,CAAC,IAAI,OAAO,CAAC,CAAC,KAAK,KAAK,WAAW,CAAC;AAC7F,CAAC;AAED;;;;;;;;;;;;GAYG;AACH,MAAa,aAAc,SAAQ,gCAAS;IACzB,KAAK,CAAY;IACjB,QAAQ,CAAoB;IAC5B,gBAAgB,CAAS;IACzB,KAAK,CAAqB;IAC1B,sBAAsB,CAAS;IAC/B,SAAS,CAAS;IAClB,MAAM,CAAsB;IAC5B,MAAM,CAAkC;IACxC,OAAO,CAAc;IAEtC,YAAY,KAAgB,EAAE,UAAgC,EAAE;QAC9D,KAAK,EAAE,CAAC;QACR,MAAM,EACJ,MAAM,EACN,MAAM,EACN,OAAO,EACP,gBAAgB,GAAG,wBAAa,EAChC,KAAK,EACL,sBAAsB,GAAG,wBAAa,EACtC,SAAS,GAAG,GAAG,EACf,MAAM,EACN,MAAM,EACN,OAAO,GAAG,yBAAc,GACzB,GAAG,OAAO,CAAC;QAEZ,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,QAAQ,GAAG,MAAM,IAAI,IAAA,sBAAW,EAAC,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,eAAe,EAAE,CAAC,CAAC;QACpF,IAAI,CAAC,gBAAgB,GAAG,gBAAgB,CAAC;QACzC,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,sBAAsB,GAAG,sBAAsB,CAAC;QACrD,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC;QACxC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;IACzB,CAAC;IAEO,cAAc,CAAC,IAAY,EAAE,SAA6B;QAChE,IAAI,IAAI,CAAC,MAAM,KAAK,SAAS,IAAI,CAAC,SAAS,KAAK,SAAS,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,EAAE,CAAC;YAC1F,OAAO,KAAK,CAAC;QACf,CAAC;QACD,OAAO,IAAA,yBAAc,EAAC,IAAI,CAAC,IAAI,IAAI,CAAC,SAAS,CAAC;IAChD,CAAC;IAEO,KAAK,CAAC,aAAa,CAAC,IAAY,EAAE,SAA6B;QACrE,OAAO,MAAM,IAAA,uBAAY,EAAC,IAAI,CAAC,QAAQ,EAAE;YACvC,OAAO,EAAE,IAAI;YACb,KAAK,EAAE,IAAI,CAAC,KAAK,IAAI,SAAS,IAAI,gBAAgB;YAClD,gBAAgB,EAAE,IAAI,CAAC,gBAAgB;YACvC,sBAAsB,EAAE,IAAI,CAAC,sBAAsB;YACnD,GAAG,CAAC,IAAI,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC7D,SAAS,EAAE,CAAC;YACZ,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,YAAY,EAAE,SAAS,SAAS,IAAI,GAAG,EAAE;SAC1C,CAAC,CAAC;IACL,CAAC;IAEO,KAAK,CAAC,IAAI,CAAC,KAAc,EAAE,SAA6B;QAC9D,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;YAC9B,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,KAAK,EAAE,SAAS,CAAC;gBAAE,OAAO,KAAK,CAAC;YACzD,OAAO,MAAM,IAAI,CAAC,aAAa,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;QACpD,CAAC;QACD,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;YACzB,OAAO,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC,CAAC,CAAC;QAChE,CAAC;QACD,IAAI,KAAK,KAAK,IAAI,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;YAChD,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,GAAG,CAC/B,MAAM,CAAC,OAAO,CAAC,KAAgC,CAAC,CAAC,GAAG,CAClD,KAAK,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,MAAM,IAAI,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC,CAAC,CAAU,CACtD,CACF,CAAC;YACF,OAAO,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;QACrC,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,KAAK,CAAC,GAAG,CACP,SAAmB,EACnB,GAAW,EACX,KAA8B,EAC9B,KAAwB;QAExB,MAAM,UAAU,GAAG,CAAC,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,SAAS,CAAC,CAA4B,CAAC;QAClF,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,EAAE,GAAG,EAAE,UAAU,EAAE,KAAK,CAAC,CAAC;IAC3D,CAAC;IAED,KAAK,CAAC,KAAK,CAAyB,UAAc;QAChD,MAAM,SAAS,GAAG,MAAM,OAAO,CAAC,GAAG,CACjC,UAAU,CAAC,GAAG,CAAC,KAAK,EAAE,EAAE,EAAE,EAAE;YAC1B,IAAI,OAAO,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC,KAAK,KAAK,IAAI,EAAE,CAAC;gBACrC,OAAO,EAAE,GAAG,EAAE,EAAE,KAAK,EAAE,CAAC,MAAM,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,KAAK,EAAE,SAAS,CAAC,CAA4B,EAAE,CAAC;YAC7F,CAAC;YACD,OAAO,EAAE,CAAC;QACZ,CAAC,CAAC,CACH,CAAC;QACF,OAAO,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,SAAe,CAAC,CAAC;IAC3C,CAAC;IAED,KAAK,CAAC,GAAG,CAAC,SAAmB,EAAE,GAAW;QACxC,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,EAAE,GAAG,CAAC,CAAC;IACxC,CAAC;IAED,KAAK,CAAC,MAAM,CACV,eAAyB,EACzB,OAA+F;QAE/F,OAAO,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,eAAe,EAAE,OAAO,CAAC,CAAC;IACrD,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,SAAmB,EAAE,GAAW;QAC3C,OAAO,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,SAAS,EAAE,GAAG,CAAC,CAAC;IAC3C,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,OAMpB;QACC,OAAO,IAAI,CAAC,KAAK,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;IAC5C,CAAC;IAED,KAAK,CAAC,KAAK;QACT,MAAM,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;IAC3B,CAAC;IAED,KAAK,CAAC,IAAI;QACR,MAAM,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC;IAC1B,CAAC;CACF;AAlID,sCAkIC"}

package/dist/cjs/integrations/llamaindex/index.d.ts

@@ -0,0 +1,10 @@
+/**
+ * LlamaIndex.TS integration for Compresr.
+ *
+ *     npm install llamaindex   # (peer)
+ */
+export { CompresrNodePostprocessor, type CompresrNodePostprocessorOptions, } from './postprocessor.js';
+export { wrapToolWithCompresr, type WrapToolOptions, type CompresrWrappableTool, } from './wrappers.js';
+export { CompresrMemoryBlock, type CompresrMemoryBlockOptions } from './memory.js';
+export type { ErrorPolicy } from '../_shared/index.js';
+//# sourceMappingURL=index.d.ts.map

package/dist/cjs/integrations/llamaindex/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/integrations/llamaindex/index.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AACH,OAAO,EACL,yBAAyB,EACzB,KAAK,gCAAgC,GACtC,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EACL,oBAAoB,EACpB,KAAK,eAAe,EACpB,KAAK,qBAAqB,GAC3B,MAAM,eAAe,CAAC;AACvB,OAAO,EAAE,mBAAmB,EAAE,KAAK,0BAA0B,EAAE,MAAM,aAAa,CAAC;AACnF,YAAY,EAAE,WAAW,EAAE,MAAM,qBAAqB,CAAC"}

package/dist/cjs/integrations/llamaindex/index.js

@@ -0,0 +1,15 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.CompresrMemoryBlock = exports.wrapToolWithCompresr = exports.CompresrNodePostprocessor = void 0;
+/**
+ * LlamaIndex.TS integration for Compresr.
+ *
+ *     npm install llamaindex   # (peer)
+ */
+var postprocessor_js_1 = require("./postprocessor.js");
+Object.defineProperty(exports, "CompresrNodePostprocessor", { enumerable: true, get: function () { return postprocessor_js_1.CompresrNodePostprocessor; } });
+var wrappers_js_1 = require("./wrappers.js");
+Object.defineProperty(exports, "wrapToolWithCompresr", { enumerable: true, get: function () { return wrappers_js_1.wrapToolWithCompresr; } });
+var memory_js_1 = require("./memory.js");
+Object.defineProperty(exports, "CompresrMemoryBlock", { enumerable: true, get: function () { return memory_js_1.CompresrMemoryBlock; } });
+//# sourceMappingURL=index.js.map

package/dist/cjs/integrations/llamaindex/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/integrations/llamaindex/index.ts"],"names":[],"mappings":";;;AAAA;;;;GAIG;AACH,uDAG4B;AAF1B,6HAAA,yBAAyB,OAAA;AAG3B,6CAIuB;AAHrB,mHAAA,oBAAoB,OAAA;AAItB,yCAAmF;AAA1E,gHAAA,mBAAmB,OAAA"}

package/dist/cjs/integrations/llamaindex/memory.d.ts

@@ -0,0 +1,74 @@
+/**
+ * `CompresrMemoryBlock` — long-term memory block that compresses the
+ * conversation buffer with Compresr before serving it back to the model.
+ *
+ * LlamaIndex.TS's `BaseMemoryBlock` exposes `put` (sink) and `get` (source).
+ * We accumulate puts into a single text buffer and, on `get`, return one
+ * compressed message — or the raw buffer if it's below the compression
+ * threshold.
+ */
+import type { CompressionClient } from '../../clients/compression.js';
+import { type ErrorPolicy } from '../_shared/index.js';
+interface PartialMemoryMessage {
+    id?: string;
+    role: string;
+    content: unknown;
+    createdAt?: Date;
+}
+interface BaseMemoryBlockShape {
+    readonly id: string;
+    readonly priority: number;
+    readonly isLongTerm: boolean;
+    get(messages?: PartialMemoryMessage[]): Promise<PartialMemoryMessage[]>;
+    put(messages: PartialMemoryMessage[]): Promise<void>;
+}
+export interface CompresrMemoryBlockOptions {
+    id?: string;
+    priority?: number;
+    apiKey?: string;
+    client?: CompressionClient;
+    baseUrl?: string;
+    compressionModel?: string;
+    /**
+     * Query passed to `latte_v1`. If unset, the most recent `user:` line in
+     * the buffer is used (falling back to a generic placeholder).
+     */
+    query?: string;
+    targetToken?: number;
+    targetCompressionRatio?: number;
+    minTokens?: number;
+    coarse?: boolean;
+    onError?: ErrorPolicy;
+}
+/**
+ * Drop-in `BaseMemoryBlock` for LlamaIndex.TS that compresses on `get`.
+ *
+ * @example
+ * const memory = new Memory({
+ *   tokenLimit: 8_000,
+ *   memoryBlocks: [
+ *     new CompresrMemoryBlock({ apiKey: process.env.COMPRESR_API_KEY!, targetToken: 2_000 }),
+ *   ],
+ * });
+ */
+export declare class CompresrMemoryBlock implements BaseMemoryBlockShape {
+    readonly id: string;
+    readonly priority: number;
+    readonly isLongTerm = true;
+    private readonly compresr;
+    private readonly compressionModel;
+    private readonly explicitQuery;
+    private readonly targetToken;
+    private readonly targetCompressionRatio;
+    private readonly minTokens;
+    private readonly coarse;
+    private readonly onError;
+    private buffer;
+    constructor(options?: CompresrMemoryBlockOptions);
+    private resolveQuery;
+    put(messages: PartialMemoryMessage[]): Promise<void>;
+    get(_messages?: PartialMemoryMessage[]): Promise<PartialMemoryMessage[]>;
+    private computeRatio;
+}
+export {};
+//# sourceMappingURL=memory.d.ts.map

package/dist/cjs/integrations/llamaindex/memory.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"memory.d.ts","sourceRoot":"","sources":["../../../../src/integrations/llamaindex/memory.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AACH,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,8BAA8B,CAAC;AACtE,OAAO,EAML,KAAK,WAAW,EAEjB,MAAM,qBAAqB,CAAC;AAE7B,UAAU,oBAAoB;IAC5B,EAAE,CAAC,EAAE,MAAM,CAAC;IACZ,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE,OAAO,CAAC;IACjB,SAAS,CAAC,EAAE,IAAI,CAAC;CAClB;AAED,UAAU,oBAAoB;IAC5B,QAAQ,CAAC,EAAE,EAAE,MAAM,CAAC;IACpB,QAAQ,CAAC,QAAQ,EAAE,MAAM,CAAC;IAC1B,QAAQ,CAAC,UAAU,EAAE,OAAO,CAAC;IAC7B,GAAG,CAAC,QAAQ,CAAC,EAAE,oBAAoB,EAAE,GAAG,OAAO,CAAC,oBAAoB,EAAE,CAAC,CAAC;IACxE,GAAG,CAAC,QAAQ,EAAE,oBAAoB,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;CACtD;AAED,MAAM,WAAW,0BAA0B;IACzC,EAAE,CAAC,EAAE,MAAM,CAAC;IACZ,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,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,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,sBAAsB,CAAC,EAAE,MAAM,CAAC;IAChC,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB,OAAO,CAAC,EAAE,WAAW,CAAC;CACvB;AAID;;;;;;;;;;GAUG;AACH,qBAAa,mBAAoB,YAAW,oBAAoB;IAC9D,QAAQ,CAAC,EAAE,EAAE,MAAM,CAAC;IACpB,QAAQ,CAAC,QAAQ,EAAE,MAAM,CAAC;IAC1B,QAAQ,CAAC,UAAU,QAAQ;IAE3B,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAoB;IAC7C,OAAO,CAAC,QAAQ,CAAC,gBAAgB,CAAS;IAC1C,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAqB;IACnD,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAqB;IACjD,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;IAEtC,OAAO,CAAC,MAAM,CAAM;gBAER,OAAO,GAAE,0BAA+B;IA4BpD,OAAO,CAAC,YAAY;IAcd,GAAG,CAAC,QAAQ,EAAE,oBAAoB,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IASpD,GAAG,CAAC,SAAS,CAAC,EAAE,oBAAoB,EAAE,GAAG,OAAO,CAAC,oBAAoB,EAAE,CAAC;IAoB9E,OAAO,CAAC,YAAY;CAMrB"}

package/dist/cjs/integrations/llamaindex/memory.js

@@ -0,0 +1,95 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.CompresrMemoryBlock = void 0;
+const index_js_1 = require("../_shared/index.js");
+const DEFAULT_PRIORITY = 2;
+/**
+ * Drop-in `BaseMemoryBlock` for LlamaIndex.TS that compresses on `get`.
+ *
+ * @example
+ * const memory = new Memory({
+ *   tokenLimit: 8_000,
+ *   memoryBlocks: [
+ *     new CompresrMemoryBlock({ apiKey: process.env.COMPRESR_API_KEY!, targetToken: 2_000 }),
+ *   ],
+ * });
+ */
+class CompresrMemoryBlock {
+    id;
+    priority;
+    isLongTerm = true;
+    compresr;
+    compressionModel;
+    explicitQuery;
+    targetToken;
+    targetCompressionRatio;
+    minTokens;
+    coarse;
+    onError;
+    buffer = '';
+    constructor(options = {}) {
+        const { id, priority = DEFAULT_PRIORITY, apiKey, client, baseUrl, compressionModel = index_js_1.DEFAULT_MODEL, query, targetToken, targetCompressionRatio = index_js_1.DEFAULT_RATIO, minTokens = 200, coarse, onError = index_js_1.DEFAULT_POLICY, } = options;
+        this.id = id ?? `compresr-${Math.random().toString(36).slice(2, 10)}`;
+        this.priority = priority;
+        this.compresr = client ?? (0, index_js_1.buildClient)({ apiKey, baseUrl, caller: 'CompresrMemoryBlock' });
+        this.compressionModel = compressionModel;
+        this.explicitQuery = query;
+        this.targetToken = targetToken;
+        this.targetCompressionRatio = targetCompressionRatio;
+        this.minTokens = Math.max(1, minTokens);
+        this.coarse = coarse;
+        this.onError = onError;
+    }
+    resolveQuery() {
+        if (this.explicitQuery)
+            return this.explicitQuery;
+        const lines = this.buffer.split('\n');
+        for (let i = lines.length - 1; i >= 0; i--) {
+            const line = lines[i];
+            if (line?.startsWith('user:')) {
+                const q = line.slice('user:'.length).trim();
+                if (q)
+                    return q;
+            }
+        }
+        return 'conversation history';
+    }
+    // eslint-disable-next-line @typescript-eslint/require-await
+    async put(messages) {
+        for (const m of messages) {
+            const content = typeof m.content === 'string' ? m.content : '';
+            if (!content)
+                continue;
+            const role = String(m.role ?? '');
+            this.buffer = `${this.buffer}\n${role}: ${content}`.trim();
+        }
+    }
+    async get(_messages) {
+        if (!this.buffer)
+            return [];
+        const tokens = (0, index_js_1.estimateTokens)(this.buffer);
+        if (tokens < this.minTokens) {
+            return [{ role: 'system', content: this.buffer }];
+        }
+        const ratio = this.computeRatio(tokens);
+        const compressed = await (0, index_js_1.compressSafe)(this.compresr, {
+            context: this.buffer,
+            query: this.resolveQuery(),
+            compressionModel: this.compressionModel,
+            targetCompressionRatio: ratio,
+            ...(this.coarse !== undefined ? { coarse: this.coarse } : {}),
+            minTokens: this.minTokens,
+            onError: this.onError,
+            contextLabel: 'memory_block',
+        });
+        return [{ role: 'system', content: compressed }];
+    }
+    computeRatio(currentTokens) {
+        if (this.targetToken && this.targetToken > 0) {
+            return Math.max(currentTokens / this.targetToken, 1.0);
+        }
+        return this.targetCompressionRatio;
+    }
+}
+exports.CompresrMemoryBlock = CompresrMemoryBlock;
+//# sourceMappingURL=memory.js.map

package/dist/cjs/integrations/llamaindex/memory.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"memory.js","sourceRoot":"","sources":["../../../../src/integrations/llamaindex/memory.ts"],"names":[],"mappings":";;;AAUA,kDAQ6B;AAoC7B,MAAM,gBAAgB,GAAG,CAAC,CAAC;AAE3B;;;;;;;;;;GAUG;AACH,MAAa,mBAAmB;IACrB,EAAE,CAAS;IACX,QAAQ,CAAS;IACjB,UAAU,GAAG,IAAI,CAAC;IAEV,QAAQ,CAAoB;IAC5B,gBAAgB,CAAS;IACzB,aAAa,CAAqB;IAClC,WAAW,CAAqB;IAChC,sBAAsB,CAAS;IAC/B,SAAS,CAAS;IAClB,MAAM,CAAsB;IAC5B,OAAO,CAAc;IAE9B,MAAM,GAAG,EAAE,CAAC;IAEpB,YAAY,UAAsC,EAAE;QAClD,MAAM,EACJ,EAAE,EACF,QAAQ,GAAG,gBAAgB,EAC3B,MAAM,EACN,MAAM,EACN,OAAO,EACP,gBAAgB,GAAG,wBAAa,EAChC,KAAK,EACL,WAAW,EACX,sBAAsB,GAAG,wBAAa,EACtC,SAAS,GAAG,GAAG,EACf,MAAM,EACN,OAAO,GAAG,yBAAc,GACzB,GAAG,OAAO,CAAC;QAEZ,IAAI,CAAC,EAAE,GAAG,EAAE,IAAI,YAAY,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC;QACtE,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;QACzB,IAAI,CAAC,QAAQ,GAAG,MAAM,IAAI,IAAA,sBAAW,EAAC,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,qBAAqB,EAAE,CAAC,CAAC;QAC1F,IAAI,CAAC,gBAAgB,GAAG,gBAAgB,CAAC;QACzC,IAAI,CAAC,aAAa,GAAG,KAAK,CAAC;QAC3B,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;QAC/B,IAAI,CAAC,sBAAsB,GAAG,sBAAsB,CAAC;QACrD,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC;QACxC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;IACzB,CAAC;IAEO,YAAY;QAClB,IAAI,IAAI,CAAC,aAAa;YAAE,OAAO,IAAI,CAAC,aAAa,CAAC;QAClD,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QACtC,KAAK,IAAI,CAAC,GAAG,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;YAC3C,MAAM,IAAI,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;YACtB,IAAI,IAAI,EAAE,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC;gBAC9B,MAAM,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,CAAC;gBAC5C,IAAI,CAAC;oBAAE,OAAO,CAAC,CAAC;YAClB,CAAC;QACH,CAAC;QACD,OAAO,sBAAsB,CAAC;IAChC,CAAC;IAED,4DAA4D;IAC5D,KAAK,CAAC,GAAG,CAAC,QAAgC;QACxC,KAAK,MAAM,CAAC,IAAI,QAAQ,EAAE,CAAC;YACzB,MAAM,OAAO,GAAG,OAAO,CAAC,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC;YAC/D,IAAI,CAAC,OAAO;gBAAE,SAAS;YACvB,MAAM,IAAI,GAAG,MAAM,CAAC,CAAC,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC;YAClC,IAAI,CAAC,MAAM,GAAG,GAAG,IAAI,CAAC,MAAM,KAAK,IAAI,KAAK,OAAO,EAAE,CAAC,IAAI,EAAE,CAAC;QAC7D,CAAC;IACH,CAAC;IAED,KAAK,CAAC,GAAG,CAAC,SAAkC;QAC1C,IAAI,CAAC,IAAI,CAAC,MAAM;YAAE,OAAO,EAAE,CAAC;QAC5B,MAAM,MAAM,GAAG,IAAA,yBAAc,EAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC3C,IAAI,MAAM,GAAG,IAAI,CAAC,SAAS,EAAE,CAAC;YAC5B,OAAO,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;QACpD,CAAC;QACD,MAAM,KAAK,GAAG,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;QACxC,MAAM,UAAU,GAAG,MAAM,IAAA,uBAAY,EAAC,IAAI,CAAC,QAAQ,EAAE;YACnD,OAAO,EAAE,IAAI,CAAC,MAAM;YACpB,KAAK,EAAE,IAAI,CAAC,YAAY,EAAE;YAC1B,gBAAgB,EAAE,IAAI,CAAC,gBAAgB;YACvC,sBAAsB,EAAE,KAAK;YAC7B,GAAG,CAAC,IAAI,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC7D,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,YAAY,EAAE,cAAc;SAC7B,CAAC,CAAC;QACH,OAAO,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC,CAAC;IACnD,CAAC;IAEO,YAAY,CAAC,aAAqB;QACxC,IAAI,IAAI,CAAC,WAAW,IAAI,IAAI,CAAC,WAAW,GAAG,CAAC,EAAE,CAAC;YAC7C,OAAO,IAAI,CAAC,GAAG,CAAC,aAAa,GAAG,IAAI,CAAC,WAAW,EAAE,GAAG,CAAC,CAAC;QACzD,CAAC;QACD,OAAO,IAAI,CAAC,sBAAsB,CAAC;IACrC,CAAC;CACF;AA7FD,kDA6FC"}

package/dist/cjs/integrations/llamaindex/postprocessor.d.ts

@@ -0,0 +1,66 @@
+/**
+ * `BaseNodePostprocessor` that compresses retrieved node content.
+ *
+ * Mirrors Python `CompresrNodePostprocessor`. Query comes from the
+ * `QueryBundle` supplied by the query engine; override via `query`.
+ */
+import type { CompressionClient } from '../../clients/compression.js';
+import { type ErrorPolicy } from '../_shared/index.js';
+import type { NodeWithScore } from '@llamaindex/core/schema';
+type NodeWithScoreType = NodeWithScore;
+/**
+ * LlamaIndex.TS query input — either the raw query string or an object
+ * carrying the query string (matching LlamaIndex.TS' `MessageContent`
+ * conventions plus Python-side `QueryBundle.query_str`).
+ */
+type QueryInput = string | {
+    queryStr?: string;
+    query_str?: string;
+} | undefined;
+export interface CompresrNodePostprocessorOptions {
+    apiKey?: string;
+    client?: CompressionClient;
+    baseUrl?: string;
+    compressionModel?: string;
+    targetCompressionRatio?: number;
+    /**
+     * Absolute output token budget per node. When set, overrides
+     * `targetCompressionRatio` with `ratio = avg_chunk_tokens / targetToken`.
+     * Approximate (chars/4 estimator).
+     */
+    targetToken?: number;
+    minTokens?: number;
+    coarse?: boolean;
+    onError?: ErrorPolicy;
+    /** Override the query supplied by the query engine. */
+    query?: string;
+}
+/**
+ * @example
+ * ```ts
+ * import { CompresrNodePostprocessor } from '@compresr/sdk/integrations/llamaindex';
+ * const pp = new CompresrNodePostprocessor({
+ *   apiKey: process.env.COMPRESR_API_KEY!,
+ *   compressionModel: 'latte_v1',
+ * });
+ * const queryEngine = index.asQueryEngine({ nodePostprocessors: [pp] });
+ * ```
+ */
+export declare class CompresrNodePostprocessor {
+    private readonly compresr;
+    private readonly compressionModel;
+    private readonly targetCompressionRatio;
+    private readonly targetToken;
+    private readonly minTokens;
+    private readonly coarse;
+    private readonly onError;
+    private readonly staticQuery;
+    constructor(options: CompresrNodePostprocessorOptions);
+    private effectiveRatio;
+    private resolveQuery;
+    private partition;
+    /** LlamaIndex.TS `BaseNodePostprocessor.postprocessNodes` signature. */
+    postprocessNodes(nodes: readonly NodeWithScoreType[], queryBundle?: QueryInput | string): Promise<NodeWithScoreType[]>;
+}
+export {};
+//# sourceMappingURL=postprocessor.d.ts.map

package/dist/cjs/integrations/llamaindex/postprocessor.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"postprocessor.d.ts","sourceRoot":"","sources":["../../../../src/integrations/llamaindex/postprocessor.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AACH,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,8BAA8B,CAAC;AAEtE,OAAO,EAOL,KAAK,WAAW,EAEjB,MAAM,qBAAqB,CAAC;AAE7B,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,yBAAyB,CAAC;AAE7D,KAAK,iBAAiB,GAAG,aAAa,CAAC;AAEvC;;;;GAIG;AACH,KAAK,UAAU,GACX,MAAM,GACN;IAAE,QAAQ,CAAC,EAAE,MAAM,CAAC;IAAC,SAAS,CAAC,EAAE,MAAM,CAAA;CAAE,GACzC,SAAS,CAAC;AAEd,MAAM,WAAW,gCAAgC;IAC/C,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;;;;OAIG;IACH,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB,OAAO,CAAC,EAAE,WAAW,CAAC;IACtB,uDAAuD;IACvD,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAED;;;;;;;;;;GAUG;AACH,qBAAa,yBAAyB;IACpC,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,WAAW,CAAqB;IACjD,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,WAAW,CAAqB;gBAErC,OAAO,EAAE,gCAAgC;IAyBrD,OAAO,CAAC,cAAc;IAStB,OAAO,CAAC,YAAY;IAgBpB,OAAO,CAAC,SAAS;IAejB,wEAAwE;IAClE,gBAAgB,CACpB,KAAK,EAAE,SAAS,iBAAiB,EAAE,EACnC,WAAW,CAAC,EAAE,UAAU,GAAG,MAAM,GAChC,OAAO,CAAC,iBAAiB,EAAE,CAAC;CAkDhC"}

package/dist/cjs/integrations/llamaindex/postprocessor.js

@@ -0,0 +1,172 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.CompresrNodePostprocessor = void 0;
+const logger_js_1 = require("../../logger.js");
+const index_js_1 = require("../_shared/index.js");
+/**
+ * @example
+ * ```ts
+ * import { CompresrNodePostprocessor } from '@compresr/sdk/integrations/llamaindex';
+ * const pp = new CompresrNodePostprocessor({
+ *   apiKey: process.env.COMPRESR_API_KEY!,
+ *   compressionModel: 'latte_v1',
+ * });
+ * const queryEngine = index.asQueryEngine({ nodePostprocessors: [pp] });
+ * ```
+ */
+class CompresrNodePostprocessor {
+    compresr;
+    compressionModel;
+    targetCompressionRatio;
+    targetToken;
+    minTokens;
+    coarse;
+    onError;
+    staticQuery;
+    constructor(options) {
+        const { apiKey, client, baseUrl, compressionModel = index_js_1.DEFAULT_MODEL, targetCompressionRatio = index_js_1.DEFAULT_RATIO, targetToken, minTokens = index_js_1.DEFAULT_MIN_TOKENS, coarse, onError = index_js_1.DEFAULT_POLICY, query, } = options;
+        this.compresr =
+            client ?? (0, index_js_1.buildClient)({ apiKey, baseUrl, caller: 'CompresrNodePostprocessor' });
+        this.compressionModel = compressionModel;
+        this.targetCompressionRatio = targetCompressionRatio;
+        this.targetToken = targetToken;
+        this.minTokens = minTokens;
+        this.coarse = coarse;
+        this.onError = onError;
+        this.staticQuery = query;
+    }
+    effectiveRatio(chunk) {
+        if (!this.targetToken || this.targetToken <= 0)
+            return this.targetCompressionRatio;
+        const avg = Math.max(1, chunk.reduce((acc, c) => acc + (0, index_js_1.estimateTokens)(c), 0) / Math.max(1, chunk.length));
+        return Math.max(avg / this.targetToken, 1.0);
+    }
+    resolveQuery(queryBundle) {
+        if (this.compressionModel !== 'latte_v1')
+            return undefined;
+        if (typeof this.staticQuery === 'string' && this.staticQuery.trim()) {
+            return this.staticQuery;
+        }
+        if (typeof queryBundle === 'string' && queryBundle.trim()) {
+            return queryBundle;
+        }
+        if (queryBundle && typeof queryBundle === 'object') {
+            const q = queryBundle.queryStr ??
+                queryBundle.query_str;
+            if (typeof q === 'string' && q.trim())
+                return q;
+        }
+        return undefined;
+    }
+    partition(nodes) {
+        const indices = [];
+        const contents = [];
+        nodes.forEach((n, i) => {
+            const text = getNodeText(n);
+            if (typeof text === 'string' && (0, index_js_1.estimateTokens)(text) >= this.minTokens) {
+                indices.push(i);
+                contents.push(text);
+            }
+        });
+        return { indices, contents };
+    }
+    /** LlamaIndex.TS `BaseNodePostprocessor.postprocessNodes` signature. */
+    async postprocessNodes(nodes, queryBundle) {
+        if (nodes.length === 0)
+            return [...nodes];
+        const { indices, contents } = this.partition(nodes);
+        if (contents.length === 0)
+            return [...nodes];
+        const query = this.resolveQuery(queryBundle);
+        if (this.compressionModel === 'latte_v1' && !query) {
+            (0, logger_js_1.getLogger)().warn('CompresrNodePostprocessor: latte_v1 needs a query; none found. Passthrough.');
+            return [...nodes];
+        }
+        const out = nodes.map(cloneNodeWithScore);
+        for (let start = 0; start < contents.length; start += index_js_1.BATCH_LIMIT) {
+            const chunk = contents.slice(start, start + index_js_1.BATCH_LIMIT);
+            let results = [];
+            try {
+                const resp = await this.compresr.compressBatch({
+                    contexts: chunk,
+                    ...(query !== undefined ? { queries: query } : {}),
+                    compressionModelName: this.compressionModel,
+                    targetCompressionRatio: this.effectiveRatio(chunk),
+                    ...(this.coarse !== undefined ? { coarse: this.coarse } : {}),
+                });
+                results = (resp.data?.results ?? []);
+            }
+            catch (exc) {
+                if (this.onError === 'raise')
+                    throw exc;
+                (0, logger_js_1.getLogger)().warn(`postprocessor batch failed (${stringifyError(exc)}); passthrough ${chunk.length} nodes.`);
+                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 node = out[globalIdx];
+                    if (node !== undefined) {
+                        setNodeText(node, newText);
+                    }
+                }
+            });
+        }
+        return out;
+    }
+}
+exports.CompresrNodePostprocessor = CompresrNodePostprocessor;
+function getNodeText(nws) {
+    const node = nws.node;
+    if (!node || typeof node !== 'object')
+        return undefined;
+    const n = node;
+    if (typeof n.getContent === 'function') {
+        try {
+            const t = n.getContent();
+            if (typeof t === 'string')
+                return t;
+        }
+        catch {
+            /* fall through */
+        }
+    }
+    if (typeof n.text === 'string')
+        return n.text;
+    return undefined;
+}
+function cloneNodeWithScore(nws) {
+    return { ...nws };
+}
+function setNodeText(nws, newText) {
+    const node = nws.node;
+    if (!node || typeof node !== 'object')
+        return;
+    const n = node;
+    if (typeof n.setContent === 'function') {
+        try {
+            n.setContent(newText);
+            return;
+        }
+        catch {
+            /* fall through */
+        }
+    }
+    try {
+        n.text = newText;
+        return;
+    }
+    catch {
+        /* fall through */
+    }
+    const md = node.metadata;
+    if (md && typeof md === 'object') {
+        md['compresr_compressed'] = newText;
+    }
+    (0, logger_js_1.getLogger)().warn('node has no writable text field; compressed text stored in metadata.compresr_compressed but the synthesizer will see the original text.');
+}
+function stringifyError(exc) {
+    return exc instanceof Error ? exc.message : String(exc);
+}
+//# sourceMappingURL=postprocessor.js.map

package/dist/cjs/integrations/llamaindex/postprocessor.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"postprocessor.js","sourceRoot":"","sources":["../../../../src/integrations/llamaindex/postprocessor.ts"],"names":[],"mappings":";;;AAOA,+CAA4C;AAC5C,kDAS6B;AAmC7B;;;;;;;;;;GAUG;AACH,MAAa,yBAAyB;IACnB,QAAQ,CAAoB;IAC5B,gBAAgB,CAAS;IACzB,sBAAsB,CAAS;IAC/B,WAAW,CAAqB;IAChC,SAAS,CAAS;IAClB,MAAM,CAAsB;IAC5B,OAAO,CAAc;IACrB,WAAW,CAAqB;IAEjD,YAAY,OAAyC;QACnD,MAAM,EACJ,MAAM,EACN,MAAM,EACN,OAAO,EACP,gBAAgB,GAAG,wBAAa,EAChC,sBAAsB,GAAG,wBAAa,EACtC,WAAW,EACX,SAAS,GAAG,6BAAkB,EAC9B,MAAM,EACN,OAAO,GAAG,yBAAc,EACxB,KAAK,GACN,GAAG,OAAO,CAAC;QAEZ,IAAI,CAAC,QAAQ;YACX,MAAM,IAAI,IAAA,sBAAW,EAAC,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,2BAA2B,EAAE,CAAC,CAAC;QAClF,IAAI,CAAC,gBAAgB,GAAG,gBAAgB,CAAC;QACzC,IAAI,CAAC,sBAAsB,GAAG,sBAAsB,CAAC;QACrD,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;QAC/B,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,WAAW,GAAG,KAAK,CAAC;IAC3B,CAAC;IAEO,cAAc,CAAC,KAAwB;QAC7C,IAAI,CAAC,IAAI,CAAC,WAAW,IAAI,IAAI,CAAC,WAAW,IAAI,CAAC;YAAE,OAAO,IAAI,CAAC,sBAAsB,CAAC;QACnF,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAClB,CAAC,EACD,KAAK,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,IAAA,yBAAc,EAAC,CAAC,CAAC,EAAE,CAAC,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,CAAC,MAAM,CAAC,CACjF,CAAC;QACF,OAAO,IAAI,CAAC,GAAG,CAAC,GAAG,GAAG,IAAI,CAAC,WAAW,EAAE,GAAG,CAAC,CAAC;IAC/C,CAAC;IAEO,YAAY,CAAC,WAAiC;QACpD,IAAI,IAAI,CAAC,gBAAgB,KAAK,UAAU;YAAE,OAAO,SAAS,CAAC;QAC3D,IAAI,OAAO,IAAI,CAAC,WAAW,KAAK,QAAQ,IAAI,IAAI,CAAC,WAAW,CAAC,IAAI,EAAE,EAAE,CAAC;YACpE,OAAO,IAAI,CAAC,WAAW,CAAC;QAC1B,CAAC;QACD,IAAI,OAAO,WAAW,KAAK,QAAQ,IAAI,WAAW,CAAC,IAAI,EAAE,EAAE,CAAC;YAC1D,OAAO,WAAW,CAAC;QACrB,CAAC;QACD,IAAI,WAAW,IAAI,OAAO,WAAW,KAAK,QAAQ,EAAE,CAAC;YACnD,MAAM,CAAC,GAAI,WAAyD,CAAC,QAAQ;gBAC1E,WAAsC,CAAC,SAAS,CAAC;YACpD,IAAI,OAAO,CAAC,KAAK,QAAQ,IAAI,CAAC,CAAC,IAAI,EAAE;gBAAE,OAAO,CAAC,CAAC;QAClD,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAEO,SAAS,CACf,KAAmC;QAEnC,MAAM,OAAO,GAAa,EAAE,CAAC;QAC7B,MAAM,QAAQ,GAAa,EAAE,CAAC;QAC9B,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE;YACrB,MAAM,IAAI,GAAG,WAAW,CAAC,CAAC,CAAC,CAAC;YAC5B,IAAI,OAAO,IAAI,KAAK,QAAQ,IAAI,IAAA,yBAAc,EAAC,IAAI,CAAC,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;gBACvE,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;gBAChB,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACtB,CAAC;QACH,CAAC,CAAC,CAAC;QACH,OAAO,EAAE,OAAO,EAAE,QAAQ,EAAE,CAAC;IAC/B,CAAC;IAED,wEAAwE;IACxE,KAAK,CAAC,gBAAgB,CACpB,KAAmC,EACnC,WAAiC;QAEjC,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,CAAC,GAAG,KAAK,CAAC,CAAC;QAE1C,MAAM,EAAE,OAAO,EAAE,QAAQ,EAAE,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QACpD,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,CAAC,GAAG,KAAK,CAAC,CAAC;QAE7C,MAAM,KAAK,GAAG,IAAI,CAAC,YAAY,CAAC,WAAW,CAAC,CAAC;QAC7C,IAAI,IAAI,CAAC,gBAAgB,KAAK,UAAU,IAAI,CAAC,KAAK,EAAE,CAAC;YACnD,IAAA,qBAAS,GAAE,CAAC,IAAI,CACd,6EAA6E,CAC9E,CAAC;YACF,OAAO,CAAC,GAAG,KAAK,CAAC,CAAC;QACpB,CAAC;QAED,MAAM,GAAG,GAAG,KAAK,CAAC,GAAG,CAAC,kBAAkB,CAAC,CAAC;QAE1C,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;oBACf,GAAG,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;oBAClD,oBAAoB,EAAE,IAAI,CAAC,gBAAgB;oBAC3C,sBAAsB,EAAE,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC;oBAClD,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,+BAA+B,cAAc,CAAC,GAAG,CAAC,kBAAkB,KAAK,CAAC,MAAM,SAAS,CAC1F,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,IAAI,GAAG,GAAG,CAAC,SAAS,CAAC,CAAC;oBAC5B,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;wBACvB,WAAW,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;oBAC7B,CAAC;gBACH,CAAC;YACH,CAAC,CAAC,CAAC;QACL,CAAC;QAED,OAAO,GAAG,CAAC;IACb,CAAC;CACF;AAjID,8DAiIC;AAGD,SAAS,WAAW,CAAC,GAAsB;IACzC,MAAM,IAAI,GAAI,GAA0B,CAAC,IAAI,CAAC;IAC9C,IAAI,CAAC,IAAI,IAAI,OAAO,IAAI,KAAK,QAAQ;QAAE,OAAO,SAAS,CAAC;IACxD,MAAM,CAAC,GAAG,IAAoD,CAAC;IAC/D,IAAI,OAAO,CAAC,CAAC,UAAU,KAAK,UAAU,EAAE,CAAC;QACvC,IAAI,CAAC;YACH,MAAM,CAAC,GAAG,CAAC,CAAC,UAAU,EAAE,CAAC;YACzB,IAAI,OAAO,CAAC,KAAK,QAAQ;gBAAE,OAAO,CAAC,CAAC;QACtC,CAAC;QAAC,MAAM,CAAC;YACP,kBAAkB;QACpB,CAAC;IACH,CAAC;IACD,IAAI,OAAO,CAAC,CAAC,IAAI,KAAK,QAAQ;QAAE,OAAO,CAAC,CAAC,IAAI,CAAC;IAC9C,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,SAAS,kBAAkB,CAAC,GAAsB;IAChD,OAAO,EAAE,GAAI,GAAc,EAAuB,CAAC;AACrD,CAAC;AAED,SAAS,WAAW,CAAC,GAAsB,EAAE,OAAe;IAC1D,MAAM,IAAI,GAAI,GAA0B,CAAC,IAAI,CAAC;IAC9C,IAAI,CAAC,IAAI,IAAI,OAAO,IAAI,KAAK,QAAQ;QAAE,OAAO;IAC9C,MAAM,CAAC,GAAG,IAA2D,CAAC;IACtE,IAAI,OAAO,CAAC,CAAC,UAAU,KAAK,UAAU,EAAE,CAAC;QACvC,IAAI,CAAC;YACH,CAAC,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;YACtB,OAAO;QACT,CAAC;QAAC,MAAM,CAAC;YACP,kBAAkB;QACpB,CAAC;IACH,CAAC;IACD,IAAI,CAAC;QACH,CAAC,CAAC,IAAI,GAAG,OAAO,CAAC;QACjB,OAAO;IACT,CAAC;IAAC,MAAM,CAAC;QACP,kBAAkB;IACpB,CAAC;IACD,MAAM,EAAE,GAAI,IAA+C,CAAC,QAAQ,CAAC;IACrE,IAAI,EAAE,IAAI,OAAO,EAAE,KAAK,QAAQ,EAAE,CAAC;QACjC,EAAE,CAAC,qBAAqB,CAAC,GAAG,OAAO,CAAC;IACtC,CAAC;IACD,IAAA,qBAAS,GAAE,CAAC,IAAI,CACd,yIAAyI,CAC1I,CAAC;AACJ,CAAC;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/llamaindex/wrappers.d.ts

@@ -0,0 +1,38 @@
+/**
+ * Wrap a LlamaIndex.TS `FunctionTool` (or any object with a `call` method)
+ * so its return value is compressed.
+ *
+ * Mirrors Python `wrap_tool_with_compresr`. Same unified query API.
+ */
+import type { CompressionClient } from '../../clients/compression.js';
+import { type ErrorPolicy } from '../_shared/index.js';
+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;
+}
+/**
+ * The narrow callable shape we wrap. `FunctionTool` in llamaindex.ts
+ * exposes `metadata` + `call(args)` / `acall(args)` — we wrap that.
+ */
+export interface CompresrWrappableTool<TArgs extends Record<string, unknown> = Record<string, unknown>> {
+    metadata: {
+        name: string;
+        description?: string;
+    };
+    call: (args: TArgs) => unknown;
+}
+/**
+ * Returns a new tool with the same `metadata` and a wrapped `call` that
+ * compresses string return values transparently.
+ */
+export declare function wrapToolWithCompresr<TArgs extends Record<string, unknown>, T extends CompresrWrappableTool<TArgs>>(tool: T, options?: WrapToolOptions): T;
+//# sourceMappingURL=wrappers.d.ts.map

package/dist/cjs/integrations/llamaindex/wrappers.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"wrappers.d.ts","sourceRoot":"","sources":["../../../../src/integrations/llamaindex/wrappers.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AACH,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,8BAA8B,CAAC;AACtE,OAAO,EAOL,KAAK,WAAW,EAEjB,MAAM,qBAAqB,CAAC;AAE7B,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;AAED;;;GAGG;AACH,MAAM,WAAW,qBAAqB,CAAC,KAAK,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;IACpG,QAAQ,EAAE;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,WAAW,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC;IACjD,IAAI,EAAE,CAAC,IAAI,EAAE,KAAK,KAAK,OAAO,CAAC;CAChC;AAED;;;GAGG;AACH,wBAAgB,oBAAoB,CAClC,KAAK,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EACrC,CAAC,SAAS,qBAAqB,CAAC,KAAK,CAAC,EACtC,IAAI,EAAE,CAAC,EAAE,OAAO,GAAE,eAAoB,GAAG,CAAC,CAqD3C"}