@promptev/context-engine 0.0.1 → 0.0.2

package/dist/{config-CdlSkKgV.d.ts → config-BODDdXJ7.d.ts}

@@ -2,39 +2,51 @@ import { z } from 'zod';
 import { R as RedactionPolicy, b as RedactionRuleInit } from './redaction-BqD_DEUQ.js';
 
 declare const embeddingSchema: z.ZodObject<{
-    provider: z.ZodEnum<["openai", "azure_openai", "gemini", "voyage", "cohere", "custom"]>;
+    provider: z.ZodEnum<["openai", "azure_openai", "gemini", "vertex_ai", "voyage", "cohere", "custom"]>;
     model: z.ZodString;
     dim: z.ZodDefault<z.ZodOptional<z.ZodNullable<z.ZodNumber>>>;
     apiKey: z.ZodDefault<z.ZodOptional<z.ZodNullable<z.ZodString>>>;
     baseUrl: z.ZodDefault<z.ZodOptional<z.ZodNullable<z.ZodString>>>;
+    project: z.ZodOptional<z.ZodNullable<z.ZodString>>;
+    location: z.ZodOptional<z.ZodNullable<z.ZodString>>;
 }, "strip", z.ZodTypeAny, {
-    provider: "custom" | "openai" | "azure_openai" | "gemini" | "voyage" | "cohere";
+    provider: "custom" | "openai" | "azure_openai" | "gemini" | "vertex_ai" | "voyage" | "cohere";
     model: string;
     dim: number | null;
     apiKey: string | null;
     baseUrl: string | null;
+    project?: string | null | undefined;
+    location?: string | null | undefined;
 }, {
-    provider: "custom" | "openai" | "azure_openai" | "gemini" | "voyage" | "cohere";
+    provider: "custom" | "openai" | "azure_openai" | "gemini" | "vertex_ai" | "voyage" | "cohere";
     model: string;
     dim?: number | null | undefined;
     apiKey?: string | null | undefined;
     baseUrl?: string | null | undefined;
+    project?: string | null | undefined;
+    location?: string | null | undefined;
 }>;
 declare const llmSchema: z.ZodObject<{
-    provider: z.ZodEnum<["anthropic", "openai", "azure_openai", "gemini", "bedrock", "custom"]>;
+    provider: z.ZodEnum<["anthropic", "openai", "azure_openai", "gemini", "vertex_ai", "bedrock", "custom"]>;
     model: z.ZodString;
     apiKey: z.ZodDefault<z.ZodOptional<z.ZodNullable<z.ZodString>>>;
     baseUrl: z.ZodDefault<z.ZodOptional<z.ZodNullable<z.ZodString>>>;
+    project: z.ZodOptional<z.ZodNullable<z.ZodString>>;
+    location: z.ZodOptional<z.ZodNullable<z.ZodString>>;
 }, "strip", z.ZodTypeAny, {
-    provider: "custom" | "openai" | "azure_openai" | "gemini" | "anthropic" | "bedrock";
+    provider: "custom" | "openai" | "azure_openai" | "gemini" | "vertex_ai" | "anthropic" | "bedrock";
     model: string;
     apiKey: string | null;
     baseUrl: string | null;
+    project?: string | null | undefined;
+    location?: string | null | undefined;
 }, {
-    provider: "custom" | "openai" | "azure_openai" | "gemini" | "anthropic" | "bedrock";
+    provider: "custom" | "openai" | "azure_openai" | "gemini" | "vertex_ai" | "anthropic" | "bedrock";
     model: string;
     apiKey?: string | null | undefined;
     baseUrl?: string | null | undefined;
+    project?: string | null | undefined;
+    location?: string | null | undefined;
 }>;
 declare const graphSchema: z.ZodObject<{
     enabled: z.ZodDefault<z.ZodBoolean>;
@@ -43,20 +55,26 @@ declare const graphSchema: z.ZodObject<{
     neo4jPassword: z.ZodDefault<z.ZodOptional<z.ZodNullable<z.ZodString>>>;
     neo4jDatabase: z.ZodDefault<z.ZodString>;
     extractionLlm: z.ZodDefault<z.ZodOptional<z.ZodNullable<z.ZodObject<{
-        provider: z.ZodEnum<["anthropic", "openai", "azure_openai", "gemini", "bedrock", "custom"]>;
+        provider: z.ZodEnum<["anthropic", "openai", "azure_openai", "gemini", "vertex_ai", "bedrock", "custom"]>;
         model: z.ZodString;
         apiKey: z.ZodDefault<z.ZodOptional<z.ZodNullable<z.ZodString>>>;
         baseUrl: z.ZodDefault<z.ZodOptional<z.ZodNullable<z.ZodString>>>;
+        project: z.ZodOptional<z.ZodNullable<z.ZodString>>;
+        location: z.ZodOptional<z.ZodNullable<z.ZodString>>;
     }, "strip", z.ZodTypeAny, {
-        provider: "custom" | "openai" | "azure_openai" | "gemini" | "anthropic" | "bedrock";
+        provider: "custom" | "openai" | "azure_openai" | "gemini" | "vertex_ai" | "anthropic" | "bedrock";
         model: string;
         apiKey: string | null;
         baseUrl: string | null;
+        project?: string | null | undefined;
+        location?: string | null | undefined;
     }, {
-        provider: "custom" | "openai" | "azure_openai" | "gemini" | "anthropic" | "bedrock";
+        provider: "custom" | "openai" | "azure_openai" | "gemini" | "vertex_ai" | "anthropic" | "bedrock";
         model: string;
         apiKey?: string | null | undefined;
         baseUrl?: string | null | undefined;
+        project?: string | null | undefined;
+        location?: string | null | undefined;
     }>>>>;
 }, "strip", z.ZodTypeAny, {
     enabled: boolean;
@@ -65,10 +83,12 @@ declare const graphSchema: z.ZodObject<{
     neo4jPassword: string | null;
     neo4jDatabase: string;
     extractionLlm: {
-        provider: "custom" | "openai" | "azure_openai" | "gemini" | "anthropic" | "bedrock";
+        provider: "custom" | "openai" | "azure_openai" | "gemini" | "vertex_ai" | "anthropic" | "bedrock";
         model: string;
         apiKey: string | null;
         baseUrl: string | null;
+        project?: string | null | undefined;
+        location?: string | null | undefined;
     } | null;
 }, {
     enabled?: boolean | undefined;
@@ -77,10 +97,12 @@ declare const graphSchema: z.ZodObject<{
     neo4jPassword?: string | null | undefined;
     neo4jDatabase?: string | undefined;
     extractionLlm?: {
-        provider: "custom" | "openai" | "azure_openai" | "gemini" | "anthropic" | "bedrock";
+        provider: "custom" | "openai" | "azure_openai" | "gemini" | "vertex_ai" | "anthropic" | "bedrock";
         model: string;
         apiKey?: string | null | undefined;
         baseUrl?: string | null | undefined;
+        project?: string | null | undefined;
+        location?: string | null | undefined;
     } | null | undefined;
 }>;
 declare const rerankerSchema: z.ZodObject<{

package/dist/{config-CNnASw5X.d.cts → config-C5RZ00W6.d.cts}

@@ -2,39 +2,51 @@ import { z } from 'zod';
 import { R as RedactionPolicy, b as RedactionRuleInit } from './redaction-BqD_DEUQ.cjs';
 
 declare const embeddingSchema: z.ZodObject<{
-    provider: z.ZodEnum<["openai", "azure_openai", "gemini", "voyage", "cohere", "custom"]>;
+    provider: z.ZodEnum<["openai", "azure_openai", "gemini", "vertex_ai", "voyage", "cohere", "custom"]>;
     model: z.ZodString;
     dim: z.ZodDefault<z.ZodOptional<z.ZodNullable<z.ZodNumber>>>;
     apiKey: z.ZodDefault<z.ZodOptional<z.ZodNullable<z.ZodString>>>;
     baseUrl: z.ZodDefault<z.ZodOptional<z.ZodNullable<z.ZodString>>>;
+    project: z.ZodOptional<z.ZodNullable<z.ZodString>>;
+    location: z.ZodOptional<z.ZodNullable<z.ZodString>>;
 }, "strip", z.ZodTypeAny, {
-    provider: "custom" | "openai" | "azure_openai" | "gemini" | "voyage" | "cohere";
+    provider: "custom" | "openai" | "azure_openai" | "gemini" | "vertex_ai" | "voyage" | "cohere";
     model: string;
     dim: number | null;
     apiKey: string | null;
     baseUrl: string | null;
+    project?: string | null | undefined;
+    location?: string | null | undefined;
 }, {
-    provider: "custom" | "openai" | "azure_openai" | "gemini" | "voyage" | "cohere";
+    provider: "custom" | "openai" | "azure_openai" | "gemini" | "vertex_ai" | "voyage" | "cohere";
     model: string;
     dim?: number | null | undefined;
     apiKey?: string | null | undefined;
     baseUrl?: string | null | undefined;
+    project?: string | null | undefined;
+    location?: string | null | undefined;
 }>;
 declare const llmSchema: z.ZodObject<{
-    provider: z.ZodEnum<["anthropic", "openai", "azure_openai", "gemini", "bedrock", "custom"]>;
+    provider: z.ZodEnum<["anthropic", "openai", "azure_openai", "gemini", "vertex_ai", "bedrock", "custom"]>;
     model: z.ZodString;
     apiKey: z.ZodDefault<z.ZodOptional<z.ZodNullable<z.ZodString>>>;
     baseUrl: z.ZodDefault<z.ZodOptional<z.ZodNullable<z.ZodString>>>;
+    project: z.ZodOptional<z.ZodNullable<z.ZodString>>;
+    location: z.ZodOptional<z.ZodNullable<z.ZodString>>;
 }, "strip", z.ZodTypeAny, {
-    provider: "custom" | "openai" | "azure_openai" | "gemini" | "anthropic" | "bedrock";
+    provider: "custom" | "openai" | "azure_openai" | "gemini" | "vertex_ai" | "anthropic" | "bedrock";
     model: string;
     apiKey: string | null;
     baseUrl: string | null;
+    project?: string | null | undefined;
+    location?: string | null | undefined;
 }, {
-    provider: "custom" | "openai" | "azure_openai" | "gemini" | "anthropic" | "bedrock";
+    provider: "custom" | "openai" | "azure_openai" | "gemini" | "vertex_ai" | "anthropic" | "bedrock";
     model: string;
     apiKey?: string | null | undefined;
     baseUrl?: string | null | undefined;
+    project?: string | null | undefined;
+    location?: string | null | undefined;
 }>;
 declare const graphSchema: z.ZodObject<{
     enabled: z.ZodDefault<z.ZodBoolean>;
@@ -43,20 +55,26 @@ declare const graphSchema: z.ZodObject<{
     neo4jPassword: z.ZodDefault<z.ZodOptional<z.ZodNullable<z.ZodString>>>;
     neo4jDatabase: z.ZodDefault<z.ZodString>;
     extractionLlm: z.ZodDefault<z.ZodOptional<z.ZodNullable<z.ZodObject<{
-        provider: z.ZodEnum<["anthropic", "openai", "azure_openai", "gemini", "bedrock", "custom"]>;
+        provider: z.ZodEnum<["anthropic", "openai", "azure_openai", "gemini", "vertex_ai", "bedrock", "custom"]>;
         model: z.ZodString;
         apiKey: z.ZodDefault<z.ZodOptional<z.ZodNullable<z.ZodString>>>;
         baseUrl: z.ZodDefault<z.ZodOptional<z.ZodNullable<z.ZodString>>>;
+        project: z.ZodOptional<z.ZodNullable<z.ZodString>>;
+        location: z.ZodOptional<z.ZodNullable<z.ZodString>>;
     }, "strip", z.ZodTypeAny, {
-        provider: "custom" | "openai" | "azure_openai" | "gemini" | "anthropic" | "bedrock";
+        provider: "custom" | "openai" | "azure_openai" | "gemini" | "vertex_ai" | "anthropic" | "bedrock";
         model: string;
         apiKey: string | null;
         baseUrl: string | null;
+        project?: string | null | undefined;
+        location?: string | null | undefined;
     }, {
-        provider: "custom" | "openai" | "azure_openai" | "gemini" | "anthropic" | "bedrock";
+        provider: "custom" | "openai" | "azure_openai" | "gemini" | "vertex_ai" | "anthropic" | "bedrock";
         model: string;
         apiKey?: string | null | undefined;
         baseUrl?: string | null | undefined;
+        project?: string | null | undefined;
+        location?: string | null | undefined;
     }>>>>;
 }, "strip", z.ZodTypeAny, {
     enabled: boolean;
@@ -65,10 +83,12 @@ declare const graphSchema: z.ZodObject<{
     neo4jPassword: string | null;
     neo4jDatabase: string;
     extractionLlm: {
-        provider: "custom" | "openai" | "azure_openai" | "gemini" | "anthropic" | "bedrock";
+        provider: "custom" | "openai" | "azure_openai" | "gemini" | "vertex_ai" | "anthropic" | "bedrock";
         model: string;
         apiKey: string | null;
         baseUrl: string | null;
+        project?: string | null | undefined;
+        location?: string | null | undefined;
     } | null;
 }, {
     enabled?: boolean | undefined;
@@ -77,10 +97,12 @@ declare const graphSchema: z.ZodObject<{
     neo4jPassword?: string | null | undefined;
     neo4jDatabase?: string | undefined;
     extractionLlm?: {
-        provider: "custom" | "openai" | "azure_openai" | "gemini" | "anthropic" | "bedrock";
+        provider: "custom" | "openai" | "azure_openai" | "gemini" | "vertex_ai" | "anthropic" | "bedrock";
         model: string;
         apiKey?: string | null | undefined;
         baseUrl?: string | null | undefined;
+        project?: string | null | undefined;
+        location?: string | null | undefined;
     } | null | undefined;
 }>;
 declare const rerankerSchema: z.ZodObject<{

package/dist/express.d.cts

@@ -1,9 +1,9 @@
 import * as express from 'express';
 import { IncomingMessage } from 'node:http';
-import { h as handleIngestJson, c as createToolsHandlers } from './router-Dsv3fv0R.cjs';
-import './governance-D8g6Wyvb.cjs';
+import { h as handleIngestJson, c as createToolsHandlers } from './router-CiFwC-EN.cjs';
+import './governance-P9pRb4Ol.cjs';
 import 'pg';
-import './config-CNnASw5X.cjs';
+import './config-C5RZ00W6.cjs';
 import 'zod';
 import './redaction-BqD_DEUQ.cjs';
 

package/dist/express.d.ts

@@ -1,9 +1,9 @@
 import * as express from 'express';
 import { IncomingMessage } from 'node:http';
-import { h as handleIngestJson, c as createToolsHandlers } from './router-B_DTkQgU.js';
-import './governance-XFVgtEdV.js';
+import { h as handleIngestJson, c as createToolsHandlers } from './router-D8gBzwLd.js';
+import './governance-BLPK7NMe.js';
 import 'pg';
-import './config-CdlSkKgV.js';
+import './config-BODDdXJ7.js';
 import 'zod';
 import './redaction-BqD_DEUQ.js';
 

package/dist/fastify.d.cts

@@ -1,7 +1,7 @@
-import { h as handleIngestJson, c as createToolsHandlers } from './router-Dsv3fv0R.cjs';
-import './governance-D8g6Wyvb.cjs';
+import { h as handleIngestJson, c as createToolsHandlers } from './router-CiFwC-EN.cjs';
+import './governance-P9pRb4Ol.cjs';
 import 'pg';
-import './config-CNnASw5X.cjs';
+import './config-C5RZ00W6.cjs';
 import 'zod';
 import './redaction-BqD_DEUQ.cjs';
 

package/dist/fastify.d.ts

@@ -1,7 +1,7 @@
-import { h as handleIngestJson, c as createToolsHandlers } from './router-B_DTkQgU.js';
-import './governance-XFVgtEdV.js';
+import { h as handleIngestJson, c as createToolsHandlers } from './router-D8gBzwLd.js';
+import './governance-BLPK7NMe.js';
 import 'pg';
-import './config-CdlSkKgV.js';
+import './config-BODDdXJ7.js';
 import 'zod';
 import './redaction-BqD_DEUQ.js';
 

package/dist/{governance-XFVgtEdV.d.ts → governance-BLPK7NMe.d.ts}

@@ -1,5 +1,5 @@
 import { Pool } from 'pg';
-import { C as ContextEngineConfig } from './config-CdlSkKgV.js';
+import { C as ContextEngineConfig } from './config-BODDdXJ7.js';
 import { H as Hooks } from './redaction-BqD_DEUQ.js';
 
 type ToolKind = "http" | "db" | "mcp" | "function";

package/dist/{governance-D8g6Wyvb.d.cts → governance-P9pRb4Ol.d.cts}

@@ -1,5 +1,5 @@
 import { Pool } from 'pg';
-import { C as ContextEngineConfig } from './config-CNnASw5X.cjs';
+import { C as ContextEngineConfig } from './config-C5RZ00W6.cjs';
 import { H as Hooks } from './redaction-BqD_DEUQ.cjs';
 
 type ToolKind = "http" | "db" | "mcp" | "function";

package/dist/graph/index.cjs

@@ -65,6 +65,47 @@ var init_hooks = __esm({
   }
 });
 
+// src/providers/google.ts
+async function buildGenaiClient(cfg, timeoutMs, purpose) {
+  const specifier = "@google/genai";
+  let mod;
+  try {
+    mod = await import(specifier);
+  } catch {
+    throw new ExtraMissingError("gemini", specifier, purpose);
+  }
+  const Ctor = mod.GoogleGenAI ?? mod.Client;
+  if (!Ctor) {
+    throw new ExtraMissingError("gemini", specifier, purpose);
+  }
+  const opts = { httpOptions: { timeout: timeoutMs } };
+  if (cfg.provider === "vertex_ai") {
+    opts.vertexai = true;
+    if (cfg.project || cfg.location) {
+      if (cfg.project) opts.project = cfg.project;
+      if (cfg.location) opts.location = cfg.location;
+    } else if (cfg.apiKey) {
+      opts.apiKey = cfg.apiKey;
+    }
+  } else {
+    opts.apiKey = cfg.apiKey ?? null;
+  }
+  try {
+    return new Ctor(opts);
+  } catch (err) {
+    const message = err instanceof Error ? err.message : String(err);
+    if (!message.includes("Authentication is not set up")) throw err;
+    throw new Error(
+      `${message} Set \`project\` on the provider config, or export GOOGLE_CLOUD_PROJECT and GOOGLE_CLOUD_LOCATION. Credentials themselves come from Application Default Credentials.`
+    );
+  }
+}
+var init_google = __esm({
+  "src/providers/google.ts"() {
+    init_errors();
+  }
+});
+
 // src/providers/llm.ts
 var llm_exports = {};
 __export(llm_exports, {
@@ -100,20 +141,6 @@ function buildOpenAIChatClient(cfg) {
     maxRetries: 0
   });
 }
-async function loadGeminiChatClient(apiKey) {
-  const specifier = "@google/genai";
-  let mod;
-  try {
-    mod = await import(specifier);
-  } catch {
-    throw new ExtraMissingError("gemini", specifier, "gemini llm");
-  }
-  const Ctor = mod.GoogleGenAI ?? mod.Client;
-  if (!Ctor) {
-    throw new ExtraMissingError("gemini", specifier, "gemini llm");
-  }
-  return new Ctor({ apiKey: apiKey ?? null, httpOptions: { timeout: GEMINI_CALL_TIMEOUT_MS } });
-}
 async function loadBedrockSdk() {
   const specifier = "@aws-sdk/client-bedrock-runtime";
   try {
@@ -132,7 +159,7 @@ function buildLlmClient(cfg, opts) {
   if (OPENAI_FAMILY.has(cfg.provider)) {
     return new LLMClient(cfg, { client: buildOpenAIChatClient(cfg) });
   }
-  if (cfg.provider === "gemini" || cfg.provider === "bedrock") {
+  if (GOOGLE_FAMILY.has(cfg.provider) || cfg.provider === "bedrock") {
     return new LLMClient(cfg);
   }
   throw new Error(`unknown llm provider: ${JSON.stringify(cfg.provider)}`);
@@ -149,16 +176,18 @@ async function callLlm(cfg, opts) {
     await owned.aclose();
   }
 }
-var CALL_TIMEOUT_MS, TIMEOUT_MS, GEMINI_CALL_TIMEOUT_MS, ANTHROPIC_VERSION, ANTHROPIC_MAX_TOKENS, OPENAI_FAMILY, LLMClient;
+var CALL_TIMEOUT_MS, TIMEOUT_MS, GEMINI_CALL_TIMEOUT_MS, ANTHROPIC_VERSION, ANTHROPIC_MAX_TOKENS, OPENAI_FAMILY, GOOGLE_FAMILY, LLMClient;
 var init_llm = __esm({
   "src/providers/llm.ts"() {
     init_errors();
+    init_google();
     CALL_TIMEOUT_MS = 24e4;
     TIMEOUT_MS = CALL_TIMEOUT_MS;
     GEMINI_CALL_TIMEOUT_MS = CALL_TIMEOUT_MS;
     ANTHROPIC_VERSION = "2023-06-01";
     ANTHROPIC_MAX_TOKENS = 4096;
     OPENAI_FAMILY = /* @__PURE__ */ new Set(["openai", "azure_openai", "custom"]);
+    GOOGLE_FAMILY = /* @__PURE__ */ new Set(["gemini", "vertex_ai"]);
     LLMClient = class {
       cfg;
       provider;
@@ -198,7 +227,7 @@ var init_llm = __esm({
         if (OPENAI_FAMILY.has(this.provider)) {
           return this.callOpenAI(system, user, jsonMode, images, maxTokens, temperature);
         }
-        if (this.provider === "gemini") {
+        if (GOOGLE_FAMILY.has(this.provider)) {
           return this.callGemini(system, user, jsonMode, images, maxTokens, thinkingBudget, temperature);
         }
         if (this.provider === "bedrock") {
@@ -281,7 +310,11 @@ Respond with valid JSON only.`;
       }
       async callGemini(system, user, jsonMode, images, maxTokens, thinkingBudget = null, temperature = null) {
         if (!this.genaiClient) {
-          this.genaiClient = await loadGeminiChatClient(this.cfg.apiKey);
+          this.genaiClient = await buildGenaiClient(
+            this.cfg,
+            GEMINI_CALL_TIMEOUT_MS,
+            "gemini llm"
+          );
         }
         const parts = [];
         for (const img of images ?? []) {