@mastra/rag 1.2.3-alpha.0 → 1.2.3-alpha.1

package/src/tools/types.ts

@@ -1,126 +0,0 @@
-import type { MastraVector, MastraEmbeddingModel } from '@mastra/core/vector';
-
-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;
-  indexName: string;
-  model: MastraEmbeddingModel<string>;
-  enableFilter?: boolean;
-  includeVectors?: boolean;
-  includeSources?: boolean;
-  reranker?: RerankConfig;
-  /** Database-specific configuration options */
-  databaseConfig?: DatabaseConfig;
-} & ProviderOptions &
-  (
-    | {
-        vectorStoreName: string;
-      }
-    | {
-        vectorStoreName?: string;
-        vectorStore: MastraVector;
-      }
-  );
-
-export type GraphRagToolOptions = {
-  id?: string;
-  description?: string;
-  indexName: string;
-  vectorStoreName: string;
-  model: MastraEmbeddingModel<string>;
-  enableFilter?: boolean;
-  includeSources?: boolean;
-  graphOptions?: {
-    dimension?: number;
-    randomWalkSteps?: number;
-    restartProb?: number;
-    threshold?: number;
-  };
-} & ProviderOptions;
-
-export type ProviderOptions = {
-  /**
-   * Provider-specific options for the embedding model (e.g., outputDimensionality).
-   *
-   * ⚠️  **IMPORTANT**: `providerOptions` only work with AI SDK v2 models.
-   *
-   * **For v1 models**: Configure options when creating the model:
-   * ✅ const model = openai.embedding('text-embedding-3-small', { dimensions: 512 });
-   *
-   * **For v2 models**: Use providerOptions:
-   * ✅ providerOptions: { openai: { dimensions: 512 } }
-   */
-  providerOptions?: Record<string, Record<string, any>>;
-};
-
-/**
- * Default options for GraphRAG
- * @default
- * ```json
- * {
- *   "dimension": 1536,
- *   "randomWalkSteps": 100,
- *   "restartProb": 0.15,
- *   "threshold": 0.7
- * }
- * ```
- */
-export const defaultGraphOptions = {
-  dimension: 1536,
-  randomWalkSteps: 100,
-  restartProb: 0.15,
-  threshold: 0.7,
-};

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

@@ -1,190 +0,0 @@
-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,
-      }),
-    );
-  });
-});