@aigne/openai 1.74.0-beta.1 → 1.74.0-beta.11

package/dist/openai-embedding-model.cjs

@@ -0,0 +1,68 @@
+const require_rolldown_runtime = require('./_virtual/rolldown_runtime.cjs');
+const require_openai = require('./openai.cjs');
+let _aigne_model_base = require("@aigne/model-base");
+let _aigne_model_base_utils_type_utils = require("@aigne/model-base/utils/type-utils");
+let zod = require("zod");
+
+//#region src/openai-embedding-model.ts
+const DEFAULT_MODEL = "text-embedding-3-small";
+const openAIEmbeddingModelInputSchema = _aigne_model_base.embeddingModelInputSchema.extend({});
+const openAIEmbeddingModelOptionsSchema = zod.z.object({
+	apiKey: zod.z.string().optional(),
+	baseURL: zod.z.string().optional(),
+	model: zod.z.string().optional(),
+	modelOptions: zod.z.object({}).optional(),
+	clientOptions: zod.z.object({}).optional()
+});
+var OpenAIEmbeddingModel = class extends _aigne_model_base.EmbeddingModel {
+	constructor(options) {
+		super({
+			...options,
+			inputSchema: openAIEmbeddingModelInputSchema,
+			description: options?.description ?? "Generate embeddings by OpenAI embedding models"
+		});
+		this.options = options;
+		if (options) (0, _aigne_model_base_utils_type_utils.checkArguments)(this.name, openAIEmbeddingModelOptionsSchema, options);
+	}
+	_client;
+	apiKeyEnvName = "OPENAI_API_KEY";
+	get client() {
+		if (this._client) return this._client;
+		const { apiKey, url } = this.credential;
+		if (!apiKey) throw new Error(`${this.name} requires an API key. Please provide it via \`options.apiKey\`, or set the \`${this.apiKeyEnvName}\` environment variable`);
+		this._client ??= new require_openai.CustomOpenAI({
+			baseURL: url,
+			apiKey,
+			...this.options?.clientOptions
+		});
+		return this._client;
+	}
+	get credential() {
+		return {
+			url: this.options?.baseURL || process.env.OPENAI_BASE_URL,
+			apiKey: this.options?.apiKey || process.env[this.apiKeyEnvName],
+			model: this.options?.model || DEFAULT_MODEL
+		};
+	}
+	async process(input, _options) {
+		const model = input.modelOptions?.model || this.credential.model;
+		const body = {
+			input: input.input,
+			model,
+			...input.dimensions ? { dimensions: input.dimensions } : {},
+			...input.modelOptions?.encoding_format ? { encoding_format: input.modelOptions.encoding_format } : {}
+		};
+		const response = await this.client.embeddings.create(body);
+		return {
+			embeddings: response.data.map((item) => item.embedding),
+			usage: {
+				inputTokens: response.usage?.prompt_tokens ?? 0,
+				outputTokens: 0
+			},
+			model: response.model
+		};
+	}
+};
+
+//#endregion
+exports.OpenAIEmbeddingModel = OpenAIEmbeddingModel;

package/dist/openai-embedding-model.d.cts

@@ -0,0 +1,28 @@
+import { EmbeddingModel, EmbeddingModelInput, EmbeddingModelOptions, EmbeddingModelOutput, ModelInvokeOptions } from "@aigne/model-base";
+import OpenAI, { ClientOptions } from "openai";
+
+//#region src/openai-embedding-model.d.ts
+interface OpenAIEmbeddingModelInput extends EmbeddingModelInput {}
+interface OpenAIEmbeddingModelOutput extends EmbeddingModelOutput {}
+interface OpenAIEmbeddingModelOptions extends EmbeddingModelOptions<OpenAIEmbeddingModelInput, OpenAIEmbeddingModelOutput> {
+  apiKey?: string;
+  baseURL?: string;
+  model?: string;
+  clientOptions?: Partial<ClientOptions>;
+}
+declare class OpenAIEmbeddingModel extends EmbeddingModel<OpenAIEmbeddingModelInput, OpenAIEmbeddingModelOutput> {
+  options?: OpenAIEmbeddingModelOptions | undefined;
+  constructor(options?: OpenAIEmbeddingModelOptions | undefined);
+  protected _client?: OpenAI;
+  protected apiKeyEnvName: string;
+  get client(): OpenAI;
+  get credential(): {
+    url: string | undefined;
+    apiKey: string | undefined;
+    model: string;
+  };
+  process(input: OpenAIEmbeddingModelInput, _options: ModelInvokeOptions): Promise<OpenAIEmbeddingModelOutput>;
+}
+//#endregion
+export { OpenAIEmbeddingModel, OpenAIEmbeddingModelInput, OpenAIEmbeddingModelOptions, OpenAIEmbeddingModelOutput };
+//# sourceMappingURL=openai-embedding-model.d.cts.map

package/dist/openai-embedding-model.d.cts.map

@@ -0,0 +1 @@
+{"version":3,"file":"openai-embedding-model.d.cts","names":[],"sources":["../src/openai-embedding-model.ts"],"mappings":";;;;UAgBiB,yBAAA,SAAkC,mBAAA;AAAA,UAElC,0BAAA,SAAmC,oBAAA;AAAA,UAEnC,2BAAA,SACP,qBAAA,CAAsB,yBAAA,EAA2B,0BAAA;EACzD,MAAA;EACA,OAAA;EACA,KAAA;EACA,aAAA,GAAgB,OAAA,CAAQ,aAAA;AAAA;AAAA,cAab,oBAAA,SAA6B,cAAA,CACxC,yBAAA,EACA,0BAAA;EAE4B,OAAA,GAAU,2BAAA;cAAV,OAAA,GAAU,2BAAA;EAAA,UAS5B,OAAA,GAAU,MAAA;EAAA,UAEV,aAAA;EAAA,IAEN,MAAA,CAAA,GAAM,MAAA;EAAA,IAmBG,UAAA,CAAA;;;;;EAQE,OAAA,CACb,KAAA,EAAO,yBAAA,EACP,QAAA,EAAU,kBAAA,GACT,OAAA,CAAQ,0BAAA;AAAA"}

package/dist/openai-embedding-model.d.mts

@@ -0,0 +1,28 @@
+import { EmbeddingModel, EmbeddingModelInput, EmbeddingModelOptions, EmbeddingModelOutput, ModelInvokeOptions } from "@aigne/model-base";
+import OpenAI, { ClientOptions } from "openai";
+
+//#region src/openai-embedding-model.d.ts
+interface OpenAIEmbeddingModelInput extends EmbeddingModelInput {}
+interface OpenAIEmbeddingModelOutput extends EmbeddingModelOutput {}
+interface OpenAIEmbeddingModelOptions extends EmbeddingModelOptions<OpenAIEmbeddingModelInput, OpenAIEmbeddingModelOutput> {
+  apiKey?: string;
+  baseURL?: string;
+  model?: string;
+  clientOptions?: Partial<ClientOptions>;
+}
+declare class OpenAIEmbeddingModel extends EmbeddingModel<OpenAIEmbeddingModelInput, OpenAIEmbeddingModelOutput> {
+  options?: OpenAIEmbeddingModelOptions | undefined;
+  constructor(options?: OpenAIEmbeddingModelOptions | undefined);
+  protected _client?: OpenAI;
+  protected apiKeyEnvName: string;
+  get client(): OpenAI;
+  get credential(): {
+    url: string | undefined;
+    apiKey: string | undefined;
+    model: string;
+  };
+  process(input: OpenAIEmbeddingModelInput, _options: ModelInvokeOptions): Promise<OpenAIEmbeddingModelOutput>;
+}
+//#endregion
+export { OpenAIEmbeddingModel, OpenAIEmbeddingModelInput, OpenAIEmbeddingModelOptions, OpenAIEmbeddingModelOutput };
+//# sourceMappingURL=openai-embedding-model.d.mts.map

package/dist/openai-embedding-model.d.mts.map

@@ -0,0 +1 @@
+{"version":3,"file":"openai-embedding-model.d.mts","names":[],"sources":["../src/openai-embedding-model.ts"],"mappings":";;;;UAgBiB,yBAAA,SAAkC,mBAAA;AAAA,UAElC,0BAAA,SAAmC,oBAAA;AAAA,UAEnC,2BAAA,SACP,qBAAA,CAAsB,yBAAA,EAA2B,0BAAA;EACzD,MAAA;EACA,OAAA;EACA,KAAA;EACA,aAAA,GAAgB,OAAA,CAAQ,aAAA;AAAA;AAAA,cAab,oBAAA,SAA6B,cAAA,CACxC,yBAAA,EACA,0BAAA;EAE4B,OAAA,GAAU,2BAAA;cAAV,OAAA,GAAU,2BAAA;EAAA,UAS5B,OAAA,GAAU,MAAA;EAAA,UAEV,aAAA;EAAA,IAEN,MAAA,CAAA,GAAM,MAAA;EAAA,IAmBG,UAAA,CAAA;;;;;EAQE,OAAA,CACb,KAAA,EAAO,yBAAA,EACP,QAAA,EAAU,kBAAA,GACT,OAAA,CAAQ,0BAAA;AAAA"}

package/dist/openai-embedding-model.mjs

@@ -0,0 +1,68 @@
+import { CustomOpenAI } from "./openai.mjs";
+import { EmbeddingModel, embeddingModelInputSchema } from "@aigne/model-base";
+import { checkArguments } from "@aigne/model-base/utils/type-utils";
+import { z } from "zod";
+
+//#region src/openai-embedding-model.ts
+const DEFAULT_MODEL = "text-embedding-3-small";
+const openAIEmbeddingModelInputSchema = embeddingModelInputSchema.extend({});
+const openAIEmbeddingModelOptionsSchema = z.object({
+	apiKey: z.string().optional(),
+	baseURL: z.string().optional(),
+	model: z.string().optional(),
+	modelOptions: z.object({}).optional(),
+	clientOptions: z.object({}).optional()
+});
+var OpenAIEmbeddingModel = class extends EmbeddingModel {
+	constructor(options) {
+		super({
+			...options,
+			inputSchema: openAIEmbeddingModelInputSchema,
+			description: options?.description ?? "Generate embeddings by OpenAI embedding models"
+		});
+		this.options = options;
+		if (options) checkArguments(this.name, openAIEmbeddingModelOptionsSchema, options);
+	}
+	_client;
+	apiKeyEnvName = "OPENAI_API_KEY";
+	get client() {
+		if (this._client) return this._client;
+		const { apiKey, url } = this.credential;
+		if (!apiKey) throw new Error(`${this.name} requires an API key. Please provide it via \`options.apiKey\`, or set the \`${this.apiKeyEnvName}\` environment variable`);
+		this._client ??= new CustomOpenAI({
+			baseURL: url,
+			apiKey,
+			...this.options?.clientOptions
+		});
+		return this._client;
+	}
+	get credential() {
+		return {
+			url: this.options?.baseURL || process.env.OPENAI_BASE_URL,
+			apiKey: this.options?.apiKey || process.env[this.apiKeyEnvName],
+			model: this.options?.model || DEFAULT_MODEL
+		};
+	}
+	async process(input, _options) {
+		const model = input.modelOptions?.model || this.credential.model;
+		const body = {
+			input: input.input,
+			model,
+			...input.dimensions ? { dimensions: input.dimensions } : {},
+			...input.modelOptions?.encoding_format ? { encoding_format: input.modelOptions.encoding_format } : {}
+		};
+		const response = await this.client.embeddings.create(body);
+		return {
+			embeddings: response.data.map((item) => item.embedding),
+			usage: {
+				inputTokens: response.usage?.prompt_tokens ?? 0,
+				outputTokens: 0
+			},
+			model: response.model
+		};
+	}
+};
+
+//#endregion
+export { OpenAIEmbeddingModel };
+//# sourceMappingURL=openai-embedding-model.mjs.map

package/dist/openai-embedding-model.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"openai-embedding-model.mjs","names":[],"sources":["../src/openai-embedding-model.ts"],"sourcesContent":["import {\n  EmbeddingModel,\n  type EmbeddingModelInput,\n  type EmbeddingModelOptions,\n  type EmbeddingModelOutput,\n  embeddingModelInputSchema,\n  type ModelInvokeOptions,\n} from \"@aigne/model-base\";\nimport { checkArguments } from \"@aigne/model-base/utils/type-utils\";\nimport type OpenAI from \"openai\";\nimport type { ClientOptions } from \"openai\";\nimport { z } from \"zod\";\nimport { CustomOpenAI } from \"./openai.js\";\n\nconst DEFAULT_MODEL = \"text-embedding-3-small\";\n\nexport interface OpenAIEmbeddingModelInput extends EmbeddingModelInput {}\n\nexport interface OpenAIEmbeddingModelOutput extends EmbeddingModelOutput {}\n\nexport interface OpenAIEmbeddingModelOptions\n  extends EmbeddingModelOptions<OpenAIEmbeddingModelInput, OpenAIEmbeddingModelOutput> {\n  apiKey?: string;\n  baseURL?: string;\n  model?: string;\n  clientOptions?: Partial<ClientOptions>;\n}\n\nconst openAIEmbeddingModelInputSchema = embeddingModelInputSchema.extend({});\n\nconst openAIEmbeddingModelOptionsSchema = z.object({\n  apiKey: z.string().optional(),\n  baseURL: z.string().optional(),\n  model: z.string().optional(),\n  modelOptions: z.object({}).optional(),\n  clientOptions: z.object({}).optional(),\n});\n\nexport class OpenAIEmbeddingModel extends EmbeddingModel<\n  OpenAIEmbeddingModelInput,\n  OpenAIEmbeddingModelOutput\n> {\n  constructor(public override options?: OpenAIEmbeddingModelOptions) {\n    super({\n      ...options,\n      inputSchema: openAIEmbeddingModelInputSchema,\n      description: options?.description ?? \"Generate embeddings by OpenAI embedding models\",\n    });\n    if (options) checkArguments(this.name, openAIEmbeddingModelOptionsSchema, options);\n  }\n\n  protected _client?: OpenAI;\n\n  protected apiKeyEnvName = \"OPENAI_API_KEY\";\n\n  get client() {\n    if (this._client) return this._client;\n\n    const { apiKey, url } = this.credential;\n\n    if (!apiKey)\n      throw new Error(\n        `${this.name} requires an API key. Please provide it via \\`options.apiKey\\`, or set the \\`${this.apiKeyEnvName}\\` environment variable`,\n      );\n\n    this._client ??= new CustomOpenAI({\n      baseURL: url,\n      apiKey,\n      ...this.options?.clientOptions,\n    });\n\n    return this._client;\n  }\n\n  override get credential() {\n    return {\n      url: this.options?.baseURL || process.env.OPENAI_BASE_URL,\n      apiKey: this.options?.apiKey || process.env[this.apiKeyEnvName],\n      model: this.options?.model || DEFAULT_MODEL,\n    };\n  }\n\n  override async process(\n    input: OpenAIEmbeddingModelInput,\n    _options: ModelInvokeOptions,\n  ): Promise<OpenAIEmbeddingModelOutput> {\n    const model = input.modelOptions?.model || this.credential.model;\n\n    const body: OpenAI.EmbeddingCreateParams = {\n      input: input.input,\n      model,\n      ...(input.dimensions ? { dimensions: input.dimensions } : {}),\n      ...(input.modelOptions?.encoding_format\n        ? { encoding_format: input.modelOptions.encoding_format as \"float\" | \"base64\" }\n        : {}),\n    };\n\n    const response = await this.client.embeddings.create(body);\n\n    return {\n      embeddings: response.data.map((item) => item.embedding),\n      usage: {\n        inputTokens: response.usage?.prompt_tokens ?? 0,\n        outputTokens: 0,\n      },\n      model: response.model,\n    };\n  }\n}\n"],"mappings":";;;;;;AAcA,MAAM,gBAAgB;AActB,MAAM,kCAAkC,0BAA0B,OAAO,EAAE,CAAC;AAE5E,MAAM,oCAAoC,EAAE,OAAO;CACjD,QAAQ,EAAE,QAAQ,CAAC,UAAU;CAC7B,SAAS,EAAE,QAAQ,CAAC,UAAU;CAC9B,OAAO,EAAE,QAAQ,CAAC,UAAU;CAC5B,cAAc,EAAE,OAAO,EAAE,CAAC,CAAC,UAAU;CACrC,eAAe,EAAE,OAAO,EAAE,CAAC,CAAC,UAAU;CACvC,CAAC;AAEF,IAAa,uBAAb,cAA0C,eAGxC;CACA,YAAY,AAAgB,SAAuC;AACjE,QAAM;GACJ,GAAG;GACH,aAAa;GACb,aAAa,SAAS,eAAe;GACtC,CAAC;EALwB;AAM1B,MAAI,QAAS,gBAAe,KAAK,MAAM,mCAAmC,QAAQ;;CAGpF,AAAU;CAEV,AAAU,gBAAgB;CAE1B,IAAI,SAAS;AACX,MAAI,KAAK,QAAS,QAAO,KAAK;EAE9B,MAAM,EAAE,QAAQ,QAAQ,KAAK;AAE7B,MAAI,CAAC,OACH,OAAM,IAAI,MACR,GAAG,KAAK,KAAK,+EAA+E,KAAK,cAAc,yBAChH;AAEH,OAAK,YAAY,IAAI,aAAa;GAChC,SAAS;GACT;GACA,GAAG,KAAK,SAAS;GAClB,CAAC;AAEF,SAAO,KAAK;;CAGd,IAAa,aAAa;AACxB,SAAO;GACL,KAAK,KAAK,SAAS,WAAW,QAAQ,IAAI;GAC1C,QAAQ,KAAK,SAAS,UAAU,QAAQ,IAAI,KAAK;GACjD,OAAO,KAAK,SAAS,SAAS;GAC/B;;CAGH,MAAe,QACb,OACA,UACqC;EACrC,MAAM,QAAQ,MAAM,cAAc,SAAS,KAAK,WAAW;EAE3D,MAAM,OAAqC;GACzC,OAAO,MAAM;GACb;GACA,GAAI,MAAM,aAAa,EAAE,YAAY,MAAM,YAAY,GAAG,EAAE;GAC5D,GAAI,MAAM,cAAc,kBACpB,EAAE,iBAAiB,MAAM,aAAa,iBAAuC,GAC7E,EAAE;GACP;EAED,MAAM,WAAW,MAAM,KAAK,OAAO,WAAW,OAAO,KAAK;AAE1D,SAAO;GACL,YAAY,SAAS,KAAK,KAAK,SAAS,KAAK,UAAU;GACvD,OAAO;IACL,aAAa,SAAS,OAAO,iBAAiB;IAC9C,cAAc;IACf;GACD,OAAO,SAAS;GACjB"}

package/dist/openai-image-model.cjs

@@ -1,9 +1,9 @@
 const require_rolldown_runtime = require('./_virtual/rolldown_runtime.cjs');
 const require_openai = require('./openai.cjs');
-let _aigne_core = require("@aigne/core");
-let _aigne_core_utils_type_utils = require("@aigne/core/utils/type-utils");
+let _aigne_model_base = require("@aigne/model-base");
+let _aigne_model_base_utils_type_utils = require("@aigne/model-base/utils/type-utils");
 let zod = require("zod");
-let _aigne_core_utils_camelize = require("@aigne/core/utils/camelize");
+let _aigne_model_base_utils_camelize = require("@aigne/model-base/utils/camelize");
 
 //#region src/openai-image-model.ts
 const DEFAULT_MODEL = "dall-e-2";
@@ -35,7 +35,7 @@ const SUPPORTED_PARAMS = {
 	]
 };
 const SUPPORT_EDIT_MODELS = ["gpt-image-1"];
-const openAIImageModelInputSchema = _aigne_core.imageModelInputSchema.extend({});
+const openAIImageModelInputSchema = _aigne_model_base.imageModelInputSchema.extend({});
 const openAIImageModelOptionsSchema = zod.z.object({
 	apiKey: zod.z.string().optional(),
 	baseURL: zod.z.string().optional(),
@@ -43,7 +43,7 @@ const openAIImageModelOptionsSchema = zod.z.object({
 	modelOptions: zod.z.object({}).optional(),
 	clientOptions: zod.z.object({}).optional()
 });
-var OpenAIImageModel = class extends _aigne_core.ImageModel {
+var OpenAIImageModel = class extends _aigne_model_base.ImageModel {
 	constructor(options) {
 		super({
 			...options,
@@ -51,7 +51,7 @@ var OpenAIImageModel = class extends _aigne_core.ImageModel {
 			description: options?.description ?? "Draw or edit image by OpenAI image models"
 		});
 		this.options = options;
-		if (options) (0, _aigne_core_utils_type_utils.checkArguments)(this.name, openAIImageModelOptionsSchema, options);
+		if (options) (0, _aigne_model_base_utils_type_utils.checkArguments)(this.name, openAIImageModelOptionsSchema, options);
 	}
 	_client;
 	apiKeyEnvName = "OPENAI_API_KEY";
@@ -85,7 +85,7 @@ var OpenAIImageModel = class extends _aigne_core.ImageModel {
 		const model = input.modelOptions?.model || this.credential.model;
 		if (input.image?.length && !SUPPORT_EDIT_MODELS.includes(model)) throw new Error(`Model ${model} does not support image editing`);
 		const body = {
-			...(0, _aigne_core_utils_camelize.snakelize)((0, _aigne_core_utils_type_utils.pick)({
+			...(0, _aigne_model_base_utils_camelize.snakelize)((0, _aigne_model_base_utils_type_utils.pick)({
 				...this.modelOptions,
 				...input.modelOptions,
 				...input

package/dist/openai-image-model.d.cts

@@ -1,36 +1,36 @@
-import { AgentInvokeOptions, ImageModel, ImageModelInput, ImageModelOptions, ImageModelOutput } from "@aigne/core";
+import { ImageModel, ImageModelInput, ImageModelOptions, ImageModelOutput, ModelInvokeOptions } from "@aigne/model-base";
 import OpenAI, { ClientOptions } from "openai";
-import { Camelize } from "@aigne/core/utils/camelize";
+import { Camelize } from "@aigne/model-base/utils/camelize";
 
 //#region src/openai-image-model.d.ts
 interface OpenAIImageModelInput extends ImageModelInput, Camelize<Omit<OpenAI.ImageGenerateParams | OpenAI.ImageEditParams, "prompt" | "model" | "n" | "response_format">> {}
 interface OpenAIImageModelOutput extends ImageModelOutput {}
 interface OpenAIImageModelOptions extends ImageModelOptions<OpenAIImageModelInput, OpenAIImageModelOutput> {
   /**
-       * API key for OpenAI API
-       *
-       * If not provided, will look for OPENAI_API_KEY in environment variables
-       */
+   * API key for OpenAI API
+   *
+   * If not provided, will look for OPENAI_API_KEY in environment variables
+   */
   apiKey?: string;
   /**
-       * Base URL for OpenAI API
-       *
-       * Useful for proxies or alternate endpoints
-       */
+   * Base URL for OpenAI API
+   *
+   * Useful for proxies or alternate endpoints
+   */
   baseURL?: string;
   /**
-       * OpenAI model to use
-       *
-       * Defaults to 'dall-e-2'
-       */
+   * OpenAI model to use
+   *
+   * Defaults to 'dall-e-2'
+   */
   model?: string;
   /**
-       * Additional model options to control behavior
-       */
+   * Additional model options to control behavior
+   */
   modelOptions?: Omit<Partial<OpenAIImageModelInput>, "model">;
   /**
-       * Client options for OpenAI API
-       */
+   * Client options for OpenAI API
+   */
   clientOptions?: Partial<ClientOptions>;
 }
 declare class OpenAIImageModel extends ImageModel<OpenAIImageModelInput, OpenAIImageModelOutput> {
@@ -46,11 +46,11 @@ declare class OpenAIImageModel extends ImageModel<OpenAIImageModelInput, OpenAII
   };
   get modelOptions(): Omit<Partial<OpenAIImageModelInput>, "model"> | undefined;
   /**
-       * Process the input and generate a response
-       * @param input The input to process
-       * @returns The generated response
-       */
-  process(input: OpenAIImageModelInput, _options: AgentInvokeOptions): Promise<OpenAIImageModelOutput>;
+   * Process the input and generate a response
+   * @param input The input to process
+   * @returns The generated response
+   */
+  process(input: OpenAIImageModelInput, _options: ModelInvokeOptions): Promise<OpenAIImageModelOutput>;
 }
 //#endregion
 export { OpenAIImageModel, OpenAIImageModelInput, OpenAIImageModelOptions, OpenAIImageModelOutput };

package/dist/openai-image-model.d.cts.map

@@ -1 +1 @@
-{"version":3,"file":"openai-image-model.d.cts","names":[],"sources":["../src/openai-image-model.ts"],"mappings":";;;;;UAqCiB,qBAAA,SACP,eAAA,EACN,QAAA,CACE,IAAA,CACE,MAAA,CAAO,mBAAA,GAAsB,MAAA,CAAO,eAAA;AAAA,UAK3B,sBAAA,SAA+B,gBAAA;AAAA,UAE/B,uBAAA,SACP,iBAAA,CAAkB,qBAAA,EAAuB,sBAAA;EAAA;;;;;EAAA,MAAA;EAAA;;;;;EAAA,OAAA;EAAA;;;;;EAAA,KAAA;EAAA;;;EAAA,YAAA,GAyBlC,IAAA,CAAK,OAAA,CAAQ,qBAAA;EAAA;;;EAAA,aAAA,GAKZ,OAAA,CAAQ,aAAA;AAAA;AAAA,cAeb,gBAAA,SAAyB,UAAA,CAAW,qBAAA,EAAuB,sBAAA;EAAA,OAAA,GAChC,uBAAA;EAAA,YAAA,OAAA,GAAA,uBAAA;EAAA,UAAA,OAAA,GASlB,MAAA;EAAA,UAAA,aAAA;EAAA,IAAA,OAAA,GAIV,MAAA;EAAA,IAAA,WAAA;IAAA,GAAA;IAAA,MAAA;IAAA,KAAA;EAAA;EAAA,IAAA,aAAA,GA2BM,IAAA,CAAA,OAAA,CAAA,qBAAA;EAAA;;;;;EAAA,QAAA,KAAA,EAUP,qBAAA,EAAA,QAAA,EACG,kBAAA,GACT,OAAA,CAAQ,sBAAA;AAAA"}
+{"version":3,"file":"openai-image-model.d.cts","names":[],"sources":["../src/openai-image-model.ts"],"mappings":";;;;;UAqCiB,qBAAA,SACP,eAAA,EACN,QAAA,CACE,IAAA,CACE,MAAA,CAAO,mBAAA,GAAsB,MAAA,CAAO,eAAA;AAAA,UAK3B,sBAAA,SAA+B,gBAAA;AAAA,UAE/B,uBAAA,SACP,iBAAA,CAAkB,qBAAA,EAAuB,sBAAA;EARd;;;;;EAcnC,MAAA;EAjBQ;;;;;EAwBR,OAAA;EArB0C;;;AAK5C;;EAuBE,KAAA;EAvB8C;;AAEhD;EA0BE,YAAA,GAAe,IAAA,CAAK,OAAA,CAAQ,qBAAA;;;;EAK5B,aAAA,GAAgB,OAAA,CAAQ,aAAA;AAAA;AAAA,cAeb,gBAAA,SAAyB,UAAA,CAAW,qBAAA,EAAuB,sBAAA;EAC1C,OAAA,GAAU,uBAAA;cAAV,OAAA,GAAU,uBAAA;EAAA,UAS5B,OAAA,GAAU,MAAA;EAAA,UAEV,aAAA;EAAA,IAEN,MAAA,CAAA,GAAM,MAAA;EAAA,IAmBG,UAAA,CAAA;;;;;MAQT,YAAA,CAAA,GAAY,IAAA,CAAA,OAAA,CAAA,qBAAA;EA7DhB;;;;;EAsEe,OAAA,CACb,KAAA,EAAO,qBAAA,EACP,QAAA,EAAU,kBAAA,GACT,OAAA,CAAQ,sBAAA;AAAA"}

package/dist/openai-image-model.d.mts

@@ -1,36 +1,36 @@
-import { AgentInvokeOptions, ImageModel, ImageModelInput, ImageModelOptions, ImageModelOutput } from "@aigne/core";
+import { ImageModel, ImageModelInput, ImageModelOptions, ImageModelOutput, ModelInvokeOptions } from "@aigne/model-base";
 import OpenAI, { ClientOptions } from "openai";
-import { Camelize } from "@aigne/core/utils/camelize";
+import { Camelize } from "@aigne/model-base/utils/camelize";
 
 //#region src/openai-image-model.d.ts
 interface OpenAIImageModelInput extends ImageModelInput, Camelize<Omit<OpenAI.ImageGenerateParams | OpenAI.ImageEditParams, "prompt" | "model" | "n" | "response_format">> {}
 interface OpenAIImageModelOutput extends ImageModelOutput {}
 interface OpenAIImageModelOptions extends ImageModelOptions<OpenAIImageModelInput, OpenAIImageModelOutput> {
   /**
-       * API key for OpenAI API
-       *
-       * If not provided, will look for OPENAI_API_KEY in environment variables
-       */
+   * API key for OpenAI API
+   *
+   * If not provided, will look for OPENAI_API_KEY in environment variables
+   */
   apiKey?: string;
   /**
-       * Base URL for OpenAI API
-       *
-       * Useful for proxies or alternate endpoints
-       */
+   * Base URL for OpenAI API
+   *
+   * Useful for proxies or alternate endpoints
+   */
   baseURL?: string;
   /**
-       * OpenAI model to use
-       *
-       * Defaults to 'dall-e-2'
-       */
+   * OpenAI model to use
+   *
+   * Defaults to 'dall-e-2'
+   */
   model?: string;
   /**
-       * Additional model options to control behavior
-       */
+   * Additional model options to control behavior
+   */
   modelOptions?: Omit<Partial<OpenAIImageModelInput>, "model">;
   /**
-       * Client options for OpenAI API
-       */
+   * Client options for OpenAI API
+   */
   clientOptions?: Partial<ClientOptions>;
 }
 declare class OpenAIImageModel extends ImageModel<OpenAIImageModelInput, OpenAIImageModelOutput> {
@@ -46,11 +46,11 @@ declare class OpenAIImageModel extends ImageModel<OpenAIImageModelInput, OpenAII
   };
   get modelOptions(): Omit<Partial<OpenAIImageModelInput>, "model"> | undefined;
   /**
-       * Process the input and generate a response
-       * @param input The input to process
-       * @returns The generated response
-       */
-  process(input: OpenAIImageModelInput, _options: AgentInvokeOptions): Promise<OpenAIImageModelOutput>;
+   * Process the input and generate a response
+   * @param input The input to process
+   * @returns The generated response
+   */
+  process(input: OpenAIImageModelInput, _options: ModelInvokeOptions): Promise<OpenAIImageModelOutput>;
 }
 //#endregion
 export { OpenAIImageModel, OpenAIImageModelInput, OpenAIImageModelOptions, OpenAIImageModelOutput };

package/dist/openai-image-model.d.mts.map

@@ -1 +1 @@
-{"version":3,"file":"openai-image-model.d.mts","names":[],"sources":["../src/openai-image-model.ts"],"mappings":";;;;;UAqCiB,qBAAA,SACP,eAAA,EACN,QAAA,CACE,IAAA,CACE,MAAA,CAAO,mBAAA,GAAsB,MAAA,CAAO,eAAA;AAAA,UAK3B,sBAAA,SAA+B,gBAAA;AAAA,UAE/B,uBAAA,SACP,iBAAA,CAAkB,qBAAA,EAAuB,sBAAA;EAAA;;;;;EAAA,MAAA;EAAA;;;;;EAAA,OAAA;EAAA;;;;;EAAA,KAAA;EAAA;;;EAAA,YAAA,GAyBlC,IAAA,CAAK,OAAA,CAAQ,qBAAA;EAAA;;;EAAA,aAAA,GAKZ,OAAA,CAAQ,aAAA;AAAA;AAAA,cAeb,gBAAA,SAAyB,UAAA,CAAW,qBAAA,EAAuB,sBAAA;EAAA,OAAA,GAChC,uBAAA;EAAA,YAAA,OAAA,GAAA,uBAAA;EAAA,UAAA,OAAA,GASlB,MAAA;EAAA,UAAA,aAAA;EAAA,IAAA,OAAA,GAIV,MAAA;EAAA,IAAA,WAAA;IAAA,GAAA;IAAA,MAAA;IAAA,KAAA;EAAA;EAAA,IAAA,aAAA,GA2BM,IAAA,CAAA,OAAA,CAAA,qBAAA;EAAA;;;;;EAAA,QAAA,KAAA,EAUP,qBAAA,EAAA,QAAA,EACG,kBAAA,GACT,OAAA,CAAQ,sBAAA;AAAA"}
+{"version":3,"file":"openai-image-model.d.mts","names":[],"sources":["../src/openai-image-model.ts"],"mappings":";;;;;UAqCiB,qBAAA,SACP,eAAA,EACN,QAAA,CACE,IAAA,CACE,MAAA,CAAO,mBAAA,GAAsB,MAAA,CAAO,eAAA;AAAA,UAK3B,sBAAA,SAA+B,gBAAA;AAAA,UAE/B,uBAAA,SACP,iBAAA,CAAkB,qBAAA,EAAuB,sBAAA;EARd;;;;;EAcnC,MAAA;EAjBQ;;;;;EAwBR,OAAA;EArB0C;;;AAK5C;;EAuBE,KAAA;EAvB8C;;AAEhD;EA0BE,YAAA,GAAe,IAAA,CAAK,OAAA,CAAQ,qBAAA;;;;EAK5B,aAAA,GAAgB,OAAA,CAAQ,aAAA;AAAA;AAAA,cAeb,gBAAA,SAAyB,UAAA,CAAW,qBAAA,EAAuB,sBAAA;EAC1C,OAAA,GAAU,uBAAA;cAAV,OAAA,GAAU,uBAAA;EAAA,UAS5B,OAAA,GAAU,MAAA;EAAA,UAEV,aAAA;EAAA,IAEN,MAAA,CAAA,GAAM,MAAA;EAAA,IAmBG,UAAA,CAAA;;;;;MAQT,YAAA,CAAA,GAAY,IAAA,CAAA,OAAA,CAAA,qBAAA;EA7DhB;;;;;EAsEe,OAAA,CACb,KAAA,EAAO,qBAAA,EACP,QAAA,EAAU,kBAAA,GACT,OAAA,CAAQ,sBAAA;AAAA"}

package/dist/openai-image-model.mjs

@@ -1,8 +1,8 @@
 import { CustomOpenAI } from "./openai.mjs";
-import { ImageModel, imageModelInputSchema } from "@aigne/core";
-import { checkArguments, pick } from "@aigne/core/utils/type-utils";
+import { ImageModel, imageModelInputSchema } from "@aigne/model-base";
+import { checkArguments, pick } from "@aigne/model-base/utils/type-utils";
 import { z } from "zod";
-import { snakelize } from "@aigne/core/utils/camelize";
+import { snakelize } from "@aigne/model-base/utils/camelize";
 
 //#region src/openai-image-model.ts
 const DEFAULT_MODEL = "dall-e-2";

package/dist/openai-image-model.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"openai-image-model.mjs","names":[],"sources":["../src/openai-image-model.ts"],"sourcesContent":["import {\n  type AgentInvokeOptions,\n  type FileUnionContent,\n  ImageModel,\n  type ImageModelInput,\n  type ImageModelOptions,\n  type ImageModelOutput,\n  imageModelInputSchema,\n} from \"@aigne/core\";\nimport { type Camelize, snakelize } from \"@aigne/core/utils/camelize\";\nimport { checkArguments, pick } from \"@aigne/core/utils/type-utils\";\nimport type OpenAI from \"openai\";\nimport type { ClientOptions } from \"openai\";\nimport { type ZodType, z } from \"zod\";\nimport { CustomOpenAI } from \"./openai.js\";\n\nconst DEFAULT_MODEL = \"dall-e-2\";\nconst OUTPUT_MIME_TYPE = \"image/png\";\n\nconst SUPPORTED_PARAMS: { [key: string]: any[] } = {\n  \"dall-e-2\": [\"prompt\", \"size\", \"n\"],\n  \"dall-e-3\": [\"prompt\", \"size\", \"n\", \"quality\", \"style\", \"user\"],\n  \"gpt-image-1\": [\n    \"prompt\",\n    \"size\",\n    \"background\",\n    \"moderation\",\n    \"outputCompression\",\n    \"outputFormat\",\n    \"quality\",\n    \"user\",\n    \"stream\",\n  ],\n};\n\nconst SUPPORT_EDIT_MODELS = [\"gpt-image-1\"];\n\nexport interface OpenAIImageModelInput\n  extends ImageModelInput,\n    Camelize<\n      Omit<\n        OpenAI.ImageGenerateParams | OpenAI.ImageEditParams,\n        \"prompt\" | \"model\" | \"n\" | \"response_format\"\n      >\n    > {}\n\nexport interface OpenAIImageModelOutput extends ImageModelOutput {}\n\nexport interface OpenAIImageModelOptions\n  extends ImageModelOptions<OpenAIImageModelInput, OpenAIImageModelOutput> {\n  /**\n   * API key for OpenAI API\n   *\n   * If not provided, will look for OPENAI_API_KEY in environment variables\n   */\n  apiKey?: string;\n\n  /**\n   * Base URL for OpenAI API\n   *\n   * Useful for proxies or alternate endpoints\n   */\n  baseURL?: string;\n\n  /**\n   * OpenAI model to use\n   *\n   * Defaults to 'dall-e-2'\n   */\n  model?: string;\n\n  /**\n   * Additional model options to control behavior\n   */\n  modelOptions?: Omit<Partial<OpenAIImageModelInput>, \"model\">;\n\n  /**\n   * Client options for OpenAI API\n   */\n  clientOptions?: Partial<ClientOptions>;\n}\n\nconst openAIImageModelInputSchema: ZodType<OpenAIImageModelInput> = imageModelInputSchema.extend(\n  {},\n);\n\nconst openAIImageModelOptionsSchema = z.object({\n  apiKey: z.string().optional(),\n  baseURL: z.string().optional(),\n  model: z.string().optional(),\n  modelOptions: z.object({}).optional(),\n  clientOptions: z.object({}).optional(),\n});\n\nexport class OpenAIImageModel extends ImageModel<OpenAIImageModelInput, OpenAIImageModelOutput> {\n  constructor(public override options?: OpenAIImageModelOptions) {\n    super({\n      ...options,\n      inputSchema: openAIImageModelInputSchema,\n      description: options?.description ?? \"Draw or edit image by OpenAI image models\",\n    });\n    if (options) checkArguments(this.name, openAIImageModelOptionsSchema, options);\n  }\n\n  protected _client?: OpenAI;\n\n  protected apiKeyEnvName = \"OPENAI_API_KEY\";\n\n  get client() {\n    if (this._client) return this._client;\n\n    const { apiKey, url } = this.credential;\n\n    if (!apiKey)\n      throw new Error(\n        `${this.name} requires an API key. Please provide it via \\`options.apiKey\\`, or set the \\`${this.apiKeyEnvName}\\` environment variable`,\n      );\n\n    this._client ??= new CustomOpenAI({\n      baseURL: url,\n      apiKey,\n      ...this.options?.clientOptions,\n    });\n\n    return this._client;\n  }\n\n  override get credential() {\n    return {\n      url: this.options?.baseURL || process.env.OPENAI_BASE_URL,\n      apiKey: this.options?.apiKey || process.env[this.apiKeyEnvName],\n      model: this.options?.model || DEFAULT_MODEL,\n    };\n  }\n\n  get modelOptions() {\n    return this.options?.modelOptions;\n  }\n\n  /**\n   * Process the input and generate a response\n   * @param input The input to process\n   * @returns The generated response\n   */\n  override async process(\n    input: OpenAIImageModelInput,\n    _options: AgentInvokeOptions,\n  ): Promise<OpenAIImageModelOutput> {\n    const model = input.modelOptions?.model || this.credential.model;\n\n    if (input.image?.length && !SUPPORT_EDIT_MODELS.includes(model)) {\n      throw new Error(`Model ${model} does not support image editing`);\n    }\n\n    const body: OpenAI.ImageGenerateParams | OpenAI.ImageEditParams = {\n      ...snakelize(\n        pick(\n          { ...this.modelOptions, ...input.modelOptions, ...input },\n          SUPPORTED_PARAMS[model] || SUPPORTED_PARAMS[DEFAULT_MODEL],\n        ),\n      ),\n      model,\n    };\n\n    const response = input.image?.length\n      ? ((await this.client.images.edit(\n          {\n            ...(body as OpenAI.ImageEditParams),\n            image: await Promise.all(\n              input.image.map((image) =>\n                this.transformFileType(\"file\", image).then(\n                  (file) =>\n                    new File([Buffer.from(file.data, \"base64\")], file.filename || \"image.png\", {\n                      type: file.mimeType,\n                    }),\n                ),\n              ),\n            ),\n          },\n          { stream: false },\n        )) as OpenAI.ImagesResponse)\n      : ((await this.client.images.generate(\n          { ...body },\n          { stream: false },\n        )) as OpenAI.ImagesResponse);\n\n    return {\n      images: (response.data ?? []).map<FileUnionContent>((image) => {\n        if (image.url) return { type: \"url\", url: image.url, mimeType: OUTPUT_MIME_TYPE };\n        if (image.b64_json)\n          return { type: \"file\", data: image.b64_json, mimeType: OUTPUT_MIME_TYPE };\n        throw new Error(\"Image response does not contain a valid URL or base64 data\");\n      }),\n      usage: {\n        inputTokens: response.usage?.input_tokens || 0,\n        outputTokens: response.usage?.output_tokens || 0,\n      },\n      model,\n    };\n  }\n}\n"],"mappings":";;;;;;;AAgBA,MAAM,gBAAgB;AACtB,MAAM,mBAAmB;AAEzB,MAAM,mBAA6C;CACjD,YAAY;EAAC;EAAU;EAAQ;EAAI;CACnC,YAAY;EAAC;EAAU;EAAQ;EAAK;EAAW;EAAS;EAAO;CAC/D,eAAe;EACb;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACD;CACF;AAED,MAAM,sBAAsB,CAAC,cAAc;AA+C3C,MAAM,8BAA8D,sBAAsB,OACxF,EAAE,CACH;AAED,MAAM,gCAAgC,EAAE,OAAO;CAC7C,QAAQ,EAAE,QAAQ,CAAC,UAAU;CAC7B,SAAS,EAAE,QAAQ,CAAC,UAAU;CAC9B,OAAO,EAAE,QAAQ,CAAC,UAAU;CAC5B,cAAc,EAAE,OAAO,EAAE,CAAC,CAAC,UAAU;CACrC,eAAe,EAAE,OAAO,EAAE,CAAC,CAAC,UAAU;CACvC,CAAC;AAEF,IAAa,mBAAb,cAAsC,WAA0D;CAC9F,YAAY,AAAgB,SAAmC;AAC7D,QAAM;GACJ,GAAG;GACH,aAAa;GACb,aAAa,SAAS,eAAe;GACtC,CAAC;EALwB;AAM1B,MAAI,QAAS,gBAAe,KAAK,MAAM,+BAA+B,QAAQ;;CAGhF,AAAU;CAEV,AAAU,gBAAgB;CAE1B,IAAI,SAAS;AACX,MAAI,KAAK,QAAS,QAAO,KAAK;EAE9B,MAAM,EAAE,QAAQ,QAAQ,KAAK;AAE7B,MAAI,CAAC,OACH,OAAM,IAAI,MACR,GAAG,KAAK,KAAK,+EAA+E,KAAK,cAAc,yBAChH;AAEH,OAAK,YAAY,IAAI,aAAa;GAChC,SAAS;GACT;GACA,GAAG,KAAK,SAAS;GAClB,CAAC;AAEF,SAAO,KAAK;;CAGd,IAAa,aAAa;AACxB,SAAO;GACL,KAAK,KAAK,SAAS,WAAW,QAAQ,IAAI;GAC1C,QAAQ,KAAK,SAAS,UAAU,QAAQ,IAAI,KAAK;GACjD,OAAO,KAAK,SAAS,SAAS;GAC/B;;CAGH,IAAI,eAAe;AACjB,SAAO,KAAK,SAAS;;;;;;;CAQvB,MAAe,QACb,OACA,UACiC;EACjC,MAAM,QAAQ,MAAM,cAAc,SAAS,KAAK,WAAW;AAE3D,MAAI,MAAM,OAAO,UAAU,CAAC,oBAAoB,SAAS,MAAM,CAC7D,OAAM,IAAI,MAAM,SAAS,MAAM,iCAAiC;EAGlE,MAAM,OAA4D;GAChE,GAAG,UACD,KACE;IAAE,GAAG,KAAK;IAAc,GAAG,MAAM;IAAc,GAAG;IAAO,EACzD,iBAAiB,UAAU,iBAAiB,eAC7C,CACF;GACD;GACD;EAED,MAAM,WAAW,MAAM,OAAO,SACxB,MAAM,KAAK,OAAO,OAAO,KACzB;GACE,GAAI;GACJ,OAAO,MAAM,QAAQ,IACnB,MAAM,MAAM,KAAK,UACf,KAAK,kBAAkB,QAAQ,MAAM,CAAC,MACnC,SACC,IAAI,KAAK,CAAC,OAAO,KAAK,KAAK,MAAM,SAAS,CAAC,EAAE,KAAK,YAAY,aAAa,EACzE,MAAM,KAAK,UACZ,CAAC,CACL,CACF,CACF;GACF,EACD,EAAE,QAAQ,OAAO,CAClB,GACC,MAAM,KAAK,OAAO,OAAO,SACzB,EAAE,GAAG,MAAM,EACX,EAAE,QAAQ,OAAO,CAClB;AAEL,SAAO;GACL,SAAS,SAAS,QAAQ,EAAE,EAAE,KAAuB,UAAU;AAC7D,QAAI,MAAM,IAAK,QAAO;KAAE,MAAM;KAAO,KAAK,MAAM;KAAK,UAAU;KAAkB;AACjF,QAAI,MAAM,SACR,QAAO;KAAE,MAAM;KAAQ,MAAM,MAAM;KAAU,UAAU;KAAkB;AAC3E,UAAM,IAAI,MAAM,6DAA6D;KAC7E;GACF,OAAO;IACL,aAAa,SAAS,OAAO,gBAAgB;IAC7C,cAAc,SAAS,OAAO,iBAAiB;IAChD;GACD;GACD"}
+{"version":3,"file":"openai-image-model.mjs","names":[],"sources":["../src/openai-image-model.ts"],"sourcesContent":["import {\n  type FileUnionContent,\n  ImageModel,\n  type ImageModelInput,\n  type ImageModelOptions,\n  type ImageModelOutput,\n  imageModelInputSchema,\n  type ModelInvokeOptions,\n} from \"@aigne/model-base\";\nimport { type Camelize, snakelize } from \"@aigne/model-base/utils/camelize\";\nimport { checkArguments, pick } from \"@aigne/model-base/utils/type-utils\";\nimport type OpenAI from \"openai\";\nimport type { ClientOptions } from \"openai\";\nimport { type ZodType, z } from \"zod\";\nimport { CustomOpenAI } from \"./openai.js\";\n\nconst DEFAULT_MODEL = \"dall-e-2\";\nconst OUTPUT_MIME_TYPE = \"image/png\";\n\nconst SUPPORTED_PARAMS: { [key: string]: any[] } = {\n  \"dall-e-2\": [\"prompt\", \"size\", \"n\"],\n  \"dall-e-3\": [\"prompt\", \"size\", \"n\", \"quality\", \"style\", \"user\"],\n  \"gpt-image-1\": [\n    \"prompt\",\n    \"size\",\n    \"background\",\n    \"moderation\",\n    \"outputCompression\",\n    \"outputFormat\",\n    \"quality\",\n    \"user\",\n    \"stream\",\n  ],\n};\n\nconst SUPPORT_EDIT_MODELS = [\"gpt-image-1\"];\n\nexport interface OpenAIImageModelInput\n  extends ImageModelInput,\n    Camelize<\n      Omit<\n        OpenAI.ImageGenerateParams | OpenAI.ImageEditParams,\n        \"prompt\" | \"model\" | \"n\" | \"response_format\"\n      >\n    > {}\n\nexport interface OpenAIImageModelOutput extends ImageModelOutput {}\n\nexport interface OpenAIImageModelOptions\n  extends ImageModelOptions<OpenAIImageModelInput, OpenAIImageModelOutput> {\n  /**\n   * API key for OpenAI API\n   *\n   * If not provided, will look for OPENAI_API_KEY in environment variables\n   */\n  apiKey?: string;\n\n  /**\n   * Base URL for OpenAI API\n   *\n   * Useful for proxies or alternate endpoints\n   */\n  baseURL?: string;\n\n  /**\n   * OpenAI model to use\n   *\n   * Defaults to 'dall-e-2'\n   */\n  model?: string;\n\n  /**\n   * Additional model options to control behavior\n   */\n  modelOptions?: Omit<Partial<OpenAIImageModelInput>, \"model\">;\n\n  /**\n   * Client options for OpenAI API\n   */\n  clientOptions?: Partial<ClientOptions>;\n}\n\nconst openAIImageModelInputSchema: ZodType<OpenAIImageModelInput> = imageModelInputSchema.extend(\n  {},\n);\n\nconst openAIImageModelOptionsSchema = z.object({\n  apiKey: z.string().optional(),\n  baseURL: z.string().optional(),\n  model: z.string().optional(),\n  modelOptions: z.object({}).optional(),\n  clientOptions: z.object({}).optional(),\n});\n\nexport class OpenAIImageModel extends ImageModel<OpenAIImageModelInput, OpenAIImageModelOutput> {\n  constructor(public override options?: OpenAIImageModelOptions) {\n    super({\n      ...options,\n      inputSchema: openAIImageModelInputSchema,\n      description: options?.description ?? \"Draw or edit image by OpenAI image models\",\n    });\n    if (options) checkArguments(this.name, openAIImageModelOptionsSchema, options);\n  }\n\n  protected _client?: OpenAI;\n\n  protected apiKeyEnvName = \"OPENAI_API_KEY\";\n\n  get client() {\n    if (this._client) return this._client;\n\n    const { apiKey, url } = this.credential;\n\n    if (!apiKey)\n      throw new Error(\n        `${this.name} requires an API key. Please provide it via \\`options.apiKey\\`, or set the \\`${this.apiKeyEnvName}\\` environment variable`,\n      );\n\n    this._client ??= new CustomOpenAI({\n      baseURL: url,\n      apiKey,\n      ...this.options?.clientOptions,\n    });\n\n    return this._client;\n  }\n\n  override get credential() {\n    return {\n      url: this.options?.baseURL || process.env.OPENAI_BASE_URL,\n      apiKey: this.options?.apiKey || process.env[this.apiKeyEnvName],\n      model: this.options?.model || DEFAULT_MODEL,\n    };\n  }\n\n  get modelOptions() {\n    return this.options?.modelOptions;\n  }\n\n  /**\n   * Process the input and generate a response\n   * @param input The input to process\n   * @returns The generated response\n   */\n  override async process(\n    input: OpenAIImageModelInput,\n    _options: ModelInvokeOptions,\n  ): Promise<OpenAIImageModelOutput> {\n    const model = input.modelOptions?.model || this.credential.model;\n\n    if (input.image?.length && !SUPPORT_EDIT_MODELS.includes(model)) {\n      throw new Error(`Model ${model} does not support image editing`);\n    }\n\n    const body: OpenAI.ImageGenerateParams | OpenAI.ImageEditParams = {\n      ...snakelize(\n        pick(\n          { ...this.modelOptions, ...input.modelOptions, ...input },\n          SUPPORTED_PARAMS[model] || SUPPORTED_PARAMS[DEFAULT_MODEL],\n        ),\n      ),\n      model,\n    };\n\n    const response = input.image?.length\n      ? ((await this.client.images.edit(\n          {\n            ...(body as OpenAI.ImageEditParams),\n            image: await Promise.all(\n              input.image.map((image) =>\n                this.transformFileType(\"file\", image).then(\n                  (file) =>\n                    new File([Buffer.from(file.data, \"base64\")], file.filename || \"image.png\", {\n                      type: file.mimeType,\n                    }),\n                ),\n              ),\n            ),\n          },\n          { stream: false },\n        )) as OpenAI.ImagesResponse)\n      : ((await this.client.images.generate(\n          { ...body },\n          { stream: false },\n        )) as OpenAI.ImagesResponse);\n\n    return {\n      images: (response.data ?? []).map<FileUnionContent>((image) => {\n        if (image.url) return { type: \"url\", url: image.url, mimeType: OUTPUT_MIME_TYPE };\n        if (image.b64_json)\n          return { type: \"file\", data: image.b64_json, mimeType: OUTPUT_MIME_TYPE };\n        throw new Error(\"Image response does not contain a valid URL or base64 data\");\n      }),\n      usage: {\n        inputTokens: response.usage?.input_tokens || 0,\n        outputTokens: response.usage?.output_tokens || 0,\n      },\n      model,\n    };\n  }\n}\n"],"mappings":";;;;;;;AAgBA,MAAM,gBAAgB;AACtB,MAAM,mBAAmB;AAEzB,MAAM,mBAA6C;CACjD,YAAY;EAAC;EAAU;EAAQ;EAAI;CACnC,YAAY;EAAC;EAAU;EAAQ;EAAK;EAAW;EAAS;EAAO;CAC/D,eAAe;EACb;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACD;CACF;AAED,MAAM,sBAAsB,CAAC,cAAc;AA+C3C,MAAM,8BAA8D,sBAAsB,OACxF,EAAE,CACH;AAED,MAAM,gCAAgC,EAAE,OAAO;CAC7C,QAAQ,EAAE,QAAQ,CAAC,UAAU;CAC7B,SAAS,EAAE,QAAQ,CAAC,UAAU;CAC9B,OAAO,EAAE,QAAQ,CAAC,UAAU;CAC5B,cAAc,EAAE,OAAO,EAAE,CAAC,CAAC,UAAU;CACrC,eAAe,EAAE,OAAO,EAAE,CAAC,CAAC,UAAU;CACvC,CAAC;AAEF,IAAa,mBAAb,cAAsC,WAA0D;CAC9F,YAAY,AAAgB,SAAmC;AAC7D,QAAM;GACJ,GAAG;GACH,aAAa;GACb,aAAa,SAAS,eAAe;GACtC,CAAC;EALwB;AAM1B,MAAI,QAAS,gBAAe,KAAK,MAAM,+BAA+B,QAAQ;;CAGhF,AAAU;CAEV,AAAU,gBAAgB;CAE1B,IAAI,SAAS;AACX,MAAI,KAAK,QAAS,QAAO,KAAK;EAE9B,MAAM,EAAE,QAAQ,QAAQ,KAAK;AAE7B,MAAI,CAAC,OACH,OAAM,IAAI,MACR,GAAG,KAAK,KAAK,+EAA+E,KAAK,cAAc,yBAChH;AAEH,OAAK,YAAY,IAAI,aAAa;GAChC,SAAS;GACT;GACA,GAAG,KAAK,SAAS;GAClB,CAAC;AAEF,SAAO,KAAK;;CAGd,IAAa,aAAa;AACxB,SAAO;GACL,KAAK,KAAK,SAAS,WAAW,QAAQ,IAAI;GAC1C,QAAQ,KAAK,SAAS,UAAU,QAAQ,IAAI,KAAK;GACjD,OAAO,KAAK,SAAS,SAAS;GAC/B;;CAGH,IAAI,eAAe;AACjB,SAAO,KAAK,SAAS;;;;;;;CAQvB,MAAe,QACb,OACA,UACiC;EACjC,MAAM,QAAQ,MAAM,cAAc,SAAS,KAAK,WAAW;AAE3D,MAAI,MAAM,OAAO,UAAU,CAAC,oBAAoB,SAAS,MAAM,CAC7D,OAAM,IAAI,MAAM,SAAS,MAAM,iCAAiC;EAGlE,MAAM,OAA4D;GAChE,GAAG,UACD,KACE;IAAE,GAAG,KAAK;IAAc,GAAG,MAAM;IAAc,GAAG;IAAO,EACzD,iBAAiB,UAAU,iBAAiB,eAC7C,CACF;GACD;GACD;EAED,MAAM,WAAW,MAAM,OAAO,SACxB,MAAM,KAAK,OAAO,OAAO,KACzB;GACE,GAAI;GACJ,OAAO,MAAM,QAAQ,IACnB,MAAM,MAAM,KAAK,UACf,KAAK,kBAAkB,QAAQ,MAAM,CAAC,MACnC,SACC,IAAI,KAAK,CAAC,OAAO,KAAK,KAAK,MAAM,SAAS,CAAC,EAAE,KAAK,YAAY,aAAa,EACzE,MAAM,KAAK,UACZ,CAAC,CACL,CACF,CACF;GACF,EACD,EAAE,QAAQ,OAAO,CAClB,GACC,MAAM,KAAK,OAAO,OAAO,SACzB,EAAE,GAAG,MAAM,EACX,EAAE,QAAQ,OAAO,CAClB;AAEL,SAAO;GACL,SAAS,SAAS,QAAQ,EAAE,EAAE,KAAuB,UAAU;AAC7D,QAAI,MAAM,IAAK,QAAO;KAAE,MAAM;KAAO,KAAK,MAAM;KAAK,UAAU;KAAkB;AACjF,QAAI,MAAM,SACR,QAAO;KAAE,MAAM;KAAQ,MAAM,MAAM;KAAU,UAAU;KAAkB;AAC3E,UAAM,IAAI,MAAM,6DAA6D;KAC7E;GACF,OAAO;IACL,aAAa,SAAS,OAAO,gBAAgB;IAC7C,cAAc,SAAS,OAAO,iBAAiB;IAChD;GACD;GACD"}

package/dist/openai-video-model.cjs

@@ -1,14 +1,14 @@
 const require_rolldown_runtime = require('./_virtual/rolldown_runtime.cjs');
 const require_openai = require('./openai.cjs');
-let _aigne_core = require("@aigne/core");
-let _aigne_core_utils_logger = require("@aigne/core/utils/logger");
-let _aigne_core_utils_type_utils = require("@aigne/core/utils/type-utils");
+let _aigne_model_base = require("@aigne/model-base");
+let _aigne_model_base_utils_logger = require("@aigne/model-base/utils/logger");
+let _aigne_model_base_utils_type_utils = require("@aigne/model-base/utils/type-utils");
 let zod = require("zod");
 
 //#region src/openai-video-model.ts
 const DEFAULT_MODEL = "sora-2";
 const DEFAULT_SECONDS = 4;
-const openAIVideoModelInputSchema = _aigne_core.videoModelInputSchema.extend({
+const openAIVideoModelInputSchema = _aigne_model_base.videoModelInputSchema.extend({
 	model: zod.z.enum(["sora-2", "sora-2-pro"]).optional(),
 	seconds: zod.z.enum([
 		"4",
@@ -30,7 +30,7 @@ const openAIVideoModelOptionsSchema = zod.z.object({
 	clientOptions: zod.z.object({}).optional(),
 	pollingInterval: zod.z.number().optional()
 });
-var OpenAIVideoModel = class extends _aigne_core.VideoModel {
+var OpenAIVideoModel = class extends _aigne_model_base.VideoModel {
 	constructor(options) {
 		super({
 			...options,
@@ -38,7 +38,7 @@ var OpenAIVideoModel = class extends _aigne_core.VideoModel {
 			inputSchema: openAIVideoModelInputSchema
 		});
 		this.options = options;
-		if (options) (0, _aigne_core_utils_type_utils.checkArguments)(this.name, openAIVideoModelOptionsSchema, options);
+		if (options) (0, _aigne_model_base_utils_type_utils.checkArguments)(this.name, openAIVideoModelOptionsSchema, options);
 	}
 	/**
 	* @hidden
@@ -66,7 +66,7 @@ var OpenAIVideoModel = class extends _aigne_core.VideoModel {
 		return this.options?.modelOptions;
 	}
 	async downloadToFile(videoId) {
-		_aigne_core_utils_logger.logger.debug("Downloading video content...");
+		_aigne_model_base_utils_logger.logger.debug("Downloading video content...");
 		const arrayBuffer = await (await this.client.videos.downloadContent(videoId)).arrayBuffer();
 		return Buffer.from(arrayBuffer).toString("base64");
 	}
@@ -80,14 +80,14 @@ var OpenAIVideoModel = class extends _aigne_core.VideoModel {
 		if (input.size) createParams.size = input.size;
 		if (input.image) createParams.input_reference = await this.transformFileType("file", input.image).then((file) => new File([Buffer.from(file.data, "base64")], file.filename || "image.png", { type: file.mimeType }));
 		let video = await this.client.videos.create(createParams);
-		_aigne_core_utils_logger.logger.debug(`Video generation started: ${video.id}`);
+		_aigne_model_base_utils_logger.logger.debug(`Video generation started: ${video.id}`);
 		const pollingInterval = this.options?.pollingInterval ?? 2e3;
 		while (video.status === "in_progress" || video.status === "queued") {
 			await new Promise((resolve) => setTimeout(resolve, pollingInterval));
 			video = await this.client.videos.retrieve(video.id);
 			const progress = video.progress ?? 0;
 			const statusText = video.status === "queued" ? "Queued" : "Processing";
-			_aigne_core_utils_logger.logger.debug(`${statusText}: ${progress.toFixed(1)}%`);
+			_aigne_model_base_utils_logger.logger.debug(`${statusText}: ${progress.toFixed(1)}%`);
 		}
 		if (video.status === "failed") throw new Error(`Video generation failed: ${video.error?.message || "Unknown error"}`);
 		if (video.status !== "completed") throw new Error(`Unexpected video status: ${video.status}`);