@ainetwork/adk 0.2.7 → 0.2.8

package/dist/cjs/chunk-KAF2EDPZ.cjs

@@ -0,0 +1,123 @@
+"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 _chunkIKPQ52KYcjs = require('./chunk-IKPQ52KY.cjs');
+
+
+var _chunkTYLHXACMcjs = require('./chunk-TYLHXACM.cjs');
+
+
+var _chunkQHQLU7BIcjs = require('./chunk-QHQLU7BI.cjs');
+
+
+var _chunk466ED5WNcjs = require('./chunk-466ED5WN.cjs');
+
+// src/services/query.service.ts
+var _crypto = require('crypto');
+var _httpstatuscodes = require('http-status-codes');
+var QueryService = class {
+  
+  
+  
+  
+  constructor(modelModule, a2aModule, mcpModule, memoryModule, prompts) {
+    this.modelModule = modelModule;
+    this.memoryModule = memoryModule;
+    this.intentTriggerService = new (0, _chunkTYLHXACMcjs.IntentTriggerService)(
+      modelModule,
+      memoryModule
+    );
+    this.intentFulfillService = new (0, _chunkIKPQ52KYcjs.IntentFulfillService)(
+      modelModule,
+      a2aModule,
+      mcpModule,
+      memoryModule,
+      prompts
+    );
+  }
+  /**
+   * 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 or creates thread from memory
+   * 2. Detects intent from the query
+   * 3. Fulfills the intent with AI response
+   * 4. Updates conversation history
+   *
+   * @param threadMetadata - Metadata containing type, userId, and optional threadId
+   * @param threadMetadata.type - The type of thread (e.g., chat, workflow)
+   * @param threadMetadata.userId - The user's unique identifier
+   * @param threadMetadata.threadId - Optional thread identifier
+   * @param query - The user's input query
+   * @returns Promise resolving to object with content property containing the response
+   */
+  async handleQuery(threadMetadata, query, isA2A) {
+    const { type, userId } = threadMetadata;
+    const threadMemory = _optionalChain([this, 'access', _ => _.memoryModule, 'optionalAccess', _2 => _2.getThreadMemory, 'call', _3 => _3()]);
+    let threadId = threadMetadata.threadId;
+    let thread;
+    if (threadId) {
+      thread = await _optionalChain([threadMemory, 'optionalAccess', _4 => _4.getThread, 'call', _5 => _5(userId, threadId)]);
+      if (!thread && !isA2A) {
+        throw new (0, _chunkQHQLU7BIcjs.AinHttpError)(_httpstatuscodes.StatusCodes.NOT_FOUND, "Thread not found");
+      }
+    }
+    threadId ??= _crypto.randomUUID.call(void 0, );
+    if (!thread) {
+      const title = await this.generateTitle(query);
+      const metadata = await _optionalChain([threadMemory, 'optionalAccess', _6 => _6.createThread, 'call', _7 => _7(
+        type,
+        userId,
+        threadId,
+        title
+      )]) || { type, userId, threadId, title };
+      thread = { ...metadata, messages: [] };
+      _chunk466ED5WNcjs.loggers.intent.info(`Create new thread: ${threadId}`);
+    }
+    await _optionalChain([threadMemory, 'optionalAccess', _8 => _8.addMessagesToThread, 'call', _9 => _9(userId, threadId, [
+      {
+        messageId: _crypto.randomUUID.call(void 0, ),
+        role: "USER" /* USER */,
+        timestamp: Date.now(),
+        content: { type: "text", parts: [query] }
+      }
+    ])]);
+    const triggeredIntent = await this.intentTriggerService.intentTriggering(query, thread);
+    const result = await this.intentFulfillService.intentFulfill(
+      triggeredIntent,
+      thread
+    );
+    return { content: result };
+  }
+};
+
+
+
+exports.QueryService = QueryService;
+//# sourceMappingURL=chunk-KAF2EDPZ.cjs.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-KAF2EDPZ.cjs","../../src/services/query.service.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACA;ACbA,gCAA2B;AAC3B,oDAA4B;AA0BrB,IAAM,aAAA,EAAN,MAAmB;AAAA,EACjB;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,aAAA,EAAe,YAAA;AACpB,IAAA,IAAA,CAAK,qBAAA,EAAuB,IAAI,2CAAA;AAAA,MAC/B,WAAA;AAAA,MACA;AAAA,IACD,CAAA;AACA,IAAA,IAAA,CAAK,qBAAA,EAAuB,IAAI,2CAAA;AAAA,MAC/B,WAAA;AAAA,MACA,SAAA;AAAA,MACA,SAAA;AAAA,MACA,YAAA;AAAA,MACA;AAAA,IACD,CAAA;AAAA,EACD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EASA,MAAc,aAAA,CAAc,KAAA,EAAgC;AAC3D,IAAA,MAAM,cAAA,EAAgB,UAAA;AACtB,IAAA,IAAI;AACH,MAAA,MAAM,cAAA,EAAgB,IAAA,CAAK,WAAA,CAAY,QAAA,CAAS,CAAA;AAChD,MAAA,MAAM,SAAA,EAAW,aAAA,CAAc,gBAAA,CAAiB;AAAA,QAC/C,KAAA;AAAA,QACA,YAAA,EAAc,CAAA;AAAA;AAAA;AAAA,oFAAA;AAAA,MAIf,CAAC,CAAA;AACD,MAAA,MAAM,SAAA,EAAW,MAAM,aAAA,CAAc,KAAA,CAAM,QAAQ,CAAA;AACnD,MAAA,OAAO,QAAA,CAAS,QAAA,GAAW,aAAA;AAAA,IAC5B,EAAA,MAAA,CAAS,KAAA,EAAO;AACf,MAAA,yBAAA,CAAQ,MAAA,CAAO,KAAA,CAAM,wBAAA,EAA0B;AAAA,QAC9C,KAAA;AAAA,QACA;AAAA,MACD,CAAC,CAAA;AACD,MAAA,OAAO,aAAA;AAAA,IACR;AAAA,EACD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAkBA,MAAa,WAAA,CACZ,cAAA,EAKA,KAAA,EACA,KAAA,EACC;AACD,IAAA,MAAM,EAAE,IAAA,EAAM,OAAO,EAAA,EAAI,cAAA;AACzB,IAAA,MAAM,aAAA,kBAAe,IAAA,mBAAK,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,mBAAU,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,IAAA,CAAK,aAAA,CAAc,KAAK,CAAA;AAC5C,MAAA,MAAM,SAAA,EAA4B,sBAAM,YAAA,6BAAc,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;AACnD,IAAA;AAGgD,IAAA;AAC/C,MAAA;AACuB,QAAA;AACtB,QAAA;AACoB,QAAA;AACoB,QAAA;AACzC,MAAA;AACA,IAAA;AAIW,IAAA;AAGmC,IAAA;AAC9C,MAAA;AACA,MAAA;AACD,IAAA;AAEyB,IAAA;AAC1B,EAAA;AACD;ADvCyD;AACA;AACA;AACA","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-KAF2EDPZ.cjs","sourcesContent":[null,"import { randomUUID } from \"node:crypto\";\nimport { StatusCodes } from \"http-status-codes\";\nimport type {\n\tA2AModule,\n\tMCPModule,\n\tMemoryModule,\n\tModelModule,\n} from \"@/modules/index.js\";\nimport { type AinAgentPrompts, AinHttpError } from \"@/types/agent.js\";\nimport {\n\tMessageRole,\n\ttype ThreadMetadata,\n\ttype ThreadObject,\n\ttype ThreadType,\n\ttype TriggeredIntent,\n} from \"@/types/memory.js\";\nimport { loggers } from \"@/utils/logger.js\";\nimport { IntentFulfillService } from \"./intents/fulfill.service\";\nimport { IntentTriggerService } from \"./intents/trigger.service\";\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 memoryModule?: MemoryModule;\n\tprivate intentTriggerService: IntentTriggerService;\n\tprivate intentFulfillService: IntentFulfillService;\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.memoryModule = memoryModule;\n\t\tthis.intentTriggerService = new IntentTriggerService(\n\t\t\tmodelModule,\n\t\t\tmemoryModule,\n\t\t);\n\t\tthis.intentFulfillService = new IntentFulfillService(\n\t\t\tmodelModule,\n\t\t\ta2aModule,\n\t\t\tmcpModule,\n\t\t\tmemoryModule,\n\t\t\tprompts,\n\t\t);\n\t}\n\n\t/**\n\t * Generates a title for the conversation based on the query.\n\t *\n\t * @param query - The user's input query\n\t * @returns Promise resolving to a generated title\n\t */\n\n\tprivate async generateTitle(query: string): Promise<string> {\n\t\tconst DEFAULT_TITLE = \"New Chat\";\n\t\ttry {\n\t\t\tconst modelInstance = this.modelModule.getModel();\n\t\t\tconst messages = modelInstance.generateMessages({\n\t\t\t\tquery,\n\t\t\t\tsystemPrompt: `You are a helpful assistant that generates titles for conversations.\n  Please analyze the user's query and create a concise title that accurately reflects the conversation's core topic.\n  The title must be no more than 5 words long.\n  Respond with only the title. Do not include any punctuation or extra explanations.`,\n\t\t\t});\n\t\t\tconst response = await modelInstance.fetch(messages);\n\t\t\treturn response.content || DEFAULT_TITLE;\n\t\t} catch (error) {\n\t\t\tloggers.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 or creates thread 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 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 Promise resolving to object with content property containing the response\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\tisA2A?: boolean,\n\t) {\n\t\tconst { type, userId } = threadMetadata;\n\t\tconst threadMemory = this.memoryModule?.getThreadMemory();\n\n\t\t// 1. Load or create thread\n\t\tlet threadId = threadMetadata.threadId;\n\t\tlet thread: ThreadObject | undefined;\n\t\tif (threadId) {\n\t\t\tthread = await threadMemory?.getThread(userId, threadId);\n\t\t\tif (!thread && !isA2A) {\n\t\t\t\tthrow new AinHttpError(StatusCodes.NOT_FOUND, \"Thread not found\");\n\t\t\t}\n\t\t}\n\n\t\tthreadId ??= randomUUID();\n\t\tif (!thread) {\n\t\t\tconst title = await this.generateTitle(query);\n\t\t\tconst metadata: ThreadMetadata = (await threadMemory?.createThread(\n\t\t\t\ttype,\n\t\t\t\tuserId,\n\t\t\t\tthreadId,\n\t\t\t\ttitle,\n\t\t\t)) || { type, userId, threadId, title };\n\t\t\tthread = { ...metadata, messages: [] };\n\t\t\tloggers.intent.info(`Create new thread: ${threadId}`);\n\t\t}\n\n\t\t// only add for storage, not for inference\n\t\tawait threadMemory?.addMessagesToThread(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},\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\n\t\t// 3. intent fulfillment\n\t\tconst result = await this.intentFulfillService.intentFulfill(\n\t\t\ttriggeredIntent,\n\t\t\tthread,\n\t\t);\n\n\t\treturn { content: result };\n\t}\n}\n"]}

package/dist/cjs/chunk-MXDXXOXE.cjs

@@ -0,0 +1,126 @@
+"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 _chunkW4PRBE6Hcjs = require('./chunk-W4PRBE6H.cjs');
+
+
+var _chunkTYLHXACMcjs = require('./chunk-TYLHXACM.cjs');
+
+
+var _chunkQHQLU7BIcjs = require('./chunk-QHQLU7BI.cjs');
+
+
+var _chunk466ED5WNcjs = require('./chunk-466ED5WN.cjs');
+
+// src/services/query-stream.service.ts
+var _crypto = require('crypto');
+var _httpstatuscodes = require('http-status-codes');
+var QueryStreamService = class {
+  
+  
+  
+  
+  constructor(modelModule, a2aModule, mcpModule, memoryModule, prompts) {
+    this.modelModule = modelModule;
+    this.memoryModule = memoryModule;
+    this.intentTriggerService = new (0, _chunkTYLHXACMcjs.IntentTriggerService)(
+      modelModule,
+      memoryModule
+    );
+    this.intentFulfillStreamService = new (0, _chunkW4PRBE6Hcjs.IntentFulfillStreamService)(
+      modelModule,
+      a2aModule,
+      mcpModule,
+      memoryModule,
+      prompts
+    );
+  }
+  /**
+   * Generates a title for the conversation based on the query.
+   *
+   * @param query - The user's input query
+   * @returns Promise resolving to a generated title
+   */
+  async generateTitle(query) {
+    const DEFAULT_TITLE = "New Chat";
+    try {
+      const modelInstance = this.modelModule.getModel();
+      const messages = modelInstance.generateMessages({
+        query,
+        systemPrompt: `You are a helpful assistant that generates titles for conversations.
+  Please analyze the user's query and create a concise title that accurately reflects the conversation's core topic.
+  The title must be no more than 5 words long.
+  Respond with only the title. Do not include any punctuation or extra explanations.`
+      });
+      const response = await modelInstance.fetch(messages);
+      return response.content || DEFAULT_TITLE;
+    } catch (error) {
+      _chunk466ED5WNcjs.loggers.intentStream.error("Error generating title", {
+        error,
+        query
+      });
+      return DEFAULT_TITLE;
+    }
+  }
+  /**
+   * Main entry point for processing streaming user queries.
+   *
+   * Handles the complete query lifecycle:
+   * 1. Loads or creates thread from memory
+   * 2. Detects intent from the query
+   * 3. Fulfills the intent with streaming AI response
+   * 4. Updates conversation history in real-time
+   *
+   * @param threadMetadata - Metadata containing type, userId, and optional threadId
+   * @param threadMetadata.type - The type of thread (e.g., chat, workflow)
+   * @param threadMetadata.userId - The user's unique identifier
+   * @param threadMetadata.threadId - Optional thread identifier
+   * @param query - The user's input query
+   * @returns AsyncGenerator yielding StreamEvent objects for SSE
+   */
+  async *handleQueryStream(threadMetadata, query, isA2A) {
+    const { type, userId } = threadMetadata;
+    const threadMemory = _optionalChain([this, 'access', _ => _.memoryModule, 'optionalAccess', _2 => _2.getThreadMemory, 'call', _3 => _3()]);
+    let threadId = threadMetadata.threadId;
+    let thread;
+    if (threadId) {
+      thread = await _optionalChain([threadMemory, 'optionalAccess', _4 => _4.getThread, 'call', _5 => _5(userId, threadId)]);
+      if (!thread && !isA2A) {
+        throw new (0, _chunkQHQLU7BIcjs.AinHttpError)(_httpstatuscodes.StatusCodes.NOT_FOUND, "Thread not found");
+      }
+    }
+    threadId ??= _crypto.randomUUID.call(void 0, );
+    if (!thread) {
+      const title = await this.generateTitle(query);
+      const metadata = await _optionalChain([threadMemory, 'optionalAccess', _6 => _6.createThread, 'call', _7 => _7(
+        type,
+        userId,
+        threadId,
+        title
+      )]) || { type, userId, threadId, title };
+      thread = { ...metadata, messages: [] };
+      _chunk466ED5WNcjs.loggers.intent.info(`Create new thread: ${threadId}`);
+      yield { event: "thread_id", data: { type, userId, threadId, title } };
+    }
+    await _optionalChain([threadMemory, 'optionalAccess', _8 => _8.addMessagesToThread, 'call', _9 => _9(userId, threadId, [
+      {
+        messageId: _crypto.randomUUID.call(void 0, ),
+        role: "USER" /* USER */,
+        timestamp: Date.now(),
+        content: { type: "text", parts: [query] }
+      }
+    ])]);
+    const triggeredIntent = await this.intentTriggerService.intentTriggering(query, thread);
+    const stream = this.intentFulfillStreamService.intentFulfillStream(
+      triggeredIntent,
+      thread
+    );
+    for await (const event of stream) {
+      yield event;
+    }
+  }
+};
+
+
+
+exports.QueryStreamService = QueryStreamService;
+//# sourceMappingURL=chunk-MXDXXOXE.cjs.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-MXDXXOXE.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;AACA;ACbA,gCAA2B;AAC3B,oDAA4B;AA2BrB,IAAM,mBAAA,EAAN,MAAyB;AAAA,EACvB;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,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,YAAA;AAAA,MACA;AAAA,IACD,CAAA;AAAA,EACD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EASA,MAAc,aAAA,CAAc,KAAA,EAAgC;AAC3D,IAAA,MAAM,cAAA,EAAgB,UAAA;AACtB,IAAA,IAAI;AACH,MAAA,MAAM,cAAA,EAAgB,IAAA,CAAK,WAAA,CAAY,QAAA,CAAS,CAAA;AAChD,MAAA,MAAM,SAAA,EAAW,aAAA,CAAc,gBAAA,CAAiB;AAAA,QAC/C,KAAA;AAAA,QACA,YAAA,EAAc,CAAA;AAAA;AAAA;AAAA,oFAAA;AAAA,MAIf,CAAC,CAAA;AACD,MAAA,MAAM,SAAA,EAAW,MAAM,aAAA,CAAc,KAAA,CAAM,QAAQ,CAAA;AACnD,MAAA,OAAO,QAAA,CAAS,QAAA,GAAW,aAAA;AAAA,IAC5B,EAAA,MAAA,CAAS,KAAA,EAAO;AACf,MAAA,yBAAA,CAAQ,YAAA,CAAa,KAAA,CAAM,wBAAA,EAA0B;AAAA,QACpD,KAAA;AAAA,QACA;AAAA,MACD,CAAC,CAAA;AACD,MAAA,OAAO,aAAA;AAAA,IACR;AAAA,EACD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAkBA,MAAA,CAAc,iBAAA,CACb,cAAA,EAKA,KAAA,EACA,KAAA,EAC8B;AAC9B,IAAA,MAAM,EAAE,IAAA,EAAM,OAAO,EAAA,EAAI,cAAA;AACzB,IAAA,MAAM,aAAA,kBAAe,IAAA,mBAAK,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,mBAAU,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,IAAA,CAAK,aAAA,CAAc,KAAK,CAAA;AAC5C,MAAA,MAAM,SAAA,EAA4B,sBAAM,YAAA,6BAAc,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;AAGgD,IAAA;AAC/C,MAAA;AACuB,QAAA;AACtB,QAAA;AACoB,QAAA;AACoB,QAAA;AACzC,MAAA;AACA,IAAA;AAIW,IAAA;AAGmC,IAAA;AAC9C,MAAA;AACA,MAAA;AACD,IAAA;AAEkC,IAAA;AAC3B,MAAA;AACP,IAAA;AACD,EAAA;AACD;ADxCyD;AACA;AACA;AACA","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-MXDXXOXE.cjs","sourcesContent":[null,"import { randomUUID } from \"node:crypto\";\nimport { StatusCodes } from \"http-status-codes\";\nimport type {\n\tA2AModule,\n\tMCPModule,\n\tMemoryModule,\n\tModelModule,\n} from \"@/modules/index.js\";\nimport { type AinAgentPrompts, AinHttpError } from \"@/types/agent.js\";\nimport {\n\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\";\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\tprompts?: AinAgentPrompts,\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\tprompts,\n\t\t);\n\t}\n\n\t/**\n\t * Generates a title for the conversation based on the query.\n\t *\n\t * @param query - The user's input query\n\t * @returns Promise resolving to a generated title\n\t */\n\n\tprivate async generateTitle(query: string): Promise<string> {\n\t\tconst DEFAULT_TITLE = \"New Chat\";\n\t\ttry {\n\t\t\tconst modelInstance = this.modelModule.getModel();\n\t\t\tconst messages = modelInstance.generateMessages({\n\t\t\t\tquery,\n\t\t\t\tsystemPrompt: `You are a helpful assistant that generates titles for conversations.\n  Please analyze the user's query and create a concise title that accurately reflects the conversation's core topic.\n  The title must be no more than 5 words long.\n  Respond with only the title. Do not include any punctuation or extra explanations.`,\n\t\t\t});\n\t\t\tconst response = await modelInstance.fetch(messages);\n\t\t\treturn response.content || DEFAULT_TITLE;\n\t\t} catch (error) {\n\t\t\tloggers.intentStream.error(\"Error generating title\", {\n\t\t\t\terror,\n\t\t\t\tquery,\n\t\t\t});\n\t\t\treturn DEFAULT_TITLE;\n\t\t}\n\t}\n\n\t/**\n\t * Main entry point for processing streaming user queries.\n\t *\n\t * Handles the complete query lifecycle:\n\t * 1. Loads or creates thread from memory\n\t * 2. Detects intent from the query\n\t * 3. Fulfills the intent with streaming AI response\n\t * 4. Updates conversation history in real-time\n\t *\n\t * @param threadMetadata - Metadata containing type, userId, and optional threadId\n\t * @param threadMetadata.type - The type of thread (e.g., chat, workflow)\n\t * @param threadMetadata.userId - The user's unique identifier\n\t * @param threadMetadata.threadId - Optional thread identifier\n\t * @param query - The user's input query\n\t * @returns AsyncGenerator yielding StreamEvent objects for SSE\n\t */\n\tpublic async *handleQueryStream(\n\t\tthreadMetadata: {\n\t\t\ttype: ThreadType;\n\t\t\tuserId: string;\n\t\t\tthreadId?: string;\n\t\t},\n\t\tquery: string,\n\t\tisA2A?: boolean,\n\t): AsyncGenerator<StreamEvent> {\n\t\tconst { type, userId } = threadMetadata;\n\t\tconst threadMemory = this.memoryModule?.getThreadMemory();\n\n\t\t// 1. Load or create thread\n\t\tlet threadId = threadMetadata.threadId;\n\t\tlet thread: ThreadObject | undefined;\n\t\tif (threadId) {\n\t\t\tthread = await threadMemory?.getThread(userId, threadId);\n\t\t\tif (!thread && !isA2A) {\n\t\t\t\tthrow new AinHttpError(StatusCodes.NOT_FOUND, \"Thread not found\");\n\t\t\t}\n\t\t}\n\n\t\tthreadId ??= randomUUID();\n\t\tif (!thread) {\n\t\t\tconst title = await this.generateTitle(query);\n\t\t\tconst metadata: ThreadMetadata = (await threadMemory?.createThread(\n\t\t\t\ttype,\n\t\t\t\tuserId,\n\t\t\t\tthreadId,\n\t\t\t\ttitle,\n\t\t\t)) || { type, userId, threadId, title };\n\t\t\tthread = { ...metadata, messages: [] };\n\t\t\tloggers.intent.info(`Create new thread: ${threadId}`);\n\t\t\tyield { event: \"thread_id\", data: { type, userId, threadId, title } };\n\t\t}\n\n\t\t// only add for storage, not for inference\n\t\tawait threadMemory?.addMessagesToThread(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},\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\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-OAPMGKE2.cjs

@@ -0,0 +1,123 @@
+"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 _chunkIKPQ52KYcjs = require('./chunk-IKPQ52KY.cjs');
+
+
+var _chunkTYLHXACMcjs = require('./chunk-TYLHXACM.cjs');
+
+
+var _chunk466ED5WNcjs = require('./chunk-466ED5WN.cjs');
+
+
+var _chunkQHQLU7BIcjs = require('./chunk-QHQLU7BI.cjs');
+
+// src/services/query.service.ts
+var _crypto = require('crypto');
+var _httpstatuscodes = require('http-status-codes');
+var QueryService = class {
+  
+  
+  
+  
+  constructor(modelModule, a2aModule, mcpModule, memoryModule, prompts) {
+    this.modelModule = modelModule;
+    this.memoryModule = memoryModule;
+    this.intentTriggerService = new (0, _chunkTYLHXACMcjs.IntentTriggerService)(
+      modelModule,
+      memoryModule
+    );
+    this.intentFulfillService = new (0, _chunkIKPQ52KYcjs.IntentFulfillService)(
+      modelModule,
+      a2aModule,
+      mcpModule,
+      memoryModule,
+      prompts
+    );
+  }
+  /**
+   * 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 or creates thread from memory
+   * 2. Detects intent from the query
+   * 3. Fulfills the intent with AI response
+   * 4. Updates conversation history
+   *
+   * @param threadMetadata - Metadata containing type, userId, and optional threadId
+   * @param threadMetadata.type - The type of thread (e.g., chat, workflow)
+   * @param threadMetadata.userId - The user's unique identifier
+   * @param threadMetadata.threadId - Optional thread identifier
+   * @param query - The user's input query
+   * @returns Promise resolving to object with content property containing the response
+   */
+  async handleQuery(threadMetadata, query, isA2A) {
+    const { type, userId } = threadMetadata;
+    const threadMemory = _optionalChain([this, 'access', _ => _.memoryModule, 'optionalAccess', _2 => _2.getThreadMemory, 'call', _3 => _3()]);
+    let threadId = threadMetadata.threadId;
+    let thread;
+    if (threadId) {
+      thread = await _optionalChain([threadMemory, 'optionalAccess', _4 => _4.getThread, 'call', _5 => _5(userId, threadId)]);
+      if (!thread && !isA2A) {
+        throw new (0, _chunkQHQLU7BIcjs.AinHttpError)(_httpstatuscodes.StatusCodes.NOT_FOUND, "Thread not found");
+      }
+    }
+    threadId ??= _crypto.randomUUID.call(void 0, );
+    if (!thread) {
+      const title = await this.generateTitle(query);
+      const metadata = await _optionalChain([threadMemory, 'optionalAccess', _6 => _6.createThread, 'call', _7 => _7(
+        type,
+        userId,
+        threadId,
+        title
+      )]) || { type, userId, threadId, title };
+      thread = { ...metadata, messages: [] };
+      _chunk466ED5WNcjs.loggers.intent.info(`Create new thread: ${threadId}`);
+    }
+    await _optionalChain([threadMemory, 'optionalAccess', _8 => _8.addMessagesToThread, 'call', _9 => _9(userId, threadId, [
+      {
+        messageId: _crypto.randomUUID.call(void 0, ),
+        role: "USER" /* USER */,
+        timestamp: Date.now(),
+        content: { type: "text", parts: [query] }
+      }
+    ])]);
+    const triggeredIntent = await this.intentTriggerService.intentTriggering(query, thread);
+    const result = await this.intentFulfillService.intentFulfill(
+      triggeredIntent,
+      thread
+    );
+    return { content: result };
+  }
+};
+
+
+
+exports.QueryService = QueryService;
+//# sourceMappingURL=chunk-OAPMGKE2.cjs.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-OAPMGKE2.cjs","../../src/services/query.service.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACA;ACbA,gCAA2B;AAC3B,oDAA4B;AA0BrB,IAAM,aAAA,EAAN,MAAmB;AAAA,EACjB;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,aAAA,EAAe,YAAA;AACpB,IAAA,IAAA,CAAK,qBAAA,EAAuB,IAAI,2CAAA;AAAA,MAC/B,WAAA;AAAA,MACA;AAAA,IACD,CAAA;AACA,IAAA,IAAA,CAAK,qBAAA,EAAuB,IAAI,2CAAA;AAAA,MAC/B,WAAA;AAAA,MACA,SAAA;AAAA,MACA,SAAA;AAAA,MACA,YAAA;AAAA,MACA;AAAA,IACD,CAAA;AAAA,EACD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EASA,MAAc,aAAA,CAAc,KAAA,EAAgC;AAC3D,IAAA,MAAM,cAAA,EAAgB,UAAA;AACtB,IAAA,IAAI;AACH,MAAA,MAAM,cAAA,EAAgB,IAAA,CAAK,WAAA,CAAY,QAAA,CAAS,CAAA;AAChD,MAAA,MAAM,SAAA,EAAW,aAAA,CAAc,gBAAA,CAAiB;AAAA,QAC/C,KAAA;AAAA,QACA,YAAA,EAAc,CAAA;AAAA;AAAA;AAAA,oFAAA;AAAA,MAIf,CAAC,CAAA;AACD,MAAA,MAAM,SAAA,EAAW,MAAM,aAAA,CAAc,KAAA,CAAM,QAAQ,CAAA;AACnD,MAAA,OAAO,QAAA,CAAS,QAAA,GAAW,aAAA;AAAA,IAC5B,EAAA,MAAA,CAAS,KAAA,EAAO;AACf,MAAA,yBAAA,CAAQ,MAAA,CAAO,KAAA,CAAM,wBAAA,EAA0B;AAAA,QAC9C,KAAA;AAAA,QACA;AAAA,MACD,CAAC,CAAA;AACD,MAAA,OAAO,aAAA;AAAA,IACR;AAAA,EACD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAkBA,MAAa,WAAA,CACZ,cAAA,EAKA,KAAA,EACA,KAAA,EACC;AACD,IAAA,MAAM,EAAE,IAAA,EAAM,OAAO,EAAA,EAAI,cAAA;AACzB,IAAA,MAAM,aAAA,kBAAe,IAAA,mBAAK,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,mBAAU,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,IAAA,CAAK,aAAA,CAAc,KAAK,CAAA;AAC5C,MAAA,MAAM,SAAA,EAA4B,sBAAM,YAAA,6BAAc,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;AACnD,IAAA;AAGgD,IAAA;AAC/C,MAAA;AACuB,QAAA;AACtB,QAAA;AACoB,QAAA;AACoB,QAAA;AACzC,MAAA;AACA,IAAA;AAIW,IAAA;AAGmC,IAAA;AAC9C,MAAA;AACA,MAAA;AACD,IAAA;AAEyB,IAAA;AAC1B,EAAA;AACD;ADvCyD;AACA;AACA;AACA","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-OAPMGKE2.cjs","sourcesContent":[null,"import { randomUUID } from \"node:crypto\";\nimport { StatusCodes } from \"http-status-codes\";\nimport type {\n\tA2AModule,\n\tMCPModule,\n\tMemoryModule,\n\tModelModule,\n} from \"@/modules/index.js\";\nimport { type AinAgentPrompts, AinHttpError } from \"@/types/agent.js\";\nimport {\n\tMessageRole,\n\ttype ThreadMetadata,\n\ttype ThreadObject,\n\ttype ThreadType,\n\ttype TriggeredIntent,\n} from \"@/types/memory.js\";\nimport { loggers } from \"@/utils/logger.js\";\nimport { IntentFulfillService } from \"./intents/fulfill.service\";\nimport { IntentTriggerService } from \"./intents/trigger.service\";\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 memoryModule?: MemoryModule;\n\tprivate intentTriggerService: IntentTriggerService;\n\tprivate intentFulfillService: IntentFulfillService;\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.memoryModule = memoryModule;\n\t\tthis.intentTriggerService = new IntentTriggerService(\n\t\t\tmodelModule,\n\t\t\tmemoryModule,\n\t\t);\n\t\tthis.intentFulfillService = new IntentFulfillService(\n\t\t\tmodelModule,\n\t\t\ta2aModule,\n\t\t\tmcpModule,\n\t\t\tmemoryModule,\n\t\t\tprompts,\n\t\t);\n\t}\n\n\t/**\n\t * Generates a title for the conversation based on the query.\n\t *\n\t * @param query - The user's input query\n\t * @returns Promise resolving to a generated title\n\t */\n\n\tprivate async generateTitle(query: string): Promise<string> {\n\t\tconst DEFAULT_TITLE = \"New Chat\";\n\t\ttry {\n\t\t\tconst modelInstance = this.modelModule.getModel();\n\t\t\tconst messages = modelInstance.generateMessages({\n\t\t\t\tquery,\n\t\t\t\tsystemPrompt: `You are a helpful assistant that generates titles for conversations.\n  Please analyze the user's query and create a concise title that accurately reflects the conversation's core topic.\n  The title must be no more than 5 words long.\n  Respond with only the title. Do not include any punctuation or extra explanations.`,\n\t\t\t});\n\t\t\tconst response = await modelInstance.fetch(messages);\n\t\t\treturn response.content || DEFAULT_TITLE;\n\t\t} catch (error) {\n\t\t\tloggers.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 or creates thread 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 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 Promise resolving to object with content property containing the response\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\tisA2A?: boolean,\n\t) {\n\t\tconst { type, userId } = threadMetadata;\n\t\tconst threadMemory = this.memoryModule?.getThreadMemory();\n\n\t\t// 1. Load or create thread\n\t\tlet threadId = threadMetadata.threadId;\n\t\tlet thread: ThreadObject | undefined;\n\t\tif (threadId) {\n\t\t\tthread = await threadMemory?.getThread(userId, threadId);\n\t\t\tif (!thread && !isA2A) {\n\t\t\t\tthrow new AinHttpError(StatusCodes.NOT_FOUND, \"Thread not found\");\n\t\t\t}\n\t\t}\n\n\t\tthreadId ??= randomUUID();\n\t\tif (!thread) {\n\t\t\tconst title = await this.generateTitle(query);\n\t\t\tconst metadata: ThreadMetadata = (await threadMemory?.createThread(\n\t\t\t\ttype,\n\t\t\t\tuserId,\n\t\t\t\tthreadId,\n\t\t\t\ttitle,\n\t\t\t)) || { type, userId, threadId, title };\n\t\t\tthread = { ...metadata, messages: [] };\n\t\t\tloggers.intent.info(`Create new thread: ${threadId}`);\n\t\t}\n\n\t\t// only add for storage, not for inference\n\t\tawait threadMemory?.addMessagesToThread(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},\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\n\t\t// 3. intent fulfillment\n\t\tconst result = await this.intentFulfillService.intentFulfill(\n\t\t\ttriggeredIntent,\n\t\t\tthread,\n\t\t);\n\n\t\treturn { content: result };\n\t}\n}\n"]}

package/dist/cjs/chunk-PNN5WGWA.cjs

@@ -0,0 +1,43 @@
+"use strict";Object.defineProperty(exports, "__esModule", {value: true});
+
+var _chunkKAF2EDPZcjs = require('./chunk-KAF2EDPZ.cjs');
+
+
+var _chunkVLMHDHXTcjs = require('./chunk-VLMHDHXT.cjs');
+
+
+var _chunkMXDXXOXEcjs = require('./chunk-MXDXXOXE.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, _chunkKAF2EDPZcjs.QueryService)(
+    agent.modelModule,
+    agent.a2aModule,
+    agent.mcpModule,
+    agent.memoryModule,
+    agent.manifest.prompts
+  );
+  let queryStreamService;
+  if (allowStream) {
+    queryStreamService = new (0, _chunkMXDXXOXEcjs.QueryStreamService)(
+      agent.modelModule,
+      agent.a2aModule,
+      agent.mcpModule,
+      agent.memoryModule,
+      agent.manifest.prompts
+    );
+  }
+  const queryController = new (0, _chunkVLMHDHXTcjs.QueryController)(queryService, queryStreamService);
+  router.post("/", queryController.handleQueryRequest);
+  if (allowStream) {
+    router.post("/stream", queryController.handleQueryStreamRequest);
+  }
+  return router;
+};
+
+
+
+exports.createQueryRouter = createQueryRouter;
+//# sourceMappingURL=chunk-PNN5WGWA.cjs.map

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

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

package/dist/cjs/chunk-QCEMS4A3.cjs

@@ -0,0 +1,168 @@
+"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 _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); }
+  /** Map of A2A server URLs to their corresponding tool instances */
+  __init() {this.a2aConnectors = /* @__PURE__ */ new Map()}
+  /** Map of session IDs to their A2A session state */
+  __init2() {this.a2aTasks = /* @__PURE__ */ new Map()}
+  __init3() {this.agentId = _crypto.randomUUID.call(void 0, )}
+  /* FIXME */
+  /**
+   * Registers a new A2A peer server URL for connection.
+   *
+   * @param conns - Set of name, url pair
+   */
+  async addA2AConnector(conns) {
+    for (const [name, url] of Object.entries(conns)) {
+      const conn = new (0, _chunkOVTUSYSUcjs.A2AConnector)(name, url);
+      this.a2aConnectors.set(name, conn);
+    }
+  }
+  getA2AConnectors() {
+    const connectors = [];
+    for (const [name, connector] of this.a2aConnectors.entries()) {
+      connectors.push({ name, url: connector.url });
+    }
+    return connectors;
+  }
+  getOrCreateClient(connector) {
+    if (!connector.client) {
+      console.log(`Creating A2AClient for ${connector.name} with URL: ${connector.url}`);
+      connector.client = new (0, _client.A2AClient)(connector.url);
+    }
+    return connector.client;
+  }
+  /**
+   * 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 [name, conn] of this.a2aConnectors.entries()) {
+      if (!conn.enabled) {
+        continue;
+      }
+      try {
+        const client = this.getOrCreateClient(conn);
+        const card = await client.getAgentCard();
+        const tool = {
+          toolName: card.name.replaceAll(" ", "-"),
+          connectorName: name,
+          protocol: "A2A" /* A2A */,
+          description: card.description
+        };
+        tools.push(tool);
+      } catch (error) {
+        console.error(`Failed to get agent card for ${name}:`, error.message);
+      }
+    }
+    return tools;
+  }
+  /**
+   * 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 */
+      },
+      parts: [
+        {
+          kind: "text",
+          text: query
+        }
+      ],
+      contextId: threadId
+    };
+    if (this.a2aTasks.has(threadId)) {
+      messagePayload.taskId = this.a2aTasks.get(threadId);
+    }
+    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 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
+   */
+  async useTool(tool, query, threadId) {
+    const finalText = [];
+    const connector = this.a2aConnectors.get(tool.connectorName);
+    const messagePayload = this.getMessagePayload(query, threadId);
+    const params = {
+      message: messagePayload
+    };
+    try {
+      const client = this.getOrCreateClient(connector);
+      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") {
+            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);
+          }
+        } else if (event.kind === "message") {
+          const msg = event;
+          const taskId = this.a2aTasks.get(threadId);
+          if (msg.taskId && msg.taskId !== taskId) {
+            this.a2aTasks.set(threadId, msg.taskId);
+          }
+        } else if (event.kind === "task") {
+          const task = event;
+          if (task.id !== this.a2aTasks.get(threadId)) {
+            this.a2aTasks.set(threadId, task.id);
+          }
+        } else {
+          _chunk466ED5WNcjs.loggers.a2a.warn("Received unknown event structure from stream:", {
+            event
+          });
+        }
+      }
+    } catch (error) {
+      _chunk466ED5WNcjs.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;
+    }
+    return `[Bot Called A2A Tool ${tool.toolName}]
+${finalText.join("\n")}`;
+  }
+}, _class);
+
+
+
+exports.A2AModule = A2AModule;
+//# sourceMappingURL=chunk-QCEMS4A3.cjs.map