@promptev/context-engine 0.0.0

package/dist/config-Bl9U789m.d.cts

@@ -0,0 +1,174 @@
+import { z } from 'zod';
+import { R as RedactionPolicy, b as RedactionRuleInit } from './redaction-BmDSWJ7h.cjs';
+
+declare const embeddingSchema: z.ZodObject<{
+    provider: z.ZodEnum<["openai", "azure_openai", "gemini", "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>>>;
+}, "strip", z.ZodTypeAny, {
+    provider: "custom" | "openai" | "azure_openai" | "gemini" | "voyage" | "cohere";
+    model: string;
+    dim: number | null;
+    apiKey: string | null;
+    baseUrl: string | null;
+}, {
+    provider: "custom" | "openai" | "azure_openai" | "gemini" | "voyage" | "cohere";
+    model: string;
+    dim?: number | null | undefined;
+    apiKey?: string | null | undefined;
+    baseUrl?: string | null | undefined;
+}>;
+declare const llmSchema: z.ZodObject<{
+    provider: z.ZodEnum<["anthropic", "openai", "azure_openai", "gemini", "bedrock", "custom"]>;
+    model: z.ZodString;
+    apiKey: z.ZodDefault<z.ZodOptional<z.ZodNullable<z.ZodString>>>;
+    baseUrl: z.ZodDefault<z.ZodOptional<z.ZodNullable<z.ZodString>>>;
+}, "strip", z.ZodTypeAny, {
+    provider: "custom" | "openai" | "azure_openai" | "gemini" | "anthropic" | "bedrock";
+    model: string;
+    apiKey: string | null;
+    baseUrl: string | null;
+}, {
+    provider: "custom" | "openai" | "azure_openai" | "gemini" | "anthropic" | "bedrock";
+    model: string;
+    apiKey?: string | null | undefined;
+    baseUrl?: string | null | undefined;
+}>;
+declare const graphSchema: z.ZodObject<{
+    enabled: z.ZodDefault<z.ZodBoolean>;
+    neo4jUri: z.ZodDefault<z.ZodOptional<z.ZodNullable<z.ZodString>>>;
+    neo4jUser: z.ZodDefault<z.ZodString>;
+    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"]>;
+        model: z.ZodString;
+        apiKey: z.ZodDefault<z.ZodOptional<z.ZodNullable<z.ZodString>>>;
+        baseUrl: z.ZodDefault<z.ZodOptional<z.ZodNullable<z.ZodString>>>;
+    }, "strip", z.ZodTypeAny, {
+        provider: "custom" | "openai" | "azure_openai" | "gemini" | "anthropic" | "bedrock";
+        model: string;
+        apiKey: string | null;
+        baseUrl: string | null;
+    }, {
+        provider: "custom" | "openai" | "azure_openai" | "gemini" | "anthropic" | "bedrock";
+        model: string;
+        apiKey?: string | null | undefined;
+        baseUrl?: string | null | undefined;
+    }>>>>;
+}, "strip", z.ZodTypeAny, {
+    enabled: boolean;
+    neo4jUri: string | null;
+    neo4jUser: string;
+    neo4jPassword: string | null;
+    neo4jDatabase: string;
+    extractionLlm: {
+        provider: "custom" | "openai" | "azure_openai" | "gemini" | "anthropic" | "bedrock";
+        model: string;
+        apiKey: string | null;
+        baseUrl: string | null;
+    } | null;
+}, {
+    enabled?: boolean | undefined;
+    neo4jUri?: string | null | undefined;
+    neo4jUser?: string | undefined;
+    neo4jPassword?: string | null | undefined;
+    neo4jDatabase?: string | undefined;
+    extractionLlm?: {
+        provider: "custom" | "openai" | "azure_openai" | "gemini" | "anthropic" | "bedrock";
+        model: string;
+        apiKey?: string | null | undefined;
+        baseUrl?: string | null | undefined;
+    } | null | undefined;
+}>;
+declare const rerankerSchema: z.ZodObject<{
+    enabled: z.ZodDefault<z.ZodBoolean>;
+    provider: z.ZodDefault<z.ZodOptional<z.ZodNullable<z.ZodEnum<["cohere", "voyage", "jina", "custom"]>>>>;
+    model: z.ZodDefault<z.ZodOptional<z.ZodNullable<z.ZodString>>>;
+    apiKey: z.ZodDefault<z.ZodOptional<z.ZodNullable<z.ZodString>>>;
+    baseUrl: z.ZodDefault<z.ZodOptional<z.ZodNullable<z.ZodString>>>;
+    candidates: z.ZodDefault<z.ZodNumber>;
+}, "strip", z.ZodTypeAny, {
+    provider: "custom" | "voyage" | "cohere" | "jina" | null;
+    model: string | null;
+    apiKey: string | null;
+    baseUrl: string | null;
+    enabled: boolean;
+    candidates: number;
+}, {
+    provider?: "custom" | "voyage" | "cohere" | "jina" | null | undefined;
+    model?: string | null | undefined;
+    apiKey?: string | null | undefined;
+    baseUrl?: string | null | undefined;
+    enabled?: boolean | undefined;
+    candidates?: number | undefined;
+}>;
+declare const fusionSchema: z.ZodObject<{
+    method: z.ZodDefault<z.ZodLiteral<"rrf">>;
+    k: z.ZodDefault<z.ZodNumber>;
+    weights: z.ZodDefault<z.ZodRecord<z.ZodString, z.ZodNumber>>;
+}, "strip", z.ZodTypeAny, {
+    method: "rrf";
+    k: number;
+    weights: Record<string, number>;
+}, {
+    method?: "rrf" | undefined;
+    k?: number | undefined;
+    weights?: Record<string, number> | undefined;
+}>;
+declare const storageSchema: z.ZodObject<{
+    backend: z.ZodDefault<z.ZodLiteral<"postgres">>;
+    annExactThreshold: z.ZodDefault<z.ZodNumber>;
+}, "strip", z.ZodTypeAny, {
+    backend: "postgres";
+    annExactThreshold: number;
+}, {
+    backend?: "postgres" | undefined;
+    annExactThreshold?: number | undefined;
+}>;
+type EmbeddingConfig = z.infer<typeof embeddingSchema>;
+type LLMConfig = z.infer<typeof llmSchema>;
+type GraphConfig = z.infer<typeof graphSchema>;
+type RerankerConfig = z.infer<typeof rerankerSchema>;
+type FusionConfig = z.infer<typeof fusionSchema>;
+type StorageConfig = z.infer<typeof storageSchema>;
+interface ContextEngineConfigInit {
+    databaseUrl: string;
+    storage?: Partial<StorageConfig>;
+    defaultMode?: "hybrid" | "graph";
+    embedding: EmbeddingConfig;
+    llm?: LLMConfig | null;
+    visionLlm?: LLMConfig | null;
+    graph?: Partial<GraphConfig>;
+    reranker?: Partial<RerankerConfig>;
+    fusion?: Partial<FusionConfig>;
+    enableCodeExecution?: boolean;
+    secretKey?: string | null;
+    redaction?: RedactionPolicy | {
+        rules?: RedactionRuleInit[];
+        customDetectors?: RedactionPolicy["customDetectors"];
+    };
+}
+declare class ContextEngineConfig {
+    databaseUrl: string;
+    storage: StorageConfig;
+    defaultMode: "hybrid" | "graph";
+    embedding: EmbeddingConfig;
+    llm: LLMConfig | null;
+    visionLlm: LLMConfig | null;
+    graph: GraphConfig;
+    reranker: RerankerConfig;
+    fusion: FusionConfig;
+    enableCodeExecution: boolean;
+    secretKey: string | null;
+    redaction: RedactionPolicy;
+    constructor(init: ContextEngineConfigInit);
+    private validate;
+    static fromEnv(overrides?: Partial<ContextEngineConfigInit> & {
+        embedding?: EmbeddingConfig;
+    }): ContextEngineConfig;
+}
+
+export { ContextEngineConfig as C, type EmbeddingConfig as E, type FusionConfig as F, type GraphConfig as G, type LLMConfig as L, type RerankerConfig as R, type StorageConfig as S, type ContextEngineConfigInit as a };

package/dist/config-Bt9bUQqU.d.ts

@@ -0,0 +1,174 @@
+import { z } from 'zod';
+import { R as RedactionPolicy, b as RedactionRuleInit } from './redaction-BmDSWJ7h.js';
+
+declare const embeddingSchema: z.ZodObject<{
+    provider: z.ZodEnum<["openai", "azure_openai", "gemini", "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>>>;
+}, "strip", z.ZodTypeAny, {
+    provider: "custom" | "openai" | "azure_openai" | "gemini" | "voyage" | "cohere";
+    model: string;
+    dim: number | null;
+    apiKey: string | null;
+    baseUrl: string | null;
+}, {
+    provider: "custom" | "openai" | "azure_openai" | "gemini" | "voyage" | "cohere";
+    model: string;
+    dim?: number | null | undefined;
+    apiKey?: string | null | undefined;
+    baseUrl?: string | null | undefined;
+}>;
+declare const llmSchema: z.ZodObject<{
+    provider: z.ZodEnum<["anthropic", "openai", "azure_openai", "gemini", "bedrock", "custom"]>;
+    model: z.ZodString;
+    apiKey: z.ZodDefault<z.ZodOptional<z.ZodNullable<z.ZodString>>>;
+    baseUrl: z.ZodDefault<z.ZodOptional<z.ZodNullable<z.ZodString>>>;
+}, "strip", z.ZodTypeAny, {
+    provider: "custom" | "openai" | "azure_openai" | "gemini" | "anthropic" | "bedrock";
+    model: string;
+    apiKey: string | null;
+    baseUrl: string | null;
+}, {
+    provider: "custom" | "openai" | "azure_openai" | "gemini" | "anthropic" | "bedrock";
+    model: string;
+    apiKey?: string | null | undefined;
+    baseUrl?: string | null | undefined;
+}>;
+declare const graphSchema: z.ZodObject<{
+    enabled: z.ZodDefault<z.ZodBoolean>;
+    neo4jUri: z.ZodDefault<z.ZodOptional<z.ZodNullable<z.ZodString>>>;
+    neo4jUser: z.ZodDefault<z.ZodString>;
+    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"]>;
+        model: z.ZodString;
+        apiKey: z.ZodDefault<z.ZodOptional<z.ZodNullable<z.ZodString>>>;
+        baseUrl: z.ZodDefault<z.ZodOptional<z.ZodNullable<z.ZodString>>>;
+    }, "strip", z.ZodTypeAny, {
+        provider: "custom" | "openai" | "azure_openai" | "gemini" | "anthropic" | "bedrock";
+        model: string;
+        apiKey: string | null;
+        baseUrl: string | null;
+    }, {
+        provider: "custom" | "openai" | "azure_openai" | "gemini" | "anthropic" | "bedrock";
+        model: string;
+        apiKey?: string | null | undefined;
+        baseUrl?: string | null | undefined;
+    }>>>>;
+}, "strip", z.ZodTypeAny, {
+    enabled: boolean;
+    neo4jUri: string | null;
+    neo4jUser: string;
+    neo4jPassword: string | null;
+    neo4jDatabase: string;
+    extractionLlm: {
+        provider: "custom" | "openai" | "azure_openai" | "gemini" | "anthropic" | "bedrock";
+        model: string;
+        apiKey: string | null;
+        baseUrl: string | null;
+    } | null;
+}, {
+    enabled?: boolean | undefined;
+    neo4jUri?: string | null | undefined;
+    neo4jUser?: string | undefined;
+    neo4jPassword?: string | null | undefined;
+    neo4jDatabase?: string | undefined;
+    extractionLlm?: {
+        provider: "custom" | "openai" | "azure_openai" | "gemini" | "anthropic" | "bedrock";
+        model: string;
+        apiKey?: string | null | undefined;
+        baseUrl?: string | null | undefined;
+    } | null | undefined;
+}>;
+declare const rerankerSchema: z.ZodObject<{
+    enabled: z.ZodDefault<z.ZodBoolean>;
+    provider: z.ZodDefault<z.ZodOptional<z.ZodNullable<z.ZodEnum<["cohere", "voyage", "jina", "custom"]>>>>;
+    model: z.ZodDefault<z.ZodOptional<z.ZodNullable<z.ZodString>>>;
+    apiKey: z.ZodDefault<z.ZodOptional<z.ZodNullable<z.ZodString>>>;
+    baseUrl: z.ZodDefault<z.ZodOptional<z.ZodNullable<z.ZodString>>>;
+    candidates: z.ZodDefault<z.ZodNumber>;
+}, "strip", z.ZodTypeAny, {
+    provider: "custom" | "voyage" | "cohere" | "jina" | null;
+    model: string | null;
+    apiKey: string | null;
+    baseUrl: string | null;
+    enabled: boolean;
+    candidates: number;
+}, {
+    provider?: "custom" | "voyage" | "cohere" | "jina" | null | undefined;
+    model?: string | null | undefined;
+    apiKey?: string | null | undefined;
+    baseUrl?: string | null | undefined;
+    enabled?: boolean | undefined;
+    candidates?: number | undefined;
+}>;
+declare const fusionSchema: z.ZodObject<{
+    method: z.ZodDefault<z.ZodLiteral<"rrf">>;
+    k: z.ZodDefault<z.ZodNumber>;
+    weights: z.ZodDefault<z.ZodRecord<z.ZodString, z.ZodNumber>>;
+}, "strip", z.ZodTypeAny, {
+    method: "rrf";
+    k: number;
+    weights: Record<string, number>;
+}, {
+    method?: "rrf" | undefined;
+    k?: number | undefined;
+    weights?: Record<string, number> | undefined;
+}>;
+declare const storageSchema: z.ZodObject<{
+    backend: z.ZodDefault<z.ZodLiteral<"postgres">>;
+    annExactThreshold: z.ZodDefault<z.ZodNumber>;
+}, "strip", z.ZodTypeAny, {
+    backend: "postgres";
+    annExactThreshold: number;
+}, {
+    backend?: "postgres" | undefined;
+    annExactThreshold?: number | undefined;
+}>;
+type EmbeddingConfig = z.infer<typeof embeddingSchema>;
+type LLMConfig = z.infer<typeof llmSchema>;
+type GraphConfig = z.infer<typeof graphSchema>;
+type RerankerConfig = z.infer<typeof rerankerSchema>;
+type FusionConfig = z.infer<typeof fusionSchema>;
+type StorageConfig = z.infer<typeof storageSchema>;
+interface ContextEngineConfigInit {
+    databaseUrl: string;
+    storage?: Partial<StorageConfig>;
+    defaultMode?: "hybrid" | "graph";
+    embedding: EmbeddingConfig;
+    llm?: LLMConfig | null;
+    visionLlm?: LLMConfig | null;
+    graph?: Partial<GraphConfig>;
+    reranker?: Partial<RerankerConfig>;
+    fusion?: Partial<FusionConfig>;
+    enableCodeExecution?: boolean;
+    secretKey?: string | null;
+    redaction?: RedactionPolicy | {
+        rules?: RedactionRuleInit[];
+        customDetectors?: RedactionPolicy["customDetectors"];
+    };
+}
+declare class ContextEngineConfig {
+    databaseUrl: string;
+    storage: StorageConfig;
+    defaultMode: "hybrid" | "graph";
+    embedding: EmbeddingConfig;
+    llm: LLMConfig | null;
+    visionLlm: LLMConfig | null;
+    graph: GraphConfig;
+    reranker: RerankerConfig;
+    fusion: FusionConfig;
+    enableCodeExecution: boolean;
+    secretKey: string | null;
+    redaction: RedactionPolicy;
+    constructor(init: ContextEngineConfigInit);
+    private validate;
+    static fromEnv(overrides?: Partial<ContextEngineConfigInit> & {
+        embedding?: EmbeddingConfig;
+    }): ContextEngineConfig;
+}
+
+export { ContextEngineConfig as C, type EmbeddingConfig as E, type FusionConfig as F, type GraphConfig as G, type LLMConfig as L, type RerankerConfig as R, type StorageConfig as S, type ContextEngineConfigInit as a };

package/dist/embeddings-B-jZ42mk.d.cts

@@ -0,0 +1,67 @@
+import { E as EmbeddingConfig } from './config-Bl9U789m.cjs';
+
+/**
+ * Pluggable embeddings provider.
+ *
+ * openai / azure_openai / custom → OpenAI SDK (OpenAI-compatible, including TEI/vLLM).
+ * gemini → lazy `@google/genai` (optional extra "gemini").
+ * voyage / cohere → fetch against their public REST APIs.
+ *
+ * `kind` is accepted everywhere but only voyage/cohere act on it.
+ * 30s timeout, no retries. Injectable `client` / `fetch` for tests.
+ */
+
+type EmbedKind = "document" | "query";
+type FetchImpl = (input: string | URL | Request, init?: RequestInit) => Promise<Response>;
+type EmbeddingsClient = {
+    embeddings: {
+        create(body: {
+            model: string;
+            input: string[];
+        }): Promise<{
+            data: Array<{
+                embedding: number[];
+            }>;
+            usage?: {
+                total_tokens?: number;
+            } | null;
+        }>;
+    };
+    close?: () => void | Promise<void>;
+    timeout?: number;
+    baseURL?: string;
+};
+type EmbedderOpts = {
+    client?: EmbeddingsClient | null;
+    fetch?: FetchImpl | null;
+    fetchImpl?: FetchImpl | null;
+};
+declare class Embedder {
+    cfg: EmbeddingConfig;
+    provider: EmbeddingConfig["provider"];
+    model: string;
+    dim: number | null;
+    client: EmbeddingsClient | null;
+    fetchImpl: FetchImpl | null;
+    private genaiClient;
+    constructor(cfg: EmbeddingConfig, opts?: EmbedderOpts);
+    aclose(): Promise<void>;
+    [Symbol.asyncDispose](): Promise<void>;
+    /**
+     * Embed `texts`, returning `[vectors, tokenCount]`.
+     * Auto-detects `this.dim` on first success; raises on mismatch.
+     */
+    embed(texts: string[], opts?: {
+        kind?: EmbedKind;
+    }): Promise<[number[][], number]>;
+    /** Provider dispatch. Overridable per-instance (tests stub this). */
+    rawEmbed(texts: string[], kind?: EmbedKind): Promise<[number[][], number]>;
+    private embedOpenAI;
+    private embedGemini;
+    private embedVoyage;
+    private embedCohere;
+    private requireFetch;
+}
+declare function buildEmbedder(cfg: EmbeddingConfig, opts?: EmbedderOpts): Embedder;
+
+export { Embedder as E, type FetchImpl as F, type EmbedKind as a, buildEmbedder as b };

package/dist/embeddings-DaSdAZN3.d.ts

@@ -0,0 +1,67 @@
+import { E as EmbeddingConfig } from './config-Bt9bUQqU.js';
+
+/**
+ * Pluggable embeddings provider.
+ *
+ * openai / azure_openai / custom → OpenAI SDK (OpenAI-compatible, including TEI/vLLM).
+ * gemini → lazy `@google/genai` (optional extra "gemini").
+ * voyage / cohere → fetch against their public REST APIs.
+ *
+ * `kind` is accepted everywhere but only voyage/cohere act on it.
+ * 30s timeout, no retries. Injectable `client` / `fetch` for tests.
+ */
+
+type EmbedKind = "document" | "query";
+type FetchImpl = (input: string | URL | Request, init?: RequestInit) => Promise<Response>;
+type EmbeddingsClient = {
+    embeddings: {
+        create(body: {
+            model: string;
+            input: string[];
+        }): Promise<{
+            data: Array<{
+                embedding: number[];
+            }>;
+            usage?: {
+                total_tokens?: number;
+            } | null;
+        }>;
+    };
+    close?: () => void | Promise<void>;
+    timeout?: number;
+    baseURL?: string;
+};
+type EmbedderOpts = {
+    client?: EmbeddingsClient | null;
+    fetch?: FetchImpl | null;
+    fetchImpl?: FetchImpl | null;
+};
+declare class Embedder {
+    cfg: EmbeddingConfig;
+    provider: EmbeddingConfig["provider"];
+    model: string;
+    dim: number | null;
+    client: EmbeddingsClient | null;
+    fetchImpl: FetchImpl | null;
+    private genaiClient;
+    constructor(cfg: EmbeddingConfig, opts?: EmbedderOpts);
+    aclose(): Promise<void>;
+    [Symbol.asyncDispose](): Promise<void>;
+    /**
+     * Embed `texts`, returning `[vectors, tokenCount]`.
+     * Auto-detects `this.dim` on first success; raises on mismatch.
+     */
+    embed(texts: string[], opts?: {
+        kind?: EmbedKind;
+    }): Promise<[number[][], number]>;
+    /** Provider dispatch. Overridable per-instance (tests stub this). */
+    rawEmbed(texts: string[], kind?: EmbedKind): Promise<[number[][], number]>;
+    private embedOpenAI;
+    private embedGemini;
+    private embedVoyage;
+    private embedCohere;
+    private requireFetch;
+}
+declare function buildEmbedder(cfg: EmbeddingConfig, opts?: EmbedderOpts): Embedder;
+
+export { Embedder as E, type FetchImpl as F, type EmbedKind as a, buildEmbedder as b };