rag-cli-tester
Version:
A lightweight CLI tool for testing RAG (Retrieval-Augmented Generation) systems with different embedding combinations
44 lines • 1.61 kB
TypeScript
import { EmbeddingConfig, LLMConfig } from './types';
export interface EmbeddingProvider {
generateEmbedding(text: string): Promise<number[]>;
initialize(): Promise<void>;
}
export interface LLMProvider {
generateText(prompt: string, context: string): Promise<string>;
initialize(): Promise<void>;
}
export declare class LocalEmbeddingProvider implements EmbeddingProvider {
private pipeline;
private config;
constructor(config: EmbeddingConfig);
initialize(): Promise<void>;
generateEmbedding(text: string): Promise<number[]>;
}
export declare class OpenAIEmbeddingProvider implements EmbeddingProvider {
private config;
constructor(config: EmbeddingConfig);
initialize(): Promise<void>;
generateEmbedding(text: string): Promise<number[]>;
}
export declare class GeminiEmbeddingProvider implements EmbeddingProvider {
private config;
constructor(config: EmbeddingConfig);
initialize(): Promise<void>;
generateEmbedding(text: string): Promise<number[]>;
}
export declare class OpenAICompatibleLLMProvider implements LLMProvider {
private config;
constructor(config: LLMConfig);
initialize(): Promise<void>;
generateText(prompt: string, context: string): Promise<string>;
private getProviderConfig;
}
export declare class ProviderManager {
static createEmbeddingProvider(config: EmbeddingConfig): EmbeddingProvider;
static createLLMProvider(config: LLMConfig): LLMProvider;
static detectAvailableProviders(): {
embedding: string[];
llm: string[];
};
}
//# sourceMappingURL=providers.d.ts.map