@antl3x/toolrag
Version:
Context-aware tool retrieval for language models - unlock the full potential of LLM function calling without context window limitations or constraints.
29 lines • 1.14 kB
TypeScript
import OpenAI from 'openai';
import { EmbeddingProvider } from './EmbeddingProvider.js';
import { z } from 'zod';
declare const EmbeddingProviderOpenAIConfigSchema: z.ZodDefault<z.ZodObject<{
model: z.ZodDefault<z.ZodEnum<["text-embedding-3-small", "text-embedding-3-large"]>>;
client: z.ZodOptional<z.ZodType<OpenAI, z.ZodTypeDef, OpenAI>>;
}, "strip", z.ZodTypeAny, {
model: "text-embedding-3-small" | "text-embedding-3-large";
client?: OpenAI | undefined;
}, {
model?: "text-embedding-3-small" | "text-embedding-3-large" | undefined;
client?: OpenAI | undefined;
}>>;
type EmbeddingProviderOpenAIConfig = z.infer<typeof EmbeddingProviderOpenAIConfigSchema>;
/**
* OpenAI implementation of the EmbeddingProvider
*/
export declare class EmbeddingProviderOpenAI implements EmbeddingProvider {
private _client;
private _dimensions;
private _config;
constructor(config?: EmbeddingProviderOpenAIConfig);
getEmbedding(text: string): Promise<number[]>;
getDimensions(): number;
getName(): string;
getModel(): string;
}
export {};
//# sourceMappingURL=EmbeddingProviderOpenAI.d.ts.map