@unbrained/pm-cli 2026.3.9

package/dist/core/search/providers.js

@@ -0,0 +1,265 @@
+const DEFAULT_EMBEDDING_TIMEOUT_MS = 30_000;
+function trimTrailingSlashes(value) {
+    return value.replaceAll(/\/+$/g, "");
+}
+function normalizeOpenAiEmbeddingsEndpoint(baseUrl) {
+    const normalizedBaseUrl = trimTrailingSlashes(baseUrl);
+    const normalizedLower = normalizedBaseUrl.toLowerCase();
+    if (normalizedLower.endsWith("/embeddings")) {
+        return normalizedBaseUrl;
+    }
+    if (normalizedLower.endsWith("/v1")) {
+        return `${normalizedBaseUrl}/embeddings`;
+    }
+    return `${normalizedBaseUrl}/v1/embeddings`;
+}
+function toNonEmptyString(value) {
+    if (typeof value !== "string") {
+        return null;
+    }
+    const normalized = value.trim();
+    return normalized.length > 0 ? normalized : null;
+}
+function resolveOpenAiProvider(settings) {
+    const baseUrl = toNonEmptyString(settings.providers?.openai?.base_url);
+    const model = toNonEmptyString(settings.providers?.openai?.model);
+    if (!baseUrl || !model) {
+        return null;
+    }
+    const apiKey = toNonEmptyString(settings.providers?.openai?.api_key);
+    return {
+        name: "openai",
+        base_url: baseUrl,
+        model,
+        ...(apiKey ? { api_key: apiKey } : {}),
+    };
+}
+function resolveOllamaProvider(settings) {
+    const baseUrl = toNonEmptyString(settings.providers?.ollama?.base_url);
+    const model = toNonEmptyString(settings.providers?.ollama?.model);
+    if (!baseUrl || !model) {
+        return null;
+    }
+    return {
+        name: "ollama",
+        base_url: baseUrl,
+        model,
+    };
+}
+function normalizeEmbeddingInputs(input) {
+    const inputs = (Array.isArray(input) ? input : [input])
+        .map((entry) => entry.trim())
+        .filter((entry) => entry.length > 0);
+    if (inputs.length === 0) {
+        throw new Error("Embedding input must include at least one non-empty string");
+    }
+    return inputs;
+}
+function dedupeEmbeddingInputs(inputs) {
+    const uniqueInputs = [];
+    const originalToUniqueIndex = [];
+    const uniqueIndexByInput = new Map();
+    for (const entry of inputs) {
+        const existingUniqueIndex = uniqueIndexByInput.get(entry);
+        if (existingUniqueIndex === undefined) {
+            const nextUniqueIndex = uniqueInputs.length;
+            uniqueInputs.push(entry);
+            uniqueIndexByInput.set(entry, nextUniqueIndex);
+            originalToUniqueIndex.push(nextUniqueIndex);
+            continue;
+        }
+        originalToUniqueIndex.push(existingUniqueIndex);
+    }
+    return {
+        uniqueInputs,
+        originalToUniqueIndex,
+    };
+}
+function isFiniteNumberArray(value) {
+    return Array.isArray(value) && value.every((entry) => typeof entry === "number" && Number.isFinite(entry));
+}
+function buildOrderedOpenAiEntries(data) {
+    const openAiEntries = data.map((entry, position) => ({
+        embedding: entry.embedding,
+        index: entry.index,
+        position,
+    }));
+    const hasExplicitIndex = openAiEntries.some((entry) => entry.index !== undefined);
+    if (hasExplicitIndex) {
+        for (const entry of openAiEntries) {
+            if (!Number.isInteger(entry.index)) {
+                throw new TypeError(`OpenAI embedding response entry at position ${entry.position} is missing a valid integer index`);
+            }
+        }
+        return [...openAiEntries].sort((a, b) => {
+            const byIndex = a.index - b.index;
+            if (byIndex !== 0) {
+                return byIndex;
+            }
+            return a.position - b.position;
+        });
+    }
+    return openAiEntries;
+}
+export function resolveEmbeddingProviders(settings) {
+    const openAi = resolveOpenAiProvider(settings);
+    const ollama = resolveOllamaProvider(settings);
+    const available = [openAi, ollama].filter((entry) => entry !== null);
+    return {
+        active: available[0] ?? null,
+        available,
+    };
+}
+export function resolveEmbeddingRequestTarget(provider) {
+    const baseUrl = trimTrailingSlashes(provider.base_url);
+    if (provider.name === "openai") {
+        return {
+            provider: "openai",
+            endpoint: normalizeOpenAiEmbeddingsEndpoint(baseUrl),
+            model: provider.model,
+        };
+    }
+    return {
+        provider: "ollama",
+        endpoint: `${baseUrl}/api/embed`,
+        model: provider.model,
+    };
+}
+export function buildEmbeddingRequestPlan(provider, input) {
+    const normalizedInputs = normalizeEmbeddingInputs(input);
+    if (provider.name === "openai") {
+        return {
+            target: resolveEmbeddingRequestTarget(provider),
+            method: "POST",
+            headers: {
+                "content-type": "application/json",
+                ...(provider.api_key ? { authorization: `Bearer ${provider.api_key}` } : {}),
+            },
+            body: {
+                model: provider.model,
+                input: normalizedInputs.length === 1 ? normalizedInputs[0] : normalizedInputs,
+            },
+        };
+    }
+    return {
+        target: resolveEmbeddingRequestTarget(provider),
+        method: "POST",
+        headers: {
+            "content-type": "application/json",
+        },
+        body: {
+            model: provider.model,
+            input: normalizedInputs,
+        },
+    };
+}
+export function normalizeEmbeddingResponse(provider, response) {
+    if (provider.name === "openai") {
+        const data = response.data;
+        if (!Array.isArray(data) || data.length === 0) {
+            throw new Error("OpenAI embedding response must include a non-empty data array");
+        }
+        const orderedEntries = buildOrderedOpenAiEntries(data);
+        return orderedEntries.map((entry, index) => {
+            const embedding = entry.embedding;
+            if (!isFiniteNumberArray(embedding)) {
+                throw new Error(`OpenAI embedding response entry at index ${index} is missing a numeric embedding vector`);
+            }
+            return [...embedding];
+        });
+    }
+    const singleEmbedding = response.embedding;
+    if (isFiniteNumberArray(singleEmbedding)) {
+        return [[...singleEmbedding]];
+    }
+    const multipleEmbeddings = response.embeddings;
+    if (Array.isArray(multipleEmbeddings) && multipleEmbeddings.length > 0) {
+        return multipleEmbeddings.map((entry, index) => {
+            if (!isFiniteNumberArray(entry)) {
+                throw new Error(`Ollama embedding response entry at index ${index} is missing a numeric embedding vector`);
+            }
+            return [...entry];
+        });
+    }
+    throw new Error("Ollama embedding response must include embedding or embeddings vectors");
+}
+function resolveEmbeddingFetcher(fetcher) {
+    if (fetcher) {
+        return fetcher;
+    }
+    if (typeof globalThis.fetch === "function") {
+        return globalThis.fetch.bind(globalThis);
+    }
+    throw new Error("Embedding request execution requires a fetch implementation");
+}
+function normalizeTimeoutMs(timeoutMs) {
+    const resolved = timeoutMs ?? DEFAULT_EMBEDDING_TIMEOUT_MS;
+    if (!Number.isFinite(resolved) || resolved <= 0) {
+        throw new Error("Embedding request timeout must be a positive finite number");
+    }
+    return Math.floor(resolved);
+}
+function toErrorMessage(error) {
+    if (error instanceof Error) {
+        const message = error.message.trim();
+        return message.length > 0 ? message : error.name;
+    }
+    return String(error);
+}
+async function readFailedResponseBody(response) {
+    try {
+        return (await response.text()).replaceAll(/\s+/g, " ").trim();
+    }
+    catch (error) {
+        return `(failed to read response body: ${toErrorMessage(error)})`;
+    }
+}
+export async function executeEmbeddingRequest(provider, input, options = {}) {
+    const timeoutMs = normalizeTimeoutMs(options.timeout_ms);
+    const fetcher = resolveEmbeddingFetcher(options.fetcher);
+    const normalizedInputs = normalizeEmbeddingInputs(input);
+    const dedupedInputs = dedupeEmbeddingInputs(normalizedInputs);
+    const requestPlan = buildEmbeddingRequestPlan(provider, dedupedInputs.uniqueInputs);
+    const controller = new AbortController();
+    const timeoutHandle = setTimeout(() => {
+        controller.abort();
+    }, timeoutMs);
+    try {
+        let response;
+        try {
+            response = await fetcher(requestPlan.target.endpoint, {
+                method: requestPlan.method,
+                headers: requestPlan.headers,
+                body: JSON.stringify(requestPlan.body),
+                signal: controller.signal,
+            });
+        }
+        catch (error) {
+            if (error instanceof Error && error.name === "AbortError") {
+                throw new Error(`Embedding request timed out after ${timeoutMs}ms`);
+            }
+            throw new Error(`Embedding request execution failed: ${toErrorMessage(error)}`);
+        }
+        if (!response.ok) {
+            const responseBody = await readFailedResponseBody(response);
+            const detail = responseBody.length > 0 ? `: ${responseBody}` : "";
+            throw new Error(`Embedding request failed with status ${response.status} ${response.statusText}${detail}`);
+        }
+        let payload;
+        try {
+            payload = await response.json();
+        }
+        catch (error) {
+            throw new Error(`Embedding response JSON parse failed: ${toErrorMessage(error)}`);
+        }
+        const vectors = normalizeEmbeddingResponse(provider, payload);
+        if (vectors.length !== dedupedInputs.uniqueInputs.length) {
+            throw new Error(`Embedding response cardinality mismatch: expected ${dedupedInputs.uniqueInputs.length} vector(s), received ${vectors.length}`);
+        }
+        return dedupedInputs.originalToUniqueIndex.map((uniqueIndex) => [...vectors[uniqueIndex]]);
+    }
+    finally {
+        clearTimeout(timeoutHandle);
+    }
+}
+//# sourceMappingURL=providers.js.map

package/dist/core/search/providers.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"providers.js","sourceRoot":"","sources":["../../../src/core/search/providers.ts"],"names":[],"mappings":"AAoDA,MAAM,4BAA4B,GAAG,MAAM,CAAC;AAgB5C,SAAS,mBAAmB,CAAC,KAAa;IACxC,OAAO,KAAK,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;AACvC,CAAC;AAED,SAAS,iCAAiC,CAAC,OAAe;IACxD,MAAM,iBAAiB,GAAG,mBAAmB,CAAC,OAAO,CAAC,CAAC;IACvD,MAAM,eAAe,GAAG,iBAAiB,CAAC,WAAW,EAAE,CAAC;IACxD,IAAI,eAAe,CAAC,QAAQ,CAAC,aAAa,CAAC,EAAE,CAAC;QAC5C,OAAO,iBAAiB,CAAC;IAC3B,CAAC;IACD,IAAI,eAAe,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;QACpC,OAAO,GAAG,iBAAiB,aAAa,CAAC;IAC3C,CAAC;IACD,OAAO,GAAG,iBAAiB,gBAAgB,CAAC;AAC9C,CAAC;AAED,SAAS,gBAAgB,CAAC,KAAc;IACtC,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QAC9B,OAAO,IAAI,CAAC;IACd,CAAC;IACD,MAAM,UAAU,GAAG,KAAK,CAAC,IAAI,EAAE,CAAC;IAChC,OAAO,UAAU,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC;AACnD,CAAC;AAED,SAAS,qBAAqB,CAAC,QAA+B;IAC5D,MAAM,OAAO,GAAG,gBAAgB,CAAC,QAAQ,CAAC,SAAS,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC;IACvE,MAAM,KAAK,GAAG,gBAAgB,CAAC,QAAQ,CAAC,SAAS,EAAE,MAAM,EAAE,KAAK,CAAC,CAAC;IAClE,IAAI,CAAC,OAAO,IAAI,CAAC,KAAK,EAAE,CAAC;QACvB,OAAO,IAAI,CAAC;IACd,CAAC;IACD,MAAM,MAAM,GAAG,gBAAgB,CAAC,QAAQ,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC;IACrE,OAAO;QACL,IAAI,EAAE,QAAQ;QACd,QAAQ,EAAE,OAAO;QACjB,KAAK;QACL,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KACvC,CAAC;AACJ,CAAC;AAED,SAAS,qBAAqB,CAAC,QAA+B;IAC5D,MAAM,OAAO,GAAG,gBAAgB,CAAC,QAAQ,CAAC,SAAS,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC;IACvE,MAAM,KAAK,GAAG,gBAAgB,CAAC,QAAQ,CAAC,SAAS,EAAE,MAAM,EAAE,KAAK,CAAC,CAAC;IAClE,IAAI,CAAC,OAAO,IAAI,CAAC,KAAK,EAAE,CAAC;QACvB,OAAO,IAAI,CAAC;IACd,CAAC;IACD,OAAO;QACL,IAAI,EAAE,QAAQ;QACd,QAAQ,EAAE,OAAO;QACjB,KAAK;KACN,CAAC;AACJ,CAAC;AAED,SAAS,wBAAwB,CAAC,KAAwB;IACxD,MAAM,MAAM,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;SACpD,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC;SAC5B,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IACvC,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACxB,MAAM,IAAI,KAAK,CAAC,4DAA4D,CAAC,CAAC;IAChF,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAOD,SAAS,qBAAqB,CAAC,MAAgB;IAC7C,MAAM,YAAY,GAAa,EAAE,CAAC;IAClC,MAAM,qBAAqB,GAAa,EAAE,CAAC;IAC3C,MAAM,kBAAkB,GAAG,IAAI,GAAG,EAAkB,CAAC;IACrD,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;QAC3B,MAAM,mBAAmB,GAAG,kBAAkB,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAC1D,IAAI,mBAAmB,KAAK,SAAS,EAAE,CAAC;YACtC,MAAM,eAAe,GAAG,YAAY,CAAC,MAAM,CAAC;YAC5C,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACzB,kBAAkB,CAAC,GAAG,CAAC,KAAK,EAAE,eAAe,CAAC,CAAC;YAC/C,qBAAqB,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;YAC5C,SAAS;QACX,CAAC;QACD,qBAAqB,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;IAClD,CAAC;IACD,OAAO;QACL,YAAY;QACZ,qBAAqB;KACtB,CAAC;AACJ,CAAC;AAED,SAAS,mBAAmB,CAAC,KAAc;IACzC,OAAO,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,OAAO,KAAK,KAAK,QAAQ,IAAI,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC;AAC7G,CAAC;AAQD,SAAS,yBAAyB,CAAC,IAAe;IAChD,MAAM,aAAa,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,QAAQ,EAAE,EAAE,CAAC,CAAC;QACnD,SAAS,EAAG,KAAiC,CAAC,SAAS;QACvD,KAAK,EAAG,KAA6B,CAAC,KAAK;QAC3C,QAAQ;KACT,CAAC,CAAC,CAAC;IACJ,MAAM,gBAAgB,GAAG,aAAa,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC;IAClF,IAAI,gBAAgB,EAAE,CAAC;QACrB,KAAK,MAAM,KAAK,IAAI,aAAa,EAAE,CAAC;YAClC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC,KAAK,CAAC,EAAE,CAAC;gBACnC,MAAM,IAAI,SAAS,CAAC,+CAA+C,KAAK,CAAC,QAAQ,mCAAmC,CAAC,CAAC;YACxH,CAAC;QACH,CAAC;QACD,OAAO,CAAC,GAAG,aAAa,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE;YACtC,MAAM,OAAO,GAAI,CAAC,CAAC,KAAgB,GAAI,CAAC,CAAC,KAAgB,CAAC;YAC1D,IAAI,OAAO,KAAK,CAAC,EAAE,CAAC;gBAClB,OAAO,OAAO,CAAC;YACjB,CAAC;YACD,OAAO,CAAC,CAAC,QAAQ,GAAG,CAAC,CAAC,QAAQ,CAAC;QACjC,CAAC,CAAC,CAAC;IACL,CAAC;IACD,OAAO,aAAa,CAAC;AACvB,CAAC;AAED,MAAM,UAAU,yBAAyB,CAAC,QAA4C;IACpF,MAAM,MAAM,GAAG,qBAAqB,CAAC,QAAQ,CAAC,CAAC;IAC/C,MAAM,MAAM,GAAG,qBAAqB,CAAC,QAAQ,CAAC,CAAC;IAC/C,MAAM,SAAS,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,MAAM,CAAC,CAAC,KAAK,EAAoC,EAAE,CAAC,KAAK,KAAK,IAAI,CAAC,CAAC;IACvG,OAAO;QACL,MAAM,EAAE,SAAS,CAAC,CAAC,CAAC,IAAI,IAAI;QAC5B,SAAS;KACV,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,6BAA6B,CAAC,QAAiC;IAC7E,MAAM,OAAO,GAAG,mBAAmB,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;IACvD,IAAI,QAAQ,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;QAC/B,OAAO;YACL,QAAQ,EAAE,QAAQ;YAClB,QAAQ,EAAE,iCAAiC,CAAC,OAAO,CAAC;YACpD,KAAK,EAAE,QAAQ,CAAC,KAAK;SACtB,CAAC;IACJ,CAAC;IACD,OAAO;QACL,QAAQ,EAAE,QAAQ;QAClB,QAAQ,EAAE,GAAG,OAAO,YAAY;QAChC,KAAK,EAAE,QAAQ,CAAC,KAAK;KACtB,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,yBAAyB,CAAC,QAAiC,EAAE,KAAwB;IACnG,MAAM,gBAAgB,GAAG,wBAAwB,CAAC,KAAK,CAAC,CAAC;IACzD,IAAI,QAAQ,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;QAC/B,OAAO;YACL,MAAM,EAAE,6BAA6B,CAAC,QAAQ,CAAC;YAC/C,MAAM,EAAE,MAAM;YACd,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;gBAClC,GAAG,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,aAAa,EAAE,UAAU,QAAQ,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aAC7E;YACD,IAAI,EAAE;gBACJ,KAAK,EAAE,QAAQ,CAAC,KAAK;gBACrB,KAAK,EAAE,gBAAgB,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,gBAAgB,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,gBAAgB;aAC9E;SACF,CAAC;IACJ,CAAC;IACD,OAAO;QACL,MAAM,EAAE,6BAA6B,CAAC,QAAQ,CAAC;QAC/C,MAAM,EAAE,MAAM;QACd,OAAO,EAAE;YACP,cAAc,EAAE,kBAAkB;SACnC;QACD,IAAI,EAAE;YACJ,KAAK,EAAE,QAAQ,CAAC,KAAK;YACrB,KAAK,EAAE,gBAAgB;SACxB;KACF,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,0BAA0B,CAAC,QAAiC,EAAE,QAAiB;IAC7F,IAAI,QAAQ,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;QAC/B,MAAM,IAAI,GAAI,QAA+B,CAAC,IAAI,CAAC;QACnD,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC9C,MAAM,IAAI,KAAK,CAAC,+DAA+D,CAAC,CAAC;QACnF,CAAC;QACD,MAAM,cAAc,GAAG,yBAAyB,CAAC,IAAI,CAAC,CAAC;QAEvD,OAAO,cAAc,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE;YACzC,MAAM,SAAS,GAAG,KAAK,CAAC,SAAS,CAAC;YAClC,IAAI,CAAC,mBAAmB,CAAC,SAAS,CAAC,EAAE,CAAC;gBACpC,MAAM,IAAI,KAAK,CAAC,4CAA4C,KAAK,wCAAwC,CAAC,CAAC;YAC7G,CAAC;YACD,OAAO,CAAC,GAAG,SAAS,CAAC,CAAC;QACxB,CAAC,CAAC,CAAC;IACL,CAAC;IAED,MAAM,eAAe,GAAI,QAAoC,CAAC,SAAS,CAAC;IACxE,IAAI,mBAAmB,CAAC,eAAe,CAAC,EAAE,CAAC;QACzC,OAAO,CAAC,CAAC,GAAG,eAAe,CAAC,CAAC,CAAC;IAChC,CAAC;IAED,MAAM,kBAAkB,GAAI,QAAqC,CAAC,UAAU,CAAC;IAC7E,IAAI,KAAK,CAAC,OAAO,CAAC,kBAAkB,CAAC,IAAI,kBAAkB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACvE,OAAO,kBAAkB,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE;YAC7C,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,EAAE,CAAC;gBAChC,MAAM,IAAI,KAAK,CAAC,4CAA4C,KAAK,wCAAwC,CAAC,CAAC;YAC7G,CAAC;YACD,OAAO,CAAC,GAAG,KAAK,CAAC,CAAC;QACpB,CAAC,CAAC,CAAC;IACL,CAAC;IAED,MAAM,IAAI,KAAK,CAAC,wEAAwE,CAAC,CAAC;AAC5F,CAAC;AAED,SAAS,uBAAuB,CAAC,OAA4C;IAC3E,IAAI,OAAO,EAAE,CAAC;QACZ,OAAO,OAAO,CAAC;IACjB,CAAC;IACD,IAAI,OAAO,UAAU,CAAC,KAAK,KAAK,UAAU,EAAE,CAAC;QAC3C,OAAO,UAAU,CAAC,KAAK,CAAC,IAAI,CAAC,UAAU,CAAuC,CAAC;IACjF,CAAC;IACD,MAAM,IAAI,KAAK,CAAC,6DAA6D,CAAC,CAAC;AACjF,CAAC;AAED,SAAS,kBAAkB,CAAC,SAA6B;IACvD,MAAM,QAAQ,GAAG,SAAS,IAAI,4BAA4B,CAAC;IAC3D,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,QAAQ,IAAI,CAAC,EAAE,CAAC;QAChD,MAAM,IAAI,KAAK,CAAC,4DAA4D,CAAC,CAAC;IAChF,CAAC;IACD,OAAO,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;AAC9B,CAAC;AAED,SAAS,cAAc,CAAC,KAAc;IACpC,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;QAC3B,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC;QACrC,OAAO,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC;IACnD,CAAC;IACD,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC;AACvB,CAAC;AAED,KAAK,UAAU,sBAAsB,CAAC,QAA+B;IACnE,IAAI,CAAC;QACH,OAAO,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC,CAAC,UAAU,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC;IAChE,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,kCAAkC,cAAc,CAAC,KAAK,CAAC,GAAG,CAAC;IACpE,CAAC;AACH,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,uBAAuB,CAC3C,QAAiC,EACjC,KAAwB,EACxB,UAA0C,EAAE;IAE5C,MAAM,SAAS,GAAG,kBAAkB,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;IACzD,MAAM,OAAO,GAAG,uBAAuB,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;IACzD,MAAM,gBAAgB,GAAG,wBAAwB,CAAC,KAAK,CAAC,CAAC;IACzD,MAAM,aAAa,GAAG,qBAAqB,CAAC,gBAAgB,CAAC,CAAC;IAC9D,MAAM,WAAW,GAAG,yBAAyB,CAAC,QAAQ,EAAE,aAAa,CAAC,YAAY,CAAC,CAAC;IACpF,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;IACzC,MAAM,aAAa,GAAG,UAAU,CAAC,GAAG,EAAE;QACpC,UAAU,CAAC,KAAK,EAAE,CAAC;IACrB,CAAC,EAAE,SAAS,CAAC,CAAC;IAEd,IAAI,CAAC;QACH,IAAI,QAA+B,CAAC;QACpC,IAAI,CAAC;YACH,QAAQ,GAAG,MAAM,OAAO,CAAC,WAAW,CAAC,MAAM,CAAC,QAAQ,EAAE;gBACpD,MAAM,EAAE,WAAW,CAAC,MAAM;gBAC1B,OAAO,EAAE,WAAW,CAAC,OAAO;gBAC5B,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC,IAAI,CAAC;gBACtC,MAAM,EAAE,UAAU,CAAC,MAAM;aAC1B,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,KAAK,YAAY,KAAK,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;gBAC1D,MAAM,IAAI,KAAK,CAAC,qCAAqC,SAAS,IAAI,CAAC,CAAC;YACtE,CAAC;YACD,MAAM,IAAI,KAAK,CAAC,uCAAuC,cAAc,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QAClF,CAAC;QAED,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,YAAY,GAAG,MAAM,sBAAsB,CAAC,QAAQ,CAAC,CAAC;YAC5D,MAAM,MAAM,GAAG,YAAY,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,YAAY,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAClE,MAAM,IAAI,KAAK,CAAC,wCAAwC,QAAQ,CAAC,MAAM,IAAI,QAAQ,CAAC,UAAU,GAAG,MAAM,EAAE,CAAC,CAAC;QAC7G,CAAC;QAED,IAAI,OAAgB,CAAC;QACrB,IAAI,CAAC;YACH,OAAO,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;QAClC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,IAAI,KAAK,CAAC,yCAAyC,cAAc,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QACpF,CAAC;QACD,MAAM,OAAO,GAAG,0BAA0B,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QAC9D,IAAI,OAAO,CAAC,MAAM,KAAK,aAAa,CAAC,YAAY,CAAC,MAAM,EAAE,CAAC;YACzD,MAAM,IAAI,KAAK,CACb,qDAAqD,aAAa,CAAC,YAAY,CAAC,MAAM,wBAAwB,OAAO,CAAC,MAAM,EAAE,CAC/H,CAAC;QACJ,CAAC;QACD,OAAO,aAAa,CAAC,qBAAqB,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC,GAAG,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC;IAC7F,CAAC;YAAS,CAAC;QACT,YAAY,CAAC,aAAa,CAAC,CAAC;IAC9B,CAAC;AACH,CAAC"}

package/dist/core/search/vector-stores.d.ts

@@ -0,0 +1,89 @@
+import type { PmSettings } from "../../types/index.js";
+export type VectorStoreName = "qdrant" | "lancedb";
+export interface QdrantVectorStoreConfig {
+    name: "qdrant";
+    url: string;
+    api_key?: string;
+}
+export interface LanceDbVectorStoreConfig {
+    name: "lancedb";
+    path: string;
+}
+export type VectorStoreConfig = QdrantVectorStoreConfig | LanceDbVectorStoreConfig;
+export interface VectorStoreResolution {
+    active: VectorStoreConfig | null;
+    available: VectorStoreConfig[];
+}
+export interface VectorStoreRequestTarget {
+    store: VectorStoreName;
+    query_target: string;
+    upsert_target: string;
+}
+export interface VectorQueryPlan {
+    target: VectorStoreRequestTarget;
+    method: "POST" | "LOCAL";
+    headers: Record<string, string>;
+    body: Record<string, unknown>;
+}
+export interface VectorRecord {
+    id: string;
+    vector: number[];
+    payload?: Record<string, unknown>;
+}
+export interface VectorUpsertPlan {
+    target: VectorStoreRequestTarget;
+    method: "POST" | "LOCAL";
+    headers: Record<string, string>;
+    body: Record<string, unknown>;
+}
+export interface VectorDeletePlan {
+    target: VectorStoreRequestTarget;
+    method: "POST" | "LOCAL";
+    headers: Record<string, string>;
+    body: Record<string, unknown>;
+}
+export interface VectorQueryHit {
+    id: string;
+    score: number;
+    payload?: Record<string, unknown>;
+}
+export interface VectorUpsertResult {
+    status: string;
+}
+export interface VectorHttpResponse {
+    ok: boolean;
+    status: number;
+    statusText: string;
+    json(): Promise<unknown>;
+    text(): Promise<string>;
+}
+export type VectorRequestFetcher = (url: string, init: {
+    method: "POST";
+    headers: Record<string, string>;
+    body: string;
+    signal: AbortSignal;
+}) => Promise<VectorHttpResponse>;
+export interface ExecuteVectorRequestOptions {
+    timeout_ms?: number;
+    fetcher?: VectorRequestFetcher;
+}
+type VectorSettingsInput = {
+    vector_store?: {
+        qdrant?: {
+            url?: string;
+            api_key?: string;
+        };
+        lancedb?: {
+            path?: string;
+        };
+    };
+};
+export declare function resolveVectorStores(settings: PmSettings | VectorSettingsInput): VectorStoreResolution;
+export declare function resolveVectorStoreRequestTarget(store: VectorStoreConfig): VectorStoreRequestTarget;
+export declare function buildVectorQueryPlan(store: VectorStoreConfig, vector: number[], limit: number): VectorQueryPlan;
+export declare function buildVectorUpsertPlan(store: VectorStoreConfig, records: VectorRecord[]): VectorUpsertPlan;
+export declare function buildVectorDeletePlan(store: VectorStoreConfig, ids: string[]): VectorDeletePlan;
+export declare function executeVectorQuery(store: VectorStoreConfig, vector: number[], limit: number, options?: ExecuteVectorRequestOptions): Promise<VectorQueryHit[]>;
+export declare function executeVectorUpsert(store: VectorStoreConfig, records: VectorRecord[], options?: ExecuteVectorRequestOptions): Promise<VectorUpsertResult>;
+export declare function executeVectorDelete(store: VectorStoreConfig, ids: string[], options?: ExecuteVectorRequestOptions): Promise<VectorUpsertResult>;
+export {};