@promptev/context-engine 0.0.1 → 0.0.3

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 ?? []) {
@@ -586,9 +619,9 @@ function buildPrompt(fullText) {
 DOCUMENT TEXT:
 ${fullText}
 
-ENTITY TYPES: PERSON, ORG, PRODUCT, LOCATION, REFERENCE, TEMPORAL, CONCEPT
+ENTITY TYPES: ${ENTITY_TYPES.join(", ")}
 
-RELATIONSHIP CATEGORIES: HIERARCHICAL, MEMBERSHIP, CREATION, TEMPORAL, SPATIAL, REFERENCE, FUNCTIONAL, QUANTITATIVE
+RELATIONSHIP CATEGORIES: ${RELATIONSHIP_CATEGORY_VALUES.join(", ")}
 
 OUTPUT FORMAT (STRICT JSON):
 {
@@ -867,7 +900,7 @@ async function extractEntitiesForDocument(documentId, opts) {
   if (opts.graphStore) await syncToNeo4j(opts.pool, documentId, chunkSnaps, opts.graphStore);
   return stats;
 }
-var ENTITY_TYPES, RELATIONSHIP_CATEGORIES, CHUNKS_PER_BATCH, EXTRACTION_SYSTEM;
+var ENTITY_TYPES, RELATIONSHIP_CATEGORY_VALUES, RELATIONSHIP_CATEGORIES, CHUNKS_PER_BATCH, EXTRACTION_SYSTEM;
 var init_entities = __esm({
   "src/graph/entities.ts"() {
     ENTITY_TYPES = [
@@ -879,7 +912,7 @@ var init_entities = __esm({
       "TEMPORAL",
       "CONCEPT"
     ];
-    RELATIONSHIP_CATEGORIES = /* @__PURE__ */ new Set([
+    RELATIONSHIP_CATEGORY_VALUES = [
       "HIERARCHICAL",
       "MEMBERSHIP",
       "CREATION",
@@ -888,7 +921,8 @@ var init_entities = __esm({
       "REFERENCE",
       "FUNCTIONAL",
       "QUANTITATIVE"
-    ]);
+    ];
+    RELATIONSHIP_CATEGORIES = new Set(RELATIONSHIP_CATEGORY_VALUES);
     CHUNKS_PER_BATCH = 11;
     EXTRACTION_SYSTEM = "You are a STRICT JSON entity and relationship extraction engine.";
   }