@mastra/rag 0.10.2-alpha.0 → 0.10.2

package/.turbo/turbo-build.log

@@ -1,23 +1,23 @@
 
-> @mastra/rag@0.10.2-alpha.0 build /home/runner/work/mastra/mastra/packages/rag
+> @mastra/rag@0.10.2-alpha.1 build /home/runner/work/mastra/mastra/packages/rag
 > tsup src/index.ts --format esm,cjs --experimental-dts --clean --treeshake=smallest --splitting
 
 CLI Building entry: src/index.ts
 CLI Using tsconfig: tsconfig.json
 CLI tsup v8.4.0
 TSC Build start
-TSC ⚡️ Build success in 13387ms
+TSC ⚡️ Build success in 12829ms
 DTS Build start
 CLI Target: es2022
 Analysis will use the bundled TypeScript version 5.8.3
 Writing package typings: /home/runner/work/mastra/mastra/packages/rag/dist/_tsup-dts-rollup.d.ts
 Analysis will use the bundled TypeScript version 5.8.3
 Writing package typings: /home/runner/work/mastra/mastra/packages/rag/dist/_tsup-dts-rollup.d.cts
-DTS ⚡️ Build success in 10836ms
+DTS ⚡️ Build success in 13312ms
 CLI Cleaning output folder
 ESM Build start
 CJS Build start
-ESM dist/index.js 239.65 KB
-ESM ⚡️ Build success in 3608ms
-CJS dist/index.cjs 241.36 KB
-CJS ⚡️ Build success in 3609ms
+ESM dist/index.js 241.64 KB
+ESM ⚡️ Build success in 4913ms
+CJS dist/index.cjs 243.35 KB
+CJS ⚡️ Build success in 4909ms

package/CHANGELOG.md

@@ -1,5 +1,40 @@
 # @mastra/rag
 
+## 0.10.2
+
+### Patch Changes
+
+- a60d630: Adds runtime and initialization DB specific configuration to createVectorQueryTool to account for different configuration between db providers.
+- f0d559f: Fix peerdeps for alpha channel
+- Updated dependencies [ee77e78]
+- Updated dependencies [592a2db]
+- Updated dependencies [e5dc18d]
+- Updated dependencies [ab5adbe]
+- Updated dependencies [1e8bb40]
+- Updated dependencies [1b5fc55]
+- Updated dependencies [195c428]
+- Updated dependencies [f73e11b]
+- Updated dependencies [37643b8]
+- Updated dependencies [99fd6cf]
+- Updated dependencies [c5bf1ce]
+- Updated dependencies [add596e]
+- Updated dependencies [8dc94d8]
+- Updated dependencies [ecebbeb]
+- Updated dependencies [79d5145]
+- Updated dependencies [12b7002]
+- Updated dependencies [2901125]
+  - @mastra/core@0.10.2
+
+## 0.10.2-alpha.1
+
+### Patch Changes
+
+- a60d630: Adds runtime and initialization DB specific configuration to createVectorQueryTool to account for different configuration between db providers.
+- Updated dependencies [1b5fc55]
+- Updated dependencies [add596e]
+- Updated dependencies [ecebbeb]
+  - @mastra/core@0.10.2-alpha.5
+
 ## 0.10.2-alpha.0
 
 ### Patch Changes

package/dist/_tsup-dts-rollup.d.cts

@@ -83,6 +83,10 @@ declare const baseSchema: {
 export { baseSchema }
 export { baseSchema as baseSchema_alias_1 }
 
+declare type BaseWhere = {
+    [key: string]: LiteralValue | OperatorExpression;
+};
+
 export declare class CharacterTransformer extends TextTransformer {
     protected separator: string;
     protected isSeparatorRegex: boolean;
@@ -115,6 +119,11 @@ declare const CHROMA_PROMPT = "When querying Chroma, you can ONLY use the operat
 export { CHROMA_PROMPT }
 export { CHROMA_PROMPT as CHROMA_PROMPT_alias_1 }
 
+export declare interface ChromaConfig {
+    where?: Where;
+    whereDocument?: WhereDocument;
+}
+
 declare type ChunkOptions = {
     headers?: [string, string][];
     returnEachLine?: boolean;
@@ -226,6 +235,13 @@ export { createVectorQueryTool }
 export { createVectorQueryTool as createVectorQueryTool_alias_1 }
 export { createVectorQueryTool as createVectorQueryTool_alias_2 }
 
+export declare type DatabaseConfig = {
+    pinecone?: PineconeConfig;
+    pgvector?: PgVectorConfig;
+    chroma?: ChromaConfig;
+    [key: string]: any;
+};
+
 /**
  * Default options for GraphRAG
  * @default
@@ -440,6 +456,8 @@ export declare class HTMLSectionTransformer {
     transformDocuments(documents: Document[]): Document[];
 }
 
+declare type InclusionOperator = '$in' | '$nin';
+
 declare type KeywordExtractArgs = {
     llm?: MastraLanguageModel;
     keywords?: number;
@@ -539,6 +557,18 @@ declare const LIBSQL_PROMPT = "When querying LibSQL Vector, you can ONLY use the
 export { LIBSQL_PROMPT }
 export { LIBSQL_PROMPT as LIBSQL_PROMPT_alias_1 }
 
+declare type ListLiteralValue = LiteralValue[];
+
+declare type LiteralNumber = number;
+
+declare type LiteralValue = string | number | boolean;
+
+declare type LogicalOperator = '$and' | '$or';
+
+declare type LogicalWhere = {
+    [key in LogicalOperator]?: Where[];
+};
+
 export declare class MarkdownHeaderTransformer {
     private headersToSplitOn;
     private returnEachLine;
@@ -637,6 +667,10 @@ declare enum ObjectType {
 export { ObjectType }
 export { ObjectType as ObjectType_alias_1 }
 
+declare type OperatorExpression = {
+    [key in WhereOperator | InclusionOperator | LogicalOperator]?: LiteralValue | ListLiteralValue;
+};
+
 declare const outputSchema: z.ZodObject<{
     relevantContext: z.ZodAny;
     sources: z.ZodArray<z.ZodObject<{
@@ -685,11 +719,25 @@ declare const PGVECTOR_PROMPT = "When querying PG Vector, you can ONLY use the o
 export { PGVECTOR_PROMPT }
 export { PGVECTOR_PROMPT as PGVECTOR_PROMPT_alias_1 }
 
+export declare interface PgVectorConfig {
+    minScore?: number;
+    ef?: number;
+    probes?: number;
+}
+
 /** @deprecated Import from @mastra/pinecone instead */
 declare const PINECONE_PROMPT = "When querying Pinecone, you can ONLY use the operators listed below. Any other operators will be rejected.\nImportant: Don't explain how to construct the filter - use the specified operators and fields to search the content and return relevant results.\nIf a user tries to give an explicit operator that is not supported, reject the filter entirely and let them know that the operator is not supported.\n\nBasic Comparison Operators:\n- $eq: Exact match (default when using field: value)\n  Example: { \"category\": \"electronics\" }\n- $ne: Not equal\n  Example: { \"category\": { \"$ne\": \"electronics\" } }\n- $gt: Greater than\n  Example: { \"price\": { \"$gt\": 100 } }\n- $gte: Greater than or equal\n  Example: { \"price\": { \"$gte\": 100 } }\n- $lt: Less than\n  Example: { \"price\": { \"$lt\": 100 } }\n- $lte: Less than or equal\n  Example: { \"price\": { \"$lte\": 100 } }\n\nArray Operators:\n- $in: Match any value in array\n  Example: { \"category\": { \"$in\": [\"electronics\", \"books\"] } }\n- $nin: Does not match any value in array\n  Example: { \"category\": { \"$nin\": [\"electronics\", \"books\"] } }\n- $all: Match all values in array\n  Example: { \"tags\": { \"$all\": [\"premium\", \"sale\"] } }\n\nLogical Operators:\n- $and: Logical AND (can be implicit or explicit)\n  Implicit Example: { \"price\": { \"$gt\": 100 }, \"category\": \"electronics\" }\n  Explicit Example: { \"$and\": [{ \"price\": { \"$gt\": 100 } }, { \"category\": \"electronics\" }] }\n- $or: Logical OR\n  Example: { \"$or\": [{ \"price\": { \"$lt\": 50 } }, { \"category\": \"books\" }] }\n\nElement Operators:\n- $exists: Check if field exists\n  Example: { \"rating\": { \"$exists\": true } }\n\nRestrictions:\n- Regex patterns are not supported\n- Only $and and $or logical operators are supported at the top level\n- Empty arrays in $in/$nin will return no results\n- A non-empty array is required for $all operator\n- Nested fields are supported using dot notation\n- Multiple conditions on the same field are supported with both implicit and explicit $and\n- At least one key-value pair is required in filter object\n- Empty objects and undefined values are treated as no filter\n- Invalid types in comparison operators will throw errors\n- All non-logical operators must be used within a field condition\n  Valid: { \"field\": { \"$gt\": 100 } }\n  Valid: { \"$and\": [...] }\n  Invalid: { \"$gt\": 100 }\n- Logical operators must contain field conditions, not direct operators\n  Valid: { \"$and\": [{ \"field\": { \"$gt\": 100 } }] }\n  Invalid: { \"$and\": [{ \"$gt\": 100 }] }\n- Logical operators ($and, $or):\n  - Can only be used at top level or nested within other logical operators\n  - Can not be used on a field level, or be nested inside a field\n  - Can not be used inside an operator\n  - Valid: { \"$and\": [{ \"field\": { \"$gt\": 100 } }] }\n  - Valid: { \"$or\": [{ \"$and\": [{ \"field\": { \"$gt\": 100 } }] }] }\n  - Invalid: { \"field\": { \"$and\": [{ \"$gt\": 100 }] } }\n  - Invalid: { \"field\": { \"$or\": [{ \"$gt\": 100 }] } }\n  - Invalid: { \"field\": { \"$gt\": { \"$and\": [{...}] } } }\nExample Complex Query:\n{\n  \"$and\": [\n    { \"category\": { \"$in\": [\"electronics\", \"computers\"] } },\n    { \"price\": { \"$gte\": 100, \"$lte\": 1000 } },\n    { \"tags\": { \"$all\": [\"premium\", \"sale\"] } },\n    { \"rating\": { \"$exists\": true, \"$gt\": 4 } },\n    { \"$or\": [\n      { \"stock\": { \"$gt\": 0 } },\n      { \"preorder\": true }\n    ]}\n  ]\n}";
 export { PINECONE_PROMPT }
 export { PINECONE_PROMPT as PINECONE_PROMPT_alias_1 }
 
+export declare interface PineconeConfig {
+    namespace?: string;
+    sparseVector?: {
+        indices: number[];
+        values: number[];
+    };
+}
+
 declare class PromptTemplate<const TemplatesVar extends readonly string[] = string[]> extends BasePromptTemplate<TemplatesVar> {
     #private;
     constructor(options: PromptTemplateOptions<TemplatesVar>);
@@ -1156,7 +1204,7 @@ declare const VECTORIZE_PROMPT = "When querying Vectorize, you can ONLY use the
 export { VECTORIZE_PROMPT }
 export { VECTORIZE_PROMPT as VECTORIZE_PROMPT_alias_1 }
 
-declare const vectorQuerySearch: ({ indexName, vectorStore, queryText, model, queryFilter, topK, includeVectors, maxRetries, }: VectorQuerySearchParams) => Promise<VectorQuerySearchResult>;
+declare const vectorQuerySearch: ({ indexName, vectorStore, queryText, model, queryFilter, topK, includeVectors, maxRetries, databaseConfig, }: VectorQuerySearchParams) => Promise<VectorQuerySearchResult>;
 export { vectorQuerySearch }
 export { vectorQuerySearch as vectorQuerySearch_alias_1 }
 
@@ -1169,6 +1217,8 @@ declare interface VectorQuerySearchParams {
     topK: number;
     includeVectors?: boolean;
     maxRetries?: number;
+    /** Database-specific configuration options */
+    databaseConfig?: DatabaseConfig;
 }
 
 declare interface VectorQuerySearchResult {
@@ -1186,6 +1236,8 @@ export declare type VectorQueryToolOptions = {
     includeVectors?: boolean;
     includeSources?: boolean;
     reranker?: RerankConfig;
+    /** Database-specific configuration options */
+    databaseConfig?: DatabaseConfig;
 };
 
 declare type WeightConfig = {
@@ -1194,4 +1246,14 @@ declare type WeightConfig = {
     position?: number;
 };
 
+declare type Where = BaseWhere | LogicalWhere;
+
+declare type WhereDocument = {
+    [key in WhereDocumentOperator]?: LiteralValue | LiteralNumber | WhereDocument[];
+};
+
+declare type WhereDocumentOperator = '$contains' | '$not_contains' | LogicalOperator;
+
+declare type WhereOperator = '$gt' | '$gte' | '$lt' | '$lte' | '$ne' | '$eq';
+
 export { }

package/dist/_tsup-dts-rollup.d.ts

@@ -83,6 +83,10 @@ declare const baseSchema: {
 export { baseSchema }
 export { baseSchema as baseSchema_alias_1 }
 
+declare type BaseWhere = {
+    [key: string]: LiteralValue | OperatorExpression;
+};
+
 export declare class CharacterTransformer extends TextTransformer {
     protected separator: string;
     protected isSeparatorRegex: boolean;
@@ -115,6 +119,11 @@ declare const CHROMA_PROMPT = "When querying Chroma, you can ONLY use the operat
 export { CHROMA_PROMPT }
 export { CHROMA_PROMPT as CHROMA_PROMPT_alias_1 }
 
+export declare interface ChromaConfig {
+    where?: Where;
+    whereDocument?: WhereDocument;
+}
+
 declare type ChunkOptions = {
     headers?: [string, string][];
     returnEachLine?: boolean;
@@ -226,6 +235,13 @@ export { createVectorQueryTool }
 export { createVectorQueryTool as createVectorQueryTool_alias_1 }
 export { createVectorQueryTool as createVectorQueryTool_alias_2 }
 
+export declare type DatabaseConfig = {
+    pinecone?: PineconeConfig;
+    pgvector?: PgVectorConfig;
+    chroma?: ChromaConfig;
+    [key: string]: any;
+};
+
 /**
  * Default options for GraphRAG
  * @default
@@ -440,6 +456,8 @@ export declare class HTMLSectionTransformer {
     transformDocuments(documents: Document[]): Document[];
 }
 
+declare type InclusionOperator = '$in' | '$nin';
+
 declare type KeywordExtractArgs = {
     llm?: MastraLanguageModel;
     keywords?: number;
@@ -539,6 +557,18 @@ declare const LIBSQL_PROMPT = "When querying LibSQL Vector, you can ONLY use the
 export { LIBSQL_PROMPT }
 export { LIBSQL_PROMPT as LIBSQL_PROMPT_alias_1 }
 
+declare type ListLiteralValue = LiteralValue[];
+
+declare type LiteralNumber = number;
+
+declare type LiteralValue = string | number | boolean;
+
+declare type LogicalOperator = '$and' | '$or';
+
+declare type LogicalWhere = {
+    [key in LogicalOperator]?: Where[];
+};
+
 export declare class MarkdownHeaderTransformer {
     private headersToSplitOn;
     private returnEachLine;
@@ -637,6 +667,10 @@ declare enum ObjectType {
 export { ObjectType }
 export { ObjectType as ObjectType_alias_1 }
 
+declare type OperatorExpression = {
+    [key in WhereOperator | InclusionOperator | LogicalOperator]?: LiteralValue | ListLiteralValue;
+};
+
 declare const outputSchema: z.ZodObject<{
     relevantContext: z.ZodAny;
     sources: z.ZodArray<z.ZodObject<{
@@ -685,11 +719,25 @@ declare const PGVECTOR_PROMPT = "When querying PG Vector, you can ONLY use the o
 export { PGVECTOR_PROMPT }
 export { PGVECTOR_PROMPT as PGVECTOR_PROMPT_alias_1 }
 
+export declare interface PgVectorConfig {
+    minScore?: number;
+    ef?: number;
+    probes?: number;
+}
+
 /** @deprecated Import from @mastra/pinecone instead */
 declare const PINECONE_PROMPT = "When querying Pinecone, you can ONLY use the operators listed below. Any other operators will be rejected.\nImportant: Don't explain how to construct the filter - use the specified operators and fields to search the content and return relevant results.\nIf a user tries to give an explicit operator that is not supported, reject the filter entirely and let them know that the operator is not supported.\n\nBasic Comparison Operators:\n- $eq: Exact match (default when using field: value)\n  Example: { \"category\": \"electronics\" }\n- $ne: Not equal\n  Example: { \"category\": { \"$ne\": \"electronics\" } }\n- $gt: Greater than\n  Example: { \"price\": { \"$gt\": 100 } }\n- $gte: Greater than or equal\n  Example: { \"price\": { \"$gte\": 100 } }\n- $lt: Less than\n  Example: { \"price\": { \"$lt\": 100 } }\n- $lte: Less than or equal\n  Example: { \"price\": { \"$lte\": 100 } }\n\nArray Operators:\n- $in: Match any value in array\n  Example: { \"category\": { \"$in\": [\"electronics\", \"books\"] } }\n- $nin: Does not match any value in array\n  Example: { \"category\": { \"$nin\": [\"electronics\", \"books\"] } }\n- $all: Match all values in array\n  Example: { \"tags\": { \"$all\": [\"premium\", \"sale\"] } }\n\nLogical Operators:\n- $and: Logical AND (can be implicit or explicit)\n  Implicit Example: { \"price\": { \"$gt\": 100 }, \"category\": \"electronics\" }\n  Explicit Example: { \"$and\": [{ \"price\": { \"$gt\": 100 } }, { \"category\": \"electronics\" }] }\n- $or: Logical OR\n  Example: { \"$or\": [{ \"price\": { \"$lt\": 50 } }, { \"category\": \"books\" }] }\n\nElement Operators:\n- $exists: Check if field exists\n  Example: { \"rating\": { \"$exists\": true } }\n\nRestrictions:\n- Regex patterns are not supported\n- Only $and and $or logical operators are supported at the top level\n- Empty arrays in $in/$nin will return no results\n- A non-empty array is required for $all operator\n- Nested fields are supported using dot notation\n- Multiple conditions on the same field are supported with both implicit and explicit $and\n- At least one key-value pair is required in filter object\n- Empty objects and undefined values are treated as no filter\n- Invalid types in comparison operators will throw errors\n- All non-logical operators must be used within a field condition\n  Valid: { \"field\": { \"$gt\": 100 } }\n  Valid: { \"$and\": [...] }\n  Invalid: { \"$gt\": 100 }\n- Logical operators must contain field conditions, not direct operators\n  Valid: { \"$and\": [{ \"field\": { \"$gt\": 100 } }] }\n  Invalid: { \"$and\": [{ \"$gt\": 100 }] }\n- Logical operators ($and, $or):\n  - Can only be used at top level or nested within other logical operators\n  - Can not be used on a field level, or be nested inside a field\n  - Can not be used inside an operator\n  - Valid: { \"$and\": [{ \"field\": { \"$gt\": 100 } }] }\n  - Valid: { \"$or\": [{ \"$and\": [{ \"field\": { \"$gt\": 100 } }] }] }\n  - Invalid: { \"field\": { \"$and\": [{ \"$gt\": 100 }] } }\n  - Invalid: { \"field\": { \"$or\": [{ \"$gt\": 100 }] } }\n  - Invalid: { \"field\": { \"$gt\": { \"$and\": [{...}] } } }\nExample Complex Query:\n{\n  \"$and\": [\n    { \"category\": { \"$in\": [\"electronics\", \"computers\"] } },\n    { \"price\": { \"$gte\": 100, \"$lte\": 1000 } },\n    { \"tags\": { \"$all\": [\"premium\", \"sale\"] } },\n    { \"rating\": { \"$exists\": true, \"$gt\": 4 } },\n    { \"$or\": [\n      { \"stock\": { \"$gt\": 0 } },\n      { \"preorder\": true }\n    ]}\n  ]\n}";
 export { PINECONE_PROMPT }
 export { PINECONE_PROMPT as PINECONE_PROMPT_alias_1 }
 
+export declare interface PineconeConfig {
+    namespace?: string;
+    sparseVector?: {
+        indices: number[];
+        values: number[];
+    };
+}
+
 declare class PromptTemplate<const TemplatesVar extends readonly string[] = string[]> extends BasePromptTemplate<TemplatesVar> {
     #private;
     constructor(options: PromptTemplateOptions<TemplatesVar>);
@@ -1156,7 +1204,7 @@ declare const VECTORIZE_PROMPT = "When querying Vectorize, you can ONLY use the
 export { VECTORIZE_PROMPT }
 export { VECTORIZE_PROMPT as VECTORIZE_PROMPT_alias_1 }
 
-declare const vectorQuerySearch: ({ indexName, vectorStore, queryText, model, queryFilter, topK, includeVectors, maxRetries, }: VectorQuerySearchParams) => Promise<VectorQuerySearchResult>;
+declare const vectorQuerySearch: ({ indexName, vectorStore, queryText, model, queryFilter, topK, includeVectors, maxRetries, databaseConfig, }: VectorQuerySearchParams) => Promise<VectorQuerySearchResult>;
 export { vectorQuerySearch }
 export { vectorQuerySearch as vectorQuerySearch_alias_1 }
 
@@ -1169,6 +1217,8 @@ declare interface VectorQuerySearchParams {
     topK: number;
     includeVectors?: boolean;
     maxRetries?: number;
+    /** Database-specific configuration options */
+    databaseConfig?: DatabaseConfig;
 }
 
 declare interface VectorQuerySearchResult {
@@ -1186,6 +1236,8 @@ export declare type VectorQueryToolOptions = {
     includeVectors?: boolean;
     includeSources?: boolean;
     reranker?: RerankConfig;
+    /** Database-specific configuration options */
+    databaseConfig?: DatabaseConfig;
 };
 
 declare type WeightConfig = {
@@ -1194,4 +1246,14 @@ declare type WeightConfig = {
     position?: number;
 };
 
+declare type Where = BaseWhere | LogicalWhere;
+
+declare type WhereDocument = {
+    [key in WhereDocumentOperator]?: LiteralValue | LiteralNumber | WhereDocument[];
+};
+
+declare type WhereDocumentOperator = '$contains' | '$not_contains' | LogicalOperator;
+
+declare type WhereOperator = '$gt' | '$gte' | '$lt' | '$lte' | '$ne' | '$eq';
+
 export { }

package/dist/index.cjs

@@ -6217,6 +6217,13 @@ var createDocumentChunkerTool = ({
     }
   });
 };
+var DatabaseType = /* @__PURE__ */ ((DatabaseType2) => {
+  DatabaseType2["Pinecone"] = "pinecone";
+  DatabaseType2["PgVector"] = "pgvector";
+  DatabaseType2["Chroma"] = "chroma";
+  return DatabaseType2;
+})(DatabaseType || {});
+var DATABASE_TYPE_MAP = Object.keys(DatabaseType);
 var vectorQuerySearch = async ({
   indexName,
   vectorStore,
@@ -6225,22 +6232,65 @@ var vectorQuerySearch = async ({
   queryFilter,
   topK,
   includeVectors = false,
-  maxRetries = 2
+  maxRetries = 2,
+  databaseConfig = {}
 }) => {
   const { embedding } = await ai.embed({
     value: queryText,
     model,
     maxRetries
   });
-  const results = await vectorStore.query({
+  const queryParams = {
     indexName,
     queryVector: embedding,
     topK,
     filter: queryFilter,
     includeVector: includeVectors
-  });
+  };
+  const results = await vectorStore.query({ ...queryParams, ...databaseSpecificParams(databaseConfig) });
   return { results, queryEmbedding: embedding };
 };
+var databaseSpecificParams = (databaseConfig) => {
+  const databaseSpecificParams2 = {};
+  if (databaseConfig) {
+    if (databaseConfig.pinecone) {
+      if (databaseConfig.pinecone.namespace) {
+        databaseSpecificParams2.namespace = databaseConfig.pinecone.namespace;
+      }
+      if (databaseConfig.pinecone.sparseVector) {
+        databaseSpecificParams2.sparseVector = databaseConfig.pinecone.sparseVector;
+      }
+    }
+    if (databaseConfig.pgvector) {
+      if (databaseConfig.pgvector.minScore !== void 0) {
+        databaseSpecificParams2.minScore = databaseConfig.pgvector.minScore;
+      }
+      if (databaseConfig.pgvector.ef !== void 0) {
+        databaseSpecificParams2.ef = databaseConfig.pgvector.ef;
+      }
+      if (databaseConfig.pgvector.probes !== void 0) {
+        databaseSpecificParams2.probes = databaseConfig.pgvector.probes;
+      }
+    }
+    if (databaseConfig.chroma) {
+      if (databaseConfig.chroma.where) {
+        databaseSpecificParams2.where = databaseConfig.chroma.where;
+      }
+      if (databaseConfig.chroma.whereDocument) {
+        databaseSpecificParams2.whereDocument = databaseConfig.chroma.whereDocument;
+      }
+    }
+    Object.keys(databaseConfig).forEach((dbName) => {
+      if (!DATABASE_TYPE_MAP.includes(dbName)) {
+        const config = databaseConfig[dbName];
+        if (config && typeof config === "object") {
+          Object.assign(databaseSpecificParams2, config);
+        }
+      }
+    });
+  }
+  return databaseSpecificParams2;
+};
 
 // src/utils/default-settings.ts
 var defaultVectorQueryDescription = () => `Access the knowledge base to find information needed to answer user questions.`;
@@ -6481,6 +6531,7 @@ var createVectorQueryTool = (options) => {
       const includeVectors = runtimeContext.get("includeVectors") ?? options.includeVectors ?? false;
       const includeSources = runtimeContext.get("includeSources") ?? options.includeSources ?? true;
       const reranker = runtimeContext.get("reranker") ?? options.reranker;
+      const databaseConfig = runtimeContext.get("databaseConfig") ?? options.databaseConfig;
       if (!indexName) throw new Error(`indexName is required, got: ${indexName}`);
       if (!vectorStoreName) throw new Error(`vectorStoreName is required, got: ${vectorStoreName}`);
       const topK = runtimeContext.get("topK") ?? context.topK ?? 10;
@@ -6494,7 +6545,7 @@ var createVectorQueryTool = (options) => {
         );
       }
       if (logger) {
-        logger.debug("[VectorQueryTool] execute called with:", { queryText, topK, filter });
+        logger.debug("[VectorQueryTool] execute called with:", { queryText, topK, filter, databaseConfig });
       }
       try {
         const topKValue = typeof topK === "number" && !isNaN(topK) ? topK : typeof topK === "string" && !isNaN(Number(topK)) ? Number(topK) : 10;
@@ -6519,7 +6570,7 @@ var createVectorQueryTool = (options) => {
           })();
         }
         if (logger) {
-          logger.debug("Prepared vector query parameters", { queryText, topK: topKValue, queryFilter });
+          logger.debug("Prepared vector query parameters", { queryText, topK: topKValue, queryFilter, databaseConfig });
         }
         const { results } = await vectorQuerySearch({
           indexName,
@@ -6528,7 +6579,8 @@ var createVectorQueryTool = (options) => {
           model,
           queryFilter: Object.keys(queryFilter || {}).length > 0 ? queryFilter : void 0,
           topK: topKValue,
-          includeVectors
+          includeVectors,
+          databaseConfig
         });
         if (logger) {
           logger.debug("vectorQuerySearch returned results", { count: results.length });

package/dist/index.js

@@ -6215,6 +6215,13 @@ var createDocumentChunkerTool = ({
     }
   });
 };
+var DatabaseType = /* @__PURE__ */ ((DatabaseType2) => {
+  DatabaseType2["Pinecone"] = "pinecone";
+  DatabaseType2["PgVector"] = "pgvector";
+  DatabaseType2["Chroma"] = "chroma";
+  return DatabaseType2;
+})(DatabaseType || {});
+var DATABASE_TYPE_MAP = Object.keys(DatabaseType);
 var vectorQuerySearch = async ({
   indexName,
   vectorStore,
@@ -6223,22 +6230,65 @@ var vectorQuerySearch = async ({
   queryFilter,
   topK,
   includeVectors = false,
-  maxRetries = 2
+  maxRetries = 2,
+  databaseConfig = {}
 }) => {
   const { embedding } = await embed({
     value: queryText,
     model,
     maxRetries
   });
-  const results = await vectorStore.query({
+  const queryParams = {
     indexName,
     queryVector: embedding,
     topK,
     filter: queryFilter,
     includeVector: includeVectors
-  });
+  };
+  const results = await vectorStore.query({ ...queryParams, ...databaseSpecificParams(databaseConfig) });
   return { results, queryEmbedding: embedding };
 };
+var databaseSpecificParams = (databaseConfig) => {
+  const databaseSpecificParams2 = {};
+  if (databaseConfig) {
+    if (databaseConfig.pinecone) {
+      if (databaseConfig.pinecone.namespace) {
+        databaseSpecificParams2.namespace = databaseConfig.pinecone.namespace;
+      }
+      if (databaseConfig.pinecone.sparseVector) {
+        databaseSpecificParams2.sparseVector = databaseConfig.pinecone.sparseVector;
+      }
+    }
+    if (databaseConfig.pgvector) {
+      if (databaseConfig.pgvector.minScore !== void 0) {
+        databaseSpecificParams2.minScore = databaseConfig.pgvector.minScore;
+      }
+      if (databaseConfig.pgvector.ef !== void 0) {
+        databaseSpecificParams2.ef = databaseConfig.pgvector.ef;
+      }
+      if (databaseConfig.pgvector.probes !== void 0) {
+        databaseSpecificParams2.probes = databaseConfig.pgvector.probes;
+      }
+    }
+    if (databaseConfig.chroma) {
+      if (databaseConfig.chroma.where) {
+        databaseSpecificParams2.where = databaseConfig.chroma.where;
+      }
+      if (databaseConfig.chroma.whereDocument) {
+        databaseSpecificParams2.whereDocument = databaseConfig.chroma.whereDocument;
+      }
+    }
+    Object.keys(databaseConfig).forEach((dbName) => {
+      if (!DATABASE_TYPE_MAP.includes(dbName)) {
+        const config = databaseConfig[dbName];
+        if (config && typeof config === "object") {
+          Object.assign(databaseSpecificParams2, config);
+        }
+      }
+    });
+  }
+  return databaseSpecificParams2;
+};
 
 // src/utils/default-settings.ts
 var defaultVectorQueryDescription = () => `Access the knowledge base to find information needed to answer user questions.`;
@@ -6479,6 +6529,7 @@ var createVectorQueryTool = (options) => {
       const includeVectors = runtimeContext.get("includeVectors") ?? options.includeVectors ?? false;
       const includeSources = runtimeContext.get("includeSources") ?? options.includeSources ?? true;
       const reranker = runtimeContext.get("reranker") ?? options.reranker;
+      const databaseConfig = runtimeContext.get("databaseConfig") ?? options.databaseConfig;
       if (!indexName) throw new Error(`indexName is required, got: ${indexName}`);
       if (!vectorStoreName) throw new Error(`vectorStoreName is required, got: ${vectorStoreName}`);
       const topK = runtimeContext.get("topK") ?? context.topK ?? 10;
@@ -6492,7 +6543,7 @@ var createVectorQueryTool = (options) => {
         );
       }
       if (logger) {
-        logger.debug("[VectorQueryTool] execute called with:", { queryText, topK, filter });
+        logger.debug("[VectorQueryTool] execute called with:", { queryText, topK, filter, databaseConfig });
       }
       try {
         const topKValue = typeof topK === "number" && !isNaN(topK) ? topK : typeof topK === "string" && !isNaN(Number(topK)) ? Number(topK) : 10;
@@ -6517,7 +6568,7 @@ var createVectorQueryTool = (options) => {
           })();
         }
         if (logger) {
-          logger.debug("Prepared vector query parameters", { queryText, topK: topKValue, queryFilter });
+          logger.debug("Prepared vector query parameters", { queryText, topK: topKValue, queryFilter, databaseConfig });
         }
         const { results } = await vectorQuerySearch({
           indexName,
@@ -6526,7 +6577,8 @@ var createVectorQueryTool = (options) => {
           model,
           queryFilter: Object.keys(queryFilter || {}).length > 0 ? queryFilter : void 0,
           topK: topKValue,
-          includeVectors
+          includeVectors,
+          databaseConfig
         });
         if (logger) {
           logger.debug("vectorQuerySearch returned results", { count: results.length });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mastra/rag",
-  "version": "0.10.2-alpha.0",
+  "version": "0.10.2",
   "description": "",
   "type": "module",
   "main": "dist/index.js",
@@ -44,8 +44,8 @@
     "tsup": "^8.4.0",
     "typescript": "^5.8.2",
     "vitest": "^3.1.2",
-    "@internal/lint": "0.0.7",
-    "@mastra/core": "0.10.2-alpha.2"
+    "@mastra/core": "0.10.2",
+    "@internal/lint": "0.0.8"
   },
   "keywords": [
     "rag",

package/src/tools/README.md

@@ -0,0 +1,153 @@
+# Vector Query Tool with Database-Specific Configurations
+
+The `createVectorQueryTool` function now supports database-specific configurations to handle unique properties and optimizations for different vector databases.
+
+## Database Configuration Types
+
+### Pinecone Configuration
+
+```typescript
+import { createVectorQueryTool } from '@mastra/rag/tools';
+
+const pineconeVectorTool = createVectorQueryTool({
+  id: 'pinecone-search',
+  indexName: 'my-index',
+  vectorStoreName: 'pinecone',
+  model: embedModel,
+  databaseConfig: {
+    pinecone: {
+      namespace: 'my-namespace', // Pinecone namespace
+      sparseVector: {
+        // For hybrid search
+        indices: [0, 1, 2],
+        values: [0.1, 0.2, 0.3],
+      },
+    },
+  },
+});
+```
+
+### pgVector Configuration
+
+```typescript
+const pgVectorTool = createVectorQueryTool({
+  id: 'pgvector-search',
+  indexName: 'my-index',
+  vectorStoreName: 'postgres',
+  model: embedModel,
+  databaseConfig: {
+    pgvector: {
+      minScore: 0.7, // Minimum similarity score
+      ef: 200, // HNSW search parameter
+      probes: 10, // IVFFlat probe parameter
+    },
+  },
+});
+```
+
+### Chroma Configuration
+
+```typescript
+const chromaTool = createVectorQueryTool({
+  id: 'chroma-search',
+  indexName: 'my-index',
+  vectorStoreName: 'chroma',
+  model: embedModel,
+  databaseConfig: {
+    chroma: {
+      where: {
+        // Metadata filtering
+        category: 'documents',
+      },
+      whereDocument: {
+        // Document content filtering
+        $contains: 'important',
+      },
+    },
+  },
+});
+```
+
+## Runtime Configuration Override
+
+You can also override database configurations at runtime using the runtime context:
+
+```typescript
+import { RuntimeContext } from '@mastra/core/runtime-context';
+
+const runtimeContext = new RuntimeContext();
+
+// Override Pinecone namespace at runtime
+runtimeContext.set('databaseConfig', {
+  pinecone: {
+    namespace: 'runtime-namespace',
+  },
+});
+
+await vectorTool.execute({
+  context: { queryText: 'search query' },
+  mastra,
+  runtimeContext,
+});
+```
+
+## Extensibility for New Databases
+
+The system is designed to be extensible. For new vector databases, you can:
+
+1. Add configuration types:
+
+```typescript
+export interface NewDatabaseConfig {
+  customParam1?: string;
+  customParam2?: number;
+}
+
+export type DatabaseConfig = {
+  pinecone?: PineconeConfig;
+  pgvector?: PgVectorConfig;
+  chroma?: ChromaConfig;
+  newdatabase?: NewDatabaseConfig; // Add your config here
+  [key: string]: any;
+};
+```
+
+2. The configuration will be automatically passed through to the vector store's query method.
+
+## Type Safety
+
+All database configurations are fully typed, providing IntelliSense and compile-time checking:
+
+```typescript
+const config: DatabaseConfig = {
+  pinecone: {
+    namespace: 'valid-namespace',
+    sparseVector: {
+      indices: [1, 2, 3],
+      values: [0.1, 0.2, 0.3],
+    },
+  },
+  pgvector: {
+    minScore: 0.8,
+    ef: 100,
+    probes: 5,
+  },
+};
+```
+
+## Migration Guide
+
+Existing code will continue to work without changes. To add database-specific configurations:
+
+```diff
+const vectorTool = createVectorQueryTool({
+  indexName: 'my-index',
+  vectorStoreName: 'pinecone',
+  model: embedModel,
++ databaseConfig: {
++   pinecone: {
++     namespace: 'my-namespace'
++   }
++ }
+});
+```

package/src/tools/types.ts

@@ -1,6 +1,57 @@
 import type { EmbeddingModel } from 'ai';
 import type { RerankConfig } from '../rerank';
 
+export interface PineconeConfig {
+  namespace?: string;
+  sparseVector?: {
+    indices: number[];
+    values: number[];
+  };
+}
+
+export interface PgVectorConfig {
+  minScore?: number;
+  ef?: number; // HNSW search parameter
+  probes?: number; // IVFFlat probe parameter
+}
+
+// Chroma types
+type LiteralValue = string | number | boolean;
+type ListLiteralValue = LiteralValue[];
+type LiteralNumber = number;
+type LogicalOperator = '$and' | '$or';
+type InclusionOperator = '$in' | '$nin';
+type WhereOperator = '$gt' | '$gte' | '$lt' | '$lte' | '$ne' | '$eq';
+type OperatorExpression = {
+  [key in WhereOperator | InclusionOperator | LogicalOperator]?: LiteralValue | ListLiteralValue;
+};
+type BaseWhere = {
+  [key: string]: LiteralValue | OperatorExpression;
+};
+type LogicalWhere = {
+  [key in LogicalOperator]?: Where[];
+};
+type Where = BaseWhere | LogicalWhere;
+type WhereDocumentOperator = '$contains' | '$not_contains' | LogicalOperator;
+type WhereDocument = {
+  [key in WhereDocumentOperator]?: LiteralValue | LiteralNumber | WhereDocument[];
+};
+
+export interface ChromaConfig {
+  // Add Chroma-specific configs here if needed
+  where?: Where;
+  whereDocument?: WhereDocument;
+}
+
+// Union type for all database-specific configs
+export type DatabaseConfig = {
+  pinecone?: PineconeConfig;
+  pgvector?: PgVectorConfig;
+  chroma?: ChromaConfig;
+  // Add other database configs as needed
+  [key: string]: any; // Allow for future database extensions
+};
+
 export type VectorQueryToolOptions = {
   id?: string;
   description?: string;
@@ -11,6 +62,8 @@ export type VectorQueryToolOptions = {
   includeVectors?: boolean;
   includeSources?: boolean;
   reranker?: RerankConfig;
+  /** Database-specific configuration options */
+  databaseConfig?: DatabaseConfig;
 };
 
 export type GraphRagToolOptions = {

package/src/tools/vector-query-database-config.test.ts

@@ -0,0 +1,190 @@
+import { RuntimeContext } from '@mastra/core/runtime-context';
+import { describe, it, expect, vi, beforeEach } from 'vitest';
+import { vectorQuerySearch } from '../utils';
+import type { DatabaseConfig } from './types';
+import { createVectorQueryTool } from './vector-query';
+
+vi.mock('../utils', async importOriginal => {
+  const actual: any = await importOriginal();
+  return {
+    ...actual,
+    vectorQuerySearch: vi.fn().mockResolvedValue({
+      results: [{ metadata: { text: 'test result' }, vector: [1, 2, 3] }],
+    }),
+  };
+});
+
+describe('createVectorQueryTool with database-specific configurations', () => {
+  const mockModel = { name: 'test-model' } as any;
+  const mockMastra = {
+    getVector: vi.fn(() => ({})),
+    getLogger: vi.fn(() => ({
+      debug: vi.fn(),
+      warn: vi.fn(),
+      error: vi.fn(),
+    })),
+  };
+
+  beforeEach(() => {
+    vi.clearAllMocks();
+  });
+
+  it('should pass Pinecone configuration to vectorQuerySearch', async () => {
+    const databaseConfig: DatabaseConfig = {
+      pinecone: {
+        namespace: 'test-namespace',
+        sparseVector: {
+          indices: [0, 1, 2],
+          values: [0.1, 0.2, 0.3],
+        },
+      },
+    };
+
+    const tool = createVectorQueryTool({
+      vectorStoreName: 'pinecone',
+      indexName: 'testIndex',
+      model: mockModel,
+      databaseConfig,
+    });
+
+    const runtimeContext = new RuntimeContext();
+
+    await tool.execute({
+      context: { queryText: 'test query', topK: 5 },
+      mastra: mockMastra as any,
+      runtimeContext,
+    });
+
+    expect(vectorQuerySearch).toHaveBeenCalledWith(
+      expect.objectContaining({
+        databaseConfig,
+      }),
+    );
+  });
+
+  it('should pass pgVector configuration to vectorQuerySearch', async () => {
+    const databaseConfig: DatabaseConfig = {
+      pgvector: {
+        minScore: 0.7,
+        ef: 200,
+        probes: 10,
+      },
+    };
+
+    const tool = createVectorQueryTool({
+      vectorStoreName: 'postgres',
+      indexName: 'testIndex',
+      model: mockModel,
+      databaseConfig,
+    });
+
+    const runtimeContext = new RuntimeContext();
+
+    await tool.execute({
+      context: { queryText: 'test query', topK: 5 },
+      mastra: mockMastra as any,
+      runtimeContext,
+    });
+
+    expect(vectorQuerySearch).toHaveBeenCalledWith(
+      expect.objectContaining({
+        databaseConfig,
+      }),
+    );
+  });
+
+  it('should allow runtime context to override database configuration', async () => {
+    const initialConfig: DatabaseConfig = {
+      pinecone: {
+        namespace: 'initial-namespace',
+      },
+    };
+
+    const runtimeConfig: DatabaseConfig = {
+      pinecone: {
+        namespace: 'runtime-namespace',
+      },
+    };
+
+    const tool = createVectorQueryTool({
+      vectorStoreName: 'pinecone',
+      indexName: 'testIndex',
+      model: mockModel,
+      databaseConfig: initialConfig,
+    });
+
+    const runtimeContext = new RuntimeContext();
+    runtimeContext.set('databaseConfig', runtimeConfig);
+
+    await tool.execute({
+      context: { queryText: 'test query', topK: 5 },
+      mastra: mockMastra as any,
+      runtimeContext,
+    });
+
+    expect(vectorQuerySearch).toHaveBeenCalledWith(
+      expect.objectContaining({
+        databaseConfig: runtimeConfig, // Should use runtime config, not initial
+      }),
+    );
+  });
+
+  it('should work without database configuration (backward compatibility)', async () => {
+    const tool = createVectorQueryTool({
+      vectorStoreName: 'testStore',
+      indexName: 'testIndex',
+      model: mockModel,
+      // No databaseConfig provided
+    });
+
+    const runtimeContext = new RuntimeContext();
+
+    await tool.execute({
+      context: { queryText: 'test query', topK: 5 },
+      mastra: mockMastra as any,
+      runtimeContext,
+    });
+
+    expect(vectorQuerySearch).toHaveBeenCalledWith(
+      expect.objectContaining({
+        databaseConfig: undefined,
+      }),
+    );
+  });
+
+  it('should handle multiple database configurations', async () => {
+    const databaseConfig: DatabaseConfig = {
+      pinecone: {
+        namespace: 'test-namespace',
+      },
+      pgvector: {
+        minScore: 0.8,
+        ef: 100,
+      },
+      chroma: {
+        where: { category: 'documents' },
+      },
+    };
+
+    const tool = createVectorQueryTool({
+      vectorStoreName: 'multidb',
+      indexName: 'testIndex',
+      model: mockModel,
+      databaseConfig,
+    });
+
+    const runtimeContext = new RuntimeContext();
+
+    await tool.execute({
+      context: { queryText: 'test query', topK: 5 },
+      mastra: mockMastra as any,
+      runtimeContext,
+    });
+
+    expect(vectorQuerySearch).toHaveBeenCalledWith(
+      expect.objectContaining({
+        databaseConfig,
+      }),
+    );
+  });
+});

package/src/tools/vector-query.ts

@@ -25,6 +25,8 @@ export const createVectorQueryTool = (options: VectorQueryToolOptions) => {
       const includeVectors: boolean = runtimeContext.get('includeVectors') ?? options.includeVectors ?? false;
       const includeSources: boolean = runtimeContext.get('includeSources') ?? options.includeSources ?? true;
       const reranker: RerankConfig = runtimeContext.get('reranker') ?? options.reranker;
+      const databaseConfig = runtimeContext.get('databaseConfig') ?? options.databaseConfig;
+
       if (!indexName) throw new Error(`indexName is required, got: ${indexName}`);
       if (!vectorStoreName) throw new Error(`vectorStoreName is required, got: ${vectorStoreName}`);
 
@@ -40,7 +42,7 @@ export const createVectorQueryTool = (options: VectorQueryToolOptions) => {
         );
       }
       if (logger) {
-        logger.debug('[VectorQueryTool] execute called with:', { queryText, topK, filter });
+        logger.debug('[VectorQueryTool] execute called with:', { queryText, topK, filter, databaseConfig });
       }
       try {
         const topKValue =
@@ -74,7 +76,7 @@ export const createVectorQueryTool = (options: VectorQueryToolOptions) => {
           })();
         }
         if (logger) {
-          logger.debug('Prepared vector query parameters', { queryText, topK: topKValue, queryFilter });
+          logger.debug('Prepared vector query parameters', { queryText, topK: topKValue, queryFilter, databaseConfig });
         }
 
         const { results } = await vectorQuerySearch({
@@ -85,6 +87,7 @@ export const createVectorQueryTool = (options: VectorQueryToolOptions) => {
           queryFilter: Object.keys(queryFilter || {}).length > 0 ? queryFilter : undefined,
           topK: topKValue,
           includeVectors,
+          databaseConfig,
         });
         if (logger) {
           logger.debug('vectorQuerySearch returned results', { count: results.length });

package/src/utils/vector-search.ts

@@ -1,7 +1,8 @@
-import type { MastraVector, QueryResult } from '@mastra/core/vector';
+import type { MastraVector, QueryResult, QueryVectorParams } from '@mastra/core/vector';
 import type { VectorFilter } from '@mastra/core/vector/filter';
 import { embed } from 'ai';
 import type { EmbeddingModel } from 'ai';
+import type { DatabaseConfig } from '../tools/types';
 
 interface VectorQuerySearchParams {
   indexName: string;
@@ -12,6 +13,8 @@ interface VectorQuerySearchParams {
   topK: number;
   includeVectors?: boolean;
   maxRetries?: number;
+  /** Database-specific configuration options */
+  databaseConfig?: DatabaseConfig;
 }
 
 interface VectorQuerySearchResult {
@@ -19,6 +22,14 @@ interface VectorQuerySearchResult {
   queryEmbedding: number[];
 }
 
+enum DatabaseType {
+  Pinecone = 'pinecone',
+  PgVector = 'pgvector',
+  Chroma = 'chroma',
+}
+
+const DATABASE_TYPE_MAP = Object.keys(DatabaseType);
+
 // Helper function to handle vector query search
 export const vectorQuerySearch = async ({
   indexName,
@@ -29,20 +40,78 @@ export const vectorQuerySearch = async ({
   topK,
   includeVectors = false,
   maxRetries = 2,
+  databaseConfig = {},
 }: VectorQuerySearchParams): Promise<VectorQuerySearchResult> => {
   const { embedding } = await embed({
     value: queryText,
     model,
     maxRetries,
   });
-  // Get relevant chunks from the vector database
-  const results = await vectorStore.query({
+
+  // Build query parameters with database-specific configurations
+  const queryParams: QueryVectorParams = {
     indexName,
     queryVector: embedding,
     topK,
     filter: queryFilter,
     includeVector: includeVectors,
-  });
+  };
+
+  // Get relevant chunks from the vector database
+  const results = await vectorStore.query({ ...queryParams, ...databaseSpecificParams(databaseConfig) });
 
   return { results, queryEmbedding: embedding };
 };
+
+const databaseSpecificParams = (databaseConfig: DatabaseConfig) => {
+  const databaseSpecificParams: DatabaseConfig = {};
+
+  // Apply database-specific configurations
+  if (databaseConfig) {
+    // Pinecone-specific configurations
+    if (databaseConfig.pinecone) {
+      if (databaseConfig.pinecone.namespace) {
+        databaseSpecificParams.namespace = databaseConfig.pinecone.namespace;
+      }
+      if (databaseConfig.pinecone.sparseVector) {
+        databaseSpecificParams.sparseVector = databaseConfig.pinecone.sparseVector;
+      }
+    }
+
+    // pgVector-specific configurations
+    if (databaseConfig.pgvector) {
+      if (databaseConfig.pgvector.minScore !== undefined) {
+        databaseSpecificParams.minScore = databaseConfig.pgvector.minScore;
+      }
+      if (databaseConfig.pgvector.ef !== undefined) {
+        databaseSpecificParams.ef = databaseConfig.pgvector.ef;
+      }
+      if (databaseConfig.pgvector.probes !== undefined) {
+        databaseSpecificParams.probes = databaseConfig.pgvector.probes;
+      }
+    }
+
+    // Chroma-specific configurations
+    if (databaseConfig.chroma) {
+      if (databaseConfig.chroma.where) {
+        databaseSpecificParams.where = databaseConfig.chroma.where;
+      }
+      if (databaseConfig.chroma.whereDocument) {
+        databaseSpecificParams.whereDocument = databaseConfig.chroma.whereDocument;
+      }
+    }
+
+    // Handle any additional database configs
+    Object.keys(databaseConfig).forEach(dbName => {
+      if (!DATABASE_TYPE_MAP.includes(dbName)) {
+        // For unknown database types, merge the config directly
+        const config = databaseConfig[dbName];
+        if (config && typeof config === 'object') {
+          Object.assign(databaseSpecificParams, config);
+        }
+      }
+    });
+  }
+
+  return databaseSpecificParams;
+};