@aigne/gemini 1.74.0-beta → 1.74.0-beta.10

package/dist/gemini-embedding-model.mjs

@@ -0,0 +1,66 @@
+import { EmbeddingModel, embeddingModelInputSchema } from "@aigne/model-base";
+import { checkArguments } from "@aigne/model-base/utils/type-utils";
+import { GoogleGenAI } from "@google/genai";
+import { z } from "zod";
+
+//#region src/gemini-embedding-model.ts
+const DEFAULT_MODEL = "text-embedding-004";
+const geminiEmbeddingModelInputSchema = embeddingModelInputSchema.extend({});
+const geminiEmbeddingModelOptionsSchema = z.object({
+	apiKey: z.string().optional(),
+	baseURL: z.string().optional(),
+	model: z.string().optional(),
+	modelOptions: z.object({}).optional(),
+	clientOptions: z.object({}).optional()
+});
+var GeminiEmbeddingModel = class extends EmbeddingModel {
+	constructor(options) {
+		super({
+			...options,
+			inputSchema: geminiEmbeddingModelInputSchema,
+			description: options?.description ?? "Generate embeddings by Gemini embedding models"
+		});
+		this.options = options;
+		if (options) checkArguments(this.name, geminiEmbeddingModelOptionsSchema, options);
+	}
+	_client;
+	apiKeyEnvName = "GEMINI_API_KEY";
+	get client() {
+		if (this._client) return this._client;
+		const { apiKey } = 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 GoogleGenAI({ apiKey });
+		return this._client;
+	}
+	get credential() {
+		return {
+			url: this.options?.baseURL || process.env.GEMINI_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 contents = Array.isArray(input.input) ? input.input : [input.input];
+		const config = {
+			...input.dimensions ? { outputDimensionality: input.dimensions } : {},
+			...input.modelOptions?.taskType ? { taskType: input.modelOptions.taskType } : {}
+		};
+		return {
+			embeddings: (await this.client.models.embedContent({
+				model,
+				contents,
+				config
+			})).embeddings?.map((e) => e.values ?? []) ?? [],
+			usage: {
+				inputTokens: 0,
+				outputTokens: 0
+			},
+			model
+		};
+	}
+};
+
+//#endregion
+export { GeminiEmbeddingModel };
+//# sourceMappingURL=gemini-embedding-model.mjs.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"gemini-embedding-model.mjs","names":[],"sources":["../src/gemini-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 EmbedContentConfig, GoogleGenAI } from \"@google/genai\";\nimport { z } from \"zod\";\n\nconst DEFAULT_MODEL = \"text-embedding-004\";\n\nexport interface GeminiEmbeddingModelInput extends EmbeddingModelInput {}\nexport interface GeminiEmbeddingModelOutput extends EmbeddingModelOutput {}\n\nexport interface GeminiEmbeddingModelOptions\n  extends EmbeddingModelOptions<GeminiEmbeddingModelInput, GeminiEmbeddingModelOutput> {\n  apiKey?: string;\n  baseURL?: string;\n  model?: string;\n  clientOptions?: Record<string, any>;\n}\n\nconst geminiEmbeddingModelInputSchema = embeddingModelInputSchema.extend({});\n\nconst geminiEmbeddingModelOptionsSchema = 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 GeminiEmbeddingModel extends EmbeddingModel<\n  GeminiEmbeddingModelInput,\n  GeminiEmbeddingModelOutput\n> {\n  constructor(public override options?: GeminiEmbeddingModelOptions) {\n    super({\n      ...options,\n      inputSchema: geminiEmbeddingModelInputSchema,\n      description: options?.description ?? \"Generate embeddings by Gemini embedding models\",\n    });\n    if (options) checkArguments(this.name, geminiEmbeddingModelOptionsSchema, options);\n  }\n\n  protected _client?: GoogleGenAI;\n\n  protected apiKeyEnvName = \"GEMINI_API_KEY\";\n\n  get client() {\n    if (this._client) return this._client;\n\n    const { apiKey } = 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 GoogleGenAI({ apiKey });\n\n    return this._client;\n  }\n\n  override get credential() {\n    return {\n      url: this.options?.baseURL || process.env.GEMINI_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: GeminiEmbeddingModelInput,\n    _options: ModelInvokeOptions,\n  ): Promise<GeminiEmbeddingModelOutput> {\n    const model = input.modelOptions?.model || this.credential.model;\n\n    const contents = Array.isArray(input.input) ? input.input : [input.input];\n\n    const config: EmbedContentConfig = {\n      ...(input.dimensions ? { outputDimensionality: input.dimensions } : {}),\n      ...(input.modelOptions?.taskType\n        ? { taskType: input.modelOptions.taskType as string }\n        : {}),\n    };\n\n    const response = await this.client.models.embedContent({\n      model,\n      contents,\n      config,\n    });\n\n    return {\n      embeddings: response.embeddings?.map((e) => e.values ?? []) ?? [],\n      usage: {\n        inputTokens: 0,\n        outputTokens: 0,\n      },\n      model,\n    };\n  }\n}\n"],"mappings":";;;;;;AAYA,MAAM,gBAAgB;AAatB,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,WAAW,KAAK;AAExB,MAAI,CAAC,OACH,OAAM,IAAI,MACR,GAAG,KAAK,KAAK,+EAA+E,KAAK,cAAc,yBAChH;AAEH,OAAK,YAAY,IAAI,YAAY,EAAE,QAAQ,CAAC;AAE5C,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,WAAW,MAAM,QAAQ,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,MAAM,MAAM;EAEzE,MAAM,SAA6B;GACjC,GAAI,MAAM,aAAa,EAAE,sBAAsB,MAAM,YAAY,GAAG,EAAE;GACtE,GAAI,MAAM,cAAc,WACpB,EAAE,UAAU,MAAM,aAAa,UAAoB,GACnD,EAAE;GACP;AAQD,SAAO;GACL,aAPe,MAAM,KAAK,OAAO,OAAO,aAAa;IACrD;IACA;IACA;IACD,CAAC,EAGqB,YAAY,KAAK,MAAM,EAAE,UAAU,EAAE,CAAC,IAAI,EAAE;GACjE,OAAO;IACL,aAAa;IACb,cAAc;IACf;GACD;GACD"}

package/dist/gemini-image-model.cjs

@@ -1,11 +1,11 @@
-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 _google_genai = require("@google/genai");
 let zod = require("zod");
 
 //#region src/gemini-image-model.ts
 const DEFAULT_MODEL = "imagen-4.0-generate-001";
-const geminiImageModelInputSchema = _aigne_core.imageModelInputSchema.extend({});
+const geminiImageModelInputSchema = _aigne_model_base.imageModelInputSchema.extend({});
 const geminiImageModelOptionsSchema = zod.z.object({
 	apiKey: zod.z.string().optional(),
 	baseURL: zod.z.string().optional(),
@@ -13,7 +13,7 @@ const geminiImageModelOptionsSchema = zod.z.object({
 	modelOptions: zod.z.object({}).optional(),
 	clientOptions: zod.z.object({}).optional()
 });
-var GeminiImageModel = class extends _aigne_core.ImageModel {
+var GeminiImageModel = class extends _aigne_model_base.ImageModel {
 	constructor(options) {
 		super({
 			...options,
@@ -21,7 +21,7 @@ var GeminiImageModel = class extends _aigne_core.ImageModel {
 			description: options?.description ?? "Draw or edit image by Gemini image models"
 		});
 		this.options = options;
-		if (options) (0, _aigne_core_utils_type_utils.checkArguments)(this.name, geminiImageModelOptionsSchema, options);
+		if (options) (0, _aigne_model_base_utils_type_utils.checkArguments)(this.name, geminiImageModelOptionsSchema, options);
 	}
 	_client;
 	apiKeyEnvName = "GEMINI_API_KEY";
@@ -66,7 +66,7 @@ var GeminiImageModel = class extends _aigne_core.ImageModel {
 				prompt: mergedInput.prompt,
 				config: {
 					numberOfImages: mergedInput.n || 1,
-					...(0, _aigne_core_utils_type_utils.pick)(mergedInput, [
+					...(0, _aigne_model_base_utils_type_utils.pick)(mergedInput, [
 						"seed",
 						"safetyFilterLevel",
 						"personGeneration",
@@ -87,7 +87,7 @@ var GeminiImageModel = class extends _aigne_core.ImageModel {
 				type: "file",
 				data: image.imageBytes,
 				mimeType: image.mimeType
-			} : void 0).filter(_aigne_core_utils_type_utils.isNonNullable) || [],
+			} : void 0).filter(_aigne_model_base_utils_type_utils.isNonNullable) || [],
 			usage: {
 				inputTokens: 0,
 				outputTokens: 0
@@ -133,7 +133,7 @@ var GeminiImageModel = class extends _aigne_core.ImageModel {
 			"topP",
 			"imageConfig"
 		];
-		const images = await Promise.all((0, _aigne_core_utils_type_utils.flat)(input.image).map(async (image) => {
+		const images = await Promise.all((0, _aigne_model_base_utils_type_utils.flat)(input.image).map(async (image) => {
 			const { data, mimeType } = await this.transformFileType("file", image);
 			return { inlineData: {
 				data,
@@ -146,7 +146,7 @@ var GeminiImageModel = class extends _aigne_core.ImageModel {
 			config: {
 				responseModalities: [_google_genai.Modality.TEXT, _google_genai.Modality.IMAGE],
 				candidateCount: input.n || 1,
-				...(0, _aigne_core_utils_type_utils.pick)(mergedInput, inputKeys)
+				...(0, _aigne_model_base_utils_type_utils.pick)(mergedInput, inputKeys)
 			}
 		});
 		return {
@@ -155,7 +155,7 @@ var GeminiImageModel = class extends _aigne_core.ImageModel {
 				data: part.inlineData.data,
 				filename: part.inlineData.displayName,
 				mimeType: part.inlineData.mimeType
-			} : null).filter(_aigne_core_utils_type_utils.isNonNullable),
+			} : null).filter(_aigne_model_base_utils_type_utils.isNonNullable),
 			usage: {
 				inputTokens: response.usageMetadata?.promptTokenCount || 0,
 				outputTokens: response.usageMetadata?.candidatesTokenCount || 0

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

@@ -1,4 +1,4 @@
-import { AgentInvokeOptions, ImageModel, ImageModelInput, ImageModelOptions, ImageModelOutput } from "@aigne/core";
+import { ImageModel, ImageModelInput, ImageModelOptions, ImageModelOutput, ModelInvokeOptions } from "@aigne/model-base";
 import { GenerateContentConfig, GenerateImagesConfig, GoogleGenAI } from "@google/genai";
 
 //#region src/gemini-image-model.d.ts
@@ -24,11 +24,11 @@ declare class GeminiImageModel extends ImageModel<GeminiImageModelInput, GeminiI
   };
   get modelOptions(): Omit<Partial<GeminiImageModelInput>, "model"> | undefined;
   /**
-       * Process the input and generate a response
-       * @param input The input to process
-       * @returns The generated response
-       */
-  process(input: GeminiImageModelInput, _options: AgentInvokeOptions): Promise<ImageModelOutput>;
+   * Process the input and generate a response
+   * @param input The input to process
+   * @returns The generated response
+   */
+  process(input: GeminiImageModelInput, _options: ModelInvokeOptions): Promise<ImageModelOutput>;
   private generateImageByImagenModel;
   private generateImageByGeminiModel;
 }

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

@@ -1 +1 @@
-{"version":3,"file":"gemini-image-model.d.cts","names":[],"sources":["../src/gemini-image-model.ts"],"mappings":";;;;UAqBiB,qBAAA,SACP,eAAA,EACN,oBAAA,EACA,qBAAA;AAAA,UACa,sBAAA,SAA+B,gBAAA;AAAA,UAE/B,uBAAA,SACP,iBAAA,CAAkB,qBAAA,EAAuB,sBAAA;EAAA,MAAA;EAAA,OAAA;EAAA,KAAA;EAAA,YAAA,GAIlC,IAAA,CAAK,OAAA,CAAQ,qBAAA;EAAA,aAAA,GACZ,MAAA;AAAA;AAAA,cAaL,gBAAA,SAAyB,UAAA,CAAW,qBAAA,EAAuB,sBAAA;EAAA,OAAA,GAChC,uBAAA;EAAA,YAAA,OAAA,GAAA,uBAAA;EAAA,UAAA,OAAA,GASlB,WAAA;EAAA,UAAA,aAAA;EAAA,IAAA,OAAA,GAIV,WAAA;EAAA,IAAA,WAAA;IAAA,GAAA;IAAA,MAAA;IAAA,KAAA;EAAA;EAAA,IAAA,aAAA,GAuBM,IAAA,CAAA,OAAA,CAAA,qBAAA;EAAA;;;;;EAAA,QAAA,KAAA,EAUP,qBAAA,EAAA,QAAA,EACG,kBAAA,GACT,OAAA,CAAQ,gBAAA;EAAA,QAAA,0BAAA;EAAA,QAAA,0BAAA;AAAA"}
+{"version":3,"file":"gemini-image-model.d.cts","names":[],"sources":["../src/gemini-image-model.ts"],"mappings":";;;;UAqBiB,qBAAA,SACP,eAAA,EACN,oBAAA,EACA,qBAAA;AAAA,UACa,sBAAA,SAA+B,gBAAA;AAAA,UAE/B,uBAAA,SACP,iBAAA,CAAkB,qBAAA,EAAuB,sBAAA;EACjD,MAAA;EACA,OAAA;EACA,KAAA;EACA,YAAA,GAAe,IAAA,CAAK,OAAA,CAAQ,qBAAA;EAC5B,aAAA,GAAgB,MAAA;AAAA;AAAA,cAaL,gBAAA,SAAyB,UAAA,CAAW,qBAAA,EAAuB,sBAAA;EAC1C,OAAA,GAAU,uBAAA;cAAV,OAAA,GAAU,uBAAA;EAAA,UAS5B,OAAA,GAAU,WAAA;EAAA,UAEV,aAAA;EAAA,IAEN,MAAA,CAAA,GAAM,WAAA;EAAA,IAeG,UAAA,CAAA;;;;;MAQT,YAAA,CAAA,GAAY,IAAA,CAAA,OAAA,CAAA,qBAAA;EAvDU;;;;;EAgEX,OAAA,CACb,KAAA,EAAO,qBAAA,EACP,QAAA,EAAU,kBAAA,GACT,OAAA,CAAQ,gBAAA;EAAA,QAcG,0BAAA;EAAA,QA+CA,0BAAA;AAAA"}

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

@@ -1,4 +1,4 @@
-import { AgentInvokeOptions, ImageModel, ImageModelInput, ImageModelOptions, ImageModelOutput } from "@aigne/core";
+import { ImageModel, ImageModelInput, ImageModelOptions, ImageModelOutput, ModelInvokeOptions } from "@aigne/model-base";
 import { GenerateContentConfig, GenerateImagesConfig, GoogleGenAI } from "@google/genai";
 
 //#region src/gemini-image-model.d.ts
@@ -24,11 +24,11 @@ declare class GeminiImageModel extends ImageModel<GeminiImageModelInput, GeminiI
   };
   get modelOptions(): Omit<Partial<GeminiImageModelInput>, "model"> | undefined;
   /**
-       * Process the input and generate a response
-       * @param input The input to process
-       * @returns The generated response
-       */
-  process(input: GeminiImageModelInput, _options: AgentInvokeOptions): Promise<ImageModelOutput>;
+   * Process the input and generate a response
+   * @param input The input to process
+   * @returns The generated response
+   */
+  process(input: GeminiImageModelInput, _options: ModelInvokeOptions): Promise<ImageModelOutput>;
   private generateImageByImagenModel;
   private generateImageByGeminiModel;
 }

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

@@ -1 +1 @@
-{"version":3,"file":"gemini-image-model.d.mts","names":[],"sources":["../src/gemini-image-model.ts"],"mappings":";;;;UAqBiB,qBAAA,SACP,eAAA,EACN,oBAAA,EACA,qBAAA;AAAA,UACa,sBAAA,SAA+B,gBAAA;AAAA,UAE/B,uBAAA,SACP,iBAAA,CAAkB,qBAAA,EAAuB,sBAAA;EAAA,MAAA;EAAA,OAAA;EAAA,KAAA;EAAA,YAAA,GAIlC,IAAA,CAAK,OAAA,CAAQ,qBAAA;EAAA,aAAA,GACZ,MAAA;AAAA;AAAA,cAaL,gBAAA,SAAyB,UAAA,CAAW,qBAAA,EAAuB,sBAAA;EAAA,OAAA,GAChC,uBAAA;EAAA,YAAA,OAAA,GAAA,uBAAA;EAAA,UAAA,OAAA,GASlB,WAAA;EAAA,UAAA,aAAA;EAAA,IAAA,OAAA,GAIV,WAAA;EAAA,IAAA,WAAA;IAAA,GAAA;IAAA,MAAA;IAAA,KAAA;EAAA;EAAA,IAAA,aAAA,GAuBM,IAAA,CAAA,OAAA,CAAA,qBAAA;EAAA;;;;;EAAA,QAAA,KAAA,EAUP,qBAAA,EAAA,QAAA,EACG,kBAAA,GACT,OAAA,CAAQ,gBAAA;EAAA,QAAA,0BAAA;EAAA,QAAA,0BAAA;AAAA"}
+{"version":3,"file":"gemini-image-model.d.mts","names":[],"sources":["../src/gemini-image-model.ts"],"mappings":";;;;UAqBiB,qBAAA,SACP,eAAA,EACN,oBAAA,EACA,qBAAA;AAAA,UACa,sBAAA,SAA+B,gBAAA;AAAA,UAE/B,uBAAA,SACP,iBAAA,CAAkB,qBAAA,EAAuB,sBAAA;EACjD,MAAA;EACA,OAAA;EACA,KAAA;EACA,YAAA,GAAe,IAAA,CAAK,OAAA,CAAQ,qBAAA;EAC5B,aAAA,GAAgB,MAAA;AAAA;AAAA,cAaL,gBAAA,SAAyB,UAAA,CAAW,qBAAA,EAAuB,sBAAA;EAC1C,OAAA,GAAU,uBAAA;cAAV,OAAA,GAAU,uBAAA;EAAA,UAS5B,OAAA,GAAU,WAAA;EAAA,UAEV,aAAA;EAAA,IAEN,MAAA,CAAA,GAAM,WAAA;EAAA,IAeG,UAAA,CAAA;;;;;MAQT,YAAA,CAAA,GAAY,IAAA,CAAA,OAAA,CAAA,qBAAA;EAvDU;;;;;EAgEX,OAAA,CACb,KAAA,EAAO,qBAAA,EACP,QAAA,EAAU,kBAAA,GACT,OAAA,CAAQ,gBAAA;EAAA,QAcG,0BAAA;EAAA,QA+CA,0BAAA;AAAA"}

package/dist/gemini-image-model.mjs

@@ -1,5 +1,5 @@
-import { ImageModel, imageModelInputSchema } from "@aigne/core";
-import { checkArguments, flat, isNonNullable, pick } from "@aigne/core/utils/type-utils";
+import { ImageModel, imageModelInputSchema } from "@aigne/model-base";
+import { checkArguments, flat, isNonNullable, pick } from "@aigne/model-base/utils/type-utils";
 import { GoogleGenAI, Modality } from "@google/genai";
 import { z } from "zod";
 

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

@@ -1 +1 @@
-{"version":3,"file":"gemini-image-model.mjs","names":[],"sources":["../src/gemini-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 { checkArguments, flat, isNonNullable, pick } from \"@aigne/core/utils/type-utils\";\nimport {\n  type GenerateContentConfig,\n  type GenerateImagesConfig,\n  GoogleGenAI,\n  Modality,\n  type PartUnion,\n} from \"@google/genai\";\nimport { z } from \"zod\";\n\nconst DEFAULT_MODEL = \"imagen-4.0-generate-001\";\n\nexport interface GeminiImageModelInput\n  extends ImageModelInput,\n    GenerateImagesConfig,\n    GenerateContentConfig {}\nexport interface GeminiImageModelOutput extends ImageModelOutput {}\n\nexport interface GeminiImageModelOptions\n  extends ImageModelOptions<GeminiImageModelInput, GeminiImageModelOutput> {\n  apiKey?: string;\n  baseURL?: string;\n  model?: string;\n  modelOptions?: Omit<Partial<GeminiImageModelInput>, \"model\">;\n  clientOptions?: Record<string, any>;\n}\n\nconst geminiImageModelInputSchema = imageModelInputSchema.extend({});\n\nconst geminiImageModelOptionsSchema = 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 GeminiImageModel extends ImageModel<GeminiImageModelInput, GeminiImageModelOutput> {\n  constructor(public override options?: GeminiImageModelOptions) {\n    super({\n      ...options,\n      inputSchema: geminiImageModelInputSchema,\n      description: options?.description ?? \"Draw or edit image by Gemini image models\",\n    });\n    if (options) checkArguments(this.name, geminiImageModelOptionsSchema, options);\n  }\n\n  protected _client?: GoogleGenAI;\n\n  protected apiKeyEnvName = \"GEMINI_API_KEY\";\n\n  get client() {\n    if (this._client) return this._client;\n\n    const { apiKey } = 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 GoogleGenAI({ apiKey });\n\n    return this._client;\n  }\n\n  override get credential() {\n    return {\n      url: this.options?.baseURL || process.env.GEMINI_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: GeminiImageModelInput,\n    _options: AgentInvokeOptions,\n  ): Promise<ImageModelOutput> {\n    const model = input.modelOptions?.model || this.credential.model;\n    const responseFormat = input.responseFormat || \"base64\";\n    if (responseFormat === \"url\") {\n      throw new Error(\"Gemini image models currently only support base64 format\");\n    }\n\n    if (model.includes(\"imagen\")) {\n      return this.generateImageByImagenModel(input);\n    }\n\n    return this.generateImageByGeminiModel(input);\n  }\n\n  private async generateImageByImagenModel(\n    input: GeminiImageModelInput,\n  ): Promise<ImageModelOutput> {\n    const model = input.modelOptions?.model || this.credential.model;\n\n    const mergedInput = { ...this.modelOptions, ...input.modelOptions, ...input };\n\n    const inputKeys = [\n      \"seed\",\n      \"safetyFilterLevel\",\n      \"personGeneration\",\n      \"outputMimeType\",\n      \"outputGcsUri\",\n      \"outputCompressionQuality\",\n      \"negativePrompt\",\n      \"language\",\n      \"includeSafetyAttributes\",\n      \"includeRaiReason\",\n      \"imageSize\",\n      \"guidanceScale\",\n      \"aspectRatio\",\n      \"addWatermark\",\n    ];\n\n    const response = await this.client.models.generateImages({\n      model,\n      prompt: mergedInput.prompt,\n      config: { numberOfImages: mergedInput.n || 1, ...pick(mergedInput, inputKeys) },\n    });\n\n    return {\n      images:\n        response.generatedImages\n          ?.map<FileUnionContent | undefined>(({ image }) =>\n            image?.imageBytes\n              ? { type: \"file\", data: image.imageBytes, mimeType: image.mimeType }\n              : undefined,\n          )\n          .filter(isNonNullable) || [],\n      usage: {\n        inputTokens: 0,\n        outputTokens: 0,\n      },\n      model,\n    };\n  }\n\n  private async generateImageByGeminiModel(\n    input: GeminiImageModelInput,\n  ): Promise<ImageModelOutput> {\n    const model = input.modelOptions?.model || this.credential.model;\n\n    const mergedInput = { ...this.modelOptions, ...input.modelOptions, ...input };\n\n    const inputKeys = [\n      \"abortSignal\",\n      \"audioTimestamp\",\n      \"automaticFunctionCalling\",\n      \"cachedContent\",\n      \"frequencyPenalty\",\n      \"httpOptions\",\n      \"labels\",\n      \"logprobs\",\n      \"maxOutputTokens\",\n      \"mediaResolution\",\n      \"modelSelectionConfig\",\n      \"presencePenalty\",\n      \"responseJsonSchema\",\n      \"responseLogprobs\",\n      \"responseMimeType\",\n      \"responseSchema\",\n      \"routingConfig\",\n      \"safetySettings\",\n      \"seed\",\n      \"speechConfig\",\n      \"stopSequences\",\n      \"systemInstruction\",\n      \"temperature\",\n      \"thinkingConfig\",\n      \"toolConfig\",\n      \"tools\",\n      \"topK\",\n      \"topP\",\n      \"imageConfig\",\n    ];\n\n    const images = await Promise.all(\n      flat(input.image).map<Promise<PartUnion>>(async (image) => {\n        const { data, mimeType } = await this.transformFileType(\"file\", image);\n        return { inlineData: { data, mimeType } };\n      }),\n    );\n\n    const response = await this.client.models.generateContent({\n      model,\n      contents: [{ text: input.prompt }, ...images],\n      config: {\n        responseModalities: [Modality.TEXT, Modality.IMAGE],\n        candidateCount: input.n || 1,\n        ...pick(mergedInput, inputKeys),\n      },\n    });\n\n    const allImages = (response.candidates ?? [])\n      .flatMap((candidate) => candidate.content?.parts ?? [])\n      .map<FileUnionContent | null>((part) =>\n        part.inlineData?.data\n          ? {\n              type: \"file\",\n              data: part.inlineData.data,\n              filename: part.inlineData.displayName,\n              mimeType: part.inlineData.mimeType,\n            }\n          : null,\n      )\n      .filter(isNonNullable);\n\n    return {\n      images: allImages,\n      usage: {\n        inputTokens: response.usageMetadata?.promptTokenCount || 0,\n        outputTokens: response.usageMetadata?.candidatesTokenCount || 0,\n      },\n      model,\n    };\n  }\n}\n"],"mappings":";;;;;;AAmBA,MAAM,gBAAgB;AAiBtB,MAAM,8BAA8B,sBAAsB,OAAO,EAAE,CAAC;AAEpE,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,WAAW,KAAK;AAExB,MAAI,CAAC,OACH,OAAM,IAAI,MACR,GAAG,KAAK,KAAK,+EAA+E,KAAK,cAAc,yBAChH;AAEH,OAAK,YAAY,IAAI,YAAY,EAAE,QAAQ,CAAC;AAE5C,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,UAC2B;EAC3B,MAAM,QAAQ,MAAM,cAAc,SAAS,KAAK,WAAW;AAE3D,OADuB,MAAM,kBAAkB,cACxB,MACrB,OAAM,IAAI,MAAM,2DAA2D;AAG7E,MAAI,MAAM,SAAS,SAAS,CAC1B,QAAO,KAAK,2BAA2B,MAAM;AAG/C,SAAO,KAAK,2BAA2B,MAAM;;CAG/C,MAAc,2BACZ,OAC2B;EAC3B,MAAM,QAAQ,MAAM,cAAc,SAAS,KAAK,WAAW;EAE3D,MAAM,cAAc;GAAE,GAAG,KAAK;GAAc,GAAG,MAAM;GAAc,GAAG;GAAO;AAyB7E,SAAO;GACL,SAPe,MAAM,KAAK,OAAO,OAAO,eAAe;IACvD;IACA,QAAQ,YAAY;IACpB,QAAQ;KAAE,gBAAgB,YAAY,KAAK;KAAG,GAAG,KAAK,aApBtC;MAChB;MACA;MACA;MACA;MACA;MACA;MACA;MACA;MACA;MACA;MACA;MACA;MACA;MACA;MACD,CAK8E;KAAE;IAChF,CAAC,EAIW,iBACL,KAAmC,EAAE,YACrC,OAAO,aACH;IAAE,MAAM;IAAQ,MAAM,MAAM;IAAY,UAAU,MAAM;IAAU,GAClE,OACL,CACA,OAAO,cAAc,IAAI,EAAE;GAChC,OAAO;IACL,aAAa;IACb,cAAc;IACf;GACD;GACD;;CAGH,MAAc,2BACZ,OAC2B;EAC3B,MAAM,QAAQ,MAAM,cAAc,SAAS,KAAK,WAAW;EAE3D,MAAM,cAAc;GAAE,GAAG,KAAK;GAAc,GAAG,MAAM;GAAc,GAAG;GAAO;EAE7E,MAAM,YAAY;GAChB;GACA;GACA;GACA;GACA;GACA;GACA;GACA;GACA;GACA;GACA;GACA;GACA;GACA;GACA;GACA;GACA;GACA;GACA;GACA;GACA;GACA;GACA;GACA;GACA;GACA;GACA;GACA;GACA;GACD;EAED,MAAM,SAAS,MAAM,QAAQ,IAC3B,KAAK,MAAM,MAAM,CAAC,IAAwB,OAAO,UAAU;GACzD,MAAM,EAAE,MAAM,aAAa,MAAM,KAAK,kBAAkB,QAAQ,MAAM;AACtE,UAAO,EAAE,YAAY;IAAE;IAAM;IAAU,EAAE;IACzC,CACH;EAED,MAAM,WAAW,MAAM,KAAK,OAAO,OAAO,gBAAgB;GACxD;GACA,UAAU,CAAC,EAAE,MAAM,MAAM,QAAQ,EAAE,GAAG,OAAO;GAC7C,QAAQ;IACN,oBAAoB,CAAC,SAAS,MAAM,SAAS,MAAM;IACnD,gBAAgB,MAAM,KAAK;IAC3B,GAAG,KAAK,aAAa,UAAU;IAChC;GACF,CAAC;AAgBF,SAAO;GACL,SAfiB,SAAS,cAAc,EAAE,EACzC,SAAS,cAAc,UAAU,SAAS,SAAS,EAAE,CAAC,CACtD,KAA8B,SAC7B,KAAK,YAAY,OACb;IACE,MAAM;IACN,MAAM,KAAK,WAAW;IACtB,UAAU,KAAK,WAAW;IAC1B,UAAU,KAAK,WAAW;IAC3B,GACD,KACL,CACA,OAAO,cAAc;GAItB,OAAO;IACL,aAAa,SAAS,eAAe,oBAAoB;IACzD,cAAc,SAAS,eAAe,wBAAwB;IAC/D;GACD;GACD"}
+{"version":3,"file":"gemini-image-model.mjs","names":[],"sources":["../src/gemini-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 { checkArguments, flat, isNonNullable, pick } from \"@aigne/model-base/utils/type-utils\";\nimport {\n  type GenerateContentConfig,\n  type GenerateImagesConfig,\n  GoogleGenAI,\n  Modality,\n  type PartUnion,\n} from \"@google/genai\";\nimport { z } from \"zod\";\n\nconst DEFAULT_MODEL = \"imagen-4.0-generate-001\";\n\nexport interface GeminiImageModelInput\n  extends ImageModelInput,\n    GenerateImagesConfig,\n    GenerateContentConfig {}\nexport interface GeminiImageModelOutput extends ImageModelOutput {}\n\nexport interface GeminiImageModelOptions\n  extends ImageModelOptions<GeminiImageModelInput, GeminiImageModelOutput> {\n  apiKey?: string;\n  baseURL?: string;\n  model?: string;\n  modelOptions?: Omit<Partial<GeminiImageModelInput>, \"model\">;\n  clientOptions?: Record<string, any>;\n}\n\nconst geminiImageModelInputSchema = imageModelInputSchema.extend({});\n\nconst geminiImageModelOptionsSchema = 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 GeminiImageModel extends ImageModel<GeminiImageModelInput, GeminiImageModelOutput> {\n  constructor(public override options?: GeminiImageModelOptions) {\n    super({\n      ...options,\n      inputSchema: geminiImageModelInputSchema,\n      description: options?.description ?? \"Draw or edit image by Gemini image models\",\n    });\n    if (options) checkArguments(this.name, geminiImageModelOptionsSchema, options);\n  }\n\n  protected _client?: GoogleGenAI;\n\n  protected apiKeyEnvName = \"GEMINI_API_KEY\";\n\n  get client() {\n    if (this._client) return this._client;\n\n    const { apiKey } = 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 GoogleGenAI({ apiKey });\n\n    return this._client;\n  }\n\n  override get credential() {\n    return {\n      url: this.options?.baseURL || process.env.GEMINI_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: GeminiImageModelInput,\n    _options: ModelInvokeOptions,\n  ): Promise<ImageModelOutput> {\n    const model = input.modelOptions?.model || this.credential.model;\n    const responseFormat = input.responseFormat || \"base64\";\n    if (responseFormat === \"url\") {\n      throw new Error(\"Gemini image models currently only support base64 format\");\n    }\n\n    if (model.includes(\"imagen\")) {\n      return this.generateImageByImagenModel(input);\n    }\n\n    return this.generateImageByGeminiModel(input);\n  }\n\n  private async generateImageByImagenModel(\n    input: GeminiImageModelInput,\n  ): Promise<ImageModelOutput> {\n    const model = input.modelOptions?.model || this.credential.model;\n\n    const mergedInput = { ...this.modelOptions, ...input.modelOptions, ...input };\n\n    const inputKeys = [\n      \"seed\",\n      \"safetyFilterLevel\",\n      \"personGeneration\",\n      \"outputMimeType\",\n      \"outputGcsUri\",\n      \"outputCompressionQuality\",\n      \"negativePrompt\",\n      \"language\",\n      \"includeSafetyAttributes\",\n      \"includeRaiReason\",\n      \"imageSize\",\n      \"guidanceScale\",\n      \"aspectRatio\",\n      \"addWatermark\",\n    ];\n\n    const response = await this.client.models.generateImages({\n      model,\n      prompt: mergedInput.prompt,\n      config: { numberOfImages: mergedInput.n || 1, ...pick(mergedInput, inputKeys) },\n    });\n\n    return {\n      images:\n        response.generatedImages\n          ?.map<FileUnionContent | undefined>(({ image }) =>\n            image?.imageBytes\n              ? { type: \"file\", data: image.imageBytes, mimeType: image.mimeType }\n              : undefined,\n          )\n          .filter(isNonNullable) || [],\n      usage: {\n        inputTokens: 0,\n        outputTokens: 0,\n      },\n      model,\n    };\n  }\n\n  private async generateImageByGeminiModel(\n    input: GeminiImageModelInput,\n  ): Promise<ImageModelOutput> {\n    const model = input.modelOptions?.model || this.credential.model;\n\n    const mergedInput = { ...this.modelOptions, ...input.modelOptions, ...input };\n\n    const inputKeys = [\n      \"abortSignal\",\n      \"audioTimestamp\",\n      \"automaticFunctionCalling\",\n      \"cachedContent\",\n      \"frequencyPenalty\",\n      \"httpOptions\",\n      \"labels\",\n      \"logprobs\",\n      \"maxOutputTokens\",\n      \"mediaResolution\",\n      \"modelSelectionConfig\",\n      \"presencePenalty\",\n      \"responseJsonSchema\",\n      \"responseLogprobs\",\n      \"responseMimeType\",\n      \"responseSchema\",\n      \"routingConfig\",\n      \"safetySettings\",\n      \"seed\",\n      \"speechConfig\",\n      \"stopSequences\",\n      \"systemInstruction\",\n      \"temperature\",\n      \"thinkingConfig\",\n      \"toolConfig\",\n      \"tools\",\n      \"topK\",\n      \"topP\",\n      \"imageConfig\",\n    ];\n\n    const images = await Promise.all(\n      flat(input.image).map<Promise<PartUnion>>(async (image) => {\n        const { data, mimeType } = await this.transformFileType(\"file\", image);\n        return { inlineData: { data, mimeType } };\n      }),\n    );\n\n    const response = await this.client.models.generateContent({\n      model,\n      contents: [{ text: input.prompt }, ...images],\n      config: {\n        responseModalities: [Modality.TEXT, Modality.IMAGE],\n        candidateCount: input.n || 1,\n        ...pick(mergedInput, inputKeys),\n      },\n    });\n\n    const allImages = (response.candidates ?? [])\n      .flatMap((candidate) => candidate.content?.parts ?? [])\n      .map<FileUnionContent | null>((part) =>\n        part.inlineData?.data\n          ? {\n              type: \"file\",\n              data: part.inlineData.data,\n              filename: part.inlineData.displayName,\n              mimeType: part.inlineData.mimeType,\n            }\n          : null,\n      )\n      .filter(isNonNullable);\n\n    return {\n      images: allImages,\n      usage: {\n        inputTokens: response.usageMetadata?.promptTokenCount || 0,\n        outputTokens: response.usageMetadata?.candidatesTokenCount || 0,\n      },\n      model,\n    };\n  }\n}\n"],"mappings":";;;;;;AAmBA,MAAM,gBAAgB;AAiBtB,MAAM,8BAA8B,sBAAsB,OAAO,EAAE,CAAC;AAEpE,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,WAAW,KAAK;AAExB,MAAI,CAAC,OACH,OAAM,IAAI,MACR,GAAG,KAAK,KAAK,+EAA+E,KAAK,cAAc,yBAChH;AAEH,OAAK,YAAY,IAAI,YAAY,EAAE,QAAQ,CAAC;AAE5C,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,UAC2B;EAC3B,MAAM,QAAQ,MAAM,cAAc,SAAS,KAAK,WAAW;AAE3D,OADuB,MAAM,kBAAkB,cACxB,MACrB,OAAM,IAAI,MAAM,2DAA2D;AAG7E,MAAI,MAAM,SAAS,SAAS,CAC1B,QAAO,KAAK,2BAA2B,MAAM;AAG/C,SAAO,KAAK,2BAA2B,MAAM;;CAG/C,MAAc,2BACZ,OAC2B;EAC3B,MAAM,QAAQ,MAAM,cAAc,SAAS,KAAK,WAAW;EAE3D,MAAM,cAAc;GAAE,GAAG,KAAK;GAAc,GAAG,MAAM;GAAc,GAAG;GAAO;AAyB7E,SAAO;GACL,SAPe,MAAM,KAAK,OAAO,OAAO,eAAe;IACvD;IACA,QAAQ,YAAY;IACpB,QAAQ;KAAE,gBAAgB,YAAY,KAAK;KAAG,GAAG,KAAK,aApBtC;MAChB;MACA;MACA;MACA;MACA;MACA;MACA;MACA;MACA;MACA;MACA;MACA;MACA;MACA;MACD,CAK8E;KAAE;IAChF,CAAC,EAIW,iBACL,KAAmC,EAAE,YACrC,OAAO,aACH;IAAE,MAAM;IAAQ,MAAM,MAAM;IAAY,UAAU,MAAM;IAAU,GAClE,OACL,CACA,OAAO,cAAc,IAAI,EAAE;GAChC,OAAO;IACL,aAAa;IACb,cAAc;IACf;GACD;GACD;;CAGH,MAAc,2BACZ,OAC2B;EAC3B,MAAM,QAAQ,MAAM,cAAc,SAAS,KAAK,WAAW;EAE3D,MAAM,cAAc;GAAE,GAAG,KAAK;GAAc,GAAG,MAAM;GAAc,GAAG;GAAO;EAE7E,MAAM,YAAY;GAChB;GACA;GACA;GACA;GACA;GACA;GACA;GACA;GACA;GACA;GACA;GACA;GACA;GACA;GACA;GACA;GACA;GACA;GACA;GACA;GACA;GACA;GACA;GACA;GACA;GACA;GACA;GACA;GACA;GACD;EAED,MAAM,SAAS,MAAM,QAAQ,IAC3B,KAAK,MAAM,MAAM,CAAC,IAAwB,OAAO,UAAU;GACzD,MAAM,EAAE,MAAM,aAAa,MAAM,KAAK,kBAAkB,QAAQ,MAAM;AACtE,UAAO,EAAE,YAAY;IAAE;IAAM;IAAU,EAAE;IACzC,CACH;EAED,MAAM,WAAW,MAAM,KAAK,OAAO,OAAO,gBAAgB;GACxD;GACA,UAAU,CAAC,EAAE,MAAM,MAAM,QAAQ,EAAE,GAAG,OAAO;GAC7C,QAAQ;IACN,oBAAoB,CAAC,SAAS,MAAM,SAAS,MAAM;IACnD,gBAAgB,MAAM,KAAK;IAC3B,GAAG,KAAK,aAAa,UAAU;IAChC;GACF,CAAC;AAgBF,SAAO;GACL,SAfiB,SAAS,cAAc,EAAE,EACzC,SAAS,cAAc,UAAU,SAAS,SAAS,EAAE,CAAC,CACtD,KAA8B,SAC7B,KAAK,YAAY,OACb;IACE,MAAM;IACN,MAAM,KAAK,WAAW;IACtB,UAAU,KAAK,WAAW;IAC1B,UAAU,KAAK,WAAW;IAC3B,GACD,KACL,CACA,OAAO,cAAc;GAItB,OAAO;IACL,aAAa,SAAS,eAAe,oBAAoB;IACzD,cAAc,SAAS,eAAe,wBAAwB;IAC/D;GACD;GACD"}

package/dist/gemini-video-model.cjs

@@ -1,15 +1,15 @@
 const require_utils = require('./utils.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_utils_nodejs = require("@aigne/utils/nodejs");
+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_nodejs = require("@aigne/model-base/utils/nodejs");
+let _aigne_model_base_utils_type_utils = require("@aigne/model-base/utils/type-utils");
 let _google_genai = require("@google/genai");
 let zod = require("zod");
 
 //#region src/gemini-video-model.ts
 const DEFAULT_MODEL = "veo-3.1-generate-preview";
 const DEFAULT_SECONDS = 8;
-const geminiVideoModelInputSchema = _aigne_core.videoModelInputSchema.extend({
+const geminiVideoModelInputSchema = _aigne_model_base.videoModelInputSchema.extend({
 	negativePrompt: zod.z.string().optional(),
 	aspectRatio: zod.z.enum(["16:9", "9:16"]).optional(),
 	size: zod.z.enum(["720p", "1080p"]).optional(),
@@ -19,8 +19,8 @@ const geminiVideoModelInputSchema = _aigne_core.videoModelInputSchema.extend({
 		"8"
 	]).optional(),
 	personGeneration: zod.z.string().optional(),
-	lastFrame: _aigne_core.fileUnionContentSchema.optional(),
-	referenceImages: _aigne_core.fileUnionContentSchema.array().optional()
+	lastFrame: _aigne_model_base.fileUnionContentSchema.optional(),
+	referenceImages: _aigne_model_base.fileUnionContentSchema.array().optional()
 });
 const geminiVideoModelOptionsSchema = zod.z.object({
 	apiKey: zod.z.string().optional(),
@@ -30,7 +30,7 @@ const geminiVideoModelOptionsSchema = zod.z.object({
 	clientOptions: zod.z.object({}).optional(),
 	pollingInterval: zod.z.number().optional()
 });
-var GeminiVideoModel = class extends _aigne_core.VideoModel {
+var GeminiVideoModel = class extends _aigne_model_base.VideoModel {
 	constructor(options) {
 		super({
 			...options,
@@ -38,7 +38,7 @@ var GeminiVideoModel = class extends _aigne_core.VideoModel {
 			inputSchema: geminiVideoModelInputSchema
 		});
 		this.options = options;
-		if (options) (0, _aigne_core_utils_type_utils.checkArguments)(this.name, geminiVideoModelOptionsSchema, options);
+		if (options) (0, _aigne_model_base_utils_type_utils.checkArguments)(this.name, geminiVideoModelOptionsSchema, options);
 	}
 	/**
 	* @hidden
@@ -65,17 +65,17 @@ var GeminiVideoModel = class extends _aigne_core.VideoModel {
 		return this.options?.modelOptions;
 	}
 	async downloadToFile(dir, videoId, videoFile) {
-		_aigne_core_utils_logger.logger.debug("Downloading video content...");
-		const localPath = _aigne_utils_nodejs.nodejs.path.join(dir, `${videoId}.mp4`);
+		_aigne_model_base_utils_logger.logger.debug("Downloading video content...");
+		const localPath = _aigne_model_base_utils_nodejs.nodejs.path.join(dir, `${videoId}.mp4`);
 		await this.client.files.download({
 			file: videoFile,
 			downloadPath: localPath
 		});
-		_aigne_core_utils_logger.logger.debug(`Generated video saved to ${localPath}`);
+		_aigne_model_base_utils_logger.logger.debug(`Generated video saved to ${localPath}`);
 		await require_utils.waitFileSizeStable(localPath);
-		return (await _aigne_utils_nodejs.nodejs.fs.readFile(localPath)).toString("base64");
+		return (await _aigne_model_base_utils_nodejs.nodejs.fs.readFile(localPath)).toString("base64");
 	}
-	async process(input, options) {
+	async process(input, _options) {
 		const model = input.model ?? input.modelOptions?.model ?? this.credential.model;
 		const mergedInput = {
 			...this.modelOptions,
@@ -114,18 +114,18 @@ var GeminiVideoModel = class extends _aigne_core.VideoModel {
 			};
 		});
 		let operation = await this.client.models.generateVideos(params);
-		_aigne_core_utils_logger.logger.debug("Video generation started...");
+		_aigne_model_base_utils_logger.logger.debug("Video generation started...");
 		const pollingInterval = this.options?.pollingInterval ?? 1e4;
 		while (!operation.done) {
-			_aigne_core_utils_logger.logger.debug("Waiting for video generation to complete...");
+			_aigne_model_base_utils_logger.logger.debug("Waiting for video generation to complete...");
 			await new Promise((resolve) => setTimeout(resolve, pollingInterval));
 			operation = await this.client.operations.getVideosOperation({ operation });
 		}
 		if (!operation.response?.generatedVideos?.[0]?.video) throw new Error("Video generation failed: No video generated");
 		const videoFile = operation.response.generatedVideos[0].video;
 		if (!videoFile) throw new Error("Video generation failed: No video file returned");
-		const dir = _aigne_utils_nodejs.nodejs.path.join(_aigne_utils_nodejs.nodejs.os.tmpdir(), options?.context?.id || "");
-		await _aigne_utils_nodejs.nodejs.fs.mkdir(dir, { recursive: true });
+		const dir = _aigne_model_base_utils_nodejs.nodejs.path.join(_aigne_model_base_utils_nodejs.nodejs.os.tmpdir(), `gemini-video-${Date.now()}`);
+		await _aigne_model_base_utils_nodejs.nodejs.fs.mkdir(dir, { recursive: true });
 		const videoId = Date.now().toString();
 		return {
 			videos: [{

package/dist/gemini-video-model.d.cts

@@ -1,4 +1,4 @@
-import { AgentInvokeOptions, FileUnionContent, VideoModel, VideoModelInput, VideoModelOptions, VideoModelOutput } from "@aigne/core";
+import { FileUnionContent, ModelInvokeOptions, VideoModel, VideoModelInput, VideoModelOptions, VideoModelOutput } from "@aigne/model-base";
 import { GoogleGenAI } from "@google/genai";
 
 //#region src/gemini-video-model.d.ts
@@ -7,47 +7,47 @@ import { GoogleGenAI } from "@google/genai";
  */
 interface GeminiVideoModelInput extends VideoModelInput {
   /**
-       * Text describing content that should not appear in the video
-       */
+   * Text describing content that should not appear in the video
+   */
   negativePrompt?: string;
   /**
-       * Aspect ratio of the video
-       *
-       * Veo 3.1: "16:9" (default, 720p and 1080p), "9:16" (720p and 1080p)
-       * Veo 3: "16:9" (default, 720p and 1080p), "9:16" (720p and 1080p)
-       */
+   * Aspect ratio of the video
+   *
+   * Veo 3.1: "16:9" (default, 720p and 1080p), "9:16" (720p and 1080p)
+   * Veo 3: "16:9" (default, 720p and 1080p), "9:16" (720p and 1080p)
+   */
   aspectRatio?: "16:9" | "9:16";
   /**
-       * Resolution of the video
-       *
-       * Veo 3.1: "720p" (default), "1080p" (only supports 8 seconds duration)
-       * Veo 3: "720p" (default), "1080p" (16:9 only)
-       */
+   * Resolution of the video
+   *
+   * Veo 3.1: "720p" (default), "1080p" (only supports 8 seconds duration)
+   * Veo 3: "720p" (default), "1080p" (16:9 only)
+   */
   size?: "720p" | "1080p";
   /**
-       * Duration of the generated video in seconds
-       *
-       * Veo 3.1: "4", "6", "8"
-       * Veo 3: "4", "6", "8"
-       */
+   * Duration of the generated video in seconds
+   *
+   * Veo 3.1: "4", "6", "8"
+   * Veo 3: "4", "6", "8"
+   */
   seconds?: "4" | "6" | "8";
   /**
-       * Control person generation
-       *
-       * For text-to-video and image-to-video:
-       * - Veo 3.1: "allow_all" for image-to-video, frame interpolation and reference images; only "allow_adult" for text-to-video
-       * - Veo 3: "allow_all" for image-to-video; only "allow_adult" for text-to-video
-       * - Veo 2: "allow_all", "allow_adult", "dont_allow"
-       */
+   * Control person generation
+   *
+   * For text-to-video and image-to-video:
+   * - Veo 3.1: "allow_all" for image-to-video, frame interpolation and reference images; only "allow_adult" for text-to-video
+   * - Veo 3: "allow_all" for image-to-video; only "allow_adult" for text-to-video
+   * - Veo 2: "allow_all", "allow_adult", "dont_allow"
+   */
   personGeneration?: string;
   /**
-       * Last frame for video generation (frame interpolation)
-       */
+   * Last frame for video generation (frame interpolation)
+   */
   lastFrame?: FileUnionContent;
   /**
-       * Reference images for video generation
-       * Only supported in Veo 3.1 models
-       */
+   * Reference images for video generation
+   * Only supported in Veo 3.1 models
+   */
   referenceImages?: FileUnionContent[];
 }
 /**
@@ -59,44 +59,44 @@ interface GeminiVideoModelOutput extends VideoModelOutput {}
  */
 interface GeminiVideoModelOptions extends VideoModelOptions<GeminiVideoModelInput, GeminiVideoModelOutput> {
   /**
-       * API key for Gemini API
-       *
-       * If not provided, will look for GEMINI_API_KEY in environment variables
-       */
+   * API key for Gemini API
+   *
+   * If not provided, will look for GEMINI_API_KEY in environment variables
+   */
   apiKey?: string;
   /**
-       * Base URL for Gemini API
-       *
-       * Useful for proxies or alternate endpoints
-       */
+   * Base URL for Gemini API
+   *
+   * Useful for proxies or alternate endpoints
+   */
   baseURL?: string;
   /**
-       * Gemini model to use
-       *
-       * Defaults to 'veo-3.1-generate-preview'
-       */
+   * Gemini model to use
+   *
+   * Defaults to 'veo-3.1-generate-preview'
+   */
   model?: string;
   /**
-       * Additional model options to control behavior
-       */
+   * Additional model options to control behavior
+   */
   modelOptions?: Omit<Partial<GeminiVideoModelInput>, "model">;
   /**
-       * Client options for Gemini API
-       */
+   * Client options for Gemini API
+   */
   clientOptions?: Record<string, any>;
   /**
-       * Polling interval in milliseconds for checking video generation status
-       *
-       * Defaults to 10000ms (10 seconds)
-       */
+   * Polling interval in milliseconds for checking video generation status
+   *
+   * Defaults to 10000ms (10 seconds)
+   */
   pollingInterval?: number;
 }
 declare class GeminiVideoModel extends VideoModel<GeminiVideoModelInput, GeminiVideoModelOutput> {
   options?: GeminiVideoModelOptions | undefined;
   constructor(options?: GeminiVideoModelOptions | undefined);
   /**
-       * @hidden
-       */
+   * @hidden
+   */
   protected _client?: GoogleGenAI;
   protected apiKeyEnvName: string;
   get client(): GoogleGenAI;
@@ -110,7 +110,7 @@ declare class GeminiVideoModel extends VideoModel<GeminiVideoModelInput, GeminiV
     uri?: string;
     videoBytes?: any;
   }): Promise<string>;
-  process(input: GeminiVideoModelInput, options: AgentInvokeOptions): Promise<GeminiVideoModelOutput>;
+  process(input: GeminiVideoModelInput, _options: ModelInvokeOptions): Promise<GeminiVideoModelOutput>;
 }
 //#endregion
 export { GeminiVideoModel, GeminiVideoModelInput, GeminiVideoModelOptions, GeminiVideoModelOutput };

package/dist/gemini-video-model.d.cts.map

@@ -1 +1 @@
-{"version":3,"file":"gemini-video-model.d.cts","names":[],"sources":["../src/gemini-video-model.ts"],"mappings":";;;;;AAqBA;;UAAiB,qBAAA,SAA8B,eAAA;EAAA;;AAuD/C;EAvD+C,cAAA;EAAA;;AAuD/C;AAKA;;;EA5D+C,WAAA;EAAA;;AAuD/C;AAKA;;;EA5D+C,IAAA;EAAA;;AAuD/C;AAKA;;;EA5D+C,OAAA;EAAA;;AAuD/C;AAKA;;;;;EA5D+C,gBAAA;EAAA;;AAuD/C;EAvD+C,SAAA,GA2CjC,gBAAA;EAAA;;;;EAAA,eAAA,GAMM,gBAAA;AAAA;AAAA;;;AAAA,UAMH,sBAAA,SAA+B,gBAAA;AAAA;AAKhD;;AALgD,UAK/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,MAAA;EAAA;;;AA6BlB;;EA7BkB,eAAA;AAAA;AAAA,cA6BL,gBAAA,SAAyB,UAAA,CAAW,qBAAA,EAAuB,sBAAA;EAAA,OAAA,GAChC,uBAAA;EAAA,YAAA,OAAA,GAAA,uBAAA;EAAA;;;EAAA,UAAA,OAAA,GAalB,WAAA;EAAA,UAAA,aAAA;EAAA,IAAA,OAAA,GAIV,WAAA;EAAA,IAAA,WAAA;IAAA,GAAA;IAAA,MAAA;IAAA,KAAA;EAAA;EAAA,IAAA,aAAA,GAmBM,IAAA,CAAA,OAAA,CAAA,qBAAA;EAAA,eAAA,GAAA,UAAA,OAAA,UAAA,SAAA;IAAA,GAAA;IAAA,UAAA;EAAA,IAQb,OAAA;EAAA,QAAA,KAAA,EAaM,qBAAA,EAAA,OAAA,EACE,kBAAA,GACR,OAAA,CAAQ,sBAAA;AAAA"}
+{"version":3,"file":"gemini-video-model.d.cts","names":[],"sources":["../src/gemini-video-model.ts"],"mappings":";;;;;AAqBA;;UAAiB,qBAAA,SAA8B,eAAA;EA2CjC;;;EAvCZ,cAAA;EAJ4D;;;;;;EAY5D,WAAA;EA+BA;;;;;;EAvBA,IAAA;EAmCsC;;;;AAKxC;;EAhCE,OAAA;EAiC0B;;;;;;;;EAvB1B,gBAAA;EAuBQ;;;EAlBR,SAAA,GAAY,gBAAA;EA+BZ;;;;EAzBA,eAAA,GAAkB,gBAAA;AAAA;;;;UAMH,sBAAA,SAA+B,gBAAA;AAiEhD;;;AAAA,UA5DiB,uBAAA,SACP,iBAAA,CAAkB,qBAAA,EAAuB,sBAAA;EA2DqB;;;;;EArDtE,MAAA;EA0FgB;;;;;EAnFhB,OAAA;EA0GG;;;;;EAnGH,KAAA;EAuCsE;;;EAlCtE,YAAA,GAAe,IAAA,CAAK,OAAA,CAAQ,qBAAA;EAmCU;;;EA9BtC,aAAA,GAAgB,MAAA;EA6CN;;;;;EAtCV,eAAA;AAAA;AAAA,cAsBW,gBAAA,SAAyB,UAAA,CAAW,qBAAA,EAAuB,sBAAA;EAC1C,OAAA,GAAU,uBAAA;cAAV,OAAA,GAAU,uBAAA;EAoCtB;;;EAAA,UAvBN,OAAA,GAAU,WAAA;EAAA,UAEV,aAAA;EAAA,IAEN,MAAA,CAAA,GAAM,WAAA;EAAA,IAWG,UAAA,CAAA;;;;;MAQT,YAAA,CAAA,GAAY,IAAA,CAAA,OAAA,CAAA,qBAAA;EAIV,cAAA,CACJ,GAAA,UACA,OAAA,UACA,SAAA;IAAa,GAAA;IAAc,UAAA;EAAA,IAC1B,OAAA;EAYY,OAAA,CACb,KAAA,EAAO,qBAAA,EACP,QAAA,EAAU,kBAAA,GACT,OAAA,CAAQ,sBAAA;AAAA"}