@retrivora-ai/rag-engine 0.3.1 → 0.3.3

package/.env.example

@@ -20,7 +20,7 @@ VECTOR_DB_REST_API_KEY=your-rest-api-key
 # VECTOR_DB_UPSERT_PATH=/upsert
 
 # ─── LLM ──────────────────────────────────────────────────────
-# Choose: openai | anthropic | ollama | universal_rest
+# Choose: openai | anthropic | ollama | gemini | universal_rest
 LLM_PROVIDER=openai
 LLM_MODEL=gpt-4o
 LLM_MAX_TOKENS=1024
@@ -33,6 +33,9 @@ OPENAI_API_KEY=sk-...
 # Anthropic (if provider=anthropic)
 ANTHROPIC_API_KEY=sk-ant-...
 
+# Gemini (if provider=gemini)
+GEMINI_API_KEY=AIza...
+
 # Ollama (if provider=ollama)
 LLM_BASE_URL=http://localhost:11434
 
@@ -52,7 +55,7 @@ VECTOR_BASE_URL=http://localhost:6333
 VECTOR_DB_INDEX=my-index
 
 # ─── Embedding ────────────────────────────────────────────────
-# Choose: openai | ollama | custom
+# Choose: openai | ollama | gemini | custom
 EMBEDDING_PROVIDER=openai
 EMBEDDING_MODEL=text-embedding-3-small
 EMBEDDING_DIMENSIONS=1536

package/dist/{RagConfig-D3Inaf9N.d.mts → RagConfig-Bpp39-um.d.mts}

@@ -20,12 +20,23 @@ interface ChatResponse {
     sources: VectorMatch[];
 }
 
+/**
+ * constants.ts — Centralized definitions of acceptable configuration values.
+ *
+ * This file serves as the single source of truth for all statically supported
+ * providers and stylistic options in the Retrivora AI RAG Engine.
+ */
+declare const VECTOR_DB_PROVIDERS: readonly ["pinecone", "pgvector", "postgresql", "mongodb", "milvus", "qdrant", "chromadb", "redis", "weaviate", "rest", "universal_rest", "custom"];
+declare const LLM_PROVIDERS: readonly ["openai", "anthropic", "ollama", "gemini", "rest", "universal_rest", "custom"];
+declare const EMBEDDING_PROVIDERS: readonly ["openai", "ollama", "gemini", "rest", "universal_rest", "custom"];
+
 /**
  * Master configuration interface for Retrivora AI.
  * Each consuming project provides one RagConfig object to drive
  * vector DB selection, LLM selection, embedding, and UI branding.
  */
-type VectorDBProvider = 'pinecone' | 'pgvector' | 'postgresql' | 'mongodb' | 'milvus' | 'qdrant' | 'chromadb' | 'redis' | 'weaviate' | 'rest' | 'universal_rest' | 'custom';
+
+type VectorDBProvider = typeof VECTOR_DB_PROVIDERS[number];
 interface VectorDBConfig {
     /** Which vector database to use */
     provider: VectorDBProvider;
@@ -51,7 +62,7 @@ interface VectorDBConfig {
      */
     options: Record<string, unknown>;
 }
-type LLMProvider = 'openai' | 'anthropic' | 'ollama' | 'rest' | 'universal_rest' | 'custom';
+type LLMProvider = typeof LLM_PROVIDERS[number];
 interface LLMConfig {
     /** Which LLM provider to use */
     provider: LLMProvider;
@@ -77,7 +88,7 @@ interface LLMConfig {
      */
     options?: Record<string, unknown>;
 }
-type EmbeddingProvider = 'openai' | 'ollama' | 'rest' | 'universal_rest' | 'custom';
+type EmbeddingProvider = typeof EMBEDDING_PROVIDERS[number];
 interface EmbeddingConfig {
     /** Which embedding provider to use */
     provider: EmbeddingProvider;

package/dist/{RagConfig-D3Inaf9N.d.ts → RagConfig-Bpp39-um.d.ts}

@@ -20,12 +20,23 @@ interface ChatResponse {
     sources: VectorMatch[];
 }
 
+/**
+ * constants.ts — Centralized definitions of acceptable configuration values.
+ *
+ * This file serves as the single source of truth for all statically supported
+ * providers and stylistic options in the Retrivora AI RAG Engine.
+ */
+declare const VECTOR_DB_PROVIDERS: readonly ["pinecone", "pgvector", "postgresql", "mongodb", "milvus", "qdrant", "chromadb", "redis", "weaviate", "rest", "universal_rest", "custom"];
+declare const LLM_PROVIDERS: readonly ["openai", "anthropic", "ollama", "gemini", "rest", "universal_rest", "custom"];
+declare const EMBEDDING_PROVIDERS: readonly ["openai", "ollama", "gemini", "rest", "universal_rest", "custom"];
+
 /**
  * Master configuration interface for Retrivora AI.
  * Each consuming project provides one RagConfig object to drive
  * vector DB selection, LLM selection, embedding, and UI branding.
  */
-type VectorDBProvider = 'pinecone' | 'pgvector' | 'postgresql' | 'mongodb' | 'milvus' | 'qdrant' | 'chromadb' | 'redis' | 'weaviate' | 'rest' | 'universal_rest' | 'custom';
+
+type VectorDBProvider = typeof VECTOR_DB_PROVIDERS[number];
 interface VectorDBConfig {
     /** Which vector database to use */
     provider: VectorDBProvider;
@@ -51,7 +62,7 @@ interface VectorDBConfig {
      */
     options: Record<string, unknown>;
 }
-type LLMProvider = 'openai' | 'anthropic' | 'ollama' | 'rest' | 'universal_rest' | 'custom';
+type LLMProvider = typeof LLM_PROVIDERS[number];
 interface LLMConfig {
     /** Which LLM provider to use */
     provider: LLMProvider;
@@ -77,7 +88,7 @@ interface LLMConfig {
      */
     options?: Record<string, unknown>;
 }
-type EmbeddingProvider = 'openai' | 'ollama' | 'rest' | 'universal_rest' | 'custom';
+type EmbeddingProvider = typeof EMBEDDING_PROVIDERS[number];
 interface EmbeddingConfig {
     /** Which embedding provider to use */
     provider: EmbeddingProvider;

package/dist/{chunk-GT72OIOD.mjs → chunk-SNPFZBDK.mjs}

@@ -11,7 +11,7 @@ import {
 // src/handlers/index.ts
 import { NextResponse } from "next/server";
 
-// src/config/serverConfig.ts
+// src/config/constants.ts
 var VECTOR_DB_PROVIDERS = [
   "pinecone",
   "pgvector",
@@ -26,8 +26,34 @@ var VECTOR_DB_PROVIDERS = [
   "universal_rest",
   "custom"
 ];
-var LLM_PROVIDERS = ["openai", "anthropic", "ollama", "rest", "universal_rest", "custom"];
-var EMBEDDING_PROVIDERS = ["openai", "ollama", "rest", "universal_rest", "custom"];
+var LLM_PROVIDERS = [
+  "openai",
+  "anthropic",
+  "ollama",
+  "gemini",
+  "rest",
+  "universal_rest",
+  "custom"
+];
+var EMBEDDING_PROVIDERS = [
+  "openai",
+  "ollama",
+  "gemini",
+  "rest",
+  "universal_rest",
+  "custom"
+];
+var PROVIDERS_WITH_EMBEDDINGS = [
+  "openai",
+  "ollama",
+  "gemini",
+  "rest",
+  "universal_rest"
+];
+var UI_VISUAL_STYLES = ["glass", "solid"];
+var UI_BORDER_RADIUS_OPTIONS = ["none", "sm", "md", "lg", "xl", "full"];
+
+// src/config/serverConfig.ts
 function readString(env, name) {
   var _a;
   const value = (_a = env[name]) == null ? void 0 : _a.trim();
@@ -94,6 +120,7 @@ function getRagConfig(env = process.env) {
   const llmApiKeyByProvider = {
     openai: readString(env, "OPENAI_API_KEY"),
     anthropic: readString(env, "ANTHROPIC_API_KEY"),
+    gemini: readString(env, "GEMINI_API_KEY"),
     ollama: void 0,
     universal_rest: readString(env, "LLM_API_KEY"),
     rest: readString(env, "LLM_API_KEY"),
@@ -101,6 +128,7 @@ function getRagConfig(env = process.env) {
   };
   const embeddingApiKeyByProvider = {
     openai: readString(env, "OPENAI_API_KEY"),
+    gemini: readString(env, "GEMINI_API_KEY"),
     ollama: void 0,
     universal_rest: (_q = readString(env, "EMBEDDING_API_KEY")) != null ? _q : readString(env, "OPENAI_API_KEY"),
     custom: (_r = readString(env, "EMBEDDING_API_KEY")) != null ? _r : readString(env, "OPENAI_API_KEY")
@@ -515,6 +543,16 @@ var ConfigValidator = class {
           });
         }
         break;
+      case "gemini":
+        if (!config.apiKey) {
+          errors.push({
+            field: "llm.apiKey",
+            message: "Gemini API key is required",
+            suggestion: "Set GEMINI_API_KEY environment variable",
+            severity: "error"
+          });
+        }
+        break;
       case "ollama":
         if (!config.baseUrl) {
           errors.push({
@@ -581,6 +619,14 @@ var ConfigValidator = class {
         severity: "error"
       });
     }
+    if (config.provider === "gemini" && !config.apiKey) {
+      errors.push({
+        field: "embedding.apiKey",
+        message: "Gemini API key is required for embedding",
+        suggestion: "Set GEMINI_API_KEY environment variable",
+        severity: "error"
+      });
+    }
     if (config.provider === "ollama" && !config.baseUrl) {
       errors.push({
         field: "embedding.baseUrl",
@@ -611,21 +657,19 @@ var ConfigValidator = class {
       }
     }
     if (config.borderRadius) {
-      const validValues = ["none", "sm", "md", "lg", "xl", "full"];
-      if (!validValues.includes(config.borderRadius)) {
+      if (!UI_BORDER_RADIUS_OPTIONS.includes(config.borderRadius)) {
         errors.push({
           field: "ui.borderRadius",
-          message: `borderRadius must be one of: ${validValues.join(", ")}`,
+          message: `borderRadius must be one of: ${UI_BORDER_RADIUS_OPTIONS.join(", ")}`,
           severity: "warning"
         });
       }
     }
     if (config.visualStyle) {
-      const validValues = ["glass", "solid"];
-      if (!validValues.includes(config.visualStyle)) {
+      if (!UI_VISUAL_STYLES.includes(config.visualStyle)) {
         errors.push({
           field: "ui.visualStyle",
-          message: `visualStyle must be one of: ${validValues.join(", ")}`,
+          message: `visualStyle must be one of: ${UI_VISUAL_STYLES.join(", ")}`,
           severity: "warning"
         });
       }
@@ -959,6 +1003,101 @@ ${context}`;
   }
 };
 
+// src/llm/providers/GeminiProvider.ts
+import { GoogleGenAI } from "@google/genai";
+var GeminiProvider = class {
+  constructor(llmConfig, embeddingConfig) {
+    if (!llmConfig.apiKey) throw new Error("[GeminiProvider] llmConfig.apiKey is required");
+    this.client = new GoogleGenAI({ apiKey: llmConfig.apiKey });
+    this.llmConfig = llmConfig;
+    this.embeddingConfig = embeddingConfig;
+  }
+  async chat(messages, context, options) {
+    var _a, _b, _c, _d, _e, _f;
+    const systemPrompt = (_a = this.llmConfig.systemPrompt) != null ? _a : `You are a helpful assistant. Answer questions based on the provided context.
+
+Context:
+${context}`;
+    const systemInstruction = systemPrompt.includes("{{context}}") ? systemPrompt.replace("{{context}}", context) : `${systemPrompt}
+
+Context:
+${context}`;
+    const geminiMessages = messages.map((m) => ({
+      role: m.role === "assistant" ? "model" : "user",
+      parts: [{ text: m.content }]
+    }));
+    const response = await this.client.models.generateContent({
+      model: this.llmConfig.model,
+      contents: geminiMessages,
+      config: {
+        systemInstruction,
+        temperature: (_c = (_b = options == null ? void 0 : options.temperature) != null ? _b : this.llmConfig.temperature) != null ? _c : 0.7,
+        maxOutputTokens: (_e = (_d = options == null ? void 0 : options.maxTokens) != null ? _d : this.llmConfig.maxTokens) != null ? _e : 1024,
+        stopSequences: options == null ? void 0 : options.stop
+      }
+    });
+    return (_f = response.text) != null ? _f : "";
+  }
+  async embed(text, options) {
+    var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j;
+    const model = (_c = (_b = options == null ? void 0 : options.model) != null ? _b : (_a = this.embeddingConfig) == null ? void 0 : _a.model) != null ? _c : "text-embedding-004";
+    const apiKey = (_e = (_d = this.embeddingConfig) == null ? void 0 : _d.apiKey) != null ? _e : this.llmConfig.apiKey;
+    const client = apiKey !== this.llmConfig.apiKey ? new GoogleGenAI({ apiKey }) : this.client;
+    let content = text;
+    const queryPrefix = (_f = this.embeddingConfig) == null ? void 0 : _f.queryPrefix;
+    const docPrefix = (_g = this.embeddingConfig) == null ? void 0 : _g.documentPrefix;
+    if ((options == null ? void 0 : options.taskType) === "query" && queryPrefix) {
+      if (!content.startsWith(queryPrefix)) {
+        content = `${queryPrefix}${text}`;
+      }
+    } else if ((options == null ? void 0 : options.taskType) === "document" && docPrefix) {
+      if (!content.startsWith(docPrefix)) {
+        content = `${docPrefix}${text}`;
+      }
+    }
+    const response = await client.models.embedContent({
+      model,
+      contents: content
+    });
+    return (_j = (_i = (_h = response.embeddings) == null ? void 0 : _h[0]) == null ? void 0 : _i.values) != null ? _j : [];
+  }
+  async batchEmbed(texts, options) {
+    var _a, _b, _c, _d, _e, _f, _g, _h, _i;
+    const vectors = [];
+    const model = (_c = (_b = options == null ? void 0 : options.model) != null ? _b : (_a = this.embeddingConfig) == null ? void 0 : _a.model) != null ? _c : "text-embedding-004";
+    const apiKey = (_e = (_d = this.embeddingConfig) == null ? void 0 : _d.apiKey) != null ? _e : this.llmConfig.apiKey;
+    const client = apiKey !== this.llmConfig.apiKey ? new GoogleGenAI({ apiKey }) : this.client;
+    let contents = texts;
+    const queryPrefix = (_f = this.embeddingConfig) == null ? void 0 : _f.queryPrefix;
+    const docPrefix = (_g = this.embeddingConfig) == null ? void 0 : _g.documentPrefix;
+    if ((options == null ? void 0 : options.taskType) === "query" && queryPrefix) {
+      contents = texts.map((text) => text.startsWith(queryPrefix) ? text : `${queryPrefix}${text}`);
+    } else if ((options == null ? void 0 : options.taskType) === "document" && docPrefix) {
+      contents = texts.map((text) => text.startsWith(docPrefix) ? text : `${docPrefix}${text}`);
+    }
+    const response = await client.models.embedContent({
+      model,
+      contents
+    });
+    return (_i = (_h = response.embeddings) == null ? void 0 : _h.map((e) => {
+      var _a2;
+      return (_a2 = e.values) != null ? _a2 : [];
+    })) != null ? _i : [];
+  }
+  async ping() {
+    try {
+      await this.client.models.embedContent({
+        model: "text-embedding-004",
+        contents: "ping"
+      });
+      return true;
+    } catch (err) {
+      console.error("[GeminiProvider] Ping failed:", err);
+      return false;
+    }
+  }
+};
+
 // src/llm/providers/UniversalLLMAdapter.ts
 import axios2 from "axios";
 
@@ -1155,6 +1294,8 @@ var LLMFactory = class _LLMFactory {
         return new AnthropicProvider(llmConfig, embeddingConfig);
       case "ollama":
         return new OllamaProvider(llmConfig, embeddingConfig);
+      case "gemini":
+        return new GeminiProvider(llmConfig, embeddingConfig);
       case "rest":
       case "universal_rest":
       case "custom":
@@ -1546,8 +1687,7 @@ var EmbeddingStrategyResolver = class {
    * Check if an LLM provider natively supports embeddings
    */
   static supportsEmbedding(provider) {
-    const providersWithEmbedding = ["openai", "ollama", "rest", "universal_rest"];
-    return providersWithEmbedding.includes(provider);
+    return PROVIDERS_WITH_EMBEDDINGS.includes(provider);
   }
   /**
    * Get a human-readable description of the strategy

package/dist/handlers/index.d.mts

@@ -1,3 +1,3 @@
-import '../RagConfig-D3Inaf9N.mjs';
-export { c as createChatHandler, a as createHealthHandler, b as createIngestHandler, d as createUploadHandler } from '../index-Ymwm-_OR.mjs';
+import '../RagConfig-Bpp39-um.mjs';
+export { c as createChatHandler, a as createHealthHandler, b as createIngestHandler, d as createUploadHandler } from '../index-CZE72wnV.mjs';
 import 'next/server';

package/dist/handlers/index.d.ts

@@ -1,3 +1,3 @@
-import '../RagConfig-D3Inaf9N.js';
-export { c as createChatHandler, a as createHealthHandler, b as createIngestHandler, d as createUploadHandler } from '../index-BhNJQ2SS.js';
+import '../RagConfig-Bpp39-um.js';
+export { c as createChatHandler, a as createHealthHandler, b as createIngestHandler, d as createUploadHandler } from '../index-DSaQwkKA.js';
 import 'next/server';

package/dist/handlers/index.js

@@ -1179,7 +1179,7 @@ var import_server = require("next/server");
 // src/core/ConfigResolver.ts
 init_templateUtils();
 
-// src/config/serverConfig.ts
+// src/config/constants.ts
 var VECTOR_DB_PROVIDERS = [
   "pinecone",
   "pgvector",
@@ -1194,8 +1194,34 @@ var VECTOR_DB_PROVIDERS = [
   "universal_rest",
   "custom"
 ];
-var LLM_PROVIDERS = ["openai", "anthropic", "ollama", "rest", "universal_rest", "custom"];
-var EMBEDDING_PROVIDERS = ["openai", "ollama", "rest", "universal_rest", "custom"];
+var LLM_PROVIDERS = [
+  "openai",
+  "anthropic",
+  "ollama",
+  "gemini",
+  "rest",
+  "universal_rest",
+  "custom"
+];
+var EMBEDDING_PROVIDERS = [
+  "openai",
+  "ollama",
+  "gemini",
+  "rest",
+  "universal_rest",
+  "custom"
+];
+var PROVIDERS_WITH_EMBEDDINGS = [
+  "openai",
+  "ollama",
+  "gemini",
+  "rest",
+  "universal_rest"
+];
+var UI_VISUAL_STYLES = ["glass", "solid"];
+var UI_BORDER_RADIUS_OPTIONS = ["none", "sm", "md", "lg", "xl", "full"];
+
+// src/config/serverConfig.ts
 function readString(env, name) {
   var _a;
   const value = (_a = env[name]) == null ? void 0 : _a.trim();
@@ -1262,6 +1288,7 @@ function getRagConfig(env = process.env) {
   const llmApiKeyByProvider = {
     openai: readString(env, "OPENAI_API_KEY"),
     anthropic: readString(env, "ANTHROPIC_API_KEY"),
+    gemini: readString(env, "GEMINI_API_KEY"),
     ollama: void 0,
     universal_rest: readString(env, "LLM_API_KEY"),
     rest: readString(env, "LLM_API_KEY"),
@@ -1269,6 +1296,7 @@ function getRagConfig(env = process.env) {
   };
   const embeddingApiKeyByProvider = {
     openai: readString(env, "OPENAI_API_KEY"),
+    gemini: readString(env, "GEMINI_API_KEY"),
     ollama: void 0,
     universal_rest: (_q = readString(env, "EMBEDDING_API_KEY")) != null ? _q : readString(env, "OPENAI_API_KEY"),
     custom: (_r = readString(env, "EMBEDDING_API_KEY")) != null ? _r : readString(env, "OPENAI_API_KEY")
@@ -1683,6 +1711,16 @@ var ConfigValidator = class {
           });
         }
         break;
+      case "gemini":
+        if (!config.apiKey) {
+          errors.push({
+            field: "llm.apiKey",
+            message: "Gemini API key is required",
+            suggestion: "Set GEMINI_API_KEY environment variable",
+            severity: "error"
+          });
+        }
+        break;
       case "ollama":
         if (!config.baseUrl) {
           errors.push({
@@ -1749,6 +1787,14 @@ var ConfigValidator = class {
         severity: "error"
       });
     }
+    if (config.provider === "gemini" && !config.apiKey) {
+      errors.push({
+        field: "embedding.apiKey",
+        message: "Gemini API key is required for embedding",
+        suggestion: "Set GEMINI_API_KEY environment variable",
+        severity: "error"
+      });
+    }
     if (config.provider === "ollama" && !config.baseUrl) {
       errors.push({
         field: "embedding.baseUrl",
@@ -1779,21 +1825,19 @@ var ConfigValidator = class {
       }
     }
     if (config.borderRadius) {
-      const validValues = ["none", "sm", "md", "lg", "xl", "full"];
-      if (!validValues.includes(config.borderRadius)) {
+      if (!UI_BORDER_RADIUS_OPTIONS.includes(config.borderRadius)) {
         errors.push({
           field: "ui.borderRadius",
-          message: `borderRadius must be one of: ${validValues.join(", ")}`,
+          message: `borderRadius must be one of: ${UI_BORDER_RADIUS_OPTIONS.join(", ")}`,
           severity: "warning"
         });
       }
     }
     if (config.visualStyle) {
-      const validValues = ["glass", "solid"];
-      if (!validValues.includes(config.visualStyle)) {
+      if (!UI_VISUAL_STYLES.includes(config.visualStyle)) {
         errors.push({
           field: "ui.visualStyle",
-          message: `visualStyle must be one of: ${validValues.join(", ")}`,
+          message: `visualStyle must be one of: ${UI_VISUAL_STYLES.join(", ")}`,
           severity: "warning"
         });
       }
@@ -2127,6 +2171,101 @@ ${context}`;
   }
 };
 
+// src/llm/providers/GeminiProvider.ts
+var import_genai = require("@google/genai");
+var GeminiProvider = class {
+  constructor(llmConfig, embeddingConfig) {
+    if (!llmConfig.apiKey) throw new Error("[GeminiProvider] llmConfig.apiKey is required");
+    this.client = new import_genai.GoogleGenAI({ apiKey: llmConfig.apiKey });
+    this.llmConfig = llmConfig;
+    this.embeddingConfig = embeddingConfig;
+  }
+  async chat(messages, context, options) {
+    var _a, _b, _c, _d, _e, _f;
+    const systemPrompt = (_a = this.llmConfig.systemPrompt) != null ? _a : `You are a helpful assistant. Answer questions based on the provided context.
+
+Context:
+${context}`;
+    const systemInstruction = systemPrompt.includes("{{context}}") ? systemPrompt.replace("{{context}}", context) : `${systemPrompt}
+
+Context:
+${context}`;
+    const geminiMessages = messages.map((m) => ({
+      role: m.role === "assistant" ? "model" : "user",
+      parts: [{ text: m.content }]
+    }));
+    const response = await this.client.models.generateContent({
+      model: this.llmConfig.model,
+      contents: geminiMessages,
+      config: {
+        systemInstruction,
+        temperature: (_c = (_b = options == null ? void 0 : options.temperature) != null ? _b : this.llmConfig.temperature) != null ? _c : 0.7,
+        maxOutputTokens: (_e = (_d = options == null ? void 0 : options.maxTokens) != null ? _d : this.llmConfig.maxTokens) != null ? _e : 1024,
+        stopSequences: options == null ? void 0 : options.stop
+      }
+    });
+    return (_f = response.text) != null ? _f : "";
+  }
+  async embed(text, options) {
+    var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j;
+    const model = (_c = (_b = options == null ? void 0 : options.model) != null ? _b : (_a = this.embeddingConfig) == null ? void 0 : _a.model) != null ? _c : "text-embedding-004";
+    const apiKey = (_e = (_d = this.embeddingConfig) == null ? void 0 : _d.apiKey) != null ? _e : this.llmConfig.apiKey;
+    const client = apiKey !== this.llmConfig.apiKey ? new import_genai.GoogleGenAI({ apiKey }) : this.client;
+    let content = text;
+    const queryPrefix = (_f = this.embeddingConfig) == null ? void 0 : _f.queryPrefix;
+    const docPrefix = (_g = this.embeddingConfig) == null ? void 0 : _g.documentPrefix;
+    if ((options == null ? void 0 : options.taskType) === "query" && queryPrefix) {
+      if (!content.startsWith(queryPrefix)) {
+        content = `${queryPrefix}${text}`;
+      }
+    } else if ((options == null ? void 0 : options.taskType) === "document" && docPrefix) {
+      if (!content.startsWith(docPrefix)) {
+        content = `${docPrefix}${text}`;
+      }
+    }
+    const response = await client.models.embedContent({
+      model,
+      contents: content
+    });
+    return (_j = (_i = (_h = response.embeddings) == null ? void 0 : _h[0]) == null ? void 0 : _i.values) != null ? _j : [];
+  }
+  async batchEmbed(texts, options) {
+    var _a, _b, _c, _d, _e, _f, _g, _h, _i;
+    const vectors = [];
+    const model = (_c = (_b = options == null ? void 0 : options.model) != null ? _b : (_a = this.embeddingConfig) == null ? void 0 : _a.model) != null ? _c : "text-embedding-004";
+    const apiKey = (_e = (_d = this.embeddingConfig) == null ? void 0 : _d.apiKey) != null ? _e : this.llmConfig.apiKey;
+    const client = apiKey !== this.llmConfig.apiKey ? new import_genai.GoogleGenAI({ apiKey }) : this.client;
+    let contents = texts;
+    const queryPrefix = (_f = this.embeddingConfig) == null ? void 0 : _f.queryPrefix;
+    const docPrefix = (_g = this.embeddingConfig) == null ? void 0 : _g.documentPrefix;
+    if ((options == null ? void 0 : options.taskType) === "query" && queryPrefix) {
+      contents = texts.map((text) => text.startsWith(queryPrefix) ? text : `${queryPrefix}${text}`);
+    } else if ((options == null ? void 0 : options.taskType) === "document" && docPrefix) {
+      contents = texts.map((text) => text.startsWith(docPrefix) ? text : `${docPrefix}${text}`);
+    }
+    const response = await client.models.embedContent({
+      model,
+      contents
+    });
+    return (_i = (_h = response.embeddings) == null ? void 0 : _h.map((e) => {
+      var _a2;
+      return (_a2 = e.values) != null ? _a2 : [];
+    })) != null ? _i : [];
+  }
+  async ping() {
+    try {
+      await this.client.models.embedContent({
+        model: "text-embedding-004",
+        contents: "ping"
+      });
+      return true;
+    } catch (err) {
+      console.error("[GeminiProvider] Ping failed:", err);
+      return false;
+    }
+  }
+};
+
 // src/llm/providers/UniversalLLMAdapter.ts
 var import_axios2 = __toESM(require("axios"));
 init_templateUtils();
@@ -2285,6 +2424,8 @@ var LLMFactory = class _LLMFactory {
         return new AnthropicProvider(llmConfig, embeddingConfig);
       case "ollama":
         return new OllamaProvider(llmConfig, embeddingConfig);
+      case "gemini":
+        return new GeminiProvider(llmConfig, embeddingConfig);
       case "rest":
       case "universal_rest":
       case "custom":
@@ -2670,8 +2811,7 @@ var EmbeddingStrategyResolver = class {
    * Check if an LLM provider natively supports embeddings
    */
   static supportsEmbedding(provider) {
-    const providersWithEmbedding = ["openai", "ollama", "rest", "universal_rest"];
-    return providersWithEmbedding.includes(provider);
+    return PROVIDERS_WITH_EMBEDDINGS.includes(provider);
   }
   /**
    * Get a human-readable description of the strategy

package/dist/handlers/index.mjs

@@ -3,7 +3,7 @@ import {
   createHealthHandler,
   createIngestHandler,
   createUploadHandler
-} from "../chunk-GT72OIOD.mjs";
+} from "../chunk-SNPFZBDK.mjs";
 import "../chunk-EDLTMSNY.mjs";
 import "../chunk-FWCSY2DS.mjs";
 export {

package/dist/{index-Ymwm-_OR.d.mts → index-CZE72wnV.d.mts}

@@ -1,4 +1,4 @@
-import { c as VectorDBConfig, L as LLMConfig, E as EmbeddingConfig, a as RagConfig, C as ChatResponse } from './RagConfig-D3Inaf9N.mjs';
+import { c as VectorDBConfig, L as LLMConfig, E as EmbeddingConfig, a as RagConfig, C as ChatResponse } from './RagConfig-Bpp39-um.mjs';
 import { NextRequest, NextResponse } from 'next/server';
 
 /**

package/dist/{index-BhNJQ2SS.d.ts → index-DSaQwkKA.d.ts}

@@ -1,4 +1,4 @@
-import { c as VectorDBConfig, L as LLMConfig, E as EmbeddingConfig, a as RagConfig, C as ChatResponse } from './RagConfig-D3Inaf9N.js';
+import { c as VectorDBConfig, L as LLMConfig, E as EmbeddingConfig, a as RagConfig, C as ChatResponse } from './RagConfig-Bpp39-um.js';
 import { NextRequest, NextResponse } from 'next/server';
 
 /**

package/dist/index.d.mts

@@ -2,8 +2,8 @@ import * as react_jsx_runtime from 'react/jsx-runtime';
 import React$1, { ReactNode } from 'react';
 import { C as ChatMessage } from './DocumentChunker-BICIjSuG.mjs';
 export { a as ChatOptions, b as Chunk, c as ChunkOptions, E as EmbedOptions, I as ILLMProvider } from './DocumentChunker-BICIjSuG.mjs';
-import { V as VectorMatch, U as UIConfig } from './RagConfig-D3Inaf9N.mjs';
-export { C as ChatResponse, E as EmbeddingConfig, I as IngestDocument, L as LLMConfig, R as RAGConfig, a as RagConfig, b as UpsertDocument, c as VectorDBConfig } from './RagConfig-D3Inaf9N.mjs';
+import { V as VectorMatch, U as UIConfig } from './RagConfig-Bpp39-um.mjs';
+export { C as ChatResponse, E as EmbeddingConfig, I as IngestDocument, L as LLMConfig, R as RAGConfig, a as RagConfig, b as UpsertDocument, c as VectorDBConfig } from './RagConfig-Bpp39-um.mjs';
 
 interface ChatWidgetProps {
     /** Position of the floating button. Defaults to bottom-right. */

package/dist/index.d.ts

@@ -2,8 +2,8 @@ import * as react_jsx_runtime from 'react/jsx-runtime';
 import React$1, { ReactNode } from 'react';
 import { C as ChatMessage } from './DocumentChunker-BICIjSuG.js';
 export { a as ChatOptions, b as Chunk, c as ChunkOptions, E as EmbedOptions, I as ILLMProvider } from './DocumentChunker-BICIjSuG.js';
-import { V as VectorMatch, U as UIConfig } from './RagConfig-D3Inaf9N.js';
-export { C as ChatResponse, E as EmbeddingConfig, I as IngestDocument, L as LLMConfig, R as RAGConfig, a as RagConfig, b as UpsertDocument, c as VectorDBConfig } from './RagConfig-D3Inaf9N.js';
+import { V as VectorMatch, U as UIConfig } from './RagConfig-Bpp39-um.js';
+export { C as ChatResponse, E as EmbeddingConfig, I as IngestDocument, L as LLMConfig, R as RAGConfig, a as RagConfig, b as UpsertDocument, c as VectorDBConfig } from './RagConfig-Bpp39-um.js';
 
 interface ChatWidgetProps {
     /** Position of the floating button. Defaults to bottom-right. */

package/dist/server.d.mts

@@ -1,9 +1,9 @@
-import { a as RagConfig, C as ChatResponse, I as IngestDocument, c as VectorDBConfig, b as UpsertDocument, V as VectorMatch, L as LLMConfig, E as EmbeddingConfig, d as VectorDBProvider, e as LLMProvider, f as EmbeddingProvider } from './RagConfig-D3Inaf9N.mjs';
-export { R as RAGConfig, U as UIConfig } from './RagConfig-D3Inaf9N.mjs';
+import { a as RagConfig, C as ChatResponse, I as IngestDocument, c as VectorDBConfig, b as UpsertDocument, V as VectorMatch, L as LLMConfig, E as EmbeddingConfig, d as VectorDBProvider, e as LLMProvider, f as EmbeddingProvider } from './RagConfig-Bpp39-um.mjs';
+export { R as RAGConfig, U as UIConfig } from './RagConfig-Bpp39-um.mjs';
 import { C as ChatMessage, I as ILLMProvider, a as ChatOptions, E as EmbedOptions } from './DocumentChunker-BICIjSuG.mjs';
 export { b as Chunk, c as ChunkOptions, D as DocumentChunker } from './DocumentChunker-BICIjSuG.mjs';
-import { H as HealthCheckResult } from './index-Ymwm-_OR.mjs';
-export { P as ProviderHealthCheck, c as createChatHandler, a as createHealthHandler, b as createIngestHandler, d as createUploadHandler } from './index-Ymwm-_OR.mjs';
+import { H as HealthCheckResult } from './index-CZE72wnV.mjs';
+export { P as ProviderHealthCheck, c as createChatHandler, a as createHealthHandler, b as createIngestHandler, d as createUploadHandler } from './index-CZE72wnV.mjs';
 import 'next/server';
 
 /**