@ainetwork/adk-provider-model-azure 0.1.1

package/dist/index.cjs

@@ -0,0 +1,140 @@
+"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, {
+  AzureOpenAI: () => AzureOpenAI
+});
+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_openai = require("openai");
+var AzureOpenAI = class extends import_modules.BaseModel {
+  client;
+  modelName;
+  constructor(baseUrl, apiKey, apiVersion, modelName) {
+    super();
+    this.client = new import_openai.AzureOpenAI({
+      baseURL: baseUrl,
+      apiKey,
+      apiVersion
+    });
+    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 "system";
+      default:
+        return "system";
+    }
+  }
+  generateMessages(params) {
+    const { query, sessionHistory, systemPrompt } = params;
+    const messages = !systemPrompt ? [] : [{ role: "system", content: systemPrompt.trim() }];
+    const sessionContent = !sessionHistory ? [] : Object.keys(sessionHistory.chats).map((chatId) => {
+      const chat = sessionHistory.chats[chatId];
+      return {
+        role: this.getMessageRole(chat.role),
+        content: chat.content.parts[0]
+      };
+    });
+    const userContent = { role: "user", content: query };
+    return messages.concat(sessionContent).concat(userContent);
+  }
+  appendMessages(messages, message) {
+    messages.push({
+      role: "user",
+      content: message
+    });
+  }
+  async fetch(messages) {
+    const response = await this.client.chat.completions.create({
+      model: this.modelName,
+      messages
+    });
+    return {
+      content: response.choices[0].message.content || void 0
+    };
+  }
+  async fetchWithContextMessage(messages, functions) {
+    if (functions.length > 0) {
+      const response = await this.client.chat.completions.create({
+        model: this.modelName,
+        messages,
+        tools: functions,
+        tool_choice: "auto"
+      });
+      const { content, tool_calls } = response.choices[0].message;
+      const toolCalls = tool_calls?.map(
+        (value) => {
+          return {
+            name: value.function.name,
+            // FIXME: value.function.arguments could not be a valid JSON
+            arguments: JSON.parse(value.function.arguments)
+          };
+        }
+      );
+      return {
+        content: content || void 0,
+        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({
+          type: "function",
+          function: {
+            name: id,
+            description: mcpTool.description,
+            parameters: mcpTool.inputSchema
+          }
+        });
+      } else {
+        const { id, card } = tool;
+        functions.push({
+          type: "function",
+          function: {
+            name: id,
+            description: card.description
+          }
+        });
+      }
+    }
+    return functions;
+  }
+};
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  AzureOpenAI
+});
+//# 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 { AzureOpenAI as AzureOpenAIClient } from \"openai\";\nimport type {\n\tChatCompletionMessageParam as CCMessageParam,\n\tChatCompletionMessageToolCall,\n\tChatCompletionTool,\n} from \"openai/resources\";\n\nexport class AzureOpenAI extends BaseModel<CCMessageParam, ChatCompletionTool> {\n\tprivate client: AzureOpenAIClient;\n\tprivate modelName: string;\n\n\tconstructor(\n\t\tbaseUrl: string,\n\t\tapiKey: string,\n\t\tapiVersion: string,\n\t\tmodelName: string,\n\t) {\n\t\tsuper();\n\t\tthis.client = new AzureOpenAIClient({\n\t\t\tbaseURL: baseUrl,\n\t\t\tapiKey: apiKey,\n\t\t\tapiVersion: apiVersion,\n\t\t});\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 \"system\";\n\t\t\tdefault:\n\t\t\t\treturn \"system\"; /*FIXME*/\n\t\t}\n\t}\n\n\tgenerateMessages(params: {\n\t\tquery: string;\n\t\tsessionHistory?: SessionObject;\n\t\tsystemPrompt?: string;\n\t}): CCMessageParam[] {\n\t\tconst { query, sessionHistory, systemPrompt } = params;\n\t\tconst messages: CCMessageParam[] = !systemPrompt\n\t\t\t? []\n\t\t\t: [{ role: \"system\", content: systemPrompt.trim() }];\n\t\tconst sessionContent: CCMessageParam[] = !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\treturn {\n\t\t\t\t\t\trole: this.getMessageRole(chat.role),\n\t\t\t\t\t\tcontent: chat.content.parts[0],\n\t\t\t\t\t};\n\t\t\t\t});\n\t\tconst userContent: CCMessageParam = { role: \"user\", content: query };\n\t\treturn messages.concat(sessionContent).concat(userContent);\n\t}\n\n\tappendMessages(messages: CCMessageParam[], message: string): void {\n\t\tmessages.push({\n\t\t\trole: \"user\",\n\t\t\tcontent: message,\n\t\t});\n\t}\n\n\tasync fetch(messages: CCMessageParam[]): Promise<FetchResponse> {\n\t\tconst response = await this.client.chat.completions.create({\n\t\t\tmodel: this.modelName,\n\t\t\tmessages,\n\t\t});\n\n\t\treturn {\n\t\t\tcontent: response.choices[0].message.content || undefined,\n\t\t};\n\t}\n\n\tasync fetchWithContextMessage(\n\t\tmessages: CCMessageParam[],\n\t\tfunctions: ChatCompletionTool[],\n\t): Promise<FetchResponse> {\n\t\tif (functions.length > 0) {\n\t\t\tconst response = await this.client.chat.completions.create({\n\t\t\t\tmodel: this.modelName,\n\t\t\t\tmessages,\n\t\t\t\ttools: functions,\n\t\t\t\ttool_choice: \"auto\",\n\t\t\t});\n\n\t\t\tconst { content, tool_calls } = response.choices[0].message;\n\n\t\t\tconst toolCalls: ToolCall[] | undefined = tool_calls?.map(\n\t\t\t\t(value: ChatCompletionMessageToolCall) => {\n\t\t\t\t\treturn {\n\t\t\t\t\t\tname: value.function.name,\n\t\t\t\t\t\t// FIXME: value.function.arguments could not be a valid JSON\n\t\t\t\t\t\targuments: JSON.parse(value.function.arguments),\n\t\t\t\t\t};\n\t\t\t\t},\n\t\t\t);\n\n\t\t\treturn {\n\t\t\t\tcontent: content || undefined,\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[]): ChatCompletionTool[] {\n\t\tconst functions: ChatCompletionTool[] = [];\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\ttype: \"function\",\n\t\t\t\t\tfunction: {\n\t\t\t\t\t\tname: id,\n\t\t\t\t\t\tdescription: mcpTool.description,\n\t\t\t\t\t\tparameters: mcpTool.inputSchema,\n\t\t\t\t\t},\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\ttype: \"function\",\n\t\t\t\t\tfunction: {\n\t\t\t\t\t\tname: id,\n\t\t\t\t\t\tdescription: card.description,\n\t\t\t\t\t},\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,oBAAiD;AAO1C,IAAM,cAAN,cAA0B,yBAA8C;AAAA,EACtE;AAAA,EACA;AAAA,EAER,YACC,SACA,QACA,YACA,WACC;AACD,UAAM;AACN,SAAK,SAAS,IAAI,cAAAA,YAAkB;AAAA,MACnC,SAAS;AAAA,MACT;AAAA,MACA;AAAA,IACD,CAAC;AACD,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,QAII;AACpB,UAAM,EAAE,OAAO,gBAAgB,aAAa,IAAI;AAChD,UAAM,WAA6B,CAAC,eACjC,CAAC,IACD,CAAC,EAAE,MAAM,UAAU,SAAS,aAAa,KAAK,EAAE,CAAC;AACpD,UAAM,iBAAmC,CAAC,iBACvC,CAAC,IACD,OAAO,KAAK,eAAe,KAAK,EAAE,IAAI,CAAC,WAAmB;AAC1D,YAAM,OAAO,eAAe,MAAM,MAAM;AACxC,aAAO;AAAA,QACN,MAAM,KAAK,eAAe,KAAK,IAAI;AAAA,QACnC,SAAS,KAAK,QAAQ,MAAM,CAAC;AAAA,MAC9B;AAAA,IACD,CAAC;AACH,UAAM,cAA8B,EAAE,MAAM,QAAQ,SAAS,MAAM;AACnE,WAAO,SAAS,OAAO,cAAc,EAAE,OAAO,WAAW;AAAA,EAC1D;AAAA,EAEA,eAAe,UAA4B,SAAuB;AACjE,aAAS,KAAK;AAAA,MACb,MAAM;AAAA,MACN,SAAS;AAAA,IACV,CAAC;AAAA,EACF;AAAA,EAEA,MAAM,MAAM,UAAoD;AAC/D,UAAM,WAAW,MAAM,KAAK,OAAO,KAAK,YAAY,OAAO;AAAA,MAC1D,OAAO,KAAK;AAAA,MACZ;AAAA,IACD,CAAC;AAED,WAAO;AAAA,MACN,SAAS,SAAS,QAAQ,CAAC,EAAE,QAAQ,WAAW;AAAA,IACjD;AAAA,EACD;AAAA,EAEA,MAAM,wBACL,UACA,WACyB;AACzB,QAAI,UAAU,SAAS,GAAG;AACzB,YAAM,WAAW,MAAM,KAAK,OAAO,KAAK,YAAY,OAAO;AAAA,QAC1D,OAAO,KAAK;AAAA,QACZ;AAAA,QACA,OAAO;AAAA,QACP,aAAa;AAAA,MACd,CAAC;AAED,YAAM,EAAE,SAAS,WAAW,IAAI,SAAS,QAAQ,CAAC,EAAE;AAEpD,YAAM,YAAoC,YAAY;AAAA,QACrD,CAAC,UAAyC;AACzC,iBAAO;AAAA,YACN,MAAM,MAAM,SAAS;AAAA;AAAA,YAErB,WAAW,KAAK,MAAM,MAAM,SAAS,SAAS;AAAA,UAC/C;AAAA,QACD;AAAA,MACD;AAEA,aAAO;AAAA,QACN,SAAS,WAAW;AAAA,QACpB;AAAA,MACD;AAAA,IACD;AACA,WAAO,MAAM,KAAK,MAAM,QAAQ;AAAA,EACjC;AAAA,EAEA,wBAAwB,OAA2C;AAClE,UAAM,YAAkC,CAAC;AACzC,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,UAAU;AAAA,YACT,MAAM;AAAA,YACN,aAAa,QAAQ;AAAA,YACrB,YAAY,QAAQ;AAAA,UACrB;AAAA,QACD,CAAC;AAAA,MACF,OAAO;AAEN,cAAM,EAAE,IAAI,KAAK,IAAI;AACrB,kBAAU,KAAK;AAAA,UACd,MAAM;AAAA,UACN,UAAU;AAAA,YACT,MAAM;AAAA,YACN,aAAa,KAAK;AAAA,UACnB;AAAA,QACD,CAAC;AAAA,MACF;AAAA,IACD;AACA,WAAO;AAAA,EACR;AACD;","names":["AzureOpenAIClient"]}

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 { ChatCompletionMessageParam, ChatCompletionTool } from 'openai/resources';
+
+declare class AzureOpenAI extends BaseModel<ChatCompletionMessageParam, ChatCompletionTool> {
+    private client;
+    private modelName;
+    constructor(baseUrl: string, apiKey: string, apiVersion: string, modelName: string);
+    private getMessageRole;
+    generateMessages(params: {
+        query: string;
+        sessionHistory?: SessionObject;
+        systemPrompt?: string;
+    }): ChatCompletionMessageParam[];
+    appendMessages(messages: ChatCompletionMessageParam[], message: string): void;
+    fetch(messages: ChatCompletionMessageParam[]): Promise<FetchResponse>;
+    fetchWithContextMessage(messages: ChatCompletionMessageParam[], functions: ChatCompletionTool[]): Promise<FetchResponse>;
+    convertToolsToFunctions(tools: IAgentTool[]): ChatCompletionTool[];
+}
+
+export { AzureOpenAI };

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 { ChatCompletionMessageParam, ChatCompletionTool } from 'openai/resources';
+
+declare class AzureOpenAI extends BaseModel<ChatCompletionMessageParam, ChatCompletionTool> {
+    private client;
+    private modelName;
+    constructor(baseUrl: string, apiKey: string, apiVersion: string, modelName: string);
+    private getMessageRole;
+    generateMessages(params: {
+        query: string;
+        sessionHistory?: SessionObject;
+        systemPrompt?: string;
+    }): ChatCompletionMessageParam[];
+    appendMessages(messages: ChatCompletionMessageParam[], message: string): void;
+    fetch(messages: ChatCompletionMessageParam[]): Promise<FetchResponse>;
+    fetchWithContextMessage(messages: ChatCompletionMessageParam[], functions: ChatCompletionTool[]): Promise<FetchResponse>;
+    convertToolsToFunctions(tools: IAgentTool[]): ChatCompletionTool[];
+}
+
+export { AzureOpenAI };

package/dist/index.js

@@ -0,0 +1,115 @@
+// 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 { AzureOpenAI as AzureOpenAIClient } from "openai";
+var AzureOpenAI = class extends BaseModel {
+  client;
+  modelName;
+  constructor(baseUrl, apiKey, apiVersion, modelName) {
+    super();
+    this.client = new AzureOpenAIClient({
+      baseURL: baseUrl,
+      apiKey,
+      apiVersion
+    });
+    this.modelName = modelName;
+  }
+  getMessageRole(role) {
+    switch (role) {
+      case ChatRole.USER:
+        return "user";
+      case ChatRole.MODEL:
+      case ChatRole.SYSTEM:
+        return "system";
+      default:
+        return "system";
+    }
+  }
+  generateMessages(params) {
+    const { query, sessionHistory, systemPrompt } = params;
+    const messages = !systemPrompt ? [] : [{ role: "system", content: systemPrompt.trim() }];
+    const sessionContent = !sessionHistory ? [] : Object.keys(sessionHistory.chats).map((chatId) => {
+      const chat = sessionHistory.chats[chatId];
+      return {
+        role: this.getMessageRole(chat.role),
+        content: chat.content.parts[0]
+      };
+    });
+    const userContent = { role: "user", content: query };
+    return messages.concat(sessionContent).concat(userContent);
+  }
+  appendMessages(messages, message) {
+    messages.push({
+      role: "user",
+      content: message
+    });
+  }
+  async fetch(messages) {
+    const response = await this.client.chat.completions.create({
+      model: this.modelName,
+      messages
+    });
+    return {
+      content: response.choices[0].message.content || void 0
+    };
+  }
+  async fetchWithContextMessage(messages, functions) {
+    if (functions.length > 0) {
+      const response = await this.client.chat.completions.create({
+        model: this.modelName,
+        messages,
+        tools: functions,
+        tool_choice: "auto"
+      });
+      const { content, tool_calls } = response.choices[0].message;
+      const toolCalls = tool_calls?.map(
+        (value) => {
+          return {
+            name: value.function.name,
+            // FIXME: value.function.arguments could not be a valid JSON
+            arguments: JSON.parse(value.function.arguments)
+          };
+        }
+      );
+      return {
+        content: content || void 0,
+        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({
+          type: "function",
+          function: {
+            name: id,
+            description: mcpTool.description,
+            parameters: mcpTool.inputSchema
+          }
+        });
+      } else {
+        const { id, card } = tool;
+        functions.push({
+          type: "function",
+          function: {
+            name: id,
+            description: card.description
+          }
+        });
+      }
+    }
+    return functions;
+  }
+};
+export {
+  AzureOpenAI
+};
+//# 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 { AzureOpenAI as AzureOpenAIClient } from \"openai\";\nimport type {\n\tChatCompletionMessageParam as CCMessageParam,\n\tChatCompletionMessageToolCall,\n\tChatCompletionTool,\n} from \"openai/resources\";\n\nexport class AzureOpenAI extends BaseModel<CCMessageParam, ChatCompletionTool> {\n\tprivate client: AzureOpenAIClient;\n\tprivate modelName: string;\n\n\tconstructor(\n\t\tbaseUrl: string,\n\t\tapiKey: string,\n\t\tapiVersion: string,\n\t\tmodelName: string,\n\t) {\n\t\tsuper();\n\t\tthis.client = new AzureOpenAIClient({\n\t\t\tbaseURL: baseUrl,\n\t\t\tapiKey: apiKey,\n\t\t\tapiVersion: apiVersion,\n\t\t});\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 \"system\";\n\t\t\tdefault:\n\t\t\t\treturn \"system\"; /*FIXME*/\n\t\t}\n\t}\n\n\tgenerateMessages(params: {\n\t\tquery: string;\n\t\tsessionHistory?: SessionObject;\n\t\tsystemPrompt?: string;\n\t}): CCMessageParam[] {\n\t\tconst { query, sessionHistory, systemPrompt } = params;\n\t\tconst messages: CCMessageParam[] = !systemPrompt\n\t\t\t? []\n\t\t\t: [{ role: \"system\", content: systemPrompt.trim() }];\n\t\tconst sessionContent: CCMessageParam[] = !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\treturn {\n\t\t\t\t\t\trole: this.getMessageRole(chat.role),\n\t\t\t\t\t\tcontent: chat.content.parts[0],\n\t\t\t\t\t};\n\t\t\t\t});\n\t\tconst userContent: CCMessageParam = { role: \"user\", content: query };\n\t\treturn messages.concat(sessionContent).concat(userContent);\n\t}\n\n\tappendMessages(messages: CCMessageParam[], message: string): void {\n\t\tmessages.push({\n\t\t\trole: \"user\",\n\t\t\tcontent: message,\n\t\t});\n\t}\n\n\tasync fetch(messages: CCMessageParam[]): Promise<FetchResponse> {\n\t\tconst response = await this.client.chat.completions.create({\n\t\t\tmodel: this.modelName,\n\t\t\tmessages,\n\t\t});\n\n\t\treturn {\n\t\t\tcontent: response.choices[0].message.content || undefined,\n\t\t};\n\t}\n\n\tasync fetchWithContextMessage(\n\t\tmessages: CCMessageParam[],\n\t\tfunctions: ChatCompletionTool[],\n\t): Promise<FetchResponse> {\n\t\tif (functions.length > 0) {\n\t\t\tconst response = await this.client.chat.completions.create({\n\t\t\t\tmodel: this.modelName,\n\t\t\t\tmessages,\n\t\t\t\ttools: functions,\n\t\t\t\ttool_choice: \"auto\",\n\t\t\t});\n\n\t\t\tconst { content, tool_calls } = response.choices[0].message;\n\n\t\t\tconst toolCalls: ToolCall[] | undefined = tool_calls?.map(\n\t\t\t\t(value: ChatCompletionMessageToolCall) => {\n\t\t\t\t\treturn {\n\t\t\t\t\t\tname: value.function.name,\n\t\t\t\t\t\t// FIXME: value.function.arguments could not be a valid JSON\n\t\t\t\t\t\targuments: JSON.parse(value.function.arguments),\n\t\t\t\t\t};\n\t\t\t\t},\n\t\t\t);\n\n\t\t\treturn {\n\t\t\t\tcontent: content || undefined,\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[]): ChatCompletionTool[] {\n\t\tconst functions: ChatCompletionTool[] = [];\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\ttype: \"function\",\n\t\t\t\t\tfunction: {\n\t\t\t\t\t\tname: id,\n\t\t\t\t\t\tdescription: mcpTool.description,\n\t\t\t\t\t\tparameters: mcpTool.inputSchema,\n\t\t\t\t\t},\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\ttype: \"function\",\n\t\t\t\t\tfunction: {\n\t\t\t\t\t\tname: id,\n\t\t\t\t\t\tdescription: card.description,\n\t\t\t\t\t},\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,SAAS,eAAe,yBAAyB;AAO1C,IAAM,cAAN,cAA0B,UAA8C;AAAA,EACtE;AAAA,EACA;AAAA,EAER,YACC,SACA,QACA,YACA,WACC;AACD,UAAM;AACN,SAAK,SAAS,IAAI,kBAAkB;AAAA,MACnC,SAAS;AAAA,MACT;AAAA,MACA;AAAA,IACD,CAAC;AACD,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,QAII;AACpB,UAAM,EAAE,OAAO,gBAAgB,aAAa,IAAI;AAChD,UAAM,WAA6B,CAAC,eACjC,CAAC,IACD,CAAC,EAAE,MAAM,UAAU,SAAS,aAAa,KAAK,EAAE,CAAC;AACpD,UAAM,iBAAmC,CAAC,iBACvC,CAAC,IACD,OAAO,KAAK,eAAe,KAAK,EAAE,IAAI,CAAC,WAAmB;AAC1D,YAAM,OAAO,eAAe,MAAM,MAAM;AACxC,aAAO;AAAA,QACN,MAAM,KAAK,eAAe,KAAK,IAAI;AAAA,QACnC,SAAS,KAAK,QAAQ,MAAM,CAAC;AAAA,MAC9B;AAAA,IACD,CAAC;AACH,UAAM,cAA8B,EAAE,MAAM,QAAQ,SAAS,MAAM;AACnE,WAAO,SAAS,OAAO,cAAc,EAAE,OAAO,WAAW;AAAA,EAC1D;AAAA,EAEA,eAAe,UAA4B,SAAuB;AACjE,aAAS,KAAK;AAAA,MACb,MAAM;AAAA,MACN,SAAS;AAAA,IACV,CAAC;AAAA,EACF;AAAA,EAEA,MAAM,MAAM,UAAoD;AAC/D,UAAM,WAAW,MAAM,KAAK,OAAO,KAAK,YAAY,OAAO;AAAA,MAC1D,OAAO,KAAK;AAAA,MACZ;AAAA,IACD,CAAC;AAED,WAAO;AAAA,MACN,SAAS,SAAS,QAAQ,CAAC,EAAE,QAAQ,WAAW;AAAA,IACjD;AAAA,EACD;AAAA,EAEA,MAAM,wBACL,UACA,WACyB;AACzB,QAAI,UAAU,SAAS,GAAG;AACzB,YAAM,WAAW,MAAM,KAAK,OAAO,KAAK,YAAY,OAAO;AAAA,QAC1D,OAAO,KAAK;AAAA,QACZ;AAAA,QACA,OAAO;AAAA,QACP,aAAa;AAAA,MACd,CAAC;AAED,YAAM,EAAE,SAAS,WAAW,IAAI,SAAS,QAAQ,CAAC,EAAE;AAEpD,YAAM,YAAoC,YAAY;AAAA,QACrD,CAAC,UAAyC;AACzC,iBAAO;AAAA,YACN,MAAM,MAAM,SAAS;AAAA;AAAA,YAErB,WAAW,KAAK,MAAM,MAAM,SAAS,SAAS;AAAA,UAC/C;AAAA,QACD;AAAA,MACD;AAEA,aAAO;AAAA,QACN,SAAS,WAAW;AAAA,QACpB;AAAA,MACD;AAAA,IACD;AACA,WAAO,MAAM,KAAK,MAAM,QAAQ;AAAA,EACjC;AAAA,EAEA,wBAAwB,OAA2C;AAClE,UAAM,YAAkC,CAAC;AACzC,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,UAAU;AAAA,YACT,MAAM;AAAA,YACN,aAAa,QAAQ;AAAA,YACrB,YAAY,QAAQ;AAAA,UACrB;AAAA,QACD,CAAC;AAAA,MACF,OAAO;AAEN,cAAM,EAAE,IAAI,KAAK,IAAI;AACrB,kBAAU,KAAK;AAAA,UACd,MAAM;AAAA,UACN,UAAU;AAAA,YACT,MAAM;AAAA,YACN,aAAa,KAAK;AAAA,UACnB;AAAA,QACD,CAAC;AAAA,MACF;AAAA,IACD;AACA,WAAO;AAAA,EACR;AACD;","names":[]}

package/index.ts

@@ -0,0 +1,151 @@
+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 { AzureOpenAI as AzureOpenAIClient } from "openai";
+import type {
+	ChatCompletionMessageParam as CCMessageParam,
+	ChatCompletionMessageToolCall,
+	ChatCompletionTool,
+} from "openai/resources";
+
+export class AzureOpenAI extends BaseModel<CCMessageParam, ChatCompletionTool> {
+	private client: AzureOpenAIClient;
+	private modelName: string;
+
+	constructor(
+		baseUrl: string,
+		apiKey: string,
+		apiVersion: string,
+		modelName: string,
+	) {
+		super();
+		this.client = new AzureOpenAIClient({
+			baseURL: baseUrl,
+			apiKey: apiKey,
+			apiVersion: apiVersion,
+		});
+		this.modelName = modelName;
+	}
+
+	private getMessageRole(role: ChatRole) {
+		switch (role) {
+			case ChatRole.USER:
+				return "user";
+			case ChatRole.MODEL:
+			case ChatRole.SYSTEM:
+				return "system";
+			default:
+				return "system"; /*FIXME*/
+		}
+	}
+
+	generateMessages(params: {
+		query: string;
+		sessionHistory?: SessionObject;
+		systemPrompt?: string;
+	}): CCMessageParam[] {
+		const { query, sessionHistory, systemPrompt } = params;
+		const messages: CCMessageParam[] = !systemPrompt
+			? []
+			: [{ role: "system", content: systemPrompt.trim() }];
+		const sessionContent: CCMessageParam[] = !sessionHistory
+			? []
+			: Object.keys(sessionHistory.chats).map((chatId: string) => {
+					const chat = sessionHistory.chats[chatId];
+					return {
+						role: this.getMessageRole(chat.role),
+						content: chat.content.parts[0],
+					};
+				});
+		const userContent: CCMessageParam = { role: "user", content: query };
+		return messages.concat(sessionContent).concat(userContent);
+	}
+
+	appendMessages(messages: CCMessageParam[], message: string): void {
+		messages.push({
+			role: "user",
+			content: message,
+		});
+	}
+
+	async fetch(messages: CCMessageParam[]): Promise<FetchResponse> {
+		const response = await this.client.chat.completions.create({
+			model: this.modelName,
+			messages,
+		});
+
+		return {
+			content: response.choices[0].message.content || undefined,
+		};
+	}
+
+	async fetchWithContextMessage(
+		messages: CCMessageParam[],
+		functions: ChatCompletionTool[],
+	): Promise<FetchResponse> {
+		if (functions.length > 0) {
+			const response = await this.client.chat.completions.create({
+				model: this.modelName,
+				messages,
+				tools: functions,
+				tool_choice: "auto",
+			});
+
+			const { content, tool_calls } = response.choices[0].message;
+
+			const toolCalls: ToolCall[] | undefined = tool_calls?.map(
+				(value: ChatCompletionMessageToolCall) => {
+					return {
+						name: value.function.name,
+						// FIXME: value.function.arguments could not be a valid JSON
+						arguments: JSON.parse(value.function.arguments),
+					};
+				},
+			);
+
+			return {
+				content: content || undefined,
+				toolCalls,
+			};
+		}
+		return await this.fetch(messages);
+	}
+
+	convertToolsToFunctions(tools: IAgentTool[]): ChatCompletionTool[] {
+		const functions: ChatCompletionTool[] = [];
+		for (const tool of tools) {
+			if (!tool.enabled) {
+				continue;
+			}
+			if (tool.protocol === TOOL_PROTOCOL_TYPE.MCP) {
+				const { mcpTool, id } = tool as IMCPTool;
+				functions.push({
+					type: "function",
+					function: {
+						name: id,
+						description: mcpTool.description,
+						parameters: mcpTool.inputSchema,
+					},
+				});
+			} else {
+				// PROTOCOL_TYPE.A2A
+				const { id, card } = tool as IA2ATool;
+				functions.push({
+					type: "function",
+					function: {
+						name: id,
+						description: card.description,
+					},
+				});
+			}
+		}
+		return functions;
+	}
+}

package/package.json

@@ -0,0 +1,35 @@
+{
+  "name": "@ainetwork/adk-provider-model-azure",
+  "version": "0.1.1",
+  "author": "AI Network (https://ainetwork.ai)",
+  "type": "module",
+  "engines": {
+    "node": ">=20"
+  },
+  "main": "./dist/cjs/index.js",
+  "module": "./dist/esm/index.js",
+  "types": "./dist/esm/index.d.ts",
+  "exports": {
+    ".": {
+      "import": "./dist/esm/index.js",
+      "require": "./dist/cjs/index.js",
+      "types": "./dist/esm/index.d.ts"
+    }
+  },
+  "scripts": {
+    "build": "tsup",
+    "clean": "rm -rf dist"
+  },
+  "dependencies": {
+    "@ainetwork/adk": "^0.1.0",
+    "openai": "^5.10.2"
+  },
+  "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,
+})