@genkit-ai/compat-oai
Version:
Genkit AI framework plugin for OpenAI APIs.
62 lines • 2.3 kB
JavaScript
;
var __defProp = Object.defineProperty;
var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
var __getOwnPropNames = Object.getOwnPropertyNames;
var __hasOwnProp = Object.prototype.hasOwnProperty;
var __export = (target, all) => {
for (var name in all)
__defProp(target, name, { get: all[name], enumerable: true });
};
var __copyProps = (to, from, except, desc) => {
if (from && typeof from === "object" || typeof from === "function") {
for (let key of __getOwnPropNames(from))
if (!__hasOwnProp.call(to, key) && key !== except)
__defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
}
return to;
};
var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
var deepseek_exports = {};
__export(deepseek_exports, {
DeepSeekChatCompletionConfigSchema: () => DeepSeekChatCompletionConfigSchema,
SUPPORTED_DEEPSEEK_MODELS: () => SUPPORTED_DEEPSEEK_MODELS,
deepSeekModelRef: () => deepSeekModelRef,
deepSeekRequestBuilder: () => deepSeekRequestBuilder
});
module.exports = __toCommonJS(deepseek_exports);
var import_genkit = require("genkit");
var import_model2 = require("../model");
const DeepSeekChatCompletionConfigSchema = import_model2.ChatCompletionCommonConfigSchema.extend({
maxTokens: import_genkit.z.number().int().min(1).max(8192).optional()
});
const deepSeekRequestBuilder = (req, params) => {
const { maxTokens } = req.config;
params.max_tokens = maxTokens;
};
function deepSeekModelRef(params) {
return (0, import_model2.compatOaiModelRef)({
...params,
configSchema: DeepSeekChatCompletionConfigSchema,
info: params.info ?? {
supports: {
multiturn: true,
tools: true,
media: false,
systemRole: true,
output: ["text", "json"]
}
}
});
}
const SUPPORTED_DEEPSEEK_MODELS = {
"deepseek-reasoner": deepSeekModelRef({ name: "deepseek/deepseek-reasoner" }),
"deepseek-chat": deepSeekModelRef({ name: "deepseek/deepseek-chat" })
};
// Annotate the CommonJS export names for ESM import in node:
0 && (module.exports = {
DeepSeekChatCompletionConfigSchema,
SUPPORTED_DEEPSEEK_MODELS,
deepSeekModelRef,
deepSeekRequestBuilder
});
//# sourceMappingURL=deepseek.js.map