@genkit-ai/compat-oai/lib/deepseek/deepseek.mjs

Genkit AI framework plugin for OpenAI APIs.

import { z } from "genkit";
import {
  ChatCompletionCommonConfigSchema,
  compatOaiModelRef
} from "../model";
const DeepSeekChatCompletionConfigSchema = ChatCompletionCommonConfigSchema.extend({
  maxTokens: z.number().int().min(1).max(8192).optional()
});
const deepSeekRequestBuilder = (req, params) => {
  const { maxTokens } = req.config;
  params.max_tokens = maxTokens;
};
function deepSeekModelRef(params) {
  return 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" })
};
export {
  DeepSeekChatCompletionConfigSchema,
  SUPPORTED_DEEPSEEK_MODELS,
  deepSeekModelRef,
  deepSeekRequestBuilder
};
//# sourceMappingURL=deepseek.mjs.map