genkitx-azure-openai/lib/embedder.d.mts

Firebase Genkit AI framework plugin for Azure OpenAI APIs.

import * as genkit from 'genkit';
import { z, Genkit } from 'genkit';
import { AzureOpenAI } from 'openai';

declare const TextEmbeddingConfigSchema: z.ZodObject<{
    dimensions: z.ZodOptional<z.ZodNumber>;
    encodingFormat: z.ZodOptional<z.ZodUnion<[z.ZodLiteral<"float">, z.ZodLiteral<"base64">]>>;
}, "strip", z.ZodTypeAny, {
    dimensions?: number | undefined;
    encodingFormat?: "float" | "base64" | undefined;
}, {
    dimensions?: number | undefined;
    encodingFormat?: "float" | "base64" | undefined;
}>;
type TextEmbeddingGeckoConfig = z.infer<typeof TextEmbeddingConfigSchema>;
declare const TextEmbeddingInputSchema: z.ZodString;
declare const textEmbedding3Small: genkit.EmbedderReference<z.ZodObject<{
    dimensions: z.ZodOptional<z.ZodNumber>;
    encodingFormat: z.ZodOptional<z.ZodUnion<[z.ZodLiteral<"float">, z.ZodLiteral<"base64">]>>;
}, "strip", z.ZodTypeAny, {
    dimensions?: number | undefined;
    encodingFormat?: "float" | "base64" | undefined;
}, {
    dimensions?: number | undefined;
    encodingFormat?: "float" | "base64" | undefined;
}>>;
declare const textEmbedding3Large: genkit.EmbedderReference<z.ZodObject<{
    dimensions: z.ZodOptional<z.ZodNumber>;
    encodingFormat: z.ZodOptional<z.ZodUnion<[z.ZodLiteral<"float">, z.ZodLiteral<"base64">]>>;
}, "strip", z.ZodTypeAny, {
    dimensions?: number | undefined;
    encodingFormat?: "float" | "base64" | undefined;
}, {
    dimensions?: number | undefined;
    encodingFormat?: "float" | "base64" | undefined;
}>>;
declare const textEmbeddingAda002: genkit.EmbedderReference<z.ZodObject<{
    dimensions: z.ZodOptional<z.ZodNumber>;
    encodingFormat: z.ZodOptional<z.ZodUnion<[z.ZodLiteral<"float">, z.ZodLiteral<"base64">]>>;
}, "strip", z.ZodTypeAny, {
    dimensions?: number | undefined;
    encodingFormat?: "float" | "base64" | undefined;
}, {
    dimensions?: number | undefined;
    encodingFormat?: "float" | "base64" | undefined;
}>>;
declare const SUPPORTED_EMBEDDING_MODELS: {
    'text-embedding-3-small': genkit.EmbedderReference<z.ZodObject<{
        dimensions: z.ZodOptional<z.ZodNumber>;
        encodingFormat: z.ZodOptional<z.ZodUnion<[z.ZodLiteral<"float">, z.ZodLiteral<"base64">]>>;
    }, "strip", z.ZodTypeAny, {
        dimensions?: number | undefined;
        encodingFormat?: "float" | "base64" | undefined;
    }, {
        dimensions?: number | undefined;
        encodingFormat?: "float" | "base64" | undefined;
    }>>;
    'text-embedding-3-large': genkit.EmbedderReference<z.ZodObject<{
        dimensions: z.ZodOptional<z.ZodNumber>;
        encodingFormat: z.ZodOptional<z.ZodUnion<[z.ZodLiteral<"float">, z.ZodLiteral<"base64">]>>;
    }, "strip", z.ZodTypeAny, {
        dimensions?: number | undefined;
        encodingFormat?: "float" | "base64" | undefined;
    }, {
        dimensions?: number | undefined;
        encodingFormat?: "float" | "base64" | undefined;
    }>>;
    'text-embedding-ada-002': genkit.EmbedderReference<z.ZodObject<{
        dimensions: z.ZodOptional<z.ZodNumber>;
        encodingFormat: z.ZodOptional<z.ZodUnion<[z.ZodLiteral<"float">, z.ZodLiteral<"base64">]>>;
    }, "strip", z.ZodTypeAny, {
        dimensions?: number | undefined;
        encodingFormat?: "float" | "base64" | undefined;
    }, {
        dimensions?: number | undefined;
        encodingFormat?: "float" | "base64" | undefined;
    }>>;
};
declare function openaiEmbedder(ai: Genkit, name: string, client: AzureOpenAI): genkit.EmbedderAction<z.ZodObject<{
    dimensions: z.ZodOptional<z.ZodNumber>;
    encodingFormat: z.ZodOptional<z.ZodUnion<[z.ZodLiteral<"float">, z.ZodLiteral<"base64">]>>;
}, "strip", z.ZodTypeAny, {
    dimensions?: number | undefined;
    encodingFormat?: "float" | "base64" | undefined;
}, {
    dimensions?: number | undefined;
    encodingFormat?: "float" | "base64" | undefined;
}>>;

export { SUPPORTED_EMBEDDING_MODELS, TextEmbeddingConfigSchema, type TextEmbeddingGeckoConfig, TextEmbeddingInputSchema, openaiEmbedder, textEmbedding3Large, textEmbedding3Small, textEmbeddingAda002 };