UNPKG

@antl3x/toolrag

Version:

Context-aware tool retrieval for language models - unlock the full potential of LLM function calling without context window limitations or constraints.

50 lines 2.08 kB
import { z } from 'zod'; import { EmbeddingProviderOpenAI } from './EmbeddingProviderOpenAI'; import { EmbeddingProviderGoogle } from './EmbeddingProviderGoogle'; declare const ToolRAGConfigSchema: z.ZodObject<{ embeddingProvider: z.ZodDefault<z.ZodUnion<[z.ZodLiteral<"openai">, z.ZodLiteral<"google">, z.ZodType<EmbeddingProviderOpenAI, z.ZodTypeDef, EmbeddingProviderOpenAI>, z.ZodType<EmbeddingProviderGoogle, z.ZodTypeDef, EmbeddingProviderGoogle>]>>; mcpServers: z.ZodDefault<z.ZodArray<z.ZodString, "many">>; rerank: z.ZodOptional<z.ZodObject<{ enabled: z.ZodOptional<z.ZodBoolean>; threshold: z.ZodOptional<z.ZodNumber>; }, "strip", z.ZodTypeAny, { enabled?: boolean | undefined; threshold?: number | undefined; }, { enabled?: boolean | undefined; threshold?: number | undefined; }>>; database: z.ZodDefault<z.ZodObject<{ url: z.ZodDefault<z.ZodString>; }, "strip", z.ZodTypeAny, { url: string; }, { url?: string | undefined; }>>; }, "strip", z.ZodTypeAny, { embeddingProvider: EmbeddingProviderGoogle | "google" | EmbeddingProviderOpenAI | "openai"; mcpServers: string[]; database: { url: string; }; rerank?: { enabled?: boolean | undefined; threshold?: number | undefined; } | undefined; }, { embeddingProvider?: EmbeddingProviderGoogle | "google" | EmbeddingProviderOpenAI | "openai" | undefined; mcpServers?: string[] | undefined; rerank?: { enabled?: boolean | undefined; threshold?: number | undefined; } | undefined; database?: { url?: string | undefined; } | undefined; }>; type ToolRAGConfigInput = z.input<typeof ToolRAGConfigSchema>; type ToolRAGConfig = z.output<typeof ToolRAGConfigSchema>; declare const setupConfig: (config: ToolRAGConfigInput | undefined) => ToolRAGConfig; declare const getConfig: () => ToolRAGConfig; export { setupConfig, getConfig, type ToolRAGConfig, type ToolRAGConfigInput }; //# sourceMappingURL=ToolRAGConfig.d.ts.map