@fllf/agent-sdk 0.1.0

package/dist/rag/retrieval/fusion.js

@@ -0,0 +1,54 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.reciprocalRankFusion = reciprocalRankFusion;
+const utils_1 = require("./utils");
+/**
+ * RRF 只依赖各检索器内部排序，不要求 dense 分数和 sparse 分数处在同一尺度。
+ * 这使它适合作为默认 hybrid 检索融合策略。
+ */
+function reciprocalRankFusion(lists, options = {}) {
+    const k = options.k ?? 60;
+    const topK = (0, utils_1.resolveTopK)(options.topK);
+    if (topK === 0) {
+        return [];
+    }
+    const entries = new Map();
+    for (const list of lists) {
+        for (const [rankIndex, item] of list.entries()) {
+            const rank = rankIndex + 1;
+            const weight = options.weights?.[item.source] ?? 1;
+            const contribution = weight / (k + rank);
+            const existing = entries.get(item.chunk.id);
+            if (existing === undefined) {
+                entries.set(item.chunk.id, {
+                    item: (0, utils_1.cloneRetrievedChunk)(item),
+                    score: contribution,
+                    sources: new Set([item.source]),
+                    sourceScores: {
+                        [item.source]: item.score,
+                    },
+                });
+                continue;
+            }
+            existing.score += contribution;
+            existing.sources.add(item.source);
+            existing.sourceScores[item.source] = Math.max(existing.sourceScores[item.source] ?? Number.NEGATIVE_INFINITY, item.score);
+        }
+    }
+    return [...entries.values()]
+        .sort((left, right) => (right.score - left.score
+        || left.item.chunk.documentId.localeCompare(right.item.chunk.documentId)
+        || left.item.chunk.id.localeCompare(right.item.chunk.id)))
+        .slice(0, topK)
+        .map((entry) => ({
+        chunk: entry.item.chunk,
+        score: entry.score,
+        source: 'hybrid',
+        metadata: {
+            sources: [...entry.sources],
+            sourceScores: entry.sourceScores,
+            rrfScore: entry.score,
+        },
+    }));
+}
+//# sourceMappingURL=fusion.js.map

package/dist/rag/retrieval/fusion.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"fusion.js","sourceRoot":"","sources":["../../../src/rag/retrieval/fusion.ts"],"names":[],"mappings":";;AA0BA,oDAyDC;AA/ED,mCAGiB;AAejB;;;GAGG;AACH,SAAgB,oBAAoB,CAChC,KAAyB,EACzB,UAAuC,EAAE;IAEzC,MAAM,CAAC,GAAG,OAAO,CAAC,CAAC,IAAI,EAAE,CAAC;IAC1B,MAAM,IAAI,GAAG,IAAA,mBAAW,EAAC,OAAO,CAAC,IAAI,CAAC,CAAC;IACvC,IAAI,IAAI,KAAK,CAAC,EAAE,CAAC;QACb,OAAO,EAAE,CAAC;IACd,CAAC;IAED,MAAM,OAAO,GAAG,IAAI,GAAG,EAAuB,CAAC;IAE/C,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACvB,KAAK,MAAM,CAAC,SAAS,EAAE,IAAI,CAAC,IAAI,IAAI,CAAC,OAAO,EAAE,EAAE,CAAC;YAC7C,MAAM,IAAI,GAAG,SAAS,GAAG,CAAC,CAAC;YAC3B,MAAM,MAAM,GAAG,OAAO,CAAC,OAAO,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YACnD,MAAM,YAAY,GAAG,MAAM,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC;YACzC,MAAM,QAAQ,GAAG,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;YAE5C,IAAI,QAAQ,KAAK,SAAS,EAAE,CAAC;gBACzB,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,EAAE;oBACvB,IAAI,EAAE,IAAA,2BAAmB,EAAC,IAAI,CAAC;oBAC/B,KAAK,EAAE,YAAY;oBACnB,OAAO,EAAE,IAAI,GAAG,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;oBAC/B,YAAY,EAAE;wBACV,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,IAAI,CAAC,KAAK;qBAC5B;iBACJ,CAAC,CAAC;gBACH,SAAS;YACb,CAAC;YAED,QAAQ,CAAC,KAAK,IAAI,YAAY,CAAC;YAC/B,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAClC,QAAQ,CAAC,YAAY,CAAC,IAAI,CAAC,MAAM,CAAC,GAAG,IAAI,CAAC,GAAG,CACzC,QAAQ,CAAC,YAAY,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,MAAM,CAAC,iBAAiB,EAC9D,IAAI,CAAC,KAAK,CACb,CAAC;QACN,CAAC;IACL,CAAC;IAED,OAAO,CAAC,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;SACvB,IAAI,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC,CACnB,KAAK,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK;WACrB,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,aAAa,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC;WACrE,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,aAAa,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,CAC3D,CAAC;SACD,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC;SACd,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QACb,KAAK,EAAE,KAAK,CAAC,IAAI,CAAC,KAAK;QACvB,KAAK,EAAE,KAAK,CAAC,KAAK;QAClB,MAAM,EAAE,QAAQ;QAChB,QAAQ,EAAE;YACN,OAAO,EAAE,CAAC,GAAG,KAAK,CAAC,OAAO,CAAC;YAC3B,YAAY,EAAE,KAAK,CAAC,YAAY;YAChC,QAAQ,EAAE,KAAK,CAAC,KAAK;SACxB;KACJ,CAAC,CAAC,CAAC;AACZ,CAAC"}

package/dist/rag/retrieval/http-rerank-model-client.d.ts

@@ -0,0 +1,41 @@
+import type { RerankModelClient, RerankModelInput, RerankModelResult } from './reranker';
+export interface HttpRerankModelClientOptions {
+    baseURL?: string;
+    baseUrl?: string;
+    endpoint?: string;
+    apiKey?: string;
+    model: string;
+    timeoutMs?: number;
+    headers?: Record<string, string>;
+    fetch?: FetchLike;
+}
+type FetchLike = (input: string, init?: {
+    method?: string;
+    headers?: Record<string, string>;
+    body?: string;
+    signal?: AbortSignal;
+}) => Promise<HttpResponseLike>;
+interface HttpResponseLike {
+    ok: boolean;
+    status: number;
+    statusText: string;
+    text(): Promise<string>;
+}
+/**
+ * 通用 HTTP rerank client，适配 Jina/Cohere 以及常见 OpenAI-style 网关的 /rerank 端点。
+ * 请求体使用 { model, query, documents, top_n }，响应读取 results[].index 与
+ * results[].relevance_score/score。
+ */
+export declare class HttpRerankModelClient implements RerankModelClient {
+    private readonly url;
+    private readonly apiKey;
+    private readonly model;
+    private readonly timeoutMs;
+    private readonly headers;
+    private readonly fetchImpl;
+    constructor(options: HttpRerankModelClientOptions);
+    rerank(input: RerankModelInput): Promise<RerankModelResult[]>;
+    private createHeaders;
+}
+export {};
+//# sourceMappingURL=http-rerank-model-client.d.ts.map

package/dist/rag/retrieval/http-rerank-model-client.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"http-rerank-model-client.d.ts","sourceRoot":"","sources":["../../../src/rag/retrieval/http-rerank-model-client.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACR,iBAAiB,EACjB,gBAAgB,EAChB,iBAAiB,EACpB,MAAM,YAAY,CAAC;AAEpB,MAAM,WAAW,4BAA4B;IACzC,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,KAAK,EAAE,MAAM,CAAC;IACd,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACjC,KAAK,CAAC,EAAE,SAAS,CAAC;CACrB;AAED,KAAK,SAAS,GAAG,CACb,KAAK,EAAE,MAAM,EACb,IAAI,CAAC,EAAE;IACH,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACjC,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,MAAM,CAAC,EAAE,WAAW,CAAC;CACxB,KACA,OAAO,CAAC,gBAAgB,CAAC,CAAC;AAE/B,UAAU,gBAAgB;IACtB,EAAE,EAAE,OAAO,CAAC;IACZ,MAAM,EAAE,MAAM,CAAC;IACf,UAAU,EAAE,MAAM,CAAC;IACnB,IAAI,IAAI,OAAO,CAAC,MAAM,CAAC,CAAC;CAC3B;AAED;;;;GAIG;AACH,qBAAa,qBAAsB,YAAW,iBAAiB;IAC3D,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAS;IAC7B,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAqB;IAC5C,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAS;IAC/B,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAS;IACnC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAyB;IACjD,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAY;gBAE1B,OAAO,EAAE,4BAA4B;IAc3C,MAAM,CAAC,KAAK,EAAE,gBAAgB,GAAG,OAAO,CAAC,iBAAiB,EAAE,CAAC;IA2CnE,OAAO,CAAC,aAAa;CAexB"}

package/dist/rag/retrieval/http-rerank-model-client.js

@@ -0,0 +1,130 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.HttpRerankModelClient = void 0;
+/**
+ * 通用 HTTP rerank client，适配 Jina/Cohere 以及常见 OpenAI-style 网关的 /rerank 端点。
+ * 请求体使用 { model, query, documents, top_n }，响应读取 results[].index 与
+ * results[].relevance_score/score。
+ */
+class HttpRerankModelClient {
+    url;
+    apiKey;
+    model;
+    timeoutMs;
+    headers;
+    fetchImpl;
+    constructor(options) {
+        const baseURL = (options.baseURL ?? options.baseUrl)?.trim();
+        if (baseURL === undefined || baseURL.length === 0) {
+            throw new Error('HTTP rerank model client requires a baseURL.');
+        }
+        this.url = joinURL(baseURL, options.endpoint ?? '/rerank');
+        this.apiKey = options.apiKey?.trim();
+        this.model = options.model;
+        this.timeoutMs = options.timeoutMs ?? 30000;
+        this.headers = { ...(options.headers ?? {}) };
+        this.fetchImpl = options.fetch ?? getGlobalFetch();
+    }
+    async rerank(input) {
+        const controller = this.timeoutMs > 0 ? new AbortController() : undefined;
+        const timeoutId = controller === undefined
+            ? undefined
+            : setTimeout(() => controller.abort(), this.timeoutMs);
+        try {
+            const request = {
+                method: 'POST',
+                headers: this.createHeaders(),
+                body: JSON.stringify({
+                    model: this.model,
+                    query: input.query,
+                    documents: input.documents,
+                    top_n: input.topK,
+                }),
+            };
+            if (controller !== undefined) {
+                request.signal = controller.signal;
+            }
+            const response = await this.fetchImpl(this.url, request);
+            const text = await response.text();
+            if (!response.ok) {
+                throw new Error(`Rerank API failed: ${response.status} ${response.statusText}${formatBody(text)}`);
+            }
+            return parseRerankResponse(text);
+        }
+        finally {
+            if (timeoutId !== undefined) {
+                clearTimeout(timeoutId);
+            }
+        }
+    }
+    createHeaders() {
+        const headers = {
+            'content-type': 'application/json',
+            ...this.headers,
+        };
+        const hasAuthorization = Object.keys(headers).some((key) => key.toLowerCase() === 'authorization');
+        if (this.apiKey !== undefined && this.apiKey.length > 0 && !hasAuthorization) {
+            headers.authorization = `Bearer ${this.apiKey}`;
+        }
+        return headers;
+    }
+}
+exports.HttpRerankModelClient = HttpRerankModelClient;
+function joinURL(baseURL, endpoint) {
+    return `${baseURL.replace(/\/+$/, '')}/${endpoint.replace(/^\/+/, '')}`;
+}
+function getGlobalFetch() {
+    if (typeof fetch !== 'function') {
+        throw new Error('HTTP rerank model client requires fetch support or a custom fetch option.');
+    }
+    return fetch;
+}
+function parseRerankResponse(text) {
+    const data = readJsonObject(text);
+    const rawResults = readResultsArray(data);
+    return rawResults.map((item) => {
+        const index = readInteger(item.index, 'rerank result index');
+        const score = readNumber(item.relevance_score ?? item.score, `rerank result score at index ${index}`);
+        return {
+            index,
+            score,
+            metadata: { ...item },
+        };
+    });
+}
+function readJsonObject(text) {
+    const parsed = JSON.parse(text);
+    if (!isRecord(parsed)) {
+        throw new TypeError('Rerank API response must be a JSON object.');
+    }
+    return parsed;
+}
+function readResultsArray(data) {
+    const rawResults = data.results ?? data.data;
+    if (!Array.isArray(rawResults) || !rawResults.every(isRecord)) {
+        throw new TypeError('Rerank API response must contain a results array.');
+    }
+    return rawResults;
+}
+function readInteger(value, label) {
+    if (typeof value !== 'number' || !Number.isInteger(value)) {
+        throw new TypeError(`${label} must be an integer.`);
+    }
+    return value;
+}
+function readNumber(value, label) {
+    if (typeof value !== 'number' || !Number.isFinite(value)) {
+        throw new TypeError(`${label} must be a finite number.`);
+    }
+    return value;
+}
+function isRecord(value) {
+    return typeof value === 'object' && value !== null && !Array.isArray(value);
+}
+function formatBody(text) {
+    if (text.length === 0) {
+        return '';
+    }
+    return `: ${text}`;
+}
+//# sourceMappingURL=http-rerank-model-client.js.map

package/dist/rag/retrieval/http-rerank-model-client.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"http-rerank-model-client.js","sourceRoot":"","sources":["../../../src/rag/retrieval/http-rerank-model-client.ts"],"names":[],"mappings":";;;AAkCA;;;;GAIG;AACH,MAAa,qBAAqB;IACb,GAAG,CAAS;IACZ,MAAM,CAAqB;IAC3B,KAAK,CAAS;IACd,SAAS,CAAS;IAClB,OAAO,CAAyB;IAChC,SAAS,CAAY;IAEtC,YAAY,OAAqC;QAC7C,MAAM,OAAO,GAAG,CAAC,OAAO,CAAC,OAAO,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,CAAC;QAC7D,IAAI,OAAO,KAAK,SAAS,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAChD,MAAM,IAAI,KAAK,CAAC,8CAA8C,CAAC,CAAC;QACpE,CAAC;QAED,IAAI,CAAC,GAAG,GAAG,OAAO,CAAC,OAAO,EAAE,OAAO,CAAC,QAAQ,IAAI,SAAS,CAAC,CAAC;QAC3D,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC;QACrC,IAAI,CAAC,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC;QAC3B,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,SAAS,IAAI,KAAK,CAAC;QAC5C,IAAI,CAAC,OAAO,GAAG,EAAE,GAAG,CAAC,OAAO,CAAC,OAAO,IAAI,EAAE,CAAC,EAAE,CAAC;QAC9C,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,KAAK,IAAI,cAAc,EAAE,CAAC;IACvD,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,KAAuB;QAChC,MAAM,UAAU,GAAG,IAAI,CAAC,SAAS,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,eAAe,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;QAC1E,MAAM,SAAS,GAAG,UAAU,KAAK,SAAS;YACtC,CAAC,CAAC,SAAS;YACX,CAAC,CAAC,UAAU,CAAC,GAAG,EAAE,CAAC,UAAU,CAAC,KAAK,EAAE,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;QAE3D,IAAI,CAAC;YACD,MAAM,OAAO,GAKT;gBACA,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE,IAAI,CAAC,aAAa,EAAE;gBAC7B,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oBACjB,KAAK,EAAE,IAAI,CAAC,KAAK;oBACjB,KAAK,EAAE,KAAK,CAAC,KAAK;oBAClB,SAAS,EAAE,KAAK,CAAC,SAAS;oBAC1B,KAAK,EAAE,KAAK,CAAC,IAAI;iBACpB,CAAC;aACL,CAAC;YACF,IAAI,UAAU,KAAK,SAAS,EAAE,CAAC;gBAC3B,OAAO,CAAC,MAAM,GAAG,UAAU,CAAC,MAAM,CAAC;YACvC,CAAC;YAED,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;YAEzD,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;YACnC,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACf,MAAM,IAAI,KAAK,CACX,sBAAsB,QAAQ,CAAC,MAAM,IAAI,QAAQ,CAAC,UAAU,GAAG,UAAU,CAAC,IAAI,CAAC,EAAE,CACpF,CAAC;YACN,CAAC;YAED,OAAO,mBAAmB,CAAC,IAAI,CAAC,CAAC;QACrC,CAAC;gBAAS,CAAC;YACP,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;gBAC1B,YAAY,CAAC,SAAS,CAAC,CAAC;YAC5B,CAAC;QACL,CAAC;IACL,CAAC;IAEO,aAAa;QACjB,MAAM,OAAO,GAA2B;YACpC,cAAc,EAAE,kBAAkB;YAClC,GAAG,IAAI,CAAC,OAAO;SAClB,CAAC;QAEF,MAAM,gBAAgB,GAAG,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,IAAI,CAC9C,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,WAAW,EAAE,KAAK,eAAe,CACjD,CAAC;QACF,IAAI,IAAI,CAAC,MAAM,KAAK,SAAS,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,IAAI,CAAC,gBAAgB,EAAE,CAAC;YAC3E,OAAO,CAAC,aAAa,GAAG,UAAU,IAAI,CAAC,MAAM,EAAE,CAAC;QACpD,CAAC;QAED,OAAO,OAAO,CAAC;IACnB,CAAC;CACJ;AAhFD,sDAgFC;AAED,SAAS,OAAO,CAAC,OAAe,EAAE,QAAgB;IAC9C,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,IAAI,QAAQ,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,EAAE,CAAC;AAC5E,CAAC;AAED,SAAS,cAAc;IACnB,IAAI,OAAO,KAAK,KAAK,UAAU,EAAE,CAAC;QAC9B,MAAM,IAAI,KAAK,CAAC,2EAA2E,CAAC,CAAC;IACjG,CAAC;IAED,OAAO,KAAK,CAAC;AACjB,CAAC;AAED,SAAS,mBAAmB,CAAC,IAAY;IACrC,MAAM,IAAI,GAAG,cAAc,CAAC,IAAI,CAAC,CAAC;IAClC,MAAM,UAAU,GAAG,gBAAgB,CAAC,IAAI,CAAC,CAAC;IAE1C,OAAO,UAAU,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;QAC3B,MAAM,KAAK,GAAG,WAAW,CAAC,IAAI,CAAC,KAAK,EAAE,qBAAqB,CAAC,CAAC;QAC7D,MAAM,KAAK,GAAG,UAAU,CACpB,IAAI,CAAC,eAAe,IAAI,IAAI,CAAC,KAAK,EAClC,gCAAgC,KAAK,EAAE,CAC1C,CAAC;QAEF,OAAO;YACH,KAAK;YACL,KAAK;YACL,QAAQ,EAAE,EAAE,GAAG,IAAI,EAAE;SACxB,CAAC;IACN,CAAC,CAAC,CAAC;AACP,CAAC;AAED,SAAS,cAAc,CAAC,IAAY;IAChC,MAAM,MAAM,GAAY,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IACzC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;QACpB,MAAM,IAAI,SAAS,CAAC,4CAA4C,CAAC,CAAC;IACtE,CAAC;IAED,OAAO,MAAM,CAAC;AAClB,CAAC;AAED,SAAS,gBAAgB,CAAC,IAA6B;IACnD,MAAM,UAAU,GAAG,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,IAAI,CAAC;IAC7C,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,QAAQ,CAAC,EAAE,CAAC;QAC5D,MAAM,IAAI,SAAS,CAAC,mDAAmD,CAAC,CAAC;IAC7E,CAAC;IAED,OAAO,UAAU,CAAC;AACtB,CAAC;AAED,SAAS,WAAW,CAAC,KAAc,EAAE,KAAa;IAC9C,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC,EAAE,CAAC;QACxD,MAAM,IAAI,SAAS,CAAC,GAAG,KAAK,sBAAsB,CAAC,CAAC;IACxD,CAAC;IAED,OAAO,KAAK,CAAC;AACjB,CAAC;AAED,SAAS,UAAU,CAAC,KAAc,EAAE,KAAa;IAC7C,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;QACvD,MAAM,IAAI,SAAS,CAAC,GAAG,KAAK,2BAA2B,CAAC,CAAC;IAC7D,CAAC;IAED,OAAO,KAAK,CAAC;AACjB,CAAC;AAED,SAAS,QAAQ,CAAC,KAAc;IAC5B,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;AAChF,CAAC;AAED,SAAS,UAAU,CAAC,IAAY;IAC5B,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACpB,OAAO,EAAE,CAAC;IACd,CAAC;IAED,OAAO,KAAK,IAAI,EAAE,CAAC;AACvB,CAAC"}

package/dist/rag/retrieval/hybrid-retriever.d.ts

@@ -0,0 +1,22 @@
+import type { RetrievedChunk, RetrieveOptions } from '../types';
+import type { ReciprocalRankFusionOptions } from './fusion';
+import type { Reranker, Retriever } from './types';
+export interface HybridRetrieverOptions {
+    denseRetriever: Retriever;
+    sparseRetriever: Retriever;
+    reranker?: Reranker;
+    branchTopKMultiplier?: number;
+    rerankCandidateMultiplier?: number;
+    fusion?: ReciprocalRankFusionOptions;
+}
+export declare class HybridRetriever implements Retriever {
+    private readonly denseRetriever;
+    private readonly sparseRetriever;
+    private readonly reranker;
+    private readonly branchTopKMultiplier;
+    private readonly rerankCandidateMultiplier;
+    private readonly fusionOptions;
+    constructor(options: HybridRetrieverOptions);
+    retrieve(query: string, options?: RetrieveOptions): Promise<RetrievedChunk[]>;
+}
+//# sourceMappingURL=hybrid-retriever.d.ts.map

package/dist/rag/retrieval/hybrid-retriever.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"hybrid-retriever.d.ts","sourceRoot":"","sources":["../../../src/rag/retrieval/hybrid-retriever.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACR,cAAc,EACd,eAAe,EAClB,MAAM,UAAU,CAAC;AAIlB,OAAO,KAAK,EACR,2BAA2B,EAC9B,MAAM,UAAU,CAAC;AAClB,OAAO,KAAK,EACR,QAAQ,EACR,SAAS,EACZ,MAAM,SAAS,CAAC;AASjB,MAAM,WAAW,sBAAsB;IACnC,cAAc,EAAE,SAAS,CAAC;IAC1B,eAAe,EAAE,SAAS,CAAC;IAC3B,QAAQ,CAAC,EAAE,QAAQ,CAAC;IACpB,oBAAoB,CAAC,EAAE,MAAM,CAAC;IAC9B,yBAAyB,CAAC,EAAE,MAAM,CAAC;IACnC,MAAM,CAAC,EAAE,2BAA2B,CAAC;CACxC;AAED,qBAAa,eAAgB,YAAW,SAAS;IAC7C,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAY;IAC3C,OAAO,CAAC,QAAQ,CAAC,eAAe,CAAY;IAC5C,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAW;IACpC,OAAO,CAAC,QAAQ,CAAC,oBAAoB,CAAS;IAC9C,OAAO,CAAC,QAAQ,CAAC,yBAAyB,CAAS;IACnD,OAAO,CAAC,QAAQ,CAAC,aAAa,CAA8B;gBAEhD,OAAO,EAAE,sBAAsB;IAWrC,QAAQ,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,eAAe,GAAG,OAAO,CAAC,cAAc,EAAE,CAAC;CA4BtF"}

package/dist/rag/retrieval/hybrid-retriever.js

@@ -0,0 +1,49 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.HybridRetriever = void 0;
+const fusion_1 = require("./fusion");
+const reranker_1 = require("./reranker");
+const utils_1 = require("./utils");
+class HybridRetriever {
+    denseRetriever;
+    sparseRetriever;
+    reranker;
+    branchTopKMultiplier;
+    rerankCandidateMultiplier;
+    fusionOptions;
+    constructor(options) {
+        this.denseRetriever = options.denseRetriever;
+        this.sparseRetriever = options.sparseRetriever;
+        this.reranker = options.reranker ?? new reranker_1.DefaultReranker();
+        this.branchTopKMultiplier = normalizeMultiplier(options.branchTopKMultiplier ?? 2);
+        this.rerankCandidateMultiplier = normalizeMultiplier(options.rerankCandidateMultiplier ?? (options.reranker === undefined ? 1 : 5));
+        this.fusionOptions = { ...options.fusion };
+    }
+    async retrieve(query, options) {
+        const topK = (0, utils_1.resolveTopK)(options?.topK);
+        if (topK === 0) {
+            return [];
+        }
+        const rerankCandidateTopK = Math.max(topK, Math.ceil(topK * this.rerankCandidateMultiplier));
+        const branchTopK = Math.max(rerankCandidateTopK, Math.ceil(topK * this.branchTopKMultiplier));
+        const branchOptions = (0, utils_1.withTopK)(options, branchTopK);
+        const [dense, sparse] = await Promise.all([
+            this.denseRetriever.retrieve(query, branchOptions),
+            this.sparseRetriever.retrieve(query, branchOptions),
+        ]);
+        const fused = (0, fusion_1.reciprocalRankFusion)([dense, sparse], {
+            ...this.fusionOptions,
+            topK: rerankCandidateTopK,
+        });
+        const reranked = await this.reranker.rerank(query, fused, options);
+        return reranked.slice(0, topK);
+    }
+}
+exports.HybridRetriever = HybridRetriever;
+function normalizeMultiplier(value) {
+    if (!Number.isFinite(value) || value <= 0) {
+        return 1;
+    }
+    return value;
+}
+//# sourceMappingURL=hybrid-retriever.js.map

package/dist/rag/retrieval/hybrid-retriever.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"hybrid-retriever.js","sourceRoot":"","sources":["../../../src/rag/retrieval/hybrid-retriever.ts"],"names":[],"mappings":";;;AAIA,qCAEkB;AAQlB,yCAEoB;AACpB,mCAGiB;AAWjB,MAAa,eAAe;IACP,cAAc,CAAY;IAC1B,eAAe,CAAY;IAC3B,QAAQ,CAAW;IACnB,oBAAoB,CAAS;IAC7B,yBAAyB,CAAS;IAClC,aAAa,CAA8B;IAE5D,YAAY,OAA+B;QACvC,IAAI,CAAC,cAAc,GAAG,OAAO,CAAC,cAAc,CAAC;QAC7C,IAAI,CAAC,eAAe,GAAG,OAAO,CAAC,eAAe,CAAC;QAC/C,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC,QAAQ,IAAI,IAAI,0BAAe,EAAE,CAAC;QAC1D,IAAI,CAAC,oBAAoB,GAAG,mBAAmB,CAAC,OAAO,CAAC,oBAAoB,IAAI,CAAC,CAAC,CAAC;QACnF,IAAI,CAAC,yBAAyB,GAAG,mBAAmB,CAChD,OAAO,CAAC,yBAAyB,IAAI,CAAC,OAAO,CAAC,QAAQ,KAAK,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAChF,CAAC;QACF,IAAI,CAAC,aAAa,GAAG,EAAE,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;IAC/C,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,KAAa,EAAE,OAAyB;QACnD,MAAM,IAAI,GAAG,IAAA,mBAAW,EAAC,OAAO,EAAE,IAAI,CAAC,CAAC;QACxC,IAAI,IAAI,KAAK,CAAC,EAAE,CAAC;YACb,OAAO,EAAE,CAAC;QACd,CAAC;QAED,MAAM,mBAAmB,GAAG,IAAI,CAAC,GAAG,CAChC,IAAI,EACJ,IAAI,CAAC,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC,yBAAyB,CAAC,CACnD,CAAC;QACF,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,CACvB,mBAAmB,EACnB,IAAI,CAAC,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC,oBAAoB,CAAC,CAC9C,CAAC;QACF,MAAM,aAAa,GAAG,IAAA,gBAAQ,EAAC,OAAO,EAAE,UAAU,CAAC,CAAC;QACpD,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;YACtC,IAAI,CAAC,cAAc,CAAC,QAAQ,CAAC,KAAK,EAAE,aAAa,CAAC;YAClD,IAAI,CAAC,eAAe,CAAC,QAAQ,CAAC,KAAK,EAAE,aAAa,CAAC;SACtD,CAAC,CAAC;QACH,MAAM,KAAK,GAAG,IAAA,6BAAoB,EAAC,CAAC,KAAK,EAAE,MAAM,CAAC,EAAE;YAChD,GAAG,IAAI,CAAC,aAAa;YACrB,IAAI,EAAE,mBAAmB;SAC5B,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,KAAK,EAAE,KAAK,EAAE,OAAO,CAAC,CAAC;QAEnE,OAAO,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC;IACnC,CAAC;CACJ;AA/CD,0CA+CC;AAED,SAAS,mBAAmB,CAAC,KAAa;IACtC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,KAAK,IAAI,CAAC,EAAE,CAAC;QACxC,OAAO,CAAC,CAAC;IACb,CAAC;IAED,OAAO,KAAK,CAAC;AACjB,CAAC"}

package/dist/rag/retrieval/index.d.ts

@@ -0,0 +1,8 @@
+export * from './types';
+export * from './dense-retriever';
+export * from './fusion';
+export * from './hybrid-retriever';
+export * from './http-rerank-model-client';
+export * from './reranker';
+export * from './sparse-retriever';
+//# sourceMappingURL=index.d.ts.map

package/dist/rag/retrieval/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/rag/retrieval/index.ts"],"names":[],"mappings":"AAAA,cAAc,SAAS,CAAC;AACxB,cAAc,mBAAmB,CAAC;AAClC,cAAc,UAAU,CAAC;AACzB,cAAc,oBAAoB,CAAC;AACnC,cAAc,4BAA4B,CAAC;AAC3C,cAAc,YAAY,CAAC;AAC3B,cAAc,oBAAoB,CAAC"}

package/dist/rag/retrieval/index.js

@@ -0,0 +1,24 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+__exportStar(require("./types"), exports);
+__exportStar(require("./dense-retriever"), exports);
+__exportStar(require("./fusion"), exports);
+__exportStar(require("./hybrid-retriever"), exports);
+__exportStar(require("./http-rerank-model-client"), exports);
+__exportStar(require("./reranker"), exports);
+__exportStar(require("./sparse-retriever"), exports);
+//# sourceMappingURL=index.js.map

package/dist/rag/retrieval/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/rag/retrieval/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,0CAAwB;AACxB,oDAAkC;AAClC,2CAAyB;AACzB,qDAAmC;AACnC,6DAA2C;AAC3C,6CAA2B;AAC3B,qDAAmC"}

package/dist/rag/retrieval/reranker.d.ts

@@ -0,0 +1,30 @@
+import type { RetrievedChunk, RetrieveOptions } from '../types';
+import type { Reranker } from './types';
+export interface RerankModelInput {
+    query: string;
+    documents: string[];
+    chunks: RetrievedChunk[];
+    topK: number;
+    options?: RetrieveOptions;
+}
+export interface RerankModelResult {
+    index: number;
+    score: number;
+    metadata?: Record<string, unknown>;
+}
+export interface RerankModelClient {
+    rerank(input: RerankModelInput): Promise<RerankModelResult[]>;
+}
+export interface DefaultRerankerOptions {
+    model?: RerankModelClient;
+}
+/**
+ * DefaultReranker 是 rerank 模型 API 的适配层。
+ * 未配置 model 时保持 no-op，配置后按模型返回的相关性分数重排候选 chunk。
+ */
+export declare class DefaultReranker implements Reranker {
+    private readonly model;
+    constructor(options?: DefaultRerankerOptions);
+    rerank(query: string, chunks: RetrievedChunk[], options?: RetrieveOptions): Promise<RetrievedChunk[]>;
+}
+//# sourceMappingURL=reranker.d.ts.map

package/dist/rag/retrieval/reranker.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"reranker.d.ts","sourceRoot":"","sources":["../../../src/rag/retrieval/reranker.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACR,cAAc,EACd,eAAe,EAClB,MAAM,UAAU,CAAC;AAClB,OAAO,KAAK,EACR,QAAQ,EACX,MAAM,SAAS,CAAC;AAMjB,MAAM,WAAW,gBAAgB;IAC7B,KAAK,EAAE,MAAM,CAAC;IACd,SAAS,EAAE,MAAM,EAAE,CAAC;IACpB,MAAM,EAAE,cAAc,EAAE,CAAC;IACzB,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,CAAC,EAAE,eAAe,CAAC;CAC7B;AAED,MAAM,WAAW,iBAAiB;IAC9B,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACtC;AAED,MAAM,WAAW,iBAAiB;IAC9B,MAAM,CAAC,KAAK,EAAE,gBAAgB,GAAG,OAAO,CAAC,iBAAiB,EAAE,CAAC,CAAC;CACjE;AAED,MAAM,WAAW,sBAAsB;IACnC,KAAK,CAAC,EAAE,iBAAiB,CAAC;CAC7B;AAED;;;GAGG;AACH,qBAAa,eAAgB,YAAW,QAAQ;IAC5C,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAgC;gBAE1C,OAAO,GAAE,sBAA2B;IAI1C,MAAM,CACR,KAAK,EAAE,MAAM,EACb,MAAM,EAAE,cAAc,EAAE,EACxB,OAAO,CAAC,EAAE,eAAe,GAC1B,OAAO,CAAC,cAAc,EAAE,CAAC;CA2E/B"}

package/dist/rag/retrieval/reranker.js

@@ -0,0 +1,85 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.DefaultReranker = void 0;
+const utils_1 = require("./utils");
+/**
+ * DefaultReranker 是 rerank 模型 API 的适配层。
+ * 未配置 model 时保持 no-op，配置后按模型返回的相关性分数重排候选 chunk。
+ */
+class DefaultReranker {
+    model;
+    constructor(options = {}) {
+        this.model = options.model;
+    }
+    async rerank(query, chunks, options) {
+        if (chunks.length === 0) {
+            return [];
+        }
+        if (this.model === undefined) {
+            return chunks;
+        }
+        const topK = Math.min(chunks.length, (0, utils_1.resolveTopK)(options?.topK, chunks.length));
+        if (topK === 0) {
+            return [];
+        }
+        const input = {
+            query,
+            documents: chunks.map(toRerankDocument),
+            chunks: chunks.map(utils_1.cloneRetrievedChunk),
+            topK,
+        };
+        if (options !== undefined) {
+            input.options = options;
+        }
+        const modelResults = await this.model.rerank(input);
+        const validResults = modelResults
+            .map((result, modelRank) => ({ result, modelRank }))
+            .filter(({ result }) => (Number.isInteger(result.index)
+            && result.index >= 0
+            && result.index < chunks.length
+            && Number.isFinite(result.score)))
+            .sort((left, right) => (right.result.score - left.result.score
+            || left.modelRank - right.modelRank));
+        const usedIndexes = new Set();
+        const reranked = [];
+        for (const { result } of validResults) {
+            if (usedIndexes.has(result.index)) {
+                continue;
+            }
+            const item = chunks[result.index];
+            if (item === undefined) {
+                continue;
+            }
+            usedIndexes.add(result.index);
+            const cloned = (0, utils_1.cloneRetrievedChunk)(item);
+            reranked.push({
+                chunk: cloned.chunk,
+                score: result.score,
+                source: 'rerank',
+                metadata: {
+                    ...cloned.metadata,
+                    ...(result.metadata ?? {}),
+                    previousScore: item.score,
+                    previousSource: item.source,
+                    rerankScore: result.score,
+                },
+            });
+        }
+        for (const [index, item] of chunks.entries()) {
+            if (usedIndexes.has(index)) {
+                continue;
+            }
+            reranked.push((0, utils_1.cloneRetrievedChunk)(item));
+        }
+        return reranked;
+    }
+}
+exports.DefaultReranker = DefaultReranker;
+function toRerankDocument(item) {
+    const title = item.chunk.metadata.title;
+    const headingPath = item.chunk.metadata.headingPath?.join(' > ');
+    return [title, headingPath, item.chunk.content]
+        .filter((part) => part !== undefined && part.length > 0)
+        .join('\n');
+}
+//# sourceMappingURL=reranker.js.map

package/dist/rag/retrieval/reranker.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"reranker.js","sourceRoot":"","sources":["../../../src/rag/retrieval/reranker.ts"],"names":[],"mappings":";;;AAOA,mCAGiB;AAwBjB;;;GAGG;AACH,MAAa,eAAe;IACP,KAAK,CAAgC;IAEtD,YAAY,UAAkC,EAAE;QAC5C,IAAI,CAAC,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC;IAC/B,CAAC;IAED,KAAK,CAAC,MAAM,CACR,KAAa,EACb,MAAwB,EACxB,OAAyB;QAEzB,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACtB,OAAO,EAAE,CAAC;QACd,CAAC;QACD,IAAI,IAAI,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;YAC3B,OAAO,MAAM,CAAC;QAClB,CAAC;QAED,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,MAAM,EAAE,IAAA,mBAAW,EAAC,OAAO,EAAE,IAAI,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC;QAChF,IAAI,IAAI,KAAK,CAAC,EAAE,CAAC;YACb,OAAO,EAAE,CAAC;QACd,CAAC;QAED,MAAM,KAAK,GAAqB;YAC5B,KAAK;YACL,SAAS,EAAE,MAAM,CAAC,GAAG,CAAC,gBAAgB,CAAC;YACvC,MAAM,EAAE,MAAM,CAAC,GAAG,CAAC,2BAAmB,CAAC;YACvC,IAAI;SACP,CAAC;QACF,IAAI,OAAO,KAAK,SAAS,EAAE,CAAC;YACxB,KAAK,CAAC,OAAO,GAAG,OAAO,CAAC;QAC5B,CAAC;QAED,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QACpD,MAAM,YAAY,GAAG,YAAY;aAC5B,GAAG,CAAC,CAAC,MAAM,EAAE,SAAS,EAAE,EAAE,CAAC,CAAC,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC,CAAC;aACnD,MAAM,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,CACpB,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC,KAAK,CAAC;eAC3B,MAAM,CAAC,KAAK,IAAI,CAAC;eACjB,MAAM,CAAC,KAAK,GAAG,MAAM,CAAC,MAAM;eAC5B,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,KAAK,CAAC,CACnC,CAAC;aACD,IAAI,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC,CACnB,KAAK,CAAC,MAAM,CAAC,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK;eACnC,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC,SAAS,CACtC,CAAC,CAAC;QACP,MAAM,WAAW,GAAG,IAAI,GAAG,EAAU,CAAC;QACtC,MAAM,QAAQ,GAAqB,EAAE,CAAC;QAEtC,KAAK,MAAM,EAAE,MAAM,EAAE,IAAI,YAAY,EAAE,CAAC;YACpC,IAAI,WAAW,CAAC,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC;gBAChC,SAAS;YACb,CAAC;YAED,MAAM,IAAI,GAAG,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YAClC,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;gBACrB,SAAS;YACb,CAAC;YAED,WAAW,CAAC,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YAC9B,MAAM,MAAM,GAAG,IAAA,2BAAmB,EAAC,IAAI,CAAC,CAAC;YACzC,QAAQ,CAAC,IAAI,CAAC;gBACV,KAAK,EAAE,MAAM,CAAC,KAAK;gBACnB,KAAK,EAAE,MAAM,CAAC,KAAK;gBACnB,MAAM,EAAE,QAAQ;gBAChB,QAAQ,EAAE;oBACN,GAAG,MAAM,CAAC,QAAQ;oBAClB,GAAG,CAAC,MAAM,CAAC,QAAQ,IAAI,EAAE,CAAC;oBAC1B,aAAa,EAAE,IAAI,CAAC,KAAK;oBACzB,cAAc,EAAE,IAAI,CAAC,MAAM;oBAC3B,WAAW,EAAE,MAAM,CAAC,KAAK;iBAC5B;aACJ,CAAC,CAAC;QACP,CAAC;QAED,KAAK,MAAM,CAAC,KAAK,EAAE,IAAI,CAAC,IAAI,MAAM,CAAC,OAAO,EAAE,EAAE,CAAC;YAC3C,IAAI,WAAW,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC;gBACzB,SAAS;YACb,CAAC;YAED,QAAQ,CAAC,IAAI,CAAC,IAAA,2BAAmB,EAAC,IAAI,CAAC,CAAC,CAAC;QAC7C,CAAC;QAED,OAAO,QAAQ,CAAC;IACpB,CAAC;CACJ;AAtFD,0CAsFC;AAED,SAAS,gBAAgB,CAAC,IAAoB;IAC1C,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC;IACxC,MAAM,WAAW,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,WAAW,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC;IAEjE,OAAO,CAAC,KAAK,EAAE,WAAW,EAAE,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC;SAC1C,MAAM,CAAC,CAAC,IAAI,EAAkB,EAAE,CAAC,IAAI,KAAK,SAAS,IAAI,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC;SACvE,IAAI,CAAC,IAAI,CAAC,CAAC;AACpB,CAAC"}

package/dist/rag/retrieval/sparse-retriever.d.ts

@@ -0,0 +1,14 @@
+import type { DocumentStore, KeywordStore } from '../stores';
+import type { RetrievedChunk, RetrieveOptions } from '../types';
+import type { Retriever } from './types';
+export interface SparseRetrieverOptions {
+    keywordStore: KeywordStore;
+    documentStore: DocumentStore;
+}
+export declare class SparseRetriever implements Retriever {
+    private readonly keywordStore;
+    private readonly documentStore;
+    constructor(options: SparseRetrieverOptions);
+    retrieve(query: string, options?: RetrieveOptions): Promise<RetrievedChunk[]>;
+}
+//# sourceMappingURL=sparse-retriever.d.ts.map

package/dist/rag/retrieval/sparse-retriever.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"sparse-retriever.d.ts","sourceRoot":"","sources":["../../../src/rag/retrieval/sparse-retriever.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACR,aAAa,EACb,YAAY,EACf,MAAM,WAAW,CAAC;AACnB,OAAO,KAAK,EACR,cAAc,EACd,eAAe,EAClB,MAAM,UAAU,CAAC;AAClB,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,SAAS,CAAC;AAMzC,MAAM,WAAW,sBAAsB;IACnC,YAAY,EAAE,YAAY,CAAC;IAC3B,aAAa,EAAE,aAAa,CAAC;CAChC;AAED,qBAAa,eAAgB,YAAW,SAAS;IAC7C,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAe;IAC5C,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAgB;gBAElC,OAAO,EAAE,sBAAsB;IAKrC,QAAQ,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,eAAe,GAAG,OAAO,CAAC,cAAc,EAAE,CAAC;CAKtF"}

package/dist/rag/retrieval/sparse-retriever.js

@@ -0,0 +1,18 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.SparseRetriever = void 0;
+const utils_1 = require("./utils");
+class SparseRetriever {
+    keywordStore;
+    documentStore;
+    constructor(options) {
+        this.keywordStore = options.keywordStore;
+        this.documentStore = options.documentStore;
+    }
+    async retrieve(query, options) {
+        const refs = await this.keywordStore.search(query, (0, utils_1.toSearchOptions)(options));
+        return await (0, utils_1.hydrateRetrievedChunks)(this.documentStore, refs, 'sparse');
+    }
+}
+exports.SparseRetriever = SparseRetriever;
+//# sourceMappingURL=sparse-retriever.js.map

package/dist/rag/retrieval/sparse-retriever.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"sparse-retriever.js","sourceRoot":"","sources":["../../../src/rag/retrieval/sparse-retriever.ts"],"names":[],"mappings":";;;AASA,mCAGiB;AAOjB,MAAa,eAAe;IACP,YAAY,CAAe;IAC3B,aAAa,CAAgB;IAE9C,YAAY,OAA+B;QACvC,IAAI,CAAC,YAAY,GAAG,OAAO,CAAC,YAAY,CAAC;QACzC,IAAI,CAAC,aAAa,GAAG,OAAO,CAAC,aAAa,CAAC;IAC/C,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,KAAa,EAAE,OAAyB;QACnD,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,KAAK,EAAE,IAAA,uBAAe,EAAC,OAAO,CAAC,CAAC,CAAC;QAE7E,OAAO,MAAM,IAAA,8BAAsB,EAAC,IAAI,CAAC,aAAa,EAAE,IAAI,EAAE,QAAQ,CAAC,CAAC;IAC5E,CAAC;CACJ;AAdD,0CAcC"}

package/dist/rag/retrieval/types.d.ts

@@ -0,0 +1,11 @@
+import type { RetrievedChunk, RetrieveOptions } from '../types';
+export interface Retriever {
+    retrieve(query: string, options?: RetrieveOptions): Promise<RetrievedChunk[]>;
+}
+export interface Reranker {
+    rerank(query: string, chunks: RetrievedChunk[], options?: RetrieveOptions): Promise<RetrievedChunk[]>;
+}
+export declare class NoopReranker implements Reranker {
+    rerank(_query: string, chunks: RetrievedChunk[], _options?: RetrieveOptions): Promise<RetrievedChunk[]>;
+}
+//# sourceMappingURL=types.d.ts.map

package/dist/rag/retrieval/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../../src/rag/retrieval/types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACR,cAAc,EACd,eAAe,EAClB,MAAM,UAAU,CAAC;AAElB,MAAM,WAAW,SAAS;IACtB,QAAQ,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,eAAe,GAAG,OAAO,CAAC,cAAc,EAAE,CAAC,CAAC;CACjF;AAED,MAAM,WAAW,QAAQ;IACrB,MAAM,CACF,KAAK,EAAE,MAAM,EACb,MAAM,EAAE,cAAc,EAAE,EACxB,OAAO,CAAC,EAAE,eAAe,GAC1B,OAAO,CAAC,cAAc,EAAE,CAAC,CAAC;CAChC;AAED,qBAAa,YAAa,YAAW,QAAQ;IACnC,MAAM,CACR,MAAM,EAAE,MAAM,EACd,MAAM,EAAE,cAAc,EAAE,EACxB,QAAQ,CAAC,EAAE,eAAe,GAC3B,OAAO,CAAC,cAAc,EAAE,CAAC;CAG/B"}

package/dist/rag/retrieval/types.js

@@ -0,0 +1,10 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.NoopReranker = void 0;
+class NoopReranker {
+    async rerank(_query, chunks, _options) {
+        return chunks;
+    }
+}
+exports.NoopReranker = NoopReranker;
+//# sourceMappingURL=types.js.map

package/dist/rag/retrieval/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../src/rag/retrieval/types.ts"],"names":[],"mappings":";;;AAiBA,MAAa,YAAY;IACrB,KAAK,CAAC,MAAM,CACR,MAAc,EACd,MAAwB,EACxB,QAA0B;QAE1B,OAAO,MAAM,CAAC;IAClB,CAAC;CACJ;AARD,oCAQC"}

package/dist/rag/retrieval/utils.d.ts

@@ -0,0 +1,8 @@
+import type { RetrievalSource, RetrievedChunk, RetrieveOptions } from '../types';
+import type { DocumentStore, ScoredChunkRef, SearchOptions } from '../stores';
+export declare function hydrateRetrievedChunks(documentStore: DocumentStore, refs: ScoredChunkRef[], source: RetrievalSource): Promise<RetrievedChunk[]>;
+export declare function toSearchOptions(options: RetrieveOptions | undefined): SearchOptions;
+export declare function resolveTopK(topK: number | undefined, fallback?: number): number;
+export declare function withTopK(options: RetrieveOptions | undefined, topK: number): RetrieveOptions;
+export declare function cloneRetrievedChunk(chunk: RetrievedChunk): RetrievedChunk;
+//# sourceMappingURL=utils.d.ts.map

package/dist/rag/retrieval/utils.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../../src/rag/retrieval/utils.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACR,eAAe,EACf,cAAc,EACd,eAAe,EAClB,MAAM,UAAU,CAAC;AAClB,OAAO,KAAK,EACR,aAAa,EACb,cAAc,EACd,aAAa,EAChB,MAAM,WAAW,CAAC;AAEnB,wBAAsB,sBAAsB,CACxC,aAAa,EAAE,aAAa,EAC5B,IAAI,EAAE,cAAc,EAAE,EACtB,MAAM,EAAE,eAAe,GACxB,OAAO,CAAC,cAAc,EAAE,CAAC,CAyB3B;AAED,wBAAgB,eAAe,CAAC,OAAO,EAAE,eAAe,GAAG,SAAS,GAAG,aAAa,CAanF;AAED,wBAAgB,WAAW,CAAC,IAAI,EAAE,MAAM,GAAG,SAAS,EAAE,QAAQ,SAAK,GAAG,MAAM,CAS3E;AAED,wBAAgB,QAAQ,CAAC,OAAO,EAAE,eAAe,GAAG,SAAS,EAAE,IAAI,EAAE,MAAM,GAAG,eAAe,CAe5F;AAED,wBAAgB,mBAAmB,CAAC,KAAK,EAAE,cAAc,GAAG,cAAc,CA0CzE"}