@ainetwork/adk 0.2.6 → 0.2.8

package/dist/cjs/{chunk-3ZFV2K7O.cjs → chunk-6WAVK3BS.cjs}

@@ -1,6 +1,6 @@
-"use strict";Object.defineProperty(exports, "__esModule", {value: true}); var _class;
+"use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _optionalChain(ops) { let lastAccessLHS = undefined; let value = ops[0]; let i = 1; while (i < ops.length) { const op = ops[i]; const fn = ops[i + 1]; i += 2; if ((op === 'optionalAccess' || op === 'optionalCall') && value == null) { return undefined; } if (op === 'access' || op === 'optionalAccess') { lastAccessLHS = value; value = fn(value); } else if (op === 'call' || op === 'optionalCall') { value = fn((...args) => value.call(lastAccessLHS, ...args)); lastAccessLHS = undefined; } } return value; } var _class;
 
-var _chunkE433ONENcjs = require('./chunk-E433ONEN.cjs');
+var _chunkCFILMIB6cjs = require('./chunk-CFILMIB6.cjs');
 
 
 var _chunk466ED5WNcjs = require('./chunk-466ED5WN.cjs');
@@ -11,21 +11,26 @@ var _ssejs = require('@modelcontextprotocol/sdk/client/sse.js');
 var _stdiojs = require('@modelcontextprotocol/sdk/client/stdio.js');
 var _streamableHttpjs = require('@modelcontextprotocol/sdk/client/streamableHttp.js');
 var _websocketjs = require('@modelcontextprotocol/sdk/client/websocket.js');
-var MCPModule = (_class = class {constructor() { _class.prototype.__init.call(this);_class.prototype.__init2.call(this);_class.prototype.__init3.call(this); }
-  __init() {this.mcpConfigs = /* @__PURE__ */ new Map()}
-  /** Map of MCP server names to their client instances */
-  __init2() {this.mcpClientMap = /* @__PURE__ */ new Map()}
-  /** Array of all discovered tools from connected MCP servers */
-  __init3() {this.tools = []}
-  addMCPServerConfig(configs) {
+var MCPModule = (_class = class {constructor() { _class.prototype.__init.call(this); }
+  __init() {this.mcpConnectors = /* @__PURE__ */ new Map()}
+  addMCPConnector(configs) {
     for (const [name, config] of Object.entries(configs)) {
-      this.mcpConfigs.set(name, config);
+      const conn = new (0, _chunkCFILMIB6cjs.MCPConnector)(name, config);
+      this.mcpConnectors.set(name, conn);
     }
   }
+  getOrCreateClient(connector) {
+    connector.client ??= new (0, _indexjs.Client)({
+      name: connector.name,
+      version: "1.0.0"
+    });
+    return connector.client;
+  }
   async connectToServers() {
-    for (const [name, config] of this.mcpConfigs.entries()) {
+    for (const [name, conn] of this.mcpConnectors.entries()) {
       try {
-        const mcpClient = new (0, _indexjs.Client)({ name, version: "1.0.0" });
+        const mcpClient = this.getOrCreateClient(conn);
+        const config = conn.config;
         switch (config.type) {
           case "stdio": {
             const transport = new (0, _stdiojs.StdioClientTransport)(config.params);
@@ -57,14 +62,19 @@ var MCPModule = (_class = class {constructor() { _class.prototype.__init.call(th
             _chunk466ED5WNcjs.loggers.mcp.error("Unsupported MCP config type");
             break;
         }
-        this.mcpClientMap.set(name, mcpClient);
         const toolList = await mcpClient.listTools();
-        const newToolList = toolList.tools.map((tool) => {
-          return new (0, _chunkE433ONENcjs.MCPTool)(name, tool);
+        conn.tools = toolList.tools.map((tool) => {
+          return {
+            toolName: `${name}-${tool.name}`,
+            // to avoid tool name duplication
+            connectorName: name,
+            protocol: "MCP" /* MCP */,
+            description: tool.description,
+            inputSchema: tool.inputSchema
+          };
         });
-        this.tools.push(...newToolList);
         _chunk466ED5WNcjs.loggers.mcp.info("Connected to MCP server with tools:", {
-          tools: newToolList.map((tool) => tool.id)
+          tools: conn.tools.map((tool) => tool.toolName)
         });
       } catch (error) {
         _chunk466ED5WNcjs.loggers.mcp.error(`Failed to connect to MCP server ${name}`, { error });
@@ -77,7 +87,11 @@ var MCPModule = (_class = class {constructor() { _class.prototype.__init.call(th
    * @returns Array of MCPTool instances representing available tools
    */
   getTools() {
-    return this.tools;
+    const allTools = [];
+    for (const conn of this.mcpConnectors.values()) {
+      allTools.push(...conn.tools);
+    }
+    return allTools;
   }
   /**
    * Executes a tool on its corresponding MCP server.
@@ -88,15 +102,15 @@ var MCPModule = (_class = class {constructor() { _class.prototype.__init.call(th
    * @throws Error if the MCP server for the tool is not found
    */
   async useTool(tool, _args) {
-    const { serverName, mcpTool } = tool;
-    const toolName = mcpTool.name;
-    const mcp = this.mcpClientMap.get(serverName);
+    const { connectorName, toolName } = tool;
+    const client = _optionalChain([this, 'access', _ => _.mcpConnectors, 'access', _2 => _2.get, 'call', _3 => _3(connectorName), 'optionalAccess', _4 => _4.client]);
     try {
-      if (!mcp) {
-        throw new Error(`Invalid MCP Tool ${serverName}-${mcpTool.name}`);
+      if (!client) {
+        throw new Error(`Invalid MCP Tool ${toolName}`);
       }
-      const result = await mcp.callTool({
-        name: toolName,
+      const mcpToolName = toolName.slice(connectorName.length + 1);
+      const result = await client.callTool({
+        name: mcpToolName,
         arguments: _args
       });
       const toolResult = `[Bot Called Tool ${toolName} with args ${JSON.stringify(_args)}]
@@ -116,13 +130,13 @@ ${typeof error === "string" ? error : JSON.stringify(error, null, 2)}`;
    * all MCP connections are properly closed.
    */
   async cleanup() {
-    this.mcpClientMap.forEach((mcpClient) => {
-      mcpClient.close();
-    });
+    for (const conn of this.mcpConnectors.values()) {
+      await _optionalChain([conn, 'access', _5 => _5.client, 'optionalAccess', _6 => _6.close, 'call', _7 => _7()]);
+    }
   }
 }, _class);
 
 
 
 exports.MCPModule = MCPModule;
-//# sourceMappingURL=chunk-3ZFV2K7O.cjs.map
+//# sourceMappingURL=chunk-6WAVK3BS.cjs.map

package/dist/cjs/chunk-6WAVK3BS.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-6WAVK3BS.cjs","../../src/modules/mcp/mcp.module.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACA;ACPA,oEAAoC;AACpC,gEAAmC;AACnC,oEAAqC;AACrC,sFAA8C;AAC9C,4EAAyC;AAgBlC,IAAM,UAAA,YAAN,MAAgB;AAAA,iBACd,cAAA,kBAA2C,IAAI,GAAA,CAAI,EAAA;AAAA,EAE3D,eAAA,CAAgB,OAAA,EAA8C;AAC7D,IAAA,IAAA,CAAA,MAAW,CAAC,IAAA,EAAM,MAAM,EAAA,GAAK,MAAA,CAAO,OAAA,CAAQ,OAAO,CAAA,EAAG;AACrD,MAAA,MAAM,KAAA,EAAO,IAAI,mCAAA,CAAa,IAAA,EAAM,MAAM,CAAA;AAC1C,MAAA,IAAA,CAAK,aAAA,CAAc,GAAA,CAAI,IAAA,EAAM,IAAI,CAAA;AAAA,IAClC;AAAA,EACD;AAAA,EAEQ,iBAAA,CAAkB,SAAA,EAAoC;AAC7D,IAAA,SAAA,CAAU,OAAA,IAAW,IAAI,oBAAA,CAAU;AAAA,MAClC,IAAA,EAAM,SAAA,CAAU,IAAA;AAAA,MAChB,OAAA,EAAS;AAAA,IACV,CAAC,CAAA;AACD,IAAA,OAAO,SAAA,CAAU,MAAA;AAAA,EAClB;AAAA,EAEA,MAAM,gBAAA,CAAA,EAAkC;AACvC,IAAA,IAAA,CAAA,MAAW,CAAC,IAAA,EAAM,IAAI,EAAA,GAAK,IAAA,CAAK,aAAA,CAAc,OAAA,CAAQ,CAAA,EAAG;AACxD,MAAA,IAAI;AACH,QAAA,MAAM,UAAA,EAAY,IAAA,CAAK,iBAAA,CAAkB,IAAI,CAAA;AAC7C,QAAA,MAAM,OAAA,EAAS,IAAA,CAAK,MAAA;AACpB,QAAA,OAAA,CAAQ,MAAA,CAAO,IAAA,EAAM;AAAA,UACpB,KAAK,OAAA,EAAS;AACb,YAAA,MAAM,UAAA,EAAY,IAAI,kCAAA,CAAqB,MAAA,CAAO,MAAM,CAAA;AACxD,YAAA,MAAM,SAAA,CAAU,OAAA,CAAQ,SAAS,CAAA;AACjC,YAAA,KAAA;AAAA,UACD;AAAA,UACA,KAAK,WAAA,EAAa;AACjB,YAAA,MAAM,UAAA,EAAY,IAAI,0CAAA,CAAyB,MAAA,CAAO,GAAG,CAAA;AACzD,YAAA,MAAM,SAAA,CAAU,OAAA,CAAQ,SAAS,CAAA;AACjC,YAAA,KAAA;AAAA,UACD;AAAA,UACA,KAAK,KAAA,EAAO;AACX,YAAA,MAAM,UAAA,EAAY,IAAI,8BAAA;AAAA,cACrB,MAAA,CAAO,GAAA;AAAA,cACP,MAAA,CAAO;AAAA,YACR,CAAA;AACA,YAAA,MAAM,SAAA,CAAU,OAAA,CAAQ,SAAS,CAAA;AACjC,YAAA,KAAA;AAAA,UACD;AAAA,UACA,KAAK,gBAAA,EAAkB;AACtB,YAAA,MAAM,UAAA,EAAY,IAAI,oDAAA;AAAA,cACrB,MAAA,CAAO,GAAA;AAAA,cACP,MAAA,CAAO;AAAA,YACR,CAAA;AACA,YAAA,MAAM,SAAA,CAAU,OAAA,CAAQ,SAAS,CAAA;AACjC,YAAA,KAAA;AAAA,UACD;AAAA,UACA,OAAA;AAEC,YAAA,yBAAA,CAAQ,GAAA,CAAI,KAAA,CAAM,6BAA6B,CAAA;AAC/C,YAAA,KAAA;AAAA,QACF;AAEA,QAAA,MAAM,SAAA,EAAW,MAAM,SAAA,CAAU,SAAA,CAAU,CAAA;AAC3C,QAAA,IAAA,CAAK,MAAA,EAAQ,QAAA,CAAS,KAAA,CAAM,GAAA,CAAI,CAAC,IAAA,EAAA,GAAS;AACzC,UAAA,OAAO;AAAA,YACN,QAAA,EAAU,CAAA,EAAA;AAAoB;AAC9B,YAAA;AACA,YAAA;AACA,YAAA;AACA,YAAA;AACD,UAAA;AACA,QAAA;AACW,QAAA;AACC,UAAA;AACZ,QAAA;AACc,MAAA;AACH,QAAA;AACb,MAAA;AACD,IAAA;AACD,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAOiC,EAAA;AACQ,IAAA;AACrB,IAAA;AACD,MAAA;AAClB,IAAA;AACO,IAAA;AACR,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAUmC,EAAA;AAC1B,IAAA;AACO,IAAA;AAEX,IAAA;AACU,MAAA;AACF,QAAA;AACX,MAAA;AAGM,MAAA;AACS,MAAA;AACR,QAAA;AACK,QAAA;AACX,MAAA;AACK,MAAA;AAEU;AACT,MAAA;AACQ,IAAA;AACH,MAAA;AACN,MAAA;AAAmG;AAClG,MAAA;AACR,IAAA;AACD,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAQgB,EAAA;AACI,IAAA;AACP,MAAA;AACZ,IAAA;AACD,EAAA;AACD;ADhBuB;AACA;AACA;AACA","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-6WAVK3BS.cjs","sourcesContent":[null,"import { Client as MCPClient } from \"@modelcontextprotocol/sdk/client/index.js\";\nimport { SSEClientTransport } from \"@modelcontextprotocol/sdk/client/sse.js\";\nimport { StdioClientTransport } from \"@modelcontextprotocol/sdk/client/stdio.js\";\nimport { StreamableHTTPClientTransport } from \"@modelcontextprotocol/sdk/client/streamableHttp.js\";\nimport { WebSocketClientTransport } from \"@modelcontextprotocol/sdk/client/websocket.js\";\nimport {\n\tCONNECTOR_PROTOCOL_TYPE,\n\ttype ConnectorTool,\n} from \"@/types/connector.js\";\nimport type { MCPConfig } from \"@/types/mcp.js\";\nimport { loggers } from \"@/utils/logger.js\";\nimport { MCPConnector } from \"./mcp.connector.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 */\nexport class MCPModule {\n\tprivate mcpConnectors: Map<string, MCPConnector> = new Map();\n\n\taddMCPConnector(configs: { [name: string]: MCPConfig }): void {\n\t\tfor (const [name, config] of Object.entries(configs)) {\n\t\t\tconst conn = new MCPConnector(name, config);\n\t\t\tthis.mcpConnectors.set(name, conn);\n\t\t}\n\t}\n\n\tprivate getOrCreateClient(connector: MCPConnector): MCPClient {\n\t\tconnector.client ??= new MCPClient({\n\t\t\tname: connector.name,\n\t\t\tversion: \"1.0.0\",\n\t\t});\n\t\treturn connector.client;\n\t}\n\n\tasync connectToServers(): Promise<void> {\n\t\tfor (const [name, conn] of this.mcpConnectors.entries()) {\n\t\t\ttry {\n\t\t\t\tconst mcpClient = this.getOrCreateClient(conn);\n\t\t\t\tconst config = conn.config;\n\t\t\t\tswitch (config.type) {\n\t\t\t\t\tcase \"stdio\": {\n\t\t\t\t\t\tconst transport = new StdioClientTransport(config.params);\n\t\t\t\t\t\tawait mcpClient.connect(transport);\n\t\t\t\t\t\tbreak;\n\t\t\t\t\t}\n\t\t\t\t\tcase \"websocket\": {\n\t\t\t\t\t\tconst transport = new WebSocketClientTransport(config.url);\n\t\t\t\t\t\tawait mcpClient.connect(transport);\n\t\t\t\t\t\tbreak;\n\t\t\t\t\t}\n\t\t\t\t\tcase \"sse\": {\n\t\t\t\t\t\tconst transport = new SSEClientTransport(\n\t\t\t\t\t\t\tconfig.url,\n\t\t\t\t\t\t\tconfig.options,\n\t\t\t\t\t\t);\n\t\t\t\t\t\tawait mcpClient.connect(transport);\n\t\t\t\t\t\tbreak;\n\t\t\t\t\t}\n\t\t\t\t\tcase \"streamableHttp\": {\n\t\t\t\t\t\tconst transport = new StreamableHTTPClientTransport(\n\t\t\t\t\t\t\tconfig.url,\n\t\t\t\t\t\t\tconfig.options,\n\t\t\t\t\t\t);\n\t\t\t\t\t\tawait mcpClient.connect(transport);\n\t\t\t\t\t\tbreak;\n\t\t\t\t\t}\n\t\t\t\t\tdefault:\n\t\t\t\t\t\t// This cannot happen.\n\t\t\t\t\t\tloggers.mcp.error(\"Unsupported MCP config type\");\n\t\t\t\t\t\tbreak;\n\t\t\t\t}\n\n\t\t\t\tconst toolList = await mcpClient.listTools();\n\t\t\t\tconn.tools = toolList.tools.map((tool) => {\n\t\t\t\t\treturn {\n\t\t\t\t\t\ttoolName: `${name}-${tool.name}`, // to avoid tool name duplication\n\t\t\t\t\t\tconnectorName: name,\n\t\t\t\t\t\tprotocol: CONNECTOR_PROTOCOL_TYPE.MCP,\n\t\t\t\t\t\tdescription: tool.description,\n\t\t\t\t\t\tinputSchema: tool.inputSchema,\n\t\t\t\t\t};\n\t\t\t\t});\n\t\t\t\tloggers.mcp.info(\"Connected to MCP server with tools:\", {\n\t\t\t\t\ttools: conn.tools.map((tool) => tool.toolName),\n\t\t\t\t});\n\t\t\t} catch (error) {\n\t\t\t\tloggers.mcp.error(`Failed to connect to MCP server ${name}`, { error });\n\t\t\t}\n\t\t}\n\t}\n\n\t/**\n\t * Returns all available tools from connected MCP servers.\n\t *\n\t * @returns Array of MCPTool instances representing available tools\n\t */\n\tgetTools(): Array<ConnectorTool> {\n\t\tconst allTools: Array<ConnectorTool> = [];\n\t\tfor (const conn of this.mcpConnectors.values()) {\n\t\t\tallTools.push(...conn.tools);\n\t\t}\n\t\treturn allTools;\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: ConnectorTool, _args?: any): Promise<string> {\n\t\tconst { connectorName, toolName } = tool;\n\t\tconst client = this.mcpConnectors.get(connectorName)?.client;\n\n\t\ttry {\n\t\t\tif (!client) {\n\t\t\t\tthrow new Error(`Invalid MCP Tool ${toolName}`);\n\t\t\t}\n\n\t\t\t// `${name}-${tool.name}` => tool.name\n\t\t\tconst mcpToolName = toolName.slice(connectorName.length + 1);\n\t\t\tconst result = await client.callTool({\n\t\t\t\tname: mcpToolName,\n\t\t\t\targuments: _args,\n\t\t\t});\n\t\t\tconst toolResult =\n\t\t\t\t`[Bot Called Tool ${toolName} with args ${JSON.stringify(_args)}]\\n` +\n\t\t\t\tJSON.stringify(result.content, null, 2);\n\t\t\treturn toolResult;\n\t\t} catch (error) {\n\t\t\tloggers.mcp.error(\"Failed to call tool\", { error });\n\t\t\tconst toolResult = `[Bot Called Tool ${toolName} with args ${JSON.stringify(_args)}]\\n${typeof error === \"string\" ? error : JSON.stringify(error, null, 2)}`;\n\t\t\treturn toolResult;\n\t\t}\n\t}\n\n\t/**\n\t * Closes all MCP client connections.\n\t *\n\t * Should be called when shutting down the application to ensure\n\t * all MCP connections are properly closed.\n\t */\n\tasync cleanup() {\n\t\tfor (const conn of this.mcpConnectors.values()) {\n\t\t\tawait conn.client?.close();\n\t\t}\n\t}\n}\n"]}

package/dist/cjs/chunk-6YHPJAGA.cjs

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

package/dist/cjs/chunk-6YHPJAGA.cjs.map

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

package/dist/cjs/chunk-7HH3N36Q.cjs

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

package/dist/cjs/chunk-7HH3N36Q.cjs.map

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

package/dist/cjs/chunk-7J5DYKDJ.cjs

@@ -0,0 +1,376 @@
+"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 _chunkQHQLU7BIcjs = require('./chunk-QHQLU7BI.cjs');
+
+
+var _chunk466ED5WNcjs = require('./chunk-466ED5WN.cjs');
+
+// src/services/query-stream.service.ts
+var _crypto = require('crypto');
+var _httpstatuscodes = require('http-status-codes');
+var QueryStreamService = class {
+  
+  
+  
+  
+  
+  constructor(modelModule, a2aModule, mcpModule, memoryModule, prompts) {
+    this.modelModule = modelModule;
+    this.a2aModule = a2aModule;
+    this.mcpModule = mcpModule;
+    this.memoryModule = memoryModule;
+    this.prompts = prompts;
+  }
+  async addToThreadMessages(thread, params) {
+    try {
+      const threadMemory = _optionalChain([this, 'access', _ => _.memoryModule, 'optionalAccess', _2 => _2.getThreadMemory, 'call', _3 => _3()]);
+      const { userId, threadId } = thread;
+      const newMessage = {
+        messageId: _crypto.randomUUID.call(void 0, ),
+        role: params.role,
+        timestamp: Date.now(),
+        content: { type: "text", parts: [params.content] },
+        metadata: params.metadata
+      };
+      thread.messages.push(newMessage);
+      await _optionalChain([threadMemory, 'optionalAccess', _4 => _4.addMessagesToThread, 'call', _5 => _5(userId, threadId, [newMessage])]);
+    } catch (error) {
+      _chunk466ED5WNcjs.loggers.intentStream.error(
+        "Error adding message to thread",
+        error
+      );
+    }
+  }
+  /**
+   * Detects the intent from context.
+   *
+   * @param query - The user's input query
+   * @param thread - The thread history
+   * @returns The detected intent
+   */
+  async intentTriggering(query, thread) {
+    const modelInstance = this.modelModule.getModel();
+    const intentMemory = _optionalChain([this, 'access', _6 => _6.memoryModule, 'optionalAccess', _7 => _7.getIntentMemory, 'call', _8 => _8()]);
+    if (!intentMemory) {
+      return [{ subquery: query }];
+    }
+    const intents = await intentMemory.listIntents();
+    if (intents.length === 0) {
+      _chunk466ED5WNcjs.loggers.intentStream.warn("No intent found");
+      return [{ subquery: query }];
+    }
+    const intentList = intents.map((intent) => `- ${intent.name}: ${intent.description}`).join("\n");
+    const threadMessages = !thread ? "" : thread.messages.sort((a, b) => a.timestamp - b.timestamp).map((message) => {
+      const role = message.role === "USER" ? "User" : message.role === "MODEL" ? "Assistant" : "System";
+      const content = Array.isArray(message.content.parts) ? message.content.parts.join(" ") : String(message.content.parts);
+      return `${role}: """${content}"""`;
+    }).join("\n");
+    const systemPrompt = `You are an expert in accurately identifying user intentions.
+
+Available intent list:
+${intentList}
+
+Please select and answer only from the above intent list. 
+Please return only the exact intent name without any additional explanations or text.`;
+    const userMessage = `The following is the conversation history with the user:
+
+${threadMessages}
+
+Last user question: "${query}"
+
+Based on the above conversation history, analyze the last user question and identify all relevant intents.
+
+Instructions:
+1. First, decompose the last user question into action-based subqueries (each representing a distinct action or task)
+2. Then, map each subquery to its corresponding intent from the available intent list
+3. For each subquery, provide a 3-4 sentence summary of what actions will be performed
+4. Multiple intents can be identified if the question covers various topics or actions
+5. Maintain the logical sequence of the original question when splitting into subqueries
+6. If any subquery doesn't match any intent in the available list, map it to "default" intent
+
+Output Format:
+Return the results as a JSON array with the following structure:
+[
+  {
+    "subquery": "<subquery_1>",
+    "intentName": "<intent_name_1>",
+		"actionPlan": "<3-4 sentence description of what will be done for this subquery>"
+	},
+  {
+    "subquery": "<subquery_2>",
+    "intentName": "<intent_name_2>",
+		"actionPlan": "<3-4 sentence description of what will be done for this subquery>"
+  },
+  ...
+]
+
+Requirements:
+- Each subquery should represent a single, actionable task or request
+- Preserve the original meaning and context when splitting queries
+- Select only from the provided intent list
+- Use "default" as the intent value for any subquery that doesn't match available intents.`;
+    const messages = modelInstance.generateMessages({
+      query: userMessage,
+      systemPrompt
+    });
+    const response = await modelInstance.fetch(messages);
+    if (!response.content) {
+      _chunk466ED5WNcjs.loggers.intent.warn("Cannot extract intent from query");
+      return [{ subquery: query }];
+    }
+    const subqueries = JSON.parse(response.content);
+    const triggeredIntent = [];
+    for (const { subquery, intentName, actionPlan } of subqueries) {
+      const intent = await intentMemory.getIntentByName(intentName);
+      triggeredIntent.push({ subquery, intent, actionPlan });
+    }
+    return triggeredIntent;
+  }
+  /**
+   * Fulfills the detected intent by generating a streaming 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
+   * - Streaming results as Server-Sent Events
+   *
+   * @param query - The user's input query
+   * @param threadId - Thread identifier for context
+   * @param thread - Previous conversation history
+   * @param intent - Optional detected intent with custom prompt
+   * @returns AsyncGenerator yielding StreamEvent objects
+   */
+  async *intentFulfilling(query, thread, intent) {
+    const systemPrompt = `
+Today is ${(/* @__PURE__ */ new Date()).toLocaleDateString()}.
+
+${_optionalChain([this, 'access', _9 => _9.prompts, 'optionalAccess', _10 => _10.agent]) || ""}
+
+${_optionalChain([this, 'access', _11 => _11.prompts, 'optionalAccess', _12 => _12.system]) || ""}
+
+${_optionalChain([intent, 'optionalAccess', _13 => _13.prompt]) || ""}
+	`.trim();
+    const modelInstance = this.modelModule.getModel();
+    const messages = modelInstance.generateMessages({
+      query,
+      thread,
+      systemPrompt: systemPrompt.trim()
+    });
+    const tools = [];
+    this.mcpModule && tools.push(...this.mcpModule.getTools());
+    this.a2aModule && tools.push(...await this.a2aModule.getTools());
+    const functions = modelInstance.convertToolsToFunctions(tools);
+    const processList = [];
+    while (true) {
+      const responseStream = await modelInstance.fetchStreamWithContextMessage(
+        messages,
+        functions
+      );
+      const assembledToolCalls = [];
+      _chunk466ED5WNcjs.loggers.intentStream.info("messages", { messages });
+      for await (const chunk of responseStream) {
+        const delta = chunk.delta;
+        if (_optionalChain([delta, 'optionalAccess', _14 => _14.tool_calls])) {
+          for (const { index, id, function: func } of delta.tool_calls) {
+            assembledToolCalls[index] ??= {
+              id: "",
+              type: "function",
+              function: { name: "", arguments: "" }
+            };
+            if (id) assembledToolCalls[index].id = id;
+            if (_optionalChain([func, 'optionalAccess', _15 => _15.name])) assembledToolCalls[index].function.name = func.name;
+            if (_optionalChain([func, 'optionalAccess', _16 => _16.arguments]))
+              assembledToolCalls[index].function.arguments += func.arguments;
+          }
+        } else if (_optionalChain([chunk, 'access', _17 => _17.delta, 'optionalAccess', _18 => _18.content])) {
+          yield {
+            event: "text_chunk",
+            data: { delta: chunk.delta.content }
+          };
+        }
+      }
+      _chunk466ED5WNcjs.loggers.intentStream.debug("assembledToolCalls", {
+        assembledToolCalls
+      });
+      if (assembledToolCalls.length > 0) {
+        for (const toolCall of assembledToolCalls) {
+          const toolCallId = _crypto.randomUUID.call(void 0, );
+          const toolName = toolCall.function.name;
+          const selectedTool = tools.filter(
+            (tool) => tool.toolName === toolName
+          )[0];
+          let toolResult = "";
+          if (this.mcpModule && selectedTool.protocol === "MCP" /* MCP */) {
+            const toolArgs = JSON.parse(toolCall.function.arguments);
+            yield {
+              event: "tool_start",
+              data: {
+                toolCallId,
+                protocol: "MCP" /* MCP */,
+                toolName,
+                toolArgs
+              }
+            };
+            _chunk466ED5WNcjs.loggers.intent.info("MCP tool call", { toolName, toolArgs });
+            toolResult = await this.mcpModule.useTool(selectedTool, toolArgs);
+          } else if (this.a2aModule && selectedTool.protocol === "A2A" /* A2A */) {
+            yield {
+              event: "tool_start",
+              data: {
+                toolCallId,
+                protocol: "A2A" /* A2A */,
+                toolName,
+                toolArgs: null
+              }
+            };
+            _chunk466ED5WNcjs.loggers.intent.info("A2A tool call", { toolName });
+            toolResult = await this.a2aModule.useTool(
+              selectedTool,
+              query,
+              thread.threadId
+            );
+          } else {
+            _chunk466ED5WNcjs.loggers.intent.warn(
+              `Unrecognized tool type: ${selectedTool.protocol}`
+            );
+            continue;
+          }
+          yield {
+            event: "tool_output",
+            data: {
+              toolCallId,
+              protocol: selectedTool.protocol,
+              toolName,
+              result: toolResult
+            }
+          };
+          _chunk466ED5WNcjs.loggers.intent.debug("toolResult", { toolResult });
+          processList.push(toolResult);
+          modelInstance.appendMessages(messages, toolResult);
+        }
+      } else {
+        break;
+      }
+    }
+  }
+  /**
+   * 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 streaming user queries.
+   *
+   * Handles the complete query lifecycle:
+   * 1. Loads or creates thread from memory
+   * 2. Detects intent from the query
+   * 3. Fulfills the intent with streaming AI response
+   * 4. Updates conversation history in real-time
+   *
+   * @param threadMetadata - Metadata containing type, userId, and optional threadId
+   * @param threadMetadata.type - The type of thread (e.g., chat, workflow)
+   * @param threadMetadata.userId - The user's unique identifier
+   * @param threadMetadata.threadId - Optional thread identifier
+   * @param query - The user's input query
+   * @returns AsyncGenerator yielding StreamEvent objects for SSE
+   */
+  async *handleQueryStream(threadMetadata, query, isA2A) {
+    const { type, userId } = threadMetadata;
+    const threadMemory = _optionalChain([this, 'access', _19 => _19.memoryModule, 'optionalAccess', _20 => _20.getThreadMemory, 'call', _21 => _21()]);
+    let threadId = threadMetadata.threadId;
+    let thread;
+    if (threadId) {
+      thread = await _optionalChain([threadMemory, 'optionalAccess', _22 => _22.getThread, 'call', _23 => _23(userId, threadId)]);
+      if (!thread && !isA2A) {
+        throw new (0, _chunkQHQLU7BIcjs.AinHttpError)(_httpstatuscodes.StatusCodes.NOT_FOUND, "Thread not found");
+      }
+    }
+    threadId ??= _crypto.randomUUID.call(void 0, );
+    if (!thread) {
+      const title = await this.generateTitle(query);
+      const metadata = await _optionalChain([threadMemory, 'optionalAccess', _24 => _24.createThread, 'call', _25 => _25(
+        type,
+        userId,
+        threadId,
+        title
+      )]) || { type, userId, threadId, title };
+      thread = { ...metadata, messages: [] };
+      _chunk466ED5WNcjs.loggers.intent.info(`Create new thread: ${threadId}`);
+      yield { event: "thread_id", data: { type, userId, threadId, title } };
+    }
+    await this.addToThreadMessages(thread, {
+      role: "USER" /* USER */,
+      content: query
+    });
+    const triggeredIntent = await this.intentTriggering(
+      query,
+      thread
+    );
+    let finalResponseText = "";
+    for (let i = 0; i < triggeredIntent.length; i++) {
+      const { subquery, intent, actionPlan } = triggeredIntent[i];
+      _chunk466ED5WNcjs.loggers.intent.info(`Process query: ${subquery}, ${_optionalChain([intent, 'optionalAccess', _26 => _26.name])}`);
+      finalResponseText !== "" && thread.messages.push({
+        messageId: _crypto.randomUUID.call(void 0, ),
+        role: "MODEL" /* MODEL */,
+        timestamp: Date.now(),
+        content: { type: "text", parts: [finalResponseText] },
+        metadata: { isThinking: true }
+      });
+      await this.addToThreadMessages(thread, {
+        role: "MODEL" /* MODEL */,
+        content: subquery,
+        metadata: {
+          isThinking: true,
+          actionPlan
+        }
+      });
+      const stream = this.intentFulfilling(
+        subquery,
+        thread,
+        intent
+      );
+      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;
+        }
+        yield event;
+      }
+    }
+    await this.addToThreadMessages(thread, {
+      role: "MODEL" /* MODEL */,
+      content: finalResponseText
+    });
+  }
+};
+
+
+
+exports.QueryStreamService = QueryStreamService;
+//# sourceMappingURL=chunk-7J5DYKDJ.cjs.map

package/dist/cjs/chunk-7J5DYKDJ.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-7J5DYKDJ.cjs","../../src/services/query-stream.service.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACA;ACPA,gCAA2B;AAC3B,oDAA4B;AAoCrB,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,EAEA,MAAc,mBAAA,CACb,MAAA,EACA,MAAA,EACC;AACD,IAAA,IAAI;AACH,MAAA,MAAM,aAAA,kBAAe,IAAA,mBAAK,YAAA,6BAAc,eAAA,mBAAgB,GAAA;AACxD,MAAA,MAAM,EAAE,MAAA,EAAQ,SAAS,EAAA,EAAI,MAAA;AAAQ,MAAA,MAAM,WAAA,EAA4B;AAAA,QACtE,SAAA,EAAW,gCAAA,CAAW;AAAA,QACtB,IAAA,EAAM,MAAA,CAAO,IAAA;AAAA,QACb,SAAA,EAAW,IAAA,CAAK,GAAA,CAAI,CAAA;AAAA,QACpB,OAAA,EAAS,EAAE,IAAA,EAAM,MAAA,EAAQ,KAAA,EAAO,CAAC,MAAA,CAAO,OAAO,EAAE,CAAA;AAAA,QACjD,QAAA,EAAU,MAAA,CAAO;AAAA,MAClB,CAAA;AACA,MAAA,MAAA,CAAO,QAAA,CAAS,IAAA,CAAK,UAAU,CAAA;AAC/B,MAAA,sBAAM,YAAA,6BAAc,mBAAA,mBAAoB,MAAA,EAAQ,QAAA,EAAU,CAAC,UAAU,CAAC,GAAA;AAAA,IACvE,EAAA,MAAA,CAAS,KAAA,EAAO;AACf,MAAA,yBAAA,CAAQ,YAAA,CAAa,KAAA;AAAA,QAAM,gCAAA;AAAA,QAC1B;AAAA,MACD,CAAA;AAAA,IACD;AAAA,EACD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EASA,MAAc,gBAAA,CACb,KAAA,EACA,MAAA,EACkC;AAClC,IAAA,MAAM,cAAA,EAAgB,IAAA,CAAK,WAAA,CAAY,QAAA,CAAS,CAAA;AAChD,IAAA,MAAM,aAAA,kBAAe,IAAA,qBAAK,YAAA,6BAAc,eAAA,mBAAgB,GAAA;AACxD,IAAA,GAAA,CAAI,CAAC,YAAA,EAAc;AAClB,MAAA,OAAO,CAAC,EAAE,QAAA,EAAU,MAAM,CAAC,CAAA;AAAA,IAC5B;AAGA,IAAA,MAAM,QAAA,EAAU,MAAM,YAAA,CAAa,WAAA,CAAY,CAAA;AAE/C,IAAA,GAAA,CAAI,OAAA,CAAQ,OAAA,IAAW,CAAA,EAAG;AACzB,MAAA,yBAAA,CAAQ,YAAA,CAAa,IAAA,CAAK,iBAAiB,CAAA;AAC3C,MAAA,OAAO,CAAC,EAAE,QAAA,EAAU,MAAM,CAAC,CAAA;AAAA,IAC5B;AAEA,IAAA,MAAM,WAAA,EAAa,OAAA,CACjB,GAAA,CAAI,CAAC,MAAA,EAAA,GAAW,CAAA,EAAA,EAAK,MAAA,CAAO,IAAI,CAAA,EAAA,EAAK,MAAA,CAAO,WAAW,CAAA,CAAA;AAOxB,IAAA;AAMxB,MAAA;AAGoB,MAAA;AAEI,MAAA;AAEpB,IAAA;AAEQ,IAAA;AAAA;AAAA;AAGX;AAAA;AAAA;AAAA,qFAAA;AAKU,IAAA;AAAA;AAEN;AAAA;AAEY,qBAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,0FAAA;AAkCsB,IAAA;AACxC,MAAA;AACP,MAAA;AACA,IAAA;AAEkD,IAAA;AAC5B,IAAA;AACgC,MAAA;AAC3B,MAAA;AAC5B,IAAA;AAE8C,IAAA;AACG,IAAA;AACc,IAAA;AACF,MAAA;AACP,MAAA;AACtD,IAAA;AAEO,IAAA;AACR,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAsB+B,EAAA;AACT,IAAA;AACmB,SAAA;AAAA;AAEf;AAAA;AAEC;AAAA;AAEN;AACd,CAAA;AAE0C,IAAA;AACA,IAAA;AAC/C,MAAA;AACA,MAAA;AACgC,MAAA;AAChC,IAAA;AAE+B,IAAA;AACyB,IAAA;AACQ,IAAA;AAEJ,IAAA;AAE9B,IAAA;AAElB,IAAA;AAC+B,MAAA;AAC1C,QAAA;AACA,QAAA;AACD,MAAA;AAMO,MAAA;AAE2C,MAAA;AAER,MAAA;AACrB,QAAA;AACG,QAAA;AACwC,UAAA;AAC/B,YAAA;AACzB,cAAA;AACE,cAAA;AAC8B,cAAA;AACrC,YAAA;AAEuC,YAAA;AACwB,YAAA;AACrD,YAAA;AAC4C,cAAA;AACvD,UAAA;AACgC,QAAA;AAC1B,UAAA;AACE,YAAA;AAC4B,YAAA;AACpC,UAAA;AACD,QAAA;AACD,MAAA;AAEiD,MAAA;AAChD,QAAA;AACA,MAAA;AAEkC,MAAA;AACS,QAAA;AACZ,UAAA;AACK,UAAA;AACR,UAAA;AACE,YAAA;AAC1B,UAAA;AAEc,UAAA;AAIf,UAAA;AACsD,YAAA;AAGjD,YAAA;AACE,cAAA;AACD,cAAA;AACL,gBAAA;AACA,gBAAA;AACA,gBAAA;AACA,gBAAA;AACD,cAAA;AACD,YAAA;AAC2D,YAAA;AACK,YAAA;AAI/D,UAAA;AACK,YAAA;AACE,cAAA;AACD,cAAA;AACL,gBAAA;AACA,gBAAA;AACA,gBAAA;AACU,gBAAA;AACX,cAAA;AACD,YAAA;AACiD,YAAA;AACf,YAAA;AACjC,cAAA;AACA,cAAA;AACO,cAAA;AACR,YAAA;AACM,UAAA;AAES,YAAA;AACkC,cAAA;AACjD,YAAA;AACA,YAAA;AACD,UAAA;AACM,UAAA;AACE,YAAA;AACD,YAAA;AACL,cAAA;AACuB,cAAA;AACvB,cAAA;AACQ,cAAA;AACT,YAAA;AACD,UAAA;AACiD,UAAA;AAEtB,UAAA;AACsB,UAAA;AAClD,QAAA;AACM,MAAA;AACN,QAAA;AACD,MAAA;AACD,IAAA;AACD,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAS4D,EAAA;AACrC,IAAA;AAClB,IAAA;AAC6C,MAAA;AACA,MAAA;AAC/C,QAAA;AACc,QAAA;AAAA;AAAA;AAAA,oFAAA;AAId,MAAA;AACkD,MAAA;AACxB,MAAA;AACZ,IAAA;AACsC,MAAA;AACpD,QAAA;AACA,QAAA;AACA,MAAA;AACM,MAAA;AACR,IAAA;AACD,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AA0B+B,EAAA;AACL,IAAA;AAC+B,IAAA;AAG1B,IAAA;AAC1B,IAAA;AACU,IAAA;AAC0C,MAAA;AAChC,MAAA;AAC0C,QAAA;AACjE,MAAA;AACD,IAAA;AAEwB,IAAA;AACX,IAAA;AACgC,MAAA;AACU,MAAA;AACrD,QAAA;AACA,QAAA;AACA,QAAA;AACA,QAAA;AACqC,MAAA;AACD,MAAA;AACe,MAAA;AACgB,MAAA;AACrE,IAAA;AAEuC,IAAA;AACtC,MAAA;AACS,MAAA;AACT,IAAA;AAG0D,IAAA;AAC1D,MAAA;AACA,MAAA;AACD,IAAA;AAGwB,IAAA;AACyB,IAAA;AACU,MAAA;AACO,MAAA;AAKhB,MAAA;AAC1B,QAAA;AACtB,QAAA;AACoB,QAAA;AACgC,QAAA;AACvB,QAAA;AAC7B,MAAA;AACsC,MAAA;AACtC,QAAA;AACS,QAAA;AACC,QAAA;AACG,UAAA;AACZ,UAAA;AACD,QAAA;AACA,MAAA;AAEmB,MAAA;AACnB,QAAA;AAAU,QAAA;AAAQ,QAAA;AACnB,MAAA;AAEoB,MAAA;AACc,MAAA;AACqB,QAAA;AACH,UAAA;AAClB,UAAA;AACjC,QAAA;AACM,QAAA;AACP,MAAA;AACD,IAAA;AAEuC,IAAA;AACtC,MAAA;AACS,MAAA;AACT,IAAA;AACF,EAAA;AACD;ADxHuF;AACA;AACA;AACA","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-7J5DYKDJ.cjs","sourcesContent":[null,"import { randomUUID } from \"node:crypto\";\nimport { StatusCodes } from \"http-status-codes\";\nimport type {\n\tA2AModule,\n\tMCPModule,\n\tMemoryModule,\n\tModelModule,\n} from \"@/modules/index.js\";\nimport { type AinAgentPrompts, AinHttpError } from \"@/types/agent.js\";\nimport {\n\tCONNECTOR_PROTOCOL_TYPE,\n\ttype ConnectorTool,\n} from \"@/types/connector.js\";\nimport {\n\ttype Intent,\n\ttype MessageObject,\n\tMessageRole,\n\tThreadMetadata,\n\ttype ThreadObject,\n\ttype ThreadType,\n} from \"@/types/memory.js\";\nimport type { StreamEvent } from \"@/types/stream\";\nimport { loggers } from \"@/utils/logger.js\";\n\ntype TriggeredIntent = {\n\tsubquery: string;\n\tintent?: Intent;\n\tactionPlan?: string;\n};\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\tprivate async addToThreadMessages(\n\t\tthread: ThreadObject,\n\t\tparams: { role: MessageRole; content: string; metadata?: any },\n\t) {\n\t\ttry {\n\t\t\tconst threadMemory = this.memoryModule?.getThreadMemory();\n\t\t\tconst { userId, threadId } = thread; const newMessage: MessageObject = {\n\t\t\t\tmessageId: randomUUID(),\n\t\t\t\trole: params.role,\n\t\t\t\ttimestamp: Date.now(),\n\t\t\t\tcontent: { type: \"text\", parts: [params.content] },\n\t\t\t\tmetadata: params.metadata,\n\t\t\t};\n\t\t\tthread.messages.push(newMessage);\n\t\t\tawait threadMemory?.addMessagesToThread(userId, threadId, [newMessage]);\n\t\t} catch (error) {\n\t\t\tloggers.intentStream.error(\"Error adding message to thread\",\n\t\t\t\terror\n\t\t\t);\n\t\t}\n\t}\n\n\t/**\n\t * Detects the intent from context.\n\t *\n\t * @param query - The user's input query\n\t * @param thread - The thread history\n\t * @returns The detected intent\n\t */\n\tprivate async intentTriggering(\n\t\tquery: string,\n\t\tthread: ThreadObject | undefined,\n\t): Promise<Array<TriggeredIntent>> {\n\t\tconst modelInstance = this.modelModule.getModel();\n\t\tconst intentMemory = this.memoryModule?.getIntentMemory();\n\t\tif (!intentMemory) {\n\t\t\treturn [{ subquery: query }];\n\t\t}\n\n\t\t// 인텐트 목록 가져오기\n\t\tconst intents = await intentMemory.listIntents();\n\n\t\tif (intents.length === 0) {\n\t\t\tloggers.intentStream.warn(\"No intent found\");\n\t\t\treturn [{ subquery: query }];\n\t\t}\n\n\t\tconst intentList = intents\n\t\t\t.map((intent) => `- ${intent.name}: ${intent.description}`)\n\t\t\t.join(\"\\n\");\n\n\t\t// Convert thread messages to a string\n\t\tconst threadMessages = !thread\n\t\t\t? \"\"\n\t\t\t: thread.messages\n\t\t\t\t\t.sort((a, b) => a.timestamp - b.timestamp)\n\t\t\t\t\t.map((message: MessageObject) => {\n\t\t\t\t\t\tconst role =\n\t\t\t\t\t\t\tmessage.role === \"USER\"\n\t\t\t\t\t\t\t\t? \"User\"\n\t\t\t\t\t\t\t\t: message.role === \"MODEL\"\n\t\t\t\t\t\t\t\t\t? \"Assistant\"\n\t\t\t\t\t\t\t\t\t: \"System\";\n\t\t\t\t\t\tconst content = Array.isArray(message.content.parts)\n\t\t\t\t\t\t\t? message.content.parts.join(\" \")\n\t\t\t\t\t\t\t: String(message.content.parts);\n\t\t\t\t\t\treturn `${role}: \"\"\"${content}\"\"\"`;\n\t\t\t\t\t})\n\t\t\t\t\t.join(\"\\n\");\n\n\t\tconst systemPrompt = `You are an expert in accurately identifying user intentions.\n\nAvailable intent list:\n${intentList}\n\nPlease select and answer only from the above intent list. \nPlease return only the exact intent name without any additional explanations or text.`;\n\n\t\tconst userMessage = `The following is the conversation history with the user:\n\n${threadMessages}\n\nLast user question: \"${query}\"\n\nBased on the above conversation history, analyze the last user question and identify all relevant intents.\n\nInstructions:\n1. First, decompose the last user question into action-based subqueries (each representing a distinct action or task)\n2. Then, map each subquery to its corresponding intent from the available intent list\n3. For each subquery, provide a 3-4 sentence summary of what actions will be performed\n4. Multiple intents can be identified if the question covers various topics or actions\n5. Maintain the logical sequence of the original question when splitting into subqueries\n6. If any subquery doesn't match any intent in the available list, map it to \"default\" intent\n\nOutput Format:\nReturn the results as a JSON array with the following structure:\n[\n  {\n    \"subquery\": \"<subquery_1>\",\n    \"intentName\": \"<intent_name_1>\",\n\t\t\"actionPlan\": \"<3-4 sentence description of what will be done for this subquery>\"\n\t},\n  {\n    \"subquery\": \"<subquery_2>\",\n    \"intentName\": \"<intent_name_2>\",\n\t\t\"actionPlan\": \"<3-4 sentence description of what will be done for this subquery>\"\n  },\n  ...\n]\n\nRequirements:\n- Each subquery should represent a single, actionable task or request\n- Preserve the original meaning and context when splitting queries\n- Select only from the provided intent list\n- Use \"default\" as the intent value for any subquery that doesn't match available intents.`;\n\n\t\tconst messages = modelInstance.generateMessages({\n\t\t\tquery: userMessage,\n\t\t\tsystemPrompt,\n\t\t});\n\n\t\tconst response = await modelInstance.fetch(messages);\n\t\tif (!response.content) {\n\t\t\tloggers.intent.warn(\"Cannot extract intent from query\");\n\t\t\treturn [{ subquery: query }];\n\t\t}\n\n\t\tconst subqueries = JSON.parse(response.content);\n\t\tconst triggeredIntent: Array<TriggeredIntent> = [];\n\t\tfor (const { subquery, intentName, actionPlan } of subqueries) {\n\t\t\tconst intent = await intentMemory.getIntentByName(intentName);\n\t\t\ttriggeredIntent.push({ subquery, intent, actionPlan });\n\t\t}\n\n\t\treturn triggeredIntent;\n\t}\n\n\t/**\n\t * Fulfills the detected intent by generating a streaming 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 * - Streaming results as Server-Sent Events\n\t *\n\t * @param query - The user's input query\n\t * @param threadId - Thread identifier for context\n\t * @param thread - Previous conversation history\n\t * @param intent - Optional detected intent with custom prompt\n\t * @returns AsyncGenerator yielding StreamEvent objects\n\t */\n\tpublic async *intentFulfilling(\n\t\tquery: string,\n\t\tthread: ThreadObject,\n\t\tintent?: Intent,\n\t): AsyncGenerator<StreamEvent> {\n\t\tconst systemPrompt = `\nToday is ${new Date().toLocaleDateString()}.\n\n${this.prompts?.agent || \"\"}\n\n${this.prompts?.system || \"\"}\n\n${intent?.prompt || \"\"}\n\t`.trim();\n\n\t\tconst modelInstance = this.modelModule.getModel();\n\t\tconst messages = modelInstance.generateMessages({\n\t\t\tquery,\n\t\t\tthread,\n\t\t\tsystemPrompt: systemPrompt.trim(),\n\t\t});\n\n\t\tconst tools: ConnectorTool[] = [];\n\t\tthis.mcpModule && tools.push(...this.mcpModule.getTools());\n\t\tthis.a2aModule && tools.push(...(await this.a2aModule.getTools()));\n\n\t\tconst functions = modelInstance.convertToolsToFunctions(tools);\n\n\t\tconst processList: string[] = [];\n\n\t\twhile (true) {\n\t\t\tconst responseStream = await modelInstance.fetchStreamWithContextMessage(\n\t\t\t\tmessages,\n\t\t\t\tfunctions,\n\t\t\t);\n\n\t\t\tconst assembledToolCalls: {\n\t\t\t\tid: string;\n\t\t\t\ttype: \"function\";\n\t\t\t\tfunction: { name: string; arguments: string };\n\t\t\t}[] = [];\n\n\t\t\tloggers.intentStream.info(\"messages\", { messages });\n\n\t\t\tfor await (const chunk of responseStream) {\n\t\t\t\tconst delta = chunk.delta;\n\t\t\t\tif (delta?.tool_calls) {\n\t\t\t\t\tfor (const { index, id, function: func } of delta.tool_calls) {\n\t\t\t\t\t\tassembledToolCalls[index] ??= {\n\t\t\t\t\t\t\tid: \"\",\n\t\t\t\t\t\t\ttype: \"function\",\n\t\t\t\t\t\t\tfunction: { name: \"\", arguments: \"\" },\n\t\t\t\t\t\t};\n\n\t\t\t\t\t\tif (id) assembledToolCalls[index].id = id;\n\t\t\t\t\t\tif (func?.name) assembledToolCalls[index].function.name = func.name;\n\t\t\t\t\t\tif (func?.arguments)\n\t\t\t\t\t\t\tassembledToolCalls[index].function.arguments += func.arguments;\n\t\t\t\t\t}\n\t\t\t\t} else if (chunk.delta?.content) {\n\t\t\t\t\tyield {\n\t\t\t\t\t\tevent: \"text_chunk\",\n\t\t\t\t\t\tdata: { delta: chunk.delta.content },\n\t\t\t\t\t};\n\t\t\t\t}\n\t\t\t}\n\n\t\t\tloggers.intentStream.debug(\"assembledToolCalls\", {\n\t\t\t\tassembledToolCalls,\n\t\t\t});\n\n\t\t\tif (assembledToolCalls.length > 0) {\n\t\t\t\tfor (const toolCall of assembledToolCalls) {\n\t\t\t\t\tconst toolCallId = randomUUID();\n\t\t\t\t\tconst toolName = toolCall.function.name;\n\t\t\t\t\tconst selectedTool = tools.filter(\n\t\t\t\t\t\t(tool) => tool.toolName === toolName,\n\t\t\t\t\t)[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 === CONNECTOR_PROTOCOL_TYPE.MCP\n\t\t\t\t\t) {\n\t\t\t\t\t\tconst toolArgs = JSON.parse(toolCall.function.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\tyield {\n\t\t\t\t\t\t\tevent: \"tool_start\",\n\t\t\t\t\t\t\tdata: {\n\t\t\t\t\t\t\t\ttoolCallId,\n\t\t\t\t\t\t\t\tprotocol: CONNECTOR_PROTOCOL_TYPE.MCP,\n\t\t\t\t\t\t\t\ttoolName,\n\t\t\t\t\t\t\t\ttoolArgs,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t};\n\t\t\t\t\t\tloggers.intent.info(\"MCP tool call\", { toolName, toolArgs });\n\t\t\t\t\t\ttoolResult = await this.mcpModule.useTool(selectedTool, toolArgs);\n\t\t\t\t\t} else if (\n\t\t\t\t\t\tthis.a2aModule &&\n\t\t\t\t\t\tselectedTool.protocol === CONNECTOR_PROTOCOL_TYPE.A2A\n\t\t\t\t\t) {\n\t\t\t\t\t\tyield {\n\t\t\t\t\t\t\tevent: \"tool_start\",\n\t\t\t\t\t\t\tdata: {\n\t\t\t\t\t\t\t\ttoolCallId,\n\t\t\t\t\t\t\t\tprotocol: CONNECTOR_PROTOCOL_TYPE.A2A,\n\t\t\t\t\t\t\t\ttoolName,\n\t\t\t\t\t\t\t\ttoolArgs: null,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t};\n\t\t\t\t\t\tloggers.intent.info(\"A2A tool call\", { toolName });\n\t\t\t\t\t\ttoolResult = await this.a2aModule.useTool(\n\t\t\t\t\t\t\tselectedTool,\n\t\t\t\t\t\t\tquery,\n\t\t\t\t\t\t\tthread.threadId,\n\t\t\t\t\t\t);\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\t\t\t\t\tyield {\n\t\t\t\t\t\tevent: \"tool_output\",\n\t\t\t\t\t\tdata: {\n\t\t\t\t\t\t\ttoolCallId,\n\t\t\t\t\t\t\tprotocol: selectedTool.protocol,\n\t\t\t\t\t\t\ttoolName,\n\t\t\t\t\t\t\tresult: toolResult,\n\t\t\t\t\t\t},\n\t\t\t\t\t};\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 {\n\t\t\t\tbreak;\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 streaming user queries.\n\t *\n\t * Handles the complete query lifecycle:\n\t * 1. Loads or creates thread from memory\n\t * 2. Detects intent from the query\n\t * 3. Fulfills the intent with streaming AI response\n\t * 4. Updates conversation history in real-time\n\t *\n\t * @param threadMetadata - Metadata containing type, userId, and optional threadId\n\t * @param threadMetadata.type - The type of thread (e.g., chat, workflow)\n\t * @param threadMetadata.userId - The user's unique identifier\n\t * @param threadMetadata.threadId - Optional thread identifier\n\t * @param query - The user's input query\n\t * @returns AsyncGenerator yielding StreamEvent objects for SSE\n\t */\n\tpublic async *handleQueryStream(\n\t\tthreadMetadata: {\n\t\t\ttype: ThreadType;\n\t\t\tuserId: string;\n\t\t\tthreadId?: string;\n\t\t},\n\t\tquery: string,\n\t\tisA2A?: boolean,\n\t): AsyncGenerator<StreamEvent> {\n\t\tconst { type, userId } = threadMetadata;\n\t\tconst threadMemory = this.memoryModule?.getThreadMemory();\n\n\t\t// 1. Load or create thread\n\t\tlet threadId = threadMetadata.threadId;\n\t\tlet thread: ThreadObject | undefined;\n\t\tif (threadId) {\n\t\t\tthread = await threadMemory?.getThread(userId, threadId);\n\t\t\tif (!thread && !isA2A) {\n\t\t\t\tthrow new AinHttpError(StatusCodes.NOT_FOUND, \"Thread not found\");\n\t\t\t}\n\t\t}\n\n\t\tthreadId ??= randomUUID();\n\t\tif (!thread) {\n\t\t\tconst title = await this.generateTitle(query);\n\t\t\tconst metadata: ThreadMetadata = (await threadMemory?.createThread(\n\t\t\t\ttype,\n\t\t\t\tuserId,\n\t\t\t\tthreadId,\n\t\t\t\ttitle,\n\t\t\t)) || { type, userId, threadId, title };\n\t\t\tthread = { ...metadata, messages: [] };\n\t\t\tloggers.intent.info(`Create new thread: ${threadId}`);\n\t\t\tyield { event: \"thread_id\", data: { type, userId, threadId, title } };\n\t\t}\n\n\t\tawait this.addToThreadMessages(thread, {\n\t\t\trole: MessageRole.USER,\n\t\t\tcontent: query,\n\t\t});\n\n\t\t// 2. intent triggering\n\t\tconst triggeredIntent: Array<TriggeredIntent> = await this.intentTriggering(\n\t\t\tquery,\n\t\t\tthread,\n\t\t);\n\n\t\t// 3. intent fulfillment\n\t\tlet finalResponseText = \"\";\n\t\tfor (let i = 0; i < triggeredIntent.length; i++) {\n\t\t\tconst { subquery, intent, actionPlan } = triggeredIntent[i];\n\t\t\tloggers.intent.info(`Process query: ${subquery}, ${intent?.name}`);\n\n\t\t\t// add subquery as user message\n\n\t\t\t// only use for inference, not stored in memory\n\t\t\tfinalResponseText !== \"\" && thread.messages.push({\n\t\t\t\tmessageId: randomUUID(),\n\t\t\t\trole: MessageRole.MODEL,\n\t\t\t\ttimestamp: Date.now(),\n\t\t\t\tcontent: { type: \"text\", parts: [finalResponseText] },\n\t\t\t\tmetadata: { isThinking: true }\n\t\t\t});\n\t\t\tawait this.addToThreadMessages(thread, {\n\t\t\t\trole: MessageRole.MODEL,\n\t\t\t\tcontent: subquery,\n\t\t\t\tmetadata: {\n\t\t\t\t\tisThinking: true,\n\t\t\t\t\tactionPlan: actionPlan,\n\t\t\t\t},\n\t\t\t});\n\n\t\t\tconst stream = this.intentFulfilling(\n\t\t\t\tsubquery, thread, intent\n\t\t\t);\n\n\t\t\tfinalResponseText = \"\";\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\t\t\t\tyield event;\n\t\t\t}\n\t\t}\n\n\t\tawait this.addToThreadMessages(thread, {\n\t\t\trole: MessageRole.MODEL,\n\t\t\tcontent: finalResponseText,\n\t\t});\n\t}\n}\n"]}