genkitx-azure-openai
Version:
Firebase Genkit AI framework plugin for Azure OpenAI APIs.
91 lines (88 loc) • 4.06 kB
text/typescript
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 };