import { z } from "zod";
export declare const embedParamsSchema: z.ZodObject<{
    modelId: z.ZodString;
    text: z.ZodUnion<readonly [z.ZodString, z.ZodArray<z.ZodString>]>;
}, z.core.$strip>;
export declare const embedRequestSchema: z.ZodObject<{
    modelId: z.ZodString;
    text: z.ZodUnion<readonly [z.ZodString, z.ZodArray<z.ZodString>]>;
    type: z.ZodLiteral<"embed">;
    requestId: z.ZodOptional<z.ZodString>;
}, z.core.$strip>;
export declare const embedStatsSchema: z.ZodObject<{
    totalTime: z.ZodOptional<z.ZodNumber>;
    tokensPerSecond: z.ZodOptional<z.ZodNumber>;
    totalTokens: z.ZodOptional<z.ZodNumber>;
    backendDevice: z.ZodOptional<z.ZodEnum<{
        cpu: "cpu";
        gpu: "gpu";
    }>>;
    contextSize: z.ZodOptional<z.ZodNumber>;
}, z.core.$strip>;
export declare const embedResponseSchema: z.ZodObject<{
    type: z.ZodLiteral<"embed">;
    success: z.ZodBoolean;
    embedding: z.ZodDefault<z.ZodUnion<readonly [z.ZodArray<z.ZodNumber>, z.ZodArray<z.ZodArray<z.ZodNumber>>]>>;
    stats: z.ZodOptional<z.ZodObject<{
        totalTime: z.ZodOptional<z.ZodNumber>;
        tokensPerSecond: z.ZodOptional<z.ZodNumber>;
        totalTokens: z.ZodOptional<z.ZodNumber>;
        backendDevice: z.ZodOptional<z.ZodEnum<{
            cpu: "cpu";
            gpu: "gpu";
        }>>;
        contextSize: z.ZodOptional<z.ZodNumber>;
    }, z.core.$strip>>;
    error: z.ZodOptional<z.ZodString>;
}, z.core.$strip>;
export type EmbedParams = z.infer<typeof embedParamsSchema>;
export type EmbedRequest = z.infer<typeof embedRequestSchema>;
export type EmbedResponse = z.infer<typeof embedResponseSchema>;
export type EmbedStats = z.infer<typeof embedStatsSchema>;
//# sourceMappingURL=embed.d.ts.map