@mastra/rag 1.2.2 → 1.2.3-alpha.1

package/src/utils/vector-search.ts

@@ -1,117 +0,0 @@
-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;
-  vectorStore: MastraVector;
-  queryText: string;
-  model: EmbeddingModel<string>;
-  queryFilter?: VectorFilter;
-  topK: number;
-  includeVectors?: boolean;
-  maxRetries?: number;
-  /** Database-specific configuration options */
-  databaseConfig?: DatabaseConfig;
-}
-
-interface VectorQuerySearchResult {
-  results: QueryResult[];
-  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,
-  vectorStore,
-  queryText,
-  model,
-  queryFilter,
-  topK,
-  includeVectors = false,
-  maxRetries = 2,
-  databaseConfig = {},
-}: VectorQuerySearchParams): Promise<VectorQuerySearchResult> => {
-  const { embedding } = await embed({
-    value: queryText,
-    model,
-    maxRetries,
-  });
-
-  // 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;
-};

package/tsconfig.build.json

@@ -1,9 +0,0 @@
-{
-  "extends": ["./tsconfig.json", "../../tsconfig.build.json"],
-  "compilerOptions": {
-    "outDir": "./dist",
-    "rootDir": "./src"
-  },
-  "include": ["src/**/*"],
-  "exclude": ["node_modules", "**/*.test.ts", "src/**/*.mock.ts"]
-}

package/tsconfig.json

@@ -1,5 +0,0 @@
-{
-  "extends": "../../tsconfig.node.json",
-  "include": ["src/**/*", "tsup.config.ts"],
-  "exclude": ["node_modules", "**/*.test.ts"]
-}

package/tsup.config.ts

@@ -1,17 +0,0 @@
-import { generateTypes } from '@internal/types-builder';
-import { defineConfig } from 'tsup';
-
-export default defineConfig({
-  entry: ['src/index.ts'],
-  format: ['esm', 'cjs'],
-  clean: true,
-  dts: false,
-  splitting: true,
-  treeshake: {
-    preset: 'smallest',
-  },
-  sourcemap: true,
-  onSuccess: async () => {
-    await generateTypes(process.cwd());
-  },
-});

package/vitest.config.ts

@@ -1,8 +0,0 @@
-import { defineConfig } from 'vitest/config';
-
-export default defineConfig({
-  test: {
-    environment: 'node',
-    include: ['src/**/*.test.ts'],
-  },
-});