@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
TypeScript
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