@ainetwork/adk 0.1.11 → 0.1.13

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

@@ -0,0 +1 @@
+{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-6BKOR3ZH.cjs","../../src/services/a2a.service.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACA;ACJA,gCAA2B;AAcpB,IAAM,WAAA,YAAN,MAA0C;AAAA,EACxC;AAAA,iBACA,cAAA,kBAA6B,IAAI,GAAA,CAAY,EAAA;AAAA,EAErD,WAAA,CAAY,YAAA,EAA4B;AACvC,IAAA,IAAA,CAAK,aAAA,EAAe,YAAA;AAAA,EACrB;AAAA,kBAEO,WAAA,EAAa,MAAA,CACnB,MAAA,EACA,SAAA,EAAA,GACmB;AACnB,IAAA,IAAA,CAAK,aAAA,CAAc,GAAA,CAAI,MAAM,CAAA;AAAA,EAC9B,EAAA;AAAA,kBAEQ,4BAAA,EAA8B,CACrC,MAAA,EACA,SAAA,EACA,KAAA,EACA,OAAA,EAAA,GAC2B;AAC3B,IAAA,OAAO;AAAA,MACN,IAAA,EAAM,eAAA;AAAA,MACN,MAAA;AAAA,MACA,SAAA;AAAA,MACA,MAAA,EAAQ;AAAA,QACP,KAAA;AAAA,QACA,OAAA,EAAS,QAAA,EACN;AAAA,UACA,IAAA,EAAM,SAAA;AAAA,UACN,IAAA,EAAM,OAAA;AAAA,UACN,SAAA,EAAW,gCAAA,CAAW;AAAA,UACtB,KAAA,EAAO,CAAC,EAAE,IAAA,EAAM,MAAA,EAAQ,IAAA,EAAM,QAAQ,CAAC,CAAA;AAAA,UACvC,MAAA;AAAA,UACA;AAAA,QACD,EAAA,EACC,KAAA,CAAA;AAAA,QACH,SAAA,EAAA,iBAAW,IAAI,IAAA,CAAK,CAAA,CAAA,CAAE,WAAA,CAAY;AAAA,MACnC,CAAA;AAAA,MACA,KAAA,EAAO,MAAA,IAAU;AAAA,IAClB,CAAA;AAAA,EACD,EAAA;AAAA,EAEA,MAAM,OAAA,CACL,cAAA,EACA,QAAA,EACgB;AAChB,IAAA,MAAM,YAAA,EAAc,cAAA,CAAe,WAAA;AACnC,IAAA,MAAM,EAAE,SAAS,EAAA,EAAI,WAAA,CAAY,QAAA;AACjC,IAAA,MAAM,aAAA,EAAe,cAAA,CAAe,IAAA;AAEpC,IAAA,MAAM,OAAA,kBAAS,YAAA,2BAAc,KAAA,GAAM,gCAAA,CAAW;AAC9C,IAAA,MAAM,UAAA,EACL,WAAA,CAAY,UAAA,mBAAa,YAAA,6BAAc,YAAA,GAAa,gCAAA,CAAW;AAEhE,IAAA,GAAA,CAAI,CAAC,YAAA,EAAc;AAClB,MAAA,MAAM,YAAA,EAAoB;AAAA,QACzB,IAAA,EAAM,MAAA;AAAA,QACN,EAAA,EAAI,MAAA;AAAA,QACJ,SAAA;AAAA,QACA,MAAA,EAAQ;AAAA,UACP,KAAA,EAAO,WAAA;AAAA,UACP,SAAA,EAAA,iBAAW,IAAI,IAAA,CAAK,CAAA,CAAA,CAAE,WAAA,CAAY;AAAA,QACnC,CAAA;AAAA,QACA,OAAA,EAAS,CAAC,WAAW,CAAA;AAAA,QACrB,QAAA,EAAU,WAAA,CAAY,QAAA;AAAA,QACtB,SAAA,EAAW,CAAC;AAAA,MACb,CAAA;AACA,MAAA,QAAA,CAAS,OAAA,CAAQ,WAAW,CAAA;AAAA,IAC7B;AAEA,IAAA,MAAM,oBAAA,EAAsB,IAAA,CAAK,2BAAA;AAAA,MAChC,MAAA;AAAA,MACA,SAAA;AAAA,MACA;AAAA,IACD,CAAA;AACA,IAAA,QAAA,CAAS,OAAA,CAAQ,mBAAmB,CAAA;AAEpC,IAAA,MAAM,QAAA,EAAkB,WAAA,CAAY,KAAA,CAClC,MAAA,CAAO,CAAC,IAAA,EAAA,GAAS,IAAA,CAAK,KAAA,IAAS,MAAM,CAAA,CACrC,GAAA,CAAI,CAAC,IAAA,EAAA,GAAS,IAAA,CAAK,IAAI,CAAA,CACvB,IAAA,CAAK,IAAI,CAAA;AACX,IAAA,GAAA,CAAI,OAAA,CAAQ,OAAA,IAAW,CAAA,EAAG;AACzB,MAAA,yBAAA,CAAQ,MAAA,CAAO,IAAA,CAAK,CAAA,gCAAA,EAAmC,MAAM,CAAA,CAAA,CAAG,CAAA;AAChE,MAAA,MAAM,cAAA,EAAgB,IAAA,CAAK,2BAAA;AAAA,QAC1B,MAAA;AAAA,QACA,SAAA;AAAA,QACA,QAAA;AAAA,QACA;AAAA,MACD,CAAA;AACA,MAAA,QAAA,CAAS,OAAA,CAAQ,aAAa,CAAA;AAC9B,MAAA,MAAA;AAAA,IACD;AAEA,IAAA,IAAI;AACH,MAAA,MAAM,SAAA,EAAW,MAAM,IAAA,CAAK,YAAA,CAAa,WAAA,CAAY,OAAA,EAAS,QAAQ,CAAA;AAEtE,MAAA,GAAA,CAAI,IAAA,CAAK,aAAA,CAAc,GAAA,CAAI,MAAM,CAAA,EAAG;AACnC,QAAA,yBAAA,CAAQ,MAAA,CAAO,IAAA,CAAK,CAAA,KAAA,EAAQ,MAAM,CAAA,cAAA,CAAgB,CAAA;AAClD,QAAA,MAAM,eAAA,EAAiB,IAAA,CAAK,2BAAA;AAAA,UAC3B,MAAA;AAAA,UACA,SAAA;AAAA,UACA;AAAA,QACD,CAAA;AACA,QAAA,QAAA,CAAS,OAAA,CAAQ,cAAc,CAAA;AAC/B,QAAA,MAAA;AAAA,MACD;AAEA,MAAA,MAAM,YAAA,EAAc,IAAA,CAAK,2BAAA;AAAA,QACxB,MAAA;AAAA,QACA,SAAA;AAAA,QACA,WAAA;AAAA,QACA,QAAA,CAAS;AAAA,MACV,CAAA;AACA,MAAA,QAAA,CAAS,OAAA,CAAQ,WAAW,CAAA;AAC5B,MAAA,yBAAA,CAAQ,MAAA,CAAO,IAAA,CAAK,CAAA,KAAA,EAAQ,MAAM,CAAA,wBAAA,CAA0B,CAAA;AAAA,IAC7D,EAAA,MAAA,CAAS,KAAA,EAAY;AACpB,MAAA,yBAAA,CAAQ,MAAA,CAAO,KAAA,CAAM,CAAA,sBAAA,EAAyB,MAAM,CAAA,CAAA,CAAA,EAAK,KAAK,CAAA;AAC9D,MAAA,MAAM,YAAA,EAAc,IAAA,CAAK,2BAAA;AAAA,QACxB,MAAA;AAAA,QACA,SAAA;AAAA,QACA,QAAA;AAAA,QACA,CAAA,aAAA,EAAgB,KAAA,CAAM,OAAO,CAAA;AAAA,MAAA;AAE9B,MAAA;AAA4B,IAAA;AAC7B,EAAA;AAEF;ADnCA;AACA;AACA;AACA","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-6BKOR3ZH.cjs","sourcesContent":[null,"import { randomUUID } from \"node:crypto\";\nimport type { Task, TaskStatusUpdateEvent } from \"@a2a-js/sdk\";\nimport type {\n\tAgentExecutor,\n\tExecutionEventBus,\n\tRequestContext,\n} from \"@a2a-js/sdk/server\";\nimport { loggers } from \"@/utils/logger.js\";\nimport type { QueryService } from \"./query.service.js\";\n\n/**\n * Implements the AgentExecutor interface from the a2a-js-sdk.\n * This service is responsible for the core business logic of executing an A2A task.\n */\nexport class A2AService implements AgentExecutor {\n\tprivate queryService: QueryService;\n\tprivate canceledTasks: Set<string> = new Set<string>();\n\n\tconstructor(queryService: QueryService) {\n\t\tthis.queryService = queryService;\n\t}\n\n\tpublic cancelTask = async (\n\t\ttaskId: string,\n\t\t_eventBus: ExecutionEventBus,\n\t): Promise<void> => {\n\t\tthis.canceledTasks.add(taskId);\n\t};\n\n\tprivate createTaskStatusUpdateEvent = (\n\t\ttaskId: string,\n\t\tcontextId: string,\n\t\tstate: \"working\" | \"failed\" | \"canceled\" | \"completed\",\n\t\tmessage?: string,\n\t): TaskStatusUpdateEvent => {\n\t\treturn {\n\t\t\tkind: \"status-update\",\n\t\t\ttaskId: taskId,\n\t\t\tcontextId: contextId,\n\t\t\tstatus: {\n\t\t\t\tstate: state,\n\t\t\t\tmessage: message\n\t\t\t\t\t? {\n\t\t\t\t\t\t\tkind: \"message\",\n\t\t\t\t\t\t\trole: \"agent\",\n\t\t\t\t\t\t\tmessageId: randomUUID(),\n\t\t\t\t\t\t\tparts: [{ kind: \"text\", text: message }],\n\t\t\t\t\t\t\ttaskId: taskId,\n\t\t\t\t\t\t\tcontextId: contextId,\n\t\t\t\t\t\t}\n\t\t\t\t\t: undefined,\n\t\t\t\ttimestamp: new Date().toISOString(),\n\t\t\t},\n\t\t\tfinal: state !== \"working\",\n\t\t};\n\t};\n\n\tasync execute(\n\t\trequestContext: RequestContext,\n\t\teventBus: ExecutionEventBus,\n\t): Promise<void> {\n\t\tconst userMessage = requestContext.userMessage;\n\t\tconst { threadId } = userMessage.metadata as { threadId: string };\n\t\tconst existingTask = requestContext.task;\n\n\t\tconst taskId = existingTask?.id || randomUUID();\n\t\tconst contextId =\n\t\t\tuserMessage.contextId || existingTask?.contextId || randomUUID();\n\n\t\tif (!existingTask) {\n\t\t\tconst initialTask: Task = {\n\t\t\t\tkind: \"task\",\n\t\t\t\tid: taskId,\n\t\t\t\tcontextId: contextId,\n\t\t\t\tstatus: {\n\t\t\t\t\tstate: \"submitted\",\n\t\t\t\t\ttimestamp: new Date().toISOString(),\n\t\t\t\t},\n\t\t\t\thistory: [userMessage],\n\t\t\t\tmetadata: userMessage.metadata,\n\t\t\t\tartifacts: [],\n\t\t\t};\n\t\t\teventBus.publish(initialTask);\n\t\t}\n\n\t\tconst workingStatusUpdate = this.createTaskStatusUpdateEvent(\n\t\t\ttaskId,\n\t\t\tcontextId,\n\t\t\t\"working\",\n\t\t);\n\t\teventBus.publish(workingStatusUpdate);\n\n\t\tconst message: string = userMessage.parts\n\t\t\t.filter((part) => part.kind === \"text\")\n\t\t\t.map((part) => part.text)\n\t\t\t.join(\"\\n\");\n\t\tif (message.length === 0) {\n\t\t\tloggers.server.warn(`Empty message received for task ${taskId}.`);\n\t\t\tconst failureUpdate = this.createTaskStatusUpdateEvent(\n\t\t\t\ttaskId,\n\t\t\t\tcontextId,\n\t\t\t\t\"failed\",\n\t\t\t\t\"No message found to process.\",\n\t\t\t);\n\t\t\teventBus.publish(failureUpdate);\n\t\t\treturn;\n\t\t}\n\n\t\ttry {\n\t\t\tconst response = await this.queryService.handleQuery(message, threadId);\n\n\t\t\tif (this.canceledTasks.has(taskId)) {\n\t\t\t\tloggers.server.info(`Task ${taskId} was canceled.`);\n\t\t\t\tconst canceledUpdate = this.createTaskStatusUpdateEvent(\n\t\t\t\t\ttaskId,\n\t\t\t\t\tcontextId,\n\t\t\t\t\t\"canceled\",\n\t\t\t\t);\n\t\t\t\teventBus.publish(canceledUpdate);\n\t\t\t\treturn;\n\t\t\t}\n\n\t\t\tconst finalUpdate = this.createTaskStatusUpdateEvent(\n\t\t\t\ttaskId,\n\t\t\t\tcontextId,\n\t\t\t\t\"completed\",\n\t\t\t\tresponse.content,\n\t\t\t);\n\t\t\teventBus.publish(finalUpdate);\n\t\t\tloggers.server.info(`Task ${taskId} completed successfully.`);\n\t\t} catch (error: any) {\n\t\t\tloggers.server.error(`Error processing task ${taskId}:`, error);\n\t\t\tconst errorUpdate = this.createTaskStatusUpdateEvent(\n\t\t\t\ttaskId,\n\t\t\t\tcontextId,\n\t\t\t\t\"failed\",\n\t\t\t\t`Agent error: ${error.message}`,\n\t\t\t);\n\t\t\teventBus.publish(errorUpdate);\n\t\t}\n\t}\n}\n"]}

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

@@ -0,0 +1,176 @@
+"use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _nullishCoalesce(lhs, rhsFn) { if (lhs != null) { return lhs; } else { return rhsFn(); } } 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 _chunkF7I4G6JGcjs = require('./chunk-F7I4G6JG.cjs');
+
+
+var _chunk466ED5WNcjs = require('./chunk-466ED5WN.cjs');
+
+// src/modules/a2a/a2a.module.ts
+var _crypto = require('crypto');
+var _client = require('@a2a-js/sdk/client');
+var A2AModule = (_class = class {constructor() { _class.prototype.__init.call(this);_class.prototype.__init2.call(this);_class.prototype.__init3.call(this);_class.prototype.__init4.call(this); }
+  /** Map of A2A server URLs to their corresponding tool instances */
+  __init() {this.a2aPeerServers = /* @__PURE__ */ new Map()}
+  /** Map of session IDs to their A2A session state */
+  __init2() {this.a2aSessions = /* @__PURE__ */ new Map()}
+  __init3() {this.agentId = _crypto.randomUUID.call(void 0, )}
+  /* FIXME */
+  /**
+   * Registers a new A2A peer server URL for connection.
+   *
+   * @param url - The URL of the A2A-compatible agent to connect to
+   */
+  async addA2APeerServer(url) {
+    this.a2aPeerServers.set(url, null);
+  }
+  /**
+   * Retrieves tools from all registered A2A peer servers.
+   *
+   * Attempts to connect to each registered server, fetch their agent cards,
+   * and create tool instances. Disables tools for unreachable servers.
+   *
+   * @returns Promise resolving to array of available A2A tools
+   */
+  async getTools() {
+    const tools = [];
+    for (const url of [...this.a2aPeerServers.keys()]) {
+      const tool = this.a2aPeerServers.get(url);
+      if (!tool || !tool.enabled) {
+        try {
+          const client = new (0, _client.A2AClient)(url);
+          const card = await client.getAgentCard();
+          const toolName = card.name.replaceAll(" ", "-");
+          const a2aTool = new (0, _chunkF7I4G6JGcjs.A2ATool)(toolName, client, card);
+          tools.push(a2aTool);
+        } catch (_error) {
+          if (tool) {
+            tool.disable();
+          }
+        }
+      } else {
+        tools.push(tool);
+      }
+    }
+    return tools;
+  }
+  /**
+   * Gets or creates an A2A session for the given session ID.
+   *
+   * @param threadId - The session identifier
+   * @returns A2ASession object with task and context IDs
+   */
+  __init4() {this.getA2ASessionWithId = (threadId) => {
+    const a2aSession = _nullishCoalesce(this.a2aSessions.get(threadId), () => ( {
+      taskId: void 0,
+      contextId: void 0
+    }));
+    if (!this.a2aSessions.has(threadId)) {
+      this.a2aSessions.set(threadId, a2aSession);
+    }
+    return a2aSession;
+  }}
+  /**
+   * Constructs a message payload for A2A communication.
+   *
+   * Includes session context (task ID and context ID) if available
+   * for maintaining conversation continuity.
+   *
+   * @param query - The message content to send
+   * @param threadId - The session identifier
+   * @returns Formatted Message object for A2A protocol
+   */
+  getMessagePayload(query, threadId) {
+    const messagePayload = {
+      messageId: _crypto.randomUUID.call(void 0, ),
+      kind: "message",
+      role: "user",
+      // FIXME: it could be 'agent'
+      metadata: {
+        agentId: this.agentId,
+        type: "CHAT" /* CHAT */,
+        threadId
+      },
+      parts: [
+        {
+          kind: "text",
+          text: query
+        }
+      ]
+    };
+    const a2aSession = this.getA2ASessionWithId(threadId);
+    if (a2aSession.taskId) {
+      messagePayload.taskId = a2aSession.taskId;
+    }
+    if (a2aSession.contextId) {
+      messagePayload.contextId = a2aSession.contextId;
+    }
+    return messagePayload;
+  }
+  /**
+   * Executes an A2A tool by sending a message to the remote agent.
+   *
+   * Handles streaming responses, maintains session state, and extracts
+   * text content from various event types in the response stream.
+   *
+   * @param tool - The A2ATool instance to use
+   * @param messagePayload - 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
+   */
+  async useTool(tool, messagePayload, threadId) {
+    const finalText = [];
+    const client = tool.client;
+    const params = {
+      message: messagePayload
+    };
+    const a2aSession = this.getA2ASessionWithId(threadId);
+    try {
+      const stream = client.sendMessageStream(params);
+      for await (const event of stream) {
+        if (event.kind === "status-update") {
+          const typedEvent = event;
+          if (typedEvent.final && typedEvent.status.state !== "input-required") {
+            a2aSession.taskId = void 0;
+          }
+          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);
+          }
+        } else if (event.kind === "message") {
+          const msg = event;
+          if (msg.taskId && msg.taskId !== a2aSession.taskId) {
+            a2aSession.taskId = msg.taskId;
+          }
+          if (msg.contextId && msg.contextId !== a2aSession.contextId) {
+            a2aSession.contextId = msg.contextId;
+          }
+        } else if (event.kind === "task") {
+          const task = event;
+          if (task.id !== a2aSession.taskId) {
+            a2aSession.taskId = task.id;
+          }
+          if (task.contextId && task.contextId !== a2aSession.contextId) {
+            a2aSession.contextId = task.contextId;
+          }
+        } else {
+          _chunk466ED5WNcjs.loggers.a2a.warn("Received unknown event structure from stream:", {
+            event
+          });
+        }
+      }
+    } catch (error) {
+      _chunk466ED5WNcjs.loggers.a2a.error("Error communicating with agent:", { error });
+      tool.disable();
+      const toolResult = `[Bot Called A2A Tool ${tool.card.name}]
+${typeof error === "string" ? error : JSON.stringify(error, null, 2)}`;
+      return toolResult;
+    }
+    return `[Bot Called A2A Tool ${tool.card.name}]
+${finalText.join("\n")}`;
+  }
+}, _class);
+
+
+
+exports.A2AModule = A2AModule;
+//# sourceMappingURL=chunk-6YVTJEMQ.cjs.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-6YVTJEMQ.cjs","../../src/modules/a2a/a2a.module.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACA;ACPA,gCAA2B;AAS3B,4CAA0B;AAsBnB,IAAM,UAAA,YAAN,MAAgB;AAAA;AAAA,iBAEd,eAAA,kBAA8C,IAAI,GAAA,CAAI,EAAA;AAAA;AAAA,kBAEtD,YAAA,kBAAuC,IAAI,GAAA,CAAI,EAAA;AAAA,kBAC/C,QAAA,EAAkB,gCAAA,EAAW;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOrC,MAAa,gBAAA,CAAiB,GAAA,EAA4B;AACzD,IAAA,IAAA,CAAK,cAAA,CAAe,GAAA,CAAI,GAAA,EAAK,IAAI,CAAA;AAAA,EAClC;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAUA,MAAa,QAAA,CAAA,EAA+B;AAC3C,IAAA,MAAM,MAAA,EAAmB,CAAC,CAAA;AAC1B,IAAA,IAAA,CAAA,MAAW,IAAA,GAAO,CAAC,GAAG,IAAA,CAAK,cAAA,CAAe,IAAA,CAAK,CAAC,CAAA,EAAG;AAClD,MAAA,MAAM,KAAA,EAAO,IAAA,CAAK,cAAA,CAAe,GAAA,CAAI,GAAG,CAAA;AACxC,MAAA,GAAA,CAAI,CAAC,KAAA,GAAQ,CAAC,IAAA,CAAK,OAAA,EAAS;AAC3B,QAAA,IAAI;AACH,UAAA,MAAM,OAAA,EAAS,IAAI,sBAAA,CAAU,GAAG,CAAA;AAChC,UAAA,MAAM,KAAA,EAAkB,MAAM,MAAA,CAAO,YAAA,CAAa,CAAA;AAClD,UAAA,MAAM,SAAA,EAAW,IAAA,CAAK,IAAA,CAAK,UAAA,CAAW,GAAA,EAAK,GAAG,CAAA;AAC9C,UAAA,MAAM,QAAA,EAAU,IAAI,8BAAA,CAAQ,QAAA,EAAU,MAAA,EAAQ,IAAI,CAAA;AAElD,UAAA,KAAA,CAAM,IAAA,CAAK,OAAO,CAAA;AAAA,QACnB,EAAA,MAAA,CAAS,MAAA,EAAa;AAErB,UAAA,GAAA,CAAI,IAAA,EAAM;AACT,YAAA,IAAA,CAAK,OAAA,CAAQ,CAAA;AAAA,UACd;AAAA,QACD;AAAA,MACD,EAAA,KAAO;AACN,QAAA,KAAA,CAAM,IAAA,CAAK,IAAI,CAAA;AAAA,MAChB;AAAA,IACD;AACA,IAAA,OAAO,KAAA;AAAA,EACR;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,kBAQQ,oBAAA,EAAsB,CAAC,QAAA,EAAA,GAAiC;AAC/D,IAAA,MAAM,WAAA,mBAAa,IAAA,CAAK,WAAA,CAAY,GAAA,CAAI,QAAQ,CAAA,UAAK;AAAA,MACpD,MAAA,EAAQ,KAAA,CAAA;AAAA,MACR,SAAA,EAAW,KAAA;AAAA,IACZ,GAAA;AACA,IAAA,GAAA,CAAI,CAAC,IAAA,CAAK,WAAA,CAAY,GAAA,CAAI,QAAQ,CAAA,EAAG;AACpC,MAAA,IAAA,CAAK,WAAA,CAAY,GAAA,CAAI,QAAA,EAAU,UAAU,CAAA;AAAA,IAC1C;AAEA,IAAA,OAAO,UAAA;AAAA,EACR,EAAA;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,iBAAA;AAAA,QACA;AAAA,MACD,CAAA;AAAA,MACA,KAAA,EAAO;AAAA,QACN;AAAA,UACC,IAAA,EAAM,MAAA;AAAA,UACN,IAAA,EAAM;AAAA,QACP;AAAA,MACD;AAAA,IACD,CAAA;AAEA,IAAA,MAAM,WAAA,EAAa,IAAA,CAAK,mBAAA,CAAoB,QAAQ,CAAA;AACpD,IAAA,GAAA,CAAI,UAAA,CAAW,MAAA,EAAQ;AACtB,MAAA,cAAA,CAAe,OAAA,EAAS,UAAA,CAAW,MAAA;AAAA,IACpC;AACA,IAAA,GAAA,CAAI,UAAA,CAAW,SAAA,EAAW;AACzB,MAAA,cAAA,CAAe,UAAA,EAAY,UAAA,CAAW,SAAA;AAAA,IACvC;AAEA,IAAA,OAAO,cAAA;AAAA,EACR;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAaA,MAAa,OAAA,CACZ,IAAA,EACA,cAAA,EACA,QAAA,EACkB;AAClB,IAAA,MAAM,UAAA,EAAsB,CAAC,CAAA;AAC7B,IAAA,MAAM,OAAA,EAAS,IAAA,CAAK,MAAA;AACpB,IAAA,MAAM,OAAA,EAA4B;AAAA,MACjC,OAAA,EAAS;AAAA,IACV,CAAA;AACA,IAAA,MAAM,WAAA,EAAa,IAAA,CAAK,mBAAA,CAAoB,QAAQ,CAAA;AAEpD,IAAA,IAAI;AACH,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,UAAA,CAAW,OAAA,EAAS,KAAA,CAAA;AAAA,UACrB;AAEA,UAAA,MAAM,MAAA,kBAAQ,UAAA,mBAAW,MAAA,qBAAO,OAAA,6BAAS,KAAA,qBACvC,MAAA,mBAAO,CAAC,IAAA,EAAA,GAAS,IAAA,CAAK,KAAA,IAAS,MAAM,CAAA,qBACrC,GAAA,mBAAI,CAAC,IAAA,EAAA,GAAmB,IAAA,CAAK,IAAI,CAAA,qBACjC,IAAA,mBAAK,IAAI,GAAA;AACX,UAAA,GAAA,CAAI,KAAA,EAAO;AACV,YAAA,SAAA,CAAU,IAAA,CAAK,KAAK,CAAA;AAAA,UACrB;AAAA,QACD,EAAA,KAAA,GAAA,CAAW,KAAA,CAAM,KAAA,IAAS,SAAA,EAAW;AAEpC,UAAA,MAAM,IAAA,EAAM,KAAA;AACZ,UAAA,GAAA,CAAI,GAAA,CAAI,OAAA,GAAU,GAAA,CAAI,OAAA,IAAW,UAAA,CAAW,MAAA,EAAQ;AACnD,YAAA,UAAA,CAAW,OAAA,EAAS,GAAA,CAAI,MAAA;AAAA,UACzB;AACA,UAAA,GAAA,CAAI,GAAA,CAAI,UAAA,GAAa,GAAA,CAAI,UAAA,IAAc,UAAA,CAAW,SAAA,EAAW;AAC5D,YAAA,UAAA,CAAW,UAAA,EAAY,GAAA,CAAI,SAAA;AAAA,UAC5B;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,UAAA,CAAW,MAAA,EAAQ;AAClC,YAAA,UAAA,CAAW,OAAA,EAAS,IAAA,CAAK,EAAA;AAAA,UAC1B;AACA,UAAA,GAAA,CAAI,IAAA,CAAK,UAAA,GAAa,IAAA,CAAK,UAAA,IAAc,UAAA,CAAW,SAAA,EAAW;AAC9D,YAAA,UAAA,CAAW,UAAA,EAAY,IAAA,CAAK,SAAA;AAAA,UAC7B;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,IAAA,CAAK,OAAA,CAAQ,CAAA;AACb,MAAA,MAAM,WAAA,EAAa,CAAA,qBAAA,EAAwB,IAAA,CAAK,IAAA,CAAK,IAAI,CAAA;AAAA,EAAM,OAAO,MAAA,IAAU,SAAA,EAAW,MAAA,EAAQ,IAAA,CAAK,SAAA,CAAU,KAAA,EAAO,IAAA,EAAM,CAAC,CAAC,CAAA,CAAA;AAC1H,MAAA;AACR,IAAA;AAE6C,IAAA;AAA0B;AACxE,EAAA;AACD;ADzCqE;AACA;AACA;AACA","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-6YVTJEMQ.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 { A2AClient } from \"@a2a-js/sdk/client\";\nimport { ThreadType } from \"@/types/memory.js\";\nimport { loggers } from \"@/utils/logger.js\";\nimport { A2ATool } from \"./a2a.tool.js\";\n\n/**\n * Represents an active A2A communication session.\n */\ninterface A2ASession {\n\t/** Current task ID for multi-turn conversations */\n\ttaskId: string | undefined;\n\t/** Context ID for maintaining conversation state */\n\tcontextId: string | undefined;\n}\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 a2aPeerServers: Map<string, A2ATool | null> = new Map();\n\t/** Map of session IDs to their A2A session state */\n\tprivate a2aSessions: Map<string, A2ASession> = 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 url - The URL of the A2A-compatible agent to connect to\n\t */\n\tpublic async addA2APeerServer(url: string): Promise<void> {\n\t\tthis.a2aPeerServers.set(url, null);\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<A2ATool[]> {\n\t\tconst tools: A2ATool[] = [];\n\t\tfor (const url of [...this.a2aPeerServers.keys()]) {\n\t\t\tconst tool = this.a2aPeerServers.get(url);\n\t\t\tif (!tool || !tool.enabled) {\n\t\t\t\ttry {\n\t\t\t\t\tconst client = new A2AClient(url);\n\t\t\t\t\tconst card: AgentCard = await client.getAgentCard();\n\t\t\t\t\tconst toolName = card.name.replaceAll(\" \", \"-\");\n\t\t\t\t\tconst a2aTool = new A2ATool(toolName, client, card);\n\n\t\t\t\t\ttools.push(a2aTool);\n\t\t\t\t} catch (_error: any) {\n\t\t\t\t\t// Agent not responded\n\t\t\t\t\tif (tool) {\n\t\t\t\t\t\ttool.disable();\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\ttools.push(tool);\n\t\t\t}\n\t\t}\n\t\treturn tools;\n\t}\n\n\t/**\n\t * Gets or creates an A2A session for the given session ID.\n\t *\n\t * @param threadId - The session identifier\n\t * @returns A2ASession object with task and context IDs\n\t */\n\tprivate getA2ASessionWithId = (threadId: string): A2ASession => {\n\t\tconst a2aSession = this.a2aSessions.get(threadId) ?? {\n\t\t\ttaskId: undefined,\n\t\t\tcontextId: undefined,\n\t\t};\n\t\tif (!this.a2aSessions.has(threadId)) {\n\t\t\tthis.a2aSessions.set(threadId, a2aSession);\n\t\t}\n\n\t\treturn a2aSession;\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\tthreadId,\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};\n\n\t\tconst a2aSession = this.getA2ASessionWithId(threadId);\n\t\tif (a2aSession.taskId) {\n\t\t\tmessagePayload.taskId = a2aSession.taskId;\n\t\t}\n\t\tif (a2aSession.contextId) {\n\t\t\tmessagePayload.contextId = a2aSession.contextId;\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 messagePayload - The message to send to the agent\n\t * @param threadId - The session identifier for context tracking\n\t * @returns Promise resolving to array of text responses from the agent\n\t */\n\tpublic async useTool(\n\t\ttool: A2ATool,\n\t\tmessagePayload: Message,\n\t\tthreadId: string,\n\t): Promise<string> {\n\t\tconst finalText: string[] = [];\n\t\tconst client = tool.client;\n\t\tconst params: MessageSendParams = {\n\t\t\tmessage: messagePayload,\n\t\t};\n\t\tconst a2aSession = this.getA2ASessionWithId(threadId);\n\n\t\ttry {\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\ta2aSession.taskId = undefined;\n\t\t\t\t\t}\n\t\t\t\t\t// TODO: handle 'file', 'data' parts\n\t\t\t\t\tconst texts = typedEvent.status.message?.parts\n\t\t\t\t\t\t.filter((part) => part.kind === \"text\")\n\t\t\t\t\t\t.map((part: TextPart) => part.text)\n\t\t\t\t\t\t.join(\"\\n\");\n\t\t\t\t\tif (texts) {\n\t\t\t\t\t\tfinalText.push(texts);\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\tif (msg.taskId && msg.taskId !== a2aSession.taskId) {\n\t\t\t\t\t\ta2aSession.taskId = msg.taskId;\n\t\t\t\t\t}\n\t\t\t\t\tif (msg.contextId && msg.contextId !== a2aSession.contextId) {\n\t\t\t\t\t\ta2aSession.contextId = msg.contextId;\n\t\t\t\t\t}\n\t\t\t\t} else if (event.kind === \"task\") {\n\t\t\t\t\t// FIXME: handling text in 'task'?\n\t\t\t\t\tconst task = event as Task;\n\t\t\t\t\tif (task.id !== a2aSession.taskId) {\n\t\t\t\t\t\ta2aSession.taskId = task.id;\n\t\t\t\t\t}\n\t\t\t\t\tif (task.contextId && task.contextId !== a2aSession.contextId) {\n\t\t\t\t\t\ta2aSession.contextId = task.contextId;\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\ttool.disable();\n\t\t\tconst toolResult = `[Bot Called A2A Tool ${tool.card.name}]\\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.card.name}]\\n${finalText.join(\"\\n\")}`;\n\t}\n}\n"]}

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

@@ -0,0 +1,205 @@
+"use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _optionalChain(ops) { let lastAccessLHS = undefined; let value = ops[0]; let i = 1; while (i < ops.length) { const op = ops[i]; const fn = ops[i + 1]; i += 2; if ((op === 'optionalAccess' || op === 'optionalCall') && value == null) { return undefined; } if (op === 'access' || op === 'optionalAccess') { lastAccessLHS = value; value = fn(value); } else if (op === 'call' || op === 'optionalCall') { value = fn((...args) => value.call(lastAccessLHS, ...args)); lastAccessLHS = undefined; } } return value; }
+
+var _chunk466ED5WNcjs = require('./chunk-466ED5WN.cjs');
+
+// src/services/query.service.ts
+var _crypto = require('crypto');
+var QueryService = class {
+  
+  
+  
+  
+  
+  constructor(modelModule, a2aModule, mcpModule, memoryModule, prompts) {
+    this.modelModule = modelModule;
+    this.a2aModule = a2aModule;
+    this.mcpModule = mcpModule;
+    this.memoryModule = memoryModule;
+    this.prompts = prompts;
+  }
+  /**
+   * Detects the intent from a user query.
+   *
+   * @param query - The user's input query
+   * @returns The detected intent (currently returns the query as-is)
+   * @todo Implement actual intent detection logic
+   */
+  async intentTriggering(query) {
+    return query;
+  }
+  /**
+   * Fulfills the detected intent by generating a response.
+   *
+   * Manages the complete inference loop including:
+   * - Loading prompts and conversation history
+   * - Collecting available tools from modules
+   * - Executing model inference with tool support
+   * - Processing tool calls iteratively until completion
+   *
+   * @param query - The user's input query
+   * @param threadId - Thread identifier for context
+   * @param thread - Previous conversation history
+   * @returns Object containing process steps and final response
+   */
+  async intentFulfilling(query, threadId, thread) {
+    const systemPrompt = `
+Today is ${(/* @__PURE__ */ new Date()).toLocaleDateString()}.
+
+${_optionalChain([this, 'access', _ => _.prompts, 'optionalAccess', _2 => _2.agent]) || ""}
+
+${_optionalChain([this, 'access', _3 => _3.prompts, 'optionalAccess', _4 => _4.system]) || ""}
+    `;
+    const modelInstance = this.modelModule.getModel();
+    const messages = modelInstance.generateMessages({
+      query,
+      thread,
+      systemPrompt: systemPrompt.trim()
+    });
+    const tools = [];
+    if (this.mcpModule) {
+      tools.push(...this.mcpModule.getTools());
+    }
+    if (this.a2aModule) {
+      tools.push(...await this.a2aModule.getTools());
+    }
+    const functions = modelInstance.convertToolsToFunctions(tools);
+    const processList = [];
+    let finalMessage = "";
+    let didCallTool = false;
+    while (true) {
+      const response = await modelInstance.fetchWithContextMessage(
+        messages,
+        functions
+      );
+      didCallTool = false;
+      _chunk466ED5WNcjs.loggers.intent.debug("messages", { messages });
+      const { content, toolCalls } = response;
+      _chunk466ED5WNcjs.loggers.intent.debug("content", { content });
+      _chunk466ED5WNcjs.loggers.intent.debug("tool_calls", { ...toolCalls });
+      if (toolCalls) {
+        const messagePayload = _optionalChain([this, 'access', _5 => _5.a2aModule, 'optionalAccess', _6 => _6.getMessagePayload, 'call', _7 => _7(
+          query,
+          threadId
+        )]);
+        for (const toolCall of toolCalls) {
+          const toolName = toolCall.name;
+          didCallTool = true;
+          const selectedTool = tools.filter((tool) => tool.id === toolName)[0];
+          let toolResult = "";
+          if (this.mcpModule && selectedTool.protocol === "MCP" /* MCP */) {
+            const toolArgs = toolCall.arguments;
+            _chunk466ED5WNcjs.loggers.intent.debug("MCP tool call", { toolName, toolArgs });
+            toolResult = await this.mcpModule.useTool(
+              selectedTool,
+              toolArgs
+            );
+          } else if (this.a2aModule && selectedTool.protocol === "A2A" /* A2A */) {
+            toolResult = await this.a2aModule.useTool(
+              selectedTool,
+              messagePayload,
+              threadId
+            );
+          } else {
+            _chunk466ED5WNcjs.loggers.intent.warn(
+              `Unrecognized tool type: ${selectedTool.protocol}`
+            );
+            continue;
+          }
+          _chunk466ED5WNcjs.loggers.intent.debug("toolResult", { toolResult });
+          processList.push(toolResult);
+          modelInstance.appendMessages(messages, toolResult);
+        }
+      } else if (content) {
+        processList.push(content);
+        finalMessage = content;
+      }
+      if (!didCallTool) break;
+    }
+    const botResponse = {
+      process: processList.join("\n"),
+      response: finalMessage
+    };
+    return botResponse;
+  }
+  /**
+   * 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.intent.error("Error generating title", {
+        error,
+        query
+      });
+      return DEFAULT_TITLE;
+    }
+  }
+  /**
+   * Main entry point for processing user queries.
+   *
+   * Handles the complete query lifecycle:
+   * 1. Loads thread history from memory
+   * 2. Detects intent from the query
+   * 3. Fulfills the intent with AI response
+   * 4. Updates conversation history
+   *
+   * @param type - The type of thread (e.g., chat, workflow)
+   * @param userId - The user's unique identifier
+   * @param threadId - Unique thread identifier
+   * @param query - The user's input query
+   */
+  async handleQuery(threadMetadata, query) {
+    const { type, userId } = threadMetadata;
+    const queryStartAt = Date.now();
+    const threadMemory = _optionalChain([this, 'access', _8 => _8.memoryModule, 'optionalAccess', _9 => _9.getThreadMemory, 'call', _10 => _10()]);
+    let threadId = threadMetadata.threadId;
+    let thread;
+    if (threadId) {
+      thread = await _optionalChain([threadMemory, 'optionalAccess', _11 => _11.getThread, 'call', _12 => _12(type, userId, threadId)]);
+    } else {
+      threadId = _crypto.randomUUID.call(void 0, );
+      const title = await this.generateTitle(query);
+      const metadata = await _optionalChain([threadMemory, 'optionalAccess', _13 => _13.createThread, 'call', _14 => _14(type, userId, threadId, title)]) || {
+        type,
+        threadId,
+        title,
+        updatedAt: Date.now()
+      };
+      _chunk466ED5WNcjs.loggers.intent.info("Create new thread", { metadata });
+    }
+    const _intent = this.intentTriggering(query);
+    const result = await this.intentFulfilling(query, threadId, thread);
+    await _optionalChain([threadMemory, 'optionalAccess', _15 => _15.addMessagesToThread, 'call', _16 => _16(userId, threadId, [
+      {
+        role: "USER" /* USER */,
+        timestamp: queryStartAt,
+        content: { type: "text", parts: [query] }
+      },
+      {
+        role: "MODEL" /* MODEL */,
+        timestamp: Date.now(),
+        content: { type: "text", parts: [result.response] }
+      }
+    ])]);
+    return { content: result.response };
+  }
+};
+
+
+
+exports.QueryService = QueryService;
+//# sourceMappingURL=chunk-7EGJAURF.cjs.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-7EGJAURF.cjs","../../src/services/query.service.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACA;ACJA,gCAA2B;AA6BpB,IAAM,aAAA,EAAN,MAAmB;AAAA,EACjB;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EAER,WAAA,CACC,WAAA,EACA,SAAA,EACA,SAAA,EACA,YAAA,EACA,OAAA,EACC;AACD,IAAA,IAAA,CAAK,YAAA,EAAc,WAAA;AACnB,IAAA,IAAA,CAAK,UAAA,EAAY,SAAA;AACjB,IAAA,IAAA,CAAK,UAAA,EAAY,SAAA;AACjB,IAAA,IAAA,CAAK,aAAA,EAAe,YAAA;AACpB,IAAA,IAAA,CAAK,QAAA,EAAU,OAAA;AAAA,EAChB;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EASA,MAAc,gBAAA,CAAiB,KAAA,EAAe;AAE7C,IAAA,OAAO,KAAA;AAAA,EACR;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAgBA,MAAc,gBAAA,CACb,KAAA,EACA,QAAA,EACA,MAAA,EACC;AAED,IAAA,MAAM,aAAA,EAAe,CAAA;AAAA,SAAA,EAAA,iBACZ,IAAI,IAAA,CAAK,CAAA,CAAA,CAAE,kBAAA,CAAmB,CAAC,CAAA;AAAA;AAAA,kBAExC,IAAA,mBAAK,OAAA,6BAAS,QAAA,GAAS,EAAE,CAAA;AAAA;AAAA,kBAEzB,IAAA,qBAAK,OAAA,6BAAS,SAAA,GAAU,EAAE,CAAA;AAAA,IAAA,CAAA;AAG1B,IAAA,MAAM,cAAA,EAAgB,IAAA,CAAK,WAAA,CAAY,QAAA,CAAS,CAAA;AAChD,IAAA,MAAM,SAAA,EAAW,aAAA,CAAc,gBAAA,CAAiB;AAAA,MAC/C,KAAA;AAAA,MACA,MAAA;AAAA,MACA,YAAA,EAAc,YAAA,CAAa,IAAA,CAAK;AAAA,IACjC,CAAC,CAAA;AAED,IAAA,MAAM,MAAA,EAAsB,CAAC,CAAA;AAC7B,IAAA,GAAA,CAAI,IAAA,CAAK,SAAA,EAAW;AACnB,MAAA,KAAA,CAAM,IAAA,CAAK,GAAG,IAAA,CAAK,SAAA,CAAU,QAAA,CAAS,CAAC,CAAA;AAAA,IACxC;AACA,IAAA,GAAA,CAAI,IAAA,CAAK,SAAA,EAAW;AACnB,MAAA,KAAA,CAAM,IAAA,CAAK,GAAI,MAAM,IAAA,CAAK,SAAA,CAAU,QAAA,CAAS,CAAE,CAAA;AAAA,IAChD;AACA,IAAA,MAAM,UAAA,EAAY,aAAA,CAAc,uBAAA,CAAwB,KAAK,CAAA;AAE7D,IAAA,MAAM,YAAA,EAAwB,CAAC,CAAA;AAC/B,IAAA,IAAI,aAAA,EAAe,EAAA;AACnB,IAAA,IAAI,YAAA,EAAc,KAAA;AAElB,IAAA,MAAA,CAAO,IAAA,EAAM;AACZ,MAAA,MAAM,SAAA,EAAW,MAAM,aAAA,CAAc,uBAAA;AAAA,QACpC,QAAA;AAAA,QACA;AAAA,MACD,CAAA;AACA,MAAA,YAAA,EAAc,KAAA;AAEd,MAAA,yBAAA,CAAQ,MAAA,CAAO,KAAA,CAAM,UAAA,EAAY,EAAE,SAAS,CAAC,CAAA;AAE7C,MAAA,MAAM,EAAE,OAAA,EAAS,UAAU,EAAA,EAAI,QAAA;AAE/B,MAAA,yBAAA,CAAQ,MAAA,CAAO,KAAA,CAAM,SAAA,EAAW,EAAE,QAAQ,CAAC,CAAA;AAC3C,MAAA,yBAAA,CAAQ,MAAA,CAAO,KAAA,CAAM,YAAA,EAAc,EAAE,GAAG,UAAU,CAAC,CAAA;AAEnD,MAAA,GAAA,CAAI,SAAA,EAAW;AACd,QAAA,MAAM,eAAA,kBAAiB,IAAA,qBAAK,SAAA,6BAAW,iBAAA;AAAA,UACtC,KAAA;AAAA,UACA;AAAA,QACD,GAAA;AAEA,QAAA,IAAA,CAAA,MAAW,SAAA,GAAY,SAAA,EAAW;AACjC,UAAA,MAAM,SAAA,EAAW,QAAA,CAAS,IAAA;AAC1B,UAAA,YAAA,EAAc,IAAA;AACd,UAAA,MAAM,aAAA,EAAe,KAAA,CAAM,MAAA,CAAO,CAAC,IAAA,EAAA,GAAS,IAAA,CAAK,GAAA,IAAO,QAAQ,CAAA,CAAE,CAAC,CAAA;AAEnE,UAAA,IAAI,WAAA,EAAa,EAAA;AACjB,UAAA,GAAA,CACC,IAAA,CAAK,UAAA,GACL,YAAA,CAAa,SAAA,IAAA,eAAA,EACZ;AACD,YAAA,MAAM,SAAA,EAAW,QAAA,CAAS,SAAA;AAG1B,YAAA,yBAAA,CAAQ,MAAA,CAAO,KAAA,CAAM,eAAA,EAAiB,EAAE,QAAA,EAAU,SAAS,CAAC,CAAA;AAC5D,YAAA,WAAA,EAAa,MAAM,IAAA,CAAK,SAAA,CAAU,OAAA;AAAA,cACjC,YAAA;AAAA,cACA;AAAA,YACD,CAAA;AAAA,UACD,EAAA,KAAA,GAAA,CACC,IAAA,CAAK,UAAA,GACL,YAAA,CAAa,SAAA,IAAA,eAAA,EACZ;AACD,YAAA,WAAA,EAAa,MAAM,IAAA,CAAK,SAAA,CAAU,OAAA;AAAA,cACjC,YAAA;AAAA,cACA,cAAA;AAAA,cACA;AAAA,YACD,CAAA;AAAA,UACD,EAAA,KAAO;AAEN,YAAA,yBAAA,CAAQ,MAAA,CAAO,IAAA;AAAA,cACd,CAAA,wBAAA,EAA2B,YAAA,CAAa,QAAQ,CAAA;AAAA,YAAA;AAEjD,YAAA;AAAA,UAAA;AAGD,UAAA;AAEA,UAAA;AACA,UAAA;AAAiD,QAAA;AAClD,MAAA;AAEA,QAAA;AACA,QAAA;AAAe,MAAA;AAGhB,MAAA;AAAkB,IAAA;AAGnB,IAAA;AAAoB,MAAA;AACW,MAAA;AACpB,IAAA;AAGX,IAAA;AAAO,EAAA;AACR;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAAA;AAUC,IAAA;AACA,IAAA;AACC,MAAA;AACA,MAAA;AAAgD,QAAA;AAC/C,QAAA;AACc;AAAA;AAAA,oFAAA;AAAA,MAAA;AAKf,MAAA;AACA,MAAA;AAA2B,IAAA;AAE3B,MAAA;AAA+C,QAAA;AAC9C,QAAA;AACA,MAAA;AAED,MAAA;AAAO,IAAA;AACR,EAAA;AACD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAAA;AAyBC,IAAA;AACA,IAAA;AACA,IAAA;AAEA,IAAA;AACA,IAAA;AAEA,IAAA;AACC,MAAA;AAA6D,IAAA;AAE7D,MAAA;AACA,MAAA;AAEA,MAAA;AAEE,QAAA;AACA,QAAA;AACA,QAAA;AACA,QAAA;AACoB,MAAA;AAEtB,MAAA;AAAqD,IAAA;AAItD,IAAA;AAGA,IAAA;AACA,IAAA;AAA0D,MAAA;AACzD,QAAA;AACC,QAAA;AACW,QAAA;AAC6B,MAAA;AACzC,MAAA;AACA,QAAA;AACC,QAAA;AACoB,QAAA;AAC8B,MAAA;AACnD,IAAA;AAGD,IAAA;AAAkC,EAAA;AAEpC;ADlFA;AACA;AACA;AACA","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-7EGJAURF.cjs","sourcesContent":[null,"import { randomUUID } from \"node:crypto\";\nimport type {\n\tA2AModule,\n\tMCPModule,\n\tMemoryModule,\n\tModelModule,\n} from \"@/modules/index.js\";\nimport type { AinAgentPrompts } from \"@/types/agent.js\";\nimport {\n\tMessageRole,\n\ttype ThreadMetadata,\n\ttype ThreadObject,\n\ttype ThreadType,\n} from \"@/types/memory.js\";\nimport {\n\ttype IA2ATool,\n\ttype IAgentTool,\n\ttype IMCPTool,\n\tTOOL_PROTOCOL_TYPE,\n} from \"@/types/tool.js\";\nimport { loggers } from \"@/utils/logger.js\";\n\n/**\n * Service for processing user queries through the agent's AI pipeline.\n *\n * Orchestrates the query processing workflow including intent detection,\n * model inference, tool execution, and response generation. Manages\n * conversation context and coordinates between different modules.\n */\nexport class QueryService {\n\tprivate modelModule: ModelModule;\n\tprivate a2aModule?: A2AModule;\n\tprivate mcpModule?: MCPModule;\n\tprivate memoryModule?: MemoryModule;\n\tprivate prompts?: AinAgentPrompts;\n\n\tconstructor(\n\t\tmodelModule: ModelModule,\n\t\ta2aModule?: A2AModule,\n\t\tmcpModule?: MCPModule,\n\t\tmemoryModule?: MemoryModule,\n\t\tprompts?: AinAgentPrompts,\n\t) {\n\t\tthis.modelModule = modelModule;\n\t\tthis.a2aModule = a2aModule;\n\t\tthis.mcpModule = mcpModule;\n\t\tthis.memoryModule = memoryModule;\n\t\tthis.prompts = prompts;\n\t}\n\n\t/**\n\t * Detects the intent from a user query.\n\t *\n\t * @param query - The user's input query\n\t * @returns The detected intent (currently returns the query as-is)\n\t * @todo Implement actual intent detection logic\n\t */\n\tprivate async intentTriggering(query: string) {\n\t\t/* TODO */\n\t\treturn query;\n\t}\n\n\t/**\n\t * Fulfills the detected intent by generating a response.\n\t *\n\t * Manages the complete inference loop including:\n\t * - Loading prompts and conversation history\n\t * - Collecting available tools from modules\n\t * - Executing model inference with tool support\n\t * - Processing tool calls iteratively until completion\n\t *\n\t * @param query - The user's input query\n\t * @param threadId - Thread identifier for context\n\t * @param thread - Previous conversation history\n\t * @returns Object containing process steps and final response\n\t */\n\tprivate async intentFulfilling(\n\t\tquery: string,\n\t\tthreadId: string,\n\t\tthread?: ThreadObject,\n\t) {\n\t\t// 1. Load agent / system prompt from memory\n\t\tconst systemPrompt = `\nToday is ${new Date().toLocaleDateString()}.\n\n${this.prompts?.agent || \"\"}\n\n${this.prompts?.system || \"\"}\n    `;\n\n\t\tconst modelInstance = this.modelModule.getModel();\n\t\tconst messages = modelInstance.generateMessages({\n\t\t\tquery,\n\t\t\tthread,\n\t\t\tsystemPrompt: systemPrompt.trim(),\n\t\t});\n\n\t\tconst tools: IAgentTool[] = [];\n\t\tif (this.mcpModule) {\n\t\t\ttools.push(...this.mcpModule.getTools());\n\t\t}\n\t\tif (this.a2aModule) {\n\t\t\ttools.push(...(await this.a2aModule.getTools()));\n\t\t}\n\t\tconst functions = modelInstance.convertToolsToFunctions(tools);\n\n\t\tconst processList: string[] = [];\n\t\tlet finalMessage = \"\";\n\t\tlet didCallTool = false;\n\n\t\twhile (true) {\n\t\t\tconst response = await modelInstance.fetchWithContextMessage(\n\t\t\t\tmessages,\n\t\t\t\tfunctions,\n\t\t\t);\n\t\t\tdidCallTool = false;\n\n\t\t\tloggers.intent.debug(\"messages\", { messages });\n\n\t\t\tconst { content, toolCalls } = response;\n\n\t\t\tloggers.intent.debug(\"content\", { content });\n\t\t\tloggers.intent.debug(\"tool_calls\", { ...toolCalls });\n\n\t\t\tif (toolCalls) {\n\t\t\t\tconst messagePayload = this.a2aModule?.getMessagePayload(\n\t\t\t\t\tquery,\n\t\t\t\t\tthreadId,\n\t\t\t\t);\n\n\t\t\t\tfor (const toolCall of toolCalls) {\n\t\t\t\t\tconst toolName = toolCall.name;\n\t\t\t\t\tdidCallTool = true;\n\t\t\t\t\tconst selectedTool = tools.filter((tool) => tool.id === toolName)[0];\n\n\t\t\t\t\tlet toolResult = \"\";\n\t\t\t\t\tif (\n\t\t\t\t\t\tthis.mcpModule &&\n\t\t\t\t\t\tselectedTool.protocol === TOOL_PROTOCOL_TYPE.MCP\n\t\t\t\t\t) {\n\t\t\t\t\t\tconst toolArgs = toolCall.arguments as\n\t\t\t\t\t\t\t| { [x: string]: unknown }\n\t\t\t\t\t\t\t| undefined;\n\t\t\t\t\t\tloggers.intent.debug(\"MCP tool call\", { toolName, toolArgs });\n\t\t\t\t\t\ttoolResult = await this.mcpModule.useTool(\n\t\t\t\t\t\t\tselectedTool as IMCPTool,\n\t\t\t\t\t\t\ttoolArgs,\n\t\t\t\t\t\t);\n\t\t\t\t\t} else if (\n\t\t\t\t\t\tthis.a2aModule &&\n\t\t\t\t\t\tselectedTool.protocol === TOOL_PROTOCOL_TYPE.A2A\n\t\t\t\t\t) {\n\t\t\t\t\t\ttoolResult = await this.a2aModule.useTool(\n\t\t\t\t\t\t\tselectedTool as IA2ATool,\n\t\t\t\t\t\t\tmessagePayload!,\n\t\t\t\t\t\t\tthreadId,\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\n\t\t\t\t\tloggers.intent.debug(\"toolResult\", { toolResult });\n\n\t\t\t\t\tprocessList.push(toolResult);\n\t\t\t\t\tmodelInstance.appendMessages(messages, toolResult);\n\t\t\t\t}\n\t\t\t} else if (content) {\n\t\t\t\tprocessList.push(content);\n\t\t\t\tfinalMessage = content;\n\t\t\t}\n\n\t\t\tif (!didCallTool) break;\n\t\t}\n\n\t\tconst botResponse = {\n\t\t\tprocess: processList.join(\"\\n\"),\n\t\t\tresponse: finalMessage,\n\t\t};\n\n\t\treturn botResponse;\n\t}\n\n\t/**\n\t * 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.intent.error(\"Error generating title\", {\n\t\t\t\terror,\n\t\t\t\tquery,\n\t\t\t});\n\t\t\treturn DEFAULT_TITLE;\n\t\t}\n\t}\n\n\t/**\n\t * Main entry point for processing user queries.\n\t *\n\t * Handles the complete query lifecycle:\n\t * 1. Loads thread history from memory\n\t * 2. Detects intent from the query\n\t * 3. Fulfills the intent with AI response\n\t * 4. Updates conversation history\n\t *\n\t * @param type - The type of thread (e.g., chat, workflow)\n\t * @param userId - The user's unique identifier\n\t * @param threadId - Unique thread identifier\n\t * @param query - The user's input query\n\t */\n\tpublic async handleQuery(\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) {\n\t\t// 1. Load thread with threadId\n\t\tconst { type, userId } = threadMetadata;\n\t\tconst queryStartAt = Date.now();\n\t\tconst threadMemory = this.memoryModule?.getThreadMemory();\n\n\t\tlet threadId = threadMetadata.threadId;\n\t\tlet thread: ThreadObject | undefined;\n\n\t\tif (threadId) {\n\t\t\tthread = await threadMemory?.getThread(type, userId, threadId);\n\t\t} else {\n\t\t\tthreadId = randomUUID();\n\t\t\tconst title = await this.generateTitle(query);\n\n\t\t\tconst metadata =\n\t\t\t\t(await threadMemory?.createThread(type, userId, threadId, title)) ||\n\t\t\t\t({\n\t\t\t\t\ttype,\n\t\t\t\t\tthreadId,\n\t\t\t\t\ttitle,\n\t\t\t\t\tupdatedAt: Date.now(),\n\t\t\t\t} as ThreadMetadata);\n\t\t\tloggers.intent.info(\"Create new thread\", { metadata });\n\t\t}\n\n\t\t// 2. intent triggering\n\t\tconst _intent = this.intentTriggering(query);\n\n\t\t// 3. intent fulfillment\n\t\tconst result = await this.intentFulfilling(query, threadId, thread);\n\t\tawait threadMemory?.addMessagesToThread(userId, threadId, [\n\t\t\t{\n\t\t\t\trole: MessageRole.USER,\n\t\t\t\ttimestamp: queryStartAt,\n\t\t\t\tcontent: { type: \"text\", parts: [query] },\n\t\t\t},\n\t\t\t{\n\t\t\t\trole: MessageRole.MODEL,\n\t\t\t\ttimestamp: Date.now(),\n\t\t\t\tcontent: { type: \"text\", parts: [result.response] },\n\t\t\t},\n\t\t]);\n\n\t\treturn { content: result.response };\n\t}\n}\n"]}