@ainetwork/adk 0.2.0 → 0.2.1

package/dist/cjs/chunk-MLUUV7UY.cjs

@@ -0,0 +1,36 @@
+"use strict";Object.defineProperty(exports, "__esModule", {value: true});
+
+var _chunkG5KEV27Wcjs = require('./chunk-G5KEV27W.cjs');
+
+
+var _chunkDX2JH3FJcjs = require('./chunk-DX2JH3FJ.cjs');
+
+// src/routes/api/threads.routes.ts
+
+
+var _express = require('express');
+var _httpstatuscodes = require('http-status-codes');
+var createThreadApiRouter = (memoryModule) => {
+  const router = _express.Router.call(void 0, );
+  const threadApiController = new (0, _chunkG5KEV27Wcjs.ThreadApiController)(memoryModule);
+  const checkThreadMemory = (req, res, next) => {
+    const threadMemory = memoryModule.getThreadMemory();
+    if (!threadMemory) {
+      const error = new (0, _chunkDX2JH3FJcjs.AinHttpError)(
+        _httpstatuscodes.StatusCodes.SERVICE_UNAVAILABLE,
+        "Memory module is not initialized"
+      );
+      throw error;
+    }
+    next();
+  };
+  router.get("/", checkThreadMemory, threadApiController.handleGetUserThreads);
+  router.get("/:id", checkThreadMemory, threadApiController.handleGetThread);
+  router.delete("/:id", checkThreadMemory, threadApiController.handleGetThread);
+  return router;
+};
+
+
+
+exports.createThreadApiRouter = createThreadApiRouter;
+//# sourceMappingURL=chunk-MLUUV7UY.cjs.map

package/dist/cjs/chunk-MLUUV7UY.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-MLUUV7UY.cjs","../../src/routes/api/threads.routes.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACA;ACPA;AAIC;AAAA,kCACM;AACP,oDAA4B;AAKrB,IAAM,sBAAA,EAAwB,CAAC,YAAA,EAAA,GAAuC;AAC5E,EAAA,MAAM,OAAA,EAAS,6BAAA,CAAO;AACtB,EAAA,MAAM,oBAAA,EAAsB,IAAI,0CAAA,CAAoB,YAAY,CAAA;AAEhE,EAAA,MAAM,kBAAA,EAAoB,CACzB,GAAA,EACA,GAAA,EACA,IAAA,EAAA,GACI;AACJ,IAAA,MAAM,aAAA,EAAe,YAAA,CAAa,eAAA,CAAgB,CAAA;AAClD,IAAA,GAAA,CAAI,CAAC,YAAA,EAAc;AAClB,MAAA,MAAM,MAAA,EAAQ,IAAI,mCAAA;AAAA,QACjB,4BAAA,CAAY,mBAAA;AAAA,QACZ;AAAA,MACD,CAAA;AACA,MAAA,MAAM,KAAA;AAAA,IACP;AACA,IAAA,IAAA,CAAK,CAAA;AAAA,EACN,CAAA;AAGA,EAAA,MAAA,CAAO,GAAA,CAAI,GAAA,EAAK,iBAAA,EAAmB,mBAAA,CAAoB,oBAAoB,CAAA;AAC3E,EAAA,MAAA,CAAO,GAAA,CAAI,MAAA,EAAQ,iBAAA,EAAmB,mBAAA,CAAoB,eAAe,CAAA;AACzE,EAAA,MAAA,CAAO,MAAA,CAAO,MAAA,EAAQ,iBAAA,EAAmB,mBAAA,CAAoB,eAAe,CAAA;AAE5E,EAAA,OAAO,MAAA;AACR,CAAA;ADNA;AACA;AACE;AACF,sDAAC","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-MLUUV7UY.cjs","sourcesContent":[null,"import {\n\ttype NextFunction,\n\ttype Request,\n\ttype Response,\n\tRouter,\n} from \"express\";\nimport { StatusCodes } from \"http-status-codes\";\nimport { ThreadApiController } from \"@/controllers/api/threads.api.controller.js\";\nimport type { MemoryModule } from \"@/modules/index.js\";\nimport { AinHttpError } from \"@/types/agent\";\n\nexport const createThreadApiRouter = (memoryModule: MemoryModule): Router => {\n\tconst router = Router();\n\tconst threadApiController = new ThreadApiController(memoryModule);\n\n\tconst checkThreadMemory = (\n\t\treq: Request,\n\t\tres: Response,\n\t\tnext: NextFunction,\n\t) => {\n\t\tconst threadMemory = memoryModule.getThreadMemory();\n\t\tif (!threadMemory) {\n\t\t\tconst error = new AinHttpError(\n\t\t\t\tStatusCodes.SERVICE_UNAVAILABLE,\n\t\t\t\t\"Memory module is not initialized\",\n\t\t\t);\n\t\t\tthrow error;\n\t\t}\n\t\tnext();\n\t};\n\n\t// APIs (prefix: /api/threads)\n\trouter.get(\"/\", checkThreadMemory, threadApiController.handleGetUserThreads);\n\trouter.get(\"/:id\", checkThreadMemory, threadApiController.handleGetThread);\n\trouter.delete(\"/:id\", checkThreadMemory, threadApiController.handleGetThread);\n\n\treturn router;\n};\n"]}

package/dist/cjs/chunk-RX72NXXR.cjs

@@ -0,0 +1,43 @@
+"use strict";Object.defineProperty(exports, "__esModule", {value: true});
+
+var _chunkAIE3SVG7cjs = require('./chunk-AIE3SVG7.cjs');
+
+
+var _chunk3J3ORQOGcjs = require('./chunk-3J3ORQOG.cjs');
+
+
+var _chunkCU5MOHJUcjs = require('./chunk-CU5MOHJU.cjs');
+
+// src/routes/query.routes.ts
+var _express = require('express');
+var createQueryRouter = (agent, allowStream = false) => {
+  const router = _express.Router.call(void 0, );
+  const queryService = new (0, _chunkCU5MOHJUcjs.QueryService)(
+    agent.modelModule,
+    agent.a2aModule,
+    agent.mcpModule,
+    agent.memoryModule,
+    agent.manifest.prompts
+  );
+  let queryStreamService;
+  if (allowStream) {
+    queryStreamService = new (0, _chunkAIE3SVG7cjs.QueryStreamService)(
+      agent.modelModule,
+      agent.a2aModule,
+      agent.mcpModule,
+      agent.memoryModule,
+      agent.manifest.prompts
+    );
+  }
+  const queryController = new (0, _chunk3J3ORQOGcjs.QueryController)(queryService, queryStreamService);
+  router.post("/", queryController.handleQueryRequest);
+  if (allowStream) {
+    router.post("/stream", queryController.handleQueryStreamRequest);
+  }
+  return router;
+};
+
+
+
+exports.createQueryRouter = createQueryRouter;
+//# sourceMappingURL=chunk-RX72NXXR.cjs.map

package/dist/cjs/chunk-RX72NXXR.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-RX72NXXR.cjs","../../src/routes/query.routes.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACA;ACVA,kCAAuB;AAMhB,IAAM,kBAAA,EAAoB,CAChC,KAAA,EACA,YAAA,EAAc,KAAA,EAAA,GACF;AACZ,EAAA,MAAM,OAAA,EAAS,6BAAA,CAAO;AAEtB,EAAA,MAAM,aAAA,EAAe,IAAI,mCAAA;AAAA,IACxB,KAAA,CAAM,WAAA;AAAA,IACN,KAAA,CAAM,SAAA;AAAA,IACN,KAAA,CAAM,SAAA;AAAA,IACN,KAAA,CAAM,YAAA;AAAA,IACN,KAAA,CAAM,QAAA,CAAS;AAAA,EAChB,CAAA;AAEA,EAAA,IAAI,kBAAA;AAEJ,EAAA,GAAA,CAAI,WAAA,EAAa;AAChB,IAAA,mBAAA,EAAqB,IAAI,yCAAA;AAAA,MACxB,KAAA,CAAM,WAAA;AAAA,MACN,KAAA,CAAM,SAAA;AAAA,MACN,KAAA,CAAM,SAAA;AAAA,MACN,KAAA,CAAM,YAAA;AAAA,MACN,KAAA,CAAM,QAAA,CAAS;AAAA,IAChB,CAAA;AAAA,EACD;AAEA,EAAA,MAAM,gBAAA,EAAkB,IAAI,sCAAA,CAAgB,YAAA,EAAc,kBAAkB,CAAA;AAC5E,EAAA,MAAA,CAAO,IAAA,CAAK,GAAA,EAAK,eAAA,CAAgB,kBAAkB,CAAA;AAEnD,EAAA,GAAA,CAAI,WAAA,EAAa;AAChB,IAAA,MAAA,CAAO,IAAA,CAAK,SAAA,EAAW,eAAA,CAAgB,wBAAwB,CAAA;AAAA,EAChE;AAEA,EAAA,OAAO,MAAA;AACR,CAAA;ADFA;AACA;AACE;AACF,8CAAC","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-RX72NXXR.cjs","sourcesContent":[null,"import { Router } from \"express\";\nimport { QueryController } from \"@/controllers/query.controller.js\";\nimport type { AINAgent } from \"@/index.js\";\nimport { QueryService } from \"@/services/query.service.js\";\nimport { QueryStreamService } from \"@/services/query-stream.service.js\";\n\nexport const createQueryRouter = (\n\tagent: AINAgent,\n\tallowStream = false,\n): 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\n\tlet queryStreamService: QueryStreamService | undefined;\n\n\tif (allowStream) {\n\t\tqueryStreamService = new QueryStreamService(\n\t\t\tagent.modelModule,\n\t\t\tagent.a2aModule,\n\t\t\tagent.mcpModule,\n\t\t\tagent.memoryModule,\n\t\t\tagent.manifest.prompts,\n\t\t);\n\t}\n\n\tconst queryController = new QueryController(queryService, queryStreamService);\n\trouter.post(\"/\", queryController.handleQueryRequest);\n\n\tif (allowStream) {\n\t\trouter.post(\"/stream\", queryController.handleQueryStreamRequest);\n\t}\n\n\treturn router;\n};\n"]}

package/dist/cjs/chunk-T2AKW5PA.cjs

@@ -0,0 +1,45 @@
+"use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _optionalChain(ops) { let lastAccessLHS = undefined; let value = ops[0]; let i = 1; while (i < ops.length) { const op = ops[i]; const fn = ops[i + 1]; i += 2; if ((op === 'optionalAccess' || op === 'optionalCall') && value == null) { return undefined; } if (op === 'access' || op === 'optionalAccess') { lastAccessLHS = value; value = fn(value); } else if (op === 'call' || op === 'optionalCall') { value = fn((...args) => value.call(lastAccessLHS, ...args)); lastAccessLHS = undefined; } } return value; } var _class;// src/controllers/api/threads.api.controller.ts
+var _httpstatuscodes = require('http-status-codes');
+var ThreadApiController = (_class = class {
+  
+  constructor(memoryModule) {;_class.prototype.__init.call(this);_class.prototype.__init2.call(this);_class.prototype.__init3.call(this);
+    this.memoryModule = memoryModule;
+  }
+  __init() {this.handleGetThread = async (req, res, next) => {
+    try {
+      const { id: threadId, type } = req.params;
+      const userId = res.locals.userId || "";
+      const threadMemory = this.memoryModule.getThreadMemory();
+      const thread = await _optionalChain([threadMemory, 'optionalAccess', _7 => _7.getThread, 'call', _8 => _8(type, userId, threadId)]);
+      res.json(thread);
+    } catch (error) {
+      next(error);
+    }
+  }}
+  __init2() {this.handleDeleteThread = async (req, res, next) => {
+    try {
+      const { id: threadId } = req.params;
+      const userId = res.locals.userId || "";
+      const threadMemory = this.memoryModule.getThreadMemory();
+      await _optionalChain([threadMemory, 'optionalAccess', _9 => _9.deleteThread, 'call', _10 => _10(userId, threadId)]);
+      res.status(_httpstatuscodes.StatusCodes.OK).send();
+    } catch (error) {
+      next(error);
+    }
+  }}
+  __init3() {this.handleGetUserThreads = async (_req, res, next) => {
+    try {
+      const userId = res.locals.userId || "";
+      const threadMemory = this.memoryModule.getThreadMemory();
+      const threads = await _optionalChain([threadMemory, 'optionalAccess', _11 => _11.listThreads, 'call', _12 => _12(userId)]);
+      res.json(threads);
+    } catch (error) {
+      next(error);
+    }
+  }}
+}, _class);
+
+
+
+exports.ThreadApiController = ThreadApiController;
+//# sourceMappingURL=chunk-T2AKW5PA.cjs.map

package/dist/cjs/chunk-T2AKW5PA.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-T2AKW5PA.cjs","../../src/controllers/api/threads.api.controller.ts"],"names":[],"mappings":"AAAA;ACCA,oDAA4B;AAKrB,IAAM,oBAAA,YAAN,MAA0B;AAAA,EACxB;AAAA,EAER,WAAA,CAAY,YAAA,EAA4B;AACvC,IAAA,IAAA,CAAK,aAAA,EAAe,YAAA;AAAA,EACrB;AAAA,iBAEO,gBAAA,EAAkB,MAAA,CACxB,GAAA,EACA,GAAA,EACA,IAAA,EAAA,GACI;AACJ,IAAA,IAAI;AACH,MAAA,MAAM,EAAE,EAAA,EAAI,QAAA,EAAU,KAAK,EAAA,EAAI,GAAA,CAAI,MAAA;AAInC,MAAA,MAAM,OAAA,EAAS,GAAA,CAAI,MAAA,CAAO,OAAA,GAAU,EAAA;AACpC,MAAA,MAAM,aAAA,EAAe,IAAA,CAAK,YAAA,CAAa,eAAA,CAAgB,CAAA;AACvD,MAAA,MAAM,OAAA,EAAS,sBAAM,YAAA,6BAAc,SAAA,mBAAU,IAAA,EAAM,MAAA,EAAQ,QAAQ,GAAA;AACnE,MAAA,GAAA,CAAI,IAAA,CAAK,MAAM,CAAA;AAAA,IAChB,EAAA,MAAA,CAAS,KAAA,EAAO;AACf,MAAA,IAAA,CAAK,KAAK,CAAA;AAAA,IACX;AAAA,EACD,EAAA;AAAA,kBAEO,mBAAA,EAAqB,MAAA,CAC3B,GAAA,EACA,GAAA,EACA,IAAA,EAAA,GACI;AACJ,IAAA,IAAI;AACH,MAAA,MAAM,EAAE,EAAA,EAAI,SAAS,EAAA,EAAI,GAAA,CAAI,MAAA;AAC7B,MAAA,MAAM,OAAA,EAAS,GAAA,CAAI,MAAA,CAAO,OAAA,GAAU,EAAA;AACpC,MAAA,MAAM,aAAA,EAAe,IAAA,CAAK,YAAA,CAAa,eAAA,CAAgB,CAAA;AACvD,MAAA,sBAAM,YAAA,6BAAc,YAAA,qBAAa,MAAA,EAAQ,QAAQ,GAAA;AACjD,MAAA,GAAA,CAAI,MAAA,CAAO,4BAAA,CAAY,EAAE,CAAA,CAAE,IAAA,CAAK,CAAA;AAAA,IACjC,EAAA,MAAA,CAAS,KAAA,EAAO;AACf,MAAA,IAAA,CAAK,KAAK,CAAA;AAAA,IACX;AAAA,EACD,EAAA;AAAA,kBAEO,qBAAA,EAAuB,MAAA,CAC7B,IAAA,EACA,GAAA,EACA,IAAA,EAAA,GACI;AACJ,IAAA,IAAI;AACH,MAAA,MAAM,OAAA,EAAS,GAAA,CAAI,MAAA,CAAO,OAAA,GAAU,EAAA;AACpC,MAAA,MAAM,aAAA,EAAe,IAAA,CAAK,YAAA,CAAa,eAAA,CAAgB,CAAA;AACvD,MAAA,MAAM,QAAA,EAAU,sBAAM,YAAA,+BAAc,WAAA,qBAAY,MAAM,GAAA;AACtD,MAAA,GAAA,CAAI,IAAA,CAAK,OAAO,CAAA;AAAA,IACjB,EAAA,MAAA,CAAS,KAAA,EAAO;AACf,MAAA,IAAA,CAAK,KAAK,CAAA;AAAA,IACX;AAAA,EACD,EAAA;AACD,UAAA;ADtBA;AACA;AACE;AACF,kDAAC","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-T2AKW5PA.cjs","sourcesContent":[null,"import type { NextFunction, Request, Response } from \"express\";\nimport { StatusCodes } from \"http-status-codes\";\nimport type { MemoryModule } from \"@/modules/index.js\";\nimport { AinHttpError } from \"@/types/agent.js\";\nimport type { ThreadType } from \"@/types/memory\";\n\nexport class ThreadApiController {\n\tprivate memoryModule: MemoryModule;\n\n\tconstructor(memoryModule: MemoryModule) {\n\t\tthis.memoryModule = memoryModule;\n\t}\n\n\tpublic handleGetThread = async (\n\t\treq: Request,\n\t\tres: Response,\n\t\tnext: NextFunction,\n\t) => {\n\t\ttry {\n\t\t\tconst { id: threadId, type } = req.params as {\n\t\t\t\tid: string;\n\t\t\t\ttype: ThreadType;\n\t\t\t};\n\t\t\tconst userId = res.locals.userId || \"\";\n\t\t\tconst threadMemory = this.memoryModule.getThreadMemory();\n\t\t\tconst thread = await threadMemory?.getThread(type, userId, threadId);\n\t\t\tres.json(thread);\n\t\t} catch (error) {\n\t\t\tnext(error);\n\t\t}\n\t};\n\n\tpublic handleDeleteThread = async (\n\t\treq: Request,\n\t\tres: Response,\n\t\tnext: NextFunction,\n\t) => {\n\t\ttry {\n\t\t\tconst { id: threadId } = req.params;\n\t\t\tconst userId = res.locals.userId || \"\";\n\t\t\tconst threadMemory = this.memoryModule.getThreadMemory();\n\t\t\tawait threadMemory?.deleteThread(userId, threadId);\n\t\t\tres.status(StatusCodes.OK).send();\n\t\t} catch (error) {\n\t\t\tnext(error);\n\t\t}\n\t};\n\n\tpublic handleGetUserThreads = async (\n\t\t_req: Request,\n\t\tres: Response,\n\t\tnext: NextFunction,\n\t) => {\n\t\ttry {\n\t\t\tconst userId = res.locals.userId || \"\";\n\t\t\tconst threadMemory = this.memoryModule.getThreadMemory();\n\t\t\tconst threads = await threadMemory?.listThreads(userId);\n\t\t\tres.json(threads);\n\t\t} catch (error) {\n\t\t\tnext(error);\n\t\t}\n\t};\n}\n"]}

package/dist/cjs/chunk-UTFWH24D.cjs

@@ -0,0 +1,22 @@
+"use strict";Object.defineProperty(exports, "__esModule", {value: true});
+
+var _chunkX46GJS7Xcjs = require('./chunk-X46GJS7X.cjs');
+
+
+var _chunkMLUUV7UYcjs = require('./chunk-MLUUV7UY.cjs');
+
+// src/routes/api.routes.ts
+var _express = require('express');
+var createApiRouter = (agent) => {
+  const router = _express.Router.call(void 0, );
+  router.use("/api/model", _chunkX46GJS7Xcjs.createModelApiRouter.call(void 0, agent.modelModule));
+  if (agent.memoryModule) {
+    router.use("/api/threads", _chunkMLUUV7UYcjs.createThreadApiRouter.call(void 0, agent.memoryModule));
+  }
+  return router;
+};
+
+
+
+exports.createApiRouter = createApiRouter;
+//# sourceMappingURL=chunk-UTFWH24D.cjs.map

package/dist/cjs/chunk-UTFWH24D.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-UTFWH24D.cjs","../../src/routes/api.routes.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACA;ACPA,kCAAuB;AAMhB,IAAM,gBAAA,EAAkB,CAAC,KAAA,EAAA,GAA4B;AAC3D,EAAA,MAAM,OAAA,EAAS,6BAAA,CAAO;AAEtB,EAAA,MAAA,CAAO,GAAA,CAAI,YAAA,EAAc,oDAAA,KAAqB,CAAM,WAAW,CAAC,CAAA;AAChE,EAAA,GAAA,CAAI,KAAA,CAAM,YAAA,EAAc;AACvB,IAAA,MAAA,CAAO,GAAA,CAAI,cAAA,EAAgB,qDAAA,KAAsB,CAAM,YAAY,CAAC,CAAA;AAAA,EACrE;AAEA,EAAA,OAAO,MAAA;AACR,CAAA;ADEA;AACA;AACE;AACF,0CAAC","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-UTFWH24D.cjs","sourcesContent":[null,"import { Router } from \"express\";\nimport type { AINAgent } from \"@/index.js\";\n\nimport { createModelApiRouter } from \"./api/model.routes.js\";\nimport { createThreadApiRouter } from \"./api/threads.routes.js\";\n\nexport const createApiRouter = (agent: AINAgent): Router => {\n\tconst router = Router();\n\n\trouter.use(\"/api/model\", createModelApiRouter(agent.modelModule));\n\tif (agent.memoryModule) {\n\t\trouter.use(\"/api/threads\", createThreadApiRouter(agent.memoryModule));\n\t}\n\n\treturn router;\n};\n"]}

package/dist/cjs/chunk-W52YCYRF.cjs

@@ -0,0 +1,43 @@
+"use strict";Object.defineProperty(exports, "__esModule", {value: true});
+
+var _chunkZZPJ5CPRcjs = require('./chunk-ZZPJ5CPR.cjs');
+
+
+var _chunk3J3ORQOGcjs = require('./chunk-3J3ORQOG.cjs');
+
+
+var _chunk7GWYMGWWcjs = require('./chunk-7GWYMGWW.cjs');
+
+// src/routes/query.routes.ts
+var _express = require('express');
+var createQueryRouter = (agent, allowStream = false) => {
+  const router = _express.Router.call(void 0, );
+  const queryService = new (0, _chunk7GWYMGWWcjs.QueryService)(
+    agent.modelModule,
+    agent.a2aModule,
+    agent.mcpModule,
+    agent.memoryModule,
+    agent.manifest.prompts
+  );
+  let queryStreamService;
+  if (allowStream) {
+    queryStreamService = new (0, _chunkZZPJ5CPRcjs.QueryStreamService)(
+      agent.modelModule,
+      agent.a2aModule,
+      agent.mcpModule,
+      agent.memoryModule,
+      agent.manifest.prompts
+    );
+  }
+  const queryController = new (0, _chunk3J3ORQOGcjs.QueryController)(queryService, queryStreamService);
+  router.post("/", queryController.handleQueryRequest);
+  if (allowStream) {
+    router.post("/stream", queryController.handleQueryStreamRequest);
+  }
+  return router;
+};
+
+
+
+exports.createQueryRouter = createQueryRouter;
+//# sourceMappingURL=chunk-W52YCYRF.cjs.map

package/dist/cjs/chunk-W52YCYRF.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-W52YCYRF.cjs","../../src/routes/query.routes.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACA;ACVA,kCAAuB;AAMhB,IAAM,kBAAA,EAAoB,CAChC,KAAA,EACA,YAAA,EAAc,KAAA,EAAA,GACF;AACZ,EAAA,MAAM,OAAA,EAAS,6BAAA,CAAO;AAEtB,EAAA,MAAM,aAAA,EAAe,IAAI,mCAAA;AAAA,IACxB,KAAA,CAAM,WAAA;AAAA,IACN,KAAA,CAAM,SAAA;AAAA,IACN,KAAA,CAAM,SAAA;AAAA,IACN,KAAA,CAAM,YAAA;AAAA,IACN,KAAA,CAAM,QAAA,CAAS;AAAA,EAChB,CAAA;AAEA,EAAA,IAAI,kBAAA;AAEJ,EAAA,GAAA,CAAI,WAAA,EAAa;AAChB,IAAA,mBAAA,EAAqB,IAAI,yCAAA;AAAA,MACxB,KAAA,CAAM,WAAA;AAAA,MACN,KAAA,CAAM,SAAA;AAAA,MACN,KAAA,CAAM,SAAA;AAAA,MACN,KAAA,CAAM,YAAA;AAAA,MACN,KAAA,CAAM,QAAA,CAAS;AAAA,IAChB,CAAA;AAAA,EACD;AAEA,EAAA,MAAM,gBAAA,EAAkB,IAAI,sCAAA,CAAgB,YAAA,EAAc,kBAAkB,CAAA;AAC5E,EAAA,MAAA,CAAO,IAAA,CAAK,GAAA,EAAK,eAAA,CAAgB,kBAAkB,CAAA;AAEnD,EAAA,GAAA,CAAI,WAAA,EAAa;AAChB,IAAA,MAAA,CAAO,IAAA,CAAK,SAAA,EAAW,eAAA,CAAgB,wBAAwB,CAAA;AAAA,EAChE;AAEA,EAAA,OAAO,MAAA;AACR,CAAA;ADFA;AACA;AACE;AACF,8CAAC","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-W52YCYRF.cjs","sourcesContent":[null,"import { Router } from \"express\";\nimport { QueryController } from \"@/controllers/query.controller.js\";\nimport type { AINAgent } from \"@/index.js\";\nimport { QueryService } from \"@/services/query.service.js\";\nimport { QueryStreamService } from \"@/services/query-stream.service.js\";\n\nexport const createQueryRouter = (\n\tagent: AINAgent,\n\tallowStream = false,\n): 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\n\tlet queryStreamService: QueryStreamService | undefined;\n\n\tif (allowStream) {\n\t\tqueryStreamService = new QueryStreamService(\n\t\t\tagent.modelModule,\n\t\t\tagent.a2aModule,\n\t\t\tagent.mcpModule,\n\t\t\tagent.memoryModule,\n\t\t\tagent.manifest.prompts,\n\t\t);\n\t}\n\n\tconst queryController = new QueryController(queryService, queryStreamService);\n\trouter.post(\"/\", queryController.handleQueryRequest);\n\n\tif (allowStream) {\n\t\trouter.post(\"/stream\", queryController.handleQueryStreamRequest);\n\t}\n\n\treturn router;\n};\n"]}

package/dist/cjs/chunk-YNBMJ5QX.cjs

@@ -0,0 +1,129 @@
+"use strict";Object.defineProperty(exports, "__esModule", {value: true}); var _class;
+
+var _chunkE433ONENcjs = require('./chunk-E433ONEN.cjs');
+
+
+var _chunk466ED5WNcjs = require('./chunk-466ED5WN.cjs');
+
+// src/modules/mcp/mcp.module.ts
+var _indexjs = require('@modelcontextprotocol/sdk/client/index.js');
+var _ssejs = require('@modelcontextprotocol/sdk/client/sse.js');
+var _stdiojs = require('@modelcontextprotocol/sdk/client/stdio.js');
+var _streamableHttpjs = require('@modelcontextprotocol/sdk/client/streamableHttp.js');
+var _websocketjs = require('@modelcontextprotocol/sdk/client/websocket.js');
+var MCPModule = (_class = class {constructor() { _class.prototype.__init.call(this);_class.prototype.__init2.call(this);_class.prototype.__init3.call(this); }
+  __init() {this.mcpConfigs = /* @__PURE__ */ new Map()}
+  /** Map of MCP server names to their client instances */
+  __init2() {this.mcpClientMap = /* @__PURE__ */ new Map()}
+  /** Array of all discovered tools from connected MCP servers */
+  __init3() {this.tools = []}
+  addMCPServerConfig(configs) {
+    for (const [name, config] of Object.entries(configs)) {
+      this.mcpConfigs.set(name, config);
+    }
+  }
+  async connectToServers() {
+    for (const [name, config] of this.mcpConfigs.entries()) {
+      try {
+        const mcpClient = new (0, _indexjs.Client)({ name, version: "1.0.0" });
+        switch (config.type) {
+          case "stdio": {
+            const transport = new (0, _stdiojs.StdioClientTransport)(config.params);
+            await mcpClient.connect(transport);
+            break;
+          }
+          case "websocket": {
+            const transport = new (0, _websocketjs.WebSocketClientTransport)(config.url);
+            await mcpClient.connect(transport);
+            break;
+          }
+          case "sse": {
+            const transport = new (0, _ssejs.SSEClientTransport)(
+              config.url,
+              config.options
+            );
+            await mcpClient.connect(transport);
+            break;
+          }
+          case "streamableHttp": {
+            const transport = new (0, _streamableHttpjs.StreamableHTTPClientTransport)(
+              config.url,
+              config.options
+            );
+            await mcpClient.connect(transport);
+            break;
+          }
+          default:
+            _chunk466ED5WNcjs.loggers.mcp.error("Unsupported MCP config type");
+            break;
+        }
+        this.mcpClientMap.set(name, mcpClient);
+        const toolList = await mcpClient.listTools();
+        this.tools.push(
+          ...toolList.tools.map((tool) => {
+            return new (0, _chunkE433ONENcjs.MCPTool)(name, tool);
+          })
+        );
+        _chunk466ED5WNcjs.loggers.mcp.info("Connected to MCP server with tools:", {
+          tools: toolList.tools.map((tool) => tool.id)
+        });
+      } catch (error) {
+        _chunk466ED5WNcjs.loggers.mcp.error(`Failed to connect to MCP server ${name}`, { 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.mcpClientMap.get(serverName);
+    try {
+      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);
+      return toolResult;
+    } catch (error) {
+      _chunk466ED5WNcjs.loggers.mcp.error("Failed to call tool", { error });
+      const toolResult = `[Bot Called Tool ${toolName} with args ${JSON.stringify(_args)}]
+${typeof error === "string" ? error : JSON.stringify(error, null, 2)}`;
+      return toolResult;
+    }
+  }
+  /**
+   * Closes all MCP client connections.
+   *
+   * Should be called when shutting down the application to ensure
+   * all MCP connections are properly closed.
+   */
+  async cleanup() {
+    this.mcpClientMap.forEach((mcpClient) => {
+      mcpClient.close();
+    });
+  }
+}, _class);
+
+
+
+exports.MCPModule = MCPModule;
+//# sourceMappingURL=chunk-YNBMJ5QX.cjs.map

package/dist/cjs/chunk-YNBMJ5QX.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-YNBMJ5QX.cjs","../../src/modules/mcp/mcp.module.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACA;ACPA,oEAAuB;AACvB,gEAAmC;AACnC,oEAAqC;AACrC,sFAA8C;AAC9C,4EAAyC;AA0BlC,IAAM,UAAA,YAAN,MAAgB;AAAA,iBACd,WAAA,kBAAqC,IAAI,GAAA,CAAI,EAAA;AAAA;AAAA,kBAE7C,aAAA,kBAAoC,IAAI,GAAA,CAAI,EAAA;AAAA;AAAA,kBAE5C,MAAA,EAAmB,CAAC,EAAA;AAAA,EAE5B,kBAAA,CAAmB,OAAA,EAA8C;AAChE,IAAA,IAAA,CAAA,MAAW,CAAC,IAAA,EAAM,MAAM,EAAA,GAAK,MAAA,CAAO,OAAA,CAAQ,OAAO,CAAA,EAAG;AACrD,MAAA,IAAA,CAAK,UAAA,CAAW,GAAA,CAAI,IAAA,EAAM,MAAM,CAAA;AAAA,IACjC;AAAA,EACD;AAAA,EAEA,MAAM,gBAAA,CAAA,EAAkC;AACvC,IAAA,IAAA,CAAA,MAAW,CAAC,IAAA,EAAM,MAAM,EAAA,GAAK,IAAA,CAAK,UAAA,CAAW,OAAA,CAAQ,CAAA,EAAG;AACvD,MAAA,IAAI;AACH,QAAA,MAAM,UAAA,EAAY,IAAI,oBAAA,CAAO,EAAE,IAAA,EAAM,OAAA,EAAS,QAAQ,CAAC,CAAA;AACvD,QAAA,OAAA,CAAQ,MAAA,CAAO,IAAA,EAAM;AAAA,UACpB,KAAK,OAAA,EAAS;AACb,YAAA,MAAM,UAAA,EAAY,IAAI,kCAAA,CAAqB,MAAA,CAAO,MAAM,CAAA;AACxD,YAAA,MAAM,SAAA,CAAU,OAAA,CAAQ,SAAS,CAAA;AACjC,YAAA,KAAA;AAAA,UACD;AAAA,UACA,KAAK,WAAA,EAAa;AACjB,YAAA,MAAM,UAAA,EAAY,IAAI,0CAAA,CAAyB,MAAA,CAAO,GAAG,CAAA;AACzD,YAAA,MAAM,SAAA,CAAU,OAAA,CAAQ,SAAS,CAAA;AACjC,YAAA,KAAA;AAAA,UACD;AAAA,UACA,KAAK,KAAA,EAAO;AACX,YAAA,MAAM,UAAA,EAAY,IAAI,8BAAA;AAAA,cACrB,MAAA,CAAO,GAAA;AAAA,cACP,MAAA,CAAO;AAAA,YACR,CAAA;AACA,YAAA,MAAM,SAAA,CAAU,OAAA,CAAQ,SAAS,CAAA;AACjC,YAAA,KAAA;AAAA,UACD;AAAA,UACA,KAAK,gBAAA,EAAkB;AACtB,YAAA,MAAM,UAAA,EAAY,IAAI,oDAAA;AAAA,cACrB,MAAA,CAAO,GAAA;AAAA,cACP,MAAA,CAAO;AAAA,YACR,CAAA;AACA,YAAA,MAAM,SAAA,CAAU,OAAA,CAAQ,SAAS,CAAA;AACjC,YAAA,KAAA;AAAA,UACD;AAAA,UACA,OAAA;AAEC,YAAA,yBAAA,CAAQ,GAAA,CAAI,KAAA,CAAM,6BAA6B,CAAA;AAC/C,YAAA,KAAA;AAAA,QACF;AAEA,QAAA,IAAA,CAAK,YAAA,CAAa,GAAA,CAAI,IAAA,EAAM,SAAS,CAAA;AACrC,QAAA,MAAM,SAAA,EAAW,MAAM,SAAA,CAAU,SAAA,CAAU,CAAA;AAC3C,QAAA,IAAA,CAAK,KAAA,CAAM,IAAA;AAAA,UACV,GAAG,QAAA,CAAS,KAAA,CAAM,GAAA,CAAI,CAAC,IAAA,EAAA,GAAS;AAC/B,YAAA,OAAO,IAAI,8BAAA,CAAQ,IAAA,EAAM,IAAI,CAAA;AAAA,UAC9B,CAAC;AAAA,QACF,CAAA;AACA,QAAA,yBAAA,CAAQ,GAAA,CAAI,IAAA,CAAK,qCAAA,EAAuC;AAAA,UACvD,KAAA,EAAO,QAAA,CAAS,KAAA,CAAM,GAAA,CAAI,CAAC,IAAA,EAAA,GAAS,IAAA,CAAK,EAAE;AAAA,QAC5C,CAAC,CAAA;AAAA,MACF,EAAA,MAAA,CAAS,KAAA,EAAO;AACf,QAAA,yBAAA,CAAQ,GAAA,CAAI,KAAA,CAAM,CAAA,gCAAA,EAAmC,IAAI,CAAA,CAAA;AAC1D,MAAA;AACD,IAAA;AACD,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAOW,EAAA;AACE,IAAA;AACb,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAU2D,EAAA;AAC1B,IAAA;AACP,IAAA;AACmB,IAAA;AAExC,IAAA;AACO,MAAA;AACiD,QAAA;AAC3D,MAAA;AAEkC,MAAA;AAC3B,QAAA;AACK,QAAA;AACX,MAAA;AAE4B,MAAA;AACU;AAChC,MAAA;AACQ,IAAA;AACmC,MAAA;AACH,MAAA;AAAyG;AACjJ,MAAA;AACR,IAAA;AACD,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAQgB,EAAA;AACkC,IAAA;AAChC,MAAA;AAChB,IAAA;AACF,EAAA;AACD;ADzBkE;AACA;AACA;AACA","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-YNBMJ5QX.cjs","sourcesContent":[null,"import { Client } from \"@modelcontextprotocol/sdk/client/index.js\";\nimport { SSEClientTransport } from \"@modelcontextprotocol/sdk/client/sse.js\";\nimport { StdioClientTransport } from \"@modelcontextprotocol/sdk/client/stdio.js\";\nimport { StreamableHTTPClientTransport } from \"@modelcontextprotocol/sdk/client/streamableHttp.js\";\nimport { WebSocketClientTransport } from \"@modelcontextprotocol/sdk/client/websocket.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\tprivate mcpConfigs: Map<string, MCPConfig> = new Map();\n\t/** Map of MCP server names to their client instances */\n\tprivate mcpClientMap: Map<string, Client> = new Map();\n\t/** Array of all discovered tools from connected MCP servers */\n\tprivate tools: MCPTool[] = [];\n\n\taddMCPServerConfig(configs: { [name: string]: MCPConfig }): void {\n\t\tfor (const [name, config] of Object.entries(configs)) {\n\t\t\tthis.mcpConfigs.set(name, config);\n\t\t}\n\t}\n\n\tasync connectToServers(): Promise<void> {\n\t\tfor (const [name, config] of this.mcpConfigs.entries()) {\n\t\t\ttry {\n\t\t\t\tconst mcpClient = new Client({ name, version: \"1.0.0\" });\n\t\t\t\tswitch (config.type) {\n\t\t\t\t\tcase \"stdio\": {\n\t\t\t\t\t\tconst transport = new StdioClientTransport(config.params);\n\t\t\t\t\t\tawait mcpClient.connect(transport);\n\t\t\t\t\t\tbreak;\n\t\t\t\t\t}\n\t\t\t\t\tcase \"websocket\": {\n\t\t\t\t\t\tconst transport = new WebSocketClientTransport(config.url);\n\t\t\t\t\t\tawait mcpClient.connect(transport);\n\t\t\t\t\t\tbreak;\n\t\t\t\t\t}\n\t\t\t\t\tcase \"sse\": {\n\t\t\t\t\t\tconst transport = new SSEClientTransport(\n\t\t\t\t\t\t\tconfig.url,\n\t\t\t\t\t\t\tconfig.options,\n\t\t\t\t\t\t);\n\t\t\t\t\t\tawait mcpClient.connect(transport);\n\t\t\t\t\t\tbreak;\n\t\t\t\t\t}\n\t\t\t\t\tcase \"streamableHttp\": {\n\t\t\t\t\t\tconst transport = new StreamableHTTPClientTransport(\n\t\t\t\t\t\t\tconfig.url,\n\t\t\t\t\t\t\tconfig.options,\n\t\t\t\t\t\t);\n\t\t\t\t\t\tawait mcpClient.connect(transport);\n\t\t\t\t\t\tbreak;\n\t\t\t\t\t}\n\t\t\t\t\tdefault:\n\t\t\t\t\t\t// It cannot be happened.\n\t\t\t\t\t\tloggers.mcp.error(\"Unsupported MCP config type\");\n\t\t\t\t\t\tbreak;\n\t\t\t\t}\n\n\t\t\t\tthis.mcpClientMap.set(name, mcpClient);\n\t\t\t\tconst toolList = await mcpClient.listTools();\n\t\t\t\tthis.tools.push(\n\t\t\t\t\t...toolList.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\tloggers.mcp.info(\"Connected to MCP server with tools:\", {\n\t\t\t\t\ttools: toolList.tools.map((tool) => tool.id),\n\t\t\t\t});\n\t\t\t} catch (error) {\n\t\t\t\tloggers.mcp.error(`Failed to connect to MCP server ${name}`, { error });\n\t\t\t}\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<string> {\n\t\tconst { serverName, mcpTool } = tool;\n\t\tconst toolName = mcpTool.name;\n\t\tconst mcp = this.mcpClientMap.get(serverName);\n\n\t\ttry {\n\t\t\tif (!mcp) {\n\t\t\t\tthrow new Error(`Invalid MCP Tool ${serverName}-${mcpTool.name}`);\n\t\t\t}\n\n\t\t\tconst result = await mcp.callTool({\n\t\t\t\tname: toolName,\n\t\t\t\targuments: _args,\n\t\t\t});\n\t\t\tconst toolResult =\n\t\t\t\t`[Bot Called Tool ${toolName} with args ${JSON.stringify(_args)}]\\n` +\n\t\t\t\tJSON.stringify(result.content, null, 2);\n\t\t\treturn toolResult;\n\t\t} catch (error) {\n\t\t\tloggers.mcp.error(\"Failed to call tool\", { error });\n\t\t\tconst toolResult = `[Bot Called Tool ${toolName} with args ${JSON.stringify(_args)}]\\n${typeof error === \"string\" ? error : JSON.stringify(error, null, 2)}`;\n\t\t\treturn toolResult;\n\t\t}\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.mcpClientMap.forEach((mcpClient: Client) => {\n\t\t\tmcpClient.close();\n\t\t});\n\t}\n}\n"]}