UNPKG

rag-cli-tester

Version:

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

44 lines 1.61 kB
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