@ainetwork/adk 0.3.1 → 0.3.2

package/dist/cjs/{chunk-ZFSXQFEK.cjs → chunk-25FZXCHU.cjs}

@@ -1,9 +1,9 @@
 "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 _chunkOVTUSYSUcjs = require('./chunk-OVTUSYSU.cjs');
+var _chunk5WWZOWMAcjs = require('./chunk-5WWZOWMA.cjs');
 
 
-var _chunk466ED5WNcjs = require('./chunk-466ED5WN.cjs');
+var _chunkCLJ3DWTAcjs = require('./chunk-CLJ3DWTA.cjs');
 
 // src/modules/a2a/a2a.module.ts
 var _crypto = require('crypto');
@@ -22,7 +22,7 @@ var A2AModule = (_class = class {constructor() { _class.prototype.__init.call(th
    */
   async addA2AConnector(conns) {
     for (const [name, url] of Object.entries(conns)) {
-      const conn = new (0, _chunkOVTUSYSUcjs.A2AConnector)(name, url);
+      const conn = new (0, _chunk5WWZOWMAcjs.A2AConnector)(name, url);
       this.a2aConnectors.set(name, conn);
     }
   }
@@ -33,8 +33,10 @@ var A2AModule = (_class = class {constructor() { _class.prototype.__init.call(th
     }
     return connectors;
   }
-  getOrCreateClient(connector) {
-    connector.client ??= new (0, _client.A2AClient)(connector.url);
+  async getOrCreateClient(connector) {
+    if (!connector.client) {
+      connector.client = await new (0, _client.ClientFactory)().createFromUrl(connector.url);
+    }
     return connector.client;
   }
   /**
@@ -52,13 +54,24 @@ var A2AModule = (_class = class {constructor() { _class.prototype.__init.call(th
         continue;
       }
       try {
-        const client = this.getOrCreateClient(conn);
+        const client = await this.getOrCreateClient(conn);
         const card = await client.getAgentCard();
         const tool = {
           toolName: card.name.replaceAll(" ", "-"),
           connectorName: name,
           protocol: "A2A" /* A2A */,
-          description: card.description
+          description: card.description,
+          // add thinking_text inputSchema
+          inputSchema: {
+            type: "object",
+            properties: {
+              thinking_text: {
+                type: "string",
+                description: "\uC0AC\uC6A9\uC790\uC758 \uC694\uCCAD\uC744 \uD574\uACB0\uD558\uAE30 \uC704\uD574 \uC774 \uB3C4\uAD6C\uB97C \uC120\uD0DD\uD55C \uAD6C\uCCB4\uC801\uC778 \uC774\uC720\uC640 \uBAA9\uC801 (Why & What). \uD55C\uB450\uC904 \uC815\uB3C4\uC758 \uBD84\uB7C9\uC73C\uB85C \uC785\uB825 \uC5B8\uC5B4\uC640 \uAC19\uC740 \uC5B8\uC5B4\uB85C \uC0DD\uC131\uD55C\uB2E4."
+              }
+            },
+            required: ["thinking_text"]
+          }
         };
         tools.push(tool);
       } catch (_error) {
@@ -108,17 +121,24 @@ var A2AModule = (_class = class {constructor() { _class.prototype.__init.call(th
    * @param tool - The A2ATool instance to use
    * @param query - The message to send to the agent
    * @param threadId - The session identifier for context tracking
-   * @returns Promise resolving to array of text responses from the agent
+   * @yields StreamEvent objects for intermediate events
+   * @returns Final text response from the agent
    */
-  async useTool(tool, query, threadId) {
+  async *useTool(tool, query, threadId) {
     const finalText = [];
     const connector = this.a2aConnectors.get(tool.connectorName);
+    if (!connector) {
+      _chunkCLJ3DWTAcjs.loggers.a2a.error("Unknown agent:", { tool });
+      const toolResult = `[Bot Called A2A Tool ${tool.connectorName}]
+"Unknown agent connector"`;
+      return toolResult;
+    }
     const messagePayload = this.getMessagePayload(query, threadId);
     const params = {
       message: messagePayload
     };
     try {
-      const client = this.getOrCreateClient(connector);
+      const client = await this.getOrCreateClient(connector);
       const stream = client.sendMessageStream(params);
       for await (const event of stream) {
         if (event.kind === "status-update") {
@@ -126,9 +146,24 @@ var A2AModule = (_class = class {constructor() { _class.prototype.__init.call(th
           if (typedEvent.final && typedEvent.status.state !== "input-required") {
             this.a2aTasks.delete(threadId);
           }
-          const texts = _optionalChain([typedEvent, 'access', _ => _.status, 'access', _2 => _2.message, 'optionalAccess', _3 => _3.parts, 'access', _4 => _4.filter, 'call', _5 => _5((part) => part.kind === "text"), 'access', _6 => _6.map, 'call', _7 => _7((part) => part.text), 'access', _8 => _8.join, 'call', _9 => _9("\n")]);
-          if (texts) {
-            finalText.push(texts);
+          if (typedEvent.status.state === "working") {
+            const eventData = JSON.parse(
+              (_optionalChain([typedEvent, 'access', _ => _.status, 'access', _2 => _2.message, 'optionalAccess', _3 => _3.parts, 'access', _4 => _4[0]])).text
+            );
+            yield {
+              event: "thinking_process",
+              data: eventData
+            };
+          } else if (typedEvent.status.state === "completed") {
+            const texts = _optionalChain([typedEvent, 'access', _5 => _5.status, 'access', _6 => _6.message, 'optionalAccess', _7 => _7.parts, 'access', _8 => _8.filter, 'call', _9 => _9((part) => part.kind === "text"), 'access', _10 => _10.map, 'call', _11 => _11((part) => part.text), 'access', _12 => _12.join, 'call', _13 => _13("\n")]);
+            if (texts) {
+              finalText.push(texts);
+              yield {
+                event: "text_chunk",
+                data: { delta: texts }
+              };
+            }
+          } else {
           }
         } else if (event.kind === "message") {
           const msg = event;
@@ -142,13 +177,13 @@ var A2AModule = (_class = class {constructor() { _class.prototype.__init.call(th
             this.a2aTasks.set(threadId, task.id);
           }
         } else {
-          _chunk466ED5WNcjs.loggers.a2a.warn("Received unknown event structure from stream:", {
+          _chunkCLJ3DWTAcjs.loggers.a2a.warn("Received unknown event structure from stream:", {
             event
           });
         }
       }
     } catch (error) {
-      _chunk466ED5WNcjs.loggers.a2a.error("Error communicating with agent:", { error });
+      _chunkCLJ3DWTAcjs.loggers.a2a.error("Error communicating with agent:", { error });
       const toolResult = `[Bot Called A2A Tool ${tool.toolName}]
 ${typeof error === "string" ? error : JSON.stringify(error, null, 2)}`;
       return toolResult;
@@ -161,4 +196,4 @@ ${finalText.join("\n")}`;
 
 
 exports.A2AModule = A2AModule;
-//# sourceMappingURL=chunk-ZFSXQFEK.cjs.map
+//# sourceMappingURL=chunk-25FZXCHU.cjs.map

package/dist/cjs/chunk-25FZXCHU.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-25FZXCHU.cjs","../../src/modules/a2a/a2a.module.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACA;ACPA,gCAA2B;AAS3B,4CAAwD;AAiBjD,IAAM,UAAA,YAAN,MAAgB;AAAA;AAAA,iBAEd,cAAA,kBAA2C,IAAI,GAAA,CAAI,EAAA;AAAA;AAAA,kBAEnD,SAAA,kBAAgC,IAAI,GAAA,CAAI,EAAA;AAAA,kBACxC,QAAA,EAAkB,gCAAA,EAAW;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOrC,MAAa,eAAA,CAAgB,KAAA,EAEX;AACjB,IAAA,IAAA,CAAA,MAAW,CAAC,IAAA,EAAM,GAAG,EAAA,GAAK,MAAA,CAAO,OAAA,CAAQ,KAAK,CAAA,EAAG;AAChD,MAAA,MAAM,KAAA,EAAO,IAAI,mCAAA,CAAa,IAAA,EAAM,GAAG,CAAA;AACvC,MAAA,IAAA,CAAK,aAAA,CAAc,GAAA,CAAI,IAAA,EAAM,IAAI,CAAA;AAAA,IAClC;AAAA,EACD;AAAA,EAEO,gBAAA,CAAA,EAAyD;AAC/D,IAAA,MAAM,WAAA,EAAmD,CAAC,CAAA;AAC1D,IAAA,IAAA,CAAA,MAAW,CAAC,IAAA,EAAM,SAAS,EAAA,GAAK,IAAA,CAAK,aAAA,CAAc,OAAA,CAAQ,CAAA,EAAG;AAC7D,MAAA,UAAA,CAAW,IAAA,CAAK,EAAE,IAAA,EAAM,GAAA,EAAK,SAAA,CAAU,IAAI,CAAC,CAAA;AAAA,IAC7C;AACA,IAAA,OAAO,UAAA;AAAA,EACR;AAAA,EAEA,MAAc,iBAAA,CAAkB,SAAA,EAA6C;AAC5E,IAAA,GAAA,CAAI,CAAC,SAAA,CAAU,MAAA,EAAQ;AACtB,MAAA,SAAA,CAAU,OAAA,EAAS,MAAM,IAAI,0BAAA,CAAc,CAAA,CAAE,aAAA,CAAc,SAAA,CAAU,GAAG,CAAA;AAAA,IACzE;AACA,IAAA,OAAO,SAAA,CAAU,MAAA;AAAA,EAClB;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAUA,MAAa,QAAA,CAAA,EAAqC;AACjD,IAAA,MAAM,MAAA,EAAyB,CAAC,CAAA;AAChC,IAAA,IAAA,CAAA,MAAW,CAAC,IAAA,EAAM,IAAI,EAAA,GAAK,IAAA,CAAK,aAAA,CAAc,OAAA,CAAQ,CAAA,EAAG;AACxD,MAAA,GAAA,CAAI,CAAC,IAAA,CAAK,OAAA,EAAS;AAClB,QAAA,QAAA;AAAA,MACD;AAEA,MAAA,IAAI;AACH,QAAA,MAAM,OAAA,EAAS,MAAM,IAAA,CAAK,iBAAA,CAAkB,IAAI,CAAA;AAChD,QAAA,MAAM,KAAA,EAAkB,MAAM,MAAA,CAAO,YAAA,CAAa,CAAA;AAElD,QAAA,MAAM,KAAA,EAAsB;AAAA,UAC3B,QAAA,EAAU,IAAA,CAAK,IAAA,CAAK,UAAA,CAAW,GAAA,EAAK,GAAG,CAAA;AAAA,UACvC,aAAA,EAAe,IAAA;AAAA,UACf,QAAA,EAAA,eAAA;AAAA,UACA,WAAA,EAAa,IAAA,CAAK,WAAA;AAAA;AAAA,UAElB,WAAA,EAAa;AAAA,YACZ,IAAA,EAAM,QAAA;AAAA,YACN,UAAA,EAAY;AAAA,cACX,aAAA,EAAe;AAAA,gBACd,IAAA,EAAM,QAAA;AAAA,gBACN,WAAA,EACC;AAAA,cACF;AAAA,YACD,CAAA;AAAA,YACA,QAAA,EAAU,CAAC,eAAe;AAAA,UAC3B;AAAA,QACD,CAAA;AAEA,QAAA,KAAA,CAAM,IAAA,CAAK,IAAI,CAAA;AAAA,MAChB,EAAA,MAAA,CAAS,MAAA,EAAa;AAAA,MAEtB;AAAA,IACD;AACA,IAAA,OAAO,KAAA;AAAA,EACR;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAYO,iBAAA,CAAkB,KAAA,EAAe,QAAA,EAA2B;AAClE,IAAA,MAAM,eAAA,EAA0B;AAAA,MAC/B,SAAA,EAAW,gCAAA,CAAW;AAAA,MACtB,IAAA,EAAM,SAAA;AAAA,MACN,IAAA,EAAM,MAAA;AAAA;AAAA,MACN,QAAA,EAAU;AAAA,QACT,OAAA,EAAS,IAAA,CAAK,OAAA;AAAA,QACd,IAAA,EAAA;AAAA,MACD,CAAA;AAAA,MACA,KAAA,EAAO;AAAA,QACN;AAAA,UACC,IAAA,EAAM,MAAA;AAAA,UACN,IAAA,EAAM;AAAA,QACP;AAAA,MACD,CAAA;AAAA,MACA,SAAA,EAAW;AAAA,IACZ,CAAA;AAEA,IAAA,GAAA,CAAI,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,QAAQ,CAAA,EAAG;AAChC,MAAA,cAAA,CAAe,OAAA,EAAS,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,QAAQ,CAAA;AAAA,IACnD;AAEA,IAAA,OAAO,cAAA;AAAA,EACR;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAcA,MAAA,CAAc,OAAA,CACb,IAAA,EACA,KAAA,EACA,QAAA,EAC+C;AAC/C,IAAA,MAAM,UAAA,EAAsB,CAAC,CAAA;AAC7B,IAAA,MAAM,UAAA,EAAY,IAAA,CAAK,aAAA,CAAc,GAAA,CAAI,IAAA,CAAK,aAAa,CAAA;AAC3D,IAAA,GAAA,CAAI,CAAC,SAAA,EAAW;AACf,MAAA,yBAAA,CAAQ,GAAA,CAAI,KAAA,CAAM,gBAAA,EAAkB,EAAE,KAAK,CAAC,CAAA;AAC5C,MAAA,MAAM,WAAA,EAAa,CAAA,qBAAA,EAAwB,IAAA,CAAK,aAAa,CAAA;AAAA,yBAAA,CAAA;AAC7D,MAAA,OAAO,UAAA;AAAA,IACR;AAEA,IAAA,MAAM,eAAA,EAAiB,IAAA,CAAK,iBAAA,CAAkB,KAAA,EAAO,QAAQ,CAAA;AAC7D,IAAA,MAAM,OAAA,EAA4B;AAAA,MACjC,OAAA,EAAS;AAAA,IACV,CAAA;AAEA,IAAA,IAAI;AACH,MAAA,MAAM,OAAA,EAAS,MAAM,IAAA,CAAK,iBAAA,CAAkB,SAAS,CAAA;AACrD,MAAA,MAAM,OAAA,EAAS,MAAA,CAAO,iBAAA,CAAkB,MAAM,CAAA;AAC9C,MAAA,IAAA,MAAA,CAAA,MAAiB,MAAA,GAAS,MAAA,EAAQ;AACjC,QAAA,GAAA,CAAI,KAAA,CAAM,KAAA,IAAS,eAAA,EAAiB;AACnC,UAAA,MAAM,WAAA,EAAa,KAAA;AACnB,UAAA,GAAA,CACC,UAAA,CAAW,MAAA,GACX,UAAA,CAAW,MAAA,CAAO,MAAA,IAAU,gBAAA,EAC3B;AACD,YAAA,IAAA,CAAK,QAAA,CAAS,MAAA,CAAO,QAAQ,CAAA;AAAA,UAC9B;AAEA,UAAA,GAAA,CAAI,UAAA,CAAW,MAAA,CAAO,MAAA,IAAU,SAAA,EAAW;AAE1C,YAAA,MAAM,UAAA,EAAY,IAAA,CAAK,KAAA;AAAA,cAAA,iBACrB,UAAA,mBAAW,MAAA,qBAAO,OAAA,6BAAS,KAAA,qBAAM,CAAC,GAAA,CAAA,CAAe;AAAA,YACnD,CAAA;AACA,YAAA,MAAM;AAAA,cACL,KAAA,EAAO,kBAAA;AAAA,cACP,IAAA,EAAM;AAAA,YACP,CAAA;AAAA,UACD,EAAA,KAAA,GAAA,CAAW,UAAA,CAAW,MAAA,CAAO,MAAA,IAAU,WAAA,EAAa;AAEnD,YAAA,MAAM,MAAA,kBAAQ,UAAA,qBAAW,MAAA,qBAAO,OAAA,6BAAS,KAAA,qBACvC,MAAA,mBAAO,CAAC,IAAA,EAAA,GAAS,IAAA,CAAK,KAAA,IAAS,MAAM,CAAA,uBACrC,GAAA,qBAAI,CAAC,IAAA,EAAA,GAAmB,IAAA,CAAK,IAAI,CAAA,uBACjC,IAAA,qBAAK,IAAI,GAAA;AACX,YAAA,GAAA,CAAI,KAAA,EAAO;AACV,cAAA,SAAA,CAAU,IAAA,CAAK,KAAK,CAAA;AACpB,cAAA,MAAM;AAAA,gBACL,KAAA,EAAO,YAAA;AAAA,gBACP,IAAA,EAAM,EAAE,KAAA,EAAO,MAAM;AAAA,cACtB,CAAA;AAAA,YACD;AAAA,UACD,EAAA,KAAO;AAAA,UAEP;AAAA,QACD,EAAA,KAAA,GAAA,CAAW,KAAA,CAAM,KAAA,IAAS,SAAA,EAAW;AAEpC,UAAA,MAAM,IAAA,EAAM,KAAA;AACZ,UAAA,MAAM,OAAA,EAAS,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,QAAQ,CAAA;AACzC,UAAA,GAAA,CAAI,GAAA,CAAI,OAAA,GAAU,GAAA,CAAI,OAAA,IAAW,MAAA,EAAQ;AACxC,YAAA,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,QAAA,EAAU,GAAA,CAAI,MAAM,CAAA;AAAA,UACvC;AAAA,QACD,EAAA,KAAA,GAAA,CAAW,KAAA,CAAM,KAAA,IAAS,MAAA,EAAQ;AAEjC,UAAA,MAAM,KAAA,EAAO,KAAA;AACb,UAAA,GAAA,CAAI,IAAA,CAAK,GAAA,IAAO,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,QAAQ,CAAA,EAAG;AAC5C,YAAA,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,QAAA,EAAU,IAAA,CAAK,EAAE,CAAA;AAAA,UACpC;AAAA,QACD,EAAA,KAAO;AACN,UAAA,yBAAA,CAAQ,GAAA,CAAI,IAAA,CAAK,+CAAA,EAAiD;AAAA,YACjE;AAAA,UACD,CAAC,CAAA;AAAA,QACF;AAAA,MACD;AAAA,IACD,EAAA,MAAA,CAAS,KAAA,EAAO;AACf,MAAA,yBAAA,CAAQ,GAAA,CAAI,KAAA,CAAM,iCAAA,EAAmC,EAAE,MAAM,CAAC,CAAA;AAC9D,MAAA,MAAM,WAAA,EAAa,CAAA,qBAAA,EAAwB,IAAA,CAAK,QAAQ,CAAA;AAAA,EAAM,OAAO,MAAA,IAAU,SAAA,EAAW,MAAA,EAAQ,IAAA,CAAK,SAAA,CAAU,KAAA,EAAO,IAAA,EAAM,CAAC,CAAC,CAAA,CAAA;AACzH,MAAA;AACR,IAAA;AAE4C,IAAA;AAA0B;AACvE,EAAA;AACD;AD5CqE;AACA;AACA;AACA","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-25FZXCHU.cjs","sourcesContent":[null,"import { randomUUID } from \"node:crypto\";\nimport type {\n\tAgentCard,\n\tMessage,\n\tMessageSendParams,\n\tTask,\n\tTaskStatusUpdateEvent,\n\tTextPart,\n} from \"@a2a-js/sdk\";\nimport { type Client as A2AClient, ClientFactory } from \"@a2a-js/sdk/client\";\nimport {\n\tCONNECTOR_PROTOCOL_TYPE,\n\ttype ConnectorTool,\n} from \"@/types/connector.js\";\nimport { ThreadType } from \"@/types/memory.js\";\nimport type { StreamEvent } from \"@/types/stream.js\";\nimport { loggers } from \"@/utils/logger.js\";\nimport { A2AConnector } from \"./a2a.connector.js\";\n\n/**\n * Module for managing Agent-to-Agent (A2A) protocol connections.\n *\n * This module handles connections to other A2A-compatible agents, manages\n * conversation sessions, and provides an interface for inter-agent communication.\n * Supports multi-turn conversations with task and context tracking.\n */\nexport class A2AModule {\n\t/** Map of A2A server URLs to their corresponding tool instances */\n\tprivate a2aConnectors: Map<string, A2AConnector> = new Map();\n\t/** Map of session IDs to their A2A session state */\n\tprivate a2aTasks: Map<string, string> = new Map();\n\tprivate agentId: string = randomUUID(); /* FIXME */\n\n\t/**\n\t * Registers a new A2A peer server URL for connection.\n\t *\n\t * @param conns - Set of name, url pair\n\t */\n\tpublic async addA2AConnector(conns: {\n\t\t[name: string]: string;\n\t}): Promise<void> {\n\t\tfor (const [name, url] of Object.entries(conns)) {\n\t\t\tconst conn = new A2AConnector(name, url);\n\t\t\tthis.a2aConnectors.set(name, conn);\n\t\t}\n\t}\n\n\tpublic getA2AConnectors(): Array<{ name: string; url: string }> {\n\t\tconst connectors: Array<{ name: string; url: string }> = [];\n\t\tfor (const [name, connector] of this.a2aConnectors.entries()) {\n\t\t\tconnectors.push({ name, url: connector.url });\n\t\t}\n\t\treturn connectors;\n\t}\n\n\tprivate async getOrCreateClient(connector: A2AConnector): Promise<A2AClient> {\n\t\tif (!connector.client) {\n\t\t\tconnector.client = await new ClientFactory().createFromUrl(connector.url);\n\t\t}\n\t\treturn connector.client;\n\t}\n\n\t/**\n\t * Retrieves tools from all registered A2A peer servers.\n\t *\n\t * Attempts to connect to each registered server, fetch their agent cards,\n\t * and create tool instances. Disables tools for unreachable servers.\n\t *\n\t * @returns Promise resolving to array of available A2A tools\n\t */\n\tpublic async getTools(): Promise<ConnectorTool[]> {\n\t\tconst tools: ConnectorTool[] = [];\n\t\tfor (const [name, conn] of this.a2aConnectors.entries()) {\n\t\t\tif (!conn.enabled) {\n\t\t\t\tcontinue; // skip disabled agent\n\t\t\t}\n\n\t\t\ttry {\n\t\t\t\tconst client = await this.getOrCreateClient(conn);\n\t\t\t\tconst card: AgentCard = await client.getAgentCard();\n\t\t\t\t/* TODO: add each skill as tool? */\n\t\t\t\tconst tool: ConnectorTool = {\n\t\t\t\t\ttoolName: card.name.replaceAll(\" \", \"-\"),\n\t\t\t\t\tconnectorName: name,\n\t\t\t\t\tprotocol: CONNECTOR_PROTOCOL_TYPE.A2A,\n\t\t\t\t\tdescription: card.description,\n\t\t\t\t\t// add thinking_text inputSchema\n\t\t\t\t\tinputSchema: {\n\t\t\t\t\t\ttype: \"object\",\n\t\t\t\t\t\tproperties: {\n\t\t\t\t\t\t\tthinking_text: {\n\t\t\t\t\t\t\t\ttype: \"string\",\n\t\t\t\t\t\t\t\tdescription:\n\t\t\t\t\t\t\t\t\t\"사용자의 요청을 해결하기 위해 이 도구를 선택한 구체적인 이유와 목적 (Why & What). 한두줄 정도의 분량으로 입력 언어와 같은 언어로 생성한다.\",\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t},\n\t\t\t\t\t\trequired: [\"thinking_text\"],\n\t\t\t\t\t},\n\t\t\t\t};\n\n\t\t\t\ttools.push(tool);\n\t\t\t} catch (_error: any) {\n\t\t\t\t// Agent not responded, just skip\n\t\t\t}\n\t\t}\n\t\treturn tools;\n\t}\n\n\t/**\n\t * Constructs a message payload for A2A communication.\n\t *\n\t * Includes session context (task ID and context ID) if available\n\t * for maintaining conversation continuity.\n\t *\n\t * @param query - The message content to send\n\t * @param threadId - The session identifier\n\t * @returns Formatted Message object for A2A protocol\n\t */\n\tpublic getMessagePayload(query: string, threadId: string): Message {\n\t\tconst messagePayload: Message = {\n\t\t\tmessageId: randomUUID(),\n\t\t\tkind: \"message\",\n\t\t\trole: \"user\", // FIXME: it could be 'agent'\n\t\t\tmetadata: {\n\t\t\t\tagentId: this.agentId,\n\t\t\t\ttype: ThreadType.CHAT,\n\t\t\t},\n\t\t\tparts: [\n\t\t\t\t{\n\t\t\t\t\tkind: \"text\",\n\t\t\t\t\ttext: query,\n\t\t\t\t},\n\t\t\t],\n\t\t\tcontextId: threadId,\n\t\t};\n\n\t\tif (this.a2aTasks.has(threadId)) {\n\t\t\tmessagePayload.taskId = this.a2aTasks.get(threadId);\n\t\t}\n\n\t\treturn messagePayload;\n\t}\n\n\t/**\n\t * Executes an A2A tool by sending a message to the remote agent.\n\t *\n\t * Handles streaming responses, maintains session state, and extracts\n\t * text content from various event types in the response stream.\n\t *\n\t * @param tool - The A2ATool instance to use\n\t * @param query - The message to send to the agent\n\t * @param threadId - The session identifier for context tracking\n\t * @yields StreamEvent objects for intermediate events\n\t * @returns Final text response from the agent\n\t */\n\tpublic async *useTool(\n\t\ttool: ConnectorTool,\n\t\tquery: string,\n\t\tthreadId: string,\n\t): AsyncGenerator<StreamEvent, string, unknown> {\n\t\tconst finalText: string[] = [];\n\t\tconst connector = this.a2aConnectors.get(tool.connectorName);\n\t\tif (!connector) {\n\t\t\tloggers.a2a.error(\"Unknown agent:\", { tool });\n\t\t\tconst toolResult = `[Bot Called A2A Tool ${tool.connectorName}]\\n\"Unknown agent connector\"`;\n\t\t\treturn toolResult;\n\t\t}\n\n\t\tconst messagePayload = this.getMessagePayload(query, threadId);\n\t\tconst params: MessageSendParams = {\n\t\t\tmessage: messagePayload,\n\t\t};\n\n\t\ttry {\n\t\t\tconst client = await this.getOrCreateClient(connector);\n\t\t\tconst stream = client.sendMessageStream(params);\n\t\t\tfor await (const event of stream) {\n\t\t\t\tif (event.kind === \"status-update\") {\n\t\t\t\t\tconst typedEvent = event as TaskStatusUpdateEvent;\n\t\t\t\t\tif (\n\t\t\t\t\t\ttypedEvent.final &&\n\t\t\t\t\t\ttypedEvent.status.state !== \"input-required\"\n\t\t\t\t\t) {\n\t\t\t\t\t\tthis.a2aTasks.delete(threadId);\n\t\t\t\t\t}\n\n\t\t\t\t\tif (typedEvent.status.state === \"working\") {\n\t\t\t\t\t\t// thinking process event\n\t\t\t\t\t\tconst eventData = JSON.parse(\n\t\t\t\t\t\t\t(typedEvent.status.message?.parts[0] as TextPart).text,\n\t\t\t\t\t\t);\n\t\t\t\t\t\tyield {\n\t\t\t\t\t\t\tevent: \"thinking_process\",\n\t\t\t\t\t\t\tdata: eventData,\n\t\t\t\t\t\t};\n\t\t\t\t\t} else if (typedEvent.status.state === \"completed\") {\n\t\t\t\t\t\t// TODO: handle 'file', 'data' parts\n\t\t\t\t\t\tconst texts = typedEvent.status.message?.parts\n\t\t\t\t\t\t\t.filter((part) => part.kind === \"text\")\n\t\t\t\t\t\t\t.map((part: TextPart) => part.text)\n\t\t\t\t\t\t\t.join(\"\\n\");\n\t\t\t\t\t\tif (texts) {\n\t\t\t\t\t\t\tfinalText.push(texts);\n\t\t\t\t\t\t\tyield {\n\t\t\t\t\t\t\t\tevent: \"text_chunk\",\n\t\t\t\t\t\t\t\tdata: { delta: texts },\n\t\t\t\t\t\t\t};\n\t\t\t\t\t\t}\n\t\t\t\t\t} else {\n\t\t\t\t\t\t// ignore other status updates\n\t\t\t\t\t}\n\t\t\t\t} else if (event.kind === \"message\") {\n\t\t\t\t\t// FIXME: handling text in 'message'?\n\t\t\t\t\tconst msg = event as Message;\n\t\t\t\t\tconst taskId = this.a2aTasks.get(threadId);\n\t\t\t\t\tif (msg.taskId && msg.taskId !== taskId) {\n\t\t\t\t\t\tthis.a2aTasks.set(threadId, msg.taskId);\n\t\t\t\t\t}\n\t\t\t\t} else if (event.kind === \"task\") {\n\t\t\t\t\t// establishing the Task ID\n\t\t\t\t\tconst task = event as Task;\n\t\t\t\t\tif (task.id !== this.a2aTasks.get(threadId)) {\n\t\t\t\t\t\tthis.a2aTasks.set(threadId, task.id);\n\t\t\t\t\t}\n\t\t\t\t} else {\n\t\t\t\t\tloggers.a2a.warn(\"Received unknown event structure from stream:\", {\n\t\t\t\t\t\tevent,\n\t\t\t\t\t});\n\t\t\t\t}\n\t\t\t}\n\t\t} catch (error) {\n\t\t\tloggers.a2a.error(\"Error communicating with agent:\", { error });\n\t\t\tconst toolResult = `[Bot Called A2A Tool ${tool.toolName}]\\n${typeof error === \"string\" ? error : JSON.stringify(error, null, 2)}`;\n\t\t\treturn toolResult;\n\t\t}\n\n\t\treturn `[Bot Called A2A Tool ${tool.toolName}]\\n${finalText.join(\"\\n\")}`;\n\t}\n}\n"]}

package/dist/cjs/{chunk-7QIF3FVV.cjs → chunk-3RS6NVSN.cjs}

@@ -1,18 +1,18 @@
 "use strict";Object.defineProperty(exports, "__esModule", {value: true});
 
-var _chunkLKVNLT2Mcjs = require('./chunk-LKVNLT2M.cjs');
+var _chunkHF7GKQ6Ycjs = require('./chunk-HF7GKQ6Y.cjs');
 
 
-var _chunkAQXHOGJDcjs = require('./chunk-AQXHOGJD.cjs');
+var _chunk5QSLQX4Hcjs = require('./chunk-5QSLQX4H.cjs');
 
 
-var _chunkM7K34W2Hcjs = require('./chunk-M7K34W2H.cjs');
+var _chunkNNYPZA6Wcjs = require('./chunk-NNYPZA6W.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, _chunkLKVNLT2Mcjs.QueryService)(
+  const queryService = new (0, _chunkHF7GKQ6Ycjs.QueryService)(
     agent.modelModule,
     agent.a2aModule,
     agent.mcpModule,
@@ -20,14 +20,14 @@ var createQueryRouter = (agent, allowStream = false) => {
   );
   let queryStreamService;
   if (allowStream) {
-    queryStreamService = new (0, _chunkM7K34W2Hcjs.QueryStreamService)(
+    queryStreamService = new (0, _chunk5QSLQX4Hcjs.QueryStreamService)(
       agent.modelModule,
       agent.a2aModule,
       agent.mcpModule,
       agent.memoryModule
     );
   }
-  const queryController = new (0, _chunkAQXHOGJDcjs.QueryController)(queryService, queryStreamService);
+  const queryController = new (0, _chunkNNYPZA6Wcjs.QueryController)(queryService, queryStreamService);
   router.post("/", queryController.handleQueryRequest);
   if (allowStream) {
     router.post("/stream", queryController.handleQueryStreamRequest);
@@ -38,4 +38,4 @@ var createQueryRouter = (agent, allowStream = false) => {
 
 
 exports.createQueryRouter = createQueryRouter;
-//# sourceMappingURL=chunk-7QIF3FVV.cjs.map
+//# sourceMappingURL=chunk-3RS6NVSN.cjs.map

package/dist/cjs/{chunk-7QIF3FVV.cjs.map → chunk-3RS6NVSN.cjs.map}

@@ -1 +1 @@
-{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-7QIF3FVV.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;AAAA,EACP,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;AAAA,IACP,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-7QIF3FVV.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);\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);\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"]}
+{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-3RS6NVSN.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;AAAA,EACP,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;AAAA,IACP,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-3RS6NVSN.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);\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);\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-6WAVK3BS.cjs → chunk-4IBKK5HW.cjs}

@@ -3,7 +3,7 @@
 var _chunkCFILMIB6cjs = require('./chunk-CFILMIB6.cjs');
 
 
-var _chunk466ED5WNcjs = require('./chunk-466ED5WN.cjs');
+var _chunkCLJ3DWTAcjs = require('./chunk-CLJ3DWTA.cjs');
 
 // src/modules/mcp/mcp.module.ts
 var _indexjs = require('@modelcontextprotocol/sdk/client/index.js');
@@ -59,7 +59,7 @@ var MCPModule = (_class = class {constructor() { _class.prototype.__init.call(th
             break;
           }
           default:
-            _chunk466ED5WNcjs.loggers.mcp.error("Unsupported MCP config type");
+            _chunkCLJ3DWTAcjs.loggers.mcp.error("Unsupported MCP config type");
             break;
         }
         const toolList = await mcpClient.listTools();
@@ -73,11 +73,11 @@ var MCPModule = (_class = class {constructor() { _class.prototype.__init.call(th
             inputSchema: tool.inputSchema
           };
         });
-        _chunk466ED5WNcjs.loggers.mcp.info("Connected to MCP server with tools:", {
+        _chunkCLJ3DWTAcjs.loggers.mcp.info("Connected to MCP server with tools:", {
           tools: conn.tools.map((tool) => tool.toolName)
         });
       } catch (error) {
-        _chunk466ED5WNcjs.loggers.mcp.error(`Failed to connect to MCP server ${name}`, { error });
+        _chunkCLJ3DWTAcjs.loggers.mcp.error(`Failed to connect to MCP server ${name}`, { error });
       }
     }
   }
@@ -89,7 +89,25 @@ var MCPModule = (_class = class {constructor() { _class.prototype.__init.call(th
   getTools() {
     const allTools = [];
     for (const conn of this.mcpConnectors.values()) {
-      allTools.push(...conn.tools);
+      for (const tool of conn.tools) {
+        const finalInputSchema = {
+          type: "object",
+          properties: { ..._optionalChain([tool, 'access', _ => _.inputSchema, 'optionalAccess', _2 => _2.properties]) || {} },
+          required: [..._optionalChain([tool, 'access', _3 => _3.inputSchema, 'optionalAccess', _4 => _4.required]) || []]
+        };
+        finalInputSchema.properties["thinking_text"] = {
+          type: "string",
+          description: "\uC0AC\uC6A9\uC790\uC758 \uC694\uCCAD\uC744 \uD574\uACB0\uD558\uAE30 \uC704\uD574 \uC774 \uB3C4\uAD6C\uB97C \uC120\uD0DD\uD55C \uAD6C\uCCB4\uC801\uC778 \uC774\uC720\uC640 \uBAA9\uC801 (Why & What). \uD55C\uB450\uC904 \uC815\uB3C4\uC758 \uBD84\uB7C9\uC73C\uB85C \uC785\uB825 \uC5B8\uC5B4\uC640 \uAC19\uC740 \uC5B8\uC5B4\uB85C \uC0DD\uC131\uD55C\uB2E4."
+        };
+        finalInputSchema.required.push("thinking_text");
+        allTools.push({
+          toolName: tool.toolName,
+          connectorName: tool.connectorName,
+          protocol: tool.protocol,
+          description: tool.description,
+          inputSchema: finalInputSchema
+        });
+      }
     }
     return allTools;
   }
@@ -103,7 +121,7 @@ var MCPModule = (_class = class {constructor() { _class.prototype.__init.call(th
    */
   async useTool(tool, _args) {
     const { connectorName, toolName } = tool;
-    const client = _optionalChain([this, 'access', _ => _.mcpConnectors, 'access', _2 => _2.get, 'call', _3 => _3(connectorName), 'optionalAccess', _4 => _4.client]);
+    const client = _optionalChain([this, 'access', _5 => _5.mcpConnectors, 'access', _6 => _6.get, 'call', _7 => _7(connectorName), 'optionalAccess', _8 => _8.client]);
     try {
       if (!client) {
         throw new Error(`Invalid MCP Tool ${toolName}`);
@@ -117,7 +135,7 @@ var MCPModule = (_class = class {constructor() { _class.prototype.__init.call(th
 ` + JSON.stringify(result.content, null, 2);
       return toolResult;
     } catch (error) {
-      _chunk466ED5WNcjs.loggers.mcp.error("Failed to call tool", { error });
+      _chunkCLJ3DWTAcjs.loggers.mcp.error("Failed to call tool", { error });
       const toolResult = `[Bot Called Tool ${toolName} with args ${JSON.stringify(_args)}]
 ${typeof error === "string" ? error : JSON.stringify(error, null, 2)}`;
       return toolResult;
@@ -131,7 +149,7 @@ ${typeof error === "string" ? error : JSON.stringify(error, null, 2)}`;
    */
   async cleanup() {
     for (const conn of this.mcpConnectors.values()) {
-      await _optionalChain([conn, 'access', _5 => _5.client, 'optionalAccess', _6 => _6.close, 'call', _7 => _7()]);
+      await _optionalChain([conn, 'access', _9 => _9.client, 'optionalAccess', _10 => _10.close, 'call', _11 => _11()]);
     }
   }
 }, _class);
@@ -139,4 +157,4 @@ ${typeof error === "string" ? error : JSON.stringify(error, null, 2)}`;
 
 
 exports.MCPModule = MCPModule;
-//# sourceMappingURL=chunk-6WAVK3BS.cjs.map
+//# sourceMappingURL=chunk-4IBKK5HW.cjs.map

package/dist/cjs/chunk-4IBKK5HW.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-4IBKK5HW.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;AACP,MAAA;AAEJ,QAAA;AACC,UAAA;AACM,UAAA;AACD,UAAA;AACZ,QAAA;AAEA,QAAA;AACO,UAAA;AAEL,UAAA;AACF,QAAA;AACA,QAAA;AAEc,QAAA;AACH,UAAA;AACV,UAAA;AACU,UAAA;AACG,UAAA;AACA,UAAA;AACb,QAAA;AACF,MAAA;AACD,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;ADpBuB;AACA;AACA;AACA","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-4IBKK5HW.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\tfor (const tool of conn.tools) {\n\t\t\t\t// add thinking_text inputSchema for each tool\n\t\t\t\tconst finalInputSchema: any = {\n\t\t\t\t\ttype: \"object\",\n\t\t\t\t\tproperties: { ...(tool.inputSchema?.properties || {}) },\n\t\t\t\t\trequired: [...(tool.inputSchema?.required || [])],\n\t\t\t\t};\n\n\t\t\t\tfinalInputSchema.properties[\"thinking_text\"] = {\n\t\t\t\t\ttype: \"string\",\n\t\t\t\t\tdescription:\n\t\t\t\t\t\t\"사용자의 요청을 해결하기 위해 이 도구를 선택한 구체적인 이유와 목적 (Why & What). 한두줄 정도의 분량으로 입력 언어와 같은 언어로 생성한다.\",\n\t\t\t\t};\n\t\t\t\tfinalInputSchema.required.push(\"thinking_text\");\n\n\t\t\t\tallTools.push({\n\t\t\t\t\ttoolName: tool.toolName,\n\t\t\t\t\tconnectorName: tool.connectorName,\n\t\t\t\t\tprotocol: tool.protocol,\n\t\t\t\t\tdescription: tool.description,\n\t\t\t\t\tinputSchema: finalInputSchema,\n\t\t\t\t});\n\t\t\t}\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-QSBUV5WJ.cjs → chunk-5QSLQX4H.cjs}

@@ -1,18 +1,18 @@
 "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 _chunk2TOW46DIcjs = require('./chunk-2TOW46DI.cjs');
+var _chunkOZPPTVUAcjs = require('./chunk-OZPPTVUA.cjs');
 
 
-var _chunkTF7S7SPOcjs = require('./chunk-TF7S7SPO.cjs');
+var _chunkNEWJ7PXHcjs = require('./chunk-NEWJ7PXH.cjs');
 
 
-var _chunkMO3H4KHEcjs = require('./chunk-MO3H4KHE.cjs');
+var _chunkXHTEJE4Ecjs = require('./chunk-XHTEJE4E.cjs');
 
 
 var _chunkZ6TQ7TFOcjs = require('./chunk-Z6TQ7TFO.cjs');
 
 
-var _chunk466ED5WNcjs = require('./chunk-466ED5WN.cjs');
+var _chunkCLJ3DWTAcjs = require('./chunk-CLJ3DWTA.cjs');
 
 // src/services/query-stream.service.ts
 var _crypto = require('crypto');
@@ -25,17 +25,21 @@ var QueryStreamService = class {
   constructor(modelModule, a2aModule, mcpModule, memoryModule) {
     this.modelModule = modelModule;
     this.memoryModule = memoryModule;
-    this.intentTriggerService = new (0, _chunkTF7S7SPOcjs.IntentTriggerService)(
+    this.intentTriggerService = new (0, _chunkNEWJ7PXHcjs.IntentTriggerService)(
       modelModule,
       memoryModule
     );
-    this.intentFulfillStreamService = new (0, _chunk2TOW46DIcjs.IntentFulfillStreamService)(
+    this.intentFulfillStreamService = new (0, _chunkOZPPTVUAcjs.IntentFulfillStreamService)(
       modelModule,
       a2aModule,
       mcpModule,
       memoryModule
     );
   }
+  async addToThreadMessages(userId, threadId, messages) {
+    const threadMemory = _optionalChain([this, 'access', _ => _.memoryModule, 'optionalAccess', _2 => _2.getThreadMemory, 'call', _3 => _3()]);
+    await _optionalChain([threadMemory, 'optionalAccess', _4 => _4.addMessagesToThread, 'call', _5 => _5(userId, threadId, messages)]);
+  }
   /**
    * Main entry point for processing streaming user queries.
    *
@@ -54,31 +58,31 @@ var QueryStreamService = class {
    */
   async *handleQueryStream(threadMetadata, query, isA2A) {
     const { type, userId, options } = threadMetadata;
-    const threadMemory = _optionalChain([this, 'access', _ => _.memoryModule, 'optionalAccess', _2 => _2.getThreadMemory, 'call', _3 => _3()]);
+    const threadMemory = _optionalChain([this, 'access', _6 => _6.memoryModule, 'optionalAccess', _7 => _7.getThreadMemory, 'call', _8 => _8()]);
     let threadId = threadMetadata.threadId;
     let thread;
     if (threadId) {
-      thread = await _optionalChain([threadMemory, 'optionalAccess', _4 => _4.getThread, 'call', _5 => _5(userId, threadId)]);
+      thread = await _optionalChain([threadMemory, 'optionalAccess', _9 => _9.getThread, 'call', _10 => _10(userId, threadId)]);
       if (!thread && !isA2A) {
         throw new (0, _chunkZ6TQ7TFOcjs.AinHttpError)(_httpstatuscodes.StatusCodes.NOT_FOUND, "Thread not found");
       }
     }
     threadId ??= _crypto.randomUUID.call(void 0, );
     if (!thread) {
-      const title = await _chunkMO3H4KHEcjs.generateTitle.call(void 0, this.modelModule, query, options);
-      const metadata = await _optionalChain([threadMemory, 'optionalAccess', _6 => _6.createThread, 'call', _7 => _7(
+      const title = await _chunkXHTEJE4Ecjs.generateTitle.call(void 0, this.modelModule, query, options);
+      const metadata = await _optionalChain([threadMemory, 'optionalAccess', _11 => _11.createThread, 'call', _12 => _12(
         type,
         userId,
         threadId,
         title
       )]) || { type, userId, threadId, title };
       thread = { ...metadata, messages: [] };
-      _chunk466ED5WNcjs.loggers.intent.info(`Create new thread: ${threadId}`);
+      _chunkCLJ3DWTAcjs.loggers.intent.info(`Create new thread: ${threadId}`);
       yield { event: "thread_id", data: { type, userId, threadId, title } };
     }
     const triggeredIntent = await this.intentTriggerService.intentTriggering(query, thread);
-    _chunk466ED5WNcjs.loggers.intent.debug("Triggered intents", { triggeredIntent });
-    await _optionalChain([threadMemory, 'optionalAccess', _8 => _8.addMessagesToThread, 'call', _9 => _9(userId, threadId, [
+    _chunkCLJ3DWTAcjs.loggers.intent.debug("Triggered intents", { triggeredIntent });
+    await this.addToThreadMessages(userId, threadId, [
       {
         messageId: _crypto.randomUUID.call(void 0, ),
         role: "USER" /* USER */,
@@ -86,12 +90,12 @@ var QueryStreamService = class {
         content: { type: "text", parts: [query] },
         metadata: {
           intents: triggeredIntent.filter((intent) => !!intent.intent).map((intent) => ({
-            id: _optionalChain([intent, 'access', _10 => _10.intent, 'optionalAccess', _11 => _11.id]),
+            id: _optionalChain([intent, 'access', _13 => _13.intent, 'optionalAccess', _14 => _14.id]),
             subquery: intent.subquery
           }))
         }
       }
-    ])]);
+    ]);
     const stream = this.intentFulfillStreamService.intentFulfillStream(
       triggeredIntent,
       thread
@@ -105,4 +109,4 @@ var QueryStreamService = class {
 
 
 exports.QueryStreamService = QueryStreamService;
-//# sourceMappingURL=chunk-QSBUV5WJ.cjs.map
+//# sourceMappingURL=chunk-5QSLQX4H.cjs.map

package/dist/cjs/chunk-5QSLQX4H.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-5QSLQX4H.cjs","../../src/services/query-stream.service.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACA;AChBA,gCAA2B;AAC3B,oDAA4B;AA8BrB,IAAM,mBAAA,EAAN,MAAyB;AAAA,EACvB;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EAER,WAAA,CACC,WAAA,EACA,SAAA,EACA,SAAA,EACA,YAAA,EACC;AACD,IAAA,IAAA,CAAK,YAAA,EAAc,WAAA;AACnB,IAAA,IAAA,CAAK,aAAA,EAAe,YAAA;AACpB,IAAA,IAAA,CAAK,qBAAA,EAAuB,IAAI,2CAAA;AAAA,MAC/B,WAAA;AAAA,MACA;AAAA,IACD,CAAA;AACA,IAAA,IAAA,CAAK,2BAAA,EAA6B,IAAI,iDAAA;AAAA,MACrC,WAAA;AAAA,MACA,SAAA;AAAA,MACA,SAAA;AAAA,MACA;AAAA,IACD,CAAA;AAAA,EACD;AAAA,EAEA,MAAa,mBAAA,CACZ,MAAA,EACA,QAAA,EACA,QAAA,EACC;AACD,IAAA,MAAM,aAAA,kBAAe,IAAA,mBAAK,YAAA,6BAAc,eAAA,mBAAgB,GAAA;AACxD,IAAA,sBAAM,YAAA,6BAAc,mBAAA,mBAAoB,MAAA,EAAQ,QAAA,EAAU,QAAQ,GAAA;AAAA,EACnE;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAkBA,MAAA,CAAc,iBAAA,CACb,cAAA,EAMA,KAAA,EACA,KAAA,EAC8B;AAC9B,IAAA,MAAM,EAAE,IAAA,EAAM,MAAA,EAAQ,QAAQ,EAAA,EAAI,cAAA;AAClC,IAAA,MAAM,aAAA,kBAAe,IAAA,qBAAK,YAAA,6BAAc,eAAA,mBAAgB,GAAA;AAGxD,IAAA,IAAI,SAAA,EAAW,cAAA,CAAe,QAAA;AAC9B,IAAA,IAAI,MAAA;AACJ,IAAA,GAAA,CAAI,QAAA,EAAU;AACb,MAAA,OAAA,EAAS,sBAAM,YAAA,6BAAc,SAAA,qBAAU,MAAA,EAAQ,QAAQ,GAAA;AACvD,MAAA,GAAA,CAAI,CAAC,OAAA,GAAU,CAAC,KAAA,EAAO;AACtB,QAAA,MAAM,IAAI,mCAAA,CAAa,4BAAA,CAAY,SAAA,EAAW,kBAAkB,CAAA;AAAA,MACjE;AAAA,IACD;AAEA,IAAA,SAAA,IAAa,gCAAA,CAAW;AACxB,IAAA,GAAA,CAAI,CAAC,MAAA,EAAQ;AACZ,MAAA,MAAM,MAAA,EAAQ,MAAM,6CAAA,IAAc,CAAK,WAAA,EAAa,KAAA,EAAO,OAAO,CAAA;AAClE,MAAA,MAAM,SAAA,EAA4B,sBAAM,YAAA,+BAAc,YAAA;AAAA,QACrD,IAAA;AAAA,QACA,MAAA;AAAA,QACA,QAAA;AAAA,QACA;AAAA,MACD,IAAA,GAAM,EAAE,IAAA,EAAM,MAAA,EAAQ,QAAA,EAAU,MAAM,CAAA;AACtC,MAAA,OAAA,EAAS,EAAE,GAAG,QAAA,EAAU,QAAA,EAAU,CAAC,EAAE,CAAA;AACrC,MAAA,yBAAA,CAAQ,MAAA,CAAO,IAAA,CAAK,CAAA,mBAAA,EAAsB,QAAQ,CAAA,CAAA;AACA,MAAA;AACnD,IAAA;AAIY,IAAA;AACgC,IAAA;AAGK,IAAA;AAChD,MAAA;AACuB,QAAA;AACtB,QAAA;AACoB,QAAA;AACoB,QAAA;AAC9B,QAAA;AAEc,UAAA;AAEF,YAAA;AACF,YAAA;AAChB,UAAA;AACJ,QAAA;AACD,MAAA;AACA,IAAA;AAG8C,IAAA;AAC9C,MAAA;AACA,MAAA;AACD,IAAA;AAEkC,IAAA;AAC3B,MAAA;AACP,IAAA;AACD,EAAA;AACD;AD7CyD;AACA;AACA;AACA","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-5QSLQX4H.cjs","sourcesContent":[null,"import { randomUUID } from \"node:crypto\";\nimport { StatusCodes } from \"http-status-codes\";\nimport type {\n\tA2AModule,\n\tMCPModule,\n\tMemoryModule,\n\tModelFetchOptions,\n\tModelModule,\n} from \"@/modules/index.js\";\nimport { AinHttpError } from \"@/types/agent.js\";\nimport {\n\ttype MessageObject,\n\tMessageRole,\n\ttype ThreadMetadata,\n\ttype ThreadObject,\n\ttype ThreadType,\n\ttype TriggeredIntent,\n} from \"@/types/memory.js\";\nimport type { StreamEvent } from \"@/types/stream\";\nimport { loggers } from \"@/utils/logger.js\";\nimport { IntentFulfillStreamService } from \"./intents/fulfill-stream.service\";\nimport { IntentTriggerService } from \"./intents/trigger.service\";\nimport { generateTitle } from \"./utils/query.common\";\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 memoryModule?: MemoryModule;\n\tprivate intentTriggerService: IntentTriggerService;\n\tprivate intentFulfillStreamService: IntentFulfillStreamService;\n\n\tconstructor(\n\t\tmodelModule: ModelModule,\n\t\ta2aModule?: A2AModule,\n\t\tmcpModule?: MCPModule,\n\t\tmemoryModule?: MemoryModule,\n\t) {\n\t\tthis.modelModule = modelModule;\n\t\tthis.memoryModule = memoryModule;\n\t\tthis.intentTriggerService = new IntentTriggerService(\n\t\t\tmodelModule,\n\t\t\tmemoryModule,\n\t\t);\n\t\tthis.intentFulfillStreamService = new IntentFulfillStreamService(\n\t\t\tmodelModule,\n\t\t\ta2aModule,\n\t\t\tmcpModule,\n\t\t\tmemoryModule,\n\t\t);\n\t}\n\n\tpublic async addToThreadMessages(\n\t\tuserId: string,\n\t\tthreadId: string,\n\t\tmessages: Array<MessageObject>,\n\t) {\n\t\tconst threadMemory = this.memoryModule?.getThreadMemory();\n\t\tawait threadMemory?.addMessagesToThread(userId, threadId, messages);\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\toptions?: ModelFetchOptions;\n\t\t},\n\t\tquery: string,\n\t\tisA2A?: boolean,\n\t): AsyncGenerator<StreamEvent> {\n\t\tconst { type, userId, options } = 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 generateTitle(this.modelModule, query, options);\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\t// 2. intent triggering\n\t\tconst triggeredIntent: Array<TriggeredIntent> =\n\t\t\tawait this.intentTriggerService.intentTriggering(query, thread);\n\t\tloggers.intent.debug(\"Triggered intents\", { triggeredIntent });\n\n\t\t// only add for storage, not for inference\n\t\tawait this.addToThreadMessages(userId, threadId, [\n\t\t\t{\n\t\t\t\tmessageId: randomUUID(),\n\t\t\t\trole: MessageRole.USER,\n\t\t\t\ttimestamp: Date.now(),\n\t\t\t\tcontent: { type: \"text\", parts: [query] },\n\t\t\t\tmetadata: {\n\t\t\t\t\tintents: triggeredIntent\n\t\t\t\t\t\t.filter((intent) => !!intent.intent)\n\t\t\t\t\t\t.map((intent) => ({\n\t\t\t\t\t\t\tid: intent.intent?.id,\n\t\t\t\t\t\t\tsubquery: intent.subquery,\n\t\t\t\t\t\t})),\n\t\t\t\t},\n\t\t\t},\n\t\t]);\n\n\t\t// 3. intent fulfillment\n\t\tconst stream = this.intentFulfillStreamService.intentFulfillStream(\n\t\t\ttriggeredIntent,\n\t\t\tthread,\n\t\t);\n\n\t\tfor await (const event of stream) {\n\t\t\tyield event;\n\t\t}\n\t}\n}\n"]}

package/dist/cjs/{chunk-OVTUSYSU.cjs → chunk-5WWZOWMA.cjs}

@@ -21,4 +21,4 @@ var A2AConnector = (_class = class {
 
 
 exports.A2AConnector = A2AConnector;
-//# sourceMappingURL=chunk-OVTUSYSU.cjs.map
+//# sourceMappingURL=chunk-5WWZOWMA.cjs.map

package/dist/cjs/chunk-5WWZOWMA.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-5WWZOWMA.cjs","../../src/modules/a2a/a2a.connector.ts"],"names":[],"mappings":"AAAA;ACMO,IAAM,aAAA,YAAN,MAA8C;AAAA,EAC7C;AAAA,iBACA,SAAA,EAAA,gBAAA;AAAA,EACA;AAAA,EACA;AAAA,kBACA,OAAA,EAA2B,KAAA;AAAA,EAElC,WAAA,CAAY,IAAA,EAAc,GAAA,EAAa;AACtC,IAAA,IAAA,CAAK,KAAA,EAAO,IAAA;AACZ,IAAA,IAAA,CAAK,QAAA,EAAU,IAAA;AACf,IAAA,IAAA,CAAK,IAAA,EAAM,GAAA;AAAA,EACZ;AAAA,EAEO,MAAA,CAAA,EAAe;AACrB,IAAA,IAAA,CAAK,QAAA,EAAU,IAAA;AAAA,EAChB;AAAA,EAEO,OAAA,CAAA,EAAgB;AACtB,IAAA,IAAA,CAAK,QAAA,EAAU,KAAA;AAAA,EAChB;AACD,UAAA;ADPA;AACA;AACE;AACF,oCAAC","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-5WWZOWMA.cjs","sourcesContent":[null,"import type { Client as A2AClient } from \"@a2a-js/sdk/client\";\nimport {\n\tCONNECTOR_PROTOCOL_TYPE,\n\ttype IAgentConnector,\n} from \"@/types/connector.js\";\n\nexport class A2AConnector implements IAgentConnector {\n\tpublic name: string;\n\tpublic protocol: CONNECTOR_PROTOCOL_TYPE = CONNECTOR_PROTOCOL_TYPE.A2A;\n\tpublic enabled: boolean;\n\tpublic url: string;\n\tpublic client: A2AClient | null = null;\n\n\tconstructor(name: string, url: string) {\n\t\tthis.name = name;\n\t\tthis.enabled = true;\n\t\tthis.url = url;\n\t}\n\n\tpublic enable(): void {\n\t\tthis.enabled = true;\n\t}\n\n\tpublic disable(): void {\n\t\tthis.enabled = false;\n\t}\n}\n"]}

package/dist/cjs/{chunk-AWHKDGXN.cjs → chunk-6T2YGFSD.cjs}

@@ -3,7 +3,7 @@
 var _chunk2P3ZBNVNcjs = require('./chunk-2P3ZBNVN.cjs');
 
 
-var _chunkOU5UB2HFcjs = require('./chunk-OU5UB2HF.cjs');
+var _chunkNEWJ7PXHcjs = require('./chunk-NEWJ7PXH.cjs');
 
 
 var _chunkS6BHSEXFcjs = require('./chunk-S6BHSEXF.cjs');
@@ -13,7 +13,7 @@ var _express = require('express');
 var createIntentRouter = (agent) => {
   const router = _express.Router.call(void 0, );
   const threadService = new (0, _chunk2P3ZBNVNcjs.ThreadService)(agent.memoryModule);
-  const triggerService = new (0, _chunkOU5UB2HFcjs.IntentTriggerService)(
+  const triggerService = new (0, _chunkNEWJ7PXHcjs.IntentTriggerService)(
     agent.modelModule,
     agent.memoryModule
   );
@@ -25,4 +25,4 @@ var createIntentRouter = (agent) => {
 
 
 exports.createIntentRouter = createIntentRouter;
-//# sourceMappingURL=chunk-AWHKDGXN.cjs.map
+//# sourceMappingURL=chunk-6T2YGFSD.cjs.map

package/dist/cjs/{chunk-AWHKDGXN.cjs.map → chunk-6T2YGFSD.cjs.map}

@@ -1 +1 @@
-{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-AWHKDGXN.cjs","../../src/routes/intent.routes.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACA;ACVA,kCAAuB;AAMhB,IAAM,mBAAA,EAAqB,CAAC,KAAA,EAAA,GAA4B;AAC9D,EAAA,MAAM,OAAA,EAAS,6BAAA,CAAO;AAEtB,EAAA,MAAM,cAAA,EAAgB,IAAI,oCAAA,CAAc,KAAA,CAAM,YAAY,CAAA;AAC1D,EAAA,MAAM,eAAA,EAAiB,IAAI,2CAAA;AAAA,IAC1B,KAAA,CAAM,WAAA;AAAA,IACN,KAAA,CAAM;AAAA,EACP,CAAA;AAEA,EAAA,MAAM,iBAAA,EAAmB,IAAI,uCAAA,CAAiB,aAAA,EAAe,cAAc,CAAA;AAE3E,EAAA,MAAA,CAAO,IAAA,CAAK,UAAA,EAAY,gBAAA,CAAiB,mBAAmB,CAAA;AAE5D,EAAA,OAAO,MAAA;AACR,CAAA;ADGA;AACA;AACE;AACF,gDAAC","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-AWHKDGXN.cjs","sourcesContent":[null,"import { Router } from \"express\";\nimport { IntentController } from \"@/controllers/intent.controller\";\nimport type { AINAgent } from \"@/index.js\";\nimport { IntentTriggerService } from \"@/services\";\nimport { ThreadService } from \"@/services/thread.service\";\n\nexport const createIntentRouter = (agent: AINAgent): Router => {\n\tconst router = Router();\n\n\tconst threadService = new ThreadService(agent.memoryModule);\n\tconst triggerService = new IntentTriggerService(\n\t\tagent.modelModule,\n\t\tagent.memoryModule,\n\t);\n\n\tconst intentController = new IntentController(threadService, triggerService);\n\n\trouter.post(\"/trigger\", intentController.handleIntentTrigger);\n\n\treturn router;\n};\n"]}
+{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-6T2YGFSD.cjs","../../src/routes/intent.routes.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACA;ACVA,kCAAuB;AAMhB,IAAM,mBAAA,EAAqB,CAAC,KAAA,EAAA,GAA4B;AAC9D,EAAA,MAAM,OAAA,EAAS,6BAAA,CAAO;AAEtB,EAAA,MAAM,cAAA,EAAgB,IAAI,oCAAA,CAAc,KAAA,CAAM,YAAY,CAAA;AAC1D,EAAA,MAAM,eAAA,EAAiB,IAAI,2CAAA;AAAA,IAC1B,KAAA,CAAM,WAAA;AAAA,IACN,KAAA,CAAM;AAAA,EACP,CAAA;AAEA,EAAA,MAAM,iBAAA,EAAmB,IAAI,uCAAA,CAAiB,aAAA,EAAe,cAAc,CAAA;AAE3E,EAAA,MAAA,CAAO,IAAA,CAAK,UAAA,EAAY,gBAAA,CAAiB,mBAAmB,CAAA;AAE5D,EAAA,OAAO,MAAA;AACR,CAAA;ADGA;AACA;AACE;AACF,gDAAC","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-6T2YGFSD.cjs","sourcesContent":[null,"import { Router } from \"express\";\nimport { IntentController } from \"@/controllers/intent.controller\";\nimport type { AINAgent } from \"@/index.js\";\nimport { IntentTriggerService } from \"@/services\";\nimport { ThreadService } from \"@/services/thread.service\";\n\nexport const createIntentRouter = (agent: AINAgent): Router => {\n\tconst router = Router();\n\n\tconst threadService = new ThreadService(agent.memoryModule);\n\tconst triggerService = new IntentTriggerService(\n\t\tagent.modelModule,\n\t\tagent.memoryModule,\n\t);\n\n\tconst intentController = new IntentController(threadService, triggerService);\n\n\trouter.post(\"/trigger\", intentController.handleIntentTrigger);\n\n\treturn router;\n};\n"]}

package/dist/cjs/{chunk-NYDNXGKB.cjs → chunk-7KXB6FOT.cjs}

@@ -1,18 +1,18 @@
 "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 _chunk7QXQYCZPcjs = require('./chunk-7QXQYCZP.cjs');
+var _chunkOZPPTVUAcjs = require('./chunk-OZPPTVUA.cjs');
 
 
-var _chunkMO3H4KHEcjs = require('./chunk-MO3H4KHE.cjs');
+var _chunkXHTEJE4Ecjs = require('./chunk-XHTEJE4E.cjs');
 
 
-var _chunkTF7S7SPOcjs = require('./chunk-TF7S7SPO.cjs');
+var _chunkNEWJ7PXHcjs = require('./chunk-NEWJ7PXH.cjs');
 
 
 var _chunkZ6TQ7TFOcjs = require('./chunk-Z6TQ7TFO.cjs');
 
 
-var _chunk466ED5WNcjs = require('./chunk-466ED5WN.cjs');
+var _chunkCLJ3DWTAcjs = require('./chunk-CLJ3DWTA.cjs');
 
 // src/services/query-stream.service.ts
 var _crypto = require('crypto');
@@ -25,17 +25,21 @@ var QueryStreamService = class {
   constructor(modelModule, a2aModule, mcpModule, memoryModule) {
     this.modelModule = modelModule;
     this.memoryModule = memoryModule;
-    this.intentTriggerService = new (0, _chunkTF7S7SPOcjs.IntentTriggerService)(
+    this.intentTriggerService = new (0, _chunkNEWJ7PXHcjs.IntentTriggerService)(
       modelModule,
       memoryModule
     );
-    this.intentFulfillStreamService = new (0, _chunk7QXQYCZPcjs.IntentFulfillStreamService)(
+    this.intentFulfillStreamService = new (0, _chunkOZPPTVUAcjs.IntentFulfillStreamService)(
       modelModule,
       a2aModule,
       mcpModule,
       memoryModule
     );
   }
+  async addToThreadMessages(userId, threadId, messages) {
+    const threadMemory = _optionalChain([this, 'access', _ => _.memoryModule, 'optionalAccess', _2 => _2.getThreadMemory, 'call', _3 => _3()]);
+    await _optionalChain([threadMemory, 'optionalAccess', _4 => _4.addMessagesToThread, 'call', _5 => _5(userId, threadId, messages)]);
+  }
   /**
    * Main entry point for processing streaming user queries.
    *
@@ -54,31 +58,31 @@ var QueryStreamService = class {
    */
   async *handleQueryStream(threadMetadata, query, isA2A) {
     const { type, userId, options } = threadMetadata;
-    const threadMemory = _optionalChain([this, 'access', _ => _.memoryModule, 'optionalAccess', _2 => _2.getThreadMemory, 'call', _3 => _3()]);
+    const threadMemory = _optionalChain([this, 'access', _6 => _6.memoryModule, 'optionalAccess', _7 => _7.getThreadMemory, 'call', _8 => _8()]);
     let threadId = threadMetadata.threadId;
     let thread;
     if (threadId) {
-      thread = await _optionalChain([threadMemory, 'optionalAccess', _4 => _4.getThread, 'call', _5 => _5(userId, threadId)]);
+      thread = await _optionalChain([threadMemory, 'optionalAccess', _9 => _9.getThread, 'call', _10 => _10(userId, threadId)]);
       if (!thread && !isA2A) {
         throw new (0, _chunkZ6TQ7TFOcjs.AinHttpError)(_httpstatuscodes.StatusCodes.NOT_FOUND, "Thread not found");
       }
     }
     threadId ??= _crypto.randomUUID.call(void 0, );
     if (!thread) {
-      const title = await _chunkMO3H4KHEcjs.generateTitle.call(void 0, this.modelModule, query, options);
-      const metadata = await _optionalChain([threadMemory, 'optionalAccess', _6 => _6.createThread, 'call', _7 => _7(
+      const title = await _chunkXHTEJE4Ecjs.generateTitle.call(void 0, this.modelModule, query, options);
+      const metadata = await _optionalChain([threadMemory, 'optionalAccess', _11 => _11.createThread, 'call', _12 => _12(
         type,
         userId,
         threadId,
         title
       )]) || { type, userId, threadId, title };
       thread = { ...metadata, messages: [] };
-      _chunk466ED5WNcjs.loggers.intent.info(`Create new thread: ${threadId}`);
+      _chunkCLJ3DWTAcjs.loggers.intent.info(`Create new thread: ${threadId}`);
       yield { event: "thread_id", data: { type, userId, threadId, title } };
     }
     const triggeredIntent = await this.intentTriggerService.intentTriggering(query, thread);
-    _chunk466ED5WNcjs.loggers.intent.debug("Triggered intents", { triggeredIntent });
-    await _optionalChain([threadMemory, 'optionalAccess', _8 => _8.addMessagesToThread, 'call', _9 => _9(userId, threadId, [
+    _chunkCLJ3DWTAcjs.loggers.intent.debug("Triggered intents", { triggeredIntent });
+    await this.addToThreadMessages(userId, threadId, [
       {
         messageId: _crypto.randomUUID.call(void 0, ),
         role: "USER" /* USER */,
@@ -86,12 +90,12 @@ var QueryStreamService = class {
         content: { type: "text", parts: [query] },
         metadata: {
           intents: triggeredIntent.filter((intent) => !!intent.intent).map((intent) => ({
-            id: _optionalChain([intent, 'access', _10 => _10.intent, 'optionalAccess', _11 => _11.id]),
+            id: _optionalChain([intent, 'access', _13 => _13.intent, 'optionalAccess', _14 => _14.id]),
             subquery: intent.subquery
           }))
         }
       }
-    ])]);
+    ]);
     const stream = this.intentFulfillStreamService.intentFulfillStream(
       triggeredIntent,
       thread
@@ -105,4 +109,4 @@ var QueryStreamService = class {
 
 
 exports.QueryStreamService = QueryStreamService;
-//# sourceMappingURL=chunk-NYDNXGKB.cjs.map
+//# sourceMappingURL=chunk-7KXB6FOT.cjs.map