@retrivora-ai/rag-engine 0.3.3 → 0.3.5

package/dist/{ChromaDBProvider-T7TK3ONZ.mjs → ChromaDBProvider-6JWXHZAE.mjs}

@@ -1,7 +1,7 @@
 import {
   ChromaDBProvider
-} from "./chunk-VKE5ZW7Y.mjs";
-import "./chunk-VOIWNO5O.mjs";
+} from "./chunk-Q2QDQ5SQ.mjs";
+import "./chunk-YVTGOFNV.mjs";
 import "./chunk-FWCSY2DS.mjs";
 export {
   ChromaDBProvider

package/dist/{MilvusProvider-Y5FV5EAE.mjs → MilvusProvider-MJQVHU4B.mjs}

@@ -1,7 +1,7 @@
 import {
   MilvusProvider
-} from "./chunk-VEJNRS4B.mjs";
-import "./chunk-VOIWNO5O.mjs";
+} from "./chunk-ENEUABSO.mjs";
+import "./chunk-YVTGOFNV.mjs";
 import "./chunk-FWCSY2DS.mjs";
 export {
   MilvusProvider

package/dist/{MongoDBProvider-KGO6N23T.mjs → MongoDBProvider-XS3QR47H.mjs}

@@ -1,7 +1,7 @@
 import {
   MongoDBProvider
-} from "./chunk-IFPISZ2S.mjs";
-import "./chunk-VOIWNO5O.mjs";
+} from "./chunk-KGW2YHV2.mjs";
+import "./chunk-YVTGOFNV.mjs";
 import "./chunk-FWCSY2DS.mjs";
 export {
   MongoDBProvider

package/dist/{PineconeProvider-A47MRRYJ.mjs → PineconeProvider-QZTC7QAP.mjs}

@@ -1,7 +1,7 @@
 import {
   PineconeProvider
-} from "./chunk-BMHJTWSU.mjs";
-import "./chunk-VOIWNO5O.mjs";
+} from "./chunk-M3AZTD7F.mjs";
+import "./chunk-YVTGOFNV.mjs";
 import "./chunk-FWCSY2DS.mjs";
 export {
   PineconeProvider

package/dist/{PostgreSQLProvider-ILWADFAP.mjs → PostgreSQLProvider-CNO72UP6.mjs}

@@ -1,7 +1,7 @@
 import {
   PostgreSQLProvider
-} from "./chunk-6GSARSCP.mjs";
-import "./chunk-VOIWNO5O.mjs";
+} from "./chunk-L54WODRW.mjs";
+import "./chunk-YVTGOFNV.mjs";
 import "./chunk-FWCSY2DS.mjs";
 export {
   PostgreSQLProvider

package/dist/{QdrantProvider-OLPJK7CY.mjs → QdrantProvider-6OBJKUIP.mjs}

@@ -1,7 +1,7 @@
 import {
   QdrantProvider
-} from "./chunk-W2PQR3UK.mjs";
-import "./chunk-VOIWNO5O.mjs";
+} from "./chunk-ULX4EF74.mjs";
+import "./chunk-YVTGOFNV.mjs";
 import "./chunk-FWCSY2DS.mjs";
 export {
   QdrantProvider

package/dist/{RedisProvider-ANEJ3BHR.mjs → RedisProvider-FF5BNQPB.mjs}

@@ -1,7 +1,7 @@
 import {
   RedisProvider
-} from "./chunk-HOMXEE3M.mjs";
-import "./chunk-VOIWNO5O.mjs";
+} from "./chunk-N2L5CFGM.mjs";
+import "./chunk-YVTGOFNV.mjs";
 import "./chunk-FWCSY2DS.mjs";
 export {
   RedisProvider

package/dist/{UniversalVectorProvider-QJIV2AJJ.mjs → UniversalVectorProvider-CAVX5VSQ.mjs}

@@ -1,8 +1,8 @@
 import {
   UniversalVectorProvider
-} from "./chunk-VV2ML6TM.mjs";
+} from "./chunk-R3GPRETL.mjs";
 import "./chunk-EDLTMSNY.mjs";
-import "./chunk-VOIWNO5O.mjs";
+import "./chunk-YVTGOFNV.mjs";
 import "./chunk-FWCSY2DS.mjs";
 export {
   UniversalVectorProvider

package/dist/{WeaviateProvider-WIK2QN23.mjs → WeaviateProvider-DIYYBTEF.mjs}

@@ -1,7 +1,7 @@
 import {
   WeaviateProvider
-} from "./chunk-RUKZC3ON.mjs";
-import "./chunk-VOIWNO5O.mjs";
+} from "./chunk-7XL3QSQ7.mjs";
+import "./chunk-YVTGOFNV.mjs";
 import "./chunk-FWCSY2DS.mjs";
 export {
   WeaviateProvider

package/dist/{chunk-RUKZC3ON.mjs → chunk-7XL3QSQ7.mjs}

@@ -1,6 +1,6 @@
 import {
   BaseVectorProvider
-} from "./chunk-VOIWNO5O.mjs";
+} from "./chunk-YVTGOFNV.mjs";
 import {
   __spreadValues
 } from "./chunk-FWCSY2DS.mjs";

package/dist/{chunk-VEJNRS4B.mjs → chunk-ENEUABSO.mjs}

@@ -1,6 +1,6 @@
 import {
   BaseVectorProvider
-} from "./chunk-VOIWNO5O.mjs";
+} from "./chunk-YVTGOFNV.mjs";
 import {
   __spreadValues
 } from "./chunk-FWCSY2DS.mjs";

package/dist/{chunk-IFPISZ2S.mjs → chunk-KGW2YHV2.mjs}

@@ -1,18 +1,12 @@
 import {
   BaseVectorProvider
-} from "./chunk-VOIWNO5O.mjs";
+} from "./chunk-YVTGOFNV.mjs";
 import {
   __spreadValues
 } from "./chunk-FWCSY2DS.mjs";
 
 // src/providers/vectordb/MongoDBProvider.ts
 import { MongoClient } from "mongodb";
-function stripInternalFilterKeys(filter) {
-  if (!filter) return {};
-  return Object.fromEntries(
-    Object.entries(filter).filter(([key]) => !key.startsWith("__"))
-  );
-}
 var MongoDBProvider = class extends BaseVectorProvider {
   constructor(config) {
     super(config);
@@ -58,7 +52,7 @@ var MongoDBProvider = class extends BaseVectorProvider {
     await this.collection.bulkWrite(operations);
   }
   async query(vector, topK, namespace, filter) {
-    const publicFilter = stripInternalFilterKeys(filter);
+    const publicFilter = this.sanitizeFilter(filter);
     const pipeline = [
       {
         $vectorSearch: __spreadValues({

package/dist/{chunk-6GSARSCP.mjs → chunk-L54WODRW.mjs}

@@ -1,15 +1,9 @@
 import {
   BaseVectorProvider
-} from "./chunk-VOIWNO5O.mjs";
+} from "./chunk-YVTGOFNV.mjs";
 
 // src/providers/vectordb/PostgreSQLProvider.ts
 import { Pool } from "pg";
-function stripInternalFilterKeys(filter) {
-  if (!filter) return {};
-  return Object.fromEntries(
-    Object.entries(filter).filter(([key]) => !key.startsWith("__"))
-  );
-}
 var PostgreSQLProvider = class extends BaseVectorProvider {
   constructor(config) {
     var _a;
@@ -96,7 +90,7 @@ var PostgreSQLProvider = class extends BaseVectorProvider {
     let whereClause = namespace ? `WHERE namespace = $3` : "";
     const params = [vectorLiteral, topK];
     if (namespace) params.push(namespace);
-    const publicFilter = stripInternalFilterKeys(filter);
+    const publicFilter = this.sanitizeFilter(filter);
     if (Object.keys(publicFilter).length > 0) {
       const filterConditions = Object.entries(publicFilter).map(([key, val]) => {
         const paramIdx = params.length + 1;

package/dist/{chunk-BMHJTWSU.mjs → chunk-M3AZTD7F.mjs}

@@ -1,6 +1,6 @@
 import {
   BaseVectorProvider
-} from "./chunk-VOIWNO5O.mjs";
+} from "./chunk-YVTGOFNV.mjs";
 import {
   __spreadValues
 } from "./chunk-FWCSY2DS.mjs";
@@ -55,11 +55,12 @@ var PineconeProvider = class extends BaseVectorProvider {
   }
   async query(vector, topK, namespace, filter) {
     var _a;
+    const pineconeFilter = this.sanitizeFilter(filter);
     const result = await this.index(namespace).query(__spreadValues({
       vector,
       topK,
       includeMetadata: true
-    }, filter ? { filter } : {}));
+    }, Object.keys(pineconeFilter).length > 0 ? { filter: pineconeFilter } : {}));
     return ((_a = result.matches) != null ? _a : []).map((m) => {
       var _a2, _b;
       return {

package/dist/{chunk-HOMXEE3M.mjs → chunk-N2L5CFGM.mjs}

@@ -1,6 +1,6 @@
 import {
   BaseVectorProvider
-} from "./chunk-VOIWNO5O.mjs";
+} from "./chunk-YVTGOFNV.mjs";
 import {
   __spreadValues
 } from "./chunk-FWCSY2DS.mjs";

package/dist/{chunk-VKE5ZW7Y.mjs → chunk-Q2QDQ5SQ.mjs}

@@ -1,6 +1,6 @@
 import {
   BaseVectorProvider
-} from "./chunk-VOIWNO5O.mjs";
+} from "./chunk-YVTGOFNV.mjs";
 import {
   __spreadValues
 } from "./chunk-FWCSY2DS.mjs";

package/dist/{chunk-VV2ML6TM.mjs → chunk-R3GPRETL.mjs}

@@ -4,7 +4,7 @@ import {
 } from "./chunk-EDLTMSNY.mjs";
 import {
   BaseVectorProvider
-} from "./chunk-VOIWNO5O.mjs";
+} from "./chunk-YVTGOFNV.mjs";
 import {
   __spreadValues
 } from "./chunk-FWCSY2DS.mjs";
@@ -75,11 +75,12 @@ var UniversalVectorProvider = class extends BaseVectorProvider {
       namespace: "{{namespace}}",
       filter: "{{filter}}"
     });
+    const sanitizedFilter = this.sanitizeFilter(filter);
     const payload = buildPayload(template, {
       vector,
       topK,
       namespace: namespace != null ? namespace : this.indexName,
-      filter: filter != null ? filter : {}
+      filter: sanitizedFilter
     });
     try {
       const response = await this.http.post(endpoint, payload);

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

@@ -1009,8 +1009,18 @@ 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;
+    this.llmConfig = __spreadProps(__spreadValues({}, llmConfig), {
+      model: this.sanitizeModel(llmConfig.model)
+    });
+    if (embeddingConfig) {
+      this.embeddingConfig = __spreadProps(__spreadValues({}, embeddingConfig), {
+        model: this.sanitizeModel(embeddingConfig.model)
+      });
+    }
+  }
+  sanitizeModel(model) {
+    if (!model) return model;
+    return model.split(":")[0];
   }
   async chat(messages, context, options) {
     var _a, _b, _c, _d, _e, _f;
@@ -1039,8 +1049,10 @@ ${context}`;
     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";
+    var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k;
+    const model = this.sanitizeModel(
+      (_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;
@@ -1057,14 +1069,20 @@ ${context}`;
     }
     const response = await client.models.embedContent({
       model,
-      contents: content
+      contents: content,
+      config: ((_h = this.embeddingConfig) == null ? void 0 : _h.dimensions) ? { outputDimensionality: this.embeddingConfig.dimensions } : void 0
+    }).catch((err) => {
+      console.error(`[GeminiProvider] Embedding failed for model "${model}":`, err.message);
+      throw err;
     });
-    return (_j = (_i = (_h = response.embeddings) == null ? void 0 : _h[0]) == null ? void 0 : _i.values) != null ? _j : [];
+    return (_k = (_j = (_i = response.embeddings) == null ? void 0 : _i[0]) == null ? void 0 : _j.values) != null ? _k : [];
   }
   async batchEmbed(texts, options) {
-    var _a, _b, _c, _d, _e, _f, _g, _h, _i;
+    var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j;
     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 model = this.sanitizeModel(
+      (_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;
@@ -1077,12 +1095,16 @@ ${context}`;
     }
     const response = await client.models.embedContent({
       model,
-      contents
+      contents,
+      config: ((_h = this.embeddingConfig) == null ? void 0 : _h.dimensions) ? { outputDimensionality: this.embeddingConfig.dimensions } : void 0
+    }).catch((err) => {
+      console.error(`[GeminiProvider] Batch embedding failed for model "${model}":`, err.message);
+      throw err;
     });
-    return (_i = (_h = response.embeddings) == null ? void 0 : _h.map((e) => {
+    return (_j = (_i = response.embeddings) == null ? void 0 : _i.map((e) => {
       var _a2;
       return (_a2 = e.values) != null ? _a2 : [];
-    })) != null ? _i : [];
+    })) != null ? _j : [];
   }
   async ping() {
     try {
@@ -1348,41 +1370,41 @@ var ProviderRegistry = class {
     }
     switch (provider) {
       case "pinecone": {
-        const { PineconeProvider } = await import("./PineconeProvider-A47MRRYJ.mjs");
+        const { PineconeProvider } = await import("./PineconeProvider-QZTC7QAP.mjs");
         return new PineconeProvider(config);
       }
       case "pgvector":
       case "postgresql": {
-        const { PostgreSQLProvider } = await import("./PostgreSQLProvider-ILWADFAP.mjs");
+        const { PostgreSQLProvider } = await import("./PostgreSQLProvider-CNO72UP6.mjs");
         return new PostgreSQLProvider(config);
       }
       case "mongodb": {
-        const { MongoDBProvider } = await import("./MongoDBProvider-KGO6N23T.mjs");
+        const { MongoDBProvider } = await import("./MongoDBProvider-XS3QR47H.mjs");
         return new MongoDBProvider(config);
       }
       case "milvus": {
-        const { MilvusProvider } = await import("./MilvusProvider-Y5FV5EAE.mjs");
+        const { MilvusProvider } = await import("./MilvusProvider-MJQVHU4B.mjs");
         return new MilvusProvider(config);
       }
       case "qdrant": {
-        const { QdrantProvider } = await import("./QdrantProvider-OLPJK7CY.mjs");
+        const { QdrantProvider } = await import("./QdrantProvider-6OBJKUIP.mjs");
         return new QdrantProvider(config);
       }
       case "chromadb": {
-        const { ChromaDBProvider } = await import("./ChromaDBProvider-T7TK3ONZ.mjs");
+        const { ChromaDBProvider } = await import("./ChromaDBProvider-6JWXHZAE.mjs");
         return new ChromaDBProvider(config);
       }
       case "redis": {
-        const { RedisProvider } = await import("./RedisProvider-ANEJ3BHR.mjs");
+        const { RedisProvider } = await import("./RedisProvider-FF5BNQPB.mjs");
         return new RedisProvider(config);
       }
       case "weaviate": {
-        const { WeaviateProvider } = await import("./WeaviateProvider-WIK2QN23.mjs");
+        const { WeaviateProvider } = await import("./WeaviateProvider-DIYYBTEF.mjs");
         return new WeaviateProvider(config);
       }
       case "universal_rest":
       case "rest": {
-        const { UniversalVectorProvider } = await import("./UniversalVectorProvider-QJIV2AJJ.mjs");
+        const { UniversalVectorProvider } = await import("./UniversalVectorProvider-CAVX5VSQ.mjs");
         return new UniversalVectorProvider(config);
       }
       default:

package/dist/{chunk-W2PQR3UK.mjs → chunk-ULX4EF74.mjs}

@@ -1,6 +1,6 @@
 import {
   BaseVectorProvider
-} from "./chunk-VOIWNO5O.mjs";
+} from "./chunk-YVTGOFNV.mjs";
 import {
   __spreadValues
 } from "./chunk-FWCSY2DS.mjs";

package/dist/chunk-YVTGOFNV.mjs

@@ -0,0 +1,23 @@
+// src/providers/vectordb/BaseVectorProvider.ts
+var BaseVectorProvider = class {
+  constructor(config) {
+    this.config = config;
+    this.indexName = config.indexName || "default";
+  }
+  /**
+   * Remove internal keys (starting with __) and queryText from a filter object
+   * before passing it to an external provider that might not support them.
+   */
+  sanitizeFilter(filter) {
+    if (!filter) return {};
+    return Object.fromEntries(
+      Object.entries(filter).filter(
+        ([key]) => !key.startsWith("__") && key !== "queryText"
+      )
+    );
+  }
+};
+
+export {
+  BaseVectorProvider
+};

package/dist/handlers/index.js

@@ -102,6 +102,18 @@ var init_BaseVectorProvider = __esm({
         this.config = config;
         this.indexName = config.indexName || "default";
       }
+      /**
+       * Remove internal keys (starting with __) and queryText from a filter object
+       * before passing it to an external provider that might not support them.
+       */
+      sanitizeFilter(filter) {
+        if (!filter) return {};
+        return Object.fromEntries(
+          Object.entries(filter).filter(
+            ([key]) => !key.startsWith("__") && key !== "queryText"
+          )
+        );
+      }
     };
   }
 });
@@ -165,11 +177,12 @@ var init_PineconeProvider = __esm({
       }
       async query(vector, topK, namespace, filter) {
         var _a;
+        const pineconeFilter = this.sanitizeFilter(filter);
         const result = await this.index(namespace).query(__spreadValues({
           vector,
           topK,
           includeMetadata: true
-        }, filter ? { filter } : {}));
+        }, Object.keys(pineconeFilter).length > 0 ? { filter: pineconeFilter } : {}));
         return ((_a = result.matches) != null ? _a : []).map((m) => {
           var _a2, _b;
           return {
@@ -205,12 +218,6 @@ var PostgreSQLProvider_exports = {};
 __export(PostgreSQLProvider_exports, {
   PostgreSQLProvider: () => PostgreSQLProvider
 });
-function stripInternalFilterKeys(filter) {
-  if (!filter) return {};
-  return Object.fromEntries(
-    Object.entries(filter).filter(([key]) => !key.startsWith("__"))
-  );
-}
 var import_pg, PostgreSQLProvider;
 var init_PostgreSQLProvider = __esm({
   "src/providers/vectordb/PostgreSQLProvider.ts"() {
@@ -303,7 +310,7 @@ var init_PostgreSQLProvider = __esm({
         let whereClause = namespace ? `WHERE namespace = $3` : "";
         const params = [vectorLiteral, topK];
         if (namespace) params.push(namespace);
-        const publicFilter = stripInternalFilterKeys(filter);
+        const publicFilter = this.sanitizeFilter(filter);
         if (Object.keys(publicFilter).length > 0) {
           const filterConditions = Object.entries(publicFilter).map(([key, val]) => {
             const paramIdx = params.length + 1;
@@ -355,12 +362,6 @@ var MongoDBProvider_exports = {};
 __export(MongoDBProvider_exports, {
   MongoDBProvider: () => MongoDBProvider
 });
-function stripInternalFilterKeys2(filter) {
-  if (!filter) return {};
-  return Object.fromEntries(
-    Object.entries(filter).filter(([key]) => !key.startsWith("__"))
-  );
-}
 var import_mongodb, MongoDBProvider;
 var init_MongoDBProvider = __esm({
   "src/providers/vectordb/MongoDBProvider.ts"() {
@@ -412,7 +413,7 @@ var init_MongoDBProvider = __esm({
         await this.collection.bulkWrite(operations);
       }
       async query(vector, topK, namespace, filter) {
-        const publicFilter = stripInternalFilterKeys2(filter);
+        const publicFilter = this.sanitizeFilter(filter);
         const pipeline = [
           {
             $vectorSearch: __spreadValues({
@@ -1087,11 +1088,12 @@ var init_UniversalVectorProvider = __esm({
           namespace: "{{namespace}}",
           filter: "{{filter}}"
         });
+        const sanitizedFilter = this.sanitizeFilter(filter);
         const payload = buildPayload(template, {
           vector,
           topK,
           namespace: namespace != null ? namespace : this.indexName,
-          filter: filter != null ? filter : {}
+          filter: sanitizedFilter
         });
         try {
           const response = await this.http.post(endpoint, payload);
@@ -2177,8 +2179,18 @@ 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;
+    this.llmConfig = __spreadProps(__spreadValues({}, llmConfig), {
+      model: this.sanitizeModel(llmConfig.model)
+    });
+    if (embeddingConfig) {
+      this.embeddingConfig = __spreadProps(__spreadValues({}, embeddingConfig), {
+        model: this.sanitizeModel(embeddingConfig.model)
+      });
+    }
+  }
+  sanitizeModel(model) {
+    if (!model) return model;
+    return model.split(":")[0];
   }
   async chat(messages, context, options) {
     var _a, _b, _c, _d, _e, _f;
@@ -2207,8 +2219,10 @@ ${context}`;
     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";
+    var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k;
+    const model = this.sanitizeModel(
+      (_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;
@@ -2225,14 +2239,20 @@ ${context}`;
     }
     const response = await client.models.embedContent({
       model,
-      contents: content
+      contents: content,
+      config: ((_h = this.embeddingConfig) == null ? void 0 : _h.dimensions) ? { outputDimensionality: this.embeddingConfig.dimensions } : void 0
+    }).catch((err) => {
+      console.error(`[GeminiProvider] Embedding failed for model "${model}":`, err.message);
+      throw err;
     });
-    return (_j = (_i = (_h = response.embeddings) == null ? void 0 : _h[0]) == null ? void 0 : _i.values) != null ? _j : [];
+    return (_k = (_j = (_i = response.embeddings) == null ? void 0 : _i[0]) == null ? void 0 : _j.values) != null ? _k : [];
   }
   async batchEmbed(texts, options) {
-    var _a, _b, _c, _d, _e, _f, _g, _h, _i;
+    var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j;
     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 model = this.sanitizeModel(
+      (_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;
@@ -2245,12 +2265,16 @@ ${context}`;
     }
     const response = await client.models.embedContent({
       model,
-      contents
+      contents,
+      config: ((_h = this.embeddingConfig) == null ? void 0 : _h.dimensions) ? { outputDimensionality: this.embeddingConfig.dimensions } : void 0
+    }).catch((err) => {
+      console.error(`[GeminiProvider] Batch embedding failed for model "${model}":`, err.message);
+      throw err;
     });
-    return (_i = (_h = response.embeddings) == null ? void 0 : _h.map((e) => {
+    return (_j = (_i = response.embeddings) == null ? void 0 : _i.map((e) => {
       var _a2;
       return (_a2 = e.values) != null ? _a2 : [];
-    })) != null ? _i : [];
+    })) != null ? _j : [];
   }
   async ping() {
     try {

package/dist/handlers/index.mjs

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

package/dist/server.d.mts

@@ -307,6 +307,11 @@ declare abstract class BaseVectorProvider {
      * Gracefully close the connection.
      */
     abstract disconnect(): Promise<void>;
+    /**
+     * Remove internal keys (starting with __) and queryText from a filter object
+     * before passing it to an external provider that might not support them.
+     */
+    protected sanitizeFilter(filter?: Record<string, unknown>): Record<string, unknown>;
 }
 
 /**

package/dist/server.d.ts

@@ -307,6 +307,11 @@ declare abstract class BaseVectorProvider {
      * Gracefully close the connection.
      */
     abstract disconnect(): Promise<void>;
+    /**
+     * Remove internal keys (starting with __) and queryText from a filter object
+     * before passing it to an external provider that might not support them.
+     */
+    protected sanitizeFilter(filter?: Record<string, unknown>): Record<string, unknown>;
 }
 
 /**

package/dist/server.js

@@ -114,6 +114,18 @@ var init_BaseVectorProvider = __esm({
         this.config = config;
         this.indexName = config.indexName || "default";
       }
+      /**
+       * Remove internal keys (starting with __) and queryText from a filter object
+       * before passing it to an external provider that might not support them.
+       */
+      sanitizeFilter(filter) {
+        if (!filter) return {};
+        return Object.fromEntries(
+          Object.entries(filter).filter(
+            ([key]) => !key.startsWith("__") && key !== "queryText"
+          )
+        );
+      }
     };
   }
 });
@@ -177,11 +189,12 @@ var init_PineconeProvider = __esm({
       }
       async query(vector, topK, namespace, filter) {
         var _a;
+        const pineconeFilter = this.sanitizeFilter(filter);
         const result = await this.index(namespace).query(__spreadValues({
           vector,
           topK,
           includeMetadata: true
-        }, filter ? { filter } : {}));
+        }, Object.keys(pineconeFilter).length > 0 ? { filter: pineconeFilter } : {}));
         return ((_a = result.matches) != null ? _a : []).map((m) => {
           var _a2, _b;
           return {
@@ -217,12 +230,6 @@ var PostgreSQLProvider_exports = {};
 __export(PostgreSQLProvider_exports, {
   PostgreSQLProvider: () => PostgreSQLProvider
 });
-function stripInternalFilterKeys(filter) {
-  if (!filter) return {};
-  return Object.fromEntries(
-    Object.entries(filter).filter(([key]) => !key.startsWith("__"))
-  );
-}
 var import_pg, PostgreSQLProvider;
 var init_PostgreSQLProvider = __esm({
   "src/providers/vectordb/PostgreSQLProvider.ts"() {
@@ -315,7 +322,7 @@ var init_PostgreSQLProvider = __esm({
         let whereClause = namespace ? `WHERE namespace = $3` : "";
         const params = [vectorLiteral, topK];
         if (namespace) params.push(namespace);
-        const publicFilter = stripInternalFilterKeys(filter);
+        const publicFilter = this.sanitizeFilter(filter);
         if (Object.keys(publicFilter).length > 0) {
           const filterConditions = Object.entries(publicFilter).map(([key, val]) => {
             const paramIdx = params.length + 1;
@@ -367,12 +374,6 @@ var MongoDBProvider_exports = {};
 __export(MongoDBProvider_exports, {
   MongoDBProvider: () => MongoDBProvider
 });
-function stripInternalFilterKeys2(filter) {
-  if (!filter) return {};
-  return Object.fromEntries(
-    Object.entries(filter).filter(([key]) => !key.startsWith("__"))
-  );
-}
 var import_mongodb, MongoDBProvider;
 var init_MongoDBProvider = __esm({
   "src/providers/vectordb/MongoDBProvider.ts"() {
@@ -424,7 +425,7 @@ var init_MongoDBProvider = __esm({
         await this.collection.bulkWrite(operations);
       }
       async query(vector, topK, namespace, filter) {
-        const publicFilter = stripInternalFilterKeys2(filter);
+        const publicFilter = this.sanitizeFilter(filter);
         const pipeline = [
           {
             $vectorSearch: __spreadValues({
@@ -1099,11 +1100,12 @@ var init_UniversalVectorProvider = __esm({
           namespace: "{{namespace}}",
           filter: "{{filter}}"
         });
+        const sanitizedFilter = this.sanitizeFilter(filter);
         const payload = buildPayload(template, {
           vector,
           topK,
           namespace: namespace != null ? namespace : this.indexName,
-          filter: filter != null ? filter : {}
+          filter: sanitizedFilter
         });
         try {
           const response = await this.http.post(endpoint, payload);
@@ -2221,8 +2223,18 @@ 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;
+    this.llmConfig = __spreadProps(__spreadValues({}, llmConfig), {
+      model: this.sanitizeModel(llmConfig.model)
+    });
+    if (embeddingConfig) {
+      this.embeddingConfig = __spreadProps(__spreadValues({}, embeddingConfig), {
+        model: this.sanitizeModel(embeddingConfig.model)
+      });
+    }
+  }
+  sanitizeModel(model) {
+    if (!model) return model;
+    return model.split(":")[0];
   }
   async chat(messages, context, options) {
     var _a, _b, _c, _d, _e, _f;
@@ -2251,8 +2263,10 @@ ${context}`;
     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";
+    var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k;
+    const model = this.sanitizeModel(
+      (_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;
@@ -2269,14 +2283,20 @@ ${context}`;
     }
     const response = await client.models.embedContent({
       model,
-      contents: content
+      contents: content,
+      config: ((_h = this.embeddingConfig) == null ? void 0 : _h.dimensions) ? { outputDimensionality: this.embeddingConfig.dimensions } : void 0
+    }).catch((err) => {
+      console.error(`[GeminiProvider] Embedding failed for model "${model}":`, err.message);
+      throw err;
     });
-    return (_j = (_i = (_h = response.embeddings) == null ? void 0 : _h[0]) == null ? void 0 : _i.values) != null ? _j : [];
+    return (_k = (_j = (_i = response.embeddings) == null ? void 0 : _i[0]) == null ? void 0 : _j.values) != null ? _k : [];
   }
   async batchEmbed(texts, options) {
-    var _a, _b, _c, _d, _e, _f, _g, _h, _i;
+    var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j;
     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 model = this.sanitizeModel(
+      (_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;
@@ -2289,12 +2309,16 @@ ${context}`;
     }
     const response = await client.models.embedContent({
       model,
-      contents
+      contents,
+      config: ((_h = this.embeddingConfig) == null ? void 0 : _h.dimensions) ? { outputDimensionality: this.embeddingConfig.dimensions } : void 0
+    }).catch((err) => {
+      console.error(`[GeminiProvider] Batch embedding failed for model "${model}":`, err.message);
+      throw err;
     });
-    return (_i = (_h = response.embeddings) == null ? void 0 : _h.map((e) => {
+    return (_j = (_i = response.embeddings) == null ? void 0 : _i.map((e) => {
       var _a2;
       return (_a2 = e.values) != null ? _a2 : [];
-    })) != null ? _i : [];
+    })) != null ? _j : [];
   }
   async ping() {
     try {

package/dist/server.mjs

@@ -1,15 +1,15 @@
 import {
   ChromaDBProvider
-} from "./chunk-VKE5ZW7Y.mjs";
+} from "./chunk-Q2QDQ5SQ.mjs";
 import {
   RedisProvider
-} from "./chunk-HOMXEE3M.mjs";
+} from "./chunk-N2L5CFGM.mjs";
 import {
   WeaviateProvider
-} from "./chunk-RUKZC3ON.mjs";
+} from "./chunk-7XL3QSQ7.mjs";
 import {
   UniversalVectorProvider
-} from "./chunk-VV2ML6TM.mjs";
+} from "./chunk-R3GPRETL.mjs";
 import {
   AnthropicProvider,
   BatchProcessor,
@@ -34,26 +34,26 @@ import {
   createIngestHandler,
   createUploadHandler,
   getRagConfig
-} from "./chunk-SNPFZBDK.mjs";
+} from "./chunk-TOKEXP3M.mjs";
 import "./chunk-EDLTMSNY.mjs";
 import {
   PineconeProvider
-} from "./chunk-BMHJTWSU.mjs";
+} from "./chunk-M3AZTD7F.mjs";
 import {
   PostgreSQLProvider
-} from "./chunk-6GSARSCP.mjs";
+} from "./chunk-L54WODRW.mjs";
 import {
   MongoDBProvider
-} from "./chunk-IFPISZ2S.mjs";
+} from "./chunk-KGW2YHV2.mjs";
 import {
   MilvusProvider
-} from "./chunk-VEJNRS4B.mjs";
+} from "./chunk-ENEUABSO.mjs";
 import {
   QdrantProvider
-} from "./chunk-W2PQR3UK.mjs";
+} from "./chunk-ULX4EF74.mjs";
 import {
   BaseVectorProvider
-} from "./chunk-VOIWNO5O.mjs";
+} from "./chunk-YVTGOFNV.mjs";
 import {
   __objRest,
   __spreadProps,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@retrivora-ai/rag-engine",
-  "version": "0.3.3",
+  "version": "0.3.5",
   "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",

package/src/llm/providers/GeminiProvider.ts

@@ -26,10 +26,28 @@ export class GeminiProvider implements ILLMProvider {
   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;
+    
+    // Sanitize model names (e.g. remove ':latest' suffix used by Ollama)
+    this.llmConfig = {
+      ...llmConfig,
+      model: this.sanitizeModel(llmConfig.model)
+    };
+    
+    if (embeddingConfig) {
+      this.embeddingConfig = {
+        ...embeddingConfig,
+        model: this.sanitizeModel(embeddingConfig.model)
+      };
+    }
   }
 
+  private sanitizeModel(model: string): string {
+    if (!model) return model;
+    // Strip :latest suffix common in Ollama but invalid in Google SDK
+    return model.split(':')[0];
+  }
+
+
   async chat(messages: ChatMessage[], context: string, options?: ChatOptions): Promise<string> {
     const systemPrompt =
       this.llmConfig.systemPrompt ??
@@ -62,10 +80,11 @@ export class GeminiProvider implements ILLMProvider {
   }
 
   async embed(text: string, options?: EmbedOptions): Promise<number[]> {
-    const model =
+    const model = this.sanitizeModel(
       options?.model ??
       this.embeddingConfig?.model ??
-      'text-embedding-004';
+      'text-embedding-004'
+    );
 
     const apiKey = this.embeddingConfig?.apiKey ?? this.llmConfig.apiKey;
     const client = apiKey !== this.llmConfig.apiKey
@@ -91,6 +110,10 @@ export class GeminiProvider implements ILLMProvider {
     const response = await client.models.embedContent({
       model,
       contents: content,
+      config: this.embeddingConfig?.dimensions ? { outputDimensionality: this.embeddingConfig.dimensions } : undefined
+    }).catch(err => {
+      console.error(`[GeminiProvider] Embedding failed for model "${model}":`, err.message);
+      throw err;
     });
 
     return response.embeddings?.[0]?.values ?? [];
@@ -101,10 +124,11 @@ export class GeminiProvider implements ILLMProvider {
     // 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 =
+    const model = this.sanitizeModel(
       options?.model ??
       this.embeddingConfig?.model ??
-      'text-embedding-004';
+      'text-embedding-004'
+    );
 
     const apiKey = this.embeddingConfig?.apiKey ?? this.llmConfig.apiKey;
     const client = apiKey !== this.llmConfig.apiKey
@@ -126,6 +150,10 @@ export class GeminiProvider implements ILLMProvider {
     const response = await client.models.embedContent({
       model,
       contents,
+      config: this.embeddingConfig?.dimensions ? { outputDimensionality: this.embeddingConfig.dimensions } : undefined
+    }).catch(err => {
+      console.error(`[GeminiProvider] Batch embedding failed for model "${model}":`, err.message);
+      throw err;
     });
 
     return response.embeddings?.map(e => e.values ?? []) ?? [];

package/src/providers/vectordb/BaseVectorProvider.ts

@@ -58,4 +58,17 @@ export abstract class BaseVectorProvider {
    * Gracefully close the connection.
    */
   abstract disconnect(): Promise<void>;
+
+  /**
+   * Remove internal keys (starting with __) and queryText from a filter object
+   * before passing it to an external provider that might not support them.
+   */
+  protected sanitizeFilter(filter?: Record<string, unknown>): Record<string, unknown> {
+    if (!filter) return {};
+    return Object.fromEntries(
+      Object.entries(filter).filter(
+        ([key]) => !key.startsWith('__') && key !== 'queryText'
+      )
+    );
+  }
 }

package/src/providers/vectordb/MongoDBProvider.ts

@@ -3,14 +3,6 @@ import { VectorDBConfig } from '../../config/RagConfig';
 import { BaseVectorProvider } from './BaseVectorProvider';
 import { VectorMatch, UpsertDocument } from '../../types';
 
-function stripInternalFilterKeys(filter?: Record<string, unknown>): Record<string, unknown> {
-  if (!filter) return {};
-
-  return Object.fromEntries(
-    Object.entries(filter).filter(([key]) => !key.startsWith('__'))
-  );
-}
-
 /**
  * MongoDBProvider — MongoDB Atlas Vector Search implementation.
  */
@@ -75,7 +67,7 @@ export class MongoDBProvider extends BaseVectorProvider {
   }
 
   async query(vector: number[], topK: number, namespace?: string, filter?: Record<string, unknown>): Promise<VectorMatch[]> {
-    const publicFilter = stripInternalFilterKeys(filter);
+    const publicFilter = this.sanitizeFilter(filter);
     const pipeline: Record<string, unknown>[] = [
       {
         $vectorSearch: {

package/src/providers/vectordb/PineconeProvider.ts

@@ -66,11 +66,12 @@ export class PineconeProvider extends BaseVectorProvider {
     namespace?: string,
     filter?: Record<string, unknown>
   ): Promise<VectorMatch[]> {
+    const pineconeFilter = this.sanitizeFilter(filter);
     const result = await this.index(namespace).query({
       vector,
       topK,
       includeMetadata: true,
-      ...(filter ? { filter } : {}),
+      ...(Object.keys(pineconeFilter).length > 0 ? { filter: pineconeFilter as any } : {}),
     });
     return (result.matches ?? []).map((m) => ({
       id: m.id,

package/src/providers/vectordb/PostgreSQLProvider.ts

@@ -3,14 +3,6 @@ import { VectorDBConfig } from '../../config/RagConfig';
 import { BaseVectorProvider } from './BaseVectorProvider';
 import { VectorMatch, UpsertDocument } from '../../types';
 
-function stripInternalFilterKeys(filter?: Record<string, unknown>): Record<string, unknown> {
-  if (!filter) return {};
-
-  return Object.fromEntries(
-    Object.entries(filter).filter(([key]) => !key.startsWith('__'))
-  );
-}
-
 /**
  * PostgreSQLProvider — PostgreSQL implementation using the pgvector extension.
  */
@@ -111,7 +103,7 @@ export class PostgreSQLProvider extends BaseVectorProvider {
     const params: unknown[] = [vectorLiteral, topK];
     if (namespace) params.push(namespace);
 
-    const publicFilter = stripInternalFilterKeys(filter);
+    const publicFilter = this.sanitizeFilter(filter);
     if (Object.keys(publicFilter).length > 0) {
       const filterConditions = Object.entries(publicFilter)
         .map(([key, val]) => {

package/src/providers/vectordb/UniversalVectorProvider.ts

@@ -132,11 +132,12 @@ export class UniversalVectorProvider extends BaseVectorProvider {
       filter: '{{filter}}',
     });
 
+    const sanitizedFilter = this.sanitizeFilter(filter);
     const payload = buildPayload(template, {
       vector,
       topK,
       namespace: namespace ?? this.indexName,
-      filter: filter ?? {},
+      filter: sanitizedFilter,
     });
 
     try {

package/dist/chunk-VOIWNO5O.mjs

@@ -1,11 +0,0 @@
-// src/providers/vectordb/BaseVectorProvider.ts
-var BaseVectorProvider = class {
-  constructor(config) {
-    this.config = config;
-    this.indexName = config.indexName || "default";
-  }
-};
-
-export {
-  BaseVectorProvider
-};