@genkit-ai/compat-oai 1.30.1 → 1.31.0

package/lib/{audio-Cr0WtG4I.d.mts → audio-CW-qdV9D.d.mts}

@@ -1,4 +1,4 @@
-import { EmbedderReference, EmbedderAction, GenerateRequest, z, ModelReference, StreamingCallback, GenerateResponseChunkData, GenerateResponseData, Role, Part, MessageData, ToolRequestPart, ActionMetadata } from 'genkit';
+import { EmbedderReference, EmbedderAction, z, GenerateRequest, ModelReference, StreamingCallback, GenerateResponseChunkData, GenerateResponseData, ToolRequestPart, MessageData, Role, Part, ActionMetadata } from 'genkit';
 import { ModelInfo, ModelAction, ToolDefinition } from 'genkit/model';
 import OpenAI, { ClientOptions } from 'openai';
 import { Response } from 'openai/core.mjs';
@@ -7,7 +7,7 @@ import * as genkit_plugin from 'genkit/plugin';
 import { ResolvableAction } from 'genkit/plugin';
 import { ActionType } from 'genkit/registry';
 import { ImageGenerateParams } from 'openai/resources/images.mjs';
-import { ChatCompletionCreateParams, ChatCompletionRole, ChatCompletionTool, ChatCompletionContentPart, ChatCompletionMessageParam, ChatCompletionMessageToolCall, ChatCompletionChunk, ChatCompletion } from 'openai/resources/index.mjs';
+import { ChatCompletionCreateParams, ChatCompletion, ChatCompletionChunk, ChatCompletionMessageToolCall, ChatCompletionMessageParam, ChatCompletionRole, ChatCompletionContentPart, ChatCompletionTool } from 'openai/resources/index.mjs';
 
 /**
  * Copyright 2024 The Fire Company
@@ -622,4 +622,4 @@ declare function compatOaiTranscriptionModelRef<CustomOptions extends z.ZodTypeA
     namespace?: string;
 }): ModelReference<any>;
 
-export { defineCompatOpenAISpeechModel as A, defineCompatOpenAITranscriptionModel as B, ChatCompletionCommonConfigSchema as C, type SpeechRequestBuilder as D, type TranscriptionRequestBuilder as E, TRANSCRIPTION_MODEL_INFO as F, SPEECH_MODEL_INFO as G, toTTSRequest as H, ImageGenerationCommonConfigSchema as I, speechToGenerateResponse as J, toSttRequest as K, transcriptionToGenerateResponse as L, type ModelRequestBuilder as M, type PluginOptions as P, RESPONSE_FORMAT_MEDIA_TYPES as R, SpeechConfigSchema as S, TranscriptionConfigSchema as T, type ImageRequestBuilder as a, IMAGE_GENERATION_MODEL_INFO as b, defineCompatOpenAIImageModel as c, defineCompatOpenAIEmbedder as d, compatOaiImageModelRef as e, toOpenAITool as f, toOpenAITextAndMedia as g, toOpenAIMessages as h, fromOpenAIToolCall as i, fromOpenAIChoice as j, fromOpenAIChunkChoice as k, toOpenAIRequestBody as l, defineCompatOpenAIModel as m, compatOaiModelRef as n, openAIModelRunner as o, type TranslationRequestBuilder as p, TRANSLATION_MODEL_INFO as q, TranslationConfigSchema as r, toTranslationRequest as s, toOpenAIRole as t, translationToGenerateResponse as u, defineCompatOpenAITranslationModel as v, compatOaiTranslationModelRef as w, openAICompatible as x, compatOaiSpeechModelRef as y, compatOaiTranscriptionModelRef as z };
+export { compatOaiTranscriptionModelRef as A, openAICompatible as B, ChatCompletionCommonConfigSchema as C, defineCompatOpenAISpeechModel as D, defineCompatOpenAITranscriptionModel as E, SPEECH_MODEL_INFO as F, TRANSCRIPTION_MODEL_INFO as G, speechToGenerateResponse as H, ImageGenerationCommonConfigSchema as I, toSttRequest as J, toTTSRequest as K, transcriptionToGenerateResponse as L, type ModelRequestBuilder as M, type PluginOptions as P, RESPONSE_FORMAT_MEDIA_TYPES as R, SpeechConfigSchema as S, TranscriptionConfigSchema as T, type ImageRequestBuilder as a, IMAGE_GENERATION_MODEL_INFO as b, compatOaiImageModelRef as c, defineCompatOpenAIEmbedder as d, defineCompatOpenAIImageModel as e, compatOaiModelRef as f, defineCompatOpenAIModel as g, fromOpenAIChoice as h, fromOpenAIChunkChoice as i, fromOpenAIToolCall as j, toOpenAIRequestBody as k, toOpenAIRole as l, toOpenAITextAndMedia as m, toOpenAITool as n, openAIModelRunner as o, TRANSLATION_MODEL_INFO as p, TranslationConfigSchema as q, type TranslationRequestBuilder as r, compatOaiTranslationModelRef as s, toOpenAIMessages as t, defineCompatOpenAITranslationModel as u, toTranslationRequest as v, translationToGenerateResponse as w, type SpeechRequestBuilder as x, type TranscriptionRequestBuilder as y, compatOaiSpeechModelRef as z };

package/lib/{audio-Cr0WtG4I.d.ts → audio-CW-qdV9D.d.ts}

@@ -1,4 +1,4 @@
-import { EmbedderReference, EmbedderAction, GenerateRequest, z, ModelReference, StreamingCallback, GenerateResponseChunkData, GenerateResponseData, Role, Part, MessageData, ToolRequestPart, ActionMetadata } from 'genkit';
+import { EmbedderReference, EmbedderAction, z, GenerateRequest, ModelReference, StreamingCallback, GenerateResponseChunkData, GenerateResponseData, ToolRequestPart, MessageData, Role, Part, ActionMetadata } from 'genkit';
 import { ModelInfo, ModelAction, ToolDefinition } from 'genkit/model';
 import OpenAI, { ClientOptions } from 'openai';
 import { Response } from 'openai/core.mjs';
@@ -7,7 +7,7 @@ import * as genkit_plugin from 'genkit/plugin';
 import { ResolvableAction } from 'genkit/plugin';
 import { ActionType } from 'genkit/registry';
 import { ImageGenerateParams } from 'openai/resources/images.mjs';
-import { ChatCompletionCreateParams, ChatCompletionRole, ChatCompletionTool, ChatCompletionContentPart, ChatCompletionMessageParam, ChatCompletionMessageToolCall, ChatCompletionChunk, ChatCompletion } from 'openai/resources/index.mjs';
+import { ChatCompletionCreateParams, ChatCompletion, ChatCompletionChunk, ChatCompletionMessageToolCall, ChatCompletionMessageParam, ChatCompletionRole, ChatCompletionContentPart, ChatCompletionTool } from 'openai/resources/index.mjs';
 
 /**
  * Copyright 2024 The Fire Company
@@ -622,4 +622,4 @@ declare function compatOaiTranscriptionModelRef<CustomOptions extends z.ZodTypeA
     namespace?: string;
 }): ModelReference<any>;
 
-export { defineCompatOpenAISpeechModel as A, defineCompatOpenAITranscriptionModel as B, ChatCompletionCommonConfigSchema as C, type SpeechRequestBuilder as D, type TranscriptionRequestBuilder as E, TRANSCRIPTION_MODEL_INFO as F, SPEECH_MODEL_INFO as G, toTTSRequest as H, ImageGenerationCommonConfigSchema as I, speechToGenerateResponse as J, toSttRequest as K, transcriptionToGenerateResponse as L, type ModelRequestBuilder as M, type PluginOptions as P, RESPONSE_FORMAT_MEDIA_TYPES as R, SpeechConfigSchema as S, TranscriptionConfigSchema as T, type ImageRequestBuilder as a, IMAGE_GENERATION_MODEL_INFO as b, defineCompatOpenAIImageModel as c, defineCompatOpenAIEmbedder as d, compatOaiImageModelRef as e, toOpenAITool as f, toOpenAITextAndMedia as g, toOpenAIMessages as h, fromOpenAIToolCall as i, fromOpenAIChoice as j, fromOpenAIChunkChoice as k, toOpenAIRequestBody as l, defineCompatOpenAIModel as m, compatOaiModelRef as n, openAIModelRunner as o, type TranslationRequestBuilder as p, TRANSLATION_MODEL_INFO as q, TranslationConfigSchema as r, toTranslationRequest as s, toOpenAIRole as t, translationToGenerateResponse as u, defineCompatOpenAITranslationModel as v, compatOaiTranslationModelRef as w, openAICompatible as x, compatOaiSpeechModelRef as y, compatOaiTranscriptionModelRef as z };
+export { compatOaiTranscriptionModelRef as A, openAICompatible as B, ChatCompletionCommonConfigSchema as C, defineCompatOpenAISpeechModel as D, defineCompatOpenAITranscriptionModel as E, SPEECH_MODEL_INFO as F, TRANSCRIPTION_MODEL_INFO as G, speechToGenerateResponse as H, ImageGenerationCommonConfigSchema as I, toSttRequest as J, toTTSRequest as K, transcriptionToGenerateResponse as L, type ModelRequestBuilder as M, type PluginOptions as P, RESPONSE_FORMAT_MEDIA_TYPES as R, SpeechConfigSchema as S, TranscriptionConfigSchema as T, type ImageRequestBuilder as a, IMAGE_GENERATION_MODEL_INFO as b, compatOaiImageModelRef as c, defineCompatOpenAIEmbedder as d, defineCompatOpenAIImageModel as e, compatOaiModelRef as f, defineCompatOpenAIModel as g, fromOpenAIChoice as h, fromOpenAIChunkChoice as i, fromOpenAIToolCall as j, toOpenAIRequestBody as k, toOpenAIRole as l, toOpenAITextAndMedia as m, toOpenAITool as n, openAIModelRunner as o, TRANSLATION_MODEL_INFO as p, TranslationConfigSchema as q, type TranslationRequestBuilder as r, compatOaiTranslationModelRef as s, toOpenAIMessages as t, defineCompatOpenAITranslationModel as u, toTranslationRequest as v, translationToGenerateResponse as w, type SpeechRequestBuilder as x, type TranscriptionRequestBuilder as y, compatOaiSpeechModelRef as z };

package/lib/audio.d.mts

@@ -3,7 +3,7 @@ import 'genkit/model';
 import 'openai';
 import 'openai/core.mjs';
 import 'openai/resources/audio/index.mjs';
-export { R as RESPONSE_FORMAT_MEDIA_TYPES, G as SPEECH_MODEL_INFO, S as SpeechConfigSchema, D as SpeechRequestBuilder, F as TRANSCRIPTION_MODEL_INFO, T as TranscriptionConfigSchema, E as TranscriptionRequestBuilder, y as compatOaiSpeechModelRef, z as compatOaiTranscriptionModelRef, A as defineCompatOpenAISpeechModel, B as defineCompatOpenAITranscriptionModel, J as speechToGenerateResponse, K as toSttRequest, H as toTTSRequest, L as transcriptionToGenerateResponse } from './audio-Cr0WtG4I.mjs';
+export { R as RESPONSE_FORMAT_MEDIA_TYPES, F as SPEECH_MODEL_INFO, S as SpeechConfigSchema, x as SpeechRequestBuilder, G as TRANSCRIPTION_MODEL_INFO, T as TranscriptionConfigSchema, y as TranscriptionRequestBuilder, z as compatOaiSpeechModelRef, A as compatOaiTranscriptionModelRef, D as defineCompatOpenAISpeechModel, E as defineCompatOpenAITranscriptionModel, H as speechToGenerateResponse, J as toSttRequest, K as toTTSRequest, L as transcriptionToGenerateResponse } from './audio-CW-qdV9D.mjs';
 import 'genkit/plugin';
 import 'genkit/registry';
 import 'openai/resources/images.mjs';

package/lib/audio.d.ts

@@ -3,7 +3,7 @@ import 'genkit/model';
 import 'openai';
 import 'openai/core.mjs';
 import 'openai/resources/audio/index.mjs';
-export { R as RESPONSE_FORMAT_MEDIA_TYPES, G as SPEECH_MODEL_INFO, S as SpeechConfigSchema, D as SpeechRequestBuilder, F as TRANSCRIPTION_MODEL_INFO, T as TranscriptionConfigSchema, E as TranscriptionRequestBuilder, y as compatOaiSpeechModelRef, z as compatOaiTranscriptionModelRef, A as defineCompatOpenAISpeechModel, B as defineCompatOpenAITranscriptionModel, J as speechToGenerateResponse, K as toSttRequest, H as toTTSRequest, L as transcriptionToGenerateResponse } from './audio-Cr0WtG4I.js';
+export { R as RESPONSE_FORMAT_MEDIA_TYPES, F as SPEECH_MODEL_INFO, S as SpeechConfigSchema, x as SpeechRequestBuilder, G as TRANSCRIPTION_MODEL_INFO, T as TranscriptionConfigSchema, y as TranscriptionRequestBuilder, z as compatOaiSpeechModelRef, A as compatOaiTranscriptionModelRef, D as defineCompatOpenAISpeechModel, E as defineCompatOpenAITranscriptionModel, H as speechToGenerateResponse, J as toSttRequest, K as toTTSRequest, L as transcriptionToGenerateResponse } from './audio-CW-qdV9D.js';
 import 'genkit/plugin';
 import 'genkit/registry';
 import 'openai/resources/images.mjs';

package/lib/deepseek/deepseek.d.mts

@@ -1,6 +1,6 @@
 import { z } from 'genkit';
-import { ModelInfo, ModelReference } from 'genkit/model';
-import { M as ModelRequestBuilder } from '../audio-Cr0WtG4I.mjs';
+import { ModelReference, ModelInfo } from 'genkit/model';
+import { M as ModelRequestBuilder } from '../audio-CW-qdV9D.mjs';
 import 'openai';
 import 'openai/core.mjs';
 import 'openai/resources/audio/index.mjs';

package/lib/deepseek/deepseek.d.ts

@@ -1,6 +1,6 @@
 import { z } from 'genkit';
-import { ModelInfo, ModelReference } from 'genkit/model';
-import { M as ModelRequestBuilder } from '../audio-Cr0WtG4I.js';
+import { ModelReference, ModelInfo } from 'genkit/model';
+import { M as ModelRequestBuilder } from '../audio-CW-qdV9D.js';
 import 'openai';
 import 'openai/core.mjs';
 import 'openai/resources/audio/index.mjs';

package/lib/deepseek/index.d.mts

@@ -1,6 +1,6 @@
 import { z, ModelReference } from 'genkit';
 import { GenkitPluginV2 } from 'genkit/plugin';
-import { P as PluginOptions } from '../audio-Cr0WtG4I.mjs';
+import { P as PluginOptions } from '../audio-CW-qdV9D.mjs';
 import { SUPPORTED_DEEPSEEK_MODELS, DeepSeekChatCompletionConfigSchema } from './deepseek.mjs';
 import 'genkit/model';
 import 'openai';

package/lib/deepseek/index.d.ts

@@ -1,6 +1,6 @@
 import { z, ModelReference } from 'genkit';
 import { GenkitPluginV2 } from 'genkit/plugin';
-import { P as PluginOptions } from '../audio-Cr0WtG4I.js';
+import { P as PluginOptions } from '../audio-CW-qdV9D.js';
 import { SUPPORTED_DEEPSEEK_MODELS, DeepSeekChatCompletionConfigSchema } from './deepseek.js';
 import 'genkit/model';
 import 'openai';

package/lib/deepseek/index.js

@@ -90,12 +90,12 @@ function deepSeekPlugin(options) {
     listActions
   });
 }
-const model = (name, config) => {
+const model = ((name, config) => {
   return (0, import_deepseek.deepSeekModelRef)({
     name,
     config
   });
-};
+});
 const deepSeek = Object.assign(deepSeekPlugin, {
   model
 });

package/lib/deepseek/index.js.map

@@ -1 +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  GenkitError,\n  modelActionMetadata,\n  ModelReference,\n  z,\n} from 'genkit';\nimport { logger } from 'genkit/logging';\nimport { type GenkitPluginV2 } from 'genkit/plugin';\nimport { ActionType } from 'genkit/registry';\nimport OpenAI from 'openai';\nimport { openAICompatible, PluginOptions } from '../index.js';\nimport { defineCompatOpenAIModel } from '../model.js';\nimport {\n  DeepSeekChatCompletionConfigSchema,\n  deepSeekModelRef,\n  deepSeekRequestBuilder,\n  SUPPORTED_DEEPSEEK_MODELS,\n} from './deepseek.js';\n\nexport type DeepSeekPluginOptions = Omit<PluginOptions, 'name' | 'baseURL'>;\n\nfunction createResolver(pluginOptions: PluginOptions) {\n  return async (client: OpenAI, actionType: ActionType, actionName: string) => {\n    if (actionType === 'model') {\n      const modelRef = deepSeekModelRef({\n        name: actionName,\n      });\n      return defineCompatOpenAIModel({\n        name: modelRef.name,\n        client,\n        pluginOptions,\n        modelRef,\n        requestBuilder: deepSeekRequestBuilder,\n      });\n    } else {\n      logger.warn('Only model actions are supported by the DeepSeek plugin');\n      return undefined;\n    }\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        const modelRef =\n          SUPPORTED_DEEPSEEK_MODELS[model.id] ??\n          deepSeekModelRef({\n            name: model.id,\n          });\n        return modelActionMetadata({\n          name: modelRef.name,\n          info: modelRef.info,\n          configSchema: modelRef.configSchema,\n        });\n      })\n  );\n};\n\nexport function deepSeekPlugin(\n  options?: DeepSeekPluginOptions\n): GenkitPluginV2 {\n  const apiKey = options?.apiKey ?? process.env.DEEPSEEK_API_KEY;\n  if (!apiKey) {\n    throw new GenkitError({\n      status: 'FAILED_PRECONDITION',\n      message:\n        'Please pass in the API key or set the DEEPSEEK_API_KEY environment variable.',\n    });\n  }\n  const pluginOptions = { name: 'deepseek', ...options };\n  return openAICompatible({\n    name: 'deepseek',\n    baseURL: 'https://api.deepseek.com',\n    apiKey,\n    ...options,\n    initializer: async (client) => {\n      return Object.values(SUPPORTED_DEEPSEEK_MODELS).map((modelRef) =>\n        defineCompatOpenAIModel({\n          name: modelRef.name,\n          client,\n          pluginOptions,\n          modelRef,\n          requestBuilder: deepSeekRequestBuilder,\n        })\n      );\n    },\n    resolver: createResolver(pluginOptions),\n    listActions,\n  });\n}\n\nexport type DeepSeekPlugin = {\n  (params?: DeepSeekPluginOptions): GenkitPluginV2;\n  model(\n    name: keyof typeof SUPPORTED_DEEPSEEK_MODELS,\n    config?: z.infer<typeof DeepSeekChatCompletionConfigSchema>\n  ): ModelReference<typeof DeepSeekChatCompletionConfigSchema>;\n  model(name: string, config?: any): ModelReference<z.ZodTypeAny>;\n};\n\nconst model = ((name: string, config?: any): ModelReference<z.ZodTypeAny> => {\n  return deepSeekModelRef({\n    name,\n    config,\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,oBAMO;AACP,qBAAuB;AAIvB,eAAgD;AAChD,mBAAwC;AACxC,sBAKO;AAIP,SAAS,eAAe,eAA8B;AACpD,SAAO,OAAO,QAAgB,YAAwB,eAAuB;AAC3E,QAAI,eAAe,SAAS;AAC1B,YAAM,eAAW,kCAAiB;AAAA,QAChC,MAAM;AAAA,MACR,CAAC;AACD,iBAAO,sCAAwB;AAAA,QAC7B,MAAM,SAAS;AAAA,QACf;AAAA,QACA;AAAA,QACA;AAAA,QACA,gBAAgB;AAAA,MAClB,CAAC;AAAA,IACH,OAAO;AACL,4BAAO,KAAK,yDAAyD;AACrE,aAAO;AAAA,IACT;AAAA,EACF;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,YAAM,WACJ,0CAA0BA,OAAM,EAAE,SAClC,kCAAiB;AAAA,QACf,MAAMA,OAAM;AAAA,MACd,CAAC;AACH,iBAAO,mCAAoB;AAAA,QACzB,MAAM,SAAS;AAAA,QACf,MAAM,SAAS;AAAA,QACf,cAAc,SAAS;AAAA,MACzB,CAAC;AAAA,IACH,CAAC;AAAA,EACL;AACF;AAEO,SAAS,eACd,SACgB;AAChB,QAAM,SAAS,SAAS,UAAU,QAAQ,IAAI;AAC9C,MAAI,CAAC,QAAQ;AACX,UAAM,IAAI,0BAAY;AAAA,MACpB,QAAQ;AAAA,MACR,SACE;AAAA,IACJ,CAAC;AAAA,EACH;AACA,QAAM,gBAAgB,EAAE,MAAM,YAAY,GAAG,QAAQ;AACrD,aAAO,2BAAiB;AAAA,IACtB,MAAM;AAAA,IACN,SAAS;AAAA,IACT;AAAA,IACA,GAAG;AAAA,IACH,aAAa,OAAO,WAAW;AAC7B,aAAO,OAAO,OAAO,yCAAyB,EAAE;AAAA,QAAI,CAAC,iBACnD,sCAAwB;AAAA,UACtB,MAAM,SAAS;AAAA,UACf;AAAA,UACA;AAAA,UACA;AAAA,UACA,gBAAgB;AAAA,QAClB,CAAC;AAAA,MACH;AAAA,IACF;AAAA,IACA,UAAU,eAAe,aAAa;AAAA,IACtC;AAAA,EACF,CAAC;AACH;AAWA,MAAM,QAAS,CAAC,MAAc,WAA+C;AAC3E,aAAO,kCAAiB;AAAA,IACtB;AAAA,IACA;AAAA,EACF,CAAC;AACH;AAgCO,MAAM,WAA2B,OAAO,OAAO,gBAAgB;AAAA,EACpE;AACF,CAAC;AAED,IAAO,mBAAQ;","names":["model"]}
+{"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  GenkitError,\n  modelActionMetadata,\n  ModelReference,\n  z,\n} from 'genkit';\nimport { logger } from 'genkit/logging';\nimport { type GenkitPluginV2 } from 'genkit/plugin';\nimport { ActionType } from 'genkit/registry';\nimport OpenAI from 'openai';\nimport { openAICompatible, PluginOptions } from '../index.js';\nimport { defineCompatOpenAIModel } from '../model.js';\nimport {\n  DeepSeekChatCompletionConfigSchema,\n  deepSeekModelRef,\n  deepSeekRequestBuilder,\n  SUPPORTED_DEEPSEEK_MODELS,\n} from './deepseek.js';\n\nexport type DeepSeekPluginOptions = Omit<PluginOptions, 'name' | 'baseURL'>;\n\nfunction createResolver(pluginOptions: PluginOptions) {\n  return async (client: OpenAI, actionType: ActionType, actionName: string) => {\n    if (actionType === 'model') {\n      const modelRef = deepSeekModelRef({\n        name: actionName,\n      });\n      return defineCompatOpenAIModel({\n        name: modelRef.name,\n        client,\n        pluginOptions,\n        modelRef,\n        requestBuilder: deepSeekRequestBuilder,\n      });\n    } else {\n      logger.warn('Only model actions are supported by the DeepSeek plugin');\n      return undefined;\n    }\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        const modelRef =\n          SUPPORTED_DEEPSEEK_MODELS[model.id] ??\n          deepSeekModelRef({\n            name: model.id,\n          });\n        return modelActionMetadata({\n          name: modelRef.name,\n          info: modelRef.info,\n          configSchema: modelRef.configSchema,\n        });\n      })\n  );\n};\n\nexport function deepSeekPlugin(\n  options?: DeepSeekPluginOptions\n): GenkitPluginV2 {\n  const apiKey = options?.apiKey ?? process.env.DEEPSEEK_API_KEY;\n  if (!apiKey) {\n    throw new GenkitError({\n      status: 'FAILED_PRECONDITION',\n      message:\n        'Please pass in the API key or set the DEEPSEEK_API_KEY environment variable.',\n    });\n  }\n  const pluginOptions = { name: 'deepseek', ...options };\n  return openAICompatible({\n    name: 'deepseek',\n    baseURL: 'https://api.deepseek.com',\n    apiKey,\n    ...options,\n    initializer: async (client) => {\n      return Object.values(SUPPORTED_DEEPSEEK_MODELS).map((modelRef) =>\n        defineCompatOpenAIModel({\n          name: modelRef.name,\n          client,\n          pluginOptions,\n          modelRef,\n          requestBuilder: deepSeekRequestBuilder,\n        })\n      );\n    },\n    resolver: createResolver(pluginOptions),\n    listActions,\n  });\n}\n\nexport type DeepSeekPlugin = {\n  (params?: DeepSeekPluginOptions): GenkitPluginV2;\n  model(\n    name: keyof typeof SUPPORTED_DEEPSEEK_MODELS,\n    config?: z.infer<typeof DeepSeekChatCompletionConfigSchema>\n  ): ModelReference<typeof DeepSeekChatCompletionConfigSchema>;\n  model(name: string, config?: any): ModelReference<z.ZodTypeAny>;\n};\n\nconst model = ((name: string, config?: any): ModelReference<z.ZodTypeAny> => {\n  return deepSeekModelRef({\n    name,\n    config,\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,oBAMO;AACP,qBAAuB;AAIvB,eAAgD;AAChD,mBAAwC;AACxC,sBAKO;AAIP,SAAS,eAAe,eAA8B;AACpD,SAAO,OAAO,QAAgB,YAAwB,eAAuB;AAC3E,QAAI,eAAe,SAAS;AAC1B,YAAM,eAAW,kCAAiB;AAAA,QAChC,MAAM;AAAA,MACR,CAAC;AACD,iBAAO,sCAAwB;AAAA,QAC7B,MAAM,SAAS;AAAA,QACf;AAAA,QACA;AAAA,QACA;AAAA,QACA,gBAAgB;AAAA,MAClB,CAAC;AAAA,IACH,OAAO;AACL,4BAAO,KAAK,yDAAyD;AACrE,aAAO;AAAA,IACT;AAAA,EACF;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,YAAM,WACJ,0CAA0BA,OAAM,EAAE,SAClC,kCAAiB;AAAA,QACf,MAAMA,OAAM;AAAA,MACd,CAAC;AACH,iBAAO,mCAAoB;AAAA,QACzB,MAAM,SAAS;AAAA,QACf,MAAM,SAAS;AAAA,QACf,cAAc,SAAS;AAAA,MACzB,CAAC;AAAA,IACH,CAAC;AAAA,EACL;AACF;AAEO,SAAS,eACd,SACgB;AAChB,QAAM,SAAS,SAAS,UAAU,QAAQ,IAAI;AAC9C,MAAI,CAAC,QAAQ;AACX,UAAM,IAAI,0BAAY;AAAA,MACpB,QAAQ;AAAA,MACR,SACE;AAAA,IACJ,CAAC;AAAA,EACH;AACA,QAAM,gBAAgB,EAAE,MAAM,YAAY,GAAG,QAAQ;AACrD,aAAO,2BAAiB;AAAA,IACtB,MAAM;AAAA,IACN,SAAS;AAAA,IACT;AAAA,IACA,GAAG;AAAA,IACH,aAAa,OAAO,WAAW;AAC7B,aAAO,OAAO,OAAO,yCAAyB,EAAE;AAAA,QAAI,CAAC,iBACnD,sCAAwB;AAAA,UACtB,MAAM,SAAS;AAAA,UACf;AAAA,UACA;AAAA,UACA;AAAA,UACA,gBAAgB;AAAA,QAClB,CAAC;AAAA,MACH;AAAA,IACF;AAAA,IACA,UAAU,eAAe,aAAa;AAAA,IACtC;AAAA,EACF,CAAC;AACH;AAWA,MAAM,SAAS,CAAC,MAAc,WAA+C;AAC3E,aAAO,kCAAiB;AAAA,IACtB;AAAA,IACA;AAAA,EACF,CAAC;AACH;AAgCO,MAAM,WAA2B,OAAO,OAAO,gBAAgB;AAAA,EACpE;AACF,CAAC;AAED,IAAO,mBAAQ;","names":["model"]}

package/lib/deepseek/index.mjs

@@ -72,12 +72,12 @@ function deepSeekPlugin(options) {
     listActions
   });
 }
-const model = (name, config) => {
+const model = ((name, config) => {
   return deepSeekModelRef({
     name,
     config
   });
-};
+});
 const deepSeek = Object.assign(deepSeekPlugin, {
   model
 });

package/lib/deepseek/index.mjs.map

@@ -1 +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  GenkitError,\n  modelActionMetadata,\n  ModelReference,\n  z,\n} from 'genkit';\nimport { logger } from 'genkit/logging';\nimport { type GenkitPluginV2 } from 'genkit/plugin';\nimport { ActionType } from 'genkit/registry';\nimport OpenAI from 'openai';\nimport { openAICompatible, PluginOptions } from '../index.js';\nimport { defineCompatOpenAIModel } from '../model.js';\nimport {\n  DeepSeekChatCompletionConfigSchema,\n  deepSeekModelRef,\n  deepSeekRequestBuilder,\n  SUPPORTED_DEEPSEEK_MODELS,\n} from './deepseek.js';\n\nexport type DeepSeekPluginOptions = Omit<PluginOptions, 'name' | 'baseURL'>;\n\nfunction createResolver(pluginOptions: PluginOptions) {\n  return async (client: OpenAI, actionType: ActionType, actionName: string) => {\n    if (actionType === 'model') {\n      const modelRef = deepSeekModelRef({\n        name: actionName,\n      });\n      return defineCompatOpenAIModel({\n        name: modelRef.name,\n        client,\n        pluginOptions,\n        modelRef,\n        requestBuilder: deepSeekRequestBuilder,\n      });\n    } else {\n      logger.warn('Only model actions are supported by the DeepSeek plugin');\n      return undefined;\n    }\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        const modelRef =\n          SUPPORTED_DEEPSEEK_MODELS[model.id] ??\n          deepSeekModelRef({\n            name: model.id,\n          });\n        return modelActionMetadata({\n          name: modelRef.name,\n          info: modelRef.info,\n          configSchema: modelRef.configSchema,\n        });\n      })\n  );\n};\n\nexport function deepSeekPlugin(\n  options?: DeepSeekPluginOptions\n): GenkitPluginV2 {\n  const apiKey = options?.apiKey ?? process.env.DEEPSEEK_API_KEY;\n  if (!apiKey) {\n    throw new GenkitError({\n      status: 'FAILED_PRECONDITION',\n      message:\n        'Please pass in the API key or set the DEEPSEEK_API_KEY environment variable.',\n    });\n  }\n  const pluginOptions = { name: 'deepseek', ...options };\n  return openAICompatible({\n    name: 'deepseek',\n    baseURL: 'https://api.deepseek.com',\n    apiKey,\n    ...options,\n    initializer: async (client) => {\n      return Object.values(SUPPORTED_DEEPSEEK_MODELS).map((modelRef) =>\n        defineCompatOpenAIModel({\n          name: modelRef.name,\n          client,\n          pluginOptions,\n          modelRef,\n          requestBuilder: deepSeekRequestBuilder,\n        })\n      );\n    },\n    resolver: createResolver(pluginOptions),\n    listActions,\n  });\n}\n\nexport type DeepSeekPlugin = {\n  (params?: DeepSeekPluginOptions): GenkitPluginV2;\n  model(\n    name: keyof typeof SUPPORTED_DEEPSEEK_MODELS,\n    config?: z.infer<typeof DeepSeekChatCompletionConfigSchema>\n  ): ModelReference<typeof DeepSeekChatCompletionConfigSchema>;\n  model(name: string, config?: any): ModelReference<z.ZodTypeAny>;\n};\n\nconst model = ((name: string, config?: any): ModelReference<z.ZodTypeAny> => {\n  return deepSeekModelRef({\n    name,\n    config,\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,EAEE;AAAA,EACA;AAAA,OAGK;AACP,SAAS,cAAc;AAIvB,SAAS,wBAAuC;AAChD,SAAS,+BAA+B;AACxC;AAAA,EAEE;AAAA,EACA;AAAA,EACA;AAAA,OACK;AAIP,SAAS,eAAe,eAA8B;AACpD,SAAO,OAAO,QAAgB,YAAwB,eAAuB;AAC3E,QAAI,eAAe,SAAS;AAC1B,YAAM,WAAW,iBAAiB;AAAA,QAChC,MAAM;AAAA,MACR,CAAC;AACD,aAAO,wBAAwB;AAAA,QAC7B,MAAM,SAAS;AAAA,QACf;AAAA,QACA;AAAA,QACA;AAAA,QACA,gBAAgB;AAAA,MAClB,CAAC;AAAA,IACH,OAAO;AACL,aAAO,KAAK,yDAAyD;AACrE,aAAO;AAAA,IACT;AAAA,EACF;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,YAAM,WACJ,0BAA0BA,OAAM,EAAE,KAClC,iBAAiB;AAAA,QACf,MAAMA,OAAM;AAAA,MACd,CAAC;AACH,aAAO,oBAAoB;AAAA,QACzB,MAAM,SAAS;AAAA,QACf,MAAM,SAAS;AAAA,QACf,cAAc,SAAS;AAAA,MACzB,CAAC;AAAA,IACH,CAAC;AAAA,EACL;AACF;AAEO,SAAS,eACd,SACgB;AAChB,QAAM,SAAS,SAAS,UAAU,QAAQ,IAAI;AAC9C,MAAI,CAAC,QAAQ;AACX,UAAM,IAAI,YAAY;AAAA,MACpB,QAAQ;AAAA,MACR,SACE;AAAA,IACJ,CAAC;AAAA,EACH;AACA,QAAM,gBAAgB,EAAE,MAAM,YAAY,GAAG,QAAQ;AACrD,SAAO,iBAAiB;AAAA,IACtB,MAAM;AAAA,IACN,SAAS;AAAA,IACT;AAAA,IACA,GAAG;AAAA,IACH,aAAa,OAAO,WAAW;AAC7B,aAAO,OAAO,OAAO,yBAAyB,EAAE;AAAA,QAAI,CAAC,aACnD,wBAAwB;AAAA,UACtB,MAAM,SAAS;AAAA,UACf;AAAA,UACA;AAAA,UACA;AAAA,UACA,gBAAgB;AAAA,QAClB,CAAC;AAAA,MACH;AAAA,IACF;AAAA,IACA,UAAU,eAAe,aAAa;AAAA,IACtC;AAAA,EACF,CAAC;AACH;AAWA,MAAM,QAAS,CAAC,MAAc,WAA+C;AAC3E,SAAO,iBAAiB;AAAA,IACtB;AAAA,IACA;AAAA,EACF,CAAC;AACH;AAgCO,MAAM,WAA2B,OAAO,OAAO,gBAAgB;AAAA,EACpE;AACF,CAAC;AAED,IAAO,mBAAQ;","names":["model"]}
+{"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  GenkitError,\n  modelActionMetadata,\n  ModelReference,\n  z,\n} from 'genkit';\nimport { logger } from 'genkit/logging';\nimport { type GenkitPluginV2 } from 'genkit/plugin';\nimport { ActionType } from 'genkit/registry';\nimport OpenAI from 'openai';\nimport { openAICompatible, PluginOptions } from '../index.js';\nimport { defineCompatOpenAIModel } from '../model.js';\nimport {\n  DeepSeekChatCompletionConfigSchema,\n  deepSeekModelRef,\n  deepSeekRequestBuilder,\n  SUPPORTED_DEEPSEEK_MODELS,\n} from './deepseek.js';\n\nexport type DeepSeekPluginOptions = Omit<PluginOptions, 'name' | 'baseURL'>;\n\nfunction createResolver(pluginOptions: PluginOptions) {\n  return async (client: OpenAI, actionType: ActionType, actionName: string) => {\n    if (actionType === 'model') {\n      const modelRef = deepSeekModelRef({\n        name: actionName,\n      });\n      return defineCompatOpenAIModel({\n        name: modelRef.name,\n        client,\n        pluginOptions,\n        modelRef,\n        requestBuilder: deepSeekRequestBuilder,\n      });\n    } else {\n      logger.warn('Only model actions are supported by the DeepSeek plugin');\n      return undefined;\n    }\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        const modelRef =\n          SUPPORTED_DEEPSEEK_MODELS[model.id] ??\n          deepSeekModelRef({\n            name: model.id,\n          });\n        return modelActionMetadata({\n          name: modelRef.name,\n          info: modelRef.info,\n          configSchema: modelRef.configSchema,\n        });\n      })\n  );\n};\n\nexport function deepSeekPlugin(\n  options?: DeepSeekPluginOptions\n): GenkitPluginV2 {\n  const apiKey = options?.apiKey ?? process.env.DEEPSEEK_API_KEY;\n  if (!apiKey) {\n    throw new GenkitError({\n      status: 'FAILED_PRECONDITION',\n      message:\n        'Please pass in the API key or set the DEEPSEEK_API_KEY environment variable.',\n    });\n  }\n  const pluginOptions = { name: 'deepseek', ...options };\n  return openAICompatible({\n    name: 'deepseek',\n    baseURL: 'https://api.deepseek.com',\n    apiKey,\n    ...options,\n    initializer: async (client) => {\n      return Object.values(SUPPORTED_DEEPSEEK_MODELS).map((modelRef) =>\n        defineCompatOpenAIModel({\n          name: modelRef.name,\n          client,\n          pluginOptions,\n          modelRef,\n          requestBuilder: deepSeekRequestBuilder,\n        })\n      );\n    },\n    resolver: createResolver(pluginOptions),\n    listActions,\n  });\n}\n\nexport type DeepSeekPlugin = {\n  (params?: DeepSeekPluginOptions): GenkitPluginV2;\n  model(\n    name: keyof typeof SUPPORTED_DEEPSEEK_MODELS,\n    config?: z.infer<typeof DeepSeekChatCompletionConfigSchema>\n  ): ModelReference<typeof DeepSeekChatCompletionConfigSchema>;\n  model(name: string, config?: any): ModelReference<z.ZodTypeAny>;\n};\n\nconst model = ((name: string, config?: any): ModelReference<z.ZodTypeAny> => {\n  return deepSeekModelRef({\n    name,\n    config,\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,EAEE;AAAA,EACA;AAAA,OAGK;AACP,SAAS,cAAc;AAIvB,SAAS,wBAAuC;AAChD,SAAS,+BAA+B;AACxC;AAAA,EAEE;AAAA,EACA;AAAA,EACA;AAAA,OACK;AAIP,SAAS,eAAe,eAA8B;AACpD,SAAO,OAAO,QAAgB,YAAwB,eAAuB;AAC3E,QAAI,eAAe,SAAS;AAC1B,YAAM,WAAW,iBAAiB;AAAA,QAChC,MAAM;AAAA,MACR,CAAC;AACD,aAAO,wBAAwB;AAAA,QAC7B,MAAM,SAAS;AAAA,QACf;AAAA,QACA;AAAA,QACA;AAAA,QACA,gBAAgB;AAAA,MAClB,CAAC;AAAA,IACH,OAAO;AACL,aAAO,KAAK,yDAAyD;AACrE,aAAO;AAAA,IACT;AAAA,EACF;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,YAAM,WACJ,0BAA0BA,OAAM,EAAE,KAClC,iBAAiB;AAAA,QACf,MAAMA,OAAM;AAAA,MACd,CAAC;AACH,aAAO,oBAAoB;AAAA,QACzB,MAAM,SAAS;AAAA,QACf,MAAM,SAAS;AAAA,QACf,cAAc,SAAS;AAAA,MACzB,CAAC;AAAA,IACH,CAAC;AAAA,EACL;AACF;AAEO,SAAS,eACd,SACgB;AAChB,QAAM,SAAS,SAAS,UAAU,QAAQ,IAAI;AAC9C,MAAI,CAAC,QAAQ;AACX,UAAM,IAAI,YAAY;AAAA,MACpB,QAAQ;AAAA,MACR,SACE;AAAA,IACJ,CAAC;AAAA,EACH;AACA,QAAM,gBAAgB,EAAE,MAAM,YAAY,GAAG,QAAQ;AACrD,SAAO,iBAAiB;AAAA,IACtB,MAAM;AAAA,IACN,SAAS;AAAA,IACT;AAAA,IACA,GAAG;AAAA,IACH,aAAa,OAAO,WAAW;AAC7B,aAAO,OAAO,OAAO,yBAAyB,EAAE;AAAA,QAAI,CAAC,aACnD,wBAAwB;AAAA,UACtB,MAAM,SAAS;AAAA,UACf;AAAA,UACA;AAAA,UACA;AAAA,UACA,gBAAgB;AAAA,QAClB,CAAC;AAAA,MACH;AAAA,IACF;AAAA,IACA,UAAU,eAAe,aAAa;AAAA,IACtC;AAAA,EACF,CAAC;AACH;AAWA,MAAM,SAAS,CAAC,MAAc,WAA+C;AAC3E,SAAO,iBAAiB;AAAA,IACtB;AAAA,IACA;AAAA,EACF,CAAC;AACH;AAgCO,MAAM,WAA2B,OAAO,OAAO,gBAAgB;AAAA,EACpE;AACF,CAAC;AAED,IAAO,mBAAQ;","names":["model"]}

package/lib/embedder.d.mts

@@ -1,6 +1,6 @@
 import 'genkit';
 import 'openai';
-export { d as defineCompatOpenAIEmbedder } from './audio-Cr0WtG4I.mjs';
+export { d as defineCompatOpenAIEmbedder } from './audio-CW-qdV9D.mjs';
 import 'genkit/model';
 import 'openai/core.mjs';
 import 'openai/resources/audio/index.mjs';

package/lib/embedder.d.ts

@@ -1,6 +1,6 @@
 import 'genkit';
 import 'openai';
-export { d as defineCompatOpenAIEmbedder } from './audio-Cr0WtG4I.js';
+export { d as defineCompatOpenAIEmbedder } from './audio-CW-qdV9D.js';
 import 'genkit/model';
 import 'openai/core.mjs';
 import 'openai/resources/audio/index.mjs';

package/lib/image.d.mts

@@ -2,7 +2,7 @@ import 'genkit';
 import 'genkit/model';
 import 'openai';
 import 'openai/resources/images.mjs';
-export { b as IMAGE_GENERATION_MODEL_INFO, I as ImageGenerationCommonConfigSchema, a as ImageRequestBuilder, e as compatOaiImageModelRef, c as defineCompatOpenAIImageModel } from './audio-Cr0WtG4I.mjs';
+export { b as IMAGE_GENERATION_MODEL_INFO, I as ImageGenerationCommonConfigSchema, a as ImageRequestBuilder, c as compatOaiImageModelRef, e as defineCompatOpenAIImageModel } from './audio-CW-qdV9D.mjs';
 import 'openai/core.mjs';
 import 'openai/resources/audio/index.mjs';
 import 'genkit/plugin';

package/lib/image.d.ts

@@ -2,7 +2,7 @@ import 'genkit';
 import 'genkit/model';
 import 'openai';
 import 'openai/resources/images.mjs';
-export { b as IMAGE_GENERATION_MODEL_INFO, I as ImageGenerationCommonConfigSchema, a as ImageRequestBuilder, e as compatOaiImageModelRef, c as defineCompatOpenAIImageModel } from './audio-Cr0WtG4I.js';
+export { b as IMAGE_GENERATION_MODEL_INFO, I as ImageGenerationCommonConfigSchema, a as ImageRequestBuilder, c as compatOaiImageModelRef, e as defineCompatOpenAIImageModel } from './audio-CW-qdV9D.js';
 import 'openai/core.mjs';
 import 'openai/resources/audio/index.mjs';
 import 'genkit/plugin';

package/lib/index.d.mts

@@ -2,7 +2,7 @@ import 'genkit/plugin';
 import 'genkit';
 import 'genkit/registry';
 import 'openai';
-export { C as ChatCompletionCommonConfigSchema, I as ImageGenerationCommonConfigSchema, a as ImageRequestBuilder, M as ModelRequestBuilder, P as PluginOptions, S as SpeechConfigSchema, D as SpeechRequestBuilder, T as TranscriptionConfigSchema, E as TranscriptionRequestBuilder, r as TranslationConfigSchema, p as TranslationRequestBuilder, e as compatOaiImageModelRef, n as compatOaiModelRef, y as compatOaiSpeechModelRef, z as compatOaiTranscriptionModelRef, w as compatOaiTranslationModelRef, x as default, d as defineCompatOpenAIEmbedder, c as defineCompatOpenAIImageModel, m as defineCompatOpenAIModel, A as defineCompatOpenAISpeechModel, B as defineCompatOpenAITranscriptionModel, v as defineCompatOpenAITranslationModel, x as openAICompatible, o as openAIModelRunner } from './audio-Cr0WtG4I.mjs';
+export { C as ChatCompletionCommonConfigSchema, I as ImageGenerationCommonConfigSchema, a as ImageRequestBuilder, M as ModelRequestBuilder, P as PluginOptions, S as SpeechConfigSchema, x as SpeechRequestBuilder, T as TranscriptionConfigSchema, y as TranscriptionRequestBuilder, q as TranslationConfigSchema, r as TranslationRequestBuilder, c as compatOaiImageModelRef, f as compatOaiModelRef, z as compatOaiSpeechModelRef, A as compatOaiTranscriptionModelRef, s as compatOaiTranslationModelRef, B as default, d as defineCompatOpenAIEmbedder, e as defineCompatOpenAIImageModel, g as defineCompatOpenAIModel, D as defineCompatOpenAISpeechModel, E as defineCompatOpenAITranscriptionModel, u as defineCompatOpenAITranslationModel, B as openAICompatible, o as openAIModelRunner } from './audio-CW-qdV9D.mjs';
 import 'genkit/model';
 import 'openai/core.mjs';
 import 'openai/resources/audio/index.mjs';

package/lib/index.d.ts

@@ -2,7 +2,7 @@ import 'genkit/plugin';
 import 'genkit';
 import 'genkit/registry';
 import 'openai';
-export { C as ChatCompletionCommonConfigSchema, I as ImageGenerationCommonConfigSchema, a as ImageRequestBuilder, M as ModelRequestBuilder, P as PluginOptions, S as SpeechConfigSchema, D as SpeechRequestBuilder, T as TranscriptionConfigSchema, E as TranscriptionRequestBuilder, r as TranslationConfigSchema, p as TranslationRequestBuilder, e as compatOaiImageModelRef, n as compatOaiModelRef, y as compatOaiSpeechModelRef, z as compatOaiTranscriptionModelRef, w as compatOaiTranslationModelRef, x as default, d as defineCompatOpenAIEmbedder, c as defineCompatOpenAIImageModel, m as defineCompatOpenAIModel, A as defineCompatOpenAISpeechModel, B as defineCompatOpenAITranscriptionModel, v as defineCompatOpenAITranslationModel, x as openAICompatible, o as openAIModelRunner } from './audio-Cr0WtG4I.js';
+export { C as ChatCompletionCommonConfigSchema, I as ImageGenerationCommonConfigSchema, a as ImageRequestBuilder, M as ModelRequestBuilder, P as PluginOptions, S as SpeechConfigSchema, x as SpeechRequestBuilder, T as TranscriptionConfigSchema, y as TranscriptionRequestBuilder, q as TranslationConfigSchema, r as TranslationRequestBuilder, c as compatOaiImageModelRef, f as compatOaiModelRef, z as compatOaiSpeechModelRef, A as compatOaiTranscriptionModelRef, s as compatOaiTranslationModelRef, B as default, d as defineCompatOpenAIEmbedder, e as defineCompatOpenAIImageModel, g as defineCompatOpenAIModel, D as defineCompatOpenAISpeechModel, E as defineCompatOpenAITranscriptionModel, u as defineCompatOpenAITranslationModel, B as openAICompatible, o as openAIModelRunner } from './audio-CW-qdV9D.js';
 import 'genkit/model';
 import 'openai/core.mjs';
 import 'openai/resources/audio/index.mjs';

package/lib/model.d.mts

@@ -2,7 +2,7 @@ import 'genkit';
 import 'genkit/model';
 import 'openai';
 import 'openai/resources/index.mjs';
-export { C as ChatCompletionCommonConfigSchema, M as ModelRequestBuilder, n as compatOaiModelRef, m as defineCompatOpenAIModel, j as fromOpenAIChoice, k as fromOpenAIChunkChoice, i as fromOpenAIToolCall, o as openAIModelRunner, h as toOpenAIMessages, l as toOpenAIRequestBody, t as toOpenAIRole, g as toOpenAITextAndMedia, f as toOpenAITool } from './audio-Cr0WtG4I.mjs';
+export { C as ChatCompletionCommonConfigSchema, M as ModelRequestBuilder, f as compatOaiModelRef, g as defineCompatOpenAIModel, h as fromOpenAIChoice, i as fromOpenAIChunkChoice, j as fromOpenAIToolCall, o as openAIModelRunner, t as toOpenAIMessages, k as toOpenAIRequestBody, l as toOpenAIRole, m as toOpenAITextAndMedia, n as toOpenAITool } from './audio-CW-qdV9D.mjs';
 import 'openai/core.mjs';
 import 'openai/resources/audio/index.mjs';
 import 'genkit/plugin';

package/lib/model.d.ts

@@ -2,7 +2,7 @@ import 'genkit';
 import 'genkit/model';
 import 'openai';
 import 'openai/resources/index.mjs';
-export { C as ChatCompletionCommonConfigSchema, M as ModelRequestBuilder, n as compatOaiModelRef, m as defineCompatOpenAIModel, j as fromOpenAIChoice, k as fromOpenAIChunkChoice, i as fromOpenAIToolCall, o as openAIModelRunner, h as toOpenAIMessages, l as toOpenAIRequestBody, t as toOpenAIRole, g as toOpenAITextAndMedia, f as toOpenAITool } from './audio-Cr0WtG4I.js';
+export { C as ChatCompletionCommonConfigSchema, M as ModelRequestBuilder, f as compatOaiModelRef, g as defineCompatOpenAIModel, h as fromOpenAIChoice, i as fromOpenAIChunkChoice, j as fromOpenAIToolCall, o as openAIModelRunner, t as toOpenAIMessages, k as toOpenAIRequestBody, l as toOpenAIRole, m as toOpenAITextAndMedia, n as toOpenAITool } from './audio-CW-qdV9D.js';
 import 'openai/core.mjs';
 import 'openai/resources/audio/index.mjs';
 import 'genkit/plugin';

package/lib/openai/dalle.d.mts

@@ -1,7 +1,7 @@
 import * as genkit from 'genkit';
 import { z } from 'genkit';
 import { ModelInfo } from 'genkit/model';
-import { a as ImageRequestBuilder } from '../audio-Cr0WtG4I.mjs';
+import { a as ImageRequestBuilder } from '../audio-CW-qdV9D.mjs';
 import 'openai';
 import 'openai/core.mjs';
 import 'openai/resources/audio/index.mjs';

package/lib/openai/dalle.d.ts

@@ -1,7 +1,7 @@
 import * as genkit from 'genkit';
 import { z } from 'genkit';
 import { ModelInfo } from 'genkit/model';
-import { a as ImageRequestBuilder } from '../audio-Cr0WtG4I.js';
+import { a as ImageRequestBuilder } from '../audio-CW-qdV9D.js';
 import 'openai';
 import 'openai/core.mjs';
 import 'openai/resources/audio/index.mjs';

package/lib/openai/embedder.d.mts

@@ -1,4 +1,4 @@
-import { z, EmbedderReference } from 'genkit';
+import { EmbedderReference, z } from 'genkit';
 
 /**
  * Copyright 2024 The Fire Company

package/lib/openai/embedder.d.ts

@@ -1,4 +1,4 @@
-import { z, EmbedderReference } from 'genkit';
+import { EmbedderReference, z } from 'genkit';
 
 /**
  * Copyright 2024 The Fire Company

package/lib/openai/gpt.d.mts

@@ -1,5 +1,5 @@
 import { z } from 'genkit';
-import { ModelInfo, ModelReference } from 'genkit/model';
+import { ModelReference, ModelInfo } from 'genkit/model';
 
 /**
  * Copyright 2024 The Fire Company

package/lib/openai/gpt.d.ts

@@ -1,5 +1,5 @@
 import { z } from 'genkit';
-import { ModelInfo, ModelReference } from 'genkit/model';
+import { ModelReference, ModelInfo } from 'genkit/model';
 
 /**
  * Copyright 2024 The Fire Company

package/lib/openai/index.d.mts

@@ -1,6 +1,6 @@
 import { z, ModelReference, EmbedderReference } from 'genkit';
 import { GenkitPluginV2 } from 'genkit/plugin';
-import { P as PluginOptions, I as ImageGenerationCommonConfigSchema, S as SpeechConfigSchema, T as TranscriptionConfigSchema } from '../audio-Cr0WtG4I.mjs';
+import { P as PluginOptions, I as ImageGenerationCommonConfigSchema, S as SpeechConfigSchema, T as TranscriptionConfigSchema } from '../audio-CW-qdV9D.mjs';
 import { SUPPORTED_IMAGE_MODELS } from './dalle.mjs';
 import { SUPPORTED_EMBEDDING_MODELS, TextEmbeddingConfigSchema } from './embedder.mjs';
 import { SUPPORTED_GPT_MODELS, OpenAIChatCompletionConfigSchema } from './gpt.mjs';

package/lib/openai/index.d.ts

@@ -1,6 +1,6 @@
 import { z, ModelReference, EmbedderReference } from 'genkit';
 import { GenkitPluginV2 } from 'genkit/plugin';
-import { P as PluginOptions, I as ImageGenerationCommonConfigSchema, S as SpeechConfigSchema, T as TranscriptionConfigSchema } from '../audio-Cr0WtG4I.js';
+import { P as PluginOptions, I as ImageGenerationCommonConfigSchema, S as SpeechConfigSchema, T as TranscriptionConfigSchema } from '../audio-CW-qdV9D.js';
 import { SUPPORTED_IMAGE_MODELS } from './dalle.js';
 import { SUPPORTED_EMBEDDING_MODELS, TextEmbeddingConfigSchema } from './embedder.js';
 import { SUPPORTED_GPT_MODELS, OpenAIChatCompletionConfigSchema } from './gpt.js';

package/lib/openai/index.js

@@ -214,7 +214,7 @@ function openAIPlugin(options) {
     listActions
   });
 }
-const model = (name, config) => {
+const model = ((name, config) => {
   if (name.includes("gpt-image-1") || name.includes("dall-e")) {
     return (0, import_dalle.openAIImageModelRef)({
       name,
@@ -243,15 +243,15 @@ const model = (name, config) => {
     name,
     config
   });
-};
-const embedder = (name, config) => {
+});
+const embedder = ((name, config) => {
   return (0, import_genkit.embedderRef)({
     name,
     config,
     configSchema: import_embedder2.TextEmbeddingConfigSchema,
     namespace: "openai"
   });
-};
+});
 const openAI = Object.assign(openAIPlugin, {
   model,
   embedder

package/lib/openai/index.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/openai/index.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\nimport {\n  ActionMetadata,\n  embedderActionMetadata,\n  embedderRef,\n  EmbedderReference,\n  modelActionMetadata,\n  ModelReference,\n  z,\n} from 'genkit';\nimport { ResolvableAction, type GenkitPluginV2 } from 'genkit/plugin';\nimport { ActionType } from 'genkit/registry';\nimport OpenAI from 'openai';\nimport {\n  defineCompatOpenAISpeechModel,\n  defineCompatOpenAITranscriptionModel,\n  SpeechConfigSchema,\n  TranscriptionConfigSchema,\n} from '../audio.js';\nimport { defineCompatOpenAIEmbedder } from '../embedder.js';\nimport {\n  defineCompatOpenAIImageModel,\n  ImageGenerationCommonConfigSchema,\n} from '../image.js';\nimport { openAICompatible, PluginOptions } from '../index.js';\nimport { defineCompatOpenAIModel } from '../model.js';\nimport {\n  gptImage1RequestBuilder,\n  openAIImageModelRef,\n  SUPPORTED_IMAGE_MODELS,\n} from './dalle.js';\nimport {\n  SUPPORTED_EMBEDDING_MODELS,\n  TextEmbeddingConfigSchema,\n} from './embedder.js';\nimport {\n  OpenAIChatCompletionConfigSchema,\n  openAIModelRef,\n  SUPPORTED_GPT_MODELS,\n} from './gpt.js';\nimport { openAITranscriptionModelRef, SUPPORTED_STT_MODELS } from './stt.js';\nimport { openAISpeechModelRef, SUPPORTED_TTS_MODELS } from './tts.js';\nimport {\n  defineOpenAIWhisperModel,\n  openAIWhisperModelRef,\n  SUPPORTED_WHISPER_MODELS,\n  WhisperConfigSchema,\n} from './whisper.js';\n\nexport type OpenAIPluginOptions = Omit<PluginOptions, 'name' | 'baseURL'>;\n\nconst UNSUPPORTED_MODEL_MATCHERS = ['babbage', 'davinci', 'codex'];\n\nfunction createResolver(pluginOptions: PluginOptions) {\n  return async (client: OpenAI, actionType: ActionType, actionName: string) => {\n    if (actionType === 'embedder') {\n      return defineCompatOpenAIEmbedder({\n        name: actionName,\n        client,\n        pluginOptions,\n      });\n    } else if (\n      actionName.includes('gpt-image-1') ||\n      actionName.includes('dall-e')\n    ) {\n      const modelRef = openAIImageModelRef({ name: actionName });\n      return defineCompatOpenAIImageModel({\n        name: modelRef.name,\n        client,\n        pluginOptions,\n        modelRef,\n      });\n    } else if (actionName.includes('tts')) {\n      const modelRef = openAISpeechModelRef({ name: actionName });\n      return defineCompatOpenAISpeechModel({\n        name: modelRef.name,\n        client,\n        pluginOptions,\n        modelRef,\n      });\n    } else if (actionName.includes('whisper')) {\n      const modelRef = openAIWhisperModelRef({ name: actionName });\n      return defineOpenAIWhisperModel({\n        name: modelRef.name,\n        client,\n        pluginOptions,\n        modelRef,\n      });\n    } else if (actionName.includes('transcribe')) {\n      const modelRef = openAITranscriptionModelRef({\n        name: actionName,\n      });\n      return defineCompatOpenAITranscriptionModel({\n        name: modelRef.name,\n        client,\n        pluginOptions,\n        modelRef,\n      });\n    } else {\n      const modelRef = openAIModelRef({ name: actionName });\n      return defineCompatOpenAIModel({\n        name: modelRef.name,\n        client,\n        pluginOptions,\n        modelRef,\n      });\n    }\n  };\n}\n\nfunction filterOpenAiModels(model: OpenAI.Model): boolean {\n  return !UNSUPPORTED_MODEL_MATCHERS.some((m) => model.id.includes(m));\n}\n\nconst listActions = async (client: OpenAI): Promise<ActionMetadata[]> => {\n  return await client.models.list().then((response) =>\n    response.data.filter(filterOpenAiModels).map((model: OpenAI.Model) => {\n      if (model.id.includes('embedding')) {\n        return embedderActionMetadata({\n          name: model.id,\n          configSchema: TextEmbeddingConfigSchema,\n          info: SUPPORTED_EMBEDDING_MODELS[model.id]?.info,\n        });\n      } else if (\n        model.id.includes('gpt-image-1') ||\n        model.id.includes('dall-e')\n      ) {\n        const modelRef =\n          SUPPORTED_IMAGE_MODELS[model.id] ??\n          openAIImageModelRef({ name: model.id });\n        return modelActionMetadata({\n          name: modelRef.name,\n          info: modelRef.info,\n          configSchema: modelRef.configSchema,\n        });\n      } else if (model.id.includes('tts')) {\n        const modelRef =\n          SUPPORTED_TTS_MODELS[model.id] ??\n          openAISpeechModelRef({ name: model.id });\n        return modelActionMetadata({\n          name: modelRef.name,\n          info: modelRef.info,\n          configSchema: modelRef.configSchema,\n        });\n      } else if (model.id.includes('whisper')) {\n        const modelRef =\n          SUPPORTED_WHISPER_MODELS[model.id] ??\n          openAIWhisperModelRef({ name: model.id });\n        return modelActionMetadata({\n          name: modelRef.name,\n          info: modelRef.info,\n          configSchema: modelRef.configSchema,\n        });\n      } else if (model.id.includes('transcribe')) {\n        const modelRef =\n          SUPPORTED_STT_MODELS[model.id] ??\n          openAITranscriptionModelRef({ name: model.id });\n        return modelActionMetadata({\n          name: modelRef.name,\n          info: modelRef.info,\n          configSchema: modelRef.configSchema,\n        });\n      } else {\n        const modelRef =\n          SUPPORTED_GPT_MODELS[model.id] ?? openAIModelRef({ name: model.id });\n        return modelActionMetadata({\n          name: modelRef.name,\n          info: modelRef.info,\n          configSchema: modelRef.configSchema,\n        });\n      }\n    })\n  );\n};\n\nexport function openAIPlugin(options?: OpenAIPluginOptions): GenkitPluginV2 {\n  const pluginOptions = { name: 'openai', ...options };\n  return openAICompatible({\n    name: 'openai',\n    ...options,\n    initializer: async (client) => {\n      const models = [] as ResolvableAction[];\n      models.push(\n        ...Object.values(SUPPORTED_GPT_MODELS).map((modelRef) =>\n          defineCompatOpenAIModel({\n            name: modelRef.name,\n            client,\n            pluginOptions,\n            modelRef,\n          })\n        )\n      );\n      models.push(\n        ...Object.values(SUPPORTED_EMBEDDING_MODELS).map((embedderRef) =>\n          defineCompatOpenAIEmbedder({\n            name: embedderRef.name,\n            client,\n            pluginOptions,\n            embedderRef,\n          })\n        )\n      );\n      models.push(\n        ...Object.values(SUPPORTED_TTS_MODELS).map((modelRef) =>\n          defineCompatOpenAISpeechModel({\n            name: modelRef.name,\n            client,\n            pluginOptions,\n            modelRef,\n          })\n        )\n      );\n      models.push(\n        ...Object.values(SUPPORTED_WHISPER_MODELS).map((modelRef) =>\n          defineOpenAIWhisperModel({\n            name: modelRef.name,\n            client,\n            pluginOptions,\n            modelRef,\n          })\n        )\n      );\n      models.push(\n        ...Object.values(SUPPORTED_STT_MODELS).map((modelRef) =>\n          defineCompatOpenAITranscriptionModel({\n            name: modelRef.name,\n            client,\n            pluginOptions,\n            modelRef,\n          })\n        )\n      );\n      models.push(\n        ...Object.values(SUPPORTED_IMAGE_MODELS).map((modelRef) =>\n          defineCompatOpenAIImageModel({\n            name: modelRef.name,\n            client,\n            pluginOptions,\n            modelRef,\n            requestBuilder: modelRef.name.includes('gpt-image-1')\n              ? gptImage1RequestBuilder\n              : undefined,\n          })\n        )\n      );\n      return models;\n    },\n    resolver: createResolver(pluginOptions),\n    listActions,\n  });\n}\n\nexport type OpenAIPlugin = {\n  (params?: OpenAIPluginOptions): GenkitPluginV2;\n  model(\n    name:\n      | keyof typeof SUPPORTED_IMAGE_MODELS\n      | (`dall-e${string}` & {})\n      | (`gpt-image-${string}` & {}),\n    config?: z.infer<typeof ImageGenerationCommonConfigSchema>\n  ): ModelReference<typeof ImageGenerationCommonConfigSchema>;\n  model(\n    name:\n      | keyof typeof SUPPORTED_TTS_MODELS\n      | (`tts-${string}` & {})\n      | (`${string}-tts` & {}),\n    config?: z.infer<typeof SpeechConfigSchema>\n  ): ModelReference<typeof SpeechConfigSchema>;\n  model(\n    name: keyof typeof SUPPORTED_WHISPER_MODELS | (`whisper-${string}` & {}),\n    config?: z.infer<typeof WhisperConfigSchema>\n  ): ModelReference<typeof WhisperConfigSchema>;\n  model(\n    name: keyof typeof SUPPORTED_STT_MODELS | (`${string}-transcribe` & {}),\n    config?: z.infer<typeof TranscriptionConfigSchema>\n  ): ModelReference<typeof TranscriptionConfigSchema>;\n  model(\n    name:\n      | keyof typeof SUPPORTED_GPT_MODELS\n      | (`gpt-${string}` & {})\n      | (`o${number}` & {}),\n    config?: z.infer<typeof OpenAIChatCompletionConfigSchema>\n  ): ModelReference<typeof OpenAIChatCompletionConfigSchema>;\n  model(name: string, config?: any): ModelReference<z.ZodTypeAny>;\n  embedder(\n    name:\n      | keyof typeof SUPPORTED_EMBEDDING_MODELS\n      | (`${string}-embedding-${string}` & {}),\n    config?: z.infer<typeof TextEmbeddingConfigSchema>\n  ): EmbedderReference<typeof TextEmbeddingConfigSchema>;\n  embedder(name: string, config?: any): EmbedderReference<z.ZodTypeAny>;\n};\n\nconst model = ((name: string, config?: any): ModelReference<z.ZodTypeAny> => {\n  if (name.includes('gpt-image-1') || name.includes('dall-e')) {\n    return openAIImageModelRef({\n      name,\n      config,\n    });\n  }\n  if (name.includes('tts')) {\n    return openAISpeechModelRef({\n      name,\n      config,\n    });\n  }\n  if (name.includes('whisper')) {\n    return openAIWhisperModelRef({\n      name,\n      config,\n    });\n  }\n  if (name.includes('transcribe')) {\n    return openAITranscriptionModelRef({\n      name,\n      config,\n    });\n  }\n  return openAIModelRef({\n    name,\n    config,\n  });\n}) as OpenAIPlugin['model'];\n\nconst embedder = ((\n  name: string,\n  config?: any\n): EmbedderReference<z.ZodTypeAny> => {\n  return embedderRef({\n    name,\n    config,\n    configSchema: TextEmbeddingConfigSchema,\n    namespace: 'openai',\n  });\n}) as OpenAIPlugin['embedder'];\n\n/**\n * This module provides an interface to the OpenAI 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 `openai` 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 * - openai: The main plugin function to interact with OpenAI.\n *\n * Usage:\n * To use the models, initialize the openai plugin inside `configureGenkit` and\n * pass the configuration options. If no API key is provided in the options, the\n * environment variable `OPENAI_API_KEY` must be set.\n *\n * Example:\n * ```\n * import { openAI } from '@genkit-ai/compat-oai/openai';\n *\n * export default configureGenkit({\n *  plugins: [\n *    openai()\n *    ... // other plugins\n *  ]\n * });\n * ```\n */\nexport const openAI: OpenAIPlugin = Object.assign(openAIPlugin, {\n  model,\n  embedder,\n});\n\nexport default openAI;\n"],"mappings":";;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAiBA,oBAQO;AAIP,mBAKO;AACP,sBAA2C;AAC3C,mBAGO;AACP,eAAgD;AAChD,mBAAwC;AACxC,mBAIO;AACP,IAAAA,mBAGO;AACP,iBAIO;AACP,iBAAkE;AAClE,iBAA2D;AAC3D,qBAKO;AAIP,MAAM,6BAA6B,CAAC,WAAW,WAAW,OAAO;AAEjE,SAAS,eAAe,eAA8B;AACpD,SAAO,OAAO,QAAgB,YAAwB,eAAuB;AAC3E,QAAI,eAAe,YAAY;AAC7B,iBAAO,4CAA2B;AAAA,QAChC,MAAM;AAAA,QACN;AAAA,QACA;AAAA,MACF,CAAC;AAAA,IACH,WACE,WAAW,SAAS,aAAa,KACjC,WAAW,SAAS,QAAQ,GAC5B;AACA,YAAM,eAAW,kCAAoB,EAAE,MAAM,WAAW,CAAC;AACzD,iBAAO,2CAA6B;AAAA,QAClC,MAAM,SAAS;AAAA,QACf;AAAA,QACA;AAAA,QACA;AAAA,MACF,CAAC;AAAA,IACH,WAAW,WAAW,SAAS,KAAK,GAAG;AACrC,YAAM,eAAW,iCAAqB,EAAE,MAAM,WAAW,CAAC;AAC1D,iBAAO,4CAA8B;AAAA,QACnC,MAAM,SAAS;AAAA,QACf;AAAA,QACA;AAAA,QACA;AAAA,MACF,CAAC;AAAA,IACH,WAAW,WAAW,SAAS,SAAS,GAAG;AACzC,YAAM,eAAW,sCAAsB,EAAE,MAAM,WAAW,CAAC;AAC3D,iBAAO,yCAAyB;AAAA,QAC9B,MAAM,SAAS;AAAA,QACf;AAAA,QACA;AAAA,QACA;AAAA,MACF,CAAC;AAAA,IACH,WAAW,WAAW,SAAS,YAAY,GAAG;AAC5C,YAAM,eAAW,wCAA4B;AAAA,QAC3C,MAAM;AAAA,MACR,CAAC;AACD,iBAAO,mDAAqC;AAAA,QAC1C,MAAM,SAAS;AAAA,QACf;AAAA,QACA;AAAA,QACA;AAAA,MACF,CAAC;AAAA,IACH,OAAO;AACL,YAAM,eAAW,2BAAe,EAAE,MAAM,WAAW,CAAC;AACpD,iBAAO,sCAAwB;AAAA,QAC7B,MAAM,SAAS;AAAA,QACf;AAAA,QACA;AAAA,QACA;AAAA,MACF,CAAC;AAAA,IACH;AAAA,EACF;AACF;AAEA,SAAS,mBAAmBC,QAA8B;AACxD,SAAO,CAAC,2BAA2B,KAAK,CAAC,MAAMA,OAAM,GAAG,SAAS,CAAC,CAAC;AACrE;AAEA,MAAM,cAAc,OAAO,WAA8C;AACvE,SAAO,MAAM,OAAO,OAAO,KAAK,EAAE;AAAA,IAAK,CAAC,aACtC,SAAS,KAAK,OAAO,kBAAkB,EAAE,IAAI,CAACA,WAAwB;AACpE,UAAIA,OAAM,GAAG,SAAS,WAAW,GAAG;AAClC,mBAAO,sCAAuB;AAAA,UAC5B,MAAMA,OAAM;AAAA,UACZ,cAAc;AAAA,UACd,MAAM,4CAA2BA,OAAM,EAAE,GAAG;AAAA,QAC9C,CAAC;AAAA,MACH,WACEA,OAAM,GAAG,SAAS,aAAa,KAC/BA,OAAM,GAAG,SAAS,QAAQ,GAC1B;AACA,cAAM,WACJ,oCAAuBA,OAAM,EAAE,SAC/B,kCAAoB,EAAE,MAAMA,OAAM,GAAG,CAAC;AACxC,mBAAO,mCAAoB;AAAA,UACzB,MAAM,SAAS;AAAA,UACf,MAAM,SAAS;AAAA,UACf,cAAc,SAAS;AAAA,QACzB,CAAC;AAAA,MACH,WAAWA,OAAM,GAAG,SAAS,KAAK,GAAG;AACnC,cAAM,WACJ,gCAAqBA,OAAM,EAAE,SAC7B,iCAAqB,EAAE,MAAMA,OAAM,GAAG,CAAC;AACzC,mBAAO,mCAAoB;AAAA,UACzB,MAAM,SAAS;AAAA,UACf,MAAM,SAAS;AAAA,UACf,cAAc,SAAS;AAAA,QACzB,CAAC;AAAA,MACH,WAAWA,OAAM,GAAG,SAAS,SAAS,GAAG;AACvC,cAAM,WACJ,wCAAyBA,OAAM,EAAE,SACjC,sCAAsB,EAAE,MAAMA,OAAM,GAAG,CAAC;AAC1C,mBAAO,mCAAoB;AAAA,UACzB,MAAM,SAAS;AAAA,UACf,MAAM,SAAS;AAAA,UACf,cAAc,SAAS;AAAA,QACzB,CAAC;AAAA,MACH,WAAWA,OAAM,GAAG,SAAS,YAAY,GAAG;AAC1C,cAAM,WACJ,gCAAqBA,OAAM,EAAE,SAC7B,wCAA4B,EAAE,MAAMA,OAAM,GAAG,CAAC;AAChD,mBAAO,mCAAoB;AAAA,UACzB,MAAM,SAAS;AAAA,UACf,MAAM,SAAS;AAAA,UACf,cAAc,SAAS;AAAA,QACzB,CAAC;AAAA,MACH,OAAO;AACL,cAAM,WACJ,gCAAqBA,OAAM,EAAE,SAAK,2BAAe,EAAE,MAAMA,OAAM,GAAG,CAAC;AACrE,mBAAO,mCAAoB;AAAA,UACzB,MAAM,SAAS;AAAA,UACf,MAAM,SAAS;AAAA,UACf,cAAc,SAAS;AAAA,QACzB,CAAC;AAAA,MACH;AAAA,IACF,CAAC;AAAA,EACH;AACF;AAEO,SAAS,aAAa,SAA+C;AAC1E,QAAM,gBAAgB,EAAE,MAAM,UAAU,GAAG,QAAQ;AACnD,aAAO,2BAAiB;AAAA,IACtB,MAAM;AAAA,IACN,GAAG;AAAA,IACH,aAAa,OAAO,WAAW;AAC7B,YAAM,SAAS,CAAC;AAChB,aAAO;AAAA,QACL,GAAG,OAAO,OAAO,+BAAoB,EAAE;AAAA,UAAI,CAAC,iBAC1C,sCAAwB;AAAA,YACtB,MAAM,SAAS;AAAA,YACf;AAAA,YACA;AAAA,YACA;AAAA,UACF,CAAC;AAAA,QACH;AAAA,MACF;AACA,aAAO;AAAA,QACL,GAAG,OAAO,OAAO,2CAA0B,EAAE;AAAA,UAAI,CAACC,qBAChD,4CAA2B;AAAA,YACzB,MAAMA,aAAY;AAAA,YAClB;AAAA,YACA;AAAA,YACA,aAAAA;AAAA,UACF,CAAC;AAAA,QACH;AAAA,MACF;AACA,aAAO;AAAA,QACL,GAAG,OAAO,OAAO,+BAAoB,EAAE;AAAA,UAAI,CAAC,iBAC1C,4CAA8B;AAAA,YAC5B,MAAM,SAAS;AAAA,YACf;AAAA,YACA;AAAA,YACA;AAAA,UACF,CAAC;AAAA,QACH;AAAA,MACF;AACA,aAAO;AAAA,QACL,GAAG,OAAO,OAAO,uCAAwB,EAAE;AAAA,UAAI,CAAC,iBAC9C,yCAAyB;AAAA,YACvB,MAAM,SAAS;AAAA,YACf;AAAA,YACA;AAAA,YACA;AAAA,UACF,CAAC;AAAA,QACH;AAAA,MACF;AACA,aAAO;AAAA,QACL,GAAG,OAAO,OAAO,+BAAoB,EAAE;AAAA,UAAI,CAAC,iBAC1C,mDAAqC;AAAA,YACnC,MAAM,SAAS;AAAA,YACf;AAAA,YACA;AAAA,YACA;AAAA,UACF,CAAC;AAAA,QACH;AAAA,MACF;AACA,aAAO;AAAA,QACL,GAAG,OAAO,OAAO,mCAAsB,EAAE;AAAA,UAAI,CAAC,iBAC5C,2CAA6B;AAAA,YAC3B,MAAM,SAAS;AAAA,YACf;AAAA,YACA;AAAA,YACA;AAAA,YACA,gBAAgB,SAAS,KAAK,SAAS,aAAa,IAChD,uCACA;AAAA,UACN,CAAC;AAAA,QACH;AAAA,MACF;AACA,aAAO;AAAA,IACT;AAAA,IACA,UAAU,eAAe,aAAa;AAAA,IACtC;AAAA,EACF,CAAC;AACH;AA2CA,MAAM,QAAS,CAAC,MAAc,WAA+C;AAC3E,MAAI,KAAK,SAAS,aAAa,KAAK,KAAK,SAAS,QAAQ,GAAG;AAC3D,eAAO,kCAAoB;AAAA,MACzB;AAAA,MACA;AAAA,IACF,CAAC;AAAA,EACH;AACA,MAAI,KAAK,SAAS,KAAK,GAAG;AACxB,eAAO,iCAAqB;AAAA,MAC1B;AAAA,MACA;AAAA,IACF,CAAC;AAAA,EACH;AACA,MAAI,KAAK,SAAS,SAAS,GAAG;AAC5B,eAAO,sCAAsB;AAAA,MAC3B;AAAA,MACA;AAAA,IACF,CAAC;AAAA,EACH;AACA,MAAI,KAAK,SAAS,YAAY,GAAG;AAC/B,eAAO,wCAA4B;AAAA,MACjC;AAAA,MACA;AAAA,IACF,CAAC;AAAA,EACH;AACA,aAAO,2BAAe;AAAA,IACpB;AAAA,IACA;AAAA,EACF,CAAC;AACH;AAEA,MAAM,WAAY,CAChB,MACA,WACoC;AACpC,aAAO,2BAAY;AAAA,IACjB;AAAA,IACA;AAAA,IACA,cAAc;AAAA,IACd,WAAW;AAAA,EACb,CAAC;AACH;AA+BO,MAAM,SAAuB,OAAO,OAAO,cAAc;AAAA,EAC9D;AAAA,EACA;AACF,CAAC;AAED,IAAO,iBAAQ;","names":["import_embedder","model","embedderRef"]}
+{"version":3,"sources":["../../src/openai/index.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\nimport {\n  ActionMetadata,\n  embedderActionMetadata,\n  embedderRef,\n  EmbedderReference,\n  modelActionMetadata,\n  ModelReference,\n  z,\n} from 'genkit';\nimport { ResolvableAction, type GenkitPluginV2 } from 'genkit/plugin';\nimport { ActionType } from 'genkit/registry';\nimport OpenAI from 'openai';\nimport {\n  defineCompatOpenAISpeechModel,\n  defineCompatOpenAITranscriptionModel,\n  SpeechConfigSchema,\n  TranscriptionConfigSchema,\n} from '../audio.js';\nimport { defineCompatOpenAIEmbedder } from '../embedder.js';\nimport {\n  defineCompatOpenAIImageModel,\n  ImageGenerationCommonConfigSchema,\n} from '../image.js';\nimport { openAICompatible, PluginOptions } from '../index.js';\nimport { defineCompatOpenAIModel } from '../model.js';\nimport {\n  gptImage1RequestBuilder,\n  openAIImageModelRef,\n  SUPPORTED_IMAGE_MODELS,\n} from './dalle.js';\nimport {\n  SUPPORTED_EMBEDDING_MODELS,\n  TextEmbeddingConfigSchema,\n} from './embedder.js';\nimport {\n  OpenAIChatCompletionConfigSchema,\n  openAIModelRef,\n  SUPPORTED_GPT_MODELS,\n} from './gpt.js';\nimport { openAITranscriptionModelRef, SUPPORTED_STT_MODELS } from './stt.js';\nimport { openAISpeechModelRef, SUPPORTED_TTS_MODELS } from './tts.js';\nimport {\n  defineOpenAIWhisperModel,\n  openAIWhisperModelRef,\n  SUPPORTED_WHISPER_MODELS,\n  WhisperConfigSchema,\n} from './whisper.js';\n\nexport type OpenAIPluginOptions = Omit<PluginOptions, 'name' | 'baseURL'>;\n\nconst UNSUPPORTED_MODEL_MATCHERS = ['babbage', 'davinci', 'codex'];\n\nfunction createResolver(pluginOptions: PluginOptions) {\n  return async (client: OpenAI, actionType: ActionType, actionName: string) => {\n    if (actionType === 'embedder') {\n      return defineCompatOpenAIEmbedder({\n        name: actionName,\n        client,\n        pluginOptions,\n      });\n    } else if (\n      actionName.includes('gpt-image-1') ||\n      actionName.includes('dall-e')\n    ) {\n      const modelRef = openAIImageModelRef({ name: actionName });\n      return defineCompatOpenAIImageModel({\n        name: modelRef.name,\n        client,\n        pluginOptions,\n        modelRef,\n      });\n    } else if (actionName.includes('tts')) {\n      const modelRef = openAISpeechModelRef({ name: actionName });\n      return defineCompatOpenAISpeechModel({\n        name: modelRef.name,\n        client,\n        pluginOptions,\n        modelRef,\n      });\n    } else if (actionName.includes('whisper')) {\n      const modelRef = openAIWhisperModelRef({ name: actionName });\n      return defineOpenAIWhisperModel({\n        name: modelRef.name,\n        client,\n        pluginOptions,\n        modelRef,\n      });\n    } else if (actionName.includes('transcribe')) {\n      const modelRef = openAITranscriptionModelRef({\n        name: actionName,\n      });\n      return defineCompatOpenAITranscriptionModel({\n        name: modelRef.name,\n        client,\n        pluginOptions,\n        modelRef,\n      });\n    } else {\n      const modelRef = openAIModelRef({ name: actionName });\n      return defineCompatOpenAIModel({\n        name: modelRef.name,\n        client,\n        pluginOptions,\n        modelRef,\n      });\n    }\n  };\n}\n\nfunction filterOpenAiModels(model: OpenAI.Model): boolean {\n  return !UNSUPPORTED_MODEL_MATCHERS.some((m) => model.id.includes(m));\n}\n\nconst listActions = async (client: OpenAI): Promise<ActionMetadata[]> => {\n  return await client.models.list().then((response) =>\n    response.data.filter(filterOpenAiModels).map((model: OpenAI.Model) => {\n      if (model.id.includes('embedding')) {\n        return embedderActionMetadata({\n          name: model.id,\n          configSchema: TextEmbeddingConfigSchema,\n          info: SUPPORTED_EMBEDDING_MODELS[model.id]?.info,\n        });\n      } else if (\n        model.id.includes('gpt-image-1') ||\n        model.id.includes('dall-e')\n      ) {\n        const modelRef =\n          SUPPORTED_IMAGE_MODELS[model.id] ??\n          openAIImageModelRef({ name: model.id });\n        return modelActionMetadata({\n          name: modelRef.name,\n          info: modelRef.info,\n          configSchema: modelRef.configSchema,\n        });\n      } else if (model.id.includes('tts')) {\n        const modelRef =\n          SUPPORTED_TTS_MODELS[model.id] ??\n          openAISpeechModelRef({ name: model.id });\n        return modelActionMetadata({\n          name: modelRef.name,\n          info: modelRef.info,\n          configSchema: modelRef.configSchema,\n        });\n      } else if (model.id.includes('whisper')) {\n        const modelRef =\n          SUPPORTED_WHISPER_MODELS[model.id] ??\n          openAIWhisperModelRef({ name: model.id });\n        return modelActionMetadata({\n          name: modelRef.name,\n          info: modelRef.info,\n          configSchema: modelRef.configSchema,\n        });\n      } else if (model.id.includes('transcribe')) {\n        const modelRef =\n          SUPPORTED_STT_MODELS[model.id] ??\n          openAITranscriptionModelRef({ name: model.id });\n        return modelActionMetadata({\n          name: modelRef.name,\n          info: modelRef.info,\n          configSchema: modelRef.configSchema,\n        });\n      } else {\n        const modelRef =\n          SUPPORTED_GPT_MODELS[model.id] ?? openAIModelRef({ name: model.id });\n        return modelActionMetadata({\n          name: modelRef.name,\n          info: modelRef.info,\n          configSchema: modelRef.configSchema,\n        });\n      }\n    })\n  );\n};\n\nexport function openAIPlugin(options?: OpenAIPluginOptions): GenkitPluginV2 {\n  const pluginOptions = { name: 'openai', ...options };\n  return openAICompatible({\n    name: 'openai',\n    ...options,\n    initializer: async (client) => {\n      const models = [] as ResolvableAction[];\n      models.push(\n        ...Object.values(SUPPORTED_GPT_MODELS).map((modelRef) =>\n          defineCompatOpenAIModel({\n            name: modelRef.name,\n            client,\n            pluginOptions,\n            modelRef,\n          })\n        )\n      );\n      models.push(\n        ...Object.values(SUPPORTED_EMBEDDING_MODELS).map((embedderRef) =>\n          defineCompatOpenAIEmbedder({\n            name: embedderRef.name,\n            client,\n            pluginOptions,\n            embedderRef,\n          })\n        )\n      );\n      models.push(\n        ...Object.values(SUPPORTED_TTS_MODELS).map((modelRef) =>\n          defineCompatOpenAISpeechModel({\n            name: modelRef.name,\n            client,\n            pluginOptions,\n            modelRef,\n          })\n        )\n      );\n      models.push(\n        ...Object.values(SUPPORTED_WHISPER_MODELS).map((modelRef) =>\n          defineOpenAIWhisperModel({\n            name: modelRef.name,\n            client,\n            pluginOptions,\n            modelRef,\n          })\n        )\n      );\n      models.push(\n        ...Object.values(SUPPORTED_STT_MODELS).map((modelRef) =>\n          defineCompatOpenAITranscriptionModel({\n            name: modelRef.name,\n            client,\n            pluginOptions,\n            modelRef,\n          })\n        )\n      );\n      models.push(\n        ...Object.values(SUPPORTED_IMAGE_MODELS).map((modelRef) =>\n          defineCompatOpenAIImageModel({\n            name: modelRef.name,\n            client,\n            pluginOptions,\n            modelRef,\n            requestBuilder: modelRef.name.includes('gpt-image-1')\n              ? gptImage1RequestBuilder\n              : undefined,\n          })\n        )\n      );\n      return models;\n    },\n    resolver: createResolver(pluginOptions),\n    listActions,\n  });\n}\n\nexport type OpenAIPlugin = {\n  (params?: OpenAIPluginOptions): GenkitPluginV2;\n  model(\n    name:\n      | keyof typeof SUPPORTED_IMAGE_MODELS\n      | (`dall-e${string}` & {})\n      | (`gpt-image-${string}` & {}),\n    config?: z.infer<typeof ImageGenerationCommonConfigSchema>\n  ): ModelReference<typeof ImageGenerationCommonConfigSchema>;\n  model(\n    name:\n      | keyof typeof SUPPORTED_TTS_MODELS\n      | (`tts-${string}` & {})\n      | (`${string}-tts` & {}),\n    config?: z.infer<typeof SpeechConfigSchema>\n  ): ModelReference<typeof SpeechConfigSchema>;\n  model(\n    name: keyof typeof SUPPORTED_WHISPER_MODELS | (`whisper-${string}` & {}),\n    config?: z.infer<typeof WhisperConfigSchema>\n  ): ModelReference<typeof WhisperConfigSchema>;\n  model(\n    name: keyof typeof SUPPORTED_STT_MODELS | (`${string}-transcribe` & {}),\n    config?: z.infer<typeof TranscriptionConfigSchema>\n  ): ModelReference<typeof TranscriptionConfigSchema>;\n  model(\n    name:\n      | keyof typeof SUPPORTED_GPT_MODELS\n      | (`gpt-${string}` & {})\n      | (`o${number}` & {}),\n    config?: z.infer<typeof OpenAIChatCompletionConfigSchema>\n  ): ModelReference<typeof OpenAIChatCompletionConfigSchema>;\n  model(name: string, config?: any): ModelReference<z.ZodTypeAny>;\n  embedder(\n    name:\n      | keyof typeof SUPPORTED_EMBEDDING_MODELS\n      | (`${string}-embedding-${string}` & {}),\n    config?: z.infer<typeof TextEmbeddingConfigSchema>\n  ): EmbedderReference<typeof TextEmbeddingConfigSchema>;\n  embedder(name: string, config?: any): EmbedderReference<z.ZodTypeAny>;\n};\n\nconst model = ((name: string, config?: any): ModelReference<z.ZodTypeAny> => {\n  if (name.includes('gpt-image-1') || name.includes('dall-e')) {\n    return openAIImageModelRef({\n      name,\n      config,\n    });\n  }\n  if (name.includes('tts')) {\n    return openAISpeechModelRef({\n      name,\n      config,\n    });\n  }\n  if (name.includes('whisper')) {\n    return openAIWhisperModelRef({\n      name,\n      config,\n    });\n  }\n  if (name.includes('transcribe')) {\n    return openAITranscriptionModelRef({\n      name,\n      config,\n    });\n  }\n  return openAIModelRef({\n    name,\n    config,\n  });\n}) as OpenAIPlugin['model'];\n\nconst embedder = ((\n  name: string,\n  config?: any\n): EmbedderReference<z.ZodTypeAny> => {\n  return embedderRef({\n    name,\n    config,\n    configSchema: TextEmbeddingConfigSchema,\n    namespace: 'openai',\n  });\n}) as OpenAIPlugin['embedder'];\n\n/**\n * This module provides an interface to the OpenAI 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 `openai` 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 * - openai: The main plugin function to interact with OpenAI.\n *\n * Usage:\n * To use the models, initialize the openai plugin inside `configureGenkit` and\n * pass the configuration options. If no API key is provided in the options, the\n * environment variable `OPENAI_API_KEY` must be set.\n *\n * Example:\n * ```\n * import { openAI } from '@genkit-ai/compat-oai/openai';\n *\n * export default configureGenkit({\n *  plugins: [\n *    openai()\n *    ... // other plugins\n *  ]\n * });\n * ```\n */\nexport const openAI: OpenAIPlugin = Object.assign(openAIPlugin, {\n  model,\n  embedder,\n});\n\nexport default openAI;\n"],"mappings":";;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAiBA,oBAQO;AAIP,mBAKO;AACP,sBAA2C;AAC3C,mBAGO;AACP,eAAgD;AAChD,mBAAwC;AACxC,mBAIO;AACP,IAAAA,mBAGO;AACP,iBAIO;AACP,iBAAkE;AAClE,iBAA2D;AAC3D,qBAKO;AAIP,MAAM,6BAA6B,CAAC,WAAW,WAAW,OAAO;AAEjE,SAAS,eAAe,eAA8B;AACpD,SAAO,OAAO,QAAgB,YAAwB,eAAuB;AAC3E,QAAI,eAAe,YAAY;AAC7B,iBAAO,4CAA2B;AAAA,QAChC,MAAM;AAAA,QACN;AAAA,QACA;AAAA,MACF,CAAC;AAAA,IACH,WACE,WAAW,SAAS,aAAa,KACjC,WAAW,SAAS,QAAQ,GAC5B;AACA,YAAM,eAAW,kCAAoB,EAAE,MAAM,WAAW,CAAC;AACzD,iBAAO,2CAA6B;AAAA,QAClC,MAAM,SAAS;AAAA,QACf;AAAA,QACA;AAAA,QACA;AAAA,MACF,CAAC;AAAA,IACH,WAAW,WAAW,SAAS,KAAK,GAAG;AACrC,YAAM,eAAW,iCAAqB,EAAE,MAAM,WAAW,CAAC;AAC1D,iBAAO,4CAA8B;AAAA,QACnC,MAAM,SAAS;AAAA,QACf;AAAA,QACA;AAAA,QACA;AAAA,MACF,CAAC;AAAA,IACH,WAAW,WAAW,SAAS,SAAS,GAAG;AACzC,YAAM,eAAW,sCAAsB,EAAE,MAAM,WAAW,CAAC;AAC3D,iBAAO,yCAAyB;AAAA,QAC9B,MAAM,SAAS;AAAA,QACf;AAAA,QACA;AAAA,QACA;AAAA,MACF,CAAC;AAAA,IACH,WAAW,WAAW,SAAS,YAAY,GAAG;AAC5C,YAAM,eAAW,wCAA4B;AAAA,QAC3C,MAAM;AAAA,MACR,CAAC;AACD,iBAAO,mDAAqC;AAAA,QAC1C,MAAM,SAAS;AAAA,QACf;AAAA,QACA;AAAA,QACA;AAAA,MACF,CAAC;AAAA,IACH,OAAO;AACL,YAAM,eAAW,2BAAe,EAAE,MAAM,WAAW,CAAC;AACpD,iBAAO,sCAAwB;AAAA,QAC7B,MAAM,SAAS;AAAA,QACf;AAAA,QACA;AAAA,QACA;AAAA,MACF,CAAC;AAAA,IACH;AAAA,EACF;AACF;AAEA,SAAS,mBAAmBC,QAA8B;AACxD,SAAO,CAAC,2BAA2B,KAAK,CAAC,MAAMA,OAAM,GAAG,SAAS,CAAC,CAAC;AACrE;AAEA,MAAM,cAAc,OAAO,WAA8C;AACvE,SAAO,MAAM,OAAO,OAAO,KAAK,EAAE;AAAA,IAAK,CAAC,aACtC,SAAS,KAAK,OAAO,kBAAkB,EAAE,IAAI,CAACA,WAAwB;AACpE,UAAIA,OAAM,GAAG,SAAS,WAAW,GAAG;AAClC,mBAAO,sCAAuB;AAAA,UAC5B,MAAMA,OAAM;AAAA,UACZ,cAAc;AAAA,UACd,MAAM,4CAA2BA,OAAM,EAAE,GAAG;AAAA,QAC9C,CAAC;AAAA,MACH,WACEA,OAAM,GAAG,SAAS,aAAa,KAC/BA,OAAM,GAAG,SAAS,QAAQ,GAC1B;AACA,cAAM,WACJ,oCAAuBA,OAAM,EAAE,SAC/B,kCAAoB,EAAE,MAAMA,OAAM,GAAG,CAAC;AACxC,mBAAO,mCAAoB;AAAA,UACzB,MAAM,SAAS;AAAA,UACf,MAAM,SAAS;AAAA,UACf,cAAc,SAAS;AAAA,QACzB,CAAC;AAAA,MACH,WAAWA,OAAM,GAAG,SAAS,KAAK,GAAG;AACnC,cAAM,WACJ,gCAAqBA,OAAM,EAAE,SAC7B,iCAAqB,EAAE,MAAMA,OAAM,GAAG,CAAC;AACzC,mBAAO,mCAAoB;AAAA,UACzB,MAAM,SAAS;AAAA,UACf,MAAM,SAAS;AAAA,UACf,cAAc,SAAS;AAAA,QACzB,CAAC;AAAA,MACH,WAAWA,OAAM,GAAG,SAAS,SAAS,GAAG;AACvC,cAAM,WACJ,wCAAyBA,OAAM,EAAE,SACjC,sCAAsB,EAAE,MAAMA,OAAM,GAAG,CAAC;AAC1C,mBAAO,mCAAoB;AAAA,UACzB,MAAM,SAAS;AAAA,UACf,MAAM,SAAS;AAAA,UACf,cAAc,SAAS;AAAA,QACzB,CAAC;AAAA,MACH,WAAWA,OAAM,GAAG,SAAS,YAAY,GAAG;AAC1C,cAAM,WACJ,gCAAqBA,OAAM,EAAE,SAC7B,wCAA4B,EAAE,MAAMA,OAAM,GAAG,CAAC;AAChD,mBAAO,mCAAoB;AAAA,UACzB,MAAM,SAAS;AAAA,UACf,MAAM,SAAS;AAAA,UACf,cAAc,SAAS;AAAA,QACzB,CAAC;AAAA,MACH,OAAO;AACL,cAAM,WACJ,gCAAqBA,OAAM,EAAE,SAAK,2BAAe,EAAE,MAAMA,OAAM,GAAG,CAAC;AACrE,mBAAO,mCAAoB;AAAA,UACzB,MAAM,SAAS;AAAA,UACf,MAAM,SAAS;AAAA,UACf,cAAc,SAAS;AAAA,QACzB,CAAC;AAAA,MACH;AAAA,IACF,CAAC;AAAA,EACH;AACF;AAEO,SAAS,aAAa,SAA+C;AAC1E,QAAM,gBAAgB,EAAE,MAAM,UAAU,GAAG,QAAQ;AACnD,aAAO,2BAAiB;AAAA,IACtB,MAAM;AAAA,IACN,GAAG;AAAA,IACH,aAAa,OAAO,WAAW;AAC7B,YAAM,SAAS,CAAC;AAChB,aAAO;AAAA,QACL,GAAG,OAAO,OAAO,+BAAoB,EAAE;AAAA,UAAI,CAAC,iBAC1C,sCAAwB;AAAA,YACtB,MAAM,SAAS;AAAA,YACf;AAAA,YACA;AAAA,YACA;AAAA,UACF,CAAC;AAAA,QACH;AAAA,MACF;AACA,aAAO;AAAA,QACL,GAAG,OAAO,OAAO,2CAA0B,EAAE;AAAA,UAAI,CAACC,qBAChD,4CAA2B;AAAA,YACzB,MAAMA,aAAY;AAAA,YAClB;AAAA,YACA;AAAA,YACA,aAAAA;AAAA,UACF,CAAC;AAAA,QACH;AAAA,MACF;AACA,aAAO;AAAA,QACL,GAAG,OAAO,OAAO,+BAAoB,EAAE;AAAA,UAAI,CAAC,iBAC1C,4CAA8B;AAAA,YAC5B,MAAM,SAAS;AAAA,YACf;AAAA,YACA;AAAA,YACA;AAAA,UACF,CAAC;AAAA,QACH;AAAA,MACF;AACA,aAAO;AAAA,QACL,GAAG,OAAO,OAAO,uCAAwB,EAAE;AAAA,UAAI,CAAC,iBAC9C,yCAAyB;AAAA,YACvB,MAAM,SAAS;AAAA,YACf;AAAA,YACA;AAAA,YACA;AAAA,UACF,CAAC;AAAA,QACH;AAAA,MACF;AACA,aAAO;AAAA,QACL,GAAG,OAAO,OAAO,+BAAoB,EAAE;AAAA,UAAI,CAAC,iBAC1C,mDAAqC;AAAA,YACnC,MAAM,SAAS;AAAA,YACf;AAAA,YACA;AAAA,YACA;AAAA,UACF,CAAC;AAAA,QACH;AAAA,MACF;AACA,aAAO;AAAA,QACL,GAAG,OAAO,OAAO,mCAAsB,EAAE;AAAA,UAAI,CAAC,iBAC5C,2CAA6B;AAAA,YAC3B,MAAM,SAAS;AAAA,YACf;AAAA,YACA;AAAA,YACA;AAAA,YACA,gBAAgB,SAAS,KAAK,SAAS,aAAa,IAChD,uCACA;AAAA,UACN,CAAC;AAAA,QACH;AAAA,MACF;AACA,aAAO;AAAA,IACT;AAAA,IACA,UAAU,eAAe,aAAa;AAAA,IACtC;AAAA,EACF,CAAC;AACH;AA2CA,MAAM,SAAS,CAAC,MAAc,WAA+C;AAC3E,MAAI,KAAK,SAAS,aAAa,KAAK,KAAK,SAAS,QAAQ,GAAG;AAC3D,eAAO,kCAAoB;AAAA,MACzB;AAAA,MACA;AAAA,IACF,CAAC;AAAA,EACH;AACA,MAAI,KAAK,SAAS,KAAK,GAAG;AACxB,eAAO,iCAAqB;AAAA,MAC1B;AAAA,MACA;AAAA,IACF,CAAC;AAAA,EACH;AACA,MAAI,KAAK,SAAS,SAAS,GAAG;AAC5B,eAAO,sCAAsB;AAAA,MAC3B;AAAA,MACA;AAAA,IACF,CAAC;AAAA,EACH;AACA,MAAI,KAAK,SAAS,YAAY,GAAG;AAC/B,eAAO,wCAA4B;AAAA,MACjC;AAAA,MACA;AAAA,IACF,CAAC;AAAA,EACH;AACA,aAAO,2BAAe;AAAA,IACpB;AAAA,IACA;AAAA,EACF,CAAC;AACH;AAEA,MAAM,YAAY,CAChB,MACA,WACoC;AACpC,aAAO,2BAAY;AAAA,IACjB;AAAA,IACA;AAAA,IACA,cAAc;AAAA,IACd,WAAW;AAAA,EACb,CAAC;AACH;AA+BO,MAAM,SAAuB,OAAO,OAAO,cAAc;AAAA,EAC9D;AAAA,EACA;AACF,CAAC;AAED,IAAO,iBAAQ;","names":["import_embedder","model","embedderRef"]}

package/lib/openai/index.mjs

@@ -212,7 +212,7 @@ function openAIPlugin(options) {
     listActions
   });
 }
-const model = (name, config) => {
+const model = ((name, config) => {
   if (name.includes("gpt-image-1") || name.includes("dall-e")) {
     return openAIImageModelRef({
       name,
@@ -241,15 +241,15 @@ const model = (name, config) => {
     name,
     config
   });
-};
-const embedder = (name, config) => {
+});
+const embedder = ((name, config) => {
   return embedderRef({
     name,
     config,
     configSchema: TextEmbeddingConfigSchema,
     namespace: "openai"
   });
-};
+});
 const openAI = Object.assign(openAIPlugin, {
   model,
   embedder

package/lib/openai/index.mjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/openai/index.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\nimport {\n  ActionMetadata,\n  embedderActionMetadata,\n  embedderRef,\n  EmbedderReference,\n  modelActionMetadata,\n  ModelReference,\n  z,\n} from 'genkit';\nimport { ResolvableAction, type GenkitPluginV2 } from 'genkit/plugin';\nimport { ActionType } from 'genkit/registry';\nimport OpenAI from 'openai';\nimport {\n  defineCompatOpenAISpeechModel,\n  defineCompatOpenAITranscriptionModel,\n  SpeechConfigSchema,\n  TranscriptionConfigSchema,\n} from '../audio.js';\nimport { defineCompatOpenAIEmbedder } from '../embedder.js';\nimport {\n  defineCompatOpenAIImageModel,\n  ImageGenerationCommonConfigSchema,\n} from '../image.js';\nimport { openAICompatible, PluginOptions } from '../index.js';\nimport { defineCompatOpenAIModel } from '../model.js';\nimport {\n  gptImage1RequestBuilder,\n  openAIImageModelRef,\n  SUPPORTED_IMAGE_MODELS,\n} from './dalle.js';\nimport {\n  SUPPORTED_EMBEDDING_MODELS,\n  TextEmbeddingConfigSchema,\n} from './embedder.js';\nimport {\n  OpenAIChatCompletionConfigSchema,\n  openAIModelRef,\n  SUPPORTED_GPT_MODELS,\n} from './gpt.js';\nimport { openAITranscriptionModelRef, SUPPORTED_STT_MODELS } from './stt.js';\nimport { openAISpeechModelRef, SUPPORTED_TTS_MODELS } from './tts.js';\nimport {\n  defineOpenAIWhisperModel,\n  openAIWhisperModelRef,\n  SUPPORTED_WHISPER_MODELS,\n  WhisperConfigSchema,\n} from './whisper.js';\n\nexport type OpenAIPluginOptions = Omit<PluginOptions, 'name' | 'baseURL'>;\n\nconst UNSUPPORTED_MODEL_MATCHERS = ['babbage', 'davinci', 'codex'];\n\nfunction createResolver(pluginOptions: PluginOptions) {\n  return async (client: OpenAI, actionType: ActionType, actionName: string) => {\n    if (actionType === 'embedder') {\n      return defineCompatOpenAIEmbedder({\n        name: actionName,\n        client,\n        pluginOptions,\n      });\n    } else if (\n      actionName.includes('gpt-image-1') ||\n      actionName.includes('dall-e')\n    ) {\n      const modelRef = openAIImageModelRef({ name: actionName });\n      return defineCompatOpenAIImageModel({\n        name: modelRef.name,\n        client,\n        pluginOptions,\n        modelRef,\n      });\n    } else if (actionName.includes('tts')) {\n      const modelRef = openAISpeechModelRef({ name: actionName });\n      return defineCompatOpenAISpeechModel({\n        name: modelRef.name,\n        client,\n        pluginOptions,\n        modelRef,\n      });\n    } else if (actionName.includes('whisper')) {\n      const modelRef = openAIWhisperModelRef({ name: actionName });\n      return defineOpenAIWhisperModel({\n        name: modelRef.name,\n        client,\n        pluginOptions,\n        modelRef,\n      });\n    } else if (actionName.includes('transcribe')) {\n      const modelRef = openAITranscriptionModelRef({\n        name: actionName,\n      });\n      return defineCompatOpenAITranscriptionModel({\n        name: modelRef.name,\n        client,\n        pluginOptions,\n        modelRef,\n      });\n    } else {\n      const modelRef = openAIModelRef({ name: actionName });\n      return defineCompatOpenAIModel({\n        name: modelRef.name,\n        client,\n        pluginOptions,\n        modelRef,\n      });\n    }\n  };\n}\n\nfunction filterOpenAiModels(model: OpenAI.Model): boolean {\n  return !UNSUPPORTED_MODEL_MATCHERS.some((m) => model.id.includes(m));\n}\n\nconst listActions = async (client: OpenAI): Promise<ActionMetadata[]> => {\n  return await client.models.list().then((response) =>\n    response.data.filter(filterOpenAiModels).map((model: OpenAI.Model) => {\n      if (model.id.includes('embedding')) {\n        return embedderActionMetadata({\n          name: model.id,\n          configSchema: TextEmbeddingConfigSchema,\n          info: SUPPORTED_EMBEDDING_MODELS[model.id]?.info,\n        });\n      } else if (\n        model.id.includes('gpt-image-1') ||\n        model.id.includes('dall-e')\n      ) {\n        const modelRef =\n          SUPPORTED_IMAGE_MODELS[model.id] ??\n          openAIImageModelRef({ name: model.id });\n        return modelActionMetadata({\n          name: modelRef.name,\n          info: modelRef.info,\n          configSchema: modelRef.configSchema,\n        });\n      } else if (model.id.includes('tts')) {\n        const modelRef =\n          SUPPORTED_TTS_MODELS[model.id] ??\n          openAISpeechModelRef({ name: model.id });\n        return modelActionMetadata({\n          name: modelRef.name,\n          info: modelRef.info,\n          configSchema: modelRef.configSchema,\n        });\n      } else if (model.id.includes('whisper')) {\n        const modelRef =\n          SUPPORTED_WHISPER_MODELS[model.id] ??\n          openAIWhisperModelRef({ name: model.id });\n        return modelActionMetadata({\n          name: modelRef.name,\n          info: modelRef.info,\n          configSchema: modelRef.configSchema,\n        });\n      } else if (model.id.includes('transcribe')) {\n        const modelRef =\n          SUPPORTED_STT_MODELS[model.id] ??\n          openAITranscriptionModelRef({ name: model.id });\n        return modelActionMetadata({\n          name: modelRef.name,\n          info: modelRef.info,\n          configSchema: modelRef.configSchema,\n        });\n      } else {\n        const modelRef =\n          SUPPORTED_GPT_MODELS[model.id] ?? openAIModelRef({ name: model.id });\n        return modelActionMetadata({\n          name: modelRef.name,\n          info: modelRef.info,\n          configSchema: modelRef.configSchema,\n        });\n      }\n    })\n  );\n};\n\nexport function openAIPlugin(options?: OpenAIPluginOptions): GenkitPluginV2 {\n  const pluginOptions = { name: 'openai', ...options };\n  return openAICompatible({\n    name: 'openai',\n    ...options,\n    initializer: async (client) => {\n      const models = [] as ResolvableAction[];\n      models.push(\n        ...Object.values(SUPPORTED_GPT_MODELS).map((modelRef) =>\n          defineCompatOpenAIModel({\n            name: modelRef.name,\n            client,\n            pluginOptions,\n            modelRef,\n          })\n        )\n      );\n      models.push(\n        ...Object.values(SUPPORTED_EMBEDDING_MODELS).map((embedderRef) =>\n          defineCompatOpenAIEmbedder({\n            name: embedderRef.name,\n            client,\n            pluginOptions,\n            embedderRef,\n          })\n        )\n      );\n      models.push(\n        ...Object.values(SUPPORTED_TTS_MODELS).map((modelRef) =>\n          defineCompatOpenAISpeechModel({\n            name: modelRef.name,\n            client,\n            pluginOptions,\n            modelRef,\n          })\n        )\n      );\n      models.push(\n        ...Object.values(SUPPORTED_WHISPER_MODELS).map((modelRef) =>\n          defineOpenAIWhisperModel({\n            name: modelRef.name,\n            client,\n            pluginOptions,\n            modelRef,\n          })\n        )\n      );\n      models.push(\n        ...Object.values(SUPPORTED_STT_MODELS).map((modelRef) =>\n          defineCompatOpenAITranscriptionModel({\n            name: modelRef.name,\n            client,\n            pluginOptions,\n            modelRef,\n          })\n        )\n      );\n      models.push(\n        ...Object.values(SUPPORTED_IMAGE_MODELS).map((modelRef) =>\n          defineCompatOpenAIImageModel({\n            name: modelRef.name,\n            client,\n            pluginOptions,\n            modelRef,\n            requestBuilder: modelRef.name.includes('gpt-image-1')\n              ? gptImage1RequestBuilder\n              : undefined,\n          })\n        )\n      );\n      return models;\n    },\n    resolver: createResolver(pluginOptions),\n    listActions,\n  });\n}\n\nexport type OpenAIPlugin = {\n  (params?: OpenAIPluginOptions): GenkitPluginV2;\n  model(\n    name:\n      | keyof typeof SUPPORTED_IMAGE_MODELS\n      | (`dall-e${string}` & {})\n      | (`gpt-image-${string}` & {}),\n    config?: z.infer<typeof ImageGenerationCommonConfigSchema>\n  ): ModelReference<typeof ImageGenerationCommonConfigSchema>;\n  model(\n    name:\n      | keyof typeof SUPPORTED_TTS_MODELS\n      | (`tts-${string}` & {})\n      | (`${string}-tts` & {}),\n    config?: z.infer<typeof SpeechConfigSchema>\n  ): ModelReference<typeof SpeechConfigSchema>;\n  model(\n    name: keyof typeof SUPPORTED_WHISPER_MODELS | (`whisper-${string}` & {}),\n    config?: z.infer<typeof WhisperConfigSchema>\n  ): ModelReference<typeof WhisperConfigSchema>;\n  model(\n    name: keyof typeof SUPPORTED_STT_MODELS | (`${string}-transcribe` & {}),\n    config?: z.infer<typeof TranscriptionConfigSchema>\n  ): ModelReference<typeof TranscriptionConfigSchema>;\n  model(\n    name:\n      | keyof typeof SUPPORTED_GPT_MODELS\n      | (`gpt-${string}` & {})\n      | (`o${number}` & {}),\n    config?: z.infer<typeof OpenAIChatCompletionConfigSchema>\n  ): ModelReference<typeof OpenAIChatCompletionConfigSchema>;\n  model(name: string, config?: any): ModelReference<z.ZodTypeAny>;\n  embedder(\n    name:\n      | keyof typeof SUPPORTED_EMBEDDING_MODELS\n      | (`${string}-embedding-${string}` & {}),\n    config?: z.infer<typeof TextEmbeddingConfigSchema>\n  ): EmbedderReference<typeof TextEmbeddingConfigSchema>;\n  embedder(name: string, config?: any): EmbedderReference<z.ZodTypeAny>;\n};\n\nconst model = ((name: string, config?: any): ModelReference<z.ZodTypeAny> => {\n  if (name.includes('gpt-image-1') || name.includes('dall-e')) {\n    return openAIImageModelRef({\n      name,\n      config,\n    });\n  }\n  if (name.includes('tts')) {\n    return openAISpeechModelRef({\n      name,\n      config,\n    });\n  }\n  if (name.includes('whisper')) {\n    return openAIWhisperModelRef({\n      name,\n      config,\n    });\n  }\n  if (name.includes('transcribe')) {\n    return openAITranscriptionModelRef({\n      name,\n      config,\n    });\n  }\n  return openAIModelRef({\n    name,\n    config,\n  });\n}) as OpenAIPlugin['model'];\n\nconst embedder = ((\n  name: string,\n  config?: any\n): EmbedderReference<z.ZodTypeAny> => {\n  return embedderRef({\n    name,\n    config,\n    configSchema: TextEmbeddingConfigSchema,\n    namespace: 'openai',\n  });\n}) as OpenAIPlugin['embedder'];\n\n/**\n * This module provides an interface to the OpenAI 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 `openai` 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 * - openai: The main plugin function to interact with OpenAI.\n *\n * Usage:\n * To use the models, initialize the openai plugin inside `configureGenkit` and\n * pass the configuration options. If no API key is provided in the options, the\n * environment variable `OPENAI_API_KEY` must be set.\n *\n * Example:\n * ```\n * import { openAI } from '@genkit-ai/compat-oai/openai';\n *\n * export default configureGenkit({\n *  plugins: [\n *    openai()\n *    ... // other plugins\n *  ]\n * });\n * ```\n */\nexport const openAI: OpenAIPlugin = Object.assign(openAIPlugin, {\n  model,\n  embedder,\n});\n\nexport default openAI;\n"],"mappings":"AAiBA;AAAA,EAEE;AAAA,EACA;AAAA,EAEA;AAAA,OAGK;AAIP;AAAA,EACE;AAAA,EACA;AAAA,OAGK;AACP,SAAS,kCAAkC;AAC3C;AAAA,EACE;AAAA,OAEK;AACP,SAAS,wBAAuC;AAChD,SAAS,+BAA+B;AACxC;AAAA,EACE;AAAA,EACA;AAAA,EACA;AAAA,OACK;AACP;AAAA,EACE;AAAA,EACA;AAAA,OACK;AACP;AAAA,EAEE;AAAA,EACA;AAAA,OACK;AACP,SAAS,6BAA6B,4BAA4B;AAClE,SAAS,sBAAsB,4BAA4B;AAC3D;AAAA,EACE;AAAA,EACA;AAAA,EACA;AAAA,OAEK;AAIP,MAAM,6BAA6B,CAAC,WAAW,WAAW,OAAO;AAEjE,SAAS,eAAe,eAA8B;AACpD,SAAO,OAAO,QAAgB,YAAwB,eAAuB;AAC3E,QAAI,eAAe,YAAY;AAC7B,aAAO,2BAA2B;AAAA,QAChC,MAAM;AAAA,QACN;AAAA,QACA;AAAA,MACF,CAAC;AAAA,IACH,WACE,WAAW,SAAS,aAAa,KACjC,WAAW,SAAS,QAAQ,GAC5B;AACA,YAAM,WAAW,oBAAoB,EAAE,MAAM,WAAW,CAAC;AACzD,aAAO,6BAA6B;AAAA,QAClC,MAAM,SAAS;AAAA,QACf;AAAA,QACA;AAAA,QACA;AAAA,MACF,CAAC;AAAA,IACH,WAAW,WAAW,SAAS,KAAK,GAAG;AACrC,YAAM,WAAW,qBAAqB,EAAE,MAAM,WAAW,CAAC;AAC1D,aAAO,8BAA8B;AAAA,QACnC,MAAM,SAAS;AAAA,QACf;AAAA,QACA;AAAA,QACA;AAAA,MACF,CAAC;AAAA,IACH,WAAW,WAAW,SAAS,SAAS,GAAG;AACzC,YAAM,WAAW,sBAAsB,EAAE,MAAM,WAAW,CAAC;AAC3D,aAAO,yBAAyB;AAAA,QAC9B,MAAM,SAAS;AAAA,QACf;AAAA,QACA;AAAA,QACA;AAAA,MACF,CAAC;AAAA,IACH,WAAW,WAAW,SAAS,YAAY,GAAG;AAC5C,YAAM,WAAW,4BAA4B;AAAA,QAC3C,MAAM;AAAA,MACR,CAAC;AACD,aAAO,qCAAqC;AAAA,QAC1C,MAAM,SAAS;AAAA,QACf;AAAA,QACA;AAAA,QACA;AAAA,MACF,CAAC;AAAA,IACH,OAAO;AACL,YAAM,WAAW,eAAe,EAAE,MAAM,WAAW,CAAC;AACpD,aAAO,wBAAwB;AAAA,QAC7B,MAAM,SAAS;AAAA,QACf;AAAA,QACA;AAAA,QACA;AAAA,MACF,CAAC;AAAA,IACH;AAAA,EACF;AACF;AAEA,SAAS,mBAAmBA,QAA8B;AACxD,SAAO,CAAC,2BAA2B,KAAK,CAAC,MAAMA,OAAM,GAAG,SAAS,CAAC,CAAC;AACrE;AAEA,MAAM,cAAc,OAAO,WAA8C;AACvE,SAAO,MAAM,OAAO,OAAO,KAAK,EAAE;AAAA,IAAK,CAAC,aACtC,SAAS,KAAK,OAAO,kBAAkB,EAAE,IAAI,CAACA,WAAwB;AACpE,UAAIA,OAAM,GAAG,SAAS,WAAW,GAAG;AAClC,eAAO,uBAAuB;AAAA,UAC5B,MAAMA,OAAM;AAAA,UACZ,cAAc;AAAA,UACd,MAAM,2BAA2BA,OAAM,EAAE,GAAG;AAAA,QAC9C,CAAC;AAAA,MACH,WACEA,OAAM,GAAG,SAAS,aAAa,KAC/BA,OAAM,GAAG,SAAS,QAAQ,GAC1B;AACA,cAAM,WACJ,uBAAuBA,OAAM,EAAE,KAC/B,oBAAoB,EAAE,MAAMA,OAAM,GAAG,CAAC;AACxC,eAAO,oBAAoB;AAAA,UACzB,MAAM,SAAS;AAAA,UACf,MAAM,SAAS;AAAA,UACf,cAAc,SAAS;AAAA,QACzB,CAAC;AAAA,MACH,WAAWA,OAAM,GAAG,SAAS,KAAK,GAAG;AACnC,cAAM,WACJ,qBAAqBA,OAAM,EAAE,KAC7B,qBAAqB,EAAE,MAAMA,OAAM,GAAG,CAAC;AACzC,eAAO,oBAAoB;AAAA,UACzB,MAAM,SAAS;AAAA,UACf,MAAM,SAAS;AAAA,UACf,cAAc,SAAS;AAAA,QACzB,CAAC;AAAA,MACH,WAAWA,OAAM,GAAG,SAAS,SAAS,GAAG;AACvC,cAAM,WACJ,yBAAyBA,OAAM,EAAE,KACjC,sBAAsB,EAAE,MAAMA,OAAM,GAAG,CAAC;AAC1C,eAAO,oBAAoB;AAAA,UACzB,MAAM,SAAS;AAAA,UACf,MAAM,SAAS;AAAA,UACf,cAAc,SAAS;AAAA,QACzB,CAAC;AAAA,MACH,WAAWA,OAAM,GAAG,SAAS,YAAY,GAAG;AAC1C,cAAM,WACJ,qBAAqBA,OAAM,EAAE,KAC7B,4BAA4B,EAAE,MAAMA,OAAM,GAAG,CAAC;AAChD,eAAO,oBAAoB;AAAA,UACzB,MAAM,SAAS;AAAA,UACf,MAAM,SAAS;AAAA,UACf,cAAc,SAAS;AAAA,QACzB,CAAC;AAAA,MACH,OAAO;AACL,cAAM,WACJ,qBAAqBA,OAAM,EAAE,KAAK,eAAe,EAAE,MAAMA,OAAM,GAAG,CAAC;AACrE,eAAO,oBAAoB;AAAA,UACzB,MAAM,SAAS;AAAA,UACf,MAAM,SAAS;AAAA,UACf,cAAc,SAAS;AAAA,QACzB,CAAC;AAAA,MACH;AAAA,IACF,CAAC;AAAA,EACH;AACF;AAEO,SAAS,aAAa,SAA+C;AAC1E,QAAM,gBAAgB,EAAE,MAAM,UAAU,GAAG,QAAQ;AACnD,SAAO,iBAAiB;AAAA,IACtB,MAAM;AAAA,IACN,GAAG;AAAA,IACH,aAAa,OAAO,WAAW;AAC7B,YAAM,SAAS,CAAC;AAChB,aAAO;AAAA,QACL,GAAG,OAAO,OAAO,oBAAoB,EAAE;AAAA,UAAI,CAAC,aAC1C,wBAAwB;AAAA,YACtB,MAAM,SAAS;AAAA,YACf;AAAA,YACA;AAAA,YACA;AAAA,UACF,CAAC;AAAA,QACH;AAAA,MACF;AACA,aAAO;AAAA,QACL,GAAG,OAAO,OAAO,0BAA0B,EAAE;AAAA,UAAI,CAACC,iBAChD,2BAA2B;AAAA,YACzB,MAAMA,aAAY;AAAA,YAClB;AAAA,YACA;AAAA,YACA,aAAAA;AAAA,UACF,CAAC;AAAA,QACH;AAAA,MACF;AACA,aAAO;AAAA,QACL,GAAG,OAAO,OAAO,oBAAoB,EAAE;AAAA,UAAI,CAAC,aAC1C,8BAA8B;AAAA,YAC5B,MAAM,SAAS;AAAA,YACf;AAAA,YACA;AAAA,YACA;AAAA,UACF,CAAC;AAAA,QACH;AAAA,MACF;AACA,aAAO;AAAA,QACL,GAAG,OAAO,OAAO,wBAAwB,EAAE;AAAA,UAAI,CAAC,aAC9C,yBAAyB;AAAA,YACvB,MAAM,SAAS;AAAA,YACf;AAAA,YACA;AAAA,YACA;AAAA,UACF,CAAC;AAAA,QACH;AAAA,MACF;AACA,aAAO;AAAA,QACL,GAAG,OAAO,OAAO,oBAAoB,EAAE;AAAA,UAAI,CAAC,aAC1C,qCAAqC;AAAA,YACnC,MAAM,SAAS;AAAA,YACf;AAAA,YACA;AAAA,YACA;AAAA,UACF,CAAC;AAAA,QACH;AAAA,MACF;AACA,aAAO;AAAA,QACL,GAAG,OAAO,OAAO,sBAAsB,EAAE;AAAA,UAAI,CAAC,aAC5C,6BAA6B;AAAA,YAC3B,MAAM,SAAS;AAAA,YACf;AAAA,YACA;AAAA,YACA;AAAA,YACA,gBAAgB,SAAS,KAAK,SAAS,aAAa,IAChD,0BACA;AAAA,UACN,CAAC;AAAA,QACH;AAAA,MACF;AACA,aAAO;AAAA,IACT;AAAA,IACA,UAAU,eAAe,aAAa;AAAA,IACtC;AAAA,EACF,CAAC;AACH;AA2CA,MAAM,QAAS,CAAC,MAAc,WAA+C;AAC3E,MAAI,KAAK,SAAS,aAAa,KAAK,KAAK,SAAS,QAAQ,GAAG;AAC3D,WAAO,oBAAoB;AAAA,MACzB;AAAA,MACA;AAAA,IACF,CAAC;AAAA,EACH;AACA,MAAI,KAAK,SAAS,KAAK,GAAG;AACxB,WAAO,qBAAqB;AAAA,MAC1B;AAAA,MACA;AAAA,IACF,CAAC;AAAA,EACH;AACA,MAAI,KAAK,SAAS,SAAS,GAAG;AAC5B,WAAO,sBAAsB;AAAA,MAC3B;AAAA,MACA;AAAA,IACF,CAAC;AAAA,EACH;AACA,MAAI,KAAK,SAAS,YAAY,GAAG;AAC/B,WAAO,4BAA4B;AAAA,MACjC;AAAA,MACA;AAAA,IACF,CAAC;AAAA,EACH;AACA,SAAO,eAAe;AAAA,IACpB;AAAA,IACA;AAAA,EACF,CAAC;AACH;AAEA,MAAM,WAAY,CAChB,MACA,WACoC;AACpC,SAAO,YAAY;AAAA,IACjB;AAAA,IACA;AAAA,IACA,cAAc;AAAA,IACd,WAAW;AAAA,EACb,CAAC;AACH;AA+BO,MAAM,SAAuB,OAAO,OAAO,cAAc;AAAA,EAC9D;AAAA,EACA;AACF,CAAC;AAED,IAAO,iBAAQ;","names":["model","embedderRef"]}
+{"version":3,"sources":["../../src/openai/index.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\nimport {\n  ActionMetadata,\n  embedderActionMetadata,\n  embedderRef,\n  EmbedderReference,\n  modelActionMetadata,\n  ModelReference,\n  z,\n} from 'genkit';\nimport { ResolvableAction, type GenkitPluginV2 } from 'genkit/plugin';\nimport { ActionType } from 'genkit/registry';\nimport OpenAI from 'openai';\nimport {\n  defineCompatOpenAISpeechModel,\n  defineCompatOpenAITranscriptionModel,\n  SpeechConfigSchema,\n  TranscriptionConfigSchema,\n} from '../audio.js';\nimport { defineCompatOpenAIEmbedder } from '../embedder.js';\nimport {\n  defineCompatOpenAIImageModel,\n  ImageGenerationCommonConfigSchema,\n} from '../image.js';\nimport { openAICompatible, PluginOptions } from '../index.js';\nimport { defineCompatOpenAIModel } from '../model.js';\nimport {\n  gptImage1RequestBuilder,\n  openAIImageModelRef,\n  SUPPORTED_IMAGE_MODELS,\n} from './dalle.js';\nimport {\n  SUPPORTED_EMBEDDING_MODELS,\n  TextEmbeddingConfigSchema,\n} from './embedder.js';\nimport {\n  OpenAIChatCompletionConfigSchema,\n  openAIModelRef,\n  SUPPORTED_GPT_MODELS,\n} from './gpt.js';\nimport { openAITranscriptionModelRef, SUPPORTED_STT_MODELS } from './stt.js';\nimport { openAISpeechModelRef, SUPPORTED_TTS_MODELS } from './tts.js';\nimport {\n  defineOpenAIWhisperModel,\n  openAIWhisperModelRef,\n  SUPPORTED_WHISPER_MODELS,\n  WhisperConfigSchema,\n} from './whisper.js';\n\nexport type OpenAIPluginOptions = Omit<PluginOptions, 'name' | 'baseURL'>;\n\nconst UNSUPPORTED_MODEL_MATCHERS = ['babbage', 'davinci', 'codex'];\n\nfunction createResolver(pluginOptions: PluginOptions) {\n  return async (client: OpenAI, actionType: ActionType, actionName: string) => {\n    if (actionType === 'embedder') {\n      return defineCompatOpenAIEmbedder({\n        name: actionName,\n        client,\n        pluginOptions,\n      });\n    } else if (\n      actionName.includes('gpt-image-1') ||\n      actionName.includes('dall-e')\n    ) {\n      const modelRef = openAIImageModelRef({ name: actionName });\n      return defineCompatOpenAIImageModel({\n        name: modelRef.name,\n        client,\n        pluginOptions,\n        modelRef,\n      });\n    } else if (actionName.includes('tts')) {\n      const modelRef = openAISpeechModelRef({ name: actionName });\n      return defineCompatOpenAISpeechModel({\n        name: modelRef.name,\n        client,\n        pluginOptions,\n        modelRef,\n      });\n    } else if (actionName.includes('whisper')) {\n      const modelRef = openAIWhisperModelRef({ name: actionName });\n      return defineOpenAIWhisperModel({\n        name: modelRef.name,\n        client,\n        pluginOptions,\n        modelRef,\n      });\n    } else if (actionName.includes('transcribe')) {\n      const modelRef = openAITranscriptionModelRef({\n        name: actionName,\n      });\n      return defineCompatOpenAITranscriptionModel({\n        name: modelRef.name,\n        client,\n        pluginOptions,\n        modelRef,\n      });\n    } else {\n      const modelRef = openAIModelRef({ name: actionName });\n      return defineCompatOpenAIModel({\n        name: modelRef.name,\n        client,\n        pluginOptions,\n        modelRef,\n      });\n    }\n  };\n}\n\nfunction filterOpenAiModels(model: OpenAI.Model): boolean {\n  return !UNSUPPORTED_MODEL_MATCHERS.some((m) => model.id.includes(m));\n}\n\nconst listActions = async (client: OpenAI): Promise<ActionMetadata[]> => {\n  return await client.models.list().then((response) =>\n    response.data.filter(filterOpenAiModels).map((model: OpenAI.Model) => {\n      if (model.id.includes('embedding')) {\n        return embedderActionMetadata({\n          name: model.id,\n          configSchema: TextEmbeddingConfigSchema,\n          info: SUPPORTED_EMBEDDING_MODELS[model.id]?.info,\n        });\n      } else if (\n        model.id.includes('gpt-image-1') ||\n        model.id.includes('dall-e')\n      ) {\n        const modelRef =\n          SUPPORTED_IMAGE_MODELS[model.id] ??\n          openAIImageModelRef({ name: model.id });\n        return modelActionMetadata({\n          name: modelRef.name,\n          info: modelRef.info,\n          configSchema: modelRef.configSchema,\n        });\n      } else if (model.id.includes('tts')) {\n        const modelRef =\n          SUPPORTED_TTS_MODELS[model.id] ??\n          openAISpeechModelRef({ name: model.id });\n        return modelActionMetadata({\n          name: modelRef.name,\n          info: modelRef.info,\n          configSchema: modelRef.configSchema,\n        });\n      } else if (model.id.includes('whisper')) {\n        const modelRef =\n          SUPPORTED_WHISPER_MODELS[model.id] ??\n          openAIWhisperModelRef({ name: model.id });\n        return modelActionMetadata({\n          name: modelRef.name,\n          info: modelRef.info,\n          configSchema: modelRef.configSchema,\n        });\n      } else if (model.id.includes('transcribe')) {\n        const modelRef =\n          SUPPORTED_STT_MODELS[model.id] ??\n          openAITranscriptionModelRef({ name: model.id });\n        return modelActionMetadata({\n          name: modelRef.name,\n          info: modelRef.info,\n          configSchema: modelRef.configSchema,\n        });\n      } else {\n        const modelRef =\n          SUPPORTED_GPT_MODELS[model.id] ?? openAIModelRef({ name: model.id });\n        return modelActionMetadata({\n          name: modelRef.name,\n          info: modelRef.info,\n          configSchema: modelRef.configSchema,\n        });\n      }\n    })\n  );\n};\n\nexport function openAIPlugin(options?: OpenAIPluginOptions): GenkitPluginV2 {\n  const pluginOptions = { name: 'openai', ...options };\n  return openAICompatible({\n    name: 'openai',\n    ...options,\n    initializer: async (client) => {\n      const models = [] as ResolvableAction[];\n      models.push(\n        ...Object.values(SUPPORTED_GPT_MODELS).map((modelRef) =>\n          defineCompatOpenAIModel({\n            name: modelRef.name,\n            client,\n            pluginOptions,\n            modelRef,\n          })\n        )\n      );\n      models.push(\n        ...Object.values(SUPPORTED_EMBEDDING_MODELS).map((embedderRef) =>\n          defineCompatOpenAIEmbedder({\n            name: embedderRef.name,\n            client,\n            pluginOptions,\n            embedderRef,\n          })\n        )\n      );\n      models.push(\n        ...Object.values(SUPPORTED_TTS_MODELS).map((modelRef) =>\n          defineCompatOpenAISpeechModel({\n            name: modelRef.name,\n            client,\n            pluginOptions,\n            modelRef,\n          })\n        )\n      );\n      models.push(\n        ...Object.values(SUPPORTED_WHISPER_MODELS).map((modelRef) =>\n          defineOpenAIWhisperModel({\n            name: modelRef.name,\n            client,\n            pluginOptions,\n            modelRef,\n          })\n        )\n      );\n      models.push(\n        ...Object.values(SUPPORTED_STT_MODELS).map((modelRef) =>\n          defineCompatOpenAITranscriptionModel({\n            name: modelRef.name,\n            client,\n            pluginOptions,\n            modelRef,\n          })\n        )\n      );\n      models.push(\n        ...Object.values(SUPPORTED_IMAGE_MODELS).map((modelRef) =>\n          defineCompatOpenAIImageModel({\n            name: modelRef.name,\n            client,\n            pluginOptions,\n            modelRef,\n            requestBuilder: modelRef.name.includes('gpt-image-1')\n              ? gptImage1RequestBuilder\n              : undefined,\n          })\n        )\n      );\n      return models;\n    },\n    resolver: createResolver(pluginOptions),\n    listActions,\n  });\n}\n\nexport type OpenAIPlugin = {\n  (params?: OpenAIPluginOptions): GenkitPluginV2;\n  model(\n    name:\n      | keyof typeof SUPPORTED_IMAGE_MODELS\n      | (`dall-e${string}` & {})\n      | (`gpt-image-${string}` & {}),\n    config?: z.infer<typeof ImageGenerationCommonConfigSchema>\n  ): ModelReference<typeof ImageGenerationCommonConfigSchema>;\n  model(\n    name:\n      | keyof typeof SUPPORTED_TTS_MODELS\n      | (`tts-${string}` & {})\n      | (`${string}-tts` & {}),\n    config?: z.infer<typeof SpeechConfigSchema>\n  ): ModelReference<typeof SpeechConfigSchema>;\n  model(\n    name: keyof typeof SUPPORTED_WHISPER_MODELS | (`whisper-${string}` & {}),\n    config?: z.infer<typeof WhisperConfigSchema>\n  ): ModelReference<typeof WhisperConfigSchema>;\n  model(\n    name: keyof typeof SUPPORTED_STT_MODELS | (`${string}-transcribe` & {}),\n    config?: z.infer<typeof TranscriptionConfigSchema>\n  ): ModelReference<typeof TranscriptionConfigSchema>;\n  model(\n    name:\n      | keyof typeof SUPPORTED_GPT_MODELS\n      | (`gpt-${string}` & {})\n      | (`o${number}` & {}),\n    config?: z.infer<typeof OpenAIChatCompletionConfigSchema>\n  ): ModelReference<typeof OpenAIChatCompletionConfigSchema>;\n  model(name: string, config?: any): ModelReference<z.ZodTypeAny>;\n  embedder(\n    name:\n      | keyof typeof SUPPORTED_EMBEDDING_MODELS\n      | (`${string}-embedding-${string}` & {}),\n    config?: z.infer<typeof TextEmbeddingConfigSchema>\n  ): EmbedderReference<typeof TextEmbeddingConfigSchema>;\n  embedder(name: string, config?: any): EmbedderReference<z.ZodTypeAny>;\n};\n\nconst model = ((name: string, config?: any): ModelReference<z.ZodTypeAny> => {\n  if (name.includes('gpt-image-1') || name.includes('dall-e')) {\n    return openAIImageModelRef({\n      name,\n      config,\n    });\n  }\n  if (name.includes('tts')) {\n    return openAISpeechModelRef({\n      name,\n      config,\n    });\n  }\n  if (name.includes('whisper')) {\n    return openAIWhisperModelRef({\n      name,\n      config,\n    });\n  }\n  if (name.includes('transcribe')) {\n    return openAITranscriptionModelRef({\n      name,\n      config,\n    });\n  }\n  return openAIModelRef({\n    name,\n    config,\n  });\n}) as OpenAIPlugin['model'];\n\nconst embedder = ((\n  name: string,\n  config?: any\n): EmbedderReference<z.ZodTypeAny> => {\n  return embedderRef({\n    name,\n    config,\n    configSchema: TextEmbeddingConfigSchema,\n    namespace: 'openai',\n  });\n}) as OpenAIPlugin['embedder'];\n\n/**\n * This module provides an interface to the OpenAI 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 `openai` 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 * - openai: The main plugin function to interact with OpenAI.\n *\n * Usage:\n * To use the models, initialize the openai plugin inside `configureGenkit` and\n * pass the configuration options. If no API key is provided in the options, the\n * environment variable `OPENAI_API_KEY` must be set.\n *\n * Example:\n * ```\n * import { openAI } from '@genkit-ai/compat-oai/openai';\n *\n * export default configureGenkit({\n *  plugins: [\n *    openai()\n *    ... // other plugins\n *  ]\n * });\n * ```\n */\nexport const openAI: OpenAIPlugin = Object.assign(openAIPlugin, {\n  model,\n  embedder,\n});\n\nexport default openAI;\n"],"mappings":"AAiBA;AAAA,EAEE;AAAA,EACA;AAAA,EAEA;AAAA,OAGK;AAIP;AAAA,EACE;AAAA,EACA;AAAA,OAGK;AACP,SAAS,kCAAkC;AAC3C;AAAA,EACE;AAAA,OAEK;AACP,SAAS,wBAAuC;AAChD,SAAS,+BAA+B;AACxC;AAAA,EACE;AAAA,EACA;AAAA,EACA;AAAA,OACK;AACP;AAAA,EACE;AAAA,EACA;AAAA,OACK;AACP;AAAA,EAEE;AAAA,EACA;AAAA,OACK;AACP,SAAS,6BAA6B,4BAA4B;AAClE,SAAS,sBAAsB,4BAA4B;AAC3D;AAAA,EACE;AAAA,EACA;AAAA,EACA;AAAA,OAEK;AAIP,MAAM,6BAA6B,CAAC,WAAW,WAAW,OAAO;AAEjE,SAAS,eAAe,eAA8B;AACpD,SAAO,OAAO,QAAgB,YAAwB,eAAuB;AAC3E,QAAI,eAAe,YAAY;AAC7B,aAAO,2BAA2B;AAAA,QAChC,MAAM;AAAA,QACN;AAAA,QACA;AAAA,MACF,CAAC;AAAA,IACH,WACE,WAAW,SAAS,aAAa,KACjC,WAAW,SAAS,QAAQ,GAC5B;AACA,YAAM,WAAW,oBAAoB,EAAE,MAAM,WAAW,CAAC;AACzD,aAAO,6BAA6B;AAAA,QAClC,MAAM,SAAS;AAAA,QACf;AAAA,QACA;AAAA,QACA;AAAA,MACF,CAAC;AAAA,IACH,WAAW,WAAW,SAAS,KAAK,GAAG;AACrC,YAAM,WAAW,qBAAqB,EAAE,MAAM,WAAW,CAAC;AAC1D,aAAO,8BAA8B;AAAA,QACnC,MAAM,SAAS;AAAA,QACf;AAAA,QACA;AAAA,QACA;AAAA,MACF,CAAC;AAAA,IACH,WAAW,WAAW,SAAS,SAAS,GAAG;AACzC,YAAM,WAAW,sBAAsB,EAAE,MAAM,WAAW,CAAC;AAC3D,aAAO,yBAAyB;AAAA,QAC9B,MAAM,SAAS;AAAA,QACf;AAAA,QACA;AAAA,QACA;AAAA,MACF,CAAC;AAAA,IACH,WAAW,WAAW,SAAS,YAAY,GAAG;AAC5C,YAAM,WAAW,4BAA4B;AAAA,QAC3C,MAAM;AAAA,MACR,CAAC;AACD,aAAO,qCAAqC;AAAA,QAC1C,MAAM,SAAS;AAAA,QACf;AAAA,QACA;AAAA,QACA;AAAA,MACF,CAAC;AAAA,IACH,OAAO;AACL,YAAM,WAAW,eAAe,EAAE,MAAM,WAAW,CAAC;AACpD,aAAO,wBAAwB;AAAA,QAC7B,MAAM,SAAS;AAAA,QACf;AAAA,QACA;AAAA,QACA;AAAA,MACF,CAAC;AAAA,IACH;AAAA,EACF;AACF;AAEA,SAAS,mBAAmBA,QAA8B;AACxD,SAAO,CAAC,2BAA2B,KAAK,CAAC,MAAMA,OAAM,GAAG,SAAS,CAAC,CAAC;AACrE;AAEA,MAAM,cAAc,OAAO,WAA8C;AACvE,SAAO,MAAM,OAAO,OAAO,KAAK,EAAE;AAAA,IAAK,CAAC,aACtC,SAAS,KAAK,OAAO,kBAAkB,EAAE,IAAI,CAACA,WAAwB;AACpE,UAAIA,OAAM,GAAG,SAAS,WAAW,GAAG;AAClC,eAAO,uBAAuB;AAAA,UAC5B,MAAMA,OAAM;AAAA,UACZ,cAAc;AAAA,UACd,MAAM,2BAA2BA,OAAM,EAAE,GAAG;AAAA,QAC9C,CAAC;AAAA,MACH,WACEA,OAAM,GAAG,SAAS,aAAa,KAC/BA,OAAM,GAAG,SAAS,QAAQ,GAC1B;AACA,cAAM,WACJ,uBAAuBA,OAAM,EAAE,KAC/B,oBAAoB,EAAE,MAAMA,OAAM,GAAG,CAAC;AACxC,eAAO,oBAAoB;AAAA,UACzB,MAAM,SAAS;AAAA,UACf,MAAM,SAAS;AAAA,UACf,cAAc,SAAS;AAAA,QACzB,CAAC;AAAA,MACH,WAAWA,OAAM,GAAG,SAAS,KAAK,GAAG;AACnC,cAAM,WACJ,qBAAqBA,OAAM,EAAE,KAC7B,qBAAqB,EAAE,MAAMA,OAAM,GAAG,CAAC;AACzC,eAAO,oBAAoB;AAAA,UACzB,MAAM,SAAS;AAAA,UACf,MAAM,SAAS;AAAA,UACf,cAAc,SAAS;AAAA,QACzB,CAAC;AAAA,MACH,WAAWA,OAAM,GAAG,SAAS,SAAS,GAAG;AACvC,cAAM,WACJ,yBAAyBA,OAAM,EAAE,KACjC,sBAAsB,EAAE,MAAMA,OAAM,GAAG,CAAC;AAC1C,eAAO,oBAAoB;AAAA,UACzB,MAAM,SAAS;AAAA,UACf,MAAM,SAAS;AAAA,UACf,cAAc,SAAS;AAAA,QACzB,CAAC;AAAA,MACH,WAAWA,OAAM,GAAG,SAAS,YAAY,GAAG;AAC1C,cAAM,WACJ,qBAAqBA,OAAM,EAAE,KAC7B,4BAA4B,EAAE,MAAMA,OAAM,GAAG,CAAC;AAChD,eAAO,oBAAoB;AAAA,UACzB,MAAM,SAAS;AAAA,UACf,MAAM,SAAS;AAAA,UACf,cAAc,SAAS;AAAA,QACzB,CAAC;AAAA,MACH,OAAO;AACL,cAAM,WACJ,qBAAqBA,OAAM,EAAE,KAAK,eAAe,EAAE,MAAMA,OAAM,GAAG,CAAC;AACrE,eAAO,oBAAoB;AAAA,UACzB,MAAM,SAAS;AAAA,UACf,MAAM,SAAS;AAAA,UACf,cAAc,SAAS;AAAA,QACzB,CAAC;AAAA,MACH;AAAA,IACF,CAAC;AAAA,EACH;AACF;AAEO,SAAS,aAAa,SAA+C;AAC1E,QAAM,gBAAgB,EAAE,MAAM,UAAU,GAAG,QAAQ;AACnD,SAAO,iBAAiB;AAAA,IACtB,MAAM;AAAA,IACN,GAAG;AAAA,IACH,aAAa,OAAO,WAAW;AAC7B,YAAM,SAAS,CAAC;AAChB,aAAO;AAAA,QACL,GAAG,OAAO,OAAO,oBAAoB,EAAE;AAAA,UAAI,CAAC,aAC1C,wBAAwB;AAAA,YACtB,MAAM,SAAS;AAAA,YACf;AAAA,YACA;AAAA,YACA;AAAA,UACF,CAAC;AAAA,QACH;AAAA,MACF;AACA,aAAO;AAAA,QACL,GAAG,OAAO,OAAO,0BAA0B,EAAE;AAAA,UAAI,CAACC,iBAChD,2BAA2B;AAAA,YACzB,MAAMA,aAAY;AAAA,YAClB;AAAA,YACA;AAAA,YACA,aAAAA;AAAA,UACF,CAAC;AAAA,QACH;AAAA,MACF;AACA,aAAO;AAAA,QACL,GAAG,OAAO,OAAO,oBAAoB,EAAE;AAAA,UAAI,CAAC,aAC1C,8BAA8B;AAAA,YAC5B,MAAM,SAAS;AAAA,YACf;AAAA,YACA;AAAA,YACA;AAAA,UACF,CAAC;AAAA,QACH;AAAA,MACF;AACA,aAAO;AAAA,QACL,GAAG,OAAO,OAAO,wBAAwB,EAAE;AAAA,UAAI,CAAC,aAC9C,yBAAyB;AAAA,YACvB,MAAM,SAAS;AAAA,YACf;AAAA,YACA;AAAA,YACA;AAAA,UACF,CAAC;AAAA,QACH;AAAA,MACF;AACA,aAAO;AAAA,QACL,GAAG,OAAO,OAAO,oBAAoB,EAAE;AAAA,UAAI,CAAC,aAC1C,qCAAqC;AAAA,YACnC,MAAM,SAAS;AAAA,YACf;AAAA,YACA;AAAA,YACA;AAAA,UACF,CAAC;AAAA,QACH;AAAA,MACF;AACA,aAAO;AAAA,QACL,GAAG,OAAO,OAAO,sBAAsB,EAAE;AAAA,UAAI,CAAC,aAC5C,6BAA6B;AAAA,YAC3B,MAAM,SAAS;AAAA,YACf;AAAA,YACA;AAAA,YACA;AAAA,YACA,gBAAgB,SAAS,KAAK,SAAS,aAAa,IAChD,0BACA;AAAA,UACN,CAAC;AAAA,QACH;AAAA,MACF;AACA,aAAO;AAAA,IACT;AAAA,IACA,UAAU,eAAe,aAAa;AAAA,IACtC;AAAA,EACF,CAAC;AACH;AA2CA,MAAM,SAAS,CAAC,MAAc,WAA+C;AAC3E,MAAI,KAAK,SAAS,aAAa,KAAK,KAAK,SAAS,QAAQ,GAAG;AAC3D,WAAO,oBAAoB;AAAA,MACzB;AAAA,MACA;AAAA,IACF,CAAC;AAAA,EACH;AACA,MAAI,KAAK,SAAS,KAAK,GAAG;AACxB,WAAO,qBAAqB;AAAA,MAC1B;AAAA,MACA;AAAA,IACF,CAAC;AAAA,EACH;AACA,MAAI,KAAK,SAAS,SAAS,GAAG;AAC5B,WAAO,sBAAsB;AAAA,MAC3B;AAAA,MACA;AAAA,IACF,CAAC;AAAA,EACH;AACA,MAAI,KAAK,SAAS,YAAY,GAAG;AAC/B,WAAO,4BAA4B;AAAA,MACjC;AAAA,MACA;AAAA,IACF,CAAC;AAAA,EACH;AACA,SAAO,eAAe;AAAA,IACpB;AAAA,IACA;AAAA,EACF,CAAC;AACH;AAEA,MAAM,YAAY,CAChB,MACA,WACoC;AACpC,SAAO,YAAY;AAAA,IACjB;AAAA,IACA;AAAA,IACA,cAAc;AAAA,IACd,WAAW;AAAA,EACb,CAAC;AACH;AA+BO,MAAM,SAAuB,OAAO,OAAO,cAAc;AAAA,EAC9D;AAAA,EACA;AACF,CAAC;AAED,IAAO,iBAAQ;","names":["model","embedderRef"]}

package/lib/openai/whisper.d.mts

@@ -1,7 +1,7 @@
 import { ModelReference, z } from 'genkit';
 import { ModelInfo, ModelAction } from 'genkit/model';
 import OpenAI from 'openai';
-import { P as PluginOptions } from '../audio-Cr0WtG4I.mjs';
+import { P as PluginOptions } from '../audio-CW-qdV9D.mjs';
 import 'openai/core.mjs';
 import 'openai/resources/audio/index.mjs';
 import 'genkit/plugin';

package/lib/openai/whisper.d.ts

@@ -1,7 +1,7 @@
 import { ModelReference, z } from 'genkit';
 import { ModelInfo, ModelAction } from 'genkit/model';
 import OpenAI from 'openai';
-import { P as PluginOptions } from '../audio-Cr0WtG4I.js';
+import { P as PluginOptions } from '../audio-CW-qdV9D.js';
 import 'openai/core.mjs';
 import 'openai/resources/audio/index.mjs';
 import 'genkit/plugin';

package/lib/translate.d.mts

@@ -2,7 +2,7 @@ import 'genkit';
 import 'genkit/model';
 import 'openai';
 import 'openai/resources/audio/index.mjs';
-export { q as TRANSLATION_MODEL_INFO, r as TranslationConfigSchema, p as TranslationRequestBuilder, w as compatOaiTranslationModelRef, v as defineCompatOpenAITranslationModel, s as toTranslationRequest, u as translationToGenerateResponse } from './audio-Cr0WtG4I.mjs';
+export { p as TRANSLATION_MODEL_INFO, q as TranslationConfigSchema, r as TranslationRequestBuilder, s as compatOaiTranslationModelRef, u as defineCompatOpenAITranslationModel, v as toTranslationRequest, w as translationToGenerateResponse } from './audio-CW-qdV9D.mjs';
 import 'openai/core.mjs';
 import 'genkit/plugin';
 import 'genkit/registry';

package/lib/translate.d.ts

@@ -2,7 +2,7 @@ import 'genkit';
 import 'genkit/model';
 import 'openai';
 import 'openai/resources/audio/index.mjs';
-export { q as TRANSLATION_MODEL_INFO, r as TranslationConfigSchema, p as TranslationRequestBuilder, w as compatOaiTranslationModelRef, v as defineCompatOpenAITranslationModel, s as toTranslationRequest, u as translationToGenerateResponse } from './audio-Cr0WtG4I.js';
+export { p as TRANSLATION_MODEL_INFO, q as TranslationConfigSchema, r as TranslationRequestBuilder, s as compatOaiTranslationModelRef, u as defineCompatOpenAITranslationModel, v as toTranslationRequest, w as translationToGenerateResponse } from './audio-CW-qdV9D.js';
 import 'openai/core.mjs';
 import 'genkit/plugin';
 import 'genkit/registry';

package/lib/utils.d.mts

@@ -1,7 +1,7 @@
 import { GenerateRequest } from 'genkit';
 import { EmbedRequest } from 'genkit/embedder';
 import OpenAI from 'openai';
-import { P as PluginOptions } from './audio-Cr0WtG4I.mjs';
+import { P as PluginOptions } from './audio-CW-qdV9D.mjs';
 import 'genkit/model';
 import 'openai/core.mjs';
 import 'openai/resources/audio/index.mjs';

package/lib/utils.d.ts

@@ -1,7 +1,7 @@
 import { GenerateRequest } from 'genkit';
 import { EmbedRequest } from 'genkit/embedder';
 import OpenAI from 'openai';
-import { P as PluginOptions } from './audio-Cr0WtG4I.js';
+import { P as PluginOptions } from './audio-CW-qdV9D.js';
 import 'genkit/model';
 import 'openai/core.mjs';
 import 'openai/resources/audio/index.mjs';

package/lib/xai/grok.d.mts

@@ -1,6 +1,6 @@
 import { z } from 'genkit';
-import { ModelInfo, ModelReference } from 'genkit/model';
-import { M as ModelRequestBuilder } from '../audio-Cr0WtG4I.mjs';
+import { ModelReference, ModelInfo } from 'genkit/model';
+import { M as ModelRequestBuilder } from '../audio-CW-qdV9D.mjs';
 import 'openai';
 import 'openai/core.mjs';
 import 'openai/resources/audio/index.mjs';

package/lib/xai/grok.d.ts

@@ -1,6 +1,6 @@
 import { z } from 'genkit';
-import { ModelInfo, ModelReference } from 'genkit/model';
-import { M as ModelRequestBuilder } from '../audio-Cr0WtG4I.js';
+import { ModelReference, ModelInfo } from 'genkit/model';
+import { M as ModelRequestBuilder } from '../audio-CW-qdV9D.js';
 import 'openai';
 import 'openai/core.mjs';
 import 'openai/resources/audio/index.mjs';

package/lib/xai/index.d.mts

@@ -1,6 +1,6 @@
 import { z, ModelReference } from 'genkit';
 import { GenkitPluginV2 } from 'genkit/plugin';
-import { P as PluginOptions, I as ImageGenerationCommonConfigSchema } from '../audio-Cr0WtG4I.mjs';
+import { P as PluginOptions, I as ImageGenerationCommonConfigSchema } from '../audio-CW-qdV9D.mjs';
 import { SUPPORTED_IMAGE_MODELS } from './grok-image.mjs';
 import { SUPPORTED_LANGUAGE_MODELS, XaiChatCompletionConfigSchema } from './grok.mjs';
 import 'genkit/model';

package/lib/xai/index.d.ts

@@ -1,6 +1,6 @@
 import { z, ModelReference } from 'genkit';
 import { GenkitPluginV2 } from 'genkit/plugin';
-import { P as PluginOptions, I as ImageGenerationCommonConfigSchema } from '../audio-Cr0WtG4I.js';
+import { P as PluginOptions, I as ImageGenerationCommonConfigSchema } from '../audio-CW-qdV9D.js';
 import { SUPPORTED_IMAGE_MODELS } from './grok-image.js';
 import { SUPPORTED_LANGUAGE_MODELS, XaiChatCompletionConfigSchema } from './grok.js';
 import 'genkit/model';

package/lib/xai/index.js

@@ -111,7 +111,7 @@ function xAIPlugin(options) {
     listActions
   });
 }
-const model = (name, config) => {
+const model = ((name, config) => {
   if (name.includes("image")) {
     return (0, import_grok_image.xaiImageModelRef)({
       name,
@@ -122,7 +122,7 @@ const model = (name, config) => {
     name,
     config
   });
-};
+});
 const xAI = Object.assign(xAIPlugin, {
   model
 });

package/lib/xai/index.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/xai/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  GenkitError,\n  modelActionMetadata,\n  ModelReference,\n  z,\n} from 'genkit';\nimport { logger } from 'genkit/logging';\nimport { ResolvableAction, type GenkitPluginV2 } from 'genkit/plugin';\nimport { ActionType } from 'genkit/registry';\nimport OpenAI from 'openai';\nimport {\n  defineCompatOpenAIImageModel,\n  ImageGenerationCommonConfigSchema,\n} from '../image.js';\nimport { openAICompatible, PluginOptions } from '../index.js';\nimport { defineCompatOpenAIModel } from '../model.js';\nimport { SUPPORTED_IMAGE_MODELS, xaiImageModelRef } from './grok-image.js';\nimport {\n  grokRequestBuilder,\n  SUPPORTED_LANGUAGE_MODELS,\n  XaiChatCompletionConfigSchema,\n  xaiModelRef,\n} from './grok.js';\n\nexport type XAIPluginOptions = Omit<PluginOptions, 'name' | 'baseURL'>;\n\nfunction createResolver(pluginOptions: PluginOptions) {\n  return async (client: OpenAI, actionType: ActionType, actionName: string) => {\n    if (actionType === 'model') {\n      const modelRef = xaiModelRef({ name: actionName });\n      return defineCompatOpenAIModel({\n        name: modelRef.name,\n        client,\n        pluginOptions,\n        modelRef,\n        requestBuilder: grokRequestBuilder,\n      });\n    } else {\n      logger.warn('Only model actions are supported by the XAI plugin');\n    }\n    return undefined;\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        if (model.id.includes('image')) {\n          const modelRef =\n            SUPPORTED_IMAGE_MODELS[model.id] ??\n            xaiImageModelRef({ name: model.id });\n          return modelActionMetadata({\n            name: modelRef.name,\n            info: modelRef.info,\n            configSchema: modelRef.configSchema,\n          });\n        } else {\n          const modelRef =\n            SUPPORTED_LANGUAGE_MODELS[model.id] ??\n            xaiModelRef({ name: model.id });\n          return modelActionMetadata({\n            name: modelRef.name,\n            info: modelRef.info,\n            configSchema: modelRef.configSchema,\n          });\n        }\n      })\n  );\n};\n\nexport function xAIPlugin(options?: XAIPluginOptions): GenkitPluginV2 {\n  const apiKey = options?.apiKey ?? process.env.XAI_API_KEY;\n  if (!apiKey) {\n    throw new GenkitError({\n      status: 'FAILED_PRECONDITION',\n      message:\n        'Please pass in the API key or set the XAI_API_KEY environment variable.',\n    });\n  }\n  const pluginOptions = { name: 'xai', ...options };\n  return openAICompatible({\n    name: 'xai',\n    baseURL: 'https://api.x.ai/v1',\n    apiKey,\n    ...options,\n    initializer: async (client) => {\n      const models = [] as ResolvableAction[];\n      models.push(\n        ...Object.values(SUPPORTED_LANGUAGE_MODELS).map((modelRef) =>\n          defineCompatOpenAIModel({\n            name: modelRef.name,\n            client,\n            pluginOptions,\n            modelRef,\n            requestBuilder: grokRequestBuilder,\n          })\n        )\n      );\n      models.push(\n        ...Object.values(SUPPORTED_IMAGE_MODELS).map((modelRef) =>\n          defineCompatOpenAIImageModel({\n            name: modelRef.name,\n            client,\n            pluginOptions,\n            modelRef,\n          })\n        )\n      );\n      return models;\n    },\n    resolver: createResolver(pluginOptions),\n    listActions,\n  });\n}\n\nexport type XAIPlugin = {\n  (params?: XAIPluginOptions): GenkitPluginV2;\n  model(\n    name: keyof typeof SUPPORTED_LANGUAGE_MODELS,\n    config?: z.infer<typeof XaiChatCompletionConfigSchema>\n  ): ModelReference<typeof XaiChatCompletionConfigSchema>;\n  model(\n    name: keyof typeof SUPPORTED_IMAGE_MODELS,\n    config?: z.infer<typeof ImageGenerationCommonConfigSchema>\n  ): ModelReference<typeof ImageGenerationCommonConfigSchema>;\n  model(name: string, config?: any): ModelReference<z.ZodTypeAny>;\n};\n\nconst model = ((name: string, config?: any): ModelReference<z.ZodTypeAny> => {\n  if (name.includes('image')) {\n    return xaiImageModelRef({\n      name,\n      config,\n    });\n  }\n  return xaiModelRef({\n    name,\n    config,\n  });\n}) as XAIPlugin['model'];\n\n/**\n * This module provides an interface to the XAI 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 `xai` 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 * - xAI: The main plugin function to interact with XAI, via OpenAI\n *   compatible API.\n *\n * Usage: To use the models, initialize the xAI 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 { xAI } from '@genkit-ai/compat-oai/xai';\n *\n * export default configureGenkit({\n *  plugins: [\n *    xAI()\n *    ... // other plugins\n *  ]\n * });\n * ```\n */\nexport const xAI: XAIPlugin = Object.assign(xAIPlugin, {\n  model,\n});\n\nexport default xAI;\n"],"mappings":";;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAgBA,oBAMO;AACP,qBAAuB;AAIvB,mBAGO;AACP,eAAgD;AAChD,mBAAwC;AACxC,wBAAyD;AACzD,kBAKO;AAIP,SAAS,eAAe,eAA8B;AACpD,SAAO,OAAO,QAAgB,YAAwB,eAAuB;AAC3E,QAAI,eAAe,SAAS;AAC1B,YAAM,eAAW,yBAAY,EAAE,MAAM,WAAW,CAAC;AACjD,iBAAO,sCAAwB;AAAA,QAC7B,MAAM,SAAS;AAAA,QACf;AAAA,QACA;AAAA,QACA;AAAA,QACA,gBAAgB;AAAA,MAClB,CAAC;AAAA,IACH,OAAO;AACL,4BAAO,KAAK,oDAAoD;AAAA,IAClE;AACA,WAAO;AAAA,EACT;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,UAAIA,OAAM,GAAG,SAAS,OAAO,GAAG;AAC9B,cAAM,WACJ,yCAAuBA,OAAM,EAAE,SAC/B,oCAAiB,EAAE,MAAMA,OAAM,GAAG,CAAC;AACrC,mBAAO,mCAAoB;AAAA,UACzB,MAAM,SAAS;AAAA,UACf,MAAM,SAAS;AAAA,UACf,cAAc,SAAS;AAAA,QACzB,CAAC;AAAA,MACH,OAAO;AACL,cAAM,WACJ,sCAA0BA,OAAM,EAAE,SAClC,yBAAY,EAAE,MAAMA,OAAM,GAAG,CAAC;AAChC,mBAAO,mCAAoB;AAAA,UACzB,MAAM,SAAS;AAAA,UACf,MAAM,SAAS;AAAA,UACf,cAAc,SAAS;AAAA,QACzB,CAAC;AAAA,MACH;AAAA,IACF,CAAC;AAAA,EACL;AACF;AAEO,SAAS,UAAU,SAA4C;AACpE,QAAM,SAAS,SAAS,UAAU,QAAQ,IAAI;AAC9C,MAAI,CAAC,QAAQ;AACX,UAAM,IAAI,0BAAY;AAAA,MACpB,QAAQ;AAAA,MACR,SACE;AAAA,IACJ,CAAC;AAAA,EACH;AACA,QAAM,gBAAgB,EAAE,MAAM,OAAO,GAAG,QAAQ;AAChD,aAAO,2BAAiB;AAAA,IACtB,MAAM;AAAA,IACN,SAAS;AAAA,IACT;AAAA,IACA,GAAG;AAAA,IACH,aAAa,OAAO,WAAW;AAC7B,YAAM,SAAS,CAAC;AAChB,aAAO;AAAA,QACL,GAAG,OAAO,OAAO,qCAAyB,EAAE;AAAA,UAAI,CAAC,iBAC/C,sCAAwB;AAAA,YACtB,MAAM,SAAS;AAAA,YACf;AAAA,YACA;AAAA,YACA;AAAA,YACA,gBAAgB;AAAA,UAClB,CAAC;AAAA,QACH;AAAA,MACF;AACA,aAAO;AAAA,QACL,GAAG,OAAO,OAAO,wCAAsB,EAAE;AAAA,UAAI,CAAC,iBAC5C,2CAA6B;AAAA,YAC3B,MAAM,SAAS;AAAA,YACf;AAAA,YACA;AAAA,YACA;AAAA,UACF,CAAC;AAAA,QACH;AAAA,MACF;AACA,aAAO;AAAA,IACT;AAAA,IACA,UAAU,eAAe,aAAa;AAAA,IACtC;AAAA,EACF,CAAC;AACH;AAeA,MAAM,QAAS,CAAC,MAAc,WAA+C;AAC3E,MAAI,KAAK,SAAS,OAAO,GAAG;AAC1B,eAAO,oCAAiB;AAAA,MACtB;AAAA,MACA;AAAA,IACF,CAAC;AAAA,EACH;AACA,aAAO,yBAAY;AAAA,IACjB;AAAA,IACA;AAAA,EACF,CAAC;AACH;AAgCO,MAAM,MAAiB,OAAO,OAAO,WAAW;AAAA,EACrD;AACF,CAAC;AAED,IAAO,cAAQ;","names":["model"]}
+{"version":3,"sources":["../../src/xai/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  GenkitError,\n  modelActionMetadata,\n  ModelReference,\n  z,\n} from 'genkit';\nimport { logger } from 'genkit/logging';\nimport { ResolvableAction, type GenkitPluginV2 } from 'genkit/plugin';\nimport { ActionType } from 'genkit/registry';\nimport OpenAI from 'openai';\nimport {\n  defineCompatOpenAIImageModel,\n  ImageGenerationCommonConfigSchema,\n} from '../image.js';\nimport { openAICompatible, PluginOptions } from '../index.js';\nimport { defineCompatOpenAIModel } from '../model.js';\nimport { SUPPORTED_IMAGE_MODELS, xaiImageModelRef } from './grok-image.js';\nimport {\n  grokRequestBuilder,\n  SUPPORTED_LANGUAGE_MODELS,\n  XaiChatCompletionConfigSchema,\n  xaiModelRef,\n} from './grok.js';\n\nexport type XAIPluginOptions = Omit<PluginOptions, 'name' | 'baseURL'>;\n\nfunction createResolver(pluginOptions: PluginOptions) {\n  return async (client: OpenAI, actionType: ActionType, actionName: string) => {\n    if (actionType === 'model') {\n      const modelRef = xaiModelRef({ name: actionName });\n      return defineCompatOpenAIModel({\n        name: modelRef.name,\n        client,\n        pluginOptions,\n        modelRef,\n        requestBuilder: grokRequestBuilder,\n      });\n    } else {\n      logger.warn('Only model actions are supported by the XAI plugin');\n    }\n    return undefined;\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        if (model.id.includes('image')) {\n          const modelRef =\n            SUPPORTED_IMAGE_MODELS[model.id] ??\n            xaiImageModelRef({ name: model.id });\n          return modelActionMetadata({\n            name: modelRef.name,\n            info: modelRef.info,\n            configSchema: modelRef.configSchema,\n          });\n        } else {\n          const modelRef =\n            SUPPORTED_LANGUAGE_MODELS[model.id] ??\n            xaiModelRef({ name: model.id });\n          return modelActionMetadata({\n            name: modelRef.name,\n            info: modelRef.info,\n            configSchema: modelRef.configSchema,\n          });\n        }\n      })\n  );\n};\n\nexport function xAIPlugin(options?: XAIPluginOptions): GenkitPluginV2 {\n  const apiKey = options?.apiKey ?? process.env.XAI_API_KEY;\n  if (!apiKey) {\n    throw new GenkitError({\n      status: 'FAILED_PRECONDITION',\n      message:\n        'Please pass in the API key or set the XAI_API_KEY environment variable.',\n    });\n  }\n  const pluginOptions = { name: 'xai', ...options };\n  return openAICompatible({\n    name: 'xai',\n    baseURL: 'https://api.x.ai/v1',\n    apiKey,\n    ...options,\n    initializer: async (client) => {\n      const models = [] as ResolvableAction[];\n      models.push(\n        ...Object.values(SUPPORTED_LANGUAGE_MODELS).map((modelRef) =>\n          defineCompatOpenAIModel({\n            name: modelRef.name,\n            client,\n            pluginOptions,\n            modelRef,\n            requestBuilder: grokRequestBuilder,\n          })\n        )\n      );\n      models.push(\n        ...Object.values(SUPPORTED_IMAGE_MODELS).map((modelRef) =>\n          defineCompatOpenAIImageModel({\n            name: modelRef.name,\n            client,\n            pluginOptions,\n            modelRef,\n          })\n        )\n      );\n      return models;\n    },\n    resolver: createResolver(pluginOptions),\n    listActions,\n  });\n}\n\nexport type XAIPlugin = {\n  (params?: XAIPluginOptions): GenkitPluginV2;\n  model(\n    name: keyof typeof SUPPORTED_LANGUAGE_MODELS,\n    config?: z.infer<typeof XaiChatCompletionConfigSchema>\n  ): ModelReference<typeof XaiChatCompletionConfigSchema>;\n  model(\n    name: keyof typeof SUPPORTED_IMAGE_MODELS,\n    config?: z.infer<typeof ImageGenerationCommonConfigSchema>\n  ): ModelReference<typeof ImageGenerationCommonConfigSchema>;\n  model(name: string, config?: any): ModelReference<z.ZodTypeAny>;\n};\n\nconst model = ((name: string, config?: any): ModelReference<z.ZodTypeAny> => {\n  if (name.includes('image')) {\n    return xaiImageModelRef({\n      name,\n      config,\n    });\n  }\n  return xaiModelRef({\n    name,\n    config,\n  });\n}) as XAIPlugin['model'];\n\n/**\n * This module provides an interface to the XAI 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 `xai` 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 * - xAI: The main plugin function to interact with XAI, via OpenAI\n *   compatible API.\n *\n * Usage: To use the models, initialize the xAI 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 { xAI } from '@genkit-ai/compat-oai/xai';\n *\n * export default configureGenkit({\n *  plugins: [\n *    xAI()\n *    ... // other plugins\n *  ]\n * });\n * ```\n */\nexport const xAI: XAIPlugin = Object.assign(xAIPlugin, {\n  model,\n});\n\nexport default xAI;\n"],"mappings":";;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAgBA,oBAMO;AACP,qBAAuB;AAIvB,mBAGO;AACP,eAAgD;AAChD,mBAAwC;AACxC,wBAAyD;AACzD,kBAKO;AAIP,SAAS,eAAe,eAA8B;AACpD,SAAO,OAAO,QAAgB,YAAwB,eAAuB;AAC3E,QAAI,eAAe,SAAS;AAC1B,YAAM,eAAW,yBAAY,EAAE,MAAM,WAAW,CAAC;AACjD,iBAAO,sCAAwB;AAAA,QAC7B,MAAM,SAAS;AAAA,QACf;AAAA,QACA;AAAA,QACA;AAAA,QACA,gBAAgB;AAAA,MAClB,CAAC;AAAA,IACH,OAAO;AACL,4BAAO,KAAK,oDAAoD;AAAA,IAClE;AACA,WAAO;AAAA,EACT;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,UAAIA,OAAM,GAAG,SAAS,OAAO,GAAG;AAC9B,cAAM,WACJ,yCAAuBA,OAAM,EAAE,SAC/B,oCAAiB,EAAE,MAAMA,OAAM,GAAG,CAAC;AACrC,mBAAO,mCAAoB;AAAA,UACzB,MAAM,SAAS;AAAA,UACf,MAAM,SAAS;AAAA,UACf,cAAc,SAAS;AAAA,QACzB,CAAC;AAAA,MACH,OAAO;AACL,cAAM,WACJ,sCAA0BA,OAAM,EAAE,SAClC,yBAAY,EAAE,MAAMA,OAAM,GAAG,CAAC;AAChC,mBAAO,mCAAoB;AAAA,UACzB,MAAM,SAAS;AAAA,UACf,MAAM,SAAS;AAAA,UACf,cAAc,SAAS;AAAA,QACzB,CAAC;AAAA,MACH;AAAA,IACF,CAAC;AAAA,EACL;AACF;AAEO,SAAS,UAAU,SAA4C;AACpE,QAAM,SAAS,SAAS,UAAU,QAAQ,IAAI;AAC9C,MAAI,CAAC,QAAQ;AACX,UAAM,IAAI,0BAAY;AAAA,MACpB,QAAQ;AAAA,MACR,SACE;AAAA,IACJ,CAAC;AAAA,EACH;AACA,QAAM,gBAAgB,EAAE,MAAM,OAAO,GAAG,QAAQ;AAChD,aAAO,2BAAiB;AAAA,IACtB,MAAM;AAAA,IACN,SAAS;AAAA,IACT;AAAA,IACA,GAAG;AAAA,IACH,aAAa,OAAO,WAAW;AAC7B,YAAM,SAAS,CAAC;AAChB,aAAO;AAAA,QACL,GAAG,OAAO,OAAO,qCAAyB,EAAE;AAAA,UAAI,CAAC,iBAC/C,sCAAwB;AAAA,YACtB,MAAM,SAAS;AAAA,YACf;AAAA,YACA;AAAA,YACA;AAAA,YACA,gBAAgB;AAAA,UAClB,CAAC;AAAA,QACH;AAAA,MACF;AACA,aAAO;AAAA,QACL,GAAG,OAAO,OAAO,wCAAsB,EAAE;AAAA,UAAI,CAAC,iBAC5C,2CAA6B;AAAA,YAC3B,MAAM,SAAS;AAAA,YACf;AAAA,YACA;AAAA,YACA;AAAA,UACF,CAAC;AAAA,QACH;AAAA,MACF;AACA,aAAO;AAAA,IACT;AAAA,IACA,UAAU,eAAe,aAAa;AAAA,IACtC;AAAA,EACF,CAAC;AACH;AAeA,MAAM,SAAS,CAAC,MAAc,WAA+C;AAC3E,MAAI,KAAK,SAAS,OAAO,GAAG;AAC1B,eAAO,oCAAiB;AAAA,MACtB;AAAA,MACA;AAAA,IACF,CAAC;AAAA,EACH;AACA,aAAO,yBAAY;AAAA,IACjB;AAAA,IACA;AAAA,EACF,CAAC;AACH;AAgCO,MAAM,MAAiB,OAAO,OAAO,WAAW;AAAA,EACrD;AACF,CAAC;AAED,IAAO,cAAQ;","names":["model"]}

package/lib/xai/index.mjs

@@ -95,7 +95,7 @@ function xAIPlugin(options) {
     listActions
   });
 }
-const model = (name, config) => {
+const model = ((name, config) => {
   if (name.includes("image")) {
     return xaiImageModelRef({
       name,
@@ -106,7 +106,7 @@ const model = (name, config) => {
     name,
     config
   });
-};
+});
 const xAI = Object.assign(xAIPlugin, {
   model
 });

package/lib/xai/index.mjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/xai/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  GenkitError,\n  modelActionMetadata,\n  ModelReference,\n  z,\n} from 'genkit';\nimport { logger } from 'genkit/logging';\nimport { ResolvableAction, type GenkitPluginV2 } from 'genkit/plugin';\nimport { ActionType } from 'genkit/registry';\nimport OpenAI from 'openai';\nimport {\n  defineCompatOpenAIImageModel,\n  ImageGenerationCommonConfigSchema,\n} from '../image.js';\nimport { openAICompatible, PluginOptions } from '../index.js';\nimport { defineCompatOpenAIModel } from '../model.js';\nimport { SUPPORTED_IMAGE_MODELS, xaiImageModelRef } from './grok-image.js';\nimport {\n  grokRequestBuilder,\n  SUPPORTED_LANGUAGE_MODELS,\n  XaiChatCompletionConfigSchema,\n  xaiModelRef,\n} from './grok.js';\n\nexport type XAIPluginOptions = Omit<PluginOptions, 'name' | 'baseURL'>;\n\nfunction createResolver(pluginOptions: PluginOptions) {\n  return async (client: OpenAI, actionType: ActionType, actionName: string) => {\n    if (actionType === 'model') {\n      const modelRef = xaiModelRef({ name: actionName });\n      return defineCompatOpenAIModel({\n        name: modelRef.name,\n        client,\n        pluginOptions,\n        modelRef,\n        requestBuilder: grokRequestBuilder,\n      });\n    } else {\n      logger.warn('Only model actions are supported by the XAI plugin');\n    }\n    return undefined;\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        if (model.id.includes('image')) {\n          const modelRef =\n            SUPPORTED_IMAGE_MODELS[model.id] ??\n            xaiImageModelRef({ name: model.id });\n          return modelActionMetadata({\n            name: modelRef.name,\n            info: modelRef.info,\n            configSchema: modelRef.configSchema,\n          });\n        } else {\n          const modelRef =\n            SUPPORTED_LANGUAGE_MODELS[model.id] ??\n            xaiModelRef({ name: model.id });\n          return modelActionMetadata({\n            name: modelRef.name,\n            info: modelRef.info,\n            configSchema: modelRef.configSchema,\n          });\n        }\n      })\n  );\n};\n\nexport function xAIPlugin(options?: XAIPluginOptions): GenkitPluginV2 {\n  const apiKey = options?.apiKey ?? process.env.XAI_API_KEY;\n  if (!apiKey) {\n    throw new GenkitError({\n      status: 'FAILED_PRECONDITION',\n      message:\n        'Please pass in the API key or set the XAI_API_KEY environment variable.',\n    });\n  }\n  const pluginOptions = { name: 'xai', ...options };\n  return openAICompatible({\n    name: 'xai',\n    baseURL: 'https://api.x.ai/v1',\n    apiKey,\n    ...options,\n    initializer: async (client) => {\n      const models = [] as ResolvableAction[];\n      models.push(\n        ...Object.values(SUPPORTED_LANGUAGE_MODELS).map((modelRef) =>\n          defineCompatOpenAIModel({\n            name: modelRef.name,\n            client,\n            pluginOptions,\n            modelRef,\n            requestBuilder: grokRequestBuilder,\n          })\n        )\n      );\n      models.push(\n        ...Object.values(SUPPORTED_IMAGE_MODELS).map((modelRef) =>\n          defineCompatOpenAIImageModel({\n            name: modelRef.name,\n            client,\n            pluginOptions,\n            modelRef,\n          })\n        )\n      );\n      return models;\n    },\n    resolver: createResolver(pluginOptions),\n    listActions,\n  });\n}\n\nexport type XAIPlugin = {\n  (params?: XAIPluginOptions): GenkitPluginV2;\n  model(\n    name: keyof typeof SUPPORTED_LANGUAGE_MODELS,\n    config?: z.infer<typeof XaiChatCompletionConfigSchema>\n  ): ModelReference<typeof XaiChatCompletionConfigSchema>;\n  model(\n    name: keyof typeof SUPPORTED_IMAGE_MODELS,\n    config?: z.infer<typeof ImageGenerationCommonConfigSchema>\n  ): ModelReference<typeof ImageGenerationCommonConfigSchema>;\n  model(name: string, config?: any): ModelReference<z.ZodTypeAny>;\n};\n\nconst model = ((name: string, config?: any): ModelReference<z.ZodTypeAny> => {\n  if (name.includes('image')) {\n    return xaiImageModelRef({\n      name,\n      config,\n    });\n  }\n  return xaiModelRef({\n    name,\n    config,\n  });\n}) as XAIPlugin['model'];\n\n/**\n * This module provides an interface to the XAI 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 `xai` 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 * - xAI: The main plugin function to interact with XAI, via OpenAI\n *   compatible API.\n *\n * Usage: To use the models, initialize the xAI 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 { xAI } from '@genkit-ai/compat-oai/xai';\n *\n * export default configureGenkit({\n *  plugins: [\n *    xAI()\n *    ... // other plugins\n *  ]\n * });\n * ```\n */\nexport const xAI: XAIPlugin = Object.assign(xAIPlugin, {\n  model,\n});\n\nexport default xAI;\n"],"mappings":"AAgBA;AAAA,EAEE;AAAA,EACA;AAAA,OAGK;AACP,SAAS,cAAc;AAIvB;AAAA,EACE;AAAA,OAEK;AACP,SAAS,wBAAuC;AAChD,SAAS,+BAA+B;AACxC,SAAS,wBAAwB,wBAAwB;AACzD;AAAA,EACE;AAAA,EACA;AAAA,EAEA;AAAA,OACK;AAIP,SAAS,eAAe,eAA8B;AACpD,SAAO,OAAO,QAAgB,YAAwB,eAAuB;AAC3E,QAAI,eAAe,SAAS;AAC1B,YAAM,WAAW,YAAY,EAAE,MAAM,WAAW,CAAC;AACjD,aAAO,wBAAwB;AAAA,QAC7B,MAAM,SAAS;AAAA,QACf;AAAA,QACA;AAAA,QACA;AAAA,QACA,gBAAgB;AAAA,MAClB,CAAC;AAAA,IACH,OAAO;AACL,aAAO,KAAK,oDAAoD;AAAA,IAClE;AACA,WAAO;AAAA,EACT;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,UAAIA,OAAM,GAAG,SAAS,OAAO,GAAG;AAC9B,cAAM,WACJ,uBAAuBA,OAAM,EAAE,KAC/B,iBAAiB,EAAE,MAAMA,OAAM,GAAG,CAAC;AACrC,eAAO,oBAAoB;AAAA,UACzB,MAAM,SAAS;AAAA,UACf,MAAM,SAAS;AAAA,UACf,cAAc,SAAS;AAAA,QACzB,CAAC;AAAA,MACH,OAAO;AACL,cAAM,WACJ,0BAA0BA,OAAM,EAAE,KAClC,YAAY,EAAE,MAAMA,OAAM,GAAG,CAAC;AAChC,eAAO,oBAAoB;AAAA,UACzB,MAAM,SAAS;AAAA,UACf,MAAM,SAAS;AAAA,UACf,cAAc,SAAS;AAAA,QACzB,CAAC;AAAA,MACH;AAAA,IACF,CAAC;AAAA,EACL;AACF;AAEO,SAAS,UAAU,SAA4C;AACpE,QAAM,SAAS,SAAS,UAAU,QAAQ,IAAI;AAC9C,MAAI,CAAC,QAAQ;AACX,UAAM,IAAI,YAAY;AAAA,MACpB,QAAQ;AAAA,MACR,SACE;AAAA,IACJ,CAAC;AAAA,EACH;AACA,QAAM,gBAAgB,EAAE,MAAM,OAAO,GAAG,QAAQ;AAChD,SAAO,iBAAiB;AAAA,IACtB,MAAM;AAAA,IACN,SAAS;AAAA,IACT;AAAA,IACA,GAAG;AAAA,IACH,aAAa,OAAO,WAAW;AAC7B,YAAM,SAAS,CAAC;AAChB,aAAO;AAAA,QACL,GAAG,OAAO,OAAO,yBAAyB,EAAE;AAAA,UAAI,CAAC,aAC/C,wBAAwB;AAAA,YACtB,MAAM,SAAS;AAAA,YACf;AAAA,YACA;AAAA,YACA;AAAA,YACA,gBAAgB;AAAA,UAClB,CAAC;AAAA,QACH;AAAA,MACF;AACA,aAAO;AAAA,QACL,GAAG,OAAO,OAAO,sBAAsB,EAAE;AAAA,UAAI,CAAC,aAC5C,6BAA6B;AAAA,YAC3B,MAAM,SAAS;AAAA,YACf;AAAA,YACA;AAAA,YACA;AAAA,UACF,CAAC;AAAA,QACH;AAAA,MACF;AACA,aAAO;AAAA,IACT;AAAA,IACA,UAAU,eAAe,aAAa;AAAA,IACtC;AAAA,EACF,CAAC;AACH;AAeA,MAAM,QAAS,CAAC,MAAc,WAA+C;AAC3E,MAAI,KAAK,SAAS,OAAO,GAAG;AAC1B,WAAO,iBAAiB;AAAA,MACtB;AAAA,MACA;AAAA,IACF,CAAC;AAAA,EACH;AACA,SAAO,YAAY;AAAA,IACjB;AAAA,IACA;AAAA,EACF,CAAC;AACH;AAgCO,MAAM,MAAiB,OAAO,OAAO,WAAW;AAAA,EACrD;AACF,CAAC;AAED,IAAO,cAAQ;","names":["model"]}
+{"version":3,"sources":["../../src/xai/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  GenkitError,\n  modelActionMetadata,\n  ModelReference,\n  z,\n} from 'genkit';\nimport { logger } from 'genkit/logging';\nimport { ResolvableAction, type GenkitPluginV2 } from 'genkit/plugin';\nimport { ActionType } from 'genkit/registry';\nimport OpenAI from 'openai';\nimport {\n  defineCompatOpenAIImageModel,\n  ImageGenerationCommonConfigSchema,\n} from '../image.js';\nimport { openAICompatible, PluginOptions } from '../index.js';\nimport { defineCompatOpenAIModel } from '../model.js';\nimport { SUPPORTED_IMAGE_MODELS, xaiImageModelRef } from './grok-image.js';\nimport {\n  grokRequestBuilder,\n  SUPPORTED_LANGUAGE_MODELS,\n  XaiChatCompletionConfigSchema,\n  xaiModelRef,\n} from './grok.js';\n\nexport type XAIPluginOptions = Omit<PluginOptions, 'name' | 'baseURL'>;\n\nfunction createResolver(pluginOptions: PluginOptions) {\n  return async (client: OpenAI, actionType: ActionType, actionName: string) => {\n    if (actionType === 'model') {\n      const modelRef = xaiModelRef({ name: actionName });\n      return defineCompatOpenAIModel({\n        name: modelRef.name,\n        client,\n        pluginOptions,\n        modelRef,\n        requestBuilder: grokRequestBuilder,\n      });\n    } else {\n      logger.warn('Only model actions are supported by the XAI plugin');\n    }\n    return undefined;\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        if (model.id.includes('image')) {\n          const modelRef =\n            SUPPORTED_IMAGE_MODELS[model.id] ??\n            xaiImageModelRef({ name: model.id });\n          return modelActionMetadata({\n            name: modelRef.name,\n            info: modelRef.info,\n            configSchema: modelRef.configSchema,\n          });\n        } else {\n          const modelRef =\n            SUPPORTED_LANGUAGE_MODELS[model.id] ??\n            xaiModelRef({ name: model.id });\n          return modelActionMetadata({\n            name: modelRef.name,\n            info: modelRef.info,\n            configSchema: modelRef.configSchema,\n          });\n        }\n      })\n  );\n};\n\nexport function xAIPlugin(options?: XAIPluginOptions): GenkitPluginV2 {\n  const apiKey = options?.apiKey ?? process.env.XAI_API_KEY;\n  if (!apiKey) {\n    throw new GenkitError({\n      status: 'FAILED_PRECONDITION',\n      message:\n        'Please pass in the API key or set the XAI_API_KEY environment variable.',\n    });\n  }\n  const pluginOptions = { name: 'xai', ...options };\n  return openAICompatible({\n    name: 'xai',\n    baseURL: 'https://api.x.ai/v1',\n    apiKey,\n    ...options,\n    initializer: async (client) => {\n      const models = [] as ResolvableAction[];\n      models.push(\n        ...Object.values(SUPPORTED_LANGUAGE_MODELS).map((modelRef) =>\n          defineCompatOpenAIModel({\n            name: modelRef.name,\n            client,\n            pluginOptions,\n            modelRef,\n            requestBuilder: grokRequestBuilder,\n          })\n        )\n      );\n      models.push(\n        ...Object.values(SUPPORTED_IMAGE_MODELS).map((modelRef) =>\n          defineCompatOpenAIImageModel({\n            name: modelRef.name,\n            client,\n            pluginOptions,\n            modelRef,\n          })\n        )\n      );\n      return models;\n    },\n    resolver: createResolver(pluginOptions),\n    listActions,\n  });\n}\n\nexport type XAIPlugin = {\n  (params?: XAIPluginOptions): GenkitPluginV2;\n  model(\n    name: keyof typeof SUPPORTED_LANGUAGE_MODELS,\n    config?: z.infer<typeof XaiChatCompletionConfigSchema>\n  ): ModelReference<typeof XaiChatCompletionConfigSchema>;\n  model(\n    name: keyof typeof SUPPORTED_IMAGE_MODELS,\n    config?: z.infer<typeof ImageGenerationCommonConfigSchema>\n  ): ModelReference<typeof ImageGenerationCommonConfigSchema>;\n  model(name: string, config?: any): ModelReference<z.ZodTypeAny>;\n};\n\nconst model = ((name: string, config?: any): ModelReference<z.ZodTypeAny> => {\n  if (name.includes('image')) {\n    return xaiImageModelRef({\n      name,\n      config,\n    });\n  }\n  return xaiModelRef({\n    name,\n    config,\n  });\n}) as XAIPlugin['model'];\n\n/**\n * This module provides an interface to the XAI 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 `xai` 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 * - xAI: The main plugin function to interact with XAI, via OpenAI\n *   compatible API.\n *\n * Usage: To use the models, initialize the xAI 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 { xAI } from '@genkit-ai/compat-oai/xai';\n *\n * export default configureGenkit({\n *  plugins: [\n *    xAI()\n *    ... // other plugins\n *  ]\n * });\n * ```\n */\nexport const xAI: XAIPlugin = Object.assign(xAIPlugin, {\n  model,\n});\n\nexport default xAI;\n"],"mappings":"AAgBA;AAAA,EAEE;AAAA,EACA;AAAA,OAGK;AACP,SAAS,cAAc;AAIvB;AAAA,EACE;AAAA,OAEK;AACP,SAAS,wBAAuC;AAChD,SAAS,+BAA+B;AACxC,SAAS,wBAAwB,wBAAwB;AACzD;AAAA,EACE;AAAA,EACA;AAAA,EAEA;AAAA,OACK;AAIP,SAAS,eAAe,eAA8B;AACpD,SAAO,OAAO,QAAgB,YAAwB,eAAuB;AAC3E,QAAI,eAAe,SAAS;AAC1B,YAAM,WAAW,YAAY,EAAE,MAAM,WAAW,CAAC;AACjD,aAAO,wBAAwB;AAAA,QAC7B,MAAM,SAAS;AAAA,QACf;AAAA,QACA;AAAA,QACA;AAAA,QACA,gBAAgB;AAAA,MAClB,CAAC;AAAA,IACH,OAAO;AACL,aAAO,KAAK,oDAAoD;AAAA,IAClE;AACA,WAAO;AAAA,EACT;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,UAAIA,OAAM,GAAG,SAAS,OAAO,GAAG;AAC9B,cAAM,WACJ,uBAAuBA,OAAM,EAAE,KAC/B,iBAAiB,EAAE,MAAMA,OAAM,GAAG,CAAC;AACrC,eAAO,oBAAoB;AAAA,UACzB,MAAM,SAAS;AAAA,UACf,MAAM,SAAS;AAAA,UACf,cAAc,SAAS;AAAA,QACzB,CAAC;AAAA,MACH,OAAO;AACL,cAAM,WACJ,0BAA0BA,OAAM,EAAE,KAClC,YAAY,EAAE,MAAMA,OAAM,GAAG,CAAC;AAChC,eAAO,oBAAoB;AAAA,UACzB,MAAM,SAAS;AAAA,UACf,MAAM,SAAS;AAAA,UACf,cAAc,SAAS;AAAA,QACzB,CAAC;AAAA,MACH;AAAA,IACF,CAAC;AAAA,EACL;AACF;AAEO,SAAS,UAAU,SAA4C;AACpE,QAAM,SAAS,SAAS,UAAU,QAAQ,IAAI;AAC9C,MAAI,CAAC,QAAQ;AACX,UAAM,IAAI,YAAY;AAAA,MACpB,QAAQ;AAAA,MACR,SACE;AAAA,IACJ,CAAC;AAAA,EACH;AACA,QAAM,gBAAgB,EAAE,MAAM,OAAO,GAAG,QAAQ;AAChD,SAAO,iBAAiB;AAAA,IACtB,MAAM;AAAA,IACN,SAAS;AAAA,IACT;AAAA,IACA,GAAG;AAAA,IACH,aAAa,OAAO,WAAW;AAC7B,YAAM,SAAS,CAAC;AAChB,aAAO;AAAA,QACL,GAAG,OAAO,OAAO,yBAAyB,EAAE;AAAA,UAAI,CAAC,aAC/C,wBAAwB;AAAA,YACtB,MAAM,SAAS;AAAA,YACf;AAAA,YACA;AAAA,YACA;AAAA,YACA,gBAAgB;AAAA,UAClB,CAAC;AAAA,QACH;AAAA,MACF;AACA,aAAO;AAAA,QACL,GAAG,OAAO,OAAO,sBAAsB,EAAE;AAAA,UAAI,CAAC,aAC5C,6BAA6B;AAAA,YAC3B,MAAM,SAAS;AAAA,YACf;AAAA,YACA;AAAA,YACA;AAAA,UACF,CAAC;AAAA,QACH;AAAA,MACF;AACA,aAAO;AAAA,IACT;AAAA,IACA,UAAU,eAAe,aAAa;AAAA,IACtC;AAAA,EACF,CAAC;AACH;AAeA,MAAM,SAAS,CAAC,MAAc,WAA+C;AAC3E,MAAI,KAAK,SAAS,OAAO,GAAG;AAC1B,WAAO,iBAAiB;AAAA,MACtB;AAAA,MACA;AAAA,IACF,CAAC;AAAA,EACH;AACA,SAAO,YAAY;AAAA,IACjB;AAAA,IACA;AAAA,EACF,CAAC;AACH;AAgCO,MAAM,MAAiB,OAAO,OAAO,WAAW;AAAA,EACrD;AACF,CAAC;AAED,IAAO,cAAQ;","names":["model"]}

package/package.json

@@ -11,7 +11,7 @@
     "genai",
     "generative-ai"
   ],
-  "version": "1.30.1",
+  "version": "1.31.0",
   "type": "commonjs",
   "repository": {
     "type": "git",
@@ -24,7 +24,7 @@
     "openai": "^4.95.0"
   },
   "peerDependencies": {
-    "genkit": "^1.30.1"
+    "genkit": "^1.31.0"
   },
   "devDependencies": {
     "@jest/globals": "^29.7.0",
@@ -33,7 +33,7 @@
     "npm-run-all": "^4.1.5",
     "ts-jest": "^29.1.2",
     "tsup": "^8.0.2",
-    "typescript": "^5.4.5"
+    "typescript": "^5.9.3"
   },
   "types": "./lib/index.d.ts",
   "exports": {