cortex-engine 0.1.0

package/dist/providers/ollama.js

@@ -0,0 +1,103 @@
+/**
+ * OllamaEmbedProvider and OllamaLLMProvider — Ollama REST API implementations.
+ *
+ * Talks to a local Ollama instance (default: http://localhost:11434).
+ * Uses Node.js built-in fetch — no extra HTTP dependencies.
+ *
+ * Defaults:
+ *   embed model  — qwen3-embedding:0.6b (1024 dimensions, MRL to 32)
+ *   LLM model    — qwen2.5:14b
+ */
+// ---------------------------------------------------------------------------
+// Internal helpers
+// ---------------------------------------------------------------------------
+async function ollamaPost(baseUrl, path, body) {
+    let response;
+    try {
+        response = await fetch(`${baseUrl}${path}`, {
+            method: 'POST',
+            headers: { 'Content-Type': 'application/json' },
+            body: JSON.stringify(body),
+        });
+    }
+    catch (cause) {
+        throw new Error(`Ollama not reachable at ${baseUrl}. Is Ollama running?`, { cause });
+    }
+    if (!response.ok) {
+        const text = await response.text().catch(() => '(no body)');
+        throw new Error(`Ollama request to ${path} failed: HTTP ${response.status} — ${text}`);
+    }
+    return response.json();
+}
+export class OllamaEmbedProvider {
+    name = 'ollama';
+    dimensions;
+    model;
+    baseUrl;
+    constructor(options) {
+        this.model = options?.model ?? 'qwen3-embedding:0.6b';
+        this.baseUrl = options?.baseUrl ?? 'http://localhost:11434';
+        this.dimensions = options?.dimensions ?? 1024;
+    }
+    async embed(text) {
+        const data = await ollamaPost(this.baseUrl, '/api/embed', { model: this.model, input: text });
+        return data.embeddings[0];
+    }
+    async embedBatch(texts) {
+        const data = await ollamaPost(this.baseUrl, '/api/embed', { model: this.model, input: texts });
+        return data.embeddings;
+    }
+}
+export class OllamaLLMProvider {
+    name = 'ollama';
+    modelId;
+    baseUrl;
+    constructor(options) {
+        this.modelId = options?.model ?? 'qwen2.5:14b';
+        this.baseUrl = options?.baseUrl ?? 'http://localhost:11434';
+    }
+    async generate(prompt, options) {
+        const body = {
+            model: this.modelId,
+            prompt,
+            stream: false,
+            options: {
+                ...(options?.temperature !== undefined && { temperature: options.temperature }),
+                ...(options?.maxTokens !== undefined && { num_predict: options.maxTokens }),
+            },
+        };
+        if (options?.systemPrompt) {
+            body['system'] = options.systemPrompt;
+        }
+        const data = await ollamaPost(this.baseUrl, '/api/generate', body);
+        return data.response;
+    }
+    async generateJSON(prompt, options) {
+        let systemPrompt = options?.systemPrompt ?? '';
+        if (options?.schema) {
+            const schemaInstruction = `Respond with JSON matching this schema: ${JSON.stringify(options.schema)}`;
+            systemPrompt = systemPrompt ? `${systemPrompt}\n\n${schemaInstruction}` : schemaInstruction;
+        }
+        const body = {
+            model: this.modelId,
+            prompt,
+            stream: false,
+            format: 'json',
+            options: {
+                ...(options?.temperature !== undefined && { temperature: options.temperature }),
+                ...(options?.maxTokens !== undefined && { num_predict: options.maxTokens }),
+            },
+        };
+        if (systemPrompt) {
+            body['system'] = systemPrompt;
+        }
+        const data = await ollamaPost(this.baseUrl, '/api/generate', body);
+        try {
+            return JSON.parse(data.response);
+        }
+        catch (cause) {
+            throw new Error(`OllamaLLMProvider.generateJSON: failed to parse JSON response.\nRaw response: ${data.response}`, { cause });
+        }
+    }
+}
+//# sourceMappingURL=ollama.js.map

package/dist/providers/ollama.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ollama.js","sourceRoot":"","sources":["../../src/providers/ollama.ts"],"names":[],"mappings":"AAAA;;;;;;;;;GASG;AAKH,8EAA8E;AAC9E,mBAAmB;AACnB,8EAA8E;AAE9E,KAAK,UAAU,UAAU,CACvB,OAAe,EACf,IAAY,EACZ,IAA6B;IAE7B,IAAI,QAAkB,CAAC;IACvB,IAAI,CAAC;QACH,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,OAAO,GAAG,IAAI,EAAE,EAAE;YAC1C,MAAM,EAAE,MAAM;YACd,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;YAC/C,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;SAC3B,CAAC,CAAC;IACL,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,IAAI,KAAK,CACb,2BAA2B,OAAO,sBAAsB,EACxD,EAAE,KAAK,EAAE,CACV,CAAC;IACJ,CAAC;IAED,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;QACjB,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,WAAW,CAAC,CAAC;QAC5D,MAAM,IAAI,KAAK,CACb,qBAAqB,IAAI,iBAAiB,QAAQ,CAAC,MAAM,MAAM,IAAI,EAAE,CACtE,CAAC;IACJ,CAAC;IAED,OAAO,QAAQ,CAAC,IAAI,EAAgB,CAAC;AACvC,CAAC;AAUD,MAAM,OAAO,mBAAmB;IACrB,IAAI,GAAG,QAAQ,CAAC;IAChB,UAAU,CAAS;IAEX,KAAK,CAAS;IACd,OAAO,CAAS;IAEjC,YAAY,OAIX;QACC,IAAI,CAAC,KAAK,GAAG,OAAO,EAAE,KAAK,IAAI,sBAAsB,CAAC;QACtD,IAAI,CAAC,OAAO,GAAG,OAAO,EAAE,OAAO,IAAI,wBAAwB,CAAC;QAC5D,IAAI,CAAC,UAAU,GAAG,OAAO,EAAE,UAAU,IAAI,IAAI,CAAC;IAChD,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,IAAY;QACtB,MAAM,IAAI,GAAG,MAAM,UAAU,CAC3B,IAAI,CAAC,OAAO,EACZ,YAAY,EACZ,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,CACnC,CAAC;QACF,OAAO,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;IAC5B,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,KAAe;QAC9B,MAAM,IAAI,GAAG,MAAM,UAAU,CAC3B,IAAI,CAAC,OAAO,EACZ,YAAY,EACZ,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,CACpC,CAAC;QACF,OAAO,IAAI,CAAC,UAAU,CAAC;IACzB,CAAC;CACF;AAUD,MAAM,OAAO,iBAAiB;IACnB,IAAI,GAAG,QAAQ,CAAC;IAChB,OAAO,CAAS;IAER,OAAO,CAAS;IAEjC,YAAY,OAGX;QACC,IAAI,CAAC,OAAO,GAAG,OAAO,EAAE,KAAK,IAAI,aAAa,CAAC;QAC/C,IAAI,CAAC,OAAO,GAAG,OAAO,EAAE,OAAO,IAAI,wBAAwB,CAAC;IAC9D,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,MAAc,EAAE,OAAyB;QACtD,MAAM,IAAI,GAA4B;YACpC,KAAK,EAAE,IAAI,CAAC,OAAO;YACnB,MAAM;YACN,MAAM,EAAE,KAAK;YACb,OAAO,EAAE;gBACP,GAAG,CAAC,OAAO,EAAE,WAAW,KAAK,SAAS,IAAI,EAAE,WAAW,EAAE,OAAO,CAAC,WAAW,EAAE,CAAC;gBAC/E,GAAG,CAAC,OAAO,EAAE,SAAS,KAAK,SAAS,IAAI,EAAE,WAAW,EAAE,OAAO,CAAC,SAAS,EAAE,CAAC;aAC5E;SACF,CAAC;QAEF,IAAI,OAAO,EAAE,YAAY,EAAE,CAAC;YAC1B,IAAI,CAAC,QAAQ,CAAC,GAAG,OAAO,CAAC,YAAY,CAAC;QACxC,CAAC;QAED,MAAM,IAAI,GAAG,MAAM,UAAU,CAC3B,IAAI,CAAC,OAAO,EACZ,eAAe,EACf,IAAI,CACL,CAAC;QAEF,OAAO,IAAI,CAAC,QAAQ,CAAC;IACvB,CAAC;IAED,KAAK,CAAC,YAAY,CAAI,MAAc,EAAE,OAA6B;QACjE,IAAI,YAAY,GAAG,OAAO,EAAE,YAAY,IAAI,EAAE,CAAC;QAE/C,IAAI,OAAO,EAAE,MAAM,EAAE,CAAC;YACpB,MAAM,iBAAiB,GAAG,2CAA2C,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;YACtG,YAAY,GAAG,YAAY,CAAC,CAAC,CAAC,GAAG,YAAY,OAAO,iBAAiB,EAAE,CAAC,CAAC,CAAC,iBAAiB,CAAC;QAC9F,CAAC;QAED,MAAM,IAAI,GAA4B;YACpC,KAAK,EAAE,IAAI,CAAC,OAAO;YACnB,MAAM;YACN,MAAM,EAAE,KAAK;YACb,MAAM,EAAE,MAAM;YACd,OAAO,EAAE;gBACP,GAAG,CAAC,OAAO,EAAE,WAAW,KAAK,SAAS,IAAI,EAAE,WAAW,EAAE,OAAO,CAAC,WAAW,EAAE,CAAC;gBAC/E,GAAG,CAAC,OAAO,EAAE,SAAS,KAAK,SAAS,IAAI,EAAE,WAAW,EAAE,OAAO,CAAC,SAAS,EAAE,CAAC;aAC5E;SACF,CAAC;QAEF,IAAI,YAAY,EAAE,CAAC;YACjB,IAAI,CAAC,QAAQ,CAAC,GAAG,YAAY,CAAC;QAChC,CAAC;QAED,MAAM,IAAI,GAAG,MAAM,UAAU,CAC3B,IAAI,CAAC,OAAO,EACZ,eAAe,EACf,IAAI,CACL,CAAC;QAEF,IAAI,CAAC;YACH,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAM,CAAC;QACxC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,IAAI,KAAK,CACb,iFAAiF,IAAI,CAAC,QAAQ,EAAE,EAChG,EAAE,KAAK,EAAE,CACV,CAAC;QACJ,CAAC;IACH,CAAC;CACF"}

package/dist/providers/vertex-embed.d.ts

@@ -0,0 +1,38 @@
+/**
+ * VertexEmbedProvider — Vertex AI text-embedding-004 via @google-cloud/aiplatform.
+ *
+ * Uses PredictionServiceClient (NOT @google-cloud/vertexai, which only supports
+ * generative models). Returns 768-dimension vectors by default.
+ */
+import type { EmbedProvider } from '../core/embed.js';
+export type VertexEmbedTaskType = 'RETRIEVAL_DOCUMENT' | 'RETRIEVAL_QUERY' | 'SEMANTIC_SIMILARITY' | 'CLASSIFICATION' | 'CLUSTERING';
+export interface VertexEmbedOptions {
+    /** GCP project ID. Falls back to GOOGLE_CLOUD_PROJECT env var. */
+    projectId?: string;
+    /** GCP region (default: us-central1). */
+    location?: string;
+    /** Embedding model name (default: text-embedding-004). */
+    model?: string;
+    /** Output dimensionality (default: 768). */
+    dimensions?: number;
+    /** Max concurrent requests per batch (default: 5). */
+    batchConcurrency?: number;
+    /** Default task type for embeddings (default: SEMANTIC_SIMILARITY). */
+    taskType?: VertexEmbedTaskType;
+}
+export declare class VertexEmbedProvider implements EmbedProvider {
+    readonly name = "vertex";
+    readonly dimensions: number;
+    private readonly projectId;
+    private readonly location;
+    private readonly model;
+    private readonly batchConcurrency;
+    private readonly taskType;
+    private readonly client;
+    private readonly helpers;
+    constructor(options: VertexEmbedOptions, client: import('@google-cloud/aiplatform').PredictionServiceClient, helpersModule: typeof import('@google-cloud/aiplatform').helpers);
+    private get endpoint();
+    embed(text: string): Promise<number[]>;
+    embedBatch(texts: string[]): Promise<number[][]>;
+}
+//# sourceMappingURL=vertex-embed.d.ts.map

package/dist/providers/vertex-embed.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"vertex-embed.d.ts","sourceRoot":"","sources":["../../src/providers/vertex-embed.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,kBAAkB,CAAC;AAEtD,MAAM,MAAM,mBAAmB,GAC3B,oBAAoB,GACpB,iBAAiB,GACjB,qBAAqB,GACrB,gBAAgB,GAChB,YAAY,CAAC;AAEjB,MAAM,WAAW,kBAAkB;IACjC,kEAAkE;IAClE,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,yCAAyC;IACzC,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,0DAA0D;IAC1D,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,4CAA4C;IAC5C,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,sDAAsD;IACtD,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,uEAAuE;IACvE,QAAQ,CAAC,EAAE,mBAAmB,CAAC;CAChC;AAED,qBAAa,mBAAoB,YAAW,aAAa;IACvD,QAAQ,CAAC,IAAI,YAAY;IACzB,QAAQ,CAAC,UAAU,EAAE,MAAM,CAAC;IAE5B,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAS;IACnC,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAS;IAClC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAS;IAC/B,OAAO,CAAC,QAAQ,CAAC,gBAAgB,CAAS;IAC1C,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAsB;IAE/C,OAAO,CAAC,QAAQ,CAAC,MAAM,CAA6D;IACpF,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAoD;gBAG1E,OAAO,EAAE,kBAAkB,EAC3B,MAAM,EAAE,OAAO,0BAA0B,EAAE,uBAAuB,EAClE,aAAa,EAAE,cAAc,0BAA0B,EAAE,OAAO;IAiBlE,OAAO,KAAK,QAAQ,GAEnB;IAEK,KAAK,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;IA+BtC,UAAU,CAAC,KAAK,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC;CAgBvD"}

package/dist/providers/vertex-embed.js

@@ -0,0 +1,69 @@
+/**
+ * VertexEmbedProvider — Vertex AI text-embedding-004 via @google-cloud/aiplatform.
+ *
+ * Uses PredictionServiceClient (NOT @google-cloud/vertexai, which only supports
+ * generative models). Returns 768-dimension vectors by default.
+ */
+export class VertexEmbedProvider {
+    name = 'vertex';
+    dimensions;
+    projectId;
+    location;
+    model;
+    batchConcurrency;
+    taskType;
+    client;
+    helpers;
+    constructor(options, client, helpersModule) {
+        this.projectId = options.projectId ?? process.env['GOOGLE_CLOUD_PROJECT'] ?? '';
+        this.location = options.location ?? 'us-central1';
+        this.model = options.model ?? 'text-embedding-004';
+        this.dimensions = options.dimensions ?? 768;
+        this.batchConcurrency = options.batchConcurrency ?? 5;
+        this.taskType = options.taskType ?? 'SEMANTIC_SIMILARITY';
+        if (!this.projectId) {
+            throw new Error('VertexEmbedProvider: projectId is required (config or GOOGLE_CLOUD_PROJECT env)');
+        }
+        this.client = client;
+        this.helpers = helpersModule;
+    }
+    get endpoint() {
+        return `projects/${this.projectId}/locations/${this.location}/publishers/google/models/${this.model}`;
+    }
+    async embed(text) {
+        const instance = this.helpers.toValue({
+            content: text,
+            task_type: this.taskType,
+        });
+        const parameters = this.helpers.toValue({
+            outputDimensionality: this.dimensions,
+        });
+        const [response] = await this.client.predict({
+            endpoint: this.endpoint,
+            instances: [instance],
+            parameters,
+        });
+        const prediction = response.predictions?.[0];
+        if (!prediction)
+            throw new Error(`VertexEmbedProvider: no prediction returned for: "${text.slice(0, 50)}"`);
+        const values = prediction.structValue?.fields?.embeddings?.structValue?.fields?.values?.listValue?.values?.map((v) => v.numberValue ?? 0);
+        if (!values || values.length === 0) {
+            throw new Error(`VertexEmbedProvider: empty embedding for: "${text.slice(0, 50)}"`);
+        }
+        return values;
+    }
+    async embedBatch(texts) {
+        const results = [];
+        for (let i = 0; i < texts.length; i += this.batchConcurrency) {
+            const chunk = texts.slice(i, i + this.batchConcurrency);
+            const embeddings = await Promise.all(chunk.map((t) => this.embed(t)));
+            results.push(...embeddings);
+            // Brief pause between batches to respect rate limits
+            if (i + this.batchConcurrency < texts.length) {
+                await new Promise((r) => setTimeout(r, 250));
+            }
+        }
+        return results;
+    }
+}
+//# sourceMappingURL=vertex-embed.js.map

package/dist/providers/vertex-embed.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"vertex-embed.js","sourceRoot":"","sources":["../../src/providers/vertex-embed.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AA0BH,MAAM,OAAO,mBAAmB;IACrB,IAAI,GAAG,QAAQ,CAAC;IAChB,UAAU,CAAS;IAEX,SAAS,CAAS;IAClB,QAAQ,CAAS;IACjB,KAAK,CAAS;IACd,gBAAgB,CAAS;IACzB,QAAQ,CAAsB;IAE9B,MAAM,CAA6D;IACnE,OAAO,CAAoD;IAE5E,YACE,OAA2B,EAC3B,MAAkE,EAClE,aAAgE;QAEhE,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,SAAS,IAAI,OAAO,CAAC,GAAG,CAAC,sBAAsB,CAAC,IAAI,EAAE,CAAC;QAChF,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC,QAAQ,IAAI,aAAa,CAAC;QAClD,IAAI,CAAC,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,oBAAoB,CAAC;QACnD,IAAI,CAAC,UAAU,GAAG,OAAO,CAAC,UAAU,IAAI,GAAG,CAAC;QAC5C,IAAI,CAAC,gBAAgB,GAAG,OAAO,CAAC,gBAAgB,IAAI,CAAC,CAAC;QACtD,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC,QAAQ,IAAI,qBAAqB,CAAC;QAE1D,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;YACpB,MAAM,IAAI,KAAK,CAAC,iFAAiF,CAAC,CAAC;QACrG,CAAC;QAED,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,OAAO,GAAG,aAAa,CAAC;IAC/B,CAAC;IAED,IAAY,QAAQ;QAClB,OAAO,YAAY,IAAI,CAAC,SAAS,cAAc,IAAI,CAAC,QAAQ,6BAA6B,IAAI,CAAC,KAAK,EAAE,CAAC;IACxG,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,IAAY;QACtB,MAAM,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC;YACpC,OAAO,EAAE,IAAI;YACb,SAAS,EAAE,IAAI,CAAC,QAAQ;SACzB,CAAC,CAAC;QAEH,MAAM,UAAU,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC;YACtC,oBAAoB,EAAE,IAAI,CAAC,UAAU;SACtC,CAAC,CAAC;QAEH,MAAM,CAAC,QAAQ,CAAC,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC;YAC3C,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,SAAS,EAAE,CAAC,QAAS,CAAC;YACtB,UAAU;SACX,CAAC,CAAC;QAEH,MAAM,UAAU,GAAG,QAAQ,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,CAAC;QAC7C,IAAI,CAAC,UAAU;YAAE,MAAM,IAAI,KAAK,CAAC,qDAAqD,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC;QAE5G,MAAM,MAAM,GACV,UAAU,CAAC,WAAW,EAAE,MAAM,EAAE,UAAU,EAAE,WAAW,EAAE,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,GAAG,CAC7F,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,WAAW,IAAI,CAAC,CAC1B,CAAC;QAEJ,IAAI,CAAC,MAAM,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACnC,MAAM,IAAI,KAAK,CAAC,8CAA8C,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC;QACtF,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,KAAe;QAC9B,MAAM,OAAO,GAAe,EAAE,CAAC;QAE/B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,IAAI,IAAI,CAAC,gBAAgB,EAAE,CAAC;YAC7D,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,gBAAgB,CAAC,CAAC;YACxD,MAAM,UAAU,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YACtE,OAAO,CAAC,IAAI,CAAC,GAAG,UAAU,CAAC,CAAC;YAE5B,qDAAqD;YACrD,IAAI,CAAC,GAAG,IAAI,CAAC,gBAAgB,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC;gBAC7C,MAAM,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,UAAU,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC;YAC/C,CAAC;QACH,CAAC;QAED,OAAO,OAAO,CAAC;IACjB,CAAC;CACF"}

package/dist/providers/vertex-llm.d.ts

@@ -0,0 +1,25 @@
+/**
+ * VertexLLMProvider — Vertex AI Gemini via @google-cloud/vertexai.
+ *
+ * Generalized provider with no idapixl-specific dependencies.
+ * Supports systemPrompt, schema instructions, and JSON fence stripping.
+ */
+import type { LLMProvider, GenerateOptions, GenerateJSONOptions } from '../core/llm.js';
+export interface VertexLLMOptions {
+    /** GCP project ID. Falls back to GOOGLE_CLOUD_PROJECT env var. */
+    projectId?: string;
+    /** GCP region (default: us-central1). */
+    location?: string;
+    /** Gemini model ID (default: gemini-2.5-flash). */
+    model?: string;
+}
+export declare class VertexLLMProvider implements LLMProvider {
+    readonly name = "vertex-gemini";
+    readonly modelId: string;
+    private readonly projectId;
+    private readonly vertexAI;
+    constructor(options: VertexLLMOptions, vertexAI: import('@google-cloud/vertexai').VertexAI);
+    generate(prompt: string, options?: GenerateOptions): Promise<string>;
+    generateJSON<T>(prompt: string, options?: GenerateJSONOptions): Promise<T>;
+}
+//# sourceMappingURL=vertex-llm.d.ts.map

package/dist/providers/vertex-llm.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"vertex-llm.d.ts","sourceRoot":"","sources":["../../src/providers/vertex-llm.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,KAAK,EAAE,WAAW,EAAE,eAAe,EAAE,mBAAmB,EAAE,MAAM,gBAAgB,CAAC;AAExF,MAAM,WAAW,gBAAgB;IAC/B,kEAAkE;IAClE,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,yCAAyC;IACzC,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,mDAAmD;IACnD,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAaD,qBAAa,iBAAkB,YAAW,WAAW;IACnD,QAAQ,CAAC,IAAI,mBAAmB;IAChC,QAAQ,CAAC,OAAO,EAAE,MAAM,CAAC;IAEzB,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAS;IACnC,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAA4C;gBAGnE,OAAO,EAAE,gBAAgB,EACzB,QAAQ,EAAE,OAAO,wBAAwB,EAAE,QAAQ;IAY/C,QAAQ,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,eAAe,GAAG,OAAO,CAAC,MAAM,CAAC;IAiBpE,YAAY,CAAC,CAAC,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,mBAAmB,GAAG,OAAO,CAAC,CAAC,CAAC;CAiCjF"}

package/dist/providers/vertex-llm.js

@@ -0,0 +1,75 @@
+/**
+ * VertexLLMProvider — Vertex AI Gemini via @google-cloud/vertexai.
+ *
+ * Generalized provider with no idapixl-specific dependencies.
+ * Supports systemPrompt, schema instructions, and JSON fence stripping.
+ */
+/**
+ * Strip common code fence wrappers from LLM JSON output.
+ * Models often return ```json ... ``` or ``` ... ``` around JSON.
+ */
+function stripJsonFences(text) {
+    const trimmed = text.trim();
+    // Match ```json ... ``` or ``` ... ```
+    const fenceMatch = trimmed.match(/^```(?:json)?\s*\n?([\s\S]*?)\n?\s*```$/);
+    return fenceMatch ? fenceMatch[1].trim() : trimmed;
+}
+export class VertexLLMProvider {
+    name = 'vertex-gemini';
+    modelId;
+    projectId;
+    vertexAI;
+    constructor(options, vertexAI) {
+        this.projectId = options.projectId ?? process.env['GOOGLE_CLOUD_PROJECT'] ?? '';
+        this.modelId = options.model ?? 'gemini-2.5-flash';
+        if (!this.projectId) {
+            throw new Error('VertexLLMProvider: projectId is required (config or GOOGLE_CLOUD_PROJECT env)');
+        }
+        this.vertexAI = vertexAI;
+    }
+    async generate(prompt, options) {
+        const model = this.vertexAI.preview.getGenerativeModel({
+            model: this.modelId,
+            generationConfig: {
+                temperature: options?.temperature ?? 0.2,
+                maxOutputTokens: options?.maxTokens ?? 1024,
+            },
+            ...(options?.systemPrompt ? { systemInstruction: options.systemPrompt } : {}),
+        });
+        const result = await model.generateContent(prompt);
+        const text = result.response.candidates?.[0]?.content?.parts?.[0]?.text;
+        if (!text)
+            throw new Error('VertexLLMProvider: empty response');
+        return text.trim();
+    }
+    async generateJSON(prompt, options) {
+        // Build system prompt with optional schema instruction
+        let systemPrompt = options?.systemPrompt ?? '';
+        if (options?.schema) {
+            const schemaInstruction = `Respond with JSON matching this schema: ${JSON.stringify(options.schema)}`;
+            systemPrompt = systemPrompt ? `${systemPrompt}\n\n${schemaInstruction}` : schemaInstruction;
+        }
+        const model = this.vertexAI.preview.getGenerativeModel({
+            model: this.modelId,
+            generationConfig: {
+                temperature: options?.temperature ?? 0.2,
+                maxOutputTokens: options?.maxTokens ?? 2048,
+                responseMimeType: 'application/json',
+            },
+            ...(systemPrompt ? { systemInstruction: systemPrompt } : {}),
+        });
+        const result = await model.generateContent(prompt);
+        const text = result.response.candidates?.[0]?.content?.parts?.[0]?.text;
+        if (!text)
+            throw new Error('VertexLLMProvider: empty response');
+        const cleaned = stripJsonFences(text);
+        try {
+            return JSON.parse(cleaned);
+        }
+        catch (err) {
+            const parseError = err instanceof Error ? err.message : String(err);
+            throw new Error(`VertexLLMProvider: failed to parse JSON response: ${parseError}\nRaw output (first 500 chars): ${text.slice(0, 500)}`, { cause: err });
+        }
+    }
+}
+//# sourceMappingURL=vertex-llm.js.map

package/dist/providers/vertex-llm.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"vertex-llm.js","sourceRoot":"","sources":["../../src/providers/vertex-llm.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAaH;;;GAGG;AACH,SAAS,eAAe,CAAC,IAAY;IACnC,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;IAC5B,uCAAuC;IACvC,MAAM,UAAU,GAAG,OAAO,CAAC,KAAK,CAAC,yCAAyC,CAAC,CAAC;IAC5E,OAAO,UAAU,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC;AACrD,CAAC;AAED,MAAM,OAAO,iBAAiB;IACnB,IAAI,GAAG,eAAe,CAAC;IACvB,OAAO,CAAS;IAER,SAAS,CAAS;IAClB,QAAQ,CAA4C;IAErE,YACE,OAAyB,EACzB,QAAmD;QAEnD,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,SAAS,IAAI,OAAO,CAAC,GAAG,CAAC,sBAAsB,CAAC,IAAI,EAAE,CAAC;QAChF,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC,KAAK,IAAI,kBAAkB,CAAC;QAEnD,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;YACpB,MAAM,IAAI,KAAK,CAAC,+EAA+E,CAAC,CAAC;QACnG,CAAC;QAED,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;IAC3B,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,MAAc,EAAE,OAAyB;QACtD,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,kBAAkB,CAAC;YACrD,KAAK,EAAE,IAAI,CAAC,OAAO;YACnB,gBAAgB,EAAE;gBAChB,WAAW,EAAE,OAAO,EAAE,WAAW,IAAI,GAAG;gBACxC,eAAe,EAAE,OAAO,EAAE,SAAS,IAAI,IAAI;aAC5C;YACD,GAAG,CAAC,OAAO,EAAE,YAAY,CAAC,CAAC,CAAC,EAAE,iBAAiB,EAAE,OAAO,CAAC,YAAY,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SAC9E,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC;QACnD,MAAM,IAAI,GAAG,MAAM,CAAC,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC;QACxE,IAAI,CAAC,IAAI;YAAE,MAAM,IAAI,KAAK,CAAC,mCAAmC,CAAC,CAAC;QAEhE,OAAO,IAAI,CAAC,IAAI,EAAE,CAAC;IACrB,CAAC;IAED,KAAK,CAAC,YAAY,CAAI,MAAc,EAAE,OAA6B;QACjE,uDAAuD;QACvD,IAAI,YAAY,GAAG,OAAO,EAAE,YAAY,IAAI,EAAE,CAAC;QAC/C,IAAI,OAAO,EAAE,MAAM,EAAE,CAAC;YACpB,MAAM,iBAAiB,GAAG,2CAA2C,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;YACtG,YAAY,GAAG,YAAY,CAAC,CAAC,CAAC,GAAG,YAAY,OAAO,iBAAiB,EAAE,CAAC,CAAC,CAAC,iBAAiB,CAAC;QAC9F,CAAC;QAED,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,kBAAkB,CAAC;YACrD,KAAK,EAAE,IAAI,CAAC,OAAO;YACnB,gBAAgB,EAAE;gBAChB,WAAW,EAAE,OAAO,EAAE,WAAW,IAAI,GAAG;gBACxC,eAAe,EAAE,OAAO,EAAE,SAAS,IAAI,IAAI;gBAC3C,gBAAgB,EAAE,kBAAkB;aACrC;YACD,GAAG,CAAC,YAAY,CAAC,CAAC,CAAC,EAAE,iBAAiB,EAAE,YAAY,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SAC7D,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC;QACnD,MAAM,IAAI,GAAG,MAAM,CAAC,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC;QACxE,IAAI,CAAC,IAAI;YAAE,MAAM,IAAI,KAAK,CAAC,mCAAmC,CAAC,CAAC;QAEhE,MAAM,OAAO,GAAG,eAAe,CAAC,IAAI,CAAC,CAAC;QACtC,IAAI,CAAC;YACH,OAAO,IAAI,CAAC,KAAK,CAAC,OAAO,CAAM,CAAC;QAClC,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,UAAU,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YACpE,MAAM,IAAI,KAAK,CACb,qDAAqD,UAAU,mCAAmC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,EACtH,EAAE,KAAK,EAAE,GAAG,EAAE,CACf,CAAC;QACJ,CAAC;IACH,CAAC;CACF"}

package/dist/stores/firestore.d.ts

@@ -0,0 +1,59 @@
+/**
+ * FirestoreCortexStore — cloud Firestore-backed implementation of CortexStore.
+ *
+ * Uses firebase-admin for Firestore access. Vector search uses Firestore native
+ * findNearest() for production-grade similarity search.
+ * Dates stored as Firestore Timestamps. Embeddings stored as VectorValue.
+ *
+ * This is the cloud counterpart to SqliteCortexStore. Both implement the same
+ * CortexStore interface, so engines work identically on either backend.
+ */
+import type { Firestore, FieldValue as FieldValueType } from '@google-cloud/firestore';
+import type { CortexStore } from '../core/store.js';
+import type { Memory, Observation, Edge, OpsEntry, OpsFilters, Signal, BeliefEntry, SearchResult, FSRSData, QueryFilter } from '../core/types.js';
+export interface FirestoreStoreOptions {
+    /** GCP project ID (required if not using Application Default Credentials) */
+    projectId?: string;
+    /** Firestore database ID (default: '(default)') */
+    databaseId?: string;
+    /** Namespace prefix for collection names (for multi-tenant isolation) */
+    namespace?: string;
+}
+export declare class FirestoreCortexStore implements CortexStore {
+    private db;
+    private ns;
+    constructor(db: Firestore, namespace?: string, fieldValue?: typeof FieldValueType);
+    /** Inject the host's FieldValue to avoid ESM dual-package class identity issues. */
+    static setFieldValue(fv: typeof FieldValueType): void;
+    /** Create a FirestoreCortexStore from firebase-admin Firestore instance. */
+    static fromFirestore(db: Firestore, namespace?: string, fieldValue?: typeof FieldValueType): FirestoreCortexStore;
+    /** Get the prefixed collection name. */
+    private col;
+    putMemory(memory: Omit<Memory, 'id'>): Promise<string>;
+    getMemory(id: string): Promise<Memory | null>;
+    updateMemory(id: string, updates: Partial<Omit<Memory, 'id'>>): Promise<void>;
+    findNearest(embedding: number[], limit: number): Promise<SearchResult[]>;
+    touchMemory(id: string, fsrsUpdates: Partial<FSRSData>): Promise<void>;
+    getAllMemories(): Promise<Memory[]>;
+    putObservation(obs: Omit<Observation, 'id'>): Promise<string>;
+    getUnprocessedObservations(limit: number): Promise<Observation[]>;
+    markObservationProcessed(id: string): Promise<void>;
+    putEdge(edge: Omit<Edge, 'id'>): Promise<string>;
+    getEdgesFrom(memoryId: string): Promise<Edge[]>;
+    getEdgesForMemories(memoryIds: string[]): Promise<Edge[]>;
+    appendOps(entry: Omit<OpsEntry, 'id'>): Promise<string>;
+    queryOps(filters: OpsFilters): Promise<OpsEntry[]>;
+    updateOps(id: string, updates: Partial<Omit<OpsEntry, 'id'>>): Promise<void>;
+    putSignal(signal: Omit<Signal, 'id'>): Promise<string>;
+    putBelief(entry: Omit<BeliefEntry, 'id'>): Promise<string>;
+    getBeliefHistory(conceptId: string): Promise<BeliefEntry[]>;
+    put(collection: string, doc: Record<string, unknown>): Promise<string>;
+    get(collection: string, id: string): Promise<Record<string, unknown> | null>;
+    update(collection: string, id: string, updates: Record<string, unknown>): Promise<void>;
+    query(collection: string, filters: QueryFilter[], options?: {
+        limit?: number;
+        orderBy?: string;
+        orderDir?: 'asc' | 'desc';
+    }): Promise<Record<string, unknown>[]>;
+}
+//# sourceMappingURL=firestore.d.ts.map

package/dist/stores/firestore.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"firestore.d.ts","sourceRoot":"","sources":["../../src/stores/firestore.ts"],"names":[],"mappings":"AAAA;;;;;;;;;GASG;AAEH,OAAO,KAAK,EAAE,SAAS,EAAqC,UAAU,IAAI,cAAc,EAAE,MAAM,yBAAyB,CAAC;AAC1H,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AACpD,OAAO,KAAK,EACV,MAAM,EAEN,WAAW,EACX,IAAI,EACJ,QAAQ,EACR,UAAU,EACV,MAAM,EACN,WAAW,EACX,YAAY,EACZ,QAAQ,EACR,WAAW,EAEZ,MAAM,kBAAkB,CAAC;AA2L1B,MAAM,WAAW,qBAAqB;IACpC,6EAA6E;IAC7E,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,mDAAmD;IACnD,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,yEAAyE;IACzE,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED,qBAAa,oBAAqB,YAAW,WAAW;IACtD,OAAO,CAAC,EAAE,CAAY;IACtB,OAAO,CAAC,EAAE,CAAS;gBAEP,EAAE,EAAE,SAAS,EAAE,SAAS,CAAC,EAAE,MAAM,EAAE,UAAU,CAAC,EAAE,OAAO,cAAc;IAQjF,oFAAoF;IACpF,MAAM,CAAC,aAAa,CAAC,EAAE,EAAE,OAAO,cAAc,GAAG,IAAI;IAIrD,4EAA4E;IAC5E,MAAM,CAAC,aAAa,CAAC,EAAE,EAAE,SAAS,EAAE,SAAS,CAAC,EAAE,MAAM,EAAE,UAAU,CAAC,EAAE,OAAO,cAAc,GAAG,oBAAoB;IAIjH,wCAAwC;IACxC,OAAO,CAAC,GAAG;IAOL,SAAS,CAAC,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC;IA8BtD,SAAS,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC;IAM7C,YAAY,CAAC,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC;IAgC7E,WAAW,CAAC,SAAS,EAAE,MAAM,EAAE,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;IAuBxE,WAAW,CAAC,EAAE,EAAE,MAAM,EAAE,WAAW,EAAE,OAAO,CAAC,QAAQ,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC;IAmBtE,cAAc,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;IAOnC,cAAc,CAAC,GAAG,EAAE,IAAI,CAAC,WAAW,EAAE,IAAI,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC;IAkB7D,0BAA0B,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;IAUjE,wBAAwB,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IASnD,OAAO,CAAC,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC;IAahD,YAAY,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,EAAE,CAAC;IAQ/C,mBAAmB,CAAC,SAAS,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,IAAI,EAAE,CAAC;IA8BzD,SAAS,CAAC,KAAK,EAAE,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC;IAiBvD,QAAQ,CAAC,OAAO,EAAE,UAAU,GAAG,OAAO,CAAC,QAAQ,EAAE,CAAC;IAmBlD,SAAS,CAAC,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,CAAC,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC;IAiB5E,SAAS,CAAC,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC;IAgBtD,SAAS,CAAC,KAAK,EAAE,IAAI,CAAC,WAAW,EAAE,IAAI,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC;IAY1D,gBAAgB,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;IAW3D,GAAG,CAAC,UAAU,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC;IAQtE,GAAG,CAAC,UAAU,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,IAAI,CAAC;IAM5E,MAAM,CAAC,UAAU,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC;IAKvF,KAAK,CACT,UAAU,EAAE,MAAM,EAClB,OAAO,EAAE,WAAW,EAAE,EACtB,OAAO,CAAC,EAAE;QAAE,KAAK,CAAC,EAAE,MAAM,CAAC;QAAC,OAAO,CAAC,EAAE,MAAM,CAAC;QAAC,QAAQ,CAAC,EAAE,KAAK,GAAG,MAAM,CAAA;KAAE,GACxE,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAAE,CAAC;CAkBtC"}