@retrivora-ai/rag-engine 0.3.1 → 0.3.3

package/dist/server.d.ts

@@ -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.js';
-export { R as RAGConfig, U as UIConfig } from './RagConfig-D3Inaf9N.js';
+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.js';
+export { R as RAGConfig, U as UIConfig } from './RagConfig-Bpp39-um.js';
 import { C as ChatMessage, I as ILLMProvider, a as ChatOptions, E as EmbedOptions } from './DocumentChunker-BICIjSuG.js';
 export { b as Chunk, c as ChunkOptions, D as DocumentChunker } from './DocumentChunker-BICIjSuG.js';
-import { H as HealthCheckResult } from './index-BhNJQ2SS.js';
-export { P as ProviderHealthCheck, c as createChatHandler, a as createHealthHandler, b as createIngestHandler, d as createUploadHandler } from './index-BhNJQ2SS.js';
+import { H as HealthCheckResult } from './index-DSaQwkKA.js';
+export { P as ProviderHealthCheck, c as createChatHandler, a as createHealthHandler, b as createIngestHandler, d as createUploadHandler } from './index-DSaQwkKA.js';
 import 'next/server';
 
 /**

package/dist/server.js

@@ -1223,7 +1223,7 @@ module.exports = __toCommonJS(server_exports);
 // src/core/ConfigResolver.ts
 init_templateUtils();
 
-// src/config/serverConfig.ts
+// src/config/constants.ts
 var VECTOR_DB_PROVIDERS = [
   "pinecone",
   "pgvector",
@@ -1238,8 +1238,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();
@@ -1306,6 +1332,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"),
@@ -1313,6 +1340,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")
@@ -1727,6 +1755,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({
@@ -1793,6 +1831,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",
@@ -1823,21 +1869,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"
         });
       }
@@ -2171,6 +2215,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();
@@ -2368,6 +2507,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":
@@ -2759,8 +2900,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/server.mjs

@@ -34,7 +34,7 @@ import {
   createIngestHandler,
   createUploadHandler,
   getRagConfig
-} from "./chunk-GT72OIOD.mjs";
+} from "./chunk-SNPFZBDK.mjs";
 import "./chunk-EDLTMSNY.mjs";
 import {
   PineconeProvider

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@retrivora-ai/rag-engine",
-  "version": "0.3.1",
+  "version": "0.3.3",
   "description": "Retrivora AI is a plug-and-play AI engine for RAG chat experiences — generic vector DB + LLM provider, embeddable or standalone.",
   "author": "Abhinav Alkuchi",
   "license": "MIT",
@@ -85,6 +85,7 @@
   },
   "dependencies": {
     "@anthropic-ai/sdk": "^0.90.0",
+    "@google/genai": "^1.50.1",
     "@pinecone-database/pinecone": "^7.2.0",
     "axios": "^1.15.0",
     "lucide-react": "^1.8.0",

package/src/config/EmbeddingStrategy.ts

@@ -12,6 +12,7 @@
 import { ILLMProvider } from '../llm/ILLMProvider';
 import { LLMFactory } from '../llm/LLMFactory';
 import { LLMConfig, EmbeddingConfig, LLMProvider, EmbeddingProvider } from './RagConfig';
+import { PROVIDERS_WITH_EMBEDDINGS } from './constants';
 
 export enum EmbeddingStrategy {
   /**
@@ -125,8 +126,7 @@ export class EmbeddingStrategyResolver {
    * Check if an LLM provider natively supports embeddings
    */
   private static supportsEmbedding(provider: LLMProvider | string): boolean {
-    const providersWithEmbedding = ['openai', 'ollama', 'rest', 'universal_rest'];
-    return providersWithEmbedding.includes(provider);
+    return (PROVIDERS_WITH_EMBEDDINGS as readonly string[]).includes(provider);
   }
 
   /**

package/src/config/RagConfig.ts

@@ -8,7 +8,13 @@
 // Vector DB
 // ---------------------------------------------------------------------------
 
-export type VectorDBProvider = 'pinecone' | 'pgvector' | 'postgresql' | 'mongodb' | 'milvus' | 'qdrant' | 'chromadb' | 'redis' | 'weaviate' | 'rest' | 'universal_rest' | 'custom';
+import {
+  VECTOR_DB_PROVIDERS,
+  LLM_PROVIDERS,
+  EMBEDDING_PROVIDERS,
+} from './constants';
+
+export type VectorDBProvider = typeof VECTOR_DB_PROVIDERS[number];
 
 export interface VectorDBConfig {
   /** Which vector database to use */
@@ -40,7 +46,7 @@ export interface VectorDBConfig {
 // LLM Provider
 // ---------------------------------------------------------------------------
 
-export type LLMProvider = 'openai' | 'anthropic' | 'ollama' | 'rest' | 'universal_rest' | 'custom';
+export type LLMProvider = typeof LLM_PROVIDERS[number];
 
 export interface LLMConfig {
   /** Which LLM provider to use */
@@ -72,7 +78,7 @@ export interface LLMConfig {
 // Embedding
 // ---------------------------------------------------------------------------
 
-export type EmbeddingProvider = 'openai' | 'ollama' | 'rest' | 'universal_rest' | 'custom';
+export type EmbeddingProvider = typeof EMBEDDING_PROVIDERS[number];
 
 export interface EmbeddingConfig {
   /** Which embedding provider to use */

package/src/config/constants.ts

@@ -0,0 +1,52 @@
+/**
+ * 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.
+ */
+
+export const VECTOR_DB_PROVIDERS = [
+  'pinecone',
+  'pgvector',
+  'postgresql',
+  'mongodb',
+  'milvus',
+  'qdrant',
+  'chromadb',
+  'redis',
+  'weaviate',
+  'rest',
+  'universal_rest',
+  'custom',
+] as const;
+
+export const LLM_PROVIDERS = [
+  'openai',
+  'anthropic',
+  'ollama',
+  'gemini',
+  'rest',
+  'universal_rest',
+  'custom',
+] as const;
+
+export const EMBEDDING_PROVIDERS = [
+  'openai',
+  'ollama',
+  'gemini',
+  'rest',
+  'universal_rest',
+  'custom',
+] as const;
+
+export const PROVIDERS_WITH_EMBEDDINGS = [
+  'openai',
+  'ollama',
+  'gemini',
+  'rest',
+  'universal_rest',
+] as const;
+
+export const UI_VISUAL_STYLES = ['glass', 'solid'] as const;
+
+export const UI_BORDER_RADIUS_OPTIONS = ['none', 'sm', 'md', 'lg', 'xl', 'full'] as const;

package/src/config/serverConfig.ts

@@ -19,14 +19,11 @@
  */
 
 import { RagConfig } from './RagConfig';
-
-const VECTOR_DB_PROVIDERS = [
-  'pinecone', 'pgvector', 'postgresql', 'mongodb', 'milvus',
-  'qdrant', 'chromadb', 'redis', 'weaviate', 'rest', 'universal_rest', 'custom',
-] as const;
-
-const LLM_PROVIDERS = ['openai', 'anthropic', 'ollama', 'rest', 'universal_rest', 'custom'] as const;
-const EMBEDDING_PROVIDERS = ['openai', 'ollama', 'rest', 'universal_rest', 'custom'] as const;
+import {
+  VECTOR_DB_PROVIDERS,
+  LLM_PROVIDERS,
+  EMBEDDING_PROVIDERS,
+} from './constants';
 
 function readString(env: Record<string, string | undefined>, name: string): string | undefined {
   const value = env[name]?.trim();
@@ -112,6 +109,7 @@ export function getRagConfig(env: Record<string, string | undefined> = process.e
   const llmApiKeyByProvider: Record<string, string | undefined> = {
     openai: readString(env, 'OPENAI_API_KEY'),
     anthropic: readString(env, 'ANTHROPIC_API_KEY'),
+    gemini: readString(env, 'GEMINI_API_KEY'),
     ollama: undefined,
     universal_rest: readString(env, 'LLM_API_KEY'),
     rest: readString(env, 'LLM_API_KEY'),
@@ -120,6 +118,7 @@ export function getRagConfig(env: Record<string, string | undefined> = process.e
 
   const embeddingApiKeyByProvider: Record<string, string | undefined> = {
     openai: readString(env, 'OPENAI_API_KEY'),
+    gemini: readString(env, 'GEMINI_API_KEY'),
     ollama: undefined,
     universal_rest: readString(env, 'EMBEDDING_API_KEY') ?? readString(env, 'OPENAI_API_KEY'),
     custom: readString(env, 'EMBEDDING_API_KEY') ?? readString(env, 'OPENAI_API_KEY'),

package/src/core/ConfigValidator.ts

@@ -10,6 +10,10 @@
  */
 
 import { RagConfig, VectorDBConfig, LLMConfig, EmbeddingConfig } from '../config/RagConfig';
+import {
+  UI_VISUAL_STYLES,
+  UI_BORDER_RADIUS_OPTIONS,
+} from '../config/constants';
 
 export interface ValidationError {
   field: string;
@@ -380,6 +384,16 @@ export class ConfigValidator {
           });
         }
         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({
@@ -454,6 +468,15 @@ export class ConfigValidator {
       });
     }
 
+    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',
@@ -489,22 +512,20 @@ export class ConfigValidator {
     }
 
     if (config.borderRadius) {
-      const validValues = ['none', 'sm', 'md', 'lg', 'xl', 'full'];
-      if (!validValues.includes(config.borderRadius as string)) {
+      if (!UI_BORDER_RADIUS_OPTIONS.includes(config.borderRadius as any)) {
         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 as string)) {
+      if (!UI_VISUAL_STYLES.includes(config.visualStyle as any)) {
         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',
         });
       }

package/src/llm/LLMFactory.ts

@@ -10,6 +10,7 @@ import { LLMConfig, EmbeddingConfig, LLMProvider } from '../config/RagConfig';
 import { OpenAIProvider } from './providers/OpenAIProvider';
 import { AnthropicProvider } from './providers/AnthropicProvider';
 import { OllamaProvider } from './providers/OllamaProvider';
+import { GeminiProvider } from './providers/GeminiProvider';
 import { UniversalLLMAdapter } from './providers/UniversalLLMAdapter';
 
 export class LLMFactory {
@@ -21,6 +22,8 @@ export 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':

package/src/llm/providers/GeminiProvider.ts

@@ -0,0 +1,147 @@
+/**
+ * Gemini LLM Provider (Google Gen AI)
+ *
+ * Handles both chat completion (gemini-2.5-flash, gemini-2.5-pro, etc.) and
+ * embedding (text-embedding-004).
+ * Uses the modern @google/genai SDK.
+ *
+ * Required LLMConfig fields:
+ *   - apiKey: string  – Gemini API key
+ *   - model:  string  – e.g. "gemini-2.5-flash"
+ *
+ * Required EmbeddingConfig fields (when used for embedding):
+ *   - apiKey: string  – same or separate key
+ *   - model:  string  – e.g. "text-embedding-004"
+ */
+
+import { GoogleGenAI } from '@google/genai';
+import { ILLMProvider, ChatMessage, ChatOptions, EmbedOptions } from '../ILLMProvider';
+import { LLMConfig, EmbeddingConfig } from '../../config/RagConfig';
+
+export class GeminiProvider implements ILLMProvider {
+  private readonly client: GoogleGenAI;
+  private readonly llmConfig: LLMConfig;
+  private readonly embeddingConfig?: EmbeddingConfig;
+
+  constructor(llmConfig: LLMConfig, embeddingConfig?: 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: ChatMessage[], context: string, options?: ChatOptions): Promise<string> {
+    const systemPrompt =
+      this.llmConfig.systemPrompt ??
+      `You are a helpful assistant. Answer questions based on the provided context.\n\nContext:\n${context}`;
+
+    const systemInstruction = systemPrompt.includes('{{context}}')
+      ? systemPrompt.replace('{{context}}', context)
+      : `${systemPrompt}\n\nContext:\n${context}`;
+
+    // Gemini requires alternating user/model messages, starting with user.
+    // If the first message is not user, or there are consecutive same-role messages, we need to adapt.
+    // To simplify and comply with typical usage, we'll map 'assistant' to 'model'.
+    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: options?.temperature ?? this.llmConfig.temperature ?? 0.7,
+        maxOutputTokens: options?.maxTokens ?? this.llmConfig.maxTokens ?? 1024,
+        stopSequences: options?.stop,
+      },
+    });
+
+    return response.text ?? '';
+  }
+
+  async embed(text: string, options?: EmbedOptions): Promise<number[]> {
+    const model =
+      options?.model ??
+      this.embeddingConfig?.model ??
+      'text-embedding-004';
+
+    const apiKey = this.embeddingConfig?.apiKey ?? this.llmConfig.apiKey;
+    const client = apiKey !== this.llmConfig.apiKey
+      ? new GoogleGenAI({ apiKey })
+      : this.client;
+
+    let content = text;
+    
+    // Dynamically apply prefixes from configuration if they exist
+    const queryPrefix = this.embeddingConfig?.queryPrefix;
+    const docPrefix = this.embeddingConfig?.documentPrefix;
+    
+    if (options?.taskType === 'query' && queryPrefix) {
+      if (!content.startsWith(queryPrefix)) {
+        content = `${queryPrefix}${text}`;
+      }
+    } else if (options?.taskType === 'document' && docPrefix) {
+      if (!content.startsWith(docPrefix)) {
+        content = `${docPrefix}${text}`;
+      }
+    }
+
+    const response = await client.models.embedContent({
+      model,
+      contents: content,
+    });
+
+    return response.embeddings?.[0]?.values ?? [];
+  }
+
+  async batchEmbed(texts: string[], options?: EmbedOptions): Promise<number[][]> {
+    const vectors: number[][] = [];
+    // Sequential fallback for Gemini (API doesn't have a direct batch endpoint in the same way, though you can pass multiple contents.
+    // For simplicity and to match the other providers' structure, we'll iterate or pass array.
+    // The new SDK supports an array of contents.
+    const model =
+      options?.model ??
+      this.embeddingConfig?.model ??
+      'text-embedding-004';
+
+    const apiKey = this.embeddingConfig?.apiKey ?? this.llmConfig.apiKey;
+    const client = apiKey !== this.llmConfig.apiKey
+      ? new GoogleGenAI({ apiKey })
+      : this.client;
+
+    let contents = texts;
+    
+    // Apply prefixes if needed
+    const queryPrefix = this.embeddingConfig?.queryPrefix;
+    const docPrefix = this.embeddingConfig?.documentPrefix;
+    
+    if (options?.taskType === 'query' && queryPrefix) {
+      contents = texts.map(text => text.startsWith(queryPrefix) ? text : `${queryPrefix}${text}`);
+    } else if (options?.taskType === 'document' && docPrefix) {
+      contents = texts.map(text => text.startsWith(docPrefix) ? text : `${docPrefix}${text}`);
+    }
+
+    const response = await client.models.embedContent({
+      model,
+      contents,
+    });
+
+    return response.embeddings?.map(e => e.values ?? []) ?? [];
+  }
+
+  async ping(): Promise<boolean> {
+    try {
+      // Perform a lightweight embed call to verify connectivity
+      await this.client.models.embedContent({
+        model: 'text-embedding-004',
+        contents: 'ping',
+      });
+      return true;
+    } catch (err) {
+      console.error('[GeminiProvider] Ping failed:', err);
+      return false;
+    }
+  }
+}