UNPKG

rag-cli-tester

Version:

A lightweight CLI tool for testing RAG (Retrieval-Augmented Generation) systems with different embedding combinations

22 lines 1.49 kB
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