@genkit-ai/compat-oai
Version:
Genkit AI framework plugin for OpenAI APIs.
33 lines • 877 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 commonRef(name, info) {
return embedderRef({
name,
configSchema: TextEmbeddingConfigSchema,
info: {
...COMMON_EMBEDDER_INFO,
...info ?? {}
}
});
}
const SUPPORTED_EMBEDDING_MODELS = {
"text-embedding-3-small": commonRef("openai/text-embedding-3-small"),
"text-embedding-3-large": commonRef("openai/text-embedding-3-large", {
dimensions: 3072
}),
"text-embedding-ada-002": commonRef("openai/text-embedding-ada-002")
};
export {
SUPPORTED_EMBEDDING_MODELS,
TextEmbeddingConfigSchema
};
//# sourceMappingURL=embedder.mjs.map