@promptev/context-engine 0.0.1 → 0.0.2

package/dist/graph/index.d.cts

@@ -1,7 +1,7 @@
 import { Pool } from 'pg';
-import { G as GraphConfig, C as ContextEngineConfig } from '../config-CNnASw5X.cjs';
+import { G as GraphConfig, C as ContextEngineConfig } from '../config-C5RZ00W6.cjs';
 import { H as Hooks } from '../redaction-BqD_DEUQ.cjs';
-import { E as Embedder, A as AnnTunable } from '../storage-DU1JRno5.cjs';
+import { E as Embedder, A as AnnTunable } from '../storage-Dvpq2xAC.cjs';
 import 'zod';
 
 type Neo4jDriver = {

package/dist/graph/index.d.ts

@@ -1,7 +1,7 @@
 import { Pool } from 'pg';
-import { G as GraphConfig, C as ContextEngineConfig } from '../config-CdlSkKgV.js';
+import { G as GraphConfig, C as ContextEngineConfig } from '../config-BODDdXJ7.js';
 import { H as Hooks } from '../redaction-BqD_DEUQ.js';
-import { E as Embedder, A as AnnTunable } from '../storage-Dvt2ZxsV.js';
+import { E as Embedder, A as AnnTunable } from '../storage-CJrKgJeJ.js';
 import 'zod';
 
 type Neo4jDriver = {

package/dist/graph/index.js

@@ -59,6 +59,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, {
@@ -94,20 +135,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 {
@@ -126,7 +153,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)}`);
@@ -143,16 +170,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;
@@ -192,7 +221,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") {
@@ -275,7 +304,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 ?? []) {