@genkit-ai/compat-oai 0.0.2

package/lib/deepseek/deepseek.mjs

@@ -0,0 +1,38 @@
+import { modelRef } from "genkit/model";
+import { ChatCompletionCommonConfigSchema } from "../model";
+const deepseekChat = modelRef({
+  name: "deepseek/deepseek-chat",
+  info: {
+    label: "DeepSeek - DeepSeek Chat",
+    supports: {
+      multiturn: true,
+      tools: true,
+      media: false,
+      systemRole: true,
+      output: ["text", "json"]
+    }
+  },
+  configSchema: ChatCompletionCommonConfigSchema
+});
+const deepseekReasoner = modelRef({
+  name: "deepseek/deepseek-reasoner",
+  info: {
+    label: "DeepSeek - DeepSeek Reasoner",
+    supports: {
+      multiturn: true,
+      tools: true,
+      media: false,
+      systemRole: true,
+      output: ["text", "json"]
+    }
+  },
+  configSchema: ChatCompletionCommonConfigSchema
+});
+const SUPPORTED_DEEPSEEK_MODELS = {
+  "deepseek-reasoner": deepseekReasoner,
+  "deepseek-chat": deepseekChat
+};
+export {
+  SUPPORTED_DEEPSEEK_MODELS
+};
+//# sourceMappingURL=deepseek.mjs.map

package/lib/deepseek/deepseek.mjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/deepseek/deepseek.ts"],"sourcesContent":["/**\n * Copyright 2024 Google LLC\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport { modelRef } from 'genkit/model';\nimport { ChatCompletionCommonConfigSchema } from '../model';\n\nconst deepseekChat = modelRef({\n  name: 'deepseek/deepseek-chat',\n  info: {\n    label: 'DeepSeek - DeepSeek Chat',\n    supports: {\n      multiturn: true,\n      tools: true,\n      media: false,\n      systemRole: true,\n      output: ['text', 'json'],\n    },\n  },\n  configSchema: ChatCompletionCommonConfigSchema,\n});\n\nconst deepseekReasoner = modelRef({\n  name: 'deepseek/deepseek-reasoner',\n  info: {\n    label: 'DeepSeek - DeepSeek Reasoner',\n    supports: {\n      multiturn: true,\n      tools: true,\n      media: false,\n      systemRole: true,\n      output: ['text', 'json'],\n    },\n  },\n  configSchema: ChatCompletionCommonConfigSchema,\n});\n\nexport const SUPPORTED_DEEPSEEK_MODELS = {\n  'deepseek-reasoner': deepseekReasoner,\n  'deepseek-chat': deepseekChat,\n};\n"],"mappings":"AAgBA,SAAS,gBAAgB;AACzB,SAAS,wCAAwC;AAEjD,MAAM,eAAe,SAAS;AAAA,EAC5B,MAAM;AAAA,EACN,MAAM;AAAA,IACJ,OAAO;AAAA,IACP,UAAU;AAAA,MACR,WAAW;AAAA,MACX,OAAO;AAAA,MACP,OAAO;AAAA,MACP,YAAY;AAAA,MACZ,QAAQ,CAAC,QAAQ,MAAM;AAAA,IACzB;AAAA,EACF;AAAA,EACA,cAAc;AAChB,CAAC;AAED,MAAM,mBAAmB,SAAS;AAAA,EAChC,MAAM;AAAA,EACN,MAAM;AAAA,IACJ,OAAO;AAAA,IACP,UAAU;AAAA,MACR,WAAW;AAAA,MACX,OAAO;AAAA,MACP,OAAO;AAAA,MACP,YAAY;AAAA,MACZ,QAAQ,CAAC,QAAQ,MAAM;AAAA,IACzB;AAAA,EACF;AAAA,EACA,cAAc;AAChB,CAAC;AAEM,MAAM,4BAA4B;AAAA,EACvC,qBAAqB;AAAA,EACrB,iBAAiB;AACnB;","names":[]}

package/lib/deepseek/index.d.mts

@@ -0,0 +1,67 @@
+import { z, ModelReference } from 'genkit';
+import { GenkitPlugin } from 'genkit/plugin';
+import { PluginOptions } from '../index.mjs';
+import { ChatCompletionCommonConfigSchema } from '../model.mjs';
+import { SUPPORTED_DEEPSEEK_MODELS } from './deepseek.mjs';
+import 'genkit/registry';
+import 'openai';
+import 'genkit/model';
+import 'openai/resources/index.mjs';
+import 'zod';
+
+/**
+ * Copyright 2024 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+type DeepSeekPluginOptions = Omit<PluginOptions, 'name' | 'baseURL'>;
+declare function deepSeekPlugin(options?: DeepSeekPluginOptions): GenkitPlugin;
+type DeepSeekPlugin = {
+    (params?: DeepSeekPluginOptions): GenkitPlugin;
+    model(name: keyof typeof SUPPORTED_DEEPSEEK_MODELS, config?: z.infer<typeof ChatCompletionCommonConfigSchema>): ModelReference<typeof ChatCompletionCommonConfigSchema>;
+    model(name: string, config?: any): ModelReference<z.ZodTypeAny>;
+};
+/**
+ * This module provides an interface to the DeepSeek models through the Genkit
+ * plugin system. It allows users to interact with various models by providing
+ * an API key and optional configuration.
+ *
+ * The main export is the `deepseek` plugin, which can be configured with an API
+ * key either directly or through environment variables. It initializes the
+ * OpenAI client and makes available the models for use.
+ *
+ * Exports:
+ * - deepSeek: The main plugin function to interact with DeepSeek, via OpenAI
+ *   compatible API.
+ *
+ * Usage: To use the models, initialize the deepseek plugin inside
+ * `configureGenkit` and pass the configuration options. If no API key is
+ * provided in the options, the environment variable `OPENAI_API_KEY` must be
+ * set.
+ *
+ * Example:
+ * ```
+ * import { deepSeek } from '@genkit-ai/compat-oai/deepseek';
+ *
+ * export default configureGenkit({
+ *  plugins: [
+ *    deepSeek()
+ *    ... // other plugins
+ *  ]
+ * });
+ * ```
+ */
+declare const deepSeek: DeepSeekPlugin;
+
+export { type DeepSeekPlugin, type DeepSeekPluginOptions, deepSeek, deepSeekPlugin, deepSeek as default };

package/lib/deepseek/index.d.ts

@@ -0,0 +1,67 @@
+import { z, ModelReference } from 'genkit';
+import { GenkitPlugin } from 'genkit/plugin';
+import { PluginOptions } from '../index.js';
+import { ChatCompletionCommonConfigSchema } from '../model.js';
+import { SUPPORTED_DEEPSEEK_MODELS } from './deepseek.js';
+import 'genkit/registry';
+import 'openai';
+import 'genkit/model';
+import 'openai/resources/index.mjs';
+import 'zod';
+
+/**
+ * Copyright 2024 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+type DeepSeekPluginOptions = Omit<PluginOptions, 'name' | 'baseURL'>;
+declare function deepSeekPlugin(options?: DeepSeekPluginOptions): GenkitPlugin;
+type DeepSeekPlugin = {
+    (params?: DeepSeekPluginOptions): GenkitPlugin;
+    model(name: keyof typeof SUPPORTED_DEEPSEEK_MODELS, config?: z.infer<typeof ChatCompletionCommonConfigSchema>): ModelReference<typeof ChatCompletionCommonConfigSchema>;
+    model(name: string, config?: any): ModelReference<z.ZodTypeAny>;
+};
+/**
+ * This module provides an interface to the DeepSeek models through the Genkit
+ * plugin system. It allows users to interact with various models by providing
+ * an API key and optional configuration.
+ *
+ * The main export is the `deepseek` plugin, which can be configured with an API
+ * key either directly or through environment variables. It initializes the
+ * OpenAI client and makes available the models for use.
+ *
+ * Exports:
+ * - deepSeek: The main plugin function to interact with DeepSeek, via OpenAI
+ *   compatible API.
+ *
+ * Usage: To use the models, initialize the deepseek plugin inside
+ * `configureGenkit` and pass the configuration options. If no API key is
+ * provided in the options, the environment variable `OPENAI_API_KEY` must be
+ * set.
+ *
+ * Example:
+ * ```
+ * import { deepSeek } from '@genkit-ai/compat-oai/deepseek';
+ *
+ * export default configureGenkit({
+ *  plugins: [
+ *    deepSeek()
+ *    ... // other plugins
+ *  ]
+ * });
+ * ```
+ */
+declare const deepSeek: DeepSeekPlugin;
+
+export { type DeepSeekPlugin, type DeepSeekPluginOptions, deepSeek, deepSeekPlugin, deepSeek as default };

package/lib/deepseek/index.js

@@ -0,0 +1,93 @@
+"use strict";
+var __create = Object.create;
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __getProtoOf = Object.getPrototypeOf;
+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 __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(
+  // If the importer is in node compatibility mode or this is not an ESM
+  // file that has been converted to a CommonJS file using a Babel-
+  // compatible transform (i.e. "__esModule" has not been set), then set
+  // "default" to the CommonJS "module.exports" for node compatibility.
+  isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target,
+  mod
+));
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+var deepseek_exports = {};
+__export(deepseek_exports, {
+  deepSeek: () => deepSeek,
+  deepSeekPlugin: () => deepSeekPlugin,
+  default: () => deepseek_default
+});
+module.exports = __toCommonJS(deepseek_exports);
+var import_genkit = require("genkit");
+var import_logging = require("genkit/logging");
+var import__ = __toESM(require("../index.js"));
+var import_model = require("../model.js");
+var import_deepseek = require("./deepseek.js");
+const resolver = async (ai, client, actionType, actionName) => {
+  if (actionType === "model") {
+    (0, import_model.defineCompatOpenAIModel)({
+      ai,
+      name: `deepseek/${actionName}`,
+      client
+    });
+  } else {
+    import_logging.logger.warn("Only model actions are supported by the DeepSeek plugin");
+  }
+};
+const listActions = async (client) => {
+  return await client.models.list().then(
+    (response) => response.data.filter((model2) => model2.object === "model").map((model2) => {
+      return (0, import_genkit.modelActionMetadata)({
+        name: `deepseek/${model2.id}`,
+        configSchema: import_model.ChatCompletionCommonConfigSchema,
+        info: import_deepseek.SUPPORTED_DEEPSEEK_MODELS[model2.id]?.info
+      });
+    })
+  );
+};
+function deepSeekPlugin(options) {
+  return (0, import__.default)({
+    name: "deepseek",
+    baseURL: "https://api.deepseek.com",
+    ...options,
+    initializer: async (ai, client) => {
+      Object.values(import_deepseek.SUPPORTED_DEEPSEEK_MODELS).forEach(
+        (modelRef2) => (0, import_model.defineCompatOpenAIModel)({ ai, name: modelRef2.name, client, modelRef: modelRef2 })
+      );
+    },
+    resolver,
+    listActions
+  });
+}
+const model = (name, config) => {
+  return (0, import_genkit.modelRef)({
+    name: `deepseek/${name}`,
+    config,
+    configSchema: import_model.ChatCompletionCommonConfigSchema
+  });
+};
+const deepSeek = Object.assign(deepSeekPlugin, {
+  model
+});
+var deepseek_default = deepSeek;
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  deepSeek,
+  deepSeekPlugin
+});
+//# sourceMappingURL=index.js.map

package/lib/deepseek/index.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/deepseek/index.ts"],"sourcesContent":["/**\n * Copyright 2024 Google LLC\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport {\n  ActionMetadata,\n  Genkit,\n  modelActionMetadata,\n  modelRef,\n  ModelReference,\n  z,\n} from 'genkit';\nimport { logger } from 'genkit/logging';\nimport { GenkitPlugin } from 'genkit/plugin';\nimport { ActionType } from 'genkit/registry';\nimport OpenAI from 'openai';\nimport openAICompatible, { PluginOptions } from '../index.js';\nimport {\n  ChatCompletionCommonConfigSchema,\n  defineCompatOpenAIModel,\n} from '../model.js';\nimport { SUPPORTED_DEEPSEEK_MODELS } from './deepseek.js';\n\nexport type DeepSeekPluginOptions = Omit<PluginOptions, 'name' | 'baseURL'>;\n\nconst resolver = async (\n  ai: Genkit,\n  client: OpenAI,\n  actionType: ActionType,\n  actionName: string\n) => {\n  if (actionType === 'model') {\n    defineCompatOpenAIModel({\n      ai,\n      name: `deepseek/${actionName}`,\n      client,\n    });\n  } else {\n    logger.warn('Only model actions are supported by the DeepSeek plugin');\n  }\n};\n\nconst listActions = async (client: OpenAI): Promise<ActionMetadata[]> => {\n  return await client.models.list().then((response) =>\n    response.data\n      .filter((model) => model.object === 'model')\n      .map((model: OpenAI.Model) => {\n        return modelActionMetadata({\n          name: `deepseek/${model.id}`,\n          configSchema: ChatCompletionCommonConfigSchema,\n          info: SUPPORTED_DEEPSEEK_MODELS[model.id]?.info,\n        });\n      })\n  );\n};\n\nexport function deepSeekPlugin(options?: DeepSeekPluginOptions): GenkitPlugin {\n  return openAICompatible({\n    name: 'deepseek',\n    baseURL: 'https://api.deepseek.com',\n    ...options,\n    initializer: async (ai, client) => {\n      Object.values(SUPPORTED_DEEPSEEK_MODELS).forEach((modelRef) =>\n        defineCompatOpenAIModel({ ai, name: modelRef.name, client, modelRef })\n      );\n    },\n    resolver,\n    listActions,\n  });\n}\n\nexport type DeepSeekPlugin = {\n  (params?: DeepSeekPluginOptions): GenkitPlugin;\n  model(\n    name: keyof typeof SUPPORTED_DEEPSEEK_MODELS,\n    config?: z.infer<typeof ChatCompletionCommonConfigSchema>\n  ): ModelReference<typeof ChatCompletionCommonConfigSchema>;\n  model(name: string, config?: any): ModelReference<z.ZodTypeAny>;\n};\n\nconst model = ((name: string, config?: any): ModelReference<z.ZodTypeAny> => {\n  return modelRef({\n    name: `deepseek/${name}`,\n    config,\n    configSchema: ChatCompletionCommonConfigSchema,\n  });\n}) as DeepSeekPlugin['model'];\n\n/**\n * This module provides an interface to the DeepSeek models through the Genkit\n * plugin system. It allows users to interact with various models by providing\n * an API key and optional configuration.\n *\n * The main export is the `deepseek` plugin, which can be configured with an API\n * key either directly or through environment variables. It initializes the\n * OpenAI client and makes available the models for use.\n *\n * Exports:\n * - deepSeek: The main plugin function to interact with DeepSeek, via OpenAI\n *   compatible API.\n *\n * Usage: To use the models, initialize the deepseek plugin inside\n * `configureGenkit` and pass the configuration options. If no API key is\n * provided in the options, the environment variable `OPENAI_API_KEY` must be\n * set.\n *\n * Example:\n * ```\n * import { deepSeek } from '@genkit-ai/compat-oai/deepseek';\n *\n * export default configureGenkit({\n *  plugins: [\n *    deepSeek()\n *    ... // other plugins\n *  ]\n * });\n * ```\n */\nexport const deepSeek: DeepSeekPlugin = Object.assign(deepSeekPlugin, {\n  model,\n});\n\nexport default deepSeek;\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAgBA,oBAOO;AACP,qBAAuB;AAIvB,eAAgD;AAChD,mBAGO;AACP,sBAA0C;AAI1C,MAAM,WAAW,OACf,IACA,QACA,YACA,eACG;AACH,MAAI,eAAe,SAAS;AAC1B,8CAAwB;AAAA,MACtB;AAAA,MACA,MAAM,YAAY,UAAU;AAAA,MAC5B;AAAA,IACF,CAAC;AAAA,EACH,OAAO;AACL,0BAAO,KAAK,yDAAyD;AAAA,EACvE;AACF;AAEA,MAAM,cAAc,OAAO,WAA8C;AACvE,SAAO,MAAM,OAAO,OAAO,KAAK,EAAE;AAAA,IAAK,CAAC,aACtC,SAAS,KACN,OAAO,CAACA,WAAUA,OAAM,WAAW,OAAO,EAC1C,IAAI,CAACA,WAAwB;AAC5B,iBAAO,mCAAoB;AAAA,QACzB,MAAM,YAAYA,OAAM,EAAE;AAAA,QAC1B,cAAc;AAAA,QACd,MAAM,0CAA0BA,OAAM,EAAE,GAAG;AAAA,MAC7C,CAAC;AAAA,IACH,CAAC;AAAA,EACL;AACF;AAEO,SAAS,eAAe,SAA+C;AAC5E,aAAO,SAAAC,SAAiB;AAAA,IACtB,MAAM;AAAA,IACN,SAAS;AAAA,IACT,GAAG;AAAA,IACH,aAAa,OAAO,IAAI,WAAW;AACjC,aAAO,OAAO,yCAAyB,EAAE;AAAA,QAAQ,CAACC,kBAChD,sCAAwB,EAAE,IAAI,MAAMA,UAAS,MAAM,QAAQ,UAAAA,UAAS,CAAC;AAAA,MACvE;AAAA,IACF;AAAA,IACA;AAAA,IACA;AAAA,EACF,CAAC;AACH;AAWA,MAAM,QAAS,CAAC,MAAc,WAA+C;AAC3E,aAAO,wBAAS;AAAA,IACd,MAAM,YAAY,IAAI;AAAA,IACtB;AAAA,IACA,cAAc;AAAA,EAChB,CAAC;AACH;AAgCO,MAAM,WAA2B,OAAO,OAAO,gBAAgB;AAAA,EACpE;AACF,CAAC;AAED,IAAO,mBAAQ;","names":["model","openAICompatible","modelRef"]}

package/lib/deepseek/index.mjs

@@ -0,0 +1,64 @@
+import {
+  modelActionMetadata,
+  modelRef
+} from "genkit";
+import { logger } from "genkit/logging";
+import openAICompatible from "../index.js";
+import {
+  ChatCompletionCommonConfigSchema,
+  defineCompatOpenAIModel
+} from "../model.js";
+import { SUPPORTED_DEEPSEEK_MODELS } from "./deepseek.js";
+const resolver = async (ai, client, actionType, actionName) => {
+  if (actionType === "model") {
+    defineCompatOpenAIModel({
+      ai,
+      name: `deepseek/${actionName}`,
+      client
+    });
+  } else {
+    logger.warn("Only model actions are supported by the DeepSeek plugin");
+  }
+};
+const listActions = async (client) => {
+  return await client.models.list().then(
+    (response) => response.data.filter((model2) => model2.object === "model").map((model2) => {
+      return modelActionMetadata({
+        name: `deepseek/${model2.id}`,
+        configSchema: ChatCompletionCommonConfigSchema,
+        info: SUPPORTED_DEEPSEEK_MODELS[model2.id]?.info
+      });
+    })
+  );
+};
+function deepSeekPlugin(options) {
+  return openAICompatible({
+    name: "deepseek",
+    baseURL: "https://api.deepseek.com",
+    ...options,
+    initializer: async (ai, client) => {
+      Object.values(SUPPORTED_DEEPSEEK_MODELS).forEach(
+        (modelRef2) => defineCompatOpenAIModel({ ai, name: modelRef2.name, client, modelRef: modelRef2 })
+      );
+    },
+    resolver,
+    listActions
+  });
+}
+const model = (name, config) => {
+  return modelRef({
+    name: `deepseek/${name}`,
+    config,
+    configSchema: ChatCompletionCommonConfigSchema
+  });
+};
+const deepSeek = Object.assign(deepSeekPlugin, {
+  model
+});
+var deepseek_default = deepSeek;
+export {
+  deepSeek,
+  deepSeekPlugin,
+  deepseek_default as default
+};
+//# sourceMappingURL=index.mjs.map

package/lib/deepseek/index.mjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/deepseek/index.ts"],"sourcesContent":["/**\n * Copyright 2024 Google LLC\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport {\n  ActionMetadata,\n  Genkit,\n  modelActionMetadata,\n  modelRef,\n  ModelReference,\n  z,\n} from 'genkit';\nimport { logger } from 'genkit/logging';\nimport { GenkitPlugin } from 'genkit/plugin';\nimport { ActionType } from 'genkit/registry';\nimport OpenAI from 'openai';\nimport openAICompatible, { PluginOptions } from '../index.js';\nimport {\n  ChatCompletionCommonConfigSchema,\n  defineCompatOpenAIModel,\n} from '../model.js';\nimport { SUPPORTED_DEEPSEEK_MODELS } from './deepseek.js';\n\nexport type DeepSeekPluginOptions = Omit<PluginOptions, 'name' | 'baseURL'>;\n\nconst resolver = async (\n  ai: Genkit,\n  client: OpenAI,\n  actionType: ActionType,\n  actionName: string\n) => {\n  if (actionType === 'model') {\n    defineCompatOpenAIModel({\n      ai,\n      name: `deepseek/${actionName}`,\n      client,\n    });\n  } else {\n    logger.warn('Only model actions are supported by the DeepSeek plugin');\n  }\n};\n\nconst listActions = async (client: OpenAI): Promise<ActionMetadata[]> => {\n  return await client.models.list().then((response) =>\n    response.data\n      .filter((model) => model.object === 'model')\n      .map((model: OpenAI.Model) => {\n        return modelActionMetadata({\n          name: `deepseek/${model.id}`,\n          configSchema: ChatCompletionCommonConfigSchema,\n          info: SUPPORTED_DEEPSEEK_MODELS[model.id]?.info,\n        });\n      })\n  );\n};\n\nexport function deepSeekPlugin(options?: DeepSeekPluginOptions): GenkitPlugin {\n  return openAICompatible({\n    name: 'deepseek',\n    baseURL: 'https://api.deepseek.com',\n    ...options,\n    initializer: async (ai, client) => {\n      Object.values(SUPPORTED_DEEPSEEK_MODELS).forEach((modelRef) =>\n        defineCompatOpenAIModel({ ai, name: modelRef.name, client, modelRef })\n      );\n    },\n    resolver,\n    listActions,\n  });\n}\n\nexport type DeepSeekPlugin = {\n  (params?: DeepSeekPluginOptions): GenkitPlugin;\n  model(\n    name: keyof typeof SUPPORTED_DEEPSEEK_MODELS,\n    config?: z.infer<typeof ChatCompletionCommonConfigSchema>\n  ): ModelReference<typeof ChatCompletionCommonConfigSchema>;\n  model(name: string, config?: any): ModelReference<z.ZodTypeAny>;\n};\n\nconst model = ((name: string, config?: any): ModelReference<z.ZodTypeAny> => {\n  return modelRef({\n    name: `deepseek/${name}`,\n    config,\n    configSchema: ChatCompletionCommonConfigSchema,\n  });\n}) as DeepSeekPlugin['model'];\n\n/**\n * This module provides an interface to the DeepSeek models through the Genkit\n * plugin system. It allows users to interact with various models by providing\n * an API key and optional configuration.\n *\n * The main export is the `deepseek` plugin, which can be configured with an API\n * key either directly or through environment variables. It initializes the\n * OpenAI client and makes available the models for use.\n *\n * Exports:\n * - deepSeek: The main plugin function to interact with DeepSeek, via OpenAI\n *   compatible API.\n *\n * Usage: To use the models, initialize the deepseek plugin inside\n * `configureGenkit` and pass the configuration options. If no API key is\n * provided in the options, the environment variable `OPENAI_API_KEY` must be\n * set.\n *\n * Example:\n * ```\n * import { deepSeek } from '@genkit-ai/compat-oai/deepseek';\n *\n * export default configureGenkit({\n *  plugins: [\n *    deepSeek()\n *    ... // other plugins\n *  ]\n * });\n * ```\n */\nexport const deepSeek: DeepSeekPlugin = Object.assign(deepSeekPlugin, {\n  model,\n});\n\nexport default deepSeek;\n"],"mappings":"AAgBA;AAAA,EAGE;AAAA,EACA;AAAA,OAGK;AACP,SAAS,cAAc;AAIvB,OAAO,sBAAyC;AAChD;AAAA,EACE;AAAA,EACA;AAAA,OACK;AACP,SAAS,iCAAiC;AAI1C,MAAM,WAAW,OACf,IACA,QACA,YACA,eACG;AACH,MAAI,eAAe,SAAS;AAC1B,4BAAwB;AAAA,MACtB;AAAA,MACA,MAAM,YAAY,UAAU;AAAA,MAC5B;AAAA,IACF,CAAC;AAAA,EACH,OAAO;AACL,WAAO,KAAK,yDAAyD;AAAA,EACvE;AACF;AAEA,MAAM,cAAc,OAAO,WAA8C;AACvE,SAAO,MAAM,OAAO,OAAO,KAAK,EAAE;AAAA,IAAK,CAAC,aACtC,SAAS,KACN,OAAO,CAACA,WAAUA,OAAM,WAAW,OAAO,EAC1C,IAAI,CAACA,WAAwB;AAC5B,aAAO,oBAAoB;AAAA,QACzB,MAAM,YAAYA,OAAM,EAAE;AAAA,QAC1B,cAAc;AAAA,QACd,MAAM,0BAA0BA,OAAM,EAAE,GAAG;AAAA,MAC7C,CAAC;AAAA,IACH,CAAC;AAAA,EACL;AACF;AAEO,SAAS,eAAe,SAA+C;AAC5E,SAAO,iBAAiB;AAAA,IACtB,MAAM;AAAA,IACN,SAAS;AAAA,IACT,GAAG;AAAA,IACH,aAAa,OAAO,IAAI,WAAW;AACjC,aAAO,OAAO,yBAAyB,EAAE;AAAA,QAAQ,CAACC,cAChD,wBAAwB,EAAE,IAAI,MAAMA,UAAS,MAAM,QAAQ,UAAAA,UAAS,CAAC;AAAA,MACvE;AAAA,IACF;AAAA,IACA;AAAA,IACA;AAAA,EACF,CAAC;AACH;AAWA,MAAM,QAAS,CAAC,MAAc,WAA+C;AAC3E,SAAO,SAAS;AAAA,IACd,MAAM,YAAY,IAAI;AAAA,IACtB;AAAA,IACA,cAAc;AAAA,EAChB,CAAC;AACH;AAgCO,MAAM,WAA2B,OAAO,OAAO,gBAAgB;AAAA,EACpE;AACF,CAAC;AAED,IAAO,mBAAQ;","names":["model","modelRef"]}

package/lib/embedder.d.mts

@@ -0,0 +1,41 @@
+import { Genkit, EmbedderReference, EmbedderAction } from 'genkit';
+import OpenAI from 'openai';
+
+/**
+ * Copyright 2024 The Fire Company
+ * Copyright 2024 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+/**
+ * Method to define a new Genkit Embedder that is compatibale with the Open AI
+ * Embeddings API.
+ *
+ * @param params An object containing parameters for defining the OpenAI embedder.
+ * @param params.ai The Genkit AI instance.
+ * @param params.name The name of the embedder.
+ * @param params.client The OpenAI client instance.
+ * @param params.embedderRef Optional reference to the embedder's configuration and
+ * custom options.
+
+ * @returns the created {@link EmbedderAction}
+ */
+declare function defineCompatOpenAIEmbedder(params: {
+    ai: Genkit;
+    name: string;
+    client: OpenAI;
+    embedderRef?: EmbedderReference;
+}): EmbedderAction;
+
+export { defineCompatOpenAIEmbedder };

package/lib/embedder.d.ts

@@ -0,0 +1,41 @@
+import { Genkit, EmbedderReference, EmbedderAction } from 'genkit';
+import OpenAI from 'openai';
+
+/**
+ * Copyright 2024 The Fire Company
+ * Copyright 2024 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+/**
+ * Method to define a new Genkit Embedder that is compatibale with the Open AI
+ * Embeddings API.
+ *
+ * @param params An object containing parameters for defining the OpenAI embedder.
+ * @param params.ai The Genkit AI instance.
+ * @param params.name The name of the embedder.
+ * @param params.client The OpenAI client instance.
+ * @param params.embedderRef Optional reference to the embedder's configuration and
+ * custom options.
+
+ * @returns the created {@link EmbedderAction}
+ */
+declare function defineCompatOpenAIEmbedder(params: {
+    ai: Genkit;
+    name: string;
+    client: OpenAI;
+    embedderRef?: EmbedderReference;
+}): EmbedderAction;
+
+export { defineCompatOpenAIEmbedder };

package/lib/embedder.js

@@ -0,0 +1,51 @@
+"use strict";
+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 embedder_exports = {};
+__export(embedder_exports, {
+  defineCompatOpenAIEmbedder: () => defineCompatOpenAIEmbedder
+});
+module.exports = __toCommonJS(embedder_exports);
+function defineCompatOpenAIEmbedder(params) {
+  const { ai, name, client, embedderRef } = params;
+  const model = name.split("/").pop();
+  return ai.defineEmbedder(
+    {
+      name,
+      configSchema: embedderRef?.configSchema,
+      ...embedderRef?.info
+    },
+    async (input, options) => {
+      const { encodingFormat: encoding_format, ...restOfConfig } = options;
+      const embeddings = await client.embeddings.create({
+        model,
+        input: input.map((d) => d.text),
+        encoding_format,
+        ...restOfConfig
+      });
+      return {
+        embeddings: embeddings.data.map((d) => ({ embedding: d.embedding }))
+      };
+    }
+  );
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  defineCompatOpenAIEmbedder
+});
+//# sourceMappingURL=embedder.js.map

package/lib/embedder.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/embedder.ts"],"sourcesContent":["/**\n * Copyright 2024 The Fire Company\n * Copyright 2024 Google LLC\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\n// import { defineEmbedder, embedderRef } from '@genkit-ai/ai/embedder';\n\nimport type { EmbedderAction, EmbedderReference, Genkit } from 'genkit';\nimport OpenAI from 'openai';\n\n/**\n * Method to define a new Genkit Embedder that is compatibale with the Open AI\n * Embeddings API. \n *\n * @param params An object containing parameters for defining the OpenAI embedder.\n * @param params.ai The Genkit AI instance.\n * @param params.name The name of the embedder.\n * @param params.client The OpenAI client instance.\n * @param params.embedderRef Optional reference to the embedder's configuration and\n * custom options.\n\n * @returns the created {@link EmbedderAction}\n */\nexport function defineCompatOpenAIEmbedder(params: {\n  ai: Genkit;\n  name: string;\n  client: OpenAI;\n  embedderRef?: EmbedderReference;\n}): EmbedderAction {\n  const { ai, name, client, embedderRef } = params;\n  const model = name.split('/').pop();\n  return ai.defineEmbedder(\n    {\n      name,\n      configSchema: embedderRef?.configSchema,\n      ...embedderRef?.info,\n    },\n    async (input, options) => {\n      const { encodingFormat: encoding_format, ...restOfConfig } = options;\n      const embeddings = await client.embeddings.create({\n        model: model!,\n        input: input.map((d) => d.text),\n        encoding_format,\n        ...restOfConfig,\n      });\n      return {\n        embeddings: embeddings.data.map((d) => ({ embedding: d.embedding })),\n      };\n    }\n  );\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAmCO,SAAS,2BAA2B,QAKxB;AACjB,QAAM,EAAE,IAAI,MAAM,QAAQ,YAAY,IAAI;AAC1C,QAAM,QAAQ,KAAK,MAAM,GAAG,EAAE,IAAI;AAClC,SAAO,GAAG;AAAA,IACR;AAAA,MACE;AAAA,MACA,cAAc,aAAa;AAAA,MAC3B,GAAG,aAAa;AAAA,IAClB;AAAA,IACA,OAAO,OAAO,YAAY;AACxB,YAAM,EAAE,gBAAgB,iBAAiB,GAAG,aAAa,IAAI;AAC7D,YAAM,aAAa,MAAM,OAAO,WAAW,OAAO;AAAA,QAChD;AAAA,QACA,OAAO,MAAM,IAAI,CAAC,MAAM,EAAE,IAAI;AAAA,QAC9B;AAAA,QACA,GAAG;AAAA,MACL,CAAC;AACD,aAAO;AAAA,QACL,YAAY,WAAW,KAAK,IAAI,CAAC,OAAO,EAAE,WAAW,EAAE,UAAU,EAAE;AAAA,MACrE;AAAA,IACF;AAAA,EACF;AACF;","names":[]}

package/lib/embedder.mjs

@@ -0,0 +1,27 @@
+function defineCompatOpenAIEmbedder(params) {
+  const { ai, name, client, embedderRef } = params;
+  const model = name.split("/").pop();
+  return ai.defineEmbedder(
+    {
+      name,
+      configSchema: embedderRef?.configSchema,
+      ...embedderRef?.info
+    },
+    async (input, options) => {
+      const { encodingFormat: encoding_format, ...restOfConfig } = options;
+      const embeddings = await client.embeddings.create({
+        model,
+        input: input.map((d) => d.text),
+        encoding_format,
+        ...restOfConfig
+      });
+      return {
+        embeddings: embeddings.data.map((d) => ({ embedding: d.embedding }))
+      };
+    }
+  );
+}
+export {
+  defineCompatOpenAIEmbedder
+};
+//# sourceMappingURL=embedder.mjs.map

package/lib/embedder.mjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/embedder.ts"],"sourcesContent":["/**\n * Copyright 2024 The Fire Company\n * Copyright 2024 Google LLC\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\n// import { defineEmbedder, embedderRef } from '@genkit-ai/ai/embedder';\n\nimport type { EmbedderAction, EmbedderReference, Genkit } from 'genkit';\nimport OpenAI from 'openai';\n\n/**\n * Method to define a new Genkit Embedder that is compatibale with the Open AI\n * Embeddings API. \n *\n * @param params An object containing parameters for defining the OpenAI embedder.\n * @param params.ai The Genkit AI instance.\n * @param params.name The name of the embedder.\n * @param params.client The OpenAI client instance.\n * @param params.embedderRef Optional reference to the embedder's configuration and\n * custom options.\n\n * @returns the created {@link EmbedderAction}\n */\nexport function defineCompatOpenAIEmbedder(params: {\n  ai: Genkit;\n  name: string;\n  client: OpenAI;\n  embedderRef?: EmbedderReference;\n}): EmbedderAction {\n  const { ai, name, client, embedderRef } = params;\n  const model = name.split('/').pop();\n  return ai.defineEmbedder(\n    {\n      name,\n      configSchema: embedderRef?.configSchema,\n      ...embedderRef?.info,\n    },\n    async (input, options) => {\n      const { encodingFormat: encoding_format, ...restOfConfig } = options;\n      const embeddings = await client.embeddings.create({\n        model: model!,\n        input: input.map((d) => d.text),\n        encoding_format,\n        ...restOfConfig,\n      });\n      return {\n        embeddings: embeddings.data.map((d) => ({ embedding: d.embedding })),\n      };\n    }\n  );\n}\n"],"mappings":"AAmCO,SAAS,2BAA2B,QAKxB;AACjB,QAAM,EAAE,IAAI,MAAM,QAAQ,YAAY,IAAI;AAC1C,QAAM,QAAQ,KAAK,MAAM,GAAG,EAAE,IAAI;AAClC,SAAO,GAAG;AAAA,IACR;AAAA,MACE;AAAA,MACA,cAAc,aAAa;AAAA,MAC3B,GAAG,aAAa;AAAA,IAClB;AAAA,IACA,OAAO,OAAO,YAAY;AACxB,YAAM,EAAE,gBAAgB,iBAAiB,GAAG,aAAa,IAAI;AAC7D,YAAM,aAAa,MAAM,OAAO,WAAW,OAAO;AAAA,QAChD;AAAA,QACA,OAAO,MAAM,IAAI,CAAC,MAAM,EAAE,IAAI;AAAA,QAC9B;AAAA,QACA,GAAG;AAAA,MACL,CAAC;AACD,aAAO;AAAA,QACL,YAAY,WAAW,KAAK,IAAI,CAAC,OAAO,EAAE,WAAW,EAAE,UAAU,EAAE;AAAA,MACrE;AAAA,IACF;AAAA,EACF;AACF;","names":[]}

package/lib/image.d.mts

@@ -0,0 +1,68 @@
+import { z, Genkit, ModelReference } from 'genkit';
+import { ModelInfo, ModelAction } from 'genkit/model';
+import OpenAI from 'openai';
+
+/**
+ * Copyright 2024 The Fire Company
+ * Copyright 2024 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+declare const IMAGE_GENERATION_MODEL_INFO: ModelInfo;
+declare const ImageGenerationCommonConfigSchema: z.ZodObject<{
+    size: z.ZodOptional<z.ZodEnum<["1024x1024", "1792x1024", "1024x1792"]>>;
+    style: z.ZodOptional<z.ZodEnum<["vivid", "natural"]>>;
+    user: z.ZodOptional<z.ZodString>;
+    n: z.ZodDefault<z.ZodNumber>;
+    quality: z.ZodOptional<z.ZodEnum<["standard", "hd"]>>;
+    response_format: z.ZodOptional<z.ZodEnum<["b64_json", "url"]>>;
+}, "strip", z.ZodTypeAny, {
+    n: number;
+    size?: "1024x1024" | "1792x1024" | "1024x1792" | undefined;
+    style?: "vivid" | "natural" | undefined;
+    user?: string | undefined;
+    quality?: "standard" | "hd" | undefined;
+    response_format?: "b64_json" | "url" | undefined;
+}, {
+    size?: "1024x1024" | "1792x1024" | "1024x1792" | undefined;
+    style?: "vivid" | "natural" | undefined;
+    user?: string | undefined;
+    n?: number | undefined;
+    quality?: "standard" | "hd" | undefined;
+    response_format?: "b64_json" | "url" | undefined;
+}>;
+/**
+ * Method to define a new Genkit Model that is compatible with Open AI
+ * Images API.
+ *
+ * These models are to be used to create images from a user prompt.
+ *
+ * @param params An object containing parameters for defining the OpenAI
+ * image model.
+ * @param params.ai The Genkit AI instance.
+ * @param params.name The name of the model.
+ * @param params.client The OpenAI client instance.
+ * @param params.modelRef Optional reference to the model's configuration and
+ * custom options.
+
+ * @returns the created {@link ModelAction}
+ */
+declare function defineCompatOpenAIImageModel<CustomOptions extends z.ZodTypeAny = z.ZodTypeAny>(params: {
+    ai: Genkit;
+    name: string;
+    client: OpenAI;
+    modelRef?: ModelReference<CustomOptions>;
+}): ModelAction<CustomOptions>;
+
+export { IMAGE_GENERATION_MODEL_INFO, ImageGenerationCommonConfigSchema, defineCompatOpenAIImageModel };