@ainetwork/adk 0.1.12 → 0.2.0

package/dist/cjs/chunk-MMWEU2MO.cjs

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

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

@@ -1 +0,0 @@
-{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-MMWEU2MO.cjs","../../src/modules/mcp/mcp.module.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACA;ACPA,oEAAuB;AACvB,oEAAqC;AA0B9B,IAAM,UAAA,YAAN,MAAgB;AAAA;AAAA,iBAEd,OAAA,kBAA8B,IAAI,GAAA,CAAI,EAAA;AAAA;AAAA,kBAEtC,aAAA,kBAAkD,IAAI,GAAA,CAAI,EAAA;AAAA;AAAA,kBAE1D,MAAA,EAAmB,CAAC,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAY5B,MAAM,YAAA,CAAa,SAAA,EAAsB;AACxC,IAAA,IAAI;AACH,MAAA,IAAA,CAAA,MAAW,CAAC,IAAA,EAAM,IAAI,EAAA,GAAK,MAAA,CAAO,OAAA,CAAQ,SAAS,CAAA,EAAG;AAErD,QAAA,GAAA,CAAI,IAAA,CAAK,MAAA,CAAO,GAAA,CAAI,IAAI,EAAA,GAAK,IAAA,CAAK,YAAA,CAAa,GAAA,CAAI,IAAI,CAAA,EAAG,QAAA;AAE1D,QAAA,MAAM,UAAA,EAAY,IAAI,kCAAA,CAAqB,IAAI,CAAA;AAC/C,QAAA,IAAA,CAAK,YAAA,CAAa,GAAA,CAAI,IAAA,EAAM,SAAS,CAAA;AACrC,QAAA,MAAM,IAAA,EAAM,IAAI,oBAAA,CAAO,EAAE,IAAA,EAAM,gBAAA,EAAkB,OAAA,EAAS,QAAQ,CAAC,CAAA;AACnE,QAAA,MAAM,GAAA,CAAI,OAAA,CAAQ,SAAS,CAAA;AAC3B,QAAA,IAAA,CAAK,MAAA,CAAO,GAAA,CAAI,IAAA,EAAM,GAAG,CAAA;AAEzB,QAAA,MAAM,YAAA,EAAc,MAAM,GAAA,CAAI,SAAA,CAAU,CAAA;AACxC,QAAA,IAAA,CAAK,KAAA,CAAM,IAAA;AAAA,UACV,GAAG,WAAA,CAAY,KAAA,CAAM,GAAA,CAAI,CAAC,IAAA,EAAA,GAAS;AAClC,YAAA,OAAO,IAAI,8BAAA,CAAQ,IAAA,EAAM,IAAI,CAAA;AAAA,UAC9B,CAAC;AAAA,QACF,CAAA;AAAA,MACD;AACA,MAAA,yBAAA,CAAQ,GAAA,CAAI,IAAA,CAAK,qCAAA,EAAuC;AAAA,QACvD,KAAA,EAAO,IAAA,CAAK,KAAA,CAAM,GAAA,CAAI,CAAC,IAAA,EAAA,GAAS,IAAA,CAAK,EAAE;AAAA,MACxC,CAAC,CAAA;AAAA,IACF,EAAA,MAAA,CAAS,KAAA,EAAgB;AACxB,MAAA,yBAAA,CAAQ,GAAA,CAAI,KAAA,CAAM,kCAAA,EAAoC,EAAE,MAAM,CAAC,CAAA;AAC/D,MAAA,MAAM,KAAA;AAAA,IACP;AAAA,EACD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,QAAA,CAAA,EAAW;AACV,IAAA,OAAO,IAAA,CAAK,KAAA;AAAA,EACb;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAUA,MAAM,OAAA,CAAQ,IAAA,EAAe,KAAA,EAA2B;AACvD,IAAA,MAAM,EAAE,UAAA,EAAY,QAAQ,EAAA,EAAI,IAAA;AAChC,IAAA,MAAM,SAAA,EAAW,OAAA,CAAQ,IAAA;AACzB,IAAA,MAAM,IAAA,EAAM,IAAA,CAAK,MAAA,CAAO,GAAA,CAAI,UAAU,CAAA;AAEtC,IAAA,GAAA,CAAI,CAAC,GAAA,EAAK;AACT,MAAA,MAAM,IAAI,KAAA,CAAM,CAAA,iBAAA,EAAoB,UAAU,CAAA,CAAA,EAAI,OAAA,CAAQ,IAAI,CAAA,CAAA;AAC/D,IAAA;AAEkC,IAAA;AAC3B,MAAA;AACK,MAAA;AACX,IAAA;AAE0C,IAAA;AACJ;AAEY,IAAA;AAC5C,IAAA;AACR,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAQgB,EAAA;AACsB,IAAA;AAC1B,MAAA;AACV,IAAA;AACF,EAAA;AACD;AD3BqE;AACA;AACA;AACA","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-MMWEU2MO.cjs","sourcesContent":[null,"import { Client } from \"@modelcontextprotocol/sdk/client/index.js\";\nimport { StdioClientTransport } from \"@modelcontextprotocol/sdk/client/stdio.js\";\nimport type { MCPConfig } from \"@/types/mcp.js\";\nimport { loggers } from \"@/utils/logger.js\";\nimport { MCPTool } from \"./mcp.tool.js\";\n\n/**\n * Module for managing Model Context Protocol (MCP) server connections.\n *\n * This module handles the lifecycle of MCP client connections, discovers\n * available tools from connected servers, and provides an interface for\n * executing those tools. Multiple MCP servers can be connected simultaneously.\n *\n * @example\n * ```typescript\n * const mcpModule = new MCPModule();\n * await mcpModule.addMCPConfig({\n *   \"filesystem\": {\n *     command: \"npx\",\n *     args: [\"@modelcontextprotocol/server-filesystem\", \"/path/to/files\"]\n *   }\n * });\n *\n * const tools = mcpModule.getTools();\n * const result = await mcpModule.useTool(tools[0], { path: \"/example.txt\" });\n * ```\n */\nexport class MCPModule {\n\t/** Map of MCP server names to their client instances */\n\tprivate mcpMap: Map<string, Client> = new Map();\n\t/** Map of MCP server names to their transport instances */\n\tprivate transportMap: Map<string, StdioClientTransport> = new Map();\n\t/** Array of all discovered tools from connected MCP servers */\n\tprivate tools: MCPTool[] = [];\n\n\t/**\n\t * Connects to MCP servers based on the provided configuration.\n\t *\n\t * For each server in the config, establishes a connection, discovers\n\t * available tools, and adds them to the module's tool collection.\n\t * Skips servers that are already connected.\n\t *\n\t * @param mcpConfig - Configuration object mapping server names to connection details\n\t * @throws Error if connection to any MCP server fails\n\t */\n\tasync addMCPConfig(mcpConfig: MCPConfig) {\n\t\ttry {\n\t\t\tfor (const [name, conf] of Object.entries(mcpConfig)) {\n\t\t\t\t// FIXME(yoojin): Need strict duplication check.\n\t\t\t\tif (this.mcpMap.get(name) && this.transportMap.get(name)) continue; // Duplicated mcp: skip\n\n\t\t\t\tconst transport = new StdioClientTransport(conf);\n\t\t\t\tthis.transportMap.set(name, transport);\n\t\t\t\tconst mcp = new Client({ name: \"mcp-client-cli\", version: \"1.0.0\" });\n\t\t\t\tawait mcp.connect(transport);\n\t\t\t\tthis.mcpMap.set(name, mcp);\n\n\t\t\t\tconst toolsResult = await mcp.listTools();\n\t\t\t\tthis.tools.push(\n\t\t\t\t\t...toolsResult.tools.map((tool) => {\n\t\t\t\t\t\treturn new MCPTool(name, tool);\n\t\t\t\t\t}),\n\t\t\t\t);\n\t\t\t}\n\t\t\tloggers.mcp.info(\"Connected to MCP server with tools:\", {\n\t\t\t\ttools: this.tools.map((tool) => tool.id),\n\t\t\t});\n\t\t} catch (error: unknown) {\n\t\t\tloggers.mcp.error(\"Failed to connect to MCP server:\", { error });\n\t\t\tthrow error;\n\t\t}\n\t}\n\n\t/**\n\t * Returns all available tools from connected MCP servers.\n\t *\n\t * @returns Array of MCPTool instances representing available tools\n\t */\n\tgetTools() {\n\t\treturn this.tools;\n\t}\n\n\t/**\n\t * Executes a tool on its corresponding MCP server.\n\t *\n\t * @param tool - The MCPTool instance to execute\n\t * @param _args - Arguments to pass to the tool\n\t * @returns Promise resolving to the tool's execution result\n\t * @throws Error if the MCP server for the tool is not found\n\t */\n\tasync useTool(tool: MCPTool, _args?: any): Promise<any> {\n\t\tconst { serverName, mcpTool } = tool;\n\t\tconst toolName = mcpTool.name;\n\t\tconst mcp = this.mcpMap.get(serverName);\n\n\t\tif (!mcp) {\n\t\t\tthrow new Error(`Invalid MCP Tool ${serverName}-${mcpTool.name}`);\n\t\t}\n\n\t\tconst result = await mcp.callTool({\n\t\t\tname: toolName,\n\t\t\targuments: _args,\n\t\t});\n\t\tconst toolResult =\n\t\t\t`[Bot Called Tool ${toolName} with args ${JSON.stringify(_args)}]\\n` +\n\t\t\tJSON.stringify(result.content, null, 2);\n\n\t\tloggers.mcp.debug(\"MCP useTool result:\", toolResult);\n\t\treturn result;\n\t}\n\n\t/**\n\t * Closes all MCP client connections.\n\t *\n\t * Should be called when shutting down the application to ensure\n\t * all MCP connections are properly closed.\n\t */\n\tasync cleanup() {\n\t\tthis.mcpMap.forEach((mcp: Client) => {\n\t\t\tmcp.close();\n\t\t});\n\t}\n}\n"]}

package/dist/cjs/chunk-MPIGSL72.cjs

@@ -1,42 +0,0 @@
-"use strict";Object.defineProperty(exports, "__esModule", {value: true});
-
-var _chunkDX2JH3FJcjs = require('./chunk-DX2JH3FJ.cjs');
-
-// src/middlewares/auth.middleware.ts
-var _httpstatuscodes = require('http-status-codes');
-var AuthMiddleware = class {
-  
-  constructor(auth) {
-    this.auth = auth;
-  }
-  middleware() {
-    return async (req, res, next) => {
-      try {
-        const isAuthenticated = await this.auth.authenticate(req, res);
-        if (isAuthenticated) {
-          next();
-        } else {
-          const error = new (0, _chunkDX2JH3FJcjs.AinHttpError)(
-            _httpstatuscodes.StatusCodes.UNAUTHORIZED,
-            "Unauthorized"
-          );
-          throw error;
-        }
-      } catch (e) {
-        if (!e.status) {
-          const error = new (0, _chunkDX2JH3FJcjs.AinHttpError)(
-            _httpstatuscodes.StatusCodes.INTERNAL_SERVER_ERROR,
-            `Authentication error: ${JSON.stringify(e)}`
-          );
-          throw error;
-        }
-        throw e;
-      }
-    };
-  }
-};
-
-
-
-exports.AuthMiddleware = AuthMiddleware;
-//# sourceMappingURL=chunk-MPIGSL72.cjs.map

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

@@ -1 +0,0 @@
-{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-MPIGSL72.cjs","../../src/middlewares/auth.middleware.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACA;ACHA,oDAA4B;AAIrB,IAAM,eAAA,EAAN,MAAqB;AAAA,EACnB;AAAA,EACR,WAAA,CAAY,IAAA,EAAgB;AAC3B,IAAA,IAAA,CAAK,KAAA,EAAO,IAAA;AAAA,EACb;AAAA,EAEO,UAAA,CAAA,EAA6B;AACnC,IAAA,OAAO,MAAA,CAAO,GAAA,EAAc,GAAA,EAAe,IAAA,EAAA,GAAuB;AACjE,MAAA,IAAI;AACH,QAAA,MAAM,gBAAA,EAAkB,MAAM,IAAA,CAAK,IAAA,CAAK,YAAA,CAAa,GAAA,EAAK,GAAG,CAAA;AAC7D,QAAA,GAAA,CAAI,eAAA,EAAiB;AACpB,UAAA,IAAA,CAAK,CAAA;AAAA,QACN,EAAA,KAAO;AACN,UAAA,MAAM,MAAA,EAAsB,IAAI,mCAAA;AAAA,YAC/B,4BAAA,CAAY,YAAA;AAAA,YACZ;AAAA,UACD,CAAA;AACA,UAAA,MAAM,KAAA;AAAA,QACP;AAAA,MACD,EAAA,MAAA,CAAS,CAAA,EAAQ;AAChB,QAAA,GAAA,CAAI,CAAC,CAAA,CAAE,MAAA,EAAQ;AACd,UAAA,MAAM,MAAA,EAAsB,IAAI,mCAAA;AAAA,YAC/B,4BAAA,CAAY,qBAAA;AAAA,YACZ,CAAA,sBAAA,EAAyB,IAAA,CAAK,SAAA,CAAU,CAAC,CAAC,CAAA;AAAA,UAAA;AAE3C,UAAA;AAAM,QAAA;AAEP,QAAA;AAAM,MAAA;AACP,IAAA;AACD,EAAA;AAEF;ADCA;AACA;AACA;AACA","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-MPIGSL72.cjs","sourcesContent":[null,"import type { NextFunction, Request, RequestHandler, Response } from \"express\";\nimport { StatusCodes } from \"http-status-codes\";\nimport type { BaseAuth } from \"@/modules/auth/base.auth\";\nimport { AinHttpError } from \"@/types/agent\";\n\nexport class AuthMiddleware {\n\tprivate auth: BaseAuth;\n\tconstructor(auth: BaseAuth) {\n\t\tthis.auth = auth;\n\t}\n\n\tpublic middleware(): RequestHandler {\n\t\treturn async (req: Request, res: Response, next: NextFunction) => {\n\t\t\ttry {\n\t\t\t\tconst isAuthenticated = await this.auth.authenticate(req, res);\n\t\t\t\tif (isAuthenticated) {\n\t\t\t\t\tnext();\n\t\t\t\t} else {\n\t\t\t\t\tconst error: AinHttpError = new AinHttpError(\n\t\t\t\t\t\tStatusCodes.UNAUTHORIZED,\n\t\t\t\t\t\t\"Unauthorized\",\n\t\t\t\t\t);\n\t\t\t\t\tthrow error;\n\t\t\t\t}\n\t\t\t} catch (e: any) {\n\t\t\t\tif (!e.status) {\n\t\t\t\t\tconst error: AinHttpError = new AinHttpError(\n\t\t\t\t\t\tStatusCodes.INTERNAL_SERVER_ERROR,\n\t\t\t\t\t\t`Authentication error: ${JSON.stringify(e)}`,\n\t\t\t\t\t);\n\t\t\t\t\tthrow error;\n\t\t\t\t}\n\t\t\t\tthrow e;\n\t\t\t}\n\t\t};\n\t}\n}\n"]}

package/dist/cjs/chunk-MSFAMZFU.cjs

@@ -1,277 +0,0 @@
-"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 _chunk466ED5WNcjs = require('./chunk-466ED5WN.cjs');
-
-// src/services/query-stream.service.ts
-var _crypto = require('crypto');
-var QueryStreamService = class {
-  
-  
-  
-  
-  
-  constructor(modelModule, a2aModule, mcpModule, memoryModule, prompts) {
-    this.modelModule = modelModule;
-    this.a2aModule = a2aModule;
-    this.mcpModule = mcpModule;
-    this.memoryModule = memoryModule;
-    this.prompts = prompts;
-  }
-  /**
-   * Detects the intent from a user query.
-   *
-   * @param query - The user's input query
-   * @returns The detected intent (currently returns the query as-is)
-   * @todo Implement actual intent detection logic
-   */
-  async intentTriggering(query) {
-    return query;
-  }
-  /**
-   * Fulfills the detected intent by generating a response.
-   *
-   * Manages the complete inference loop including:
-   * - Loading prompts and conversation history
-   * - Collecting available tools from modules
-   * - Executing model inference with tool support
-   * - Processing tool calls iteratively until completion
-   *
-   * @param query - The user's input query
-   * @param sessionId - Session identifier for context
-   * @param sessionHistory - Previous conversation history
-   * @returns Object containing process steps and final response
-   */
-  async *intentFulfilling(query, sessionId, sessionHistory) {
-    try {
-      const systemPrompt = `
-Today is ${(/* @__PURE__ */ new Date()).toLocaleDateString()}.
-
-${_optionalChain([this, 'access', _ => _.prompts, 'optionalAccess', _2 => _2.agent]) || ""}
-
-${_optionalChain([this, 'access', _3 => _3.prompts, 'optionalAccess', _4 => _4.system]) || ""}
-    `;
-      const modelInstance = this.modelModule.getModel();
-      const messages = modelInstance.generateMessages({
-        query,
-        sessionHistory,
-        systemPrompt: systemPrompt.trim()
-      });
-      const tools = [];
-      if (this.mcpModule) {
-        tools.push(...this.mcpModule.getTools());
-      }
-      if (this.a2aModule) {
-        tools.push(...await this.a2aModule.getTools());
-      }
-      console.log(tools);
-      const functions = modelInstance.convertToolsToFunctions(tools);
-      console.log(functions);
-      const processList = [];
-      let didCallTool = false;
-      while (true) {
-        const responseStream = await modelInstance.fetchStreamWithContextMessage(
-          messages,
-          functions
-        );
-        didCallTool = false;
-        const assembledToolCalls = [];
-        _chunk466ED5WNcjs.loggers.intentStream.debug("messages", { messages });
-        for await (const chunk of responseStream) {
-          const delta = chunk.delta;
-          if (_optionalChain([delta, 'optionalAccess', _5 => _5.tool_calls])) {
-            didCallTool = true;
-            for (const toolCallDelta of delta.tool_calls) {
-              const index = toolCallDelta.index;
-              if (!assembledToolCalls[index]) {
-                assembledToolCalls[index] = {
-                  id: "",
-                  type: "function",
-                  function: { name: "", arguments: "" }
-                };
-              }
-              if (toolCallDelta.id) {
-                assembledToolCalls[index].id = toolCallDelta.id;
-              }
-              if (_optionalChain([toolCallDelta, 'access', _6 => _6.function, 'optionalAccess', _7 => _7.name])) {
-                assembledToolCalls[index].function.name = toolCallDelta.function.name;
-              }
-              if (_optionalChain([toolCallDelta, 'access', _8 => _8.function, 'optionalAccess', _9 => _9.arguments])) {
-                assembledToolCalls[index].function.arguments += toolCallDelta.function.arguments;
-              }
-            }
-          } else if (_optionalChain([chunk, 'access', _10 => _10.delta, 'optionalAccess', _11 => _11.content])) {
-            yield {
-              event: "text_chunk",
-              data: { delta: chunk.delta.content }
-            };
-          }
-        }
-        _chunk466ED5WNcjs.loggers.intentStream.debug("assembledToolCalls", {
-          assembledToolCalls
-        });
-        if (didCallTool && assembledToolCalls.length > 0) {
-          const messagePayload = _optionalChain([this, 'access', _12 => _12.a2aModule, 'optionalAccess', _13 => _13.getMessagePayload, 'call', _14 => _14(
-            query,
-            sessionId
-          )]);
-          for (const toolCall of assembledToolCalls) {
-            const toolName = toolCall.function.name;
-            const toolArgs = JSON.parse(toolCall.function.arguments);
-            const selectedTool = tools.filter(
-              (tool) => tool.id === toolName
-            )[0];
-            yield { event: "tool_start", data: { toolName, toolArgs } };
-            let toolResult = "";
-            if (this.mcpModule && selectedTool.protocol === "MCP" /* MCP */) {
-              const toolArgs2 = JSON.parse(toolCall.function.arguments);
-              _chunk466ED5WNcjs.loggers.intent.debug("MCP tool call", { toolName, toolArgs: toolArgs2 });
-              toolResult = await this.mcpModule.useTool(
-                selectedTool,
-                toolArgs2
-              );
-            } else if (this.a2aModule && selectedTool.protocol === "A2A" /* A2A */) {
-              toolResult = await this.a2aModule.useTool(
-                selectedTool,
-                messagePayload,
-                sessionId
-              );
-            } else {
-              _chunk466ED5WNcjs.loggers.intent.warn(
-                `Unrecognized tool type: ${selectedTool.protocol}`
-              );
-              continue;
-            }
-            yield {
-              event: "tool_output",
-              data: { toolName, result: toolResult }
-            };
-            _chunk466ED5WNcjs.loggers.intent.debug("toolResult", { toolResult });
-            processList.push(toolResult);
-            modelInstance.appendMessages(messages, toolResult);
-          }
-        }
-        if (!didCallTool) break;
-      }
-    } catch (error) {
-      _chunk466ED5WNcjs.loggers.intent.error("Error in intentFulfilling generator", { error });
-      if (error instanceof Error) {
-        yield {
-          event: "error",
-          data: { message: error.message || "An unknown error occurred." }
-        };
-      } else {
-        yield {
-          event: "error",
-          data: { message: "An unknown error occurred." }
-        };
-      }
-    }
-  }
-  /**
-   * Generates a title for the conversation based on the query.
-   *
-   * @param query - The user's input query
-   * @returns Promise resolving to a generated title
-   */
-  async generateTitle(query) {
-    const DEFAULT_TITLE = "New Chat";
-    try {
-      const modelInstance = this.modelModule.getModel();
-      const messages = modelInstance.generateMessages({
-        query,
-        systemPrompt: `You are a helpful assistant that generates titles for conversations.
-  Please analyze the user's query and create a concise title that accurately reflects the conversation's core topic.
-  The title must be no more than 5 words long.
-  Respond with only the title. Do not include any punctuation or extra explanations.`
-      });
-      const response = await modelInstance.fetch(messages);
-      return response.content || DEFAULT_TITLE;
-    } catch (error) {
-      _chunk466ED5WNcjs.loggers.intentStream.error("Error generating title", {
-        error,
-        query
-      });
-      return DEFAULT_TITLE;
-    }
-  }
-  /**
-   * Main entry point for processing user queries.
-   *
-   * Handles the complete query lifecycle:
-   * 1. Loads session history from memory
-   * 2. Detects intent from the query
-   * 3. Fulfills the intent with AI response
-   * 4. Updates conversation history
-   *
-   * @param query - The user's input query
-   * @param sessionId - Unique session identifier
-   * @returns Object containing the AI-generated response
-   */
-  async handleQueryStream(query, userId, res, _sessionId) {
-    let sessionId = _sessionId;
-    const queryStartAt = Date.now();
-    const sessionMemory = _optionalChain([this, 'access', _15 => _15.memoryModule, 'optionalAccess', _16 => _16.getSessionMemory, 'call', _17 => _17()]);
-    const session = !userId || !sessionId ? void 0 : await _optionalChain([sessionMemory, 'optionalAccess', _18 => _18.getSession, 'call', _19 => _19(userId, sessionId)]);
-    try {
-      if (!sessionId) {
-        sessionId = _crypto.randomUUID.call(void 0, );
-        const title = await this.generateTitle(query);
-        const metadata = await _optionalChain([sessionMemory, 'optionalAccess', _20 => _20.createSession, 'call', _21 => _21(userId, sessionId, title)]) || {
-          sessionId,
-          title,
-          updatedAt: Date.now()
-        };
-        _chunk466ED5WNcjs.loggers.intentStream.info("Create new session", { metadata });
-        res.write(`event: session_id
-data: ${JSON.stringify(metadata)}
-
-`);
-      }
-    } catch (error) {
-      throw new Error("Failed to create new session");
-    }
-    const intent = this.intentTriggering(query);
-    try {
-      const stream = await this.intentFulfilling(query, sessionId, session);
-      let finalResponseText = "";
-      for await (const event of stream) {
-        if (event.event === "text_chunk" && event.data.delta) {
-          _chunk466ED5WNcjs.loggers.intentStream.debug("text_chunk", { event });
-          finalResponseText += event.data.delta;
-        }
-        const sseFormattedEvent = `event: ${event.event}
-data: ${JSON.stringify(event.data)}
-
-`;
-        res.write(sseFormattedEvent);
-      }
-      if (userId) {
-        await _optionalChain([sessionMemory, 'optionalAccess', _22 => _22.addChatToSession, 'call', _23 => _23(userId, sessionId, {
-          role: "USER" /* USER */,
-          timestamp: queryStartAt,
-          content: { type: "text", parts: [query] }
-        })]);
-        await _optionalChain([sessionMemory, 'optionalAccess', _24 => _24.addChatToSession, 'call', _25 => _25(userId, sessionId, {
-          role: "MODEL" /* MODEL */,
-          timestamp: Date.now(),
-          content: { type: "text", parts: [finalResponseText] }
-        })]);
-      }
-    } catch (error) {
-      _chunk466ED5WNcjs.loggers.intent.error("Error in handleQuery", { error });
-      res.write(
-        `event: error
-data: ${JSON.stringify({ message: "Stream failed" })}
-
-`
-      );
-    } finally {
-      res.end();
-    }
-  }
-};
-
-
-
-exports.QueryStreamService = QueryStreamService;
-//# sourceMappingURL=chunk-MSFAMZFU.cjs.map

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

@@ -1 +0,0 @@
-{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-MSFAMZFU.cjs","../../src/services/query-stream.service.ts"],"names":["toolArgs"],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACA;ACJA,gCAA2B;AA8BpB,IAAM,mBAAA,EAAN,MAAyB;AAAA,EACvB;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EAER,WAAA,CACC,WAAA,EACA,SAAA,EACA,SAAA,EACA,YAAA,EACA,OAAA,EACC;AACD,IAAA,IAAA,CAAK,YAAA,EAAc,WAAA;AACnB,IAAA,IAAA,CAAK,UAAA,EAAY,SAAA;AACjB,IAAA,IAAA,CAAK,UAAA,EAAY,SAAA;AACjB,IAAA,IAAA,CAAK,aAAA,EAAe,YAAA;AACpB,IAAA,IAAA,CAAK,QAAA,EAAU,OAAA;AAAA,EAChB;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EASA,MAAc,gBAAA,CAAiB,KAAA,EAAe;AAE7C,IAAA,OAAO,KAAA;AAAA,EACR;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAgBA,MAAA,CAAc,gBAAA,CACb,KAAA,EACA,SAAA,EACA,cAAA,EAC8B;AAC9B,IAAA,IAAI;AACH,MAAA,MAAM,aAAA,EAAe,CAAA;AAAA,SAAA,EAAA,iBACb,IAAI,IAAA,CAAK,CAAA,CAAA,CAAE,kBAAA,CAAmB,CAAC,CAAA;AAAA;AAAA,kBAExC,IAAA,mBAAK,OAAA,6BAAS,QAAA,GAAS,EAAE,CAAA;AAAA;AAAA,kBAEzB,IAAA,qBAAK,OAAA,6BAAS,SAAA,GAAU,EAAE,CAAA;AAAA,IAAA,CAAA;AAGzB,MAAA,MAAM,cAAA,EAAgB,IAAA,CAAK,WAAA,CAAY,QAAA,CAAS,CAAA;AAChD,MAAA,MAAM,SAAA,EAAW,aAAA,CAAc,gBAAA,CAAiB;AAAA,QAC/C,KAAA;AAAA,QACA,cAAA;AAAA,QACA,YAAA,EAAc,YAAA,CAAa,IAAA,CAAK;AAAA,MACjC,CAAC,CAAA;AAED,MAAA,MAAM,MAAA,EAAsB,CAAC,CAAA;AAC7B,MAAA,GAAA,CAAI,IAAA,CAAK,SAAA,EAAW;AACnB,QAAA,KAAA,CAAM,IAAA,CAAK,GAAG,IAAA,CAAK,SAAA,CAAU,QAAA,CAAS,CAAC,CAAA;AAAA,MACxC;AACA,MAAA,GAAA,CAAI,IAAA,CAAK,SAAA,EAAW;AACnB,QAAA,KAAA,CAAM,IAAA,CAAK,GAAI,MAAM,IAAA,CAAK,SAAA,CAAU,QAAA,CAAS,CAAE,CAAA;AAAA,MAChD;AACA,MAAA,OAAA,CAAQ,GAAA,CAAI,KAAK,CAAA;AACjB,MAAA,MAAM,UAAA,EAAY,aAAA,CAAc,uBAAA,CAAwB,KAAK,CAAA;AAC7D,MAAA,OAAA,CAAQ,GAAA,CAAI,SAAS,CAAA;AAErB,MAAA,MAAM,YAAA,EAAwB,CAAC,CAAA;AAC/B,MAAA,IAAI,YAAA,EAAc,KAAA;AAElB,MAAA,MAAA,CAAO,IAAA,EAAM;AACZ,QAAA,MAAM,eAAA,EACL,MAAM,aAAA,CAAc,6BAAA;AAAA,UACnB,QAAA;AAAA,UACA;AAAA,QACD,CAAA;AACD,QAAA,YAAA,EAAc,KAAA;AAEd,QAAA,MAAM,mBAAA,EAIA,CAAC,CAAA;AAEP,QAAA,yBAAA,CAAQ,YAAA,CAAa,KAAA,CAAM,UAAA,EAAY,EAAE,SAAS,CAAC,CAAA;AAEnD,QAAA,IAAA,MAAA,CAAA,MAAiB,MAAA,GAAS,cAAA,EAAgB;AACzC,UAAA,MAAM,MAAA,EAAQ,KAAA,CAAM,KAAA;AACpB,UAAA,GAAA,iBAAI,KAAA,6BAAO,YAAA,EAAY;AACtB,YAAA,YAAA,EAAc,IAAA;AACd,YAAA,IAAA,CAAA,MAAW,cAAA,GAAiB,KAAA,CAAM,UAAA,EAAY;AAC7C,cAAA,MAAM,MAAA,EAAQ,aAAA,CAAc,KAAA;AAE5B,cAAA,GAAA,CAAI,CAAC,kBAAA,CAAmB,KAAK,CAAA,EAAG;AAC/B,gBAAA,kBAAA,CAAmB,KAAK,EAAA,EAAI;AAAA,kBAC3B,EAAA,EAAI,EAAA;AAAA,kBACJ,IAAA,EAAM,UAAA;AAAA,kBACN,QAAA,EAAU,EAAE,IAAA,EAAM,EAAA,EAAI,SAAA,EAAW,GAAG;AAAA,gBACrC,CAAA;AAAA,cACD;AAEA,cAAA,GAAA,CAAI,aAAA,CAAc,EAAA,EAAI;AACrB,gBAAA,kBAAA,CAAmB,KAAK,CAAA,CAAE,GAAA,EAAK,aAAA,CAAc,EAAA;AAAA,cAC9C;AAEA,cAAA,GAAA,iBAAI,aAAA,qBAAc,QAAA,6BAAU,MAAA,EAAM;AACjC,gBAAA,kBAAA,CAAmB,KAAK,CAAA,CAAE,QAAA,CAAS,KAAA,EAClC,aAAA,CAAc,QAAA,CAAS,IAAA;AAAA,cACzB;AAEA,cAAA,GAAA,iBAAI,aAAA,qBAAc,QAAA,6BAAU,WAAA,EAAW;AACtC,gBAAA,kBAAA,CAAmB,KAAK,CAAA,CAAE,QAAA,CAAS,UAAA,GAClC,aAAA,CAAc,QAAA,CAAS,SAAA;AAAA,cACzB;AAAA,YACD;AAAA,UACD,EAAA,KAAA,GAAA,iBAAW,KAAA,uBAAM,KAAA,+BAAO,SAAA,EAAS;AAChC,YAAA,MAAM;AAAA,cACL,KAAA,EAAO,YAAA;AAAA,cACP,IAAA,EAAM,EAAE,KAAA,EAAO,KAAA,CAAM,KAAA,CAAM,QAAQ;AAAA,YACpC,CAAA;AAAA,UACD;AAAA,QACD;AAEA,QAAA,yBAAA,CAAQ,YAAA,CAAa,KAAA,CAAM,oBAAA,EAAsB;AAAA,UAChD;AAAA,QACD,CAAC,CAAA;AAED,QAAA,GAAA,CAAI,YAAA,GAAe,kBAAA,CAAmB,OAAA,EAAS,CAAA,EAAG;AACjD,UAAA,MAAM,eAAA,kBAAiB,IAAA,uBAAK,SAAA,+BAAW,iBAAA;AAAA,YACtC,KAAA;AAAA,YACA;AAAA,UACD,GAAA;AACA,UAAA,IAAA,CAAA,MAAW,SAAA,GAAY,kBAAA,EAAoB;AAC1C,YAAA,MAAM,SAAA,EAAW,QAAA,CAAS,QAAA,CAAS,IAAA;AACnC,YAAA,MAAM,SAAA,EAAW,IAAA,CAAK,KAAA,CAAM,QAAA,CAAS,QAAA,CAAS,SAAS,CAAA;AACvD,YAAA,MAAM,aAAA,EAAe,KAAA,CAAM,MAAA;AAAA,cAC1B,CAAC,IAAA,EAAA,GAAS,IAAA,CAAK,GAAA,IAAO;AAAA,YACvB,CAAA,CAAE,CAAC,CAAA;AAEH,YAAA,MAAM,EAAE,KAAA,EAAO,YAAA,EAAc,IAAA,EAAM,EAAE,QAAA,EAAU,SAAS,EAAE,CAAA;AAE1D,YAAA,IAAI,WAAA,EAAa,EAAA;AAEjB,YAAA,GAAA,CACC,IAAA,CAAK,UAAA,GACL,YAAA,CAAa,SAAA,IAAA,eAAA,EACZ;AACD,cAAA,MAAMA,UAAAA,EAAW,IAAA,CAAK,KAAA,CAAM,QAAA,CAAS,QAAA,CAAS,SAAS,CAAA;AAGvD,cAAA,yBAAA,CAAQ,MAAA,CAAO,KAAA,CAAM,eAAA,EAAiB,EAAE,QAAA,EAAU,QAAA,EAAAA,UAAS,CAAC,CAAA;AAC5D,cAAA,WAAA,EAAa,MAAM,IAAA,CAAK,SAAA,CAAU,OAAA;AAAA,gBACjC,YAAA;AAAA,gBACAA;AAAA,cACD,CAAA;AAAA,YACD,EAAA,KAAA,GAAA,CACC,IAAA,CAAK,UAAA,GACL,YAAA,CAAa,SAAA,IAAA,eAAA,EACZ;AACD,cAAA,WAAA,EAAa,MAAM,IAAA,CAAK,SAAA,CAAU,OAAA;AAAA,gBACjC,YAAA;AAAA,gBACA,cAAA;AAAA,gBACA;AAAA,cACD,CAAA;AAAA,YACD,EAAA,KAAO;AAEN,cAAA,yBAAA,CAAQ,MAAA,CAAO,IAAA;AAAA,gBACd,CAAA,wBAAA,EAA2B,YAAA,CAAa,QAAQ,CAAA;AAAA,cAAA;AAEjD,cAAA;AAAA,YAAA;AAED,YAAA;AAAM,cAAA;AACE,cAAA;AAC8B,YAAA;AAEtC,YAAA;AAEA,YAAA;AACA,YAAA;AAAiD,UAAA;AAClD,QAAA;AAGD,QAAA;AAAkB,MAAA;AACnB,IAAA;AAEA,MAAA;AACA,MAAA;AACC,QAAA;AAAM,UAAA;AACE,UAAA;AACwD,QAAA;AAChE,MAAA;AAEA,QAAA;AAAM,UAAA;AACE,UAAA;AACuC,QAAA;AAC/C,MAAA;AACD,IAAA;AACD,EAAA;AACD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAAA;AAUC,IAAA;AACA,IAAA;AACC,MAAA;AACA,MAAA;AAAgD,QAAA;AAC/C,QAAA;AACc;AAAA;AAAA,oFAAA;AAAA,MAAA;AAKf,MAAA;AACA,MAAA;AAA2B,IAAA;AAE3B,MAAA;AAAqD,QAAA;AACpD,QAAA;AACA,MAAA;AAED,MAAA;AAAO,IAAA;AACR,EAAA;AACD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAAA;AAsBC,IAAA;AACA,IAAA;AACA,IAAA;AACA,IAAA;AAKA,IAAA;AACC,MAAA;AACC,QAAA;AACA,QAAA;AAEA,QAAA;AAEE,UAAA;AACA,UAAA;AACA,UAAA;AACoB,QAAA;AAEtB,QAAA;AACA,QAAA;AAAU,MAAA;AAAoD;AAAA;AAAM,MAAA;AACrE,IAAA;AAEA,MAAA;AAA8C,IAAA;AAI/C,IAAA;AAEA,IAAA;AAEC,MAAA;AAEA,MAAA;AACA,MAAA;AACC,QAAA;AACC,UAAA;AACA,UAAA;AAAgC,QAAA;AAGjC,QAAA;AAA+C,MAAA;AAAqC;AAAA;AACpF,QAAA;AAA2B,MAAA;AAG5B,MAAA;AACC,QAAA;AAAyD,UAAA;AACxD,UAAA;AACW,UAAA;AAC6B,QAAA;AAEzC,QAAA;AAAyD,UAAA;AACxD,UAAA;AACoB,UAAA;AACgC,QAAA;AACpD,MAAA;AACF,IAAA;AAEA,MAAA;AACA,MAAA;AAAI,QAAA;AACH,MAAA;AAAmE;AAAA;AAAA,MAAA;AACpE,IAAA;AAEA,MAAA;AAAQ,IAAA;AACT,EAAA;AAEF;ADrFA;AACA;AACA;AACA","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-MSFAMZFU.cjs","sourcesContent":[null,"import { randomUUID } from \"node:crypto\";\nimport type { Response } from \"express\";\nimport type {\n\tA2AModule,\n\tMCPModule,\n\tMemoryModule,\n\tModelModule,\n} from \"@/modules/index.js\";\nimport type { AinAgentPrompts } from \"@/types/agent.js\";\nimport {\n\tChatRole,\n\ttype SessionMetadata,\n\ttype SessionObject,\n} from \"@/types/memory.js\";\nimport type { StreamEvent } from \"@/types/stream\";\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 QueryStreamService {\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\tpublic async *intentFulfilling(\n\t\tquery: string,\n\t\tsessionId: string,\n\t\tsessionHistory?: SessionObject,\n\t): AsyncGenerator<StreamEvent> {\n\t\ttry {\n\t\t\tconst systemPrompt = `\nToday is ${new Date().toLocaleDateString()}.\n\n${this.prompts?.agent || \"\"}\n\n${this.prompts?.system || \"\"}\n    `;\n\n\t\t\tconst modelInstance = this.modelModule.getModel();\n\t\t\tconst messages = modelInstance.generateMessages({\n\t\t\t\tquery,\n\t\t\t\tsessionHistory,\n\t\t\t\tsystemPrompt: systemPrompt.trim(),\n\t\t\t});\n\n\t\t\tconst tools: IAgentTool[] = [];\n\t\t\tif (this.mcpModule) {\n\t\t\t\ttools.push(...this.mcpModule.getTools());\n\t\t\t}\n\t\t\tif (this.a2aModule) {\n\t\t\t\ttools.push(...(await this.a2aModule.getTools()));\n\t\t\t}\n\t\t\tconsole.log(tools);\n\t\t\tconst functions = modelInstance.convertToolsToFunctions(tools);\n\t\t\tconsole.log(functions);\n\n\t\t\tconst processList: string[] = [];\n\t\t\tlet didCallTool = false;\n\n\t\t\twhile (true) {\n\t\t\t\tconst responseStream =\n\t\t\t\t\tawait modelInstance.fetchStreamWithContextMessage(\n\t\t\t\t\t\tmessages,\n\t\t\t\t\t\tfunctions,\n\t\t\t\t\t);\n\t\t\t\tdidCallTool = false;\n\n\t\t\t\tconst assembledToolCalls: {\n\t\t\t\t\tid: string;\n\t\t\t\t\ttype: \"function\";\n\t\t\t\t\tfunction: { name: string; arguments: string };\n\t\t\t\t}[] = [];\n\n\t\t\t\tloggers.intentStream.debug(\"messages\", { messages });\n\n\t\t\t\tfor await (const chunk of responseStream) {\n\t\t\t\t\tconst delta = chunk.delta;\n\t\t\t\t\tif (delta?.tool_calls) {\n\t\t\t\t\t\tdidCallTool = true;\n\t\t\t\t\t\tfor (const toolCallDelta of delta.tool_calls) {\n\t\t\t\t\t\t\tconst index = toolCallDelta.index;\n\n\t\t\t\t\t\t\tif (!assembledToolCalls[index]) {\n\t\t\t\t\t\t\t\tassembledToolCalls[index] = {\n\t\t\t\t\t\t\t\t\tid: \"\",\n\t\t\t\t\t\t\t\t\ttype: \"function\",\n\t\t\t\t\t\t\t\t\tfunction: { name: \"\", arguments: \"\" },\n\t\t\t\t\t\t\t\t};\n\t\t\t\t\t\t\t}\n\n\t\t\t\t\t\t\tif (toolCallDelta.id) {\n\t\t\t\t\t\t\t\tassembledToolCalls[index].id = toolCallDelta.id;\n\t\t\t\t\t\t\t}\n\n\t\t\t\t\t\t\tif (toolCallDelta.function?.name) {\n\t\t\t\t\t\t\t\tassembledToolCalls[index].function.name =\n\t\t\t\t\t\t\t\t\ttoolCallDelta.function.name;\n\t\t\t\t\t\t\t}\n\n\t\t\t\t\t\t\tif (toolCallDelta.function?.arguments) {\n\t\t\t\t\t\t\t\tassembledToolCalls[index].function.arguments +=\n\t\t\t\t\t\t\t\t\ttoolCallDelta.function.arguments;\n\t\t\t\t\t\t\t}\n\t\t\t\t\t\t}\n\t\t\t\t\t} else if (chunk.delta?.content) {\n\t\t\t\t\t\tyield {\n\t\t\t\t\t\t\tevent: \"text_chunk\",\n\t\t\t\t\t\t\tdata: { delta: chunk.delta.content },\n\t\t\t\t\t\t};\n\t\t\t\t\t}\n\t\t\t\t}\n\n\t\t\t\tloggers.intentStream.debug(\"assembledToolCalls\", {\n\t\t\t\t\tassembledToolCalls,\n\t\t\t\t});\n\n\t\t\t\tif (didCallTool && assembledToolCalls.length > 0) {\n\t\t\t\t\tconst messagePayload = this.a2aModule?.getMessagePayload(\n\t\t\t\t\t\tquery,\n\t\t\t\t\t\tsessionId,\n\t\t\t\t\t);\n\t\t\t\t\tfor (const toolCall of assembledToolCalls) {\n\t\t\t\t\t\tconst toolName = toolCall.function.name;\n\t\t\t\t\t\tconst toolArgs = JSON.parse(toolCall.function.arguments);\n\t\t\t\t\t\tconst selectedTool = tools.filter(\n\t\t\t\t\t\t\t(tool) => tool.id === toolName,\n\t\t\t\t\t\t)[0];\n\n\t\t\t\t\t\tyield { event: \"tool_start\", data: { toolName, toolArgs } };\n\n\t\t\t\t\t\tlet toolResult = \"\";\n\n\t\t\t\t\t\tif (\n\t\t\t\t\t\t\tthis.mcpModule &&\n\t\t\t\t\t\t\tselectedTool.protocol === TOOL_PROTOCOL_TYPE.MCP\n\t\t\t\t\t\t) {\n\t\t\t\t\t\t\tconst toolArgs = JSON.parse(toolCall.function.arguments) as\n\t\t\t\t\t\t\t\t| { [x: string]: unknown }\n\t\t\t\t\t\t\t\t| undefined;\n\t\t\t\t\t\t\tloggers.intent.debug(\"MCP tool call\", { toolName, toolArgs });\n\t\t\t\t\t\t\ttoolResult = await this.mcpModule.useTool(\n\t\t\t\t\t\t\t\tselectedTool as IMCPTool,\n\t\t\t\t\t\t\t\ttoolArgs,\n\t\t\t\t\t\t\t);\n\t\t\t\t\t\t} else if (\n\t\t\t\t\t\t\tthis.a2aModule &&\n\t\t\t\t\t\t\tselectedTool.protocol === TOOL_PROTOCOL_TYPE.A2A\n\t\t\t\t\t\t) {\n\t\t\t\t\t\t\ttoolResult = await this.a2aModule.useTool(\n\t\t\t\t\t\t\t\tselectedTool as IA2ATool,\n\t\t\t\t\t\t\t\tmessagePayload!,\n\t\t\t\t\t\t\t\tsessionId,\n\t\t\t\t\t\t\t);\n\t\t\t\t\t\t} else {\n\t\t\t\t\t\t\t// Unrecognized tool type. It cannot be happened...\n\t\t\t\t\t\t\tloggers.intent.warn(\n\t\t\t\t\t\t\t\t`Unrecognized tool type: ${selectedTool.protocol}`,\n\t\t\t\t\t\t\t);\n\t\t\t\t\t\t\tcontinue;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tyield {\n\t\t\t\t\t\t\tevent: \"tool_output\",\n\t\t\t\t\t\t\tdata: { toolName, result: toolResult },\n\t\t\t\t\t\t};\n\t\t\t\t\t\tloggers.intent.debug(\"toolResult\", { toolResult });\n\n\t\t\t\t\t\tprocessList.push(toolResult);\n\t\t\t\t\t\tmodelInstance.appendMessages(messages, toolResult);\n\t\t\t\t\t}\n\t\t\t\t}\n\n\t\t\t\tif (!didCallTool) break;\n\t\t\t}\n\t\t} catch (error) {\n\t\t\tloggers.intent.error(\"Error in intentFulfilling generator\", { error });\n\t\t\tif (error instanceof Error) {\n\t\t\t\tyield {\n\t\t\t\t\tevent: \"error\",\n\t\t\t\t\tdata: { message: error.message || \"An unknown error occurred.\" },\n\t\t\t\t};\n\t\t\t} else {\n\t\t\t\tyield {\n\t\t\t\t\tevent: \"error\",\n\t\t\t\t\tdata: { message: \"An unknown error occurred.\" },\n\t\t\t\t};\n\t\t\t}\n\t\t}\n\t}\n\n\t/**\n\t * Generates a title for the conversation based on the query.\n\t *\n\t * @param query - The user's input query\n\t * @returns Promise resolving to a generated title\n\t */\n\n\tprivate async generateTitle(query: string): Promise<string> {\n\t\tconst DEFAULT_TITLE = \"New Chat\";\n\t\ttry {\n\t\t\tconst modelInstance = this.modelModule.getModel();\n\t\t\tconst messages = modelInstance.generateMessages({\n\t\t\t\tquery,\n\t\t\t\tsystemPrompt: `You are a helpful assistant that generates titles for conversations.\n  Please analyze the user's query and create a concise title that accurately reflects the conversation's core topic.\n  The title must be no more than 5 words long.\n  Respond with only the title. Do not include any punctuation or extra explanations.`,\n\t\t\t});\n\t\t\tconst response = await modelInstance.fetch(messages);\n\t\t\treturn response.content || DEFAULT_TITLE;\n\t\t} catch (error) {\n\t\t\tloggers.intentStream.error(\"Error generating title\", {\n\t\t\t\terror,\n\t\t\t\tquery,\n\t\t\t});\n\t\t\treturn DEFAULT_TITLE;\n\t\t}\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 handleQueryStream(\n\t\tquery: string,\n\t\tuserId: string,\n\t\tres: Response,\n\t\t_sessionId?: string,\n\t) {\n\t\t// 1. Load session history with sessionId\n\t\tlet sessionId = _sessionId;\n\t\tconst queryStartAt = Date.now();\n\t\tconst sessionMemory = this.memoryModule?.getSessionMemory();\n\t\tconst session =\n\t\t\t!userId || !sessionId\n\t\t\t\t? undefined\n\t\t\t\t: await sessionMemory?.getSession(userId, sessionId);\n\n\t\ttry {\n\t\t\tif (!sessionId) {\n\t\t\t\tsessionId = randomUUID();\n\t\t\t\tconst title = await this.generateTitle(query);\n\n\t\t\t\tconst metadata =\n\t\t\t\t\t(await sessionMemory?.createSession(userId, sessionId, title)) ||\n\t\t\t\t\t({\n\t\t\t\t\t\tsessionId,\n\t\t\t\t\t\ttitle,\n\t\t\t\t\t\tupdatedAt: Date.now(),\n\t\t\t\t\t} as SessionMetadata);\n\t\t\t\tloggers.intentStream.info(\"Create new session\", { metadata });\n\t\t\t\tres.write(`event: session_id\\ndata: ${JSON.stringify(metadata)}\\n\\n`);\n\t\t\t}\n\t\t} catch (error) {\n\t\t\tthrow new Error(\"Failed to create new session\");\n\t\t}\n\n\t\t// 2. intent triggering\n\t\tconst intent = this.intentTriggering(query);\n\n\t\ttry {\n\t\t\t// 3. intent fulfillment\n\t\t\tconst stream = await this.intentFulfilling(query, sessionId, session);\n\n\t\t\tlet finalResponseText = \"\";\n\t\t\tfor await (const event of stream) {\n\t\t\t\tif (event.event === \"text_chunk\" && event.data.delta) {\n\t\t\t\t\tloggers.intentStream.debug(\"text_chunk\", { event });\n\t\t\t\t\tfinalResponseText += event.data.delta;\n\t\t\t\t}\n\n\t\t\t\tconst sseFormattedEvent = `event: ${event.event}\\ndata: ${JSON.stringify(event.data)}\\n\\n`;\n\t\t\t\tres.write(sseFormattedEvent);\n\t\t\t}\n\n\t\t\tif (userId) {\n\t\t\t\tawait sessionMemory?.addChatToSession(userId, sessionId, {\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\tawait sessionMemory?.addChatToSession(userId, sessionId, {\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: [finalResponseText] },\n\t\t\t\t});\n\t\t\t}\n\t\t} catch (error) {\n\t\t\tloggers.intent.error(\"Error in handleQuery\", { error });\n\t\t\tres.write(\n\t\t\t\t`event: error\\ndata: ${JSON.stringify({ message: \"Stream failed\" })}\\n\\n`,\n\t\t\t);\n\t\t} finally {\n\t\t\tres.end();\n\t\t}\n\t}\n}\n"]}

package/dist/cjs/chunk-MTAGYTHO.cjs

@@ -1,17 +0,0 @@
-"use strict";Object.defineProperty(exports, "__esModule", {value: true});
-
-var _chunkJ3BFUVZScjs = require('./chunk-J3BFUVZS.cjs');
-
-// src/routes/api/sessions.routes.ts
-var _express = require('express');
-var createSessionApiRouter = (memoryModule) => {
-  const router = _express.Router.call(void 0, );
-  const sessionApiController = new (0, _chunkJ3BFUVZScjs.SessionApiController)(memoryModule);
-  router.get("/:id", sessionApiController.handleSessionHistory);
-  return router;
-};
-
-
-
-exports.createSessionApiRouter = createSessionApiRouter;
-//# sourceMappingURL=chunk-MTAGYTHO.cjs.map

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

@@ -1 +0,0 @@
-{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-MTAGYTHO.cjs","../../src/routes/api/sessions.routes.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACA;ACJA,kCAAuB;AAIhB,IAAM,uBAAA,EAAyB,CAAC,YAAA,EAAA,GAAuC;AAC7E,EAAA,MAAM,OAAA,EAAS,6BAAA,CAAO;AACtB,EAAA,MAAM,qBAAA,EAAuB,IAAI,2CAAA,CAAqB,YAAY,CAAA;AAGlE,EAAA,MAAA,CAAO,GAAA,CAAI,MAAA,EAAQ,oBAAA,CAAqB,oBAAoB,CAAA;AAE5D,EAAA,OAAO,MAAA;AACR,CAAA;ADAA;AACA;AACE;AACF,wDAAC","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-MTAGYTHO.cjs","sourcesContent":[null,"import { Router } from \"express\";\nimport { SessionApiController } from \"@/controllers/api/session.api.controller.js\";\nimport type { MemoryModule } from \"@/modules/index.js\";\n\nexport const createSessionApiRouter = (memoryModule: MemoryModule): Router => {\n\tconst router = Router();\n\tconst sessionApiController = new SessionApiController(memoryModule);\n\n\t// APIs (prefix: /api/sessions)\n\trouter.get(\"/:id\", sessionApiController.handleSessionHistory);\n\n\treturn router;\n};\n"]}

package/dist/cjs/chunk-MWX4THZT.cjs

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

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

@@ -1 +0,0 @@
-{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-MWX4THZT.cjs","../../src/modules/mcp/mcp.module.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACA;ACPA,oEAAuB;AACvB,oEAAqC;AA0B9B,IAAM,UAAA,YAAN,MAAgB;AAAA;AAAA,iBAEd,OAAA,kBAA8B,IAAI,GAAA,CAAI,EAAA;AAAA;AAAA,kBAEtC,aAAA,kBAAkD,IAAI,GAAA,CAAI,EAAA;AAAA;AAAA,kBAE1D,MAAA,EAAmB,CAAC,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAY5B,MAAM,YAAA,CAAa,SAAA,EAAsB;AACxC,IAAA,IAAI;AACH,MAAA,IAAA,CAAA,MAAW,CAAC,IAAA,EAAM,IAAI,EAAA,GAAK,MAAA,CAAO,OAAA,CAAQ,SAAS,CAAA,EAAG;AAErD,QAAA,GAAA,CAAI,IAAA,CAAK,MAAA,CAAO,GAAA,CAAI,IAAI,EAAA,GAAK,IAAA,CAAK,YAAA,CAAa,GAAA,CAAI,IAAI,CAAA,EAAG,QAAA;AAE1D,QAAA,MAAM,UAAA,EAAY,IAAI,kCAAA,CAAqB,IAAI,CAAA;AAC/C,QAAA,IAAA,CAAK,YAAA,CAAa,GAAA,CAAI,IAAA,EAAM,SAAS,CAAA;AACrC,QAAA,MAAM,IAAA,EAAM,IAAI,oBAAA,CAAO,EAAE,IAAA,EAAM,gBAAA,EAAkB,OAAA,EAAS,QAAQ,CAAC,CAAA;AACnE,QAAA,MAAM,GAAA,CAAI,OAAA,CAAQ,SAAS,CAAA;AAC3B,QAAA,IAAA,CAAK,MAAA,CAAO,GAAA,CAAI,IAAA,EAAM,GAAG,CAAA;AAEzB,QAAA,MAAM,YAAA,EAAc,MAAM,GAAA,CAAI,SAAA,CAAU,CAAA;AACxC,QAAA,IAAA,CAAK,KAAA,CAAM,IAAA;AAAA,UACV,GAAG,WAAA,CAAY,KAAA,CAAM,GAAA,CAAI,CAAC,IAAA,EAAA,GAAS;AAClC,YAAA,OAAO,IAAI,8BAAA,CAAQ,IAAA,EAAM,IAAI,CAAA;AAAA,UAC9B,CAAC;AAAA,QACF,CAAA;AAAA,MACD;AACA,MAAA,yBAAA,CAAQ,GAAA,CAAI,IAAA,CAAK,qCAAA,EAAuC;AAAA,QACvD,KAAA,EAAO,IAAA,CAAK,KAAA,CAAM,GAAA,CAAI,CAAC,IAAA,EAAA,GAAS,IAAA,CAAK,EAAE;AAAA,MACxC,CAAC,CAAA;AAAA,IACF,EAAA,MAAA,CAAS,KAAA,EAAgB;AACxB,MAAA,yBAAA,CAAQ,GAAA,CAAI,KAAA,CAAM,kCAAA,EAAoC,EAAE,MAAM,CAAC,CAAA;AAC/D,MAAA,MAAM,KAAA;AAAA,IACP;AAAA,EACD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,QAAA,CAAA,EAAW;AACV,IAAA,OAAO,IAAA,CAAK,KAAA;AAAA,EACb;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAUA,MAAM,OAAA,CAAQ,IAAA,EAAe,KAAA,EAA2B;AACvD,IAAA,MAAM,EAAE,UAAA,EAAY,QAAQ,EAAA,EAAI,IAAA;AAChC,IAAA,MAAM,SAAA,EAAW,OAAA,CAAQ,IAAA;AACzB,IAAA,MAAM,IAAA,EAAM,IAAA,CAAK,MAAA,CAAO,GAAA,CAAI,UAAU,CAAA;AAEtC,IAAA,GAAA,CAAI,CAAC,GAAA,EAAK;AACT,MAAA,MAAM,IAAI,KAAA,CAAM,CAAA,iBAAA,EAAoB,UAAU,CAAA,CAAA,EAAI,OAAA,CAAQ,IAAI,CAAA,CAAA;AAC/D,IAAA;AAEkC,IAAA;AAC3B,MAAA;AACK,MAAA;AACX,IAAA;AAE0C,IAAA;AACJ;AAEY,IAAA;AAC5C,IAAA;AACR,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAQgB,EAAA;AACsB,IAAA;AAC1B,MAAA;AACV,IAAA;AACF,EAAA;AACD;AD3BqE;AACA;AACA;AACA","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-MWX4THZT.cjs","sourcesContent":[null,"import { Client } from \"@modelcontextprotocol/sdk/client/index.js\";\nimport { StdioClientTransport } from \"@modelcontextprotocol/sdk/client/stdio.js\";\nimport type { MCPConfig } from \"@/types/mcp.js\";\nimport { loggers } from \"@/utils/logger.js\";\nimport { MCPTool } from \"./mcp.tool.js\";\n\n/**\n * Module for managing Model Context Protocol (MCP) server connections.\n *\n * This module handles the lifecycle of MCP client connections, discovers\n * available tools from connected servers, and provides an interface for\n * executing those tools. Multiple MCP servers can be connected simultaneously.\n *\n * @example\n * ```typescript\n * const mcpModule = new MCPModule();\n * await mcpModule.addMCPConfig({\n *   \"filesystem\": {\n *     command: \"npx\",\n *     args: [\"@modelcontextprotocol/server-filesystem\", \"/path/to/files\"]\n *   }\n * });\n *\n * const tools = mcpModule.getTools();\n * const result = await mcpModule.useTool(tools[0], { path: \"/example.txt\" });\n * ```\n */\nexport class MCPModule {\n\t/** Map of MCP server names to their client instances */\n\tprivate mcpMap: Map<string, Client> = new Map();\n\t/** Map of MCP server names to their transport instances */\n\tprivate transportMap: Map<string, StdioClientTransport> = new Map();\n\t/** Array of all discovered tools from connected MCP servers */\n\tprivate tools: MCPTool[] = [];\n\n\t/**\n\t * Connects to MCP servers based on the provided configuration.\n\t *\n\t * For each server in the config, establishes a connection, discovers\n\t * available tools, and adds them to the module's tool collection.\n\t * Skips servers that are already connected.\n\t *\n\t * @param mcpConfig - Configuration object mapping server names to connection details\n\t * @throws Error if connection to any MCP server fails\n\t */\n\tasync addMCPConfig(mcpConfig: MCPConfig) {\n\t\ttry {\n\t\t\tfor (const [name, conf] of Object.entries(mcpConfig)) {\n\t\t\t\t// FIXME(yoojin): Need strict duplication check.\n\t\t\t\tif (this.mcpMap.get(name) && this.transportMap.get(name)) continue; // Duplicated mcp: skip\n\n\t\t\t\tconst transport = new StdioClientTransport(conf);\n\t\t\t\tthis.transportMap.set(name, transport);\n\t\t\t\tconst mcp = new Client({ name: \"mcp-client-cli\", version: \"1.0.0\" });\n\t\t\t\tawait mcp.connect(transport);\n\t\t\t\tthis.mcpMap.set(name, mcp);\n\n\t\t\t\tconst toolsResult = await mcp.listTools();\n\t\t\t\tthis.tools.push(\n\t\t\t\t\t...toolsResult.tools.map((tool) => {\n\t\t\t\t\t\treturn new MCPTool(name, tool);\n\t\t\t\t\t}),\n\t\t\t\t);\n\t\t\t}\n\t\t\tloggers.mcp.info(\"Connected to MCP server with tools:\", {\n\t\t\t\ttools: this.tools.map((tool) => tool.id),\n\t\t\t});\n\t\t} catch (error: unknown) {\n\t\t\tloggers.mcp.error(\"Failed to connect to MCP server:\", { error });\n\t\t\tthrow error;\n\t\t}\n\t}\n\n\t/**\n\t * Returns all available tools from connected MCP servers.\n\t *\n\t * @returns Array of MCPTool instances representing available tools\n\t */\n\tgetTools() {\n\t\treturn this.tools;\n\t}\n\n\t/**\n\t * Executes a tool on its corresponding MCP server.\n\t *\n\t * @param tool - The MCPTool instance to execute\n\t * @param _args - Arguments to pass to the tool\n\t * @returns Promise resolving to the tool's execution result\n\t * @throws Error if the MCP server for the tool is not found\n\t */\n\tasync useTool(tool: MCPTool, _args?: any): Promise<any> {\n\t\tconst { serverName, mcpTool } = tool;\n\t\tconst toolName = mcpTool.name;\n\t\tconst mcp = this.mcpMap.get(serverName);\n\n\t\tif (!mcp) {\n\t\t\tthrow new Error(`Invalid MCP Tool ${serverName}-${mcpTool.name}`);\n\t\t}\n\n\t\tconst result = await mcp.callTool({\n\t\t\tname: toolName,\n\t\t\targuments: _args,\n\t\t});\n\t\tconst toolResult =\n\t\t\t`[Bot Called Tool ${toolName} with args ${JSON.stringify(_args)}]\\n` +\n\t\t\tJSON.stringify(result.content, null, 2);\n\n\t\tloggers.mcp.debug(\"MCP useTool result:\", toolResult);\n\t\treturn result;\n\t}\n\n\t/**\n\t * Closes all MCP client connections.\n\t *\n\t * Should be called when shutting down the application to ensure\n\t * all MCP connections are properly closed.\n\t */\n\tasync cleanup() {\n\t\tthis.mcpMap.forEach((mcp: Client) => {\n\t\t\tmcp.close();\n\t\t});\n\t}\n}\n"]}

package/dist/cjs/chunk-N26HPJTO.cjs

@@ -1,27 +0,0 @@
-"use strict";Object.defineProperty(exports, "__esModule", {value: true}); var _class;// src/controllers/query.controller.ts
-var QueryController = (_class = class {
-  
-  constructor(queryService) {;_class.prototype.__init.call(this);
-    this.queryService = queryService;
-  }
-  __init() {this.handleQueryRequest = async (req, res, next) => {
-    console.log("bbbbbbbbbbbbbbbbbbbb");
-    try {
-      const { message, sessionId } = req.body;
-      const userId = res.locals.userId;
-      const result = await this.queryService.handleQuery(
-        message,
-        sessionId,
-        userId
-      );
-      res.status(200).json(result);
-    } catch (error) {
-      next(error);
-    }
-  }}
-}, _class);
-
-
-
-exports.QueryController = QueryController;
-//# sourceMappingURL=chunk-N26HPJTO.cjs.map

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

@@ -1 +0,0 @@
-{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-N26HPJTO.cjs","../../src/controllers/query.controller.ts"],"names":[],"mappings":"AAAA;ACGO,IAAM,gBAAA,YAAN,MAAsB;AAAA,EACpB;AAAA,EAER,WAAA,CAAY,YAAA,EAA4B;AACvC,IAAA,IAAA,CAAK,aAAA,EAAe,YAAA;AAAA,EACrB;AAAA,iBAEO,mBAAA,EAAqB,MAAA,CAC3B,GAAA,EACA,GAAA,EACA,IAAA,EAAA,GACI;AACJ,IAAA,OAAA,CAAQ,GAAA,CAAI,sBAAsB,CAAA;AAClC,IAAA,IAAI;AACH,MAAA,MAAM,EAAE,OAAA,EAAS,UAAU,EAAA,EAAI,GAAA,CAAI,IAAA;AACnC,MAAA,MAAM,OAAA,EAAS,GAAA,CAAI,MAAA,CAAO,MAAA;AAE1B,MAAA,MAAM,OAAA,EAAS,MAAM,IAAA,CAAK,YAAA,CAAa,WAAA;AAAA,QACtC,OAAA;AAAA,QACA,SAAA;AAAA,QACA;AAAA,MACD,CAAA;AAEA,MAAA,GAAA,CAAI,MAAA,CAAO,GAAG,CAAA,CAAE,IAAA,CAAK,MAAM,CAAA;AAAA,IAC5B,EAAA,MAAA,CAAS,KAAA,EAAO;AACf,MAAA,IAAA,CAAK,KAAK,CAAA;AAAA,IACX;AAAA,EACD,EAAA;AACD,UAAA;ADTA;AACA;AACE;AACF,0CAAC","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-N26HPJTO.cjs","sourcesContent":[null,"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\tconsole.log('bbbbbbbbbbbbbbbbbbbb');\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"]}