@ainetwork/adk-provider-model-gemini 0.1.1

package/dist/index.cjs

@@ -0,0 +1,129 @@
+"use strict";
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+
+// index.ts
+var index_exports = {};
+__export(index_exports, {
+  GeminiModel: () => GeminiModel
+});
+module.exports = __toCommonJS(index_exports);
+var import_modules = require("@ainetwork/adk/modules");
+var import_memory = require("@ainetwork/adk/types/memory");
+var import_tool = require("@ainetwork/adk/types/tool");
+var import_genai = require("@google/genai");
+var GeminiModel = class extends import_modules.BaseModel {
+  client;
+  modelName;
+  constructor(apiKey, modelName) {
+    super();
+    this.client = new import_genai.GoogleGenAI({ apiKey });
+    this.modelName = modelName;
+  }
+  getMessageRole(role) {
+    switch (role) {
+      case import_memory.ChatRole.USER:
+        return "user";
+      case import_memory.ChatRole.MODEL:
+      case import_memory.ChatRole.SYSTEM:
+        return "model";
+      default:
+        return "model";
+    }
+  }
+  generateMessages(params) {
+    const { query, sessionHistory, systemPrompt } = params;
+    const messages = !systemPrompt ? [] : [{ role: "model", parts: [{ text: systemPrompt.trim() }] }];
+    const sessionContent = !sessionHistory ? [] : Object.keys(sessionHistory.chats).map((chatId) => {
+      const chat = sessionHistory.chats[chatId];
+      return {
+        role: this.getMessageRole(chat.role),
+        parts: [{ text: chat.content.parts[0] }]
+      };
+    });
+    const userContent = { role: "user", parts: [{ text: query }] };
+    return messages.concat(sessionContent).concat(userContent);
+  }
+  appendMessages(messages, message) {
+    messages.push({
+      role: "user",
+      parts: [{ text: message }]
+    });
+  }
+  async fetch(messages) {
+    const response = await this.client.models.generateContent({
+      model: this.modelName,
+      contents: messages
+    });
+    return { content: response.text };
+  }
+  async fetchWithContextMessage(messages, functions) {
+    if (functions.length > 0) {
+      const response = await this.client.models.generateContent({
+        model: this.modelName,
+        contents: messages,
+        config: {
+          tools: [{ functionDeclarations: functions }]
+        }
+      });
+      const { text, functionCalls } = response;
+      const hasName = (value) => {
+        return value.name !== void 0;
+      };
+      const toolCalls = functionCalls?.filter(hasName).map((value) => {
+        return {
+          name: value.name,
+          arguments: value.args
+        };
+      });
+      return {
+        content: text,
+        toolCalls
+      };
+    }
+    return await this.fetch(messages);
+  }
+  convertToolsToFunctions(tools) {
+    const functions = [];
+    for (const tool of tools) {
+      if (!tool.enabled) {
+        continue;
+      }
+      if (tool.protocol === import_tool.TOOL_PROTOCOL_TYPE.MCP) {
+        const { mcpTool, id } = tool;
+        functions.push({
+          name: id,
+          description: mcpTool.description,
+          parametersJsonSchema: mcpTool.inputSchema
+        });
+      } else {
+        const { id, card } = tool;
+        functions.push({
+          name: id,
+          description: card.description
+        });
+      }
+    }
+    return functions;
+  }
+};
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  GeminiModel
+});
+//# sourceMappingURL=index.cjs.map

package/dist/index.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../index.ts"],"sourcesContent":["import { BaseModel } from \"@ainetwork/adk/modules\";\nimport { ChatRole, type SessionObject } from \"@ainetwork/adk/types/memory\";\nimport type {\n\tFetchResponse,\n\tIA2ATool,\n\tIAgentTool,\n\tIMCPTool,\n\tToolCall,\n} from \"@ainetwork/adk/types/tool\";\nimport { TOOL_PROTOCOL_TYPE } from \"@ainetwork/adk/types/tool\";\nimport {\n\ttype Content,\n\ttype FunctionCall,\n\ttype FunctionDeclaration,\n\tGoogleGenAI,\n} from \"@google/genai\";\n\nexport class GeminiModel extends BaseModel<Content, FunctionDeclaration> {\n\tprivate client: GoogleGenAI;\n\tprivate modelName: string;\n\n\tconstructor(apiKey: string, modelName: string) {\n\t\tsuper();\n\t\tthis.client = new GoogleGenAI({ apiKey });\n\t\tthis.modelName = modelName;\n\t}\n\n\tprivate getMessageRole(role: ChatRole) {\n\t\tswitch (role) {\n\t\t\tcase ChatRole.USER:\n\t\t\t\treturn \"user\";\n\t\t\tcase ChatRole.MODEL:\n\t\t\tcase ChatRole.SYSTEM:\n\t\t\t\treturn \"model\";\n\t\t\tdefault:\n\t\t\t\treturn \"model\"; /*FIXME*/\n\t\t}\n\t}\n\n\tgenerateMessages(params: {\n\t\tquery: string;\n\t\tsessionHistory?: SessionObject;\n\t\tsystemPrompt?: string;\n\t}): Content[] {\n\t\tconst { query, sessionHistory, systemPrompt } = params;\n\t\tconst messages: Content[] = !systemPrompt\n\t\t\t? []\n\t\t\t: [{ role: \"model\", parts: [{ text: systemPrompt.trim() }] }];\n\t\tconst sessionContent: Content[] = !sessionHistory\n\t\t\t? []\n\t\t\t: Object.keys(sessionHistory.chats).map((chatId: string) => {\n\t\t\t\t\tconst chat = sessionHistory.chats[chatId];\n\t\t\t\t\t// TODO: check message.content.type\n\t\t\t\t\treturn {\n\t\t\t\t\t\trole: this.getMessageRole(chat.role),\n\t\t\t\t\t\tparts: [{ text: chat.content.parts[0] }],\n\t\t\t\t\t};\n\t\t\t\t});\n\t\tconst userContent: Content = { role: \"user\", parts: [{ text: query }] };\n\t\treturn messages.concat(sessionContent).concat(userContent);\n\t}\n\n\tappendMessages(messages: Content[], message: string): void {\n\t\tmessages.push({\n\t\t\trole: \"user\",\n\t\t\tparts: [{ text: message }],\n\t\t});\n\t}\n\n\tasync fetch(messages: Content[]): Promise<FetchResponse> {\n\t\tconst response = await this.client.models.generateContent({\n\t\t\tmodel: this.modelName,\n\t\t\tcontents: messages,\n\t\t});\n\n\t\treturn { content: response.text };\n\t}\n\n\tasync fetchWithContextMessage(\n\t\tmessages: Content[],\n\t\tfunctions: FunctionDeclaration[],\n\t): Promise<FetchResponse> {\n\t\tif (functions.length > 0) {\n\t\t\tconst response = await this.client.models.generateContent({\n\t\t\t\tmodel: this.modelName,\n\t\t\t\tcontents: messages,\n\t\t\t\tconfig: {\n\t\t\t\t\ttools: [{ functionDeclarations: functions }],\n\t\t\t\t},\n\t\t\t});\n\n\t\t\tconst { text, functionCalls } = response;\n\t\t\tconst hasName = (\n\t\t\t\tvalue: FunctionCall,\n\t\t\t): value is FunctionCall & { name: string } => {\n\t\t\t\treturn value.name !== undefined;\n\t\t\t};\n\t\t\tconst toolCalls: ToolCall[] | undefined = functionCalls\n\t\t\t\t?.filter(hasName)\n\t\t\t\t.map((value) => {\n\t\t\t\t\treturn {\n\t\t\t\t\t\tname: value.name,\n\t\t\t\t\t\targuments: value.args,\n\t\t\t\t\t};\n\t\t\t\t});\n\n\t\t\treturn {\n\t\t\t\tcontent: text,\n\t\t\t\ttoolCalls,\n\t\t\t};\n\t\t}\n\t\treturn await this.fetch(messages);\n\t}\n\n\tconvertToolsToFunctions(tools: IAgentTool[]): FunctionDeclaration[] {\n\t\tconst functions: FunctionDeclaration[] = [];\n\t\tfor (const tool of tools) {\n\t\t\tif (!tool.enabled) {\n\t\t\t\tcontinue;\n\t\t\t}\n\t\t\tif (tool.protocol === TOOL_PROTOCOL_TYPE.MCP) {\n\t\t\t\tconst { mcpTool, id } = tool as IMCPTool;\n\t\t\t\tfunctions.push({\n\t\t\t\t\tname: id,\n\t\t\t\t\tdescription: mcpTool.description,\n\t\t\t\t\tparametersJsonSchema: mcpTool.inputSchema,\n\t\t\t\t});\n\t\t\t} else {\n\t\t\t\t// PROTOCOL_TYPE.A2A\n\t\t\t\tconst { id, card } = tool as IA2ATool;\n\t\t\t\tfunctions.push({\n\t\t\t\t\tname: id,\n\t\t\t\t\tdescription: card.description,\n\t\t\t\t});\n\t\t\t}\n\t\t}\n\t\treturn functions;\n\t}\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,qBAA0B;AAC1B,oBAA6C;AAQ7C,kBAAmC;AACnC,mBAKO;AAEA,IAAM,cAAN,cAA0B,yBAAwC;AAAA,EAChE;AAAA,EACA;AAAA,EAER,YAAY,QAAgB,WAAmB;AAC9C,UAAM;AACN,SAAK,SAAS,IAAI,yBAAY,EAAE,OAAO,CAAC;AACxC,SAAK,YAAY;AAAA,EAClB;AAAA,EAEQ,eAAe,MAAgB;AACtC,YAAQ,MAAM;AAAA,MACb,KAAK,uBAAS;AACb,eAAO;AAAA,MACR,KAAK,uBAAS;AAAA,MACd,KAAK,uBAAS;AACb,eAAO;AAAA,MACR;AACC,eAAO;AAAA,IACT;AAAA,EACD;AAAA,EAEA,iBAAiB,QAIH;AACb,UAAM,EAAE,OAAO,gBAAgB,aAAa,IAAI;AAChD,UAAM,WAAsB,CAAC,eAC1B,CAAC,IACD,CAAC,EAAE,MAAM,SAAS,OAAO,CAAC,EAAE,MAAM,aAAa,KAAK,EAAE,CAAC,EAAE,CAAC;AAC7D,UAAM,iBAA4B,CAAC,iBAChC,CAAC,IACD,OAAO,KAAK,eAAe,KAAK,EAAE,IAAI,CAAC,WAAmB;AAC1D,YAAM,OAAO,eAAe,MAAM,MAAM;AAExC,aAAO;AAAA,QACN,MAAM,KAAK,eAAe,KAAK,IAAI;AAAA,QACnC,OAAO,CAAC,EAAE,MAAM,KAAK,QAAQ,MAAM,CAAC,EAAE,CAAC;AAAA,MACxC;AAAA,IACD,CAAC;AACH,UAAM,cAAuB,EAAE,MAAM,QAAQ,OAAO,CAAC,EAAE,MAAM,MAAM,CAAC,EAAE;AACtE,WAAO,SAAS,OAAO,cAAc,EAAE,OAAO,WAAW;AAAA,EAC1D;AAAA,EAEA,eAAe,UAAqB,SAAuB;AAC1D,aAAS,KAAK;AAAA,MACb,MAAM;AAAA,MACN,OAAO,CAAC,EAAE,MAAM,QAAQ,CAAC;AAAA,IAC1B,CAAC;AAAA,EACF;AAAA,EAEA,MAAM,MAAM,UAA6C;AACxD,UAAM,WAAW,MAAM,KAAK,OAAO,OAAO,gBAAgB;AAAA,MACzD,OAAO,KAAK;AAAA,MACZ,UAAU;AAAA,IACX,CAAC;AAED,WAAO,EAAE,SAAS,SAAS,KAAK;AAAA,EACjC;AAAA,EAEA,MAAM,wBACL,UACA,WACyB;AACzB,QAAI,UAAU,SAAS,GAAG;AACzB,YAAM,WAAW,MAAM,KAAK,OAAO,OAAO,gBAAgB;AAAA,QACzD,OAAO,KAAK;AAAA,QACZ,UAAU;AAAA,QACV,QAAQ;AAAA,UACP,OAAO,CAAC,EAAE,sBAAsB,UAAU,CAAC;AAAA,QAC5C;AAAA,MACD,CAAC;AAED,YAAM,EAAE,MAAM,cAAc,IAAI;AAChC,YAAM,UAAU,CACf,UAC8C;AAC9C,eAAO,MAAM,SAAS;AAAA,MACvB;AACA,YAAM,YAAoC,eACvC,OAAO,OAAO,EACf,IAAI,CAAC,UAAU;AACf,eAAO;AAAA,UACN,MAAM,MAAM;AAAA,UACZ,WAAW,MAAM;AAAA,QAClB;AAAA,MACD,CAAC;AAEF,aAAO;AAAA,QACN,SAAS;AAAA,QACT;AAAA,MACD;AAAA,IACD;AACA,WAAO,MAAM,KAAK,MAAM,QAAQ;AAAA,EACjC;AAAA,EAEA,wBAAwB,OAA4C;AACnE,UAAM,YAAmC,CAAC;AAC1C,eAAW,QAAQ,OAAO;AACzB,UAAI,CAAC,KAAK,SAAS;AAClB;AAAA,MACD;AACA,UAAI,KAAK,aAAa,+BAAmB,KAAK;AAC7C,cAAM,EAAE,SAAS,GAAG,IAAI;AACxB,kBAAU,KAAK;AAAA,UACd,MAAM;AAAA,UACN,aAAa,QAAQ;AAAA,UACrB,sBAAsB,QAAQ;AAAA,QAC/B,CAAC;AAAA,MACF,OAAO;AAEN,cAAM,EAAE,IAAI,KAAK,IAAI;AACrB,kBAAU,KAAK;AAAA,UACd,MAAM;AAAA,UACN,aAAa,KAAK;AAAA,QACnB,CAAC;AAAA,MACF;AAAA,IACD;AACA,WAAO;AAAA,EACR;AACD;","names":[]}

package/dist/index.d.cts

@@ -0,0 +1,22 @@
+import { BaseModel } from '@ainetwork/adk/modules';
+import { SessionObject } from '@ainetwork/adk/types/memory';
+import { FetchResponse, IAgentTool } from '@ainetwork/adk/types/tool';
+import { Content, FunctionDeclaration } from '@google/genai';
+
+declare class GeminiModel extends BaseModel<Content, FunctionDeclaration> {
+    private client;
+    private modelName;
+    constructor(apiKey: string, modelName: string);
+    private getMessageRole;
+    generateMessages(params: {
+        query: string;
+        sessionHistory?: SessionObject;
+        systemPrompt?: string;
+    }): Content[];
+    appendMessages(messages: Content[], message: string): void;
+    fetch(messages: Content[]): Promise<FetchResponse>;
+    fetchWithContextMessage(messages: Content[], functions: FunctionDeclaration[]): Promise<FetchResponse>;
+    convertToolsToFunctions(tools: IAgentTool[]): FunctionDeclaration[];
+}
+
+export { GeminiModel };

package/dist/index.d.ts

@@ -0,0 +1,22 @@
+import { BaseModel } from '@ainetwork/adk/modules';
+import { SessionObject } from '@ainetwork/adk/types/memory';
+import { FetchResponse, IAgentTool } from '@ainetwork/adk/types/tool';
+import { Content, FunctionDeclaration } from '@google/genai';
+
+declare class GeminiModel extends BaseModel<Content, FunctionDeclaration> {
+    private client;
+    private modelName;
+    constructor(apiKey: string, modelName: string);
+    private getMessageRole;
+    generateMessages(params: {
+        query: string;
+        sessionHistory?: SessionObject;
+        systemPrompt?: string;
+    }): Content[];
+    appendMessages(messages: Content[], message: string): void;
+    fetch(messages: Content[]): Promise<FetchResponse>;
+    fetchWithContextMessage(messages: Content[], functions: FunctionDeclaration[]): Promise<FetchResponse>;
+    convertToolsToFunctions(tools: IAgentTool[]): FunctionDeclaration[];
+}
+
+export { GeminiModel };

package/dist/index.js

@@ -0,0 +1,106 @@
+// index.ts
+import { BaseModel } from "@ainetwork/adk/modules";
+import { ChatRole } from "@ainetwork/adk/types/memory";
+import { TOOL_PROTOCOL_TYPE } from "@ainetwork/adk/types/tool";
+import {
+  GoogleGenAI
+} from "@google/genai";
+var GeminiModel = class extends BaseModel {
+  client;
+  modelName;
+  constructor(apiKey, modelName) {
+    super();
+    this.client = new GoogleGenAI({ apiKey });
+    this.modelName = modelName;
+  }
+  getMessageRole(role) {
+    switch (role) {
+      case ChatRole.USER:
+        return "user";
+      case ChatRole.MODEL:
+      case ChatRole.SYSTEM:
+        return "model";
+      default:
+        return "model";
+    }
+  }
+  generateMessages(params) {
+    const { query, sessionHistory, systemPrompt } = params;
+    const messages = !systemPrompt ? [] : [{ role: "model", parts: [{ text: systemPrompt.trim() }] }];
+    const sessionContent = !sessionHistory ? [] : Object.keys(sessionHistory.chats).map((chatId) => {
+      const chat = sessionHistory.chats[chatId];
+      return {
+        role: this.getMessageRole(chat.role),
+        parts: [{ text: chat.content.parts[0] }]
+      };
+    });
+    const userContent = { role: "user", parts: [{ text: query }] };
+    return messages.concat(sessionContent).concat(userContent);
+  }
+  appendMessages(messages, message) {
+    messages.push({
+      role: "user",
+      parts: [{ text: message }]
+    });
+  }
+  async fetch(messages) {
+    const response = await this.client.models.generateContent({
+      model: this.modelName,
+      contents: messages
+    });
+    return { content: response.text };
+  }
+  async fetchWithContextMessage(messages, functions) {
+    if (functions.length > 0) {
+      const response = await this.client.models.generateContent({
+        model: this.modelName,
+        contents: messages,
+        config: {
+          tools: [{ functionDeclarations: functions }]
+        }
+      });
+      const { text, functionCalls } = response;
+      const hasName = (value) => {
+        return value.name !== void 0;
+      };
+      const toolCalls = functionCalls?.filter(hasName).map((value) => {
+        return {
+          name: value.name,
+          arguments: value.args
+        };
+      });
+      return {
+        content: text,
+        toolCalls
+      };
+    }
+    return await this.fetch(messages);
+  }
+  convertToolsToFunctions(tools) {
+    const functions = [];
+    for (const tool of tools) {
+      if (!tool.enabled) {
+        continue;
+      }
+      if (tool.protocol === TOOL_PROTOCOL_TYPE.MCP) {
+        const { mcpTool, id } = tool;
+        functions.push({
+          name: id,
+          description: mcpTool.description,
+          parametersJsonSchema: mcpTool.inputSchema
+        });
+      } else {
+        const { id, card } = tool;
+        functions.push({
+          name: id,
+          description: card.description
+        });
+      }
+    }
+    return functions;
+  }
+};
+export {
+  GeminiModel
+};
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../index.ts"],"sourcesContent":["import { BaseModel } from \"@ainetwork/adk/modules\";\nimport { ChatRole, type SessionObject } from \"@ainetwork/adk/types/memory\";\nimport type {\n\tFetchResponse,\n\tIA2ATool,\n\tIAgentTool,\n\tIMCPTool,\n\tToolCall,\n} from \"@ainetwork/adk/types/tool\";\nimport { TOOL_PROTOCOL_TYPE } from \"@ainetwork/adk/types/tool\";\nimport {\n\ttype Content,\n\ttype FunctionCall,\n\ttype FunctionDeclaration,\n\tGoogleGenAI,\n} from \"@google/genai\";\n\nexport class GeminiModel extends BaseModel<Content, FunctionDeclaration> {\n\tprivate client: GoogleGenAI;\n\tprivate modelName: string;\n\n\tconstructor(apiKey: string, modelName: string) {\n\t\tsuper();\n\t\tthis.client = new GoogleGenAI({ apiKey });\n\t\tthis.modelName = modelName;\n\t}\n\n\tprivate getMessageRole(role: ChatRole) {\n\t\tswitch (role) {\n\t\t\tcase ChatRole.USER:\n\t\t\t\treturn \"user\";\n\t\t\tcase ChatRole.MODEL:\n\t\t\tcase ChatRole.SYSTEM:\n\t\t\t\treturn \"model\";\n\t\t\tdefault:\n\t\t\t\treturn \"model\"; /*FIXME*/\n\t\t}\n\t}\n\n\tgenerateMessages(params: {\n\t\tquery: string;\n\t\tsessionHistory?: SessionObject;\n\t\tsystemPrompt?: string;\n\t}): Content[] {\n\t\tconst { query, sessionHistory, systemPrompt } = params;\n\t\tconst messages: Content[] = !systemPrompt\n\t\t\t? []\n\t\t\t: [{ role: \"model\", parts: [{ text: systemPrompt.trim() }] }];\n\t\tconst sessionContent: Content[] = !sessionHistory\n\t\t\t? []\n\t\t\t: Object.keys(sessionHistory.chats).map((chatId: string) => {\n\t\t\t\t\tconst chat = sessionHistory.chats[chatId];\n\t\t\t\t\t// TODO: check message.content.type\n\t\t\t\t\treturn {\n\t\t\t\t\t\trole: this.getMessageRole(chat.role),\n\t\t\t\t\t\tparts: [{ text: chat.content.parts[0] }],\n\t\t\t\t\t};\n\t\t\t\t});\n\t\tconst userContent: Content = { role: \"user\", parts: [{ text: query }] };\n\t\treturn messages.concat(sessionContent).concat(userContent);\n\t}\n\n\tappendMessages(messages: Content[], message: string): void {\n\t\tmessages.push({\n\t\t\trole: \"user\",\n\t\t\tparts: [{ text: message }],\n\t\t});\n\t}\n\n\tasync fetch(messages: Content[]): Promise<FetchResponse> {\n\t\tconst response = await this.client.models.generateContent({\n\t\t\tmodel: this.modelName,\n\t\t\tcontents: messages,\n\t\t});\n\n\t\treturn { content: response.text };\n\t}\n\n\tasync fetchWithContextMessage(\n\t\tmessages: Content[],\n\t\tfunctions: FunctionDeclaration[],\n\t): Promise<FetchResponse> {\n\t\tif (functions.length > 0) {\n\t\t\tconst response = await this.client.models.generateContent({\n\t\t\t\tmodel: this.modelName,\n\t\t\t\tcontents: messages,\n\t\t\t\tconfig: {\n\t\t\t\t\ttools: [{ functionDeclarations: functions }],\n\t\t\t\t},\n\t\t\t});\n\n\t\t\tconst { text, functionCalls } = response;\n\t\t\tconst hasName = (\n\t\t\t\tvalue: FunctionCall,\n\t\t\t): value is FunctionCall & { name: string } => {\n\t\t\t\treturn value.name !== undefined;\n\t\t\t};\n\t\t\tconst toolCalls: ToolCall[] | undefined = functionCalls\n\t\t\t\t?.filter(hasName)\n\t\t\t\t.map((value) => {\n\t\t\t\t\treturn {\n\t\t\t\t\t\tname: value.name,\n\t\t\t\t\t\targuments: value.args,\n\t\t\t\t\t};\n\t\t\t\t});\n\n\t\t\treturn {\n\t\t\t\tcontent: text,\n\t\t\t\ttoolCalls,\n\t\t\t};\n\t\t}\n\t\treturn await this.fetch(messages);\n\t}\n\n\tconvertToolsToFunctions(tools: IAgentTool[]): FunctionDeclaration[] {\n\t\tconst functions: FunctionDeclaration[] = [];\n\t\tfor (const tool of tools) {\n\t\t\tif (!tool.enabled) {\n\t\t\t\tcontinue;\n\t\t\t}\n\t\t\tif (tool.protocol === TOOL_PROTOCOL_TYPE.MCP) {\n\t\t\t\tconst { mcpTool, id } = tool as IMCPTool;\n\t\t\t\tfunctions.push({\n\t\t\t\t\tname: id,\n\t\t\t\t\tdescription: mcpTool.description,\n\t\t\t\t\tparametersJsonSchema: mcpTool.inputSchema,\n\t\t\t\t});\n\t\t\t} else {\n\t\t\t\t// PROTOCOL_TYPE.A2A\n\t\t\t\tconst { id, card } = tool as IA2ATool;\n\t\t\t\tfunctions.push({\n\t\t\t\t\tname: id,\n\t\t\t\t\tdescription: card.description,\n\t\t\t\t});\n\t\t\t}\n\t\t}\n\t\treturn functions;\n\t}\n}\n"],"mappings":";AAAA,SAAS,iBAAiB;AAC1B,SAAS,gBAAoC;AAQ7C,SAAS,0BAA0B;AACnC;AAAA,EAIC;AAAA,OACM;AAEA,IAAM,cAAN,cAA0B,UAAwC;AAAA,EAChE;AAAA,EACA;AAAA,EAER,YAAY,QAAgB,WAAmB;AAC9C,UAAM;AACN,SAAK,SAAS,IAAI,YAAY,EAAE,OAAO,CAAC;AACxC,SAAK,YAAY;AAAA,EAClB;AAAA,EAEQ,eAAe,MAAgB;AACtC,YAAQ,MAAM;AAAA,MACb,KAAK,SAAS;AACb,eAAO;AAAA,MACR,KAAK,SAAS;AAAA,MACd,KAAK,SAAS;AACb,eAAO;AAAA,MACR;AACC,eAAO;AAAA,IACT;AAAA,EACD;AAAA,EAEA,iBAAiB,QAIH;AACb,UAAM,EAAE,OAAO,gBAAgB,aAAa,IAAI;AAChD,UAAM,WAAsB,CAAC,eAC1B,CAAC,IACD,CAAC,EAAE,MAAM,SAAS,OAAO,CAAC,EAAE,MAAM,aAAa,KAAK,EAAE,CAAC,EAAE,CAAC;AAC7D,UAAM,iBAA4B,CAAC,iBAChC,CAAC,IACD,OAAO,KAAK,eAAe,KAAK,EAAE,IAAI,CAAC,WAAmB;AAC1D,YAAM,OAAO,eAAe,MAAM,MAAM;AAExC,aAAO;AAAA,QACN,MAAM,KAAK,eAAe,KAAK,IAAI;AAAA,QACnC,OAAO,CAAC,EAAE,MAAM,KAAK,QAAQ,MAAM,CAAC,EAAE,CAAC;AAAA,MACxC;AAAA,IACD,CAAC;AACH,UAAM,cAAuB,EAAE,MAAM,QAAQ,OAAO,CAAC,EAAE,MAAM,MAAM,CAAC,EAAE;AACtE,WAAO,SAAS,OAAO,cAAc,EAAE,OAAO,WAAW;AAAA,EAC1D;AAAA,EAEA,eAAe,UAAqB,SAAuB;AAC1D,aAAS,KAAK;AAAA,MACb,MAAM;AAAA,MACN,OAAO,CAAC,EAAE,MAAM,QAAQ,CAAC;AAAA,IAC1B,CAAC;AAAA,EACF;AAAA,EAEA,MAAM,MAAM,UAA6C;AACxD,UAAM,WAAW,MAAM,KAAK,OAAO,OAAO,gBAAgB;AAAA,MACzD,OAAO,KAAK;AAAA,MACZ,UAAU;AAAA,IACX,CAAC;AAED,WAAO,EAAE,SAAS,SAAS,KAAK;AAAA,EACjC;AAAA,EAEA,MAAM,wBACL,UACA,WACyB;AACzB,QAAI,UAAU,SAAS,GAAG;AACzB,YAAM,WAAW,MAAM,KAAK,OAAO,OAAO,gBAAgB;AAAA,QACzD,OAAO,KAAK;AAAA,QACZ,UAAU;AAAA,QACV,QAAQ;AAAA,UACP,OAAO,CAAC,EAAE,sBAAsB,UAAU,CAAC;AAAA,QAC5C;AAAA,MACD,CAAC;AAED,YAAM,EAAE,MAAM,cAAc,IAAI;AAChC,YAAM,UAAU,CACf,UAC8C;AAC9C,eAAO,MAAM,SAAS;AAAA,MACvB;AACA,YAAM,YAAoC,eACvC,OAAO,OAAO,EACf,IAAI,CAAC,UAAU;AACf,eAAO;AAAA,UACN,MAAM,MAAM;AAAA,UACZ,WAAW,MAAM;AAAA,QAClB;AAAA,MACD,CAAC;AAEF,aAAO;AAAA,QACN,SAAS;AAAA,QACT;AAAA,MACD;AAAA,IACD;AACA,WAAO,MAAM,KAAK,MAAM,QAAQ;AAAA,EACjC;AAAA,EAEA,wBAAwB,OAA4C;AACnE,UAAM,YAAmC,CAAC;AAC1C,eAAW,QAAQ,OAAO;AACzB,UAAI,CAAC,KAAK,SAAS;AAClB;AAAA,MACD;AACA,UAAI,KAAK,aAAa,mBAAmB,KAAK;AAC7C,cAAM,EAAE,SAAS,GAAG,IAAI;AACxB,kBAAU,KAAK;AAAA,UACd,MAAM;AAAA,UACN,aAAa,QAAQ;AAAA,UACrB,sBAAsB,QAAQ;AAAA,QAC/B,CAAC;AAAA,MACF,OAAO;AAEN,cAAM,EAAE,IAAI,KAAK,IAAI;AACrB,kBAAU,KAAK;AAAA,UACd,MAAM;AAAA,UACN,aAAa,KAAK;AAAA,QACnB,CAAC;AAAA,MACF;AAAA,IACD;AACA,WAAO;AAAA,EACR;AACD;","names":[]}

package/index.ts

@@ -0,0 +1,139 @@
+import { BaseModel } from "@ainetwork/adk/modules";
+import { ChatRole, type SessionObject } from "@ainetwork/adk/types/memory";
+import type {
+	FetchResponse,
+	IA2ATool,
+	IAgentTool,
+	IMCPTool,
+	ToolCall,
+} from "@ainetwork/adk/types/tool";
+import { TOOL_PROTOCOL_TYPE } from "@ainetwork/adk/types/tool";
+import {
+	type Content,
+	type FunctionCall,
+	type FunctionDeclaration,
+	GoogleGenAI,
+} from "@google/genai";
+
+export class GeminiModel extends BaseModel<Content, FunctionDeclaration> {
+	private client: GoogleGenAI;
+	private modelName: string;
+
+	constructor(apiKey: string, modelName: string) {
+		super();
+		this.client = new GoogleGenAI({ apiKey });
+		this.modelName = modelName;
+	}
+
+	private getMessageRole(role: ChatRole) {
+		switch (role) {
+			case ChatRole.USER:
+				return "user";
+			case ChatRole.MODEL:
+			case ChatRole.SYSTEM:
+				return "model";
+			default:
+				return "model"; /*FIXME*/
+		}
+	}
+
+	generateMessages(params: {
+		query: string;
+		sessionHistory?: SessionObject;
+		systemPrompt?: string;
+	}): Content[] {
+		const { query, sessionHistory, systemPrompt } = params;
+		const messages: Content[] = !systemPrompt
+			? []
+			: [{ role: "model", parts: [{ text: systemPrompt.trim() }] }];
+		const sessionContent: Content[] = !sessionHistory
+			? []
+			: Object.keys(sessionHistory.chats).map((chatId: string) => {
+					const chat = sessionHistory.chats[chatId];
+					// TODO: check message.content.type
+					return {
+						role: this.getMessageRole(chat.role),
+						parts: [{ text: chat.content.parts[0] }],
+					};
+				});
+		const userContent: Content = { role: "user", parts: [{ text: query }] };
+		return messages.concat(sessionContent).concat(userContent);
+	}
+
+	appendMessages(messages: Content[], message: string): void {
+		messages.push({
+			role: "user",
+			parts: [{ text: message }],
+		});
+	}
+
+	async fetch(messages: Content[]): Promise<FetchResponse> {
+		const response = await this.client.models.generateContent({
+			model: this.modelName,
+			contents: messages,
+		});
+
+		return { content: response.text };
+	}
+
+	async fetchWithContextMessage(
+		messages: Content[],
+		functions: FunctionDeclaration[],
+	): Promise<FetchResponse> {
+		if (functions.length > 0) {
+			const response = await this.client.models.generateContent({
+				model: this.modelName,
+				contents: messages,
+				config: {
+					tools: [{ functionDeclarations: functions }],
+				},
+			});
+
+			const { text, functionCalls } = response;
+			const hasName = (
+				value: FunctionCall,
+			): value is FunctionCall & { name: string } => {
+				return value.name !== undefined;
+			};
+			const toolCalls: ToolCall[] | undefined = functionCalls
+				?.filter(hasName)
+				.map((value) => {
+					return {
+						name: value.name,
+						arguments: value.args,
+					};
+				});
+
+			return {
+				content: text,
+				toolCalls,
+			};
+		}
+		return await this.fetch(messages);
+	}
+
+	convertToolsToFunctions(tools: IAgentTool[]): FunctionDeclaration[] {
+		const functions: FunctionDeclaration[] = [];
+		for (const tool of tools) {
+			if (!tool.enabled) {
+				continue;
+			}
+			if (tool.protocol === TOOL_PROTOCOL_TYPE.MCP) {
+				const { mcpTool, id } = tool as IMCPTool;
+				functions.push({
+					name: id,
+					description: mcpTool.description,
+					parametersJsonSchema: mcpTool.inputSchema,
+				});
+			} else {
+				// PROTOCOL_TYPE.A2A
+				const { id, card } = tool as IA2ATool;
+				functions.push({
+					name: id,
+					description: card.description,
+				});
+			}
+		}
+		return functions;
+	}
+}

package/package.json

@@ -0,0 +1,35 @@
+{
+  "name": "@ainetwork/adk-provider-model-gemini",
+  "version": "0.1.1",
+  "author": "AI Network (https://ainetwork.ai)",
+  "type": "module",
+  "engines": {
+    "node": ">=20"
+  },
+  "main": "./dist/index.cjs",
+  "module": "./dist/index.js",
+  "types": "./dist/index.d.ts",
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "import": "./dist/index.js",
+      "require": "./dist/index.cjs"
+    }
+  },
+  "scripts": {
+    "build": "tsup",
+    "clean": "rm -rf dist"
+  },
+  "dependencies": {
+    "@ainetwork/adk": "^0.1.2",
+    "@google/genai": "^1.11.0"
+  },
+  "devDependencies": {
+    "typescript": "^5.0.0"
+  },
+  "license": "MIT",
+  "publishConfig": {
+    "access": "public"
+  },
+  "gitHead": "cb6205738fd84533ce56bbeeab00ea9dfe2aaf3a"
+}

package/tsconfig.json

@@ -0,0 +1,9 @@
+{
+	"extends": "../../../tsconfig.json",
+	"compilerOptions": {
+		"outDir": "./dist",
+		"rootDir": ".",
+	},
+	"include": ["index.ts"],
+	"exclude": ["node_modules", "dist"]
+}

package/tsup.config.ts

@@ -0,0 +1,9 @@
+import { defineConfig } from 'tsup'
+
+export default defineConfig({
+  entry: ['index.ts'],
+  format: ['cjs', 'esm'],
+  dts: true,
+  sourcemap: true,
+  clean: true,
+})