@ainetwork/adk 0.1.2 → 0.1.4

package/dist/cjs/modules/memory/memory.module.cjs

@@ -1,8 +1,7 @@
 "use strict";Object.defineProperty(exports, "__esModule", {value: true});
 
-var _chunk5VIAH4HEcjs = require('../../chunk-5VIAH4HE.cjs');
-require('../../chunk-LISZOU2X.cjs');
+var _chunkQ4FP72ZYcjs = require('../../chunk-Q4FP72ZY.cjs');
 
 
-exports.MemoryModule = _chunk5VIAH4HEcjs.MemoryModule;
+exports.MemoryModule = _chunkQ4FP72ZYcjs.MemoryModule;
 //# sourceMappingURL=memory.module.cjs.map

package/dist/cjs/modules/memory/memory.module.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/modules/memory/memory.module.cjs"],"names":[],"mappings":"AAAA;AACE;AACF,4DAAiC;AACjC,oCAAiC;AACjC;AACE;AACF,sDAAC","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/modules/memory/memory.module.cjs"}
+{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/modules/memory/memory.module.cjs"],"names":[],"mappings":"AAAA;AACE;AACF,4DAAiC;AACjC;AACE;AACF,sDAAC","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/modules/memory/memory.module.cjs"}

package/dist/cjs/routes/a2a.routes.cjs

@@ -1,13 +1,13 @@
 "use strict";Object.defineProperty(exports, "__esModule", {value: true});
 
-var _chunkCR2I7S4Ucjs = require('../chunk-CR2I7S4U.cjs');
-require('../chunk-ZMYG5HDT.cjs');
+var _chunkD75FARQ5cjs = require('../chunk-D75FARQ5.cjs');
 require('../chunk-7BZO57TG.cjs');
+require('../chunk-573HTYTZ.cjs');
+require('../chunk-3IGIYVGA.cjs');
 require('../chunk-ILGIEJCX.cjs');
-require('../chunk-LISZOU2X.cjs');
 require('../chunk-LPXYZH22.cjs');
 require('../chunk-JTAXZJQ2.cjs');
 
 
-exports.createA2ARouter = _chunkCR2I7S4Ucjs.createA2ARouter;
+exports.createA2ARouter = _chunkD75FARQ5cjs.createA2ARouter;
 //# sourceMappingURL=a2a.routes.cjs.map

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 _chunkMTAGYTHOcjs = require('../../chunk-MTAGYTHO.cjs');
+require('../../chunk-J3BFUVZS.cjs');
 require('../../chunk-DX2JH3FJ.cjs');
 
 
-exports.createSessionApiRouter = _chunkD525HP3Rcjs.createSessionApiRouter;
+exports.createSessionApiRouter = _chunkMTAGYTHOcjs.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 _chunk74J4UIJ4cjs = require('../chunk-74J4UIJ4.cjs');
+require('../chunk-MTAGYTHO.cjs');
 require('../chunk-N7NGEMKL.cjs');
-require('../chunk-D525HP3R.cjs');
 require('../chunk-V5RSN2DY.cjs');
-require('../chunk-YRNIZTPF.cjs');
+require('../chunk-J3BFUVZS.cjs');
 require('../chunk-DX2JH3FJ.cjs');
 
 
-exports.createApiRouter = _chunk6YJFC7BYcjs.createApiRouter;
+exports.createApiRouter = _chunk74J4UIJ4cjs.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 _chunk74J4UIJ4cjs = require('../chunk-74J4UIJ4.cjs');
 
 
-var _chunk6YJFC7BYcjs = require('../chunk-6YJFC7BY.cjs');
+var _chunkMTAGYTHOcjs = require('../chunk-MTAGYTHO.cjs');
 
 
 var _chunkN7NGEMKLcjs = require('../chunk-N7NGEMKL.cjs');
 
 
-var _chunkD525HP3Rcjs = require('../chunk-D525HP3R.cjs');
+var _chunkTWUZNIBNcjs = require('../chunk-TWUZNIBN.cjs');
+require('../chunk-V5RSN2DY.cjs');
+require('../chunk-J3BFUVZS.cjs');
+require('../chunk-RGKCBQGZ.cjs');
+require('../chunk-DX2JH3FJ.cjs');
 
 
-var _chunk7UL6NAT6cjs = require('../chunk-7UL6NAT6.cjs');
-require('../chunk-ZMYG5HDT.cjs');
+var _chunkD75FARQ5cjs = require('../chunk-D75FARQ5.cjs');
 require('../chunk-7BZO57TG.cjs');
-require('../chunk-V5RSN2DY.cjs');
-require('../chunk-YRNIZTPF.cjs');
+require('../chunk-573HTYTZ.cjs');
+require('../chunk-3IGIYVGA.cjs');
 require('../chunk-ILGIEJCX.cjs');
-require('../chunk-DOFTYGJR.cjs');
-require('../chunk-LISZOU2X.cjs');
 require('../chunk-LPXYZH22.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 = _chunkD75FARQ5cjs.createA2ARouter; exports.createApiRouter = _chunk74J4UIJ4cjs.createApiRouter; exports.createModelApiRouter = _chunkN7NGEMKLcjs.createModelApiRouter; exports.createQueryRouter = _chunkTWUZNIBNcjs.createQueryRouter; exports.createSessionApiRouter = _chunkMTAGYTHOcjs.createSessionApiRouter;
 //# sourceMappingURL=index.cjs.map

package/dist/cjs/routes/index.cjs.map

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

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');
-require('../chunk-DOFTYGJR.cjs');
-require('../chunk-LISZOU2X.cjs');
+var _chunkTWUZNIBNcjs = require('../chunk-TWUZNIBN.cjs');
+require('../chunk-RGKCBQGZ.cjs');
+require('../chunk-573HTYTZ.cjs');
+require('../chunk-3IGIYVGA.cjs');
 require('../chunk-LPXYZH22.cjs');
 require('../chunk-JTAXZJQ2.cjs');
 
 
-exports.createQueryRouter = _chunk7UL6NAT6cjs.createQueryRouter;
+exports.createQueryRouter = _chunkTWUZNIBNcjs.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 _chunk7BZO57TGcjs = require('../chunk-7BZO57TG.cjs');
 
 
-var _chunk7BZO57TGcjs = require('../chunk-7BZO57TG.cjs');
-require('../chunk-LISZOU2X.cjs');
+var _chunk573HTYTZcjs = require('../chunk-573HTYTZ.cjs');
+require('../chunk-3IGIYVGA.cjs');
 require('../chunk-LPXYZH22.cjs');
 require('../chunk-JTAXZJQ2.cjs');
 
 
 
-exports.A2AService = _chunk7BZO57TGcjs.A2AService; exports.QueryService = _chunkZMYG5HDTcjs.QueryService;
+exports.A2AService = _chunk7BZO57TGcjs.A2AService; exports.QueryService = _chunk573HTYTZcjs.QueryService;
 //# sourceMappingURL=index.cjs.map

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 _chunk573HTYTZcjs = require('../chunk-573HTYTZ.cjs');
+require('../chunk-3IGIYVGA.cjs');
 require('../chunk-LPXYZH22.cjs');
 require('../chunk-JTAXZJQ2.cjs');
 
 
-exports.QueryService = _chunkZMYG5HDTcjs.QueryService;
+exports.QueryService = _chunk573HTYTZcjs.QueryService;
 //# sourceMappingURL=query.service.cjs.map

package/dist/cjs/types/auth.cjs

@@ -0,0 +1 @@
+"use strict";//# sourceMappingURL=auth.cjs.map

package/dist/cjs/types/auth.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/types/auth.cjs"],"names":[],"mappings":"AAAA","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/types/auth.cjs"}

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-WZBXLGQK.js → chunk-2K5L4GFC.js}

@@ -137,12 +137,10 @@ ${result.join("\n")}`;
    * @param sessionId - Unique session identifier
    * @returns Object containing the AI-generated response
    */
-  async handleQuery(query, sessionId) {
+  async handleQuery(query, sessionId, userId) {
     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, userId);
     const intent = this.intentTriggering(query);
     const result = await this.intentFulfilling(
       query,
@@ -150,16 +148,24 @@ ${result.join("\n")}`;
       sessionHistory
     );
     if (sessionId) {
-      await memoryInstance?.updateSessionHistory(sessionId, {
-        role: "USER" /* USER */,
-        timestamp: queryStartAt,
-        content: { type: "text", parts: [query] }
-      });
-      await memoryInstance?.updateSessionHistory(sessionId, {
-        role: "MODEL" /* MODEL */,
-        timestamp: Date.now(),
-        content: { type: "text", parts: [result.response] }
-      });
+      await sessionMemory?.addChatToSession(
+        sessionId,
+        {
+          role: "USER" /* USER */,
+          timestamp: queryStartAt,
+          content: { type: "text", parts: [query] }
+        },
+        userId
+      );
+      await sessionMemory?.addChatToSession(
+        sessionId,
+        {
+          role: "MODEL" /* MODEL */,
+          timestamp: Date.now(),
+          content: { type: "text", parts: [result.response] }
+        },
+        userId
+      );
     }
     return { content: result.response };
   }
@@ -168,4 +174,4 @@ ${result.join("\n")}`;
 export {
   QueryService
 };
-//# sourceMappingURL=chunk-WZBXLGQK.js.map
+//# sourceMappingURL=chunk-2K5L4GFC.js.map

package/dist/esm/chunk-2K5L4GFC.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, userId?: 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, userId);\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(\n\t\t\t\tsessionId,\n\t\t\t\t{\n\t\t\t\t\trole: ChatRole.USER,\n\t\t\t\t\ttimestamp: queryStartAt,\n\t\t\t\t\tcontent: { type: \"text\", parts: [query] },\n\t\t\t\t},\n\t\t\t\tuserId,\n\t\t\t);\n\t\t\tawait sessionMemory?.addChatToSession(\n\t\t\t\tsessionId,\n\t\t\t\t{\n\t\t\t\t\trole: ChatRole.MODEL,\n\t\t\t\t\ttimestamp: Date.now(),\n\t\t\t\t\tcontent: { type: \"text\", parts: [result.response] },\n\t\t\t\t},\n\t\t\t\tuserId,\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,QAAiB;AAE3E,UAAM,eAAe,KAAK,IAAI;AAC9B,UAAM,gBAAgB,KAAK,cAAc,iBAAiB;AAC1D,UAAM,iBAAiB,MAAM,eAAe,WAAW,WAAW,MAAM;AAGxE,UAAM,SAAS,KAAK,iBAAiB,KAAK;AAG1C,UAAM,SAAS,MAAM,KAAK;AAAA,MACzB;AAAA,MACA;AAAA,MACA;AAAA,IACD;AACA,QAAI,WAAW;AACd,YAAM,eAAe;AAAA,QACpB;AAAA,QACA;AAAA,UACC;AAAA,UACA,WAAW;AAAA,UACX,SAAS,EAAE,MAAM,QAAQ,OAAO,CAAC,KAAK,EAAE;AAAA,QACzC;AAAA,QACA;AAAA,MACD;AACA,YAAM,eAAe;AAAA,QACpB;AAAA,QACA;AAAA,UACC;AAAA,UACA,WAAW,KAAK,IAAI;AAAA,UACpB,SAAS,EAAE,MAAM,QAAQ,OAAO,CAAC,OAAO,QAAQ,EAAE;AAAA,QACnD;AAAA,QACA;AAAA,MACD;AAAA,IACD;AAEA,WAAO,EAAE,SAAS,OAAO,SAAS;AAAA,EACnC;AACD;","names":[]}

package/dist/esm/{chunk-635VQ4CH.js → chunk-6BK7LDIA.js}

@@ -12,8 +12,9 @@ var AuthMiddleware = class {
   middleware() {
     return async (req, res, next) => {
       try {
-        const isAuthenticated = await this.auth.authenticate(req, res);
-        if (isAuthenticated) {
+        const authRes = await this.auth.authenticate(req, res);
+        if (authRes.isAuthenticated) {
+          res.locals.userId = authRes.userId;
           next();
         } else {
           const error = new AinHttpError(
@@ -39,4 +40,4 @@ var AuthMiddleware = class {
 export {
   AuthMiddleware
 };
-//# sourceMappingURL=chunk-635VQ4CH.js.map
+//# sourceMappingURL=chunk-6BK7LDIA.js.map

package/dist/esm/chunk-6BK7LDIA.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\";\nimport type { AuthResponse } from \"@/types/auth\";\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 authRes: AuthResponse = await this.auth.authenticate(req, res);\n\t\t\t\tif (authRes.isAuthenticated) {\n\t\t\t\t\tres.locals.userId = authRes.userId;\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;AAKrB,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,UAAwB,MAAM,KAAK,KAAK,aAAa,KAAK,GAAG;AACnE,YAAI,QAAQ,iBAAiB;AAC5B,cAAI,OAAO,SAAS,QAAQ;AAC5B,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-CKTURH34.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?.isConnected()) {
+      disconnectPromises.push(this.agentMemory.disconnect());
+    }
+    if (this.sessionMemory?.isConnected()) {
+      disconnectPromises.push(this.sessionMemory.disconnect());
+    }
+    if (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-CKTURH34.js.map

package/dist/esm/chunk-CKTURH34.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?.isConnected()) {\n\t\t\tdisconnectPromises.push(this.agentMemory.disconnect());\n\t\t}\n\t\tif (this.sessionMemory?.isConnected()) {\n\t\t\tdisconnectPromises.push(this.sessionMemory.disconnect());\n\t\t}\n\t\tif (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,aAAa,YAAY,GAAG;AACpC,yBAAmB,KAAK,KAAK,YAAY,WAAW,CAAC;AAAA,IACtD;AACA,QAAI,KAAK,eAAe,YAAY,GAAG;AACtC,yBAAmB,KAAK,KAAK,cAAc,WAAW,CAAC;AAAA,IACxD;AACA,QAAI,KAAK,cAAc,YAAY,GAAG;AACrC,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/{chunk-CQ6GXYN2.js → chunk-EJIQKPIV.js}

@@ -1,9 +1,9 @@
 import {
   QueryController
-} from "./chunk-XOMQNCCR.js";
+} from "./chunk-WIJDRP2A.js";
 import {
   QueryService
-} from "./chunk-WZBXLGQK.js";
+} from "./chunk-2K5L4GFC.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-EJIQKPIV.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-KJ4MVCYE.js → chunk-FF5RJ4L2.js}

@@ -5,4 +5,4 @@ var BaseAuth = class {
 export {
   BaseAuth
 };
-//# sourceMappingURL=chunk-KJ4MVCYE.js.map
+//# sourceMappingURL=chunk-FF5RJ4L2.js.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/modules/auth/base.auth.ts"],"sourcesContent":["import type { Request, Response } from \"express\";\nimport type { AuthResponse } from \"@/types/auth\";\n\nexport abstract class BaseAuth {\n\tabstract authenticate(req: Request, res: Response): Promise<AuthResponse>;\n}\n"],"mappings":";AAGO,IAAe,WAAf,MAAwB;AAE/B;","names":[]}

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

@@ -1,9 +1,9 @@
+import {
+  createSessionApiRouter
+} from "./chunk-RSSLUA2N.js";
 import {
   createModelApiRouter
 } from "./chunk-2Z7GJVJJ.js";
-import {
-  createSessionApiRouter
-} from "./chunk-M4HGSCXF.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-NSKXMJWE.js.map

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

@@ -3,7 +3,7 @@ import {
 } from "./chunk-AQ5YZERE.js";
 import {
   QueryService
-} from "./chunk-WZBXLGQK.js";
+} from "./chunk-2K5L4GFC.js";
 import {
   A2AController
 } from "./chunk-EODMRV4H.js";
@@ -34,4 +34,4 @@ var createA2ARouter = (agent) => {
 export {
   createA2ARouter
 };
-//# sourceMappingURL=chunk-CH4XHVJY.js.map
+//# sourceMappingURL=chunk-OIIR42TQ.js.map

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

@@ -12,15 +12,16 @@ var SessionApiController = class {
   handleSessionHistory = async (req, res, next) => {
     try {
       const { id: sessionId } = req.params;
-      const memoryInstance = this.memoryModule.getMemory();
-      if (!memoryInstance) {
+      const userId = res.locals.userId || "";
+      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, userId);
       res.json(session);
     } catch (error) {
       next(error);
@@ -31,4 +32,4 @@ var SessionApiController = class {
 export {
   SessionApiController
 };
-//# sourceMappingURL=chunk-X47DAWPH.js.map
+//# sourceMappingURL=chunk-RB4IFZJA.js.map

package/dist/esm/chunk-RB4IFZJA.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 userId = res.locals.userId || \"\";\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, userId);\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,SAAS,IAAI,OAAO,UAAU;AACpC,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,WAAW,MAAM;AAChE,UAAI,KAAK,OAAO;AAAA,IACjB,SAAS,OAAO;AACf,WAAK,KAAK;AAAA,IACX;AAAA,EACD;AACD;","names":[]}

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

@@ -1,6 +1,6 @@
 import {
   SessionApiController
-} from "./chunk-X47DAWPH.js";
+} from "./chunk-RB4IFZJA.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-RSSLUA2N.js.map

package/dist/esm/{chunk-XOMQNCCR.js → chunk-WIJDRP2A.js}

@@ -7,7 +7,12 @@ var QueryController = class {
   handleQueryRequest = async (req, res, next) => {
     try {
       const { message, sessionId } = req.body;
-      const result = await this.queryService.handleQuery(message, sessionId);
+      const userId = res.locals.userId;
+      const result = await this.queryService.handleQuery(
+        message,
+        sessionId,
+        userId
+      );
       res.status(200).json(result);
     } catch (error) {
       next(error);
@@ -18,4 +23,4 @@ var QueryController = class {
 export {
   QueryController
 };
-//# sourceMappingURL=chunk-XOMQNCCR.js.map
+//# sourceMappingURL=chunk-WIJDRP2A.js.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/controllers/query.controller.ts"],"sourcesContent":["import type { NextFunction, Request, Response } from \"express\";\nimport type { QueryService } from \"@/services\";\n\nexport class QueryController {\n\tprivate queryService;\n\n\tconstructor(queryService: QueryService) {\n\t\tthis.queryService = queryService;\n\t}\n\n\tpublic handleQueryRequest = async (\n\t\treq: Request,\n\t\tres: Response,\n\t\tnext: NextFunction,\n\t) => {\n\t\ttry {\n\t\t\tconst { message, sessionId } = req.body;\n\t\t\tconst userId = res.locals.userId;\n\n\t\t\tconst result = await this.queryService.handleQuery(\n\t\t\t\tmessage,\n\t\t\t\tsessionId,\n\t\t\t\tuserId,\n\t\t\t);\n\n\t\t\tres.status(200).json(result);\n\t\t} catch (error) {\n\t\t\tnext(error);\n\t\t}\n\t};\n}\n"],"mappings":";AAGO,IAAM,kBAAN,MAAsB;AAAA,EACpB;AAAA,EAER,YAAY,cAA4B;AACvC,SAAK,eAAe;AAAA,EACrB;AAAA,EAEO,qBAAqB,OAC3B,KACA,KACA,SACI;AACJ,QAAI;AACH,YAAM,EAAE,SAAS,UAAU,IAAI,IAAI;AACnC,YAAM,SAAS,IAAI,OAAO;AAE1B,YAAM,SAAS,MAAM,KAAK,aAAa;AAAA,QACtC;AAAA,QACA;AAAA,QACA;AAAA,MACD;AAEA,UAAI,OAAO,GAAG,EAAE,KAAK,MAAM;AAAA,IAC5B,SAAS,OAAO;AACf,WAAK,KAAK;AAAA,IACX;AAAA,EACD;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-RB4IFZJA.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,10 +3,10 @@ import {
 } from "../chunk-UA44VKFE.js";
 import {
   SessionApiController
-} from "../chunk-X47DAWPH.js";
+} from "../chunk-RB4IFZJA.js";
 import {
   QueryController
-} from "../chunk-XOMQNCCR.js";
+} from "../chunk-WIJDRP2A.js";
 import "../chunk-FL3JRQM4.js";
 import {
   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/controllers/query.controller.js

@@ -1,6 +1,6 @@
 import {
   QueryController
-} from "../chunk-XOMQNCCR.js";
+} from "../chunk-WIJDRP2A.js";
 export {
   QueryController
 };

package/dist/esm/index.d.ts

@@ -10,11 +10,12 @@ import './modules/a2a/a2a.tool.js';
 import '@a2a-js/sdk/client';
 import './types/tool.js';
 import '@modelcontextprotocol/sdk/types.js';
+import './types/auth.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 +110,7 @@ declare class AINAgent {
      *
      * @param port - The port number to listen on
      */
-    start(port: number): void;
+    start(port: number): Promise<void>;
 }
 
 export { AINAgent };