@genkit-ai/compat-oai
Version:
Genkit AI framework plugin for OpenAI APIs.
34 lines • 913 B
JavaScript
import { embedderRef, z } from "genkit";
const TextEmbeddingConfigSchema = z.object({
dimensions: z.number().optional(),
encodingFormat: z.union([z.literal("float"), z.literal("base64")]).optional()
});
const COMMON_EMBEDDER_INFO = {
dimensions: 1536,
supports: {
input: ["text"]
}
};
function openAIEmbedderRef(name, info) {
return embedderRef({
name,
configSchema: TextEmbeddingConfigSchema,
info: {
...COMMON_EMBEDDER_INFO,
...info ?? {}
},
namespace: "openai"
});
}
const SUPPORTED_EMBEDDING_MODELS = {
"text-embedding-3-small": openAIEmbedderRef("text-embedding-3-small"),
"text-embedding-3-large": openAIEmbedderRef("text-embedding-3-large", {
dimensions: 3072
}),
"text-embedding-ada-002": openAIEmbedderRef("text-embedding-ada-002")
};
export {
SUPPORTED_EMBEDDING_MODELS,
TextEmbeddingConfigSchema
};
//# sourceMappingURL=embedder.mjs.map