rag-cli-tester
Version:
A lightweight CLI tool for testing RAG (Retrieval-Augmented Generation) systems with different embedding combinations
22 lines • 1.49 kB
TypeScript
import { DatabaseConfig, TableInfo } from './types';
export declare class DatabaseConnection {
private supabase;
private isConnected;
constructor(config: DatabaseConfig);
testConnection(): Promise<boolean>;
getTables(): Promise<string[]>;
getTableInfo(tableName: string): Promise<TableInfo | null>;
getTableData(tableName: string, columns?: string[], limit?: number, offset?: number): Promise<any[]>;
isConnectionActive(): boolean;
getColumnDataType(tableName: string, columnName: string): Promise<string | null>;
updateRowEmbedding(tableName: string, rowId: any, embeddingColumn: string, embedding: number[]): Promise<boolean>;
updateRowColumn(tableName: string, rowId: any, columnName: string, value: any): Promise<boolean>;
getRowsWithoutEmbeddings(tableName: string, embeddingColumn: string, sourceColumns: string[], limit?: number): Promise<any[]>;
getRowsWithEmptyColumn(tableName: string, targetColumn: string, sourceColumns: string[], limit?: number): Promise<any[]>;
checkColumnExists(tableName: string, columnName: string): Promise<boolean>;
getColumnDataCount(tableName: string, columnName: string): Promise<number>;
getEmptyColumnCount(tableName: string, columnName: string): Promise<number>;
getRowColumnValue(tableName: string, rowId: any, columnName: string): Promise<any>;
getTableDataSample(tableName: string, sampleSize: number, ratio: number): Promise<any[]>;
}
//# sourceMappingURL=database.d.ts.map