@ainetwork/adk 0.1.2 → 0.1.3

package/dist/cjs/routes/api/sessions.routes.cjs

@@ -1,9 +1,9 @@
 "use strict";Object.defineProperty(exports, "__esModule", {value: true});
 
-var _chunkD525HP3Rcjs = require('../../chunk-D525HP3R.cjs');
-require('../../chunk-YRNIZTPF.cjs');
+var _chunk2DKKHZA5cjs = require('../../chunk-2DKKHZA5.cjs');
+require('../../chunk-E4FRJCLU.cjs');
 require('../../chunk-DX2JH3FJ.cjs');
 
 
-exports.createSessionApiRouter = _chunkD525HP3Rcjs.createSessionApiRouter;
+exports.createSessionApiRouter = _chunk2DKKHZA5cjs.createSessionApiRouter;
 //# sourceMappingURL=sessions.routes.cjs.map

package/dist/cjs/routes/api.routes.cjs

@@ -1,12 +1,12 @@
 "use strict";Object.defineProperty(exports, "__esModule", {value: true});
 
-var _chunk6YJFC7BYcjs = require('../chunk-6YJFC7BY.cjs');
+var _chunkD3ST34H3cjs = require('../chunk-D3ST34H3.cjs');
 require('../chunk-N7NGEMKL.cjs');
-require('../chunk-D525HP3R.cjs');
+require('../chunk-2DKKHZA5.cjs');
 require('../chunk-V5RSN2DY.cjs');
-require('../chunk-YRNIZTPF.cjs');
+require('../chunk-E4FRJCLU.cjs');
 require('../chunk-DX2JH3FJ.cjs');
 
 
-exports.createApiRouter = _chunk6YJFC7BYcjs.createApiRouter;
+exports.createApiRouter = _chunkD3ST34H3cjs.createApiRouter;
 //# sourceMappingURL=api.routes.cjs.map

package/dist/cjs/routes/index.cjs

@@ -1,34 +1,34 @@
 "use strict";Object.defineProperty(exports, "__esModule", {value: true});require('../chunk-SSSDO7PA.cjs');
 
 
-var _chunkCR2I7S4Ucjs = require('../chunk-CR2I7S4U.cjs');
+var _chunkFFTEHDIHcjs = require('../chunk-FFTEHDIH.cjs');
 
 
-var _chunk6YJFC7BYcjs = require('../chunk-6YJFC7BY.cjs');
+var _chunkD3ST34H3cjs = require('../chunk-D3ST34H3.cjs');
 
 
 var _chunkN7NGEMKLcjs = require('../chunk-N7NGEMKL.cjs');
 
 
-var _chunkD525HP3Rcjs = require('../chunk-D525HP3R.cjs');
+var _chunk2DKKHZA5cjs = require('../chunk-2DKKHZA5.cjs');
 
 
-var _chunk7UL6NAT6cjs = require('../chunk-7UL6NAT6.cjs');
-require('../chunk-ZMYG5HDT.cjs');
-require('../chunk-7BZO57TG.cjs');
+var _chunkLUBMRUQXcjs = require('../chunk-LUBMRUQX.cjs');
 require('../chunk-V5RSN2DY.cjs');
-require('../chunk-YRNIZTPF.cjs');
+require('../chunk-E4FRJCLU.cjs');
 require('../chunk-ILGIEJCX.cjs');
 require('../chunk-DOFTYGJR.cjs');
-require('../chunk-LISZOU2X.cjs');
+require('../chunk-DX2JH3FJ.cjs');
+require('../chunk-LGFVKRGI.cjs');
+require('../chunk-3IGIYVGA.cjs');
 require('../chunk-LPXYZH22.cjs');
+require('../chunk-7BZO57TG.cjs');
 require('../chunk-JTAXZJQ2.cjs');
-require('../chunk-DX2JH3FJ.cjs');
 
 
 
 
 
 
-exports.createA2ARouter = _chunkCR2I7S4Ucjs.createA2ARouter; exports.createApiRouter = _chunk6YJFC7BYcjs.createApiRouter; exports.createModelApiRouter = _chunkN7NGEMKLcjs.createModelApiRouter; exports.createQueryRouter = _chunk7UL6NAT6cjs.createQueryRouter; exports.createSessionApiRouter = _chunkD525HP3Rcjs.createSessionApiRouter;
+exports.createA2ARouter = _chunkFFTEHDIHcjs.createA2ARouter; exports.createApiRouter = _chunkD3ST34H3cjs.createApiRouter; exports.createModelApiRouter = _chunkN7NGEMKLcjs.createModelApiRouter; exports.createQueryRouter = _chunkLUBMRUQXcjs.createQueryRouter; exports.createSessionApiRouter = _chunk2DKKHZA5cjs.createSessionApiRouter;
 //# sourceMappingURL=index.cjs.map

package/dist/cjs/routes/query.routes.cjs

@@ -1,12 +1,12 @@
 "use strict";Object.defineProperty(exports, "__esModule", {value: true});
 
-var _chunk7UL6NAT6cjs = require('../chunk-7UL6NAT6.cjs');
-require('../chunk-ZMYG5HDT.cjs');
+var _chunkLUBMRUQXcjs = require('../chunk-LUBMRUQX.cjs');
 require('../chunk-DOFTYGJR.cjs');
-require('../chunk-LISZOU2X.cjs');
+require('../chunk-LGFVKRGI.cjs');
+require('../chunk-3IGIYVGA.cjs');
 require('../chunk-LPXYZH22.cjs');
 require('../chunk-JTAXZJQ2.cjs');
 
 
-exports.createQueryRouter = _chunk7UL6NAT6cjs.createQueryRouter;
+exports.createQueryRouter = _chunkLUBMRUQXcjs.createQueryRouter;
 //# sourceMappingURL=query.routes.cjs.map

package/dist/cjs/services/index.cjs

@@ -1,14 +1,14 @@
 "use strict";Object.defineProperty(exports, "__esModule", {value: true});
 
-var _chunkZMYG5HDTcjs = require('../chunk-ZMYG5HDT.cjs');
+var _chunkLGFVKRGIcjs = require('../chunk-LGFVKRGI.cjs');
+require('../chunk-3IGIYVGA.cjs');
+require('../chunk-LPXYZH22.cjs');
 
 
 var _chunk7BZO57TGcjs = require('../chunk-7BZO57TG.cjs');
-require('../chunk-LISZOU2X.cjs');
-require('../chunk-LPXYZH22.cjs');
 require('../chunk-JTAXZJQ2.cjs');
 
 
 
-exports.A2AService = _chunk7BZO57TGcjs.A2AService; exports.QueryService = _chunkZMYG5HDTcjs.QueryService;
+exports.A2AService = _chunk7BZO57TGcjs.A2AService; exports.QueryService = _chunkLGFVKRGIcjs.QueryService;
 //# sourceMappingURL=index.cjs.map

package/dist/cjs/services/index.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/services/index.cjs"],"names":[],"mappings":"AAAA;AACE;AACF,yDAA8B;AAC9B;AACE;AACF,yDAA8B;AAC9B,iCAA8B;AAC9B,iCAA8B;AAC9B,iCAA8B;AAC9B;AACE;AACA;AACF,yGAAC","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/services/index.cjs"}
+{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/services/index.cjs"],"names":[],"mappings":"AAAA;AACE;AACF,yDAA8B;AAC9B,iCAA8B;AAC9B,iCAA8B;AAC9B;AACE;AACF,yDAA8B;AAC9B,iCAA8B;AAC9B;AACE;AACA;AACF,yGAAC","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/services/index.cjs"}

package/dist/cjs/services/query.service.cjs

@@ -1,10 +1,10 @@
 "use strict";Object.defineProperty(exports, "__esModule", {value: true});
 
-var _chunkZMYG5HDTcjs = require('../chunk-ZMYG5HDT.cjs');
-require('../chunk-LISZOU2X.cjs');
+var _chunkLGFVKRGIcjs = require('../chunk-LGFVKRGI.cjs');
+require('../chunk-3IGIYVGA.cjs');
 require('../chunk-LPXYZH22.cjs');
 require('../chunk-JTAXZJQ2.cjs');
 
 
-exports.QueryService = _chunkZMYG5HDTcjs.QueryService;
+exports.QueryService = _chunkLGFVKRGIcjs.QueryService;
 //# sourceMappingURL=query.service.cjs.map

package/dist/cjs/types/memory.cjs

@@ -1,9 +1,9 @@
 "use strict";Object.defineProperty(exports, "__esModule", {value: true});
 
 
-var _chunkLISZOU2Xcjs = require('../chunk-LISZOU2X.cjs');
+var _chunk3IGIYVGAcjs = require('../chunk-3IGIYVGA.cjs');
 
 
 
-exports.ChatRole = _chunkLISZOU2Xcjs.ChatRole; exports.MemoryType = _chunkLISZOU2Xcjs.MemoryType;
+exports.ChatRole = _chunk3IGIYVGAcjs.ChatRole; exports.MemoryType = _chunk3IGIYVGAcjs.MemoryType;
 //# sourceMappingURL=memory.cjs.map

package/dist/esm/{chunk-CH4XHVJY.js → chunk-423RKLBO.js}

@@ -1,12 +1,12 @@
-import {
-  A2AService
-} from "./chunk-AQ5YZERE.js";
 import {
   QueryService
-} from "./chunk-WZBXLGQK.js";
+} from "./chunk-LVKAW3WE.js";
 import {
   A2AController
 } from "./chunk-EODMRV4H.js";
+import {
+  A2AService
+} from "./chunk-AQ5YZERE.js";
 
 // src/routes/a2a.routes.ts
 import { InMemoryTaskStore } from "@a2a-js/sdk/server";
@@ -34,4 +34,4 @@ var createA2ARouter = (agent) => {
 export {
   createA2ARouter
 };
-//# sourceMappingURL=chunk-CH4XHVJY.js.map
+//# sourceMappingURL=chunk-423RKLBO.js.map

package/dist/esm/{chunk-CQ6GXYN2.js → chunk-6SDU6Z5Q.js}

@@ -1,9 +1,9 @@
+import {
+  QueryService
+} from "./chunk-LVKAW3WE.js";
 import {
   QueryController
 } from "./chunk-XOMQNCCR.js";
-import {
-  QueryService
-} from "./chunk-WZBXLGQK.js";
 
 // src/routes/query.routes.ts
 import { Router } from "express";
@@ -24,4 +24,4 @@ var createQueryRouter = (agent) => {
 export {
   createQueryRouter
 };
-//# sourceMappingURL=chunk-CQ6GXYN2.js.map
+//# sourceMappingURL=chunk-6SDU6Z5Q.js.map

package/dist/esm/{chunk-TMHRCGTE.js → chunk-F727XG4N.js}

@@ -1,8 +1,8 @@
 // src/types/memory.ts
 var MemoryType = /* @__PURE__ */ ((MemoryType2) => {
-  MemoryType2["_DEFAULT"] = "_DEFAULT";
   MemoryType2["SESSION"] = "SESSION";
   MemoryType2["INTENT"] = "INTENT";
+  MemoryType2["AGENT"] = "AGENT";
   MemoryType2["USER"] = "USER";
   return MemoryType2;
 })(MemoryType || {});
@@ -17,4 +17,4 @@ export {
   MemoryType,
   ChatRole
 };
-//# sourceMappingURL=chunk-TMHRCGTE.js.map
+//# sourceMappingURL=chunk-F727XG4N.js.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/types/memory.ts"],"sourcesContent":["export enum MemoryType {\n\tSESSION = \"SESSION\",\n\tINTENT = \"INTENT\",\n\tAGENT = \"AGENT\",\n\tUSER = \"USER\",\n}\n\n/**\n * Roles for participants in a chat conversation.\n */\nexport enum ChatRole {\n\t/** User/human participant */\n\tUSER = \"USER\",\n\t/** System-generated messages or instructions */\n\tSYSTEM = \"SYSTEM\",\n\t/** AI model responses */\n\tMODEL = \"MODEL\",\n}\n\n/**\n * Content structure for chat messages.\n *\n * Supports multi-part content with different types (text, images, etc.).\n */\nexport type ChatContentObject = {\n\t/** Content type (e.g., \"text\", \"image\", \"tool_use\") */\n\ttype: string;\n\t/** Array of content parts, structure depends on content type */\n\tparts: any[];\n};\n\n/**\n * Represents a single message in a chat conversation.\n *\n * @example\n * ```typescript\n * const message: ChatObject = {\n *   role: ChatRole.USER,\n *   content: {\n *     type: \"text\",\n *     parts: [\"Hello, how can you help me?\"]\n *   },\n *   timestamp: Date.now(),\n *   metadata: { source: \"web-ui\" }\n * };\n * ```\n */\nexport type ChatObject = {\n\t/** Role of the message sender */\n\trole: ChatRole;\n\t/** Message content with type and parts */\n\tcontent: ChatContentObject;\n\t/** Unix timestamp when the message was created */\n\ttimestamp: number;\n\t/** Optional metadata for additional context */\n\tmetadata?: { [key: string]: unknown };\n};\n\n/**\n * Represents a conversation session containing multiple chat messages.\n *\n * Messages are stored in a key-value structure where keys are unique chat IDs\n * and values are the corresponding chat objects.\n *\n * @example\n * ```typescript\n * const session: SessionObject = {\n *   chats: {\n *     \"<UUID_1>\": { role: ChatRole.USER, content: {...}, timestamp: 1234567890 },\n *     \"<UUID_2>\": { role: ChatRole.MODEL, content: {...}, timestamp: 1234567891 }\n *   }\n * };\n * ```\n */\nexport type SessionObject = {\n\t/** Collection of chat messages indexed by unique chat ID */\n\tchats: {\n\t\t[chatId: string]: ChatObject;\n\t};\n};\n\nexport interface Intent {\n\tname: string;\n\tdescription: string;\n}\n"],"mappings":";AAAO,IAAK,aAAL,kBAAKA,gBAAL;AACN,EAAAA,YAAA,aAAU;AACV,EAAAA,YAAA,YAAS;AACT,EAAAA,YAAA,WAAQ;AACR,EAAAA,YAAA,UAAO;AAJI,SAAAA;AAAA,GAAA;AAUL,IAAK,WAAL,kBAAKC,cAAL;AAEN,EAAAA,UAAA,UAAO;AAEP,EAAAA,UAAA,YAAS;AAET,EAAAA,UAAA,WAAQ;AANG,SAAAA;AAAA,GAAA;","names":["MemoryType","ChatRole"]}

package/dist/esm/{chunk-X47DAWPH.js → chunk-IXXK7ZZM.js}

@@ -12,15 +12,15 @@ var SessionApiController = class {
   handleSessionHistory = async (req, res, next) => {
     try {
       const { id: sessionId } = req.params;
-      const memoryInstance = this.memoryModule.getMemory();
-      if (!memoryInstance) {
+      const sessionMemory = this.memoryModule.getSessionMemory();
+      if (!sessionMemory) {
         const error = new AinHttpError(
           StatusCodes.SERVICE_UNAVAILABLE,
           "Memory module is not initialized"
         );
         throw error;
       }
-      const session = await memoryInstance.getSessionHistory(sessionId);
+      const session = await sessionMemory.getSession(sessionId);
       res.json(session);
     } catch (error) {
       next(error);
@@ -31,4 +31,4 @@ var SessionApiController = class {
 export {
   SessionApiController
 };
-//# sourceMappingURL=chunk-X47DAWPH.js.map
+//# sourceMappingURL=chunk-IXXK7ZZM.js.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/controllers/api/session.api.controller.ts"],"sourcesContent":["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\";\nexport class SessionApiController {\n\tprivate memoryModule: MemoryModule;\n\n\tconstructor(memoryModule: MemoryModule) {\n\t\tthis.memoryModule = memoryModule;\n\t}\n\n\tpublic handleSessionHistory = async (\n\t\treq: Request,\n\t\tres: Response,\n\t\tnext: NextFunction,\n\t) => {\n\t\ttry {\n\t\t\tconst { id: sessionId } = req.params;\n\t\t\tconst sessionMemory = this.memoryModule.getSessionMemory();\n\t\t\tif (!sessionMemory) {\n\t\t\t\tconst error = new AinHttpError(\n\t\t\t\t\tStatusCodes.SERVICE_UNAVAILABLE,\n\t\t\t\t\t\"Memory module is not initialized\",\n\t\t\t\t);\n\t\t\t\tthrow error;\n\t\t\t}\n\t\t\tconst session = await sessionMemory.getSession(sessionId);\n\t\t\tres.json(session);\n\t\t} catch (error) {\n\t\t\tnext(error);\n\t\t}\n\t};\n}\n"],"mappings":";;;;;AACA,SAAS,mBAAmB;AAGrB,IAAM,uBAAN,MAA2B;AAAA,EACzB;AAAA,EAER,YAAY,cAA4B;AACvC,SAAK,eAAe;AAAA,EACrB;AAAA,EAEO,uBAAuB,OAC7B,KACA,KACA,SACI;AACJ,QAAI;AACH,YAAM,EAAE,IAAI,UAAU,IAAI,IAAI;AAC9B,YAAM,gBAAgB,KAAK,aAAa,iBAAiB;AACzD,UAAI,CAAC,eAAe;AACnB,cAAM,QAAQ,IAAI;AAAA,UACjB,YAAY;AAAA,UACZ;AAAA,QACD;AACA,cAAM;AAAA,MACP;AACA,YAAM,UAAU,MAAM,cAAc,WAAW,SAAS;AACxD,UAAI,KAAK,OAAO;AAAA,IACjB,SAAS,OAAO;AACf,WAAK,KAAK;AAAA,IACX;AAAA,EACD;AACD;","names":[]}

package/dist/esm/{chunk-UZQZXNND.js → chunk-JLH6DGBK.js}

@@ -3,7 +3,7 @@ import {
 } from "./chunk-2Z7GJVJJ.js";
 import {
   createSessionApiRouter
-} from "./chunk-M4HGSCXF.js";
+} from "./chunk-KR6W7QIW.js";
 
 // src/routes/api.routes.ts
 import { Router } from "express";
@@ -19,4 +19,4 @@ var createApiRouter = (agent) => {
 export {
   createApiRouter
 };
-//# sourceMappingURL=chunk-UZQZXNND.js.map
+//# sourceMappingURL=chunk-JLH6DGBK.js.map

package/dist/esm/{chunk-M4HGSCXF.js → chunk-KR6W7QIW.js}

@@ -1,6 +1,6 @@
 import {
   SessionApiController
-} from "./chunk-X47DAWPH.js";
+} from "./chunk-IXXK7ZZM.js";
 
 // src/routes/api/sessions.routes.ts
 import { Router } from "express";
@@ -14,4 +14,4 @@ var createSessionApiRouter = (memoryModule) => {
 export {
   createSessionApiRouter
 };
-//# sourceMappingURL=chunk-M4HGSCXF.js.map
+//# sourceMappingURL=chunk-KR6W7QIW.js.map

package/dist/esm/{chunk-WZBXLGQK.js → chunk-LVKAW3WE.js}

@@ -139,10 +139,8 @@ ${result.join("\n")}`;
    */
   async handleQuery(query, sessionId) {
     const queryStartAt = Date.now();
-    const memoryInstance = this.memoryModule?.getMemory();
-    const sessionHistory = await memoryInstance?.getSessionHistory(
-      sessionId
-    ) || { chats: {} };
+    const sessionMemory = this.memoryModule?.getSessionMemory();
+    const sessionHistory = await sessionMemory?.getSession(sessionId);
     const intent = this.intentTriggering(query);
     const result = await this.intentFulfilling(
       query,
@@ -150,12 +148,12 @@ ${result.join("\n")}`;
       sessionHistory
     );
     if (sessionId) {
-      await memoryInstance?.updateSessionHistory(sessionId, {
+      await sessionMemory?.addChatToSession(sessionId, {
         role: "USER" /* USER */,
         timestamp: queryStartAt,
         content: { type: "text", parts: [query] }
       });
-      await memoryInstance?.updateSessionHistory(sessionId, {
+      await sessionMemory?.addChatToSession(sessionId, {
         role: "MODEL" /* MODEL */,
         timestamp: Date.now(),
         content: { type: "text", parts: [result.response] }
@@ -168,4 +166,4 @@ ${result.join("\n")}`;
 export {
   QueryService
 };
-//# sourceMappingURL=chunk-WZBXLGQK.js.map
+//# sourceMappingURL=chunk-LVKAW3WE.js.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/services/query.service.ts"],"sourcesContent":["import type {\n\tA2AModule,\n\tMCPModule,\n\tMemoryModule,\n\tModelModule,\n} from \"@/modules/index.js\";\nimport type { AinAgentPrompts } from \"@/types/agent.js\";\nimport { ChatRole, type SessionObject } from \"@/types/memory.js\";\nimport {\n\ttype IA2ATool,\n\ttype IAgentTool,\n\ttype IMCPTool,\n\tTOOL_PROTOCOL_TYPE,\n} from \"@/types/tool.js\";\nimport { loggers } from \"@/utils/logger.js\";\n\n/**\n * Service for processing user queries through the agent's AI pipeline.\n *\n * Orchestrates the query processing workflow including intent detection,\n * model inference, tool execution, and response generation. Manages\n * conversation context and coordinates between different modules.\n */\nexport class QueryService {\n\tprivate modelModule: ModelModule;\n\tprivate a2aModule?: A2AModule;\n\tprivate mcpModule?: MCPModule;\n\tprivate memoryModule?: MemoryModule;\n\tprivate prompts?: AinAgentPrompts;\n\n\tconstructor(\n\t\tmodelModule: ModelModule,\n\t\ta2aModule?: A2AModule,\n\t\tmcpModule?: MCPModule,\n\t\tmemoryModule?: MemoryModule,\n\t\tprompts?: AinAgentPrompts,\n\t) {\n\t\tthis.modelModule = modelModule;\n\t\tthis.a2aModule = a2aModule;\n\t\tthis.mcpModule = mcpModule;\n\t\tthis.memoryModule = memoryModule;\n\t\tthis.prompts = prompts;\n\t}\n\n\t/**\n\t * Detects the intent from a user query.\n\t *\n\t * @param query - The user's input query\n\t * @returns The detected intent (currently returns the query as-is)\n\t * @todo Implement actual intent detection logic\n\t */\n\tprivate async intentTriggering(query: string) {\n\t\t/* TODO */\n\t\treturn query;\n\t}\n\n\t/**\n\t * Fulfills the detected intent by generating a response.\n\t *\n\t * Manages the complete inference loop including:\n\t * - Loading prompts and conversation history\n\t * - Collecting available tools from modules\n\t * - Executing model inference with tool support\n\t * - Processing tool calls iteratively until completion\n\t *\n\t * @param query - The user's input query\n\t * @param sessionId - Session identifier for context\n\t * @param sessionHistory - Previous conversation history\n\t * @returns Object containing process steps and final response\n\t */\n\tprivate async intentFulfilling(\n\t\tquery: string,\n\t\tsessionId: string,\n\t\tsessionHistory: SessionObject | undefined,\n\t) {\n\t\t// 1. Load agent / system prompt from memory\n\t\tconst systemPrompt = `\nToday is ${new Date().toLocaleDateString()}.\n\n${this.prompts?.agent || \"\"}\n\n${this.prompts?.system || \"\"}\n    `;\n\n\t\tconst modelInstance = this.modelModule.getModel();\n\t\tconst messages = modelInstance.generateMessages({\n\t\t\tquery,\n\t\t\tsessionHistory,\n\t\t\tsystemPrompt: systemPrompt.trim(),\n\t\t});\n\n\t\tconst tools: IAgentTool[] = [];\n\t\tif (this.mcpModule) {\n\t\t\ttools.push(...this.mcpModule.getTools());\n\t\t}\n\t\tif (this.a2aModule) {\n\t\t\ttools.push(...(await this.a2aModule.getTools()));\n\t\t}\n\t\tconst functions = modelInstance.convertToolsToFunctions(tools);\n\n\t\tconst processList: string[] = [];\n\t\tlet finalMessage = \"\";\n\t\tlet didCallTool = false;\n\n\t\twhile (true) {\n\t\t\tconst response = await modelInstance.fetchWithContextMessage(\n\t\t\t\tmessages,\n\t\t\t\tfunctions,\n\t\t\t);\n\t\t\tdidCallTool = false;\n\n\t\t\tloggers.intent.debug(\"messages\", { messages });\n\n\t\t\tconst { content, toolCalls } = response;\n\n\t\t\tloggers.intent.debug(\"content\", { content });\n\t\t\tloggers.intent.debug(\"tool_calls\", { ...toolCalls });\n\n\t\t\tif (toolCalls) {\n\t\t\t\tconst messagePayload = this.a2aModule?.getMessagePayload(\n\t\t\t\t\tquery,\n\t\t\t\t\tsessionId,\n\t\t\t\t);\n\n\t\t\t\tfor (const toolCall of toolCalls) {\n\t\t\t\t\tconst toolName = toolCall.name;\n\t\t\t\t\tdidCallTool = true;\n\t\t\t\t\tconst selectedTool = tools.filter((tool) => tool.id === toolName)[0];\n\n\t\t\t\t\tlet toolResult = \"\";\n\t\t\t\t\tif (\n\t\t\t\t\t\tthis.mcpModule &&\n\t\t\t\t\t\tselectedTool.protocol === TOOL_PROTOCOL_TYPE.MCP\n\t\t\t\t\t) {\n\t\t\t\t\t\tconst toolArgs = toolCall.arguments as\n\t\t\t\t\t\t\t| { [x: string]: unknown }\n\t\t\t\t\t\t\t| undefined;\n\t\t\t\t\t\tloggers.intent.debug(\"MCP tool call\", { toolName, toolArgs });\n\t\t\t\t\t\tconst result = await this.mcpModule.useTool(\n\t\t\t\t\t\t\tselectedTool as IMCPTool,\n\t\t\t\t\t\t\ttoolArgs,\n\t\t\t\t\t\t);\n\t\t\t\t\t\ttoolResult =\n\t\t\t\t\t\t\t`[Bot Called MCP Tool ${toolName} with args ${JSON.stringify(toolArgs)}]\\n` +\n\t\t\t\t\t\t\tJSON.stringify(result.content, null, 2);\n\t\t\t\t\t} else if (\n\t\t\t\t\t\tthis.a2aModule &&\n\t\t\t\t\t\tselectedTool.protocol === TOOL_PROTOCOL_TYPE.A2A\n\t\t\t\t\t) {\n\t\t\t\t\t\tconst result = await this.a2aModule.useTool(\n\t\t\t\t\t\t\tselectedTool as IA2ATool,\n\t\t\t\t\t\t\tmessagePayload!,\n\t\t\t\t\t\t\tsessionId,\n\t\t\t\t\t\t);\n\t\t\t\t\t\ttoolResult = `[Bot Called A2A Tool ${toolName}]\\n${result.join(\"\\n\")}`;\n\t\t\t\t\t} else {\n\t\t\t\t\t\t// Unrecognized tool type. It cannot be happened...\n\t\t\t\t\t\tloggers.intent.warn(\n\t\t\t\t\t\t\t`Unrecognized tool type: ${selectedTool.protocol}`,\n\t\t\t\t\t\t);\n\t\t\t\t\t\tcontinue;\n\t\t\t\t\t}\n\n\t\t\t\t\tloggers.intent.debug(\"toolResult\", { toolResult });\n\n\t\t\t\t\tprocessList.push(toolResult);\n\t\t\t\t\tmodelInstance.appendMessages(messages, toolResult);\n\t\t\t\t}\n\t\t\t} else if (content) {\n\t\t\t\tprocessList.push(content);\n\t\t\t\tfinalMessage = content;\n\t\t\t}\n\n\t\t\tif (!didCallTool) break;\n\t\t}\n\n\t\tconst botResponse = {\n\t\t\tprocess: processList.join(\"\\n\"),\n\t\t\tresponse: finalMessage,\n\t\t};\n\n\t\treturn botResponse;\n\t}\n\n\t/**\n\t * Main entry point for processing user queries.\n\t *\n\t * Handles the complete query lifecycle:\n\t * 1. Loads session history from memory\n\t * 2. Detects intent from the query\n\t * 3. Fulfills the intent with AI response\n\t * 4. Updates conversation history\n\t *\n\t * @param query - The user's input query\n\t * @param sessionId - Unique session identifier\n\t * @returns Object containing the AI-generated response\n\t */\n\tpublic async handleQuery(query: string, sessionId: string) {\n\t\t// 1. Load session history with sessionId\n\t\tconst queryStartAt = Date.now();\n\t\tconst sessionMemory = this.memoryModule?.getSessionMemory();\n\t\tconst sessionHistory = await sessionMemory?.getSession(sessionId);\n\n\t\t// 2. intent triggering\n\t\tconst intent = this.intentTriggering(query);\n\n\t\t// 3. intent fulfillment\n\t\tconst result = await this.intentFulfilling(\n\t\t\tquery,\n\t\t\tsessionId,\n\t\t\tsessionHistory,\n\t\t);\n\t\tif (sessionId) {\n\t\t\tawait sessionMemory?.addChatToSession(sessionId, {\n\t\t\t\trole: ChatRole.USER,\n\t\t\t\ttimestamp: queryStartAt,\n\t\t\t\tcontent: { type: \"text\", parts: [query] },\n\t\t\t});\n\t\t\tawait sessionMemory?.addChatToSession(sessionId, {\n\t\t\t\trole: ChatRole.MODEL,\n\t\t\t\ttimestamp: Date.now(),\n\t\t\t\tcontent: { type: \"text\", parts: [result.response] },\n\t\t\t});\n\t\t}\n\n\t\treturn { content: result.response };\n\t}\n}\n"],"mappings":";;;;;AAuBO,IAAM,eAAN,MAAmB;AAAA,EACjB;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EAER,YACC,aACA,WACA,WACA,cACA,SACC;AACD,SAAK,cAAc;AACnB,SAAK,YAAY;AACjB,SAAK,YAAY;AACjB,SAAK,eAAe;AACpB,SAAK,UAAU;AAAA,EAChB;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EASA,MAAc,iBAAiB,OAAe;AAE7C,WAAO;AAAA,EACR;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAgBA,MAAc,iBACb,OACA,WACA,gBACC;AAED,UAAM,eAAe;AAAA,YACZ,oBAAI,KAAK,GAAE,mBAAmB,CAAC;AAAA;AAAA,EAExC,KAAK,SAAS,SAAS,EAAE;AAAA;AAAA,EAEzB,KAAK,SAAS,UAAU,EAAE;AAAA;AAG1B,UAAM,gBAAgB,KAAK,YAAY,SAAS;AAChD,UAAM,WAAW,cAAc,iBAAiB;AAAA,MAC/C;AAAA,MACA;AAAA,MACA,cAAc,aAAa,KAAK;AAAA,IACjC,CAAC;AAED,UAAM,QAAsB,CAAC;AAC7B,QAAI,KAAK,WAAW;AACnB,YAAM,KAAK,GAAG,KAAK,UAAU,SAAS,CAAC;AAAA,IACxC;AACA,QAAI,KAAK,WAAW;AACnB,YAAM,KAAK,GAAI,MAAM,KAAK,UAAU,SAAS,CAAE;AAAA,IAChD;AACA,UAAM,YAAY,cAAc,wBAAwB,KAAK;AAE7D,UAAM,cAAwB,CAAC;AAC/B,QAAI,eAAe;AACnB,QAAI,cAAc;AAElB,WAAO,MAAM;AACZ,YAAM,WAAW,MAAM,cAAc;AAAA,QACpC;AAAA,QACA;AAAA,MACD;AACA,oBAAc;AAEd,cAAQ,OAAO,MAAM,YAAY,EAAE,SAAS,CAAC;AAE7C,YAAM,EAAE,SAAS,UAAU,IAAI;AAE/B,cAAQ,OAAO,MAAM,WAAW,EAAE,QAAQ,CAAC;AAC3C,cAAQ,OAAO,MAAM,cAAc,EAAE,GAAG,UAAU,CAAC;AAEnD,UAAI,WAAW;AACd,cAAM,iBAAiB,KAAK,WAAW;AAAA,UACtC;AAAA,UACA;AAAA,QACD;AAEA,mBAAW,YAAY,WAAW;AACjC,gBAAM,WAAW,SAAS;AAC1B,wBAAc;AACd,gBAAM,eAAe,MAAM,OAAO,CAAC,SAAS,KAAK,OAAO,QAAQ,EAAE,CAAC;AAEnE,cAAI,aAAa;AACjB,cACC,KAAK,aACL,aAAa,8BACZ;AACD,kBAAM,WAAW,SAAS;AAG1B,oBAAQ,OAAO,MAAM,iBAAiB,EAAE,UAAU,SAAS,CAAC;AAC5D,kBAAM,SAAS,MAAM,KAAK,UAAU;AAAA,cACnC;AAAA,cACA;AAAA,YACD;AACA,yBACC,wBAAwB,QAAQ,cAAc,KAAK,UAAU,QAAQ,CAAC;AAAA,IACtE,KAAK,UAAU,OAAO,SAAS,MAAM,CAAC;AAAA,UACxC,WACC,KAAK,aACL,aAAa,8BACZ;AACD,kBAAM,SAAS,MAAM,KAAK,UAAU;AAAA,cACnC;AAAA,cACA;AAAA,cACA;AAAA,YACD;AACA,yBAAa,wBAAwB,QAAQ;AAAA,EAAM,OAAO,KAAK,IAAI,CAAC;AAAA,UACrE,OAAO;AAEN,oBAAQ,OAAO;AAAA,cACd,2BAA2B,aAAa,QAAQ;AAAA,YACjD;AACA;AAAA,UACD;AAEA,kBAAQ,OAAO,MAAM,cAAc,EAAE,WAAW,CAAC;AAEjD,sBAAY,KAAK,UAAU;AAC3B,wBAAc,eAAe,UAAU,UAAU;AAAA,QAClD;AAAA,MACD,WAAW,SAAS;AACnB,oBAAY,KAAK,OAAO;AACxB,uBAAe;AAAA,MAChB;AAEA,UAAI,CAAC,YAAa;AAAA,IACnB;AAEA,UAAM,cAAc;AAAA,MACnB,SAAS,YAAY,KAAK,IAAI;AAAA,MAC9B,UAAU;AAAA,IACX;AAEA,WAAO;AAAA,EACR;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAeA,MAAa,YAAY,OAAe,WAAmB;AAE1D,UAAM,eAAe,KAAK,IAAI;AAC9B,UAAM,gBAAgB,KAAK,cAAc,iBAAiB;AAC1D,UAAM,iBAAiB,MAAM,eAAe,WAAW,SAAS;AAGhE,UAAM,SAAS,KAAK,iBAAiB,KAAK;AAG1C,UAAM,SAAS,MAAM,KAAK;AAAA,MACzB;AAAA,MACA;AAAA,MACA;AAAA,IACD;AACA,QAAI,WAAW;AACd,YAAM,eAAe,iBAAiB,WAAW;AAAA,QAChD;AAAA,QACA,WAAW;AAAA,QACX,SAAS,EAAE,MAAM,QAAQ,OAAO,CAAC,KAAK,EAAE;AAAA,MACzC,CAAC;AACD,YAAM,eAAe,iBAAiB,WAAW;AAAA,QAChD;AAAA,QACA,WAAW,KAAK,IAAI;AAAA,QACpB,SAAS,EAAE,MAAM,QAAQ,OAAO,CAAC,OAAO,QAAQ,EAAE;AAAA,MACnD,CAAC;AAAA,IACF;AAEA,WAAO,EAAE,SAAS,OAAO,SAAS;AAAA,EACnC;AACD;","names":[]}

package/dist/esm/chunk-MEKGLPDX.js

@@ -0,0 +1,51 @@
+// src/modules/memory/memory.module.ts
+var MemoryModule = class {
+  agentMemory;
+  sessionMemory;
+  intentMemory;
+  constructor(config) {
+    this.agentMemory = config.agent;
+    this.sessionMemory = config.session;
+    this.intentMemory = config.intent;
+  }
+  async initialize() {
+    const connectPromises = [];
+    if (this.agentMemory) {
+      connectPromises.push(this.agentMemory.connect());
+    }
+    if (this.sessionMemory) {
+      connectPromises.push(this.sessionMemory.connect());
+    }
+    if (this.intentMemory) {
+      connectPromises.push(this.intentMemory.connect());
+    }
+    await Promise.all(connectPromises);
+  }
+  async shutdown() {
+    const disconnectPromises = [];
+    if (this.agentMemory && this.agentMemory.isConnected()) {
+      disconnectPromises.push(this.agentMemory.disconnect());
+    }
+    if (this.sessionMemory && this.sessionMemory.isConnected()) {
+      disconnectPromises.push(this.sessionMemory.disconnect());
+    }
+    if (this.intentMemory && this.intentMemory.isConnected()) {
+      disconnectPromises.push(this.intentMemory.disconnect());
+    }
+    await Promise.all(disconnectPromises);
+  }
+  getAgentMemory() {
+    return this.agentMemory;
+  }
+  getSessionMemory() {
+    return this.sessionMemory;
+  }
+  getIntentMemory() {
+    return this.intentMemory;
+  }
+};
+
+export {
+  MemoryModule
+};
+//# sourceMappingURL=chunk-MEKGLPDX.js.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/modules/memory/memory.module.ts"],"sourcesContent":["import type {\n\tIAgentMemory,\n\tIIntentMemory,\n\tISessionMemory,\n} from \"./base.memory.js\";\n\nexport interface MemoryConfig {\n\tagent?: IAgentMemory;\n\tsession?: ISessionMemory;\n\tintent?: IIntentMemory;\n}\n\nexport class MemoryModule {\n\tprivate agentMemory?: IAgentMemory;\n\tprivate sessionMemory?: ISessionMemory;\n\tprivate intentMemory?: IIntentMemory;\n\n\tconstructor(config: MemoryConfig) {\n\t\tthis.agentMemory = config.agent;\n\t\tthis.sessionMemory = config.session;\n\t\tthis.intentMemory = config.intent;\n\t}\n\n\tasync initialize(): Promise<void> {\n\t\tconst connectPromises: Promise<void>[] = [];\n\n\t\tif (this.agentMemory) {\n\t\t\tconnectPromises.push(this.agentMemory.connect());\n\t\t}\n\t\tif (this.sessionMemory) {\n\t\t\tconnectPromises.push(this.sessionMemory.connect());\n\t\t}\n\t\tif (this.intentMemory) {\n\t\t\tconnectPromises.push(this.intentMemory.connect());\n\t\t}\n\n\t\tawait Promise.all(connectPromises);\n\t}\n\n\tasync shutdown(): Promise<void> {\n\t\tconst disconnectPromises: Promise<void>[] = [];\n\n\t\tif (this.agentMemory && this.agentMemory.isConnected()) {\n\t\t\tdisconnectPromises.push(this.agentMemory.disconnect());\n\t\t}\n\t\tif (this.sessionMemory && this.sessionMemory.isConnected()) {\n\t\t\tdisconnectPromises.push(this.sessionMemory.disconnect());\n\t\t}\n\t\tif (this.intentMemory && this.intentMemory.isConnected()) {\n\t\t\tdisconnectPromises.push(this.intentMemory.disconnect());\n\t\t}\n\n\t\tawait Promise.all(disconnectPromises);\n\t}\n\n\tpublic getAgentMemory(): IAgentMemory | undefined {\n\t\treturn this.agentMemory;\n\t}\n\n\tpublic getSessionMemory(): ISessionMemory | undefined {\n\t\treturn this.sessionMemory;\n\t}\n\n\tpublic getIntentMemory(): IIntentMemory | undefined {\n\t\treturn this.intentMemory;\n\t}\n}\n"],"mappings":";AAYO,IAAM,eAAN,MAAmB;AAAA,EACjB;AAAA,EACA;AAAA,EACA;AAAA,EAER,YAAY,QAAsB;AACjC,SAAK,cAAc,OAAO;AAC1B,SAAK,gBAAgB,OAAO;AAC5B,SAAK,eAAe,OAAO;AAAA,EAC5B;AAAA,EAEA,MAAM,aAA4B;AACjC,UAAM,kBAAmC,CAAC;AAE1C,QAAI,KAAK,aAAa;AACrB,sBAAgB,KAAK,KAAK,YAAY,QAAQ,CAAC;AAAA,IAChD;AACA,QAAI,KAAK,eAAe;AACvB,sBAAgB,KAAK,KAAK,cAAc,QAAQ,CAAC;AAAA,IAClD;AACA,QAAI,KAAK,cAAc;AACtB,sBAAgB,KAAK,KAAK,aAAa,QAAQ,CAAC;AAAA,IACjD;AAEA,UAAM,QAAQ,IAAI,eAAe;AAAA,EAClC;AAAA,EAEA,MAAM,WAA0B;AAC/B,UAAM,qBAAsC,CAAC;AAE7C,QAAI,KAAK,eAAe,KAAK,YAAY,YAAY,GAAG;AACvD,yBAAmB,KAAK,KAAK,YAAY,WAAW,CAAC;AAAA,IACtD;AACA,QAAI,KAAK,iBAAiB,KAAK,cAAc,YAAY,GAAG;AAC3D,yBAAmB,KAAK,KAAK,cAAc,WAAW,CAAC;AAAA,IACxD;AACA,QAAI,KAAK,gBAAgB,KAAK,aAAa,YAAY,GAAG;AACzD,yBAAmB,KAAK,KAAK,aAAa,WAAW,CAAC;AAAA,IACvD;AAEA,UAAM,QAAQ,IAAI,kBAAkB;AAAA,EACrC;AAAA,EAEO,iBAA2C;AACjD,WAAO,KAAK;AAAA,EACb;AAAA,EAEO,mBAA+C;AACrD,WAAO,KAAK;AAAA,EACb;AAAA,EAEO,kBAA6C;AACnD,WAAO,KAAK;AAAA,EACb;AACD;","names":[]}

package/dist/esm/controllers/a2a.controller.d.ts

@@ -13,8 +13,8 @@ import '../types/mcp.js';
 import '@modelcontextprotocol/sdk/client/stdio.js';
 import '../modules/mcp/mcp.tool.js';
 import '../modules/memory/memory.module.js';
-import '../types/memory.js';
 import '../modules/memory/base.memory.js';
+import '../types/memory.js';
 import '../modules/models/model.module.js';
 import '../modules/models/base.model.js';
 import '../types/agent.js';

package/dist/esm/controllers/api/session.api.controller.d.ts

@@ -1,7 +1,7 @@
 import { Request, Response, NextFunction } from 'express';
 import { MemoryModule } from '../../modules/memory/memory.module.js';
-import '../../types/memory.js';
 import '../../modules/memory/base.memory.js';
+import '../../types/memory.js';
 
 declare class SessionApiController {
     private memoryModule;

package/dist/esm/controllers/api/session.api.controller.js

@@ -1,6 +1,6 @@
 import {
   SessionApiController
-} from "../../chunk-X47DAWPH.js";
+} from "../../chunk-IXXK7ZZM.js";
 import "../../chunk-FL3JRQM4.js";
 export {
   SessionApiController

package/dist/esm/controllers/index.d.ts

@@ -17,8 +17,8 @@ import '../types/mcp.js';
 import '@modelcontextprotocol/sdk/client/stdio.js';
 import '../modules/mcp/mcp.tool.js';
 import '../modules/memory/memory.module.js';
-import '../types/memory.js';
 import '../modules/memory/base.memory.js';
+import '../types/memory.js';
 import '../modules/models/model.module.js';
 import '../modules/models/base.model.js';
 import '../types/agent.js';

package/dist/esm/controllers/index.js

@@ -3,14 +3,14 @@ import {
 } from "../chunk-UA44VKFE.js";
 import {
   SessionApiController
-} from "../chunk-X47DAWPH.js";
+} from "../chunk-IXXK7ZZM.js";
+import {
+  A2AController
+} from "../chunk-EODMRV4H.js";
 import {
   QueryController
 } from "../chunk-XOMQNCCR.js";
 import "../chunk-FL3JRQM4.js";
-import {
-  A2AController
-} from "../chunk-EODMRV4H.js";
 import "../chunk-64GEXWSE.js";
 export {
   A2AController,

package/dist/esm/controllers/query.controller.d.ts

@@ -11,8 +11,8 @@ import '../types/mcp.js';
 import '@modelcontextprotocol/sdk/client/stdio.js';
 import '../modules/mcp/mcp.tool.js';
 import '../modules/memory/memory.module.js';
-import '../types/memory.js';
 import '../modules/memory/base.memory.js';
+import '../types/memory.js';
 import '../modules/models/model.module.js';
 import '../modules/models/base.model.js';
 import '../types/agent.js';

package/dist/esm/index.d.ts

@@ -13,8 +13,8 @@ import '@modelcontextprotocol/sdk/types.js';
 import './types/mcp.js';
 import '@modelcontextprotocol/sdk/client/stdio.js';
 import './modules/mcp/mcp.tool.js';
-import './types/memory.js';
 import './modules/memory/base.memory.js';
+import './types/memory.js';
 import './modules/models/base.model.js';
 import 'http-status-codes';
 
@@ -109,7 +109,7 @@ declare class AINAgent {
      *
      * @param port - The port number to listen on
      */
-    start(port: number): void;
+    start(port: number): Promise<void>;
 }
 
 export { AINAgent };

package/dist/esm/index.js

@@ -1,14 +1,20 @@
 import "./chunk-XYCWK2ZM.js";
+import {
+  createA2ARouter
+} from "./chunk-423RKLBO.js";
 import {
   createApiRouter
-} from "./chunk-UZQZXNND.js";
+} from "./chunk-JLH6DGBK.js";
 import "./chunk-2Z7GJVJJ.js";
-import "./chunk-M4HGSCXF.js";
+import "./chunk-KR6W7QIW.js";
 import {
   createQueryRouter
-} from "./chunk-CQ6GXYN2.js";
+} from "./chunk-6SDU6Z5Q.js";
+import "./chunk-LVKAW3WE.js";
+import "./chunk-F727XG4N.js";
 import "./chunk-UA44VKFE.js";
-import "./chunk-X47DAWPH.js";
+import "./chunk-IXXK7ZZM.js";
+import "./chunk-EODMRV4H.js";
 import "./chunk-XOMQNCCR.js";
 import {
   AuthMiddleware
@@ -17,14 +23,8 @@ import "./chunk-FL3JRQM4.js";
 import {
   errorMiddleware
 } from "./chunk-ILFWNK3W.js";
-import {
-  createA2ARouter
-} from "./chunk-CH4XHVJY.js";
-import "./chunk-AQ5YZERE.js";
-import "./chunk-WZBXLGQK.js";
-import "./chunk-EODMRV4H.js";
-import "./chunk-TMHRCGTE.js";
 import "./chunk-FI2UQLVC.js";
+import "./chunk-AQ5YZERE.js";
 import {
   loggers
 } from "./chunk-64GEXWSE.js";
@@ -716,7 +716,8 @@ var AINAgent = class {
    *
    * @param port - The port number to listen on
    */
-  start(port) {
+  async start(port) {
+    await this.memoryModule?.initialize();
     this.app.listen(port, () => {
       loggers.agent.info(`AINAgent is running on port ${port}`);
     });