@ainetwork/adk 0.0.2 → 0.1.1

package/dist/esm/chunk-635VQ4CH.js

@@ -0,0 +1,42 @@
+import {
+  AinHttpError
+} from "./chunk-FL3JRQM4.js";
+
+// src/middlewares/auth.middleware.ts
+import { StatusCodes } from "http-status-codes";
+var AuthMiddleware = class {
+  auth;
+  constructor(auth) {
+    this.auth = auth;
+  }
+  middleware() {
+    return async (req, res, next) => {
+      try {
+        const isAuthenticated = await this.auth.authenticate(req, res);
+        if (isAuthenticated) {
+          next();
+        } else {
+          const error = new AinHttpError(
+            StatusCodes.UNAUTHORIZED,
+            "Unauthorized"
+          );
+          throw error;
+        }
+      } catch (e) {
+        if (!e.status) {
+          const error = new AinHttpError(
+            StatusCodes.INTERNAL_SERVER_ERROR,
+            `Authentication error: ${JSON.stringify(e)}`
+          );
+          throw error;
+        }
+        throw e;
+      }
+    };
+  }
+};
+
+export {
+  AuthMiddleware
+};
+//# sourceMappingURL=chunk-635VQ4CH.js.map

package/dist/esm/chunk-635VQ4CH.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/middlewares/auth.middleware.ts"],"sourcesContent":["import type { NextFunction, Request, RequestHandler, Response } from \"express\";\nimport { StatusCodes } from \"http-status-codes\";\nimport type { BaseAuth } from \"@/modules/auth/base.auth\";\nimport { AinHttpError } from \"@/types/agent\";\n\nexport class AuthMiddleware {\n\tprivate auth: BaseAuth;\n\tconstructor(auth: BaseAuth) {\n\t\tthis.auth = auth;\n\t}\n\n\tpublic middleware(): RequestHandler {\n\t\treturn async (req: Request, res: Response, next: NextFunction) => {\n\t\t\ttry {\n\t\t\t\tconst isAuthenticated = await this.auth.authenticate(req, res);\n\t\t\t\tif (isAuthenticated) {\n\t\t\t\t\tnext();\n\t\t\t\t} else {\n\t\t\t\t\tconst error: AinHttpError = new AinHttpError(\n\t\t\t\t\t\tStatusCodes.UNAUTHORIZED,\n\t\t\t\t\t\t\"Unauthorized\",\n\t\t\t\t\t);\n\t\t\t\t\tthrow error;\n\t\t\t\t}\n\t\t\t} catch (e: any) {\n\t\t\t\tif (!e.status) {\n\t\t\t\t\tconst error: AinHttpError = new AinHttpError(\n\t\t\t\t\t\tStatusCodes.INTERNAL_SERVER_ERROR,\n\t\t\t\t\t\t`Authentication error: ${JSON.stringify(e)}`,\n\t\t\t\t\t);\n\t\t\t\t\tthrow error;\n\t\t\t\t}\n\t\t\t\tthrow e;\n\t\t\t}\n\t\t};\n\t}\n}\n"],"mappings":";;;;;AACA,SAAS,mBAAmB;AAIrB,IAAM,iBAAN,MAAqB;AAAA,EACnB;AAAA,EACR,YAAY,MAAgB;AAC3B,SAAK,OAAO;AAAA,EACb;AAAA,EAEO,aAA6B;AACnC,WAAO,OAAO,KAAc,KAAe,SAAuB;AACjE,UAAI;AACH,cAAM,kBAAkB,MAAM,KAAK,KAAK,aAAa,KAAK,GAAG;AAC7D,YAAI,iBAAiB;AACpB,eAAK;AAAA,QACN,OAAO;AACN,gBAAM,QAAsB,IAAI;AAAA,YAC/B,YAAY;AAAA,YACZ;AAAA,UACD;AACA,gBAAM;AAAA,QACP;AAAA,MACD,SAAS,GAAQ;AAChB,YAAI,CAAC,EAAE,QAAQ;AACd,gBAAM,QAAsB,IAAI;AAAA,YAC/B,YAAY;AAAA,YACZ,yBAAyB,KAAK,UAAU,CAAC,CAAC;AAAA,UAC3C;AACA,gBAAM;AAAA,QACP;AACA,cAAM;AAAA,MACP;AAAA,IACD;AAAA,EACD;AACD;","names":[]}

package/dist/esm/chunk-64GEXWSE.js

@@ -0,0 +1,40 @@
+// src/utils/logger.ts
+import winston from "winston";
+var { combine, timestamp, printf, colorize, errors } = winston.format;
+var createLogger = (service) => {
+  const logger2 = winston.createLogger({
+    level: (process.env.LOG_LEVEL || "info").toLowerCase(),
+    format: combine(
+      errors({ stack: true }),
+      colorize(),
+      timestamp({ format: "HH:mm:ss" }),
+      printf(({ level, message, timestamp: timestamp2, service: service2, stack, ...meta }) => {
+        const metaStr = Object.keys(meta).length ? ` | ${JSON.stringify(meta)}` : "";
+        const errorStack = stack ? `
+${stack}` : "";
+        return `${timestamp2} [${service2}] ${level}: ${message}${metaStr}${errorStack}`;
+      })
+    ),
+    defaultMeta: { service },
+    transports: [new winston.transports.Console()]
+  });
+  return logger2;
+};
+var logger = createLogger("ain-adk");
+var getLogger = (service) => createLogger(service);
+var loggers = {
+  agent: getLogger("AINAgent"),
+  intent: getLogger("Intent"),
+  mcp: getLogger("MCPModule"),
+  a2a: getLogger("A2AModule"),
+  model: getLogger("Model"),
+  server: getLogger("A2AServer"),
+  fol: getLogger("FOL")
+};
+
+export {
+  logger,
+  getLogger,
+  loggers
+};
+//# sourceMappingURL=chunk-64GEXWSE.js.map

package/dist/esm/chunk-64GEXWSE.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/utils/logger.ts"],"sourcesContent":["import winston from \"winston\";\n\nconst { combine, timestamp, printf, colorize, errors } = winston.format;\n\nconst createLogger = (service: string) => {\n\tconst logger = winston.createLogger({\n\t\tlevel: (process.env.LOG_LEVEL || \"info\").toLowerCase(),\n\t\tformat: combine(\n\t\t\terrors({ stack: true }),\n\t\t\tcolorize(),\n\t\t\ttimestamp({ format: \"HH:mm:ss\" }),\n\t\t\tprintf(({ level, message, timestamp, service, stack, ...meta }) => {\n\t\t\t\tconst metaStr = Object.keys(meta).length\n\t\t\t\t\t? ` | ${JSON.stringify(meta)}`\n\t\t\t\t\t: \"\";\n\t\t\t\tconst errorStack = stack ? `\\n${stack}` : \"\";\n\t\t\t\treturn `${timestamp} [${service}] ${level}: ${message}${metaStr}${errorStack}`;\n\t\t\t}),\n\t\t),\n\t\tdefaultMeta: { service },\n\t\ttransports: [new winston.transports.Console()],\n\t});\n\n\treturn logger;\n};\n\n// Default logger instance\nexport const logger = createLogger(\"ain-adk\");\n\n// Factory function for creating service-specific loggers\nexport const getLogger = (service: string) => createLogger(service);\n\n// Convenience methods for different components\nexport const loggers = {\n\tagent: getLogger(\"AINAgent\"),\n\tintent: getLogger(\"Intent\"),\n\tmcp: getLogger(\"MCPModule\"),\n\ta2a: getLogger(\"A2AModule\"),\n\tmodel: getLogger(\"Model\"),\n\tserver: getLogger(\"A2AServer\"),\n\tfol: getLogger(\"FOL\"),\n} as const;\n"],"mappings":";AAAA,OAAO,aAAa;AAEpB,IAAM,EAAE,SAAS,WAAW,QAAQ,UAAU,OAAO,IAAI,QAAQ;AAEjE,IAAM,eAAe,CAAC,YAAoB;AACzC,QAAMA,UAAS,QAAQ,aAAa;AAAA,IACnC,QAAQ,QAAQ,IAAI,aAAa,QAAQ,YAAY;AAAA,IACrD,QAAQ;AAAA,MACP,OAAO,EAAE,OAAO,KAAK,CAAC;AAAA,MACtB,SAAS;AAAA,MACT,UAAU,EAAE,QAAQ,WAAW,CAAC;AAAA,MAChC,OAAO,CAAC,EAAE,OAAO,SAAS,WAAAC,YAAW,SAAAC,UAAS,OAAO,GAAG,KAAK,MAAM;AAClE,cAAM,UAAU,OAAO,KAAK,IAAI,EAAE,SAC/B,MAAM,KAAK,UAAU,IAAI,CAAC,KAC1B;AACH,cAAM,aAAa,QAAQ;AAAA,EAAK,KAAK,KAAK;AAC1C,eAAO,GAAGD,UAAS,KAAKC,QAAO,KAAK,KAAK,KAAK,OAAO,GAAG,OAAO,GAAG,UAAU;AAAA,MAC7E,CAAC;AAAA,IACF;AAAA,IACA,aAAa,EAAE,QAAQ;AAAA,IACvB,YAAY,CAAC,IAAI,QAAQ,WAAW,QAAQ,CAAC;AAAA,EAC9C,CAAC;AAED,SAAOF;AACR;AAGO,IAAM,SAAS,aAAa,SAAS;AAGrC,IAAM,YAAY,CAAC,YAAoB,aAAa,OAAO;AAG3D,IAAM,UAAU;AAAA,EACtB,OAAO,UAAU,UAAU;AAAA,EAC3B,QAAQ,UAAU,QAAQ;AAAA,EAC1B,KAAK,UAAU,WAAW;AAAA,EAC1B,KAAK,UAAU,WAAW;AAAA,EAC1B,OAAO,UAAU,OAAO;AAAA,EACxB,QAAQ,UAAU,WAAW;AAAA,EAC7B,KAAK,UAAU,KAAK;AACrB;","names":["logger","timestamp","service"]}

package/dist/esm/chunk-AQ5YZERE.js

@@ -0,0 +1,111 @@
+import {
+  loggers
+} from "./chunk-64GEXWSE.js";
+
+// src/services/a2a.service.ts
+import { randomUUID } from "crypto";
+var A2AService = class {
+  queryService;
+  canceledTasks = /* @__PURE__ */ new Set();
+  constructor(queryService) {
+    this.queryService = queryService;
+  }
+  cancelTask = async (taskId, _eventBus) => {
+    this.canceledTasks.add(taskId);
+  };
+  createTaskStatusUpdateEvent = (taskId, contextId, state, message) => {
+    return {
+      kind: "status-update",
+      taskId,
+      contextId,
+      status: {
+        state,
+        message: message ? {
+          kind: "message",
+          role: "agent",
+          messageId: randomUUID(),
+          parts: [{ kind: "text", text: message }],
+          taskId,
+          contextId
+        } : void 0,
+        timestamp: (/* @__PURE__ */ new Date()).toISOString()
+      },
+      final: state !== "working"
+    };
+  };
+  async execute(requestContext, eventBus) {
+    const userMessage = requestContext.userMessage;
+    const { sessionId } = userMessage.metadata;
+    const existingTask = requestContext.task;
+    const taskId = existingTask?.id || randomUUID();
+    const contextId = userMessage.contextId || existingTask?.contextId || randomUUID();
+    if (!existingTask) {
+      const initialTask = {
+        kind: "task",
+        id: taskId,
+        contextId,
+        status: {
+          state: "submitted",
+          timestamp: (/* @__PURE__ */ new Date()).toISOString()
+        },
+        history: [userMessage],
+        metadata: userMessage.metadata,
+        artifacts: []
+      };
+      eventBus.publish(initialTask);
+    }
+    const workingStatusUpdate = this.createTaskStatusUpdateEvent(
+      taskId,
+      contextId,
+      "working"
+    );
+    eventBus.publish(workingStatusUpdate);
+    const message = userMessage.parts.filter((part) => part.kind === "text").map((part) => part.text).join("\n");
+    if (message.length === 0) {
+      loggers.server.warn(`Empty message received for task ${taskId}.`);
+      const failureUpdate = this.createTaskStatusUpdateEvent(
+        taskId,
+        contextId,
+        "failed",
+        "No message found to process."
+      );
+      eventBus.publish(failureUpdate);
+      return;
+    }
+    try {
+      const response = await this.queryService.handleQuery(message, sessionId);
+      if (this.canceledTasks.has(taskId)) {
+        loggers.server.info(`Task ${taskId} was canceled.`);
+        const canceledUpdate = this.createTaskStatusUpdateEvent(
+          taskId,
+          contextId,
+          "canceled"
+        );
+        eventBus.publish(canceledUpdate);
+        return;
+      }
+      const finalUpdate = this.createTaskStatusUpdateEvent(
+        taskId,
+        contextId,
+        "completed",
+        response.content
+      );
+      eventBus.publish(finalUpdate);
+      loggers.server.info(`Task ${taskId} completed successfully.`);
+    } catch (error) {
+      loggers.server.error(`Error processing task ${taskId}:`, error);
+      const errorUpdate = this.createTaskStatusUpdateEvent(
+        taskId,
+        contextId,
+        "failed",
+        `Agent error: ${error.message}`
+      );
+      eventBus.publish(errorUpdate);
+    }
+  }
+};
+
+export {
+  A2AService
+};
+//# sourceMappingURL=chunk-AQ5YZERE.js.map

package/dist/esm/chunk-AQ5YZERE.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/services/a2a.service.ts"],"sourcesContent":["import { randomUUID } from \"node:crypto\";\nimport type { Task, TaskStatusUpdateEvent } from \"@a2a-js/sdk\";\nimport type {\n\tAgentExecutor,\n\tExecutionEventBus,\n\tRequestContext,\n} from \"@a2a-js/sdk/server\";\nimport { loggers } from \"@/utils/logger.js\";\nimport type { QueryService } from \"./query.service.js\";\n\n/**\n * Implements the AgentExecutor interface from the a2a-js-sdk.\n * This service is responsible for the core business logic of executing an A2A task.\n */\nexport class A2AService implements AgentExecutor {\n\tprivate queryService: QueryService;\n\tprivate canceledTasks: Set<string> = new Set<string>();\n\n\tconstructor(queryService: QueryService) {\n\t\tthis.queryService = queryService;\n\t}\n\n\tpublic cancelTask = async (\n\t\ttaskId: string,\n\t\t_eventBus: ExecutionEventBus,\n\t): Promise<void> => {\n\t\tthis.canceledTasks.add(taskId);\n\t};\n\n\tprivate createTaskStatusUpdateEvent = (\n\t\ttaskId: string,\n\t\tcontextId: string,\n\t\tstate: \"working\" | \"failed\" | \"canceled\" | \"completed\",\n\t\tmessage?: string,\n\t): TaskStatusUpdateEvent => {\n\t\treturn {\n\t\t\tkind: \"status-update\",\n\t\t\ttaskId: taskId,\n\t\t\tcontextId: contextId,\n\t\t\tstatus: {\n\t\t\t\tstate: state,\n\t\t\t\tmessage: message\n\t\t\t\t\t? {\n\t\t\t\t\t\t\tkind: \"message\",\n\t\t\t\t\t\t\trole: \"agent\",\n\t\t\t\t\t\t\tmessageId: randomUUID(),\n\t\t\t\t\t\t\tparts: [{ kind: \"text\", text: message }],\n\t\t\t\t\t\t\ttaskId: taskId,\n\t\t\t\t\t\t\tcontextId: contextId,\n\t\t\t\t\t\t}\n\t\t\t\t\t: undefined,\n\t\t\t\ttimestamp: new Date().toISOString(),\n\t\t\t},\n\t\t\tfinal: state !== \"working\",\n\t\t};\n\t};\n\n\tasync execute(\n\t\trequestContext: RequestContext,\n\t\teventBus: ExecutionEventBus,\n\t): Promise<void> {\n\t\tconst userMessage = requestContext.userMessage;\n\t\tconst { sessionId } = userMessage.metadata as { sessionId: string };\n\t\tconst existingTask = requestContext.task;\n\n\t\tconst taskId = existingTask?.id || randomUUID();\n\t\tconst contextId =\n\t\t\tuserMessage.contextId || existingTask?.contextId || randomUUID();\n\n\t\tif (!existingTask) {\n\t\t\tconst initialTask: Task = {\n\t\t\t\tkind: \"task\",\n\t\t\t\tid: taskId,\n\t\t\t\tcontextId: contextId,\n\t\t\t\tstatus: {\n\t\t\t\t\tstate: \"submitted\",\n\t\t\t\t\ttimestamp: new Date().toISOString(),\n\t\t\t\t},\n\t\t\t\thistory: [userMessage],\n\t\t\t\tmetadata: userMessage.metadata,\n\t\t\t\tartifacts: [],\n\t\t\t};\n\t\t\teventBus.publish(initialTask);\n\t\t}\n\n\t\tconst workingStatusUpdate = this.createTaskStatusUpdateEvent(\n\t\t\ttaskId,\n\t\t\tcontextId,\n\t\t\t\"working\",\n\t\t);\n\t\teventBus.publish(workingStatusUpdate);\n\n\t\tconst message: string = userMessage.parts\n\t\t\t.filter((part) => part.kind === \"text\")\n\t\t\t.map((part) => part.text)\n\t\t\t.join(\"\\n\");\n\t\tif (message.length === 0) {\n\t\t\tloggers.server.warn(`Empty message received for task ${taskId}.`);\n\t\t\tconst failureUpdate = this.createTaskStatusUpdateEvent(\n\t\t\t\ttaskId,\n\t\t\t\tcontextId,\n\t\t\t\t\"failed\",\n\t\t\t\t\"No message found to process.\",\n\t\t\t);\n\t\t\teventBus.publish(failureUpdate);\n\t\t\treturn;\n\t\t}\n\n\t\ttry {\n\t\t\tconst response = await this.queryService.handleQuery(message, sessionId);\n\n\t\t\tif (this.canceledTasks.has(taskId)) {\n\t\t\t\tloggers.server.info(`Task ${taskId} was canceled.`);\n\t\t\t\tconst canceledUpdate = this.createTaskStatusUpdateEvent(\n\t\t\t\t\ttaskId,\n\t\t\t\t\tcontextId,\n\t\t\t\t\t\"canceled\",\n\t\t\t\t);\n\t\t\t\teventBus.publish(canceledUpdate);\n\t\t\t\treturn;\n\t\t\t}\n\n\t\t\tconst finalUpdate = this.createTaskStatusUpdateEvent(\n\t\t\t\ttaskId,\n\t\t\t\tcontextId,\n\t\t\t\t\"completed\",\n\t\t\t\tresponse.content,\n\t\t\t);\n\t\t\teventBus.publish(finalUpdate);\n\t\t\tloggers.server.info(`Task ${taskId} completed successfully.`);\n\t\t} catch (error: any) {\n\t\t\tloggers.server.error(`Error processing task ${taskId}:`, error);\n\t\t\tconst errorUpdate = this.createTaskStatusUpdateEvent(\n\t\t\t\ttaskId,\n\t\t\t\tcontextId,\n\t\t\t\t\"failed\",\n\t\t\t\t`Agent error: ${error.message}`,\n\t\t\t);\n\t\t\teventBus.publish(errorUpdate);\n\t\t}\n\t}\n}\n"],"mappings":";;;;;AAAA,SAAS,kBAAkB;AAcpB,IAAM,aAAN,MAA0C;AAAA,EACxC;AAAA,EACA,gBAA6B,oBAAI,IAAY;AAAA,EAErD,YAAY,cAA4B;AACvC,SAAK,eAAe;AAAA,EACrB;AAAA,EAEO,aAAa,OACnB,QACA,cACmB;AACnB,SAAK,cAAc,IAAI,MAAM;AAAA,EAC9B;AAAA,EAEQ,8BAA8B,CACrC,QACA,WACA,OACA,YAC2B;AAC3B,WAAO;AAAA,MACN,MAAM;AAAA,MACN;AAAA,MACA;AAAA,MACA,QAAQ;AAAA,QACP;AAAA,QACA,SAAS,UACN;AAAA,UACA,MAAM;AAAA,UACN,MAAM;AAAA,UACN,WAAW,WAAW;AAAA,UACtB,OAAO,CAAC,EAAE,MAAM,QAAQ,MAAM,QAAQ,CAAC;AAAA,UACvC;AAAA,UACA;AAAA,QACD,IACC;AAAA,QACH,YAAW,oBAAI,KAAK,GAAE,YAAY;AAAA,MACnC;AAAA,MACA,OAAO,UAAU;AAAA,IAClB;AAAA,EACD;AAAA,EAEA,MAAM,QACL,gBACA,UACgB;AAChB,UAAM,cAAc,eAAe;AACnC,UAAM,EAAE,UAAU,IAAI,YAAY;AAClC,UAAM,eAAe,eAAe;AAEpC,UAAM,SAAS,cAAc,MAAM,WAAW;AAC9C,UAAM,YACL,YAAY,aAAa,cAAc,aAAa,WAAW;AAEhE,QAAI,CAAC,cAAc;AAClB,YAAM,cAAoB;AAAA,QACzB,MAAM;AAAA,QACN,IAAI;AAAA,QACJ;AAAA,QACA,QAAQ;AAAA,UACP,OAAO;AAAA,UACP,YAAW,oBAAI,KAAK,GAAE,YAAY;AAAA,QACnC;AAAA,QACA,SAAS,CAAC,WAAW;AAAA,QACrB,UAAU,YAAY;AAAA,QACtB,WAAW,CAAC;AAAA,MACb;AACA,eAAS,QAAQ,WAAW;AAAA,IAC7B;AAEA,UAAM,sBAAsB,KAAK;AAAA,MAChC;AAAA,MACA;AAAA,MACA;AAAA,IACD;AACA,aAAS,QAAQ,mBAAmB;AAEpC,UAAM,UAAkB,YAAY,MAClC,OAAO,CAAC,SAAS,KAAK,SAAS,MAAM,EACrC,IAAI,CAAC,SAAS,KAAK,IAAI,EACvB,KAAK,IAAI;AACX,QAAI,QAAQ,WAAW,GAAG;AACzB,cAAQ,OAAO,KAAK,mCAAmC,MAAM,GAAG;AAChE,YAAM,gBAAgB,KAAK;AAAA,QAC1B;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,MACD;AACA,eAAS,QAAQ,aAAa;AAC9B;AAAA,IACD;AAEA,QAAI;AACH,YAAM,WAAW,MAAM,KAAK,aAAa,YAAY,SAAS,SAAS;AAEvE,UAAI,KAAK,cAAc,IAAI,MAAM,GAAG;AACnC,gBAAQ,OAAO,KAAK,QAAQ,MAAM,gBAAgB;AAClD,cAAM,iBAAiB,KAAK;AAAA,UAC3B;AAAA,UACA;AAAA,UACA;AAAA,QACD;AACA,iBAAS,QAAQ,cAAc;AAC/B;AAAA,MACD;AAEA,YAAM,cAAc,KAAK;AAAA,QACxB;AAAA,QACA;AAAA,QACA;AAAA,QACA,SAAS;AAAA,MACV;AACA,eAAS,QAAQ,WAAW;AAC5B,cAAQ,OAAO,KAAK,QAAQ,MAAM,0BAA0B;AAAA,IAC7D,SAAS,OAAY;AACpB,cAAQ,OAAO,MAAM,yBAAyB,MAAM,KAAK,KAAK;AAC9D,YAAM,cAAc,KAAK;AAAA,QACxB;AAAA,QACA;AAAA,QACA;AAAA,QACA,gBAAgB,MAAM,OAAO;AAAA,MAC9B;AACA,eAAS,QAAQ,WAAW;AAAA,IAC7B;AAAA,EACD;AACD;","names":[]}

package/dist/esm/chunk-BTZVK3EA.js

@@ -0,0 +1,100 @@
+import {
+  MCPTool
+} from "./chunk-3LLHEUY2.js";
+import {
+  loggers
+} from "./chunk-64GEXWSE.js";
+
+// src/modules/mcp/mcp.module.ts
+import { Client } from "@modelcontextprotocol/sdk/client/index.js";
+import { StdioClientTransport } from "@modelcontextprotocol/sdk/client/stdio.js";
+var MCPModule = class {
+  /** Map of MCP server names to their client instances */
+  mcpMap = /* @__PURE__ */ new Map();
+  /** Map of MCP server names to their transport instances */
+  transportMap = /* @__PURE__ */ new Map();
+  /** Array of all discovered tools from connected MCP servers */
+  tools = [];
+  /**
+   * Connects to MCP servers based on the provided configuration.
+   *
+   * For each server in the config, establishes a connection, discovers
+   * available tools, and adds them to the module's tool collection.
+   * Skips servers that are already connected.
+   *
+   * @param mcpConfig - Configuration object mapping server names to connection details
+   * @throws Error if connection to any MCP server fails
+   */
+  async addMCPConfig(mcpConfig) {
+    try {
+      for (const [name, conf] of Object.entries(mcpConfig)) {
+        if (this.mcpMap.get(name) && this.transportMap.get(name)) continue;
+        const transport = new StdioClientTransport(conf);
+        this.transportMap.set(name, transport);
+        const mcp = new Client({ name: "mcp-client-cli", version: "1.0.0" });
+        await mcp.connect(transport);
+        this.mcpMap.set(name, mcp);
+        const toolsResult = await mcp.listTools();
+        this.tools.push(
+          ...toolsResult.tools.map((tool) => {
+            return new MCPTool(name, tool);
+          })
+        );
+      }
+      loggers.mcp.info("Connected to MCP server with tools:", {
+        tools: this.tools.map((tool) => tool.id)
+      });
+    } catch (error) {
+      loggers.mcp.error("Failed to connect to MCP server:", { error });
+      throw error;
+    }
+  }
+  /**
+   * Returns all available tools from connected MCP servers.
+   *
+   * @returns Array of MCPTool instances representing available tools
+   */
+  getTools() {
+    return this.tools;
+  }
+  /**
+   * Executes a tool on its corresponding MCP server.
+   *
+   * @param tool - The MCPTool instance to execute
+   * @param _args - Arguments to pass to the tool
+   * @returns Promise resolving to the tool's execution result
+   * @throws Error if the MCP server for the tool is not found
+   */
+  async useTool(tool, _args) {
+    const { serverName, mcpTool } = tool;
+    const toolName = mcpTool.name;
+    const mcp = this.mcpMap.get(serverName);
+    if (!mcp) {
+      throw new Error(`Invalid MCP Tool ${serverName}-${mcpTool.name}`);
+    }
+    const result = await mcp.callTool({
+      name: toolName,
+      arguments: _args
+    });
+    const toolResult = `[Bot Called Tool ${toolName} with args ${JSON.stringify(_args)}]
+` + JSON.stringify(result.content, null, 2);
+    loggers.mcp.debug("MCP useTool result:", toolResult);
+    return result;
+  }
+  /**
+   * Closes all MCP client connections.
+   *
+   * Should be called when shutting down the application to ensure
+   * all MCP connections are properly closed.
+   */
+  async cleanup() {
+    this.mcpMap.forEach((mcp) => {
+      mcp.close();
+    });
+  }
+};
+
+export {
+  MCPModule
+};
+//# sourceMappingURL=chunk-BTZVK3EA.js.map

package/dist/esm/chunk-BTZVK3EA.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/modules/mcp/mcp.module.ts"],"sourcesContent":["import { Client } from \"@modelcontextprotocol/sdk/client/index.js\";\nimport { StdioClientTransport } from \"@modelcontextprotocol/sdk/client/stdio.js\";\nimport type { MCPConfig } from \"@/types/mcp.js\";\nimport { loggers } from \"@/utils/logger.js\";\nimport { MCPTool } from \"./mcp.tool.js\";\n\n/**\n * Module for managing Model Context Protocol (MCP) server connections.\n *\n * This module handles the lifecycle of MCP client connections, discovers\n * available tools from connected servers, and provides an interface for\n * executing those tools. Multiple MCP servers can be connected simultaneously.\n *\n * @example\n * ```typescript\n * const mcpModule = new MCPModule();\n * await mcpModule.addMCPConfig({\n *   \"filesystem\": {\n *     command: \"npx\",\n *     args: [\"@modelcontextprotocol/server-filesystem\", \"/path/to/files\"]\n *   }\n * });\n *\n * const tools = mcpModule.getTools();\n * const result = await mcpModule.useTool(tools[0], { path: \"/example.txt\" });\n * ```\n */\nexport class MCPModule {\n\t/** Map of MCP server names to their client instances */\n\tprivate mcpMap: Map<string, Client> = new Map();\n\t/** Map of MCP server names to their transport instances */\n\tprivate transportMap: Map<string, StdioClientTransport> = new Map();\n\t/** Array of all discovered tools from connected MCP servers */\n\tprivate tools: MCPTool[] = [];\n\n\t/**\n\t * Connects to MCP servers based on the provided configuration.\n\t *\n\t * For each server in the config, establishes a connection, discovers\n\t * available tools, and adds them to the module's tool collection.\n\t * Skips servers that are already connected.\n\t *\n\t * @param mcpConfig - Configuration object mapping server names to connection details\n\t * @throws Error if connection to any MCP server fails\n\t */\n\tasync addMCPConfig(mcpConfig: MCPConfig) {\n\t\ttry {\n\t\t\tfor (const [name, conf] of Object.entries(mcpConfig)) {\n\t\t\t\t// FIXME(yoojin): Need strict duplication check.\n\t\t\t\tif (this.mcpMap.get(name) && this.transportMap.get(name)) continue; // Duplicated mcp: skip\n\n\t\t\t\tconst transport = new StdioClientTransport(conf);\n\t\t\t\tthis.transportMap.set(name, transport);\n\t\t\t\tconst mcp = new Client({ name: \"mcp-client-cli\", version: \"1.0.0\" });\n\t\t\t\tawait mcp.connect(transport);\n\t\t\t\tthis.mcpMap.set(name, mcp);\n\n\t\t\t\tconst toolsResult = await mcp.listTools();\n\t\t\t\tthis.tools.push(\n\t\t\t\t\t...toolsResult.tools.map((tool) => {\n\t\t\t\t\t\treturn new MCPTool(name, tool);\n\t\t\t\t\t}),\n\t\t\t\t);\n\t\t\t}\n\t\t\tloggers.mcp.info(\"Connected to MCP server with tools:\", {\n\t\t\t\ttools: this.tools.map((tool) => tool.id),\n\t\t\t});\n\t\t} catch (error: unknown) {\n\t\t\tloggers.mcp.error(\"Failed to connect to MCP server:\", { error });\n\t\t\tthrow error;\n\t\t}\n\t}\n\n\t/**\n\t * Returns all available tools from connected MCP servers.\n\t *\n\t * @returns Array of MCPTool instances representing available tools\n\t */\n\tgetTools() {\n\t\treturn this.tools;\n\t}\n\n\t/**\n\t * Executes a tool on its corresponding MCP server.\n\t *\n\t * @param tool - The MCPTool instance to execute\n\t * @param _args - Arguments to pass to the tool\n\t * @returns Promise resolving to the tool's execution result\n\t * @throws Error if the MCP server for the tool is not found\n\t */\n\tasync useTool(tool: MCPTool, _args?: any): Promise<any> {\n\t\tconst { serverName, mcpTool } = tool;\n\t\tconst toolName = mcpTool.name;\n\t\tconst mcp = this.mcpMap.get(serverName);\n\n\t\tif (!mcp) {\n\t\t\tthrow new Error(`Invalid MCP Tool ${serverName}-${mcpTool.name}`);\n\t\t}\n\n\t\tconst result = await mcp.callTool({\n\t\t\tname: toolName,\n\t\t\targuments: _args,\n\t\t});\n\t\tconst toolResult =\n\t\t\t`[Bot Called Tool ${toolName} with args ${JSON.stringify(_args)}]\\n` +\n\t\t\tJSON.stringify(result.content, null, 2);\n\n\t\tloggers.mcp.debug(\"MCP useTool result:\", toolResult);\n\t\treturn result;\n\t}\n\n\t/**\n\t * Closes all MCP client connections.\n\t *\n\t * Should be called when shutting down the application to ensure\n\t * all MCP connections are properly closed.\n\t */\n\tasync cleanup() {\n\t\tthis.mcpMap.forEach((mcp: Client) => {\n\t\t\tmcp.close();\n\t\t});\n\t}\n}\n"],"mappings":";;;;;;;;AAAA,SAAS,cAAc;AACvB,SAAS,4BAA4B;AA0B9B,IAAM,YAAN,MAAgB;AAAA;AAAA,EAEd,SAA8B,oBAAI,IAAI;AAAA;AAAA,EAEtC,eAAkD,oBAAI,IAAI;AAAA;AAAA,EAE1D,QAAmB,CAAC;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAY5B,MAAM,aAAa,WAAsB;AACxC,QAAI;AACH,iBAAW,CAAC,MAAM,IAAI,KAAK,OAAO,QAAQ,SAAS,GAAG;AAErD,YAAI,KAAK,OAAO,IAAI,IAAI,KAAK,KAAK,aAAa,IAAI,IAAI,EAAG;AAE1D,cAAM,YAAY,IAAI,qBAAqB,IAAI;AAC/C,aAAK,aAAa,IAAI,MAAM,SAAS;AACrC,cAAM,MAAM,IAAI,OAAO,EAAE,MAAM,kBAAkB,SAAS,QAAQ,CAAC;AACnE,cAAM,IAAI,QAAQ,SAAS;AAC3B,aAAK,OAAO,IAAI,MAAM,GAAG;AAEzB,cAAM,cAAc,MAAM,IAAI,UAAU;AACxC,aAAK,MAAM;AAAA,UACV,GAAG,YAAY,MAAM,IAAI,CAAC,SAAS;AAClC,mBAAO,IAAI,QAAQ,MAAM,IAAI;AAAA,UAC9B,CAAC;AAAA,QACF;AAAA,MACD;AACA,cAAQ,IAAI,KAAK,uCAAuC;AAAA,QACvD,OAAO,KAAK,MAAM,IAAI,CAAC,SAAS,KAAK,EAAE;AAAA,MACxC,CAAC;AAAA,IACF,SAAS,OAAgB;AACxB,cAAQ,IAAI,MAAM,oCAAoC,EAAE,MAAM,CAAC;AAC/D,YAAM;AAAA,IACP;AAAA,EACD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,WAAW;AACV,WAAO,KAAK;AAAA,EACb;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAUA,MAAM,QAAQ,MAAe,OAA2B;AACvD,UAAM,EAAE,YAAY,QAAQ,IAAI;AAChC,UAAM,WAAW,QAAQ;AACzB,UAAM,MAAM,KAAK,OAAO,IAAI,UAAU;AAEtC,QAAI,CAAC,KAAK;AACT,YAAM,IAAI,MAAM,oBAAoB,UAAU,IAAI,QAAQ,IAAI,EAAE;AAAA,IACjE;AAEA,UAAM,SAAS,MAAM,IAAI,SAAS;AAAA,MACjC,MAAM;AAAA,MACN,WAAW;AAAA,IACZ,CAAC;AACD,UAAM,aACL,oBAAoB,QAAQ,cAAc,KAAK,UAAU,KAAK,CAAC;AAAA,IAC/D,KAAK,UAAU,OAAO,SAAS,MAAM,CAAC;AAEvC,YAAQ,IAAI,MAAM,uBAAuB,UAAU;AACnD,WAAO;AAAA,EACR;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQA,MAAM,UAAU;AACf,SAAK,OAAO,QAAQ,CAAC,QAAgB;AACpC,UAAI,MAAM;AAAA,IACX,CAAC;AAAA,EACF;AACD;","names":[]}

package/dist/esm/chunk-CH4XHVJY.js

@@ -0,0 +1,37 @@
+import {
+  A2AService
+} from "./chunk-AQ5YZERE.js";
+import {
+  QueryService
+} from "./chunk-WZBXLGQK.js";
+import {
+  A2AController
+} from "./chunk-EODMRV4H.js";
+
+// src/routes/a2a.routes.ts
+import { InMemoryTaskStore } from "@a2a-js/sdk/server";
+import { Router } from "express";
+var createA2ARouter = (agent) => {
+  const router = Router();
+  const taskStore = new InMemoryTaskStore();
+  const queryService = new QueryService(
+    agent.modelModule,
+    agent.a2aModule,
+    agent.mcpModule,
+    agent.memoryModule,
+    agent.manifest.prompts
+  );
+  const a2aService = new A2AService(queryService);
+  const a2aController = new A2AController(
+    a2aService,
+    taskStore,
+    agent.generateAgentCard
+  );
+  router.post("/a2a", a2aController.handleA2ARequest);
+  return router;
+};
+
+export {
+  createA2ARouter
+};
+//# sourceMappingURL=chunk-CH4XHVJY.js.map

package/dist/esm/chunk-CH4XHVJY.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/routes/a2a.routes.ts"],"sourcesContent":["import { InMemoryTaskStore } from \"@a2a-js/sdk/server\";\nimport { Router } from \"express\";\nimport type { AINAgent } from \"@/index.js\";\nimport { QueryService } from \"@/services/query.service.js\";\nimport { A2AController } from \"../controllers/a2a.controller.js\";\nimport { A2AService } from \"../services/a2a.service.js\";\n\n/**\n * Creates and configures the A2A router.\n * This function is a \"composition root\" for the A2A feature,\n * creating and injecting all necessary dependencies.\n * @param intentAnalyzer The core intent analyzer.\n * @param agentCard The agent's card.\n * @returns An Express Router instance.\n */\nexport const createA2ARouter = (agent: AINAgent): Router => {\n\tconst router = Router();\n\n\t// 1. Create dependencies for the A2A feature\n\tconst taskStore = new InMemoryTaskStore();\n\tconst queryService = new QueryService(\n\t\tagent.modelModule,\n\t\tagent.a2aModule,\n\t\tagent.mcpModule,\n\t\tagent.memoryModule,\n\t\tagent.manifest.prompts,\n\t);\n\tconst a2aService = new A2AService(queryService);\n\tconst a2aController = new A2AController(\n\t\ta2aService,\n\t\ttaskStore,\n\t\tagent.generateAgentCard,\n\t);\n\n\t// 2. Define the route\n\trouter.post(\"/a2a\", a2aController.handleA2ARequest);\n\n\treturn router;\n};\n"],"mappings":";;;;;;;;;;;AAAA,SAAS,yBAAyB;AAClC,SAAS,cAAc;AAchB,IAAM,kBAAkB,CAAC,UAA4B;AAC3D,QAAM,SAAS,OAAO;AAGtB,QAAM,YAAY,IAAI,kBAAkB;AACxC,QAAM,eAAe,IAAI;AAAA,IACxB,MAAM;AAAA,IACN,MAAM;AAAA,IACN,MAAM;AAAA,IACN,MAAM;AAAA,IACN,MAAM,SAAS;AAAA,EAChB;AACA,QAAM,aAAa,IAAI,WAAW,YAAY;AAC9C,QAAM,gBAAgB,IAAI;AAAA,IACzB;AAAA,IACA;AAAA,IACA,MAAM;AAAA,EACP;AAGA,SAAO,KAAK,QAAQ,cAAc,gBAAgB;AAElD,SAAO;AACR;","names":[]}

package/dist/esm/chunk-CQ6GXYN2.js

@@ -0,0 +1,27 @@
+import {
+  QueryController
+} from "./chunk-XOMQNCCR.js";
+import {
+  QueryService
+} from "./chunk-WZBXLGQK.js";
+
+// src/routes/query.routes.ts
+import { Router } from "express";
+var createQueryRouter = (agent) => {
+  const router = Router();
+  const queryService = new QueryService(
+    agent.modelModule,
+    agent.a2aModule,
+    agent.mcpModule,
+    agent.memoryModule,
+    agent.manifest.prompts
+  );
+  const queryController = new QueryController(queryService);
+  router.post("/query", queryController.handleQueryRequest);
+  return router;
+};
+
+export {
+  createQueryRouter
+};
+//# sourceMappingURL=chunk-CQ6GXYN2.js.map

package/dist/esm/chunk-CQ6GXYN2.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/routes/query.routes.ts"],"sourcesContent":["import { Router } from \"express\";\nimport { QueryController } from \"@/controllers/query.controller.js\";\nimport type { AINAgent } from \"@/index.js\";\nimport { QueryService } from \"@/services/query.service.js\";\n\nexport const createQueryRouter = (agent: AINAgent): Router => {\n\tconst router = Router();\n\n\tconst queryService = new QueryService(\n\t\tagent.modelModule,\n\t\tagent.a2aModule,\n\t\tagent.mcpModule,\n\t\tagent.memoryModule,\n\t\tagent.manifest.prompts,\n\t);\n\tconst queryController = new QueryController(queryService);\n\trouter.post(\"/query\", queryController.handleQueryRequest);\n\n\treturn router;\n};\n"],"mappings":";;;;;;;;AAAA,SAAS,cAAc;AAKhB,IAAM,oBAAoB,CAAC,UAA4B;AAC7D,QAAM,SAAS,OAAO;AAEtB,QAAM,eAAe,IAAI;AAAA,IACxB,MAAM;AAAA,IACN,MAAM;AAAA,IACN,MAAM;AAAA,IACN,MAAM;AAAA,IACN,MAAM,SAAS;AAAA,EAChB;AACA,QAAM,kBAAkB,IAAI,gBAAgB,YAAY;AACxD,SAAO,KAAK,UAAU,gBAAgB,kBAAkB;AAExD,SAAO;AACR;","names":[]}

package/dist/esm/chunk-DGH3GB5I.js

@@ -0,0 +1,26 @@
+// src/modules/a2a/a2a.tool.ts
+var A2ATool = class {
+  id;
+  protocol;
+  enabled;
+  client;
+  card;
+  constructor(name, client, card) {
+    this.id = name;
+    this.protocol = "A2A" /* A2A */;
+    this.enabled = true;
+    this.client = client;
+    this.card = card;
+  }
+  enable() {
+    this.enabled = true;
+  }
+  disable() {
+    this.enabled = false;
+  }
+};
+
+export {
+  A2ATool
+};
+//# sourceMappingURL=chunk-DGH3GB5I.js.map

package/dist/esm/chunk-DGH3GB5I.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/modules/a2a/a2a.tool.ts"],"sourcesContent":["import type { AgentCard } from \"@a2a-js/sdk\";\nimport type { A2AClient } from \"@a2a-js/sdk/client\";\nimport { type IA2ATool, TOOL_PROTOCOL_TYPE } from \"@/types/tool.js\";\n\nexport class A2ATool implements IA2ATool {\n\tpublic id: string;\n\tpublic protocol: TOOL_PROTOCOL_TYPE;\n\tpublic enabled: boolean;\n\tpublic client: A2AClient;\n\tpublic card: AgentCard;\n\n\tconstructor(name: string, client: A2AClient, card: AgentCard) {\n\t\tthis.id = name;\n\t\tthis.protocol = TOOL_PROTOCOL_TYPE.A2A;\n\t\tthis.enabled = true;\n\t\tthis.client = client;\n\t\tthis.card = card;\n\t}\n\n\tpublic enable(): void {\n\t\tthis.enabled = true;\n\t}\n\n\tpublic disable(): void {\n\t\tthis.enabled = false;\n\t}\n}\n"],"mappings":";AAIO,IAAM,UAAN,MAAkC;AAAA,EACjC;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EAEP,YAAY,MAAc,QAAmB,MAAiB;AAC7D,SAAK,KAAK;AACV,SAAK;AACL,SAAK,UAAU;AACf,SAAK,SAAS;AACd,SAAK,OAAO;AAAA,EACb;AAAA,EAEO,SAAe;AACrB,SAAK,UAAU;AAAA,EAChB;AAAA,EAEO,UAAgB;AACtB,SAAK,UAAU;AAAA,EAChB;AACD;","names":[]}

package/dist/esm/chunk-EODMRV4H.js

@@ -0,0 +1,111 @@
+import {
+  loggers
+} from "./chunk-64GEXWSE.js";
+
+// src/controllers/a2a.controller.ts
+import {
+  A2AError,
+  DefaultExecutionEventBusManager,
+  DefaultRequestHandler,
+  JsonRpcTransportHandler
+} from "@a2a-js/sdk/server";
+var AINRequestHandler = class extends DefaultRequestHandler {
+  cardGenerator;
+  constructor(cardGenerator, taskStore, executor, eventBusManager = new DefaultExecutionEventBusManager()) {
+    const card = cardGenerator();
+    super(card, taskStore, executor, eventBusManager);
+    this.cardGenerator = cardGenerator;
+  }
+  async getAgentCard() {
+    return this.cardGenerator();
+  }
+};
+var A2AController = class {
+  jsonRpcTransportHandler;
+  constructor(a2aService, taskStore, cardGenerator) {
+    const requestHandler = new AINRequestHandler(
+      cardGenerator,
+      taskStore,
+      a2aService
+      // Injecting the service here as the executor
+    );
+    this.jsonRpcTransportHandler = new JsonRpcTransportHandler(requestHandler);
+  }
+  /**
+   * Handles the POST /a2a request.
+   */
+  handleA2ARequest = async (req, res) => {
+    try {
+      const rpcResponseOrStream = await this.jsonRpcTransportHandler.handle(
+        req.body
+      );
+      if (typeof rpcResponseOrStream?.[Symbol.asyncIterator] === "function") {
+        const stream = rpcResponseOrStream;
+        this.streamSse(req, res, stream);
+      } else {
+        const rpcResponse = rpcResponseOrStream;
+        res.status(200).json(rpcResponse);
+      }
+    } catch (error) {
+      this.handleError(req, res, error);
+    }
+  };
+  async streamSse(req, res, stream) {
+    res.setHeader("Content-Type", "text/event-stream");
+    res.setHeader("Cache-Control", "no-cache");
+    res.setHeader("Connection", "keep-alive");
+    res.flushHeaders();
+    try {
+      for await (const event of stream) {
+        res.write(`id: ${Date.now()}
+`);
+        res.write(`data: ${JSON.stringify(event)}
+
+`);
+      }
+    } catch (streamError) {
+      loggers.server.error(
+        `Error during SSE streaming (request ${req.body?.id}):`,
+        streamError
+      );
+      const a2aError = streamError instanceof A2AError ? streamError : A2AError.internalError("Streaming error");
+      const errorResponse = {
+        jsonrpc: "2.0",
+        id: req.body?.id || null,
+        error: a2aError.toJSONRPCError()
+      };
+      res.write(`id: ${Date.now()}
+`);
+      res.write("event: error\n");
+      res.write(`data: ${JSON.stringify(errorResponse)}
+
+`);
+    } finally {
+      if (!res.writableEnded) {
+        res.end();
+      }
+    }
+  }
+  handleError(req, res, error) {
+    loggers.server.error(
+      "Unhandled error in AINAgent A2A POST handler:",
+      error
+    );
+    const a2aError = error instanceof A2AError ? error : A2AError.internalError("General processing error");
+    const errorResponse = {
+      jsonrpc: "2.0",
+      id: req.body?.id || null,
+      error: a2aError.toJSONRPCError()
+    };
+    if (!res.headersSent) {
+      res.status(500).json(errorResponse);
+    } else if (!res.writableEnded) {
+      res.end();
+    }
+  }
+};
+
+export {
+  A2AController
+};
+//# sourceMappingURL=chunk-EODMRV4H.js.map

package/dist/esm/chunk-EODMRV4H.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/controllers/a2a.controller.ts"],"sourcesContent":["import type {\n\tA2AResponse,\n\tAgentCard,\n\tJSONRPCErrorResponse,\n\tJSONRPCSuccessResponse,\n} from \"@a2a-js/sdk\";\nimport type {\n\tAgentExecutor,\n\tExecutionEventBusManager,\n\tTaskStore,\n} from \"@a2a-js/sdk/server\";\nimport {\n\tA2AError,\n\tDefaultExecutionEventBusManager,\n\tDefaultRequestHandler,\n\tJsonRpcTransportHandler,\n} from \"@a2a-js/sdk/server\";\nimport type { Request, Response } from \"express\";\nimport type { A2AService } from \"@/services/a2a.service.js\";\nimport { loggers } from \"@/utils/logger.js\";\n\n/**\n * Handles the transport layer for A2A communication.\n * It orchestrates the A2A-JS SDK components and manages the HTTP request/response lifecycle.\n */\n\nclass AINRequestHandler extends DefaultRequestHandler {\n\tprivate cardGenerator: () => AgentCard;\n\n\tconstructor(\n\t\tcardGenerator: () => AgentCard,\n\t\ttaskStore: TaskStore,\n\t\texecutor: AgentExecutor,\n\t\teventBusManager: ExecutionEventBusManager = new DefaultExecutionEventBusManager(),\n\t) {\n\t\tconst card = cardGenerator();\n\t\tsuper(card, taskStore, executor, eventBusManager);\n\t\tthis.cardGenerator = cardGenerator;\n\t}\n\n\tasync getAgentCard(): Promise<AgentCard> {\n\t\treturn this.cardGenerator();\n\t}\n}\n\nexport class A2AController {\n\tprivate jsonRpcTransportHandler: JsonRpcTransportHandler;\n\n\tconstructor(\n\t\ta2aService: A2AService,\n\t\ttaskStore: TaskStore,\n\t\tcardGenerator: () => AgentCard,\n\t) {\n\t\t// The controller is responsible for setting up the SDK components.\n\t\tconst requestHandler = new AINRequestHandler(\n\t\t\tcardGenerator,\n\t\t\ttaskStore,\n\t\t\ta2aService, // Injecting the service here as the executor\n\t\t);\n\t\tthis.jsonRpcTransportHandler = new JsonRpcTransportHandler(requestHandler);\n\t}\n\n\t/**\n\t * Handles the POST /a2a request.\n\t */\n\tpublic handleA2ARequest = async (req: Request, res: Response) => {\n\t\ttry {\n\t\t\tconst rpcResponseOrStream = await this.jsonRpcTransportHandler.handle(\n\t\t\t\treq.body,\n\t\t\t);\n\n\t\t\t// Handle streaming responses (AsyncGenerator)\n\t\t\tif (\n\t\t\t\ttypeof (rpcResponseOrStream as any)?.[Symbol.asyncIterator] ===\n\t\t\t\t\"function\"\n\t\t\t) {\n\t\t\t\tconst stream = rpcResponseOrStream as AsyncGenerator<\n\t\t\t\t\tJSONRPCSuccessResponse,\n\t\t\t\t\tvoid,\n\t\t\t\t\tundefined\n\t\t\t\t>;\n\t\t\t\tthis.streamSse(req, res, stream);\n\t\t\t} else {\n\t\t\t\t// Handle single JSON-RPC response\n\t\t\t\tconst rpcResponse = rpcResponseOrStream as A2AResponse;\n\t\t\t\tres.status(200).json(rpcResponse);\n\t\t\t}\n\t\t} catch (error: unknown) {\n\t\t\tthis.handleError(req, res, error);\n\t\t}\n\t};\n\n\tprivate async streamSse(\n\t\treq: Request,\n\t\tres: Response,\n\t\tstream: AsyncGenerator<JSONRPCSuccessResponse, void, undefined>,\n\t) {\n\t\tres.setHeader(\"Content-Type\", \"text/event-stream\");\n\t\tres.setHeader(\"Cache-Control\", \"no-cache\");\n\t\tres.setHeader(\"Connection\", \"keep-alive\");\n\t\tres.flushHeaders();\n\n\t\ttry {\n\t\t\tfor await (const event of stream) {\n\t\t\t\tres.write(`id: ${Date.now()}\\n`);\n\t\t\t\tres.write(`data: ${JSON.stringify(event)}\\n\\n`);\n\t\t\t}\n\t\t} catch (streamError: unknown) {\n\t\t\tloggers.server.error(\n\t\t\t\t`Error during SSE streaming (request ${req.body?.id}):`,\n\t\t\t\tstreamError,\n\t\t\t);\n\t\t\tconst a2aError =\n\t\t\t\tstreamError instanceof A2AError\n\t\t\t\t\t? streamError\n\t\t\t\t\t: A2AError.internalError(\"Streaming error\");\n\t\t\tconst errorResponse: JSONRPCErrorResponse = {\n\t\t\t\tjsonrpc: \"2.0\",\n\t\t\t\tid: req.body?.id || null,\n\t\t\t\terror: a2aError.toJSONRPCError(),\n\t\t\t};\n\t\t\tres.write(`id: ${Date.now()}\\n`);\n\t\t\tres.write(\"event: error\\n\");\n\t\t\tres.write(`data: ${JSON.stringify(errorResponse)}\\n\\n`);\n\t\t} finally {\n\t\t\tif (!res.writableEnded) {\n\t\t\t\tres.end();\n\t\t\t}\n\t\t}\n\t}\n\n\tprivate handleError(req: Request, res: Response, error: unknown) {\n\t\tloggers.server.error(\n\t\t\t\"Unhandled error in AINAgent A2A POST handler:\",\n\t\t\terror,\n\t\t);\n\t\tconst a2aError =\n\t\t\terror instanceof A2AError\n\t\t\t\t? error\n\t\t\t\t: A2AError.internalError(\"General processing error\");\n\t\tconst errorResponse: JSONRPCErrorResponse = {\n\t\t\tjsonrpc: \"2.0\",\n\t\t\tid: req.body?.id || null,\n\t\t\terror: a2aError.toJSONRPCError(),\n\t\t};\n\t\tif (!res.headersSent) {\n\t\t\tres.status(500).json(errorResponse);\n\t\t} else if (!res.writableEnded) {\n\t\t\tres.end();\n\t\t}\n\t}\n}\n"],"mappings":";;;;;AAWA;AAAA,EACC;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,OACM;AAUP,IAAM,oBAAN,cAAgC,sBAAsB;AAAA,EAC7C;AAAA,EAER,YACC,eACA,WACA,UACA,kBAA4C,IAAI,gCAAgC,GAC/E;AACD,UAAM,OAAO,cAAc;AAC3B,UAAM,MAAM,WAAW,UAAU,eAAe;AAChD,SAAK,gBAAgB;AAAA,EACtB;AAAA,EAEA,MAAM,eAAmC;AACxC,WAAO,KAAK,cAAc;AAAA,EAC3B;AACD;AAEO,IAAM,gBAAN,MAAoB;AAAA,EAClB;AAAA,EAER,YACC,YACA,WACA,eACC;AAED,UAAM,iBAAiB,IAAI;AAAA,MAC1B;AAAA,MACA;AAAA,MACA;AAAA;AAAA,IACD;AACA,SAAK,0BAA0B,IAAI,wBAAwB,cAAc;AAAA,EAC1E;AAAA;AAAA;AAAA;AAAA,EAKO,mBAAmB,OAAO,KAAc,QAAkB;AAChE,QAAI;AACH,YAAM,sBAAsB,MAAM,KAAK,wBAAwB;AAAA,QAC9D,IAAI;AAAA,MACL;AAGA,UACC,OAAQ,sBAA8B,OAAO,aAAa,MAC1D,YACC;AACD,cAAM,SAAS;AAKf,aAAK,UAAU,KAAK,KAAK,MAAM;AAAA,MAChC,OAAO;AAEN,cAAM,cAAc;AACpB,YAAI,OAAO,GAAG,EAAE,KAAK,WAAW;AAAA,MACjC;AAAA,IACD,SAAS,OAAgB;AACxB,WAAK,YAAY,KAAK,KAAK,KAAK;AAAA,IACjC;AAAA,EACD;AAAA,EAEA,MAAc,UACb,KACA,KACA,QACC;AACD,QAAI,UAAU,gBAAgB,mBAAmB;AACjD,QAAI,UAAU,iBAAiB,UAAU;AACzC,QAAI,UAAU,cAAc,YAAY;AACxC,QAAI,aAAa;AAEjB,QAAI;AACH,uBAAiB,SAAS,QAAQ;AACjC,YAAI,MAAM,OAAO,KAAK,IAAI,CAAC;AAAA,CAAI;AAC/B,YAAI,MAAM,SAAS,KAAK,UAAU,KAAK,CAAC;AAAA;AAAA,CAAM;AAAA,MAC/C;AAAA,IACD,SAAS,aAAsB;AAC9B,cAAQ,OAAO;AAAA,QACd,uCAAuC,IAAI,MAAM,EAAE;AAAA,QACnD;AAAA,MACD;AACA,YAAM,WACL,uBAAuB,WACpB,cACA,SAAS,cAAc,iBAAiB;AAC5C,YAAM,gBAAsC;AAAA,QAC3C,SAAS;AAAA,QACT,IAAI,IAAI,MAAM,MAAM;AAAA,QACpB,OAAO,SAAS,eAAe;AAAA,MAChC;AACA,UAAI,MAAM,OAAO,KAAK,IAAI,CAAC;AAAA,CAAI;AAC/B,UAAI,MAAM,gBAAgB;AAC1B,UAAI,MAAM,SAAS,KAAK,UAAU,aAAa,CAAC;AAAA;AAAA,CAAM;AAAA,IACvD,UAAE;AACD,UAAI,CAAC,IAAI,eAAe;AACvB,YAAI,IAAI;AAAA,MACT;AAAA,IACD;AAAA,EACD;AAAA,EAEQ,YAAY,KAAc,KAAe,OAAgB;AAChE,YAAQ,OAAO;AAAA,MACd;AAAA,MACA;AAAA,IACD;AACA,UAAM,WACL,iBAAiB,WACd,QACA,SAAS,cAAc,0BAA0B;AACrD,UAAM,gBAAsC;AAAA,MAC3C,SAAS;AAAA,MACT,IAAI,IAAI,MAAM,MAAM;AAAA,MACpB,OAAO,SAAS,eAAe;AAAA,IAChC;AACA,QAAI,CAAC,IAAI,aAAa;AACrB,UAAI,OAAO,GAAG,EAAE,KAAK,aAAa;AAAA,IACnC,WAAW,CAAC,IAAI,eAAe;AAC9B,UAAI,IAAI;AAAA,IACT;AAAA,EACD;AACD;","names":[]}

package/dist/esm/chunk-FI2UQLVC.js

@@ -0,0 +1,11 @@
+// src/types/tool.ts
+var TOOL_PROTOCOL_TYPE = /* @__PURE__ */ ((TOOL_PROTOCOL_TYPE2) => {
+  TOOL_PROTOCOL_TYPE2["A2A"] = "A2A";
+  TOOL_PROTOCOL_TYPE2["MCP"] = "MCP";
+  return TOOL_PROTOCOL_TYPE2;
+})(TOOL_PROTOCOL_TYPE || {});
+
+export {
+  TOOL_PROTOCOL_TYPE
+};
+//# sourceMappingURL=chunk-FI2UQLVC.js.map

package/dist/esm/chunk-FI2UQLVC.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/types/tool.ts"],"sourcesContent":["import type { AgentCard } from \"@a2a-js/sdk\";\nimport type { A2AClient } from \"@a2a-js/sdk/client\";\nimport type { Tool } from \"@modelcontextprotocol/sdk/types.js\";\n\n/**\n * Supported tool protocol types in the AIN-ADK framework.\n */\nexport enum TOOL_PROTOCOL_TYPE {\n\t/** Agent-to-Agent protocol */\n\tA2A = \"A2A\",\n\t/** Model Context Protocol */\n\tMCP = \"MCP\",\n}\n\n/**\n * Represents a tool invocation request.\n */\nexport type ToolCall = {\n\t/** Name of the tool to invoke */\n\tname: string;\n\t/** Arguments to pass to the tool */\n\targuments?: Record<string, unknown>;\n};\n\n/**\n * Response from a model fetch operation.\n *\n * Contains either content (text response) or tool calls (function invocations),\n * or both in the case of mixed responses.\n */\nexport type FetchResponse = {\n\t/** Text content response from the model */\n\tcontent?: string;\n\t/** Array of tool calls requested by the model */\n\ttoolCalls?: ToolCall[];\n};\n\n/**\n * MCP-specific tool implementation.\n *\n * Wraps an MCP tool with additional metadata and functionality\n * required for integration with the AIN-ADK framework.\n */\nexport interface IMCPTool extends IAgentTool {\n\t/** The underlying MCP tool definition */\n\tmcpTool: Tool;\n\t/** Name of the MCP server providing this tool */\n\tserverName: string;\n}\n\n/**\n * A2A-specific tool implementation.\n *\n * Represents a tool provided by another agent through the A2A protocol,\n * including the client connection and agent metadata.\n */\nexport interface IA2ATool extends IAgentTool {\n\t/** A2A client instance for communication with the remote agent */\n\tclient: A2AClient;\n\t/** Agent card containing metadata about the remote agent */\n\tcard: AgentCard;\n}\n\n/**\n * Base interface for all tools in the AIN-ADK framework.\n *\n * Provides a protocol-agnostic interface for tool management,\n * allowing tools from different sources (MCP, A2A) to be used\n * interchangeably throughout the system.\n *\n * @example\n * ```typescript\n * const tool: IAgentTool = {\n *   id: \"search-tool\",\n *   protocol: TOOL_PROTOCOL_TYPE.MCP,\n *   enabled: true,\n *   enable: () => { this.enabled = true; },\n *   disable: () => { this.enabled = false; }\n * };\n * ```\n */\nexport interface IAgentTool {\n\t/** Unique identifier for the tool */\n\tid: string;\n\t/** Protocol type (MCP or A2A) */\n\tprotocol: TOOL_PROTOCOL_TYPE;\n\t/** Whether the tool is currently enabled for use */\n\tenabled: boolean;\n\t/** Enables the tool for use */\n\tenable: () => void;\n\t/** Disables the tool from being used */\n\tdisable: () => void;\n}\n"],"mappings":";AAOO,IAAK,qBAAL,kBAAKA,wBAAL;AAEN,EAAAA,oBAAA,SAAM;AAEN,EAAAA,oBAAA,SAAM;AAJK,SAAAA;AAAA,GAAA;","names":["TOOL_PROTOCOL_TYPE"]}

package/dist/esm/chunk-FL3JRQM4.js

@@ -0,0 +1,13 @@
+// src/types/agent.ts
+var AinHttpError = class extends Error {
+  status;
+  constructor(status, message) {
+    super(message);
+    this.status = status;
+  }
+};
+
+export {
+  AinHttpError
+};
+//# sourceMappingURL=chunk-FL3JRQM4.js.map

package/dist/esm/chunk-FL3JRQM4.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/types/agent.ts"],"sourcesContent":["import type { StatusCodes } from \"http-status-codes\";\n\n/**\n * Custom prompts configuration for the agent.\n *\n * Allows customization of agent behavior through system and agent-level prompts.\n */\nexport type AinAgentPrompts = {\n\t/** Agent-level prompt that defines the agent's personality and behavior */\n\tagent?: string;\n\t/** System-level prompt that provides context and instructions */\n\tsystem?: string;\n};\n\n/**\n * Agent manifest containing essential metadata and configuration.\n *\n * The manifest defines the agent's identity, version, and optional configuration\n * such as the public URL for A2A protocol support and custom prompts.\n *\n * @example\n * ```typescript\n * const manifest: AinAgentManifest = {\n *   name: \"CustomerSupportAgent\",\n *   description: \"AI agent for handling customer support queries\",\n *   version: \"1.0.0\",\n *   url: \"https://api.example.com/agent\",\n *   prompts: {\n *     system: \"You are a helpful customer support assistant.\",\n *     agent: \"Always be polite and provide detailed solutions.\"\n *   }\n * };\n * ```\n */\nexport type AinAgentManifest = {\n\t/** Unique name identifier for the agent */\n\tname: string;\n\t/** Human-readable description of the agent's purpose and capabilities */\n\tdescription: string;\n\t/** Semantic version of the agent (e.g., \"1.0.0\") */\n\tversion: string;\n\t/** Optional public URL for A2A protocol discovery and communication */\n\turl?: string;\n\t/** Optional custom prompts to configure agent behavior */\n\tprompts?: AinAgentPrompts;\n};\n\nexport class AinHttpError extends Error {\n\tpublic status?: StatusCodes;\n\n\tconstructor(status: StatusCodes, message: string) {\n\t\tsuper(message);\n\t\tthis.status = status;\n\t}\n}\n"],"mappings":";AA+CO,IAAM,eAAN,cAA2B,MAAM;AAAA,EAChC;AAAA,EAEP,YAAY,QAAqB,SAAiB;AACjD,UAAM,OAAO;AACb,SAAK,SAAS;AAAA,EACf;AACD;","names":[]}

package/dist/esm/chunk-GZFRRFOE.js

@@ -0,0 +1,8 @@
+// src/modules/models/base.model.ts
+var BaseModel = class {
+};
+
+export {
+  BaseModel
+};
+//# sourceMappingURL=chunk-GZFRRFOE.js.map

package/dist/esm/chunk-GZFRRFOE.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/modules/models/base.model.ts"],"sourcesContent":["import type { SessionObject } from \"@/types/memory.js\";\nimport type { FetchResponse, IAgentTool } from \"@/types/tool.js\";\n\n/**\n * Abstract base class for AI model implementations.\n *\n * Provides a common interface for different AI model providers (OpenAI, Gemini, etc.)\n * to integrate with the AIN-ADK framework. Each model implementation must handle\n * message formatting, tool conversion, and API communication.\n *\n * @typeParam MessageType - The message format used by the specific model API\n * @typeParam FunctionType - The function/tool format used by the specific model API\n */\nexport abstract class BaseModel<MessageType, FunctionType> {\n\t/**\n\t * Generates an array of messages from session history and current query.\n\t *\n\t * @param sessionHistory - Previous conversation history\n\t * @param query - Current user query\n\t * @param systemPrompt - Optional system prompt to set context\n\t * @returns Array of messages formatted for the specific model API\n\t */\n\tabstract generateMessages(params: {\n\t\tquery: string;\n\t\tsessionHistory?: SessionObject;\n\t\tsystemPrompt?: string;\n\t}): MessageType[];\n\n\t/**\n\t * Appends a new message to the existing message array.\n\t *\n\t * @param messages - Existing message array to expand\n\t * @param message - New message content to append\n\t */\n\tabstract appendMessages(messages: MessageType[], message: string): void;\n\n\t/**\n\t * Converts protocol-agnostic tools to model-specific function format.\n\t *\n\t * @param tools - Array of agent tools from MCP or A2A sources\n\t * @returns Array of functions in the format required by the model API\n\t */\n\tabstract convertToolsToFunctions(tools: IAgentTool[]): FunctionType[];\n\n\t/**\n\t * Fetches a response from the model API without tool support.\n\t *\n\t * @param messages - Array of messages to send to the model\n\t * @returns Promise resolving to the model's response\n\t */\n\tabstract fetch(messages: MessageType[]): Promise<FetchResponse>;\n\n\t/**\n\t * Fetches a response from the model API with tool/function support.\n\t *\n\t * @param messages - Array of messages to send to the model\n\t * @param functions - Array of available functions/tools the model can call\n\t * @returns Promise resolving to the model's response, possibly including tool calls\n\t */\n\tabstract fetchWithContextMessage(\n\t\tmessages: MessageType[],\n\t\tfunctions: FunctionType[],\n\t): Promise<FetchResponse>;\n}\n"],"mappings":";AAaO,IAAe,YAAf,MAAoD;AAkD3D;","names":[]}