extend IndexStats

mastra-ai · Feb 21, 2025 · 6d50fb6 · 6d50fb6
1 parent e0a1307
commit 6d50fb6
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 8 deletions.
diff --git a/packages/core/src/vector/index.ts b/packages/core/src/vector/index.ts
@@ -13,13 +13,6 @@ export interface IndexStats {
   dimension: number;
   count: number;
   metric?: 'cosine' | 'euclidean' | 'dotproduct';
-  type?: 'flat' | 'hnsw' | 'ivfflat';
-  config?: {
-    m?: number;
-    efConstruction?: number;
-    lists?: number;
-    probes?: number;
-  };
 }
 
 export abstract class MastraVector extends MastraBase {

diff --git a/stores/pg/src/vector/index.ts b/stores/pg/src/vector/index.ts
@@ -6,6 +6,16 @@ import { PGFilterTranslator } from './filter';
 import { buildFilterQuery } from './sql-builder';
 import { type IndexConfig } from './types';
 
+export interface PGIndexStats extends IndexStats {
+  type: 'flat' | 'hnsw' | 'ivfflat';
+  config: {
+    m?: number;
+    efConstruction?: number;
+    lists?: number;
+    probes?: number;
+  };
+}
+
 export class PgVector extends MastraVector {
   private pool: pg.Pool;
 
@@ -251,7 +261,7 @@ export class PgVector extends MastraVector {
     }
   }
 
-  async describeIndex(indexName: string): Promise<IndexStats> {
+  async describeIndex(indexName: string): Promise<PGIndexStats> {
     const client = await this.pool.connect();
     try {
       // Get vector dimension