UNPKG

genkitx-azure-openai

Version:

Firebase Genkit AI framework plugin for Azure OpenAI APIs.

91 lines (88 loc) 4.06 kB
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 };