@aigne/gemini 1.74.0-beta.8 → 1.74.0-beta.9

package/dist/gemini-embedding-model.cjs

@@ -0,0 +1,65 @@
+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-embedding-model.ts
+const DEFAULT_MODEL = "text-embedding-004";
+const geminiEmbeddingModelInputSchema = _aigne_model_base.embeddingModelInputSchema.extend({});
+const geminiEmbeddingModelOptionsSchema = 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 GeminiEmbeddingModel = class extends _aigne_model_base.EmbeddingModel {
+	constructor(options) {
+		super({
+			...options,
+			inputSchema: geminiEmbeddingModelInputSchema,
+			description: options?.description ?? "Generate embeddings by Gemini embedding models"
+		});
+		this.options = options;
+		if (options) (0, _aigne_model_base_utils_type_utils.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 _google_genai.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
+exports.GeminiEmbeddingModel = GeminiEmbeddingModel;

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

@@ -0,0 +1,28 @@
+import { EmbeddingModel, EmbeddingModelInput, EmbeddingModelOptions, EmbeddingModelOutput, ModelInvokeOptions } from "@aigne/model-base";
+import { GoogleGenAI } from "@google/genai";
+
+//#region src/gemini-embedding-model.d.ts
+interface GeminiEmbeddingModelInput extends EmbeddingModelInput {}
+interface GeminiEmbeddingModelOutput extends EmbeddingModelOutput {}
+interface GeminiEmbeddingModelOptions extends EmbeddingModelOptions<GeminiEmbeddingModelInput, GeminiEmbeddingModelOutput> {
+  apiKey?: string;
+  baseURL?: string;
+  model?: string;
+  clientOptions?: Record<string, any>;
+}
+declare class GeminiEmbeddingModel extends EmbeddingModel<GeminiEmbeddingModelInput, GeminiEmbeddingModelOutput> {
+  options?: GeminiEmbeddingModelOptions | undefined;
+  constructor(options?: GeminiEmbeddingModelOptions | undefined);
+  protected _client?: GoogleGenAI;
+  protected apiKeyEnvName: string;
+  get client(): GoogleGenAI;
+  get credential(): {
+    url: string | undefined;
+    apiKey: string | undefined;
+    model: string;
+  };
+  process(input: GeminiEmbeddingModelInput, _options: ModelInvokeOptions): Promise<GeminiEmbeddingModelOutput>;
+}
+//#endregion
+export { GeminiEmbeddingModel, GeminiEmbeddingModelInput, GeminiEmbeddingModelOptions, GeminiEmbeddingModelOutput };
+//# sourceMappingURL=gemini-embedding-model.d.cts.map

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

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

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

@@ -0,0 +1,28 @@
+import { EmbeddingModel, EmbeddingModelInput, EmbeddingModelOptions, EmbeddingModelOutput, ModelInvokeOptions } from "@aigne/model-base";
+import { GoogleGenAI } from "@google/genai";
+
+//#region src/gemini-embedding-model.d.ts
+interface GeminiEmbeddingModelInput extends EmbeddingModelInput {}
+interface GeminiEmbeddingModelOutput extends EmbeddingModelOutput {}
+interface GeminiEmbeddingModelOptions extends EmbeddingModelOptions<GeminiEmbeddingModelInput, GeminiEmbeddingModelOutput> {
+  apiKey?: string;
+  baseURL?: string;
+  model?: string;
+  clientOptions?: Record<string, any>;
+}
+declare class GeminiEmbeddingModel extends EmbeddingModel<GeminiEmbeddingModelInput, GeminiEmbeddingModelOutput> {
+  options?: GeminiEmbeddingModelOptions | undefined;
+  constructor(options?: GeminiEmbeddingModelOptions | undefined);
+  protected _client?: GoogleGenAI;
+  protected apiKeyEnvName: string;
+  get client(): GoogleGenAI;
+  get credential(): {
+    url: string | undefined;
+    apiKey: string | undefined;
+    model: string;
+  };
+  process(input: GeminiEmbeddingModelInput, _options: ModelInvokeOptions): Promise<GeminiEmbeddingModelOutput>;
+}
+//#endregion
+export { GeminiEmbeddingModel, GeminiEmbeddingModelInput, GeminiEmbeddingModelOptions, GeminiEmbeddingModelOutput };
+//# sourceMappingURL=gemini-embedding-model.d.mts.map

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

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

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/index.cjs

@@ -1,7 +1,9 @@
 const require_gemini_chat_model = require('./gemini-chat-model.cjs');
+const require_gemini_embedding_model = require('./gemini-embedding-model.cjs');
 const require_gemini_image_model = require('./gemini-image-model.cjs');
 const require_gemini_video_model = require('./gemini-video-model.cjs');
 
 exports.GeminiChatModel = require_gemini_chat_model.GeminiChatModel;
+exports.GeminiEmbeddingModel = require_gemini_embedding_model.GeminiEmbeddingModel;
 exports.GeminiImageModel = require_gemini_image_model.GeminiImageModel;
 exports.GeminiVideoModel = require_gemini_video_model.GeminiVideoModel;

package/dist/index.d.cts

@@ -1,4 +1,5 @@
 import { GeminiChatModel, GeminiChatModelOptions } from "./gemini-chat-model.cjs";
+import { GeminiEmbeddingModel, GeminiEmbeddingModelInput, GeminiEmbeddingModelOptions, GeminiEmbeddingModelOutput } from "./gemini-embedding-model.cjs";
 import { GeminiImageModel, GeminiImageModelInput, GeminiImageModelOptions, GeminiImageModelOutput } from "./gemini-image-model.cjs";
 import { GeminiVideoModel, GeminiVideoModelInput, GeminiVideoModelOptions, GeminiVideoModelOutput } from "./gemini-video-model.cjs";
-export { GeminiChatModel, GeminiChatModelOptions, GeminiImageModel, GeminiImageModelInput, GeminiImageModelOptions, GeminiImageModelOutput, GeminiVideoModel, GeminiVideoModelInput, GeminiVideoModelOptions, GeminiVideoModelOutput };
+export { GeminiChatModel, GeminiChatModelOptions, GeminiEmbeddingModel, GeminiEmbeddingModelInput, GeminiEmbeddingModelOptions, GeminiEmbeddingModelOutput, GeminiImageModel, GeminiImageModelInput, GeminiImageModelOptions, GeminiImageModelOutput, GeminiVideoModel, GeminiVideoModelInput, GeminiVideoModelOptions, GeminiVideoModelOutput };

package/dist/index.d.mts

@@ -1,4 +1,5 @@
 import { GeminiChatModel, GeminiChatModelOptions } from "./gemini-chat-model.mjs";
+import { GeminiEmbeddingModel, GeminiEmbeddingModelInput, GeminiEmbeddingModelOptions, GeminiEmbeddingModelOutput } from "./gemini-embedding-model.mjs";
 import { GeminiImageModel, GeminiImageModelInput, GeminiImageModelOptions, GeminiImageModelOutput } from "./gemini-image-model.mjs";
 import { GeminiVideoModel, GeminiVideoModelInput, GeminiVideoModelOptions, GeminiVideoModelOutput } from "./gemini-video-model.mjs";
-export { GeminiChatModel, GeminiChatModelOptions, GeminiImageModel, GeminiImageModelInput, GeminiImageModelOptions, GeminiImageModelOutput, GeminiVideoModel, GeminiVideoModelInput, GeminiVideoModelOptions, GeminiVideoModelOutput };
+export { GeminiChatModel, GeminiChatModelOptions, GeminiEmbeddingModel, GeminiEmbeddingModelInput, GeminiEmbeddingModelOptions, GeminiEmbeddingModelOutput, GeminiImageModel, GeminiImageModelInput, GeminiImageModelOptions, GeminiImageModelOutput, GeminiVideoModel, GeminiVideoModelInput, GeminiVideoModelOptions, GeminiVideoModelOutput };

package/dist/index.mjs

@@ -1,5 +1,6 @@
 import { GeminiChatModel } from "./gemini-chat-model.mjs";
+import { GeminiEmbeddingModel } from "./gemini-embedding-model.mjs";
 import { GeminiImageModel } from "./gemini-image-model.mjs";
 import { GeminiVideoModel } from "./gemini-video-model.mjs";
 
-export { GeminiChatModel, GeminiImageModel, GeminiVideoModel };
+export { GeminiChatModel, GeminiEmbeddingModel, GeminiImageModel, GeminiVideoModel };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aigne/gemini",
-  "version": "1.74.0-beta.8",
+  "version": "1.74.0-beta.9",
   "description": "AIGNE Gemini SDK for integrating with Google's Gemini AI models",
   "license": "Elastic-2.0",
   "publishConfig": {
@@ -38,7 +38,7 @@
     "yaml": "^2.8.1",
     "zod": "^3.25.67",
     "zod-to-json-schema": "^3.24.6",
-    "@aigne/model-base": "^1.74.0-beta.8"
+    "@aigne/model-base": "^1.74.0-beta.9"
   },
   "devDependencies": {
     "@types/bun": "^1.3.6",