UNPKG

rag-aiquest

Version:

### Aiquest is an npm package that streamlines the process of parsing websites, splitting content into manageable chunks, embedding these chunks into machine-friendly vectors, and subsequently storing and retrieving these embeddings from AWS. This documen

57 lines (49 loc) 1.62 kB
import { OpenAI } from 'openai'; interface EmbeddingEntry$1 { content: string; embedding: number[]; } declare class VectorStoreAWS { private s3Bucket; private bucketName; constructor(accessKeyId: string, secretAccessKey: string, bucketName: string); uploadEmbededModeltoAWS(embeddingStore: { content: string; embedding: number[]; }[], fileName: string): Promise<{ embededFileLocation: string; }>; getKnowledgeData(fileName: string): Promise<EmbeddingEntry$1[]>; } interface EmbeddingEntry { content: string; embedding: number[]; } declare class Retrival { private openai; constructor(apiKey: string); QnARetrival(embeddingStore: EmbeddingEntry[], question: string): Promise<OpenAI.Chat.Completions.ChatCompletion>; semanticSearch(query: string, embeddingStore: EmbeddingEntry[], topN?: number): Promise<EmbeddingEntry[]>; private findNearestParagraph; private Prompt; private compareEmbeddings; private cosineSimilarity; } interface IParse { parse(input: string | Buffer): Promise<string>; } declare class UnifiedParser implements IParse { parse(input: string | Buffer): Promise<string>; } declare class ChunkUtility { static splitIntoChunks(text: string, numOfChunks: number, overlapSplitChunks: number): string[]; } declare class EmbeddingUtility { private openai; constructor(apiKey: string); createEmbedding(chunks: string[]): Promise<{ content: string; embedding: number[]; }[]>; } export { ChunkUtility, EmbeddingUtility, Retrival, UnifiedParser, VectorStoreAWS };