@ainetwork/adk 0.2.14 → 0.3.1

package/dist/cjs/chunk-WGAEW3R3.cjs

@@ -0,0 +1,134 @@
+"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 _chunkQDX7LVBMcjs = require('./chunk-QDX7LVBM.cjs');
+
+
+var _chunkOU5UB2HFcjs = require('./chunk-OU5UB2HF.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, _chunkOU5UB2HFcjs.IntentTriggerService)(
+      modelModule,
+      memoryModule
+    );
+    this.intentFulfillStreamService = new (0, _chunkQDX7LVBMcjs.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: `Today's date: ${(/* @__PURE__ */ new Date()).toISOString().split("T")[0]} (YYYY-MM-DD format).
+	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 } };
+    }
+    const triggeredIntent = await this.intentTriggerService.intentTriggering(query, thread);
+    _chunk466ED5WNcjs.loggers.intent.debug("Triggered intents", { triggeredIntent });
+    await _optionalChain([threadMemory, 'optionalAccess', _8 => _8.addMessagesToThread, 'call', _9 => _9(userId, threadId, [
+      {
+        messageId: _crypto.randomUUID.call(void 0, ),
+        role: "USER" /* USER */,
+        timestamp: Date.now(),
+        content: { type: "text", parts: [query] },
+        metadata: {
+          intents: triggeredIntent.filter((intent) => !!intent.intent).map((intent) => ({
+            id: _optionalChain([intent, 'access', _10 => _10.intent, 'optionalAccess', _11 => _11.id]),
+            subquery: intent.subquery
+          }))
+        }
+      }
+    ])]);
+    const stream = this.intentFulfillStreamService.intentFulfillStream(
+      triggeredIntent,
+      thread
+    );
+    for await (const event of stream) {
+      yield event;
+    }
+  }
+};
+
+
+
+exports.QueryStreamService = QueryStreamService;
+//# sourceMappingURL=chunk-WGAEW3R3.cjs.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-WGAEW3R3.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,cAAA,EAAA,iBAAiB,IAAI,IAAA,CAAK,CAAA,CAAA,CAAE,WAAA,CAAY,CAAA,CAAE,KAAA,CAAM,GAAG,CAAA,CAAE,CAAC,CAAC,CAAA;AAAA;AAAA;AAAA;AAAA,oFAAA;AAAA,MAKtE,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;AAIY,IAAA;AACgC,IAAA;AAGI,IAAA;AAC/C,MAAA;AACuB,QAAA;AACtB,QAAA;AACoB,QAAA;AACoB,QAAA;AAC9B,QAAA;AAEc,UAAA;AAEF,YAAA;AACF,YAAA;AAChB,UAAA;AACJ,QAAA;AACD,MAAA;AACA,IAAA;AAG8C,IAAA;AAC9C,MAAA;AACA,MAAA;AACD,IAAA;AAEkC,IAAA;AAC3B,MAAA;AACP,IAAA;AACD,EAAA;AACD;AD1CyD;AACA;AACA;AACA","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-WGAEW3R3.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: `Today's date: ${new Date().toISOString().split(\"T\")[0]} (YYYY-MM-DD format).\n\tYou 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// 2. intent triggering\n\t\tconst triggeredIntent: Array<TriggeredIntent> =\n\t\t\tawait this.intentTriggerService.intentTriggering(query, thread);\n\t\tloggers.intent.debug(\"Triggered intents\", { triggeredIntent });\n\n\t\t// only add for storage, not for inference\n\t\tawait 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\tmetadata: {\n\t\t\t\t\tintents: triggeredIntent\n\t\t\t\t\t\t.filter((intent) => !!intent.intent)\n\t\t\t\t\t\t.map((intent) => ({\n\t\t\t\t\t\t\tid: intent.intent?.id,\n\t\t\t\t\t\t\tsubquery: intent.subquery,\n\t\t\t\t\t\t})),\n\t\t\t\t},\n\t\t\t},\n\t\t]);\n\n\t\t// 3. intent fulfillment\n\t\tconst stream = this.intentFulfillStreamService.intentFulfillStream(\n\t\t\ttriggeredIntent,\n\t\t\tthread,\n\t\t);\n\n\t\tfor await (const event of stream) {\n\t\t\tyield event;\n\t\t}\n\t}\n}\n"]}

package/dist/cjs/chunk-WKGN6P3D.cjs

@@ -0,0 +1,131 @@
+"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 _chunkUGDFRSMScjs = require('./chunk-UGDFRSMS.cjs');
+
+
+var _chunkOU5UB2HFcjs = require('./chunk-OU5UB2HF.cjs');
+
+
+var _chunk466ED5WNcjs = require('./chunk-466ED5WN.cjs');
+
+
+var _chunkZ6TQ7TFOcjs = require('./chunk-Z6TQ7TFO.cjs');
+
+// src/services/query.service.ts
+var _crypto = require('crypto');
+var _httpstatuscodes = require('http-status-codes');
+var QueryService = class {
+  
+  
+  
+  
+  constructor(modelModule, a2aModule, mcpModule, memoryModule) {
+    this.modelModule = modelModule;
+    this.memoryModule = memoryModule;
+    this.intentTriggerService = new (0, _chunkOU5UB2HFcjs.IntentTriggerService)(
+      modelModule,
+      memoryModule
+    );
+    this.intentFulfillService = new (0, _chunkUGDFRSMScjs.IntentFulfillService)(
+      modelModule,
+      a2aModule,
+      mcpModule,
+      memoryModule
+    );
+  }
+  /**
+   * 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: `
+Today is ${(/* @__PURE__ */ new Date()).toLocaleDateString()}.
+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.
+Always respond in the same language as the user's input.`
+      });
+      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, _chunkZ6TQ7TFOcjs.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}`);
+    }
+    const triggeredIntent = await this.intentTriggerService.intentTriggering(query, thread);
+    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] },
+        metadata: {
+          intents: triggeredIntent.filter((intent) => !!intent.intent).map((intent) => ({
+            id: _optionalChain([intent, 'access', _10 => _10.intent, 'optionalAccess', _11 => _11.id]),
+            subquery: intent.subquery
+          }))
+        }
+      }
+    ])]);
+    const result = await this.intentFulfillService.intentFulfill(
+      triggeredIntent,
+      thread
+    );
+    return { content: result };
+  }
+};
+
+
+
+exports.QueryService = QueryService;
+//# sourceMappingURL=chunk-WKGN6P3D.cjs.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-WKGN6P3D.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,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;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,SAAA,EAAA,iBACP,IAAI,IAAA,CAAK,CAAA,CAAA,CAAE,kBAAA,CAAmB,CAAC,CAAA;AAAA;AAAA;AAAA;AAAA;AAAA,wDAAA;AAAA,MAMvC,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;AAIY,IAAA;AAGoC,IAAA;AAC/C,MAAA;AACuB,QAAA;AACtB,QAAA;AACoB,QAAA;AACoB,QAAA;AAC9B,QAAA;AAEc,UAAA;AAEF,YAAA;AACF,YAAA;AAChB,UAAA;AACJ,QAAA;AACD,MAAA;AACA,IAAA;AAG8C,IAAA;AAC9C,MAAA;AACA,MAAA;AACD,IAAA;AAEyB,IAAA;AAC1B,EAAA;AACD;ADxCyD;AACA;AACA;AACA","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-WKGN6P3D.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 { 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) {\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);\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: `\nToday is ${new Date().toLocaleDateString()}.\nYou are a helpful assistant that generates titles for conversations.\nPlease analyze the user's query and create a concise title that accurately reflects the conversation's core topic.\nThe title must be no more than 5 words long.\nRespond with only the title. Do not include any punctuation or extra explanations.\nAlways respond in the same language as the user's input.`,\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// 2. intent triggering\n\t\tconst triggeredIntent: Array<TriggeredIntent> =\n\t\t\tawait this.intentTriggerService.intentTriggering(query, thread);\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\tmetadata: {\n\t\t\t\t\tintents: triggeredIntent\n\t\t\t\t\t\t.filter((intent) => !!intent.intent)\n\t\t\t\t\t\t.map((intent) => ({\n\t\t\t\t\t\t\tid: intent.intent?.id,\n\t\t\t\t\t\t\tsubquery: intent.subquery,\n\t\t\t\t\t\t})),\n\t\t\t\t},\n\t\t\t},\n\t\t]);\n\n\t\t// 3. intent fulfillment\n\t\tconst 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-X5INT5HE.cjs

@@ -0,0 +1,163 @@
+"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 _chunk7HBOMBZGcjs = require('./chunk-7HBOMBZG.cjs');
+
+
+var _chunk466ED5WNcjs = require('./chunk-466ED5WN.cjs');
+
+// src/services/intents/fulfill.service.ts
+var _crypto = require('crypto');
+var IntentFulfillService = class {
+  
+  
+  
+  
+  constructor(modelModule, a2aModule, mcpModule, memoryModule) {
+    this.modelModule = modelModule;
+    this.a2aModule = a2aModule;
+    this.mcpModule = mcpModule;
+    this.memoryModule = memoryModule;
+  }
+  async addToThreadMessages(thread, params) {
+    try {
+      const threadMemory = _optionalChain([this, 'access', _ => _.memoryModule, 'optionalAccess', _2 => _2.getThreadMemory, 'call', _3 => _3()]);
+      const { userId, threadId } = thread;
+      const newMessage = {
+        messageId: _crypto.randomUUID.call(void 0, ),
+        role: params.role,
+        timestamp: Date.now(),
+        content: { type: "text", parts: [params.content] },
+        metadata: params.metadata
+      };
+      thread.messages.push(newMessage);
+      await _optionalChain([threadMemory, 'optionalAccess', _4 => _4.addMessagesToThread, 'call', _5 => _5(userId, threadId, [newMessage])]);
+    } catch (error) {
+      _chunk466ED5WNcjs.loggers.intentStream.error("Error adding message to thread", error);
+    }
+  }
+  /**
+   * Fulfills the detected intent by generating a streaming response.
+   *
+   * Manages the complete inference loop including:
+   * - Loading prompts and conversation history
+   * - Collecting available tools from modules
+   * - Executing model inference with tool support
+   * - Processing tool calls iteratively until completion
+   * - Streaming results as Server-Sent Events
+   *
+   * @param query - The user's input query
+   * @param threadId - Thread identifier for context
+   * @param thread - Previous conversation history
+   * @param intent - Optional detected intent with custom prompt
+   * @returns AsyncGenerator yielding StreamEvent objects
+   */
+  async intentFulfilling(query, thread, intent) {
+    const agentMemory = _optionalChain([this, 'access', _6 => _6.memoryModule, 'optionalAccess', _7 => _7.getAgentMemory, 'call', _8 => _8()]);
+    const fulfillPrompt = await _chunk7HBOMBZGcjs.createFulfillPrompt.call(void 0, agentMemory, intent);
+    const modelInstance = this.modelModule.getModel();
+    const modelOptions = this.modelModule.getModelOptions();
+    const messages = modelInstance.generateMessages({
+      query,
+      thread,
+      systemPrompt: fulfillPrompt.trim()
+    });
+    _chunk466ED5WNcjs.loggers.intent.debug("Intent fulfillment start", {
+      threadId: thread.threadId,
+      messages
+    });
+    const tools = [];
+    this.mcpModule && tools.push(...this.mcpModule.getTools());
+    this.a2aModule && tools.push(...await this.a2aModule.getTools());
+    const functions = modelInstance.convertToolsToFunctions(tools);
+    let finalMessage = "";
+    while (true) {
+      const response = await modelInstance.fetchWithContextMessage(
+        messages,
+        functions,
+        modelOptions
+      );
+      const { content, toolCalls } = response;
+      _chunk466ED5WNcjs.loggers.intent.debug("Tool calls", {
+        threadId: thread.threadId,
+        content,
+        toolCalls
+      });
+      if (toolCalls) {
+        for (const toolCall of toolCalls) {
+          const toolName = toolCall.name;
+          const selectedTool = tools.filter(
+            (tool) => tool.toolName === 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,
+              query,
+              thread.threadId
+            );
+          } else {
+            _chunk466ED5WNcjs.loggers.intent.warn(
+              `Unrecognized tool type: ${selectedTool.protocol}`
+            );
+            continue;
+          }
+          _chunk466ED5WNcjs.loggers.intent.debug("Tool Result", {
+            threadId: thread.threadId,
+            toolResult
+          });
+          modelInstance.appendMessages(messages, toolResult);
+        }
+      } else if (content) {
+        finalMessage = content;
+        break;
+      }
+    }
+    return finalMessage;
+  }
+  /**
+   * Detects the intent from context.
+   *
+   * @param intents - The user's input query
+   * @param thread - The thread history
+   * @returns The detected intent
+   */
+  async intentFulfill(intents, thread) {
+    let finalResponseText = "";
+    for (let i = 0; i < intents.length; i++) {
+      const { subquery, intent, actionPlan } = intents[i];
+      _chunk466ED5WNcjs.loggers.intent.info(`Process query: ${subquery}, ${_optionalChain([intent, 'optionalAccess', _9 => _9.name])}`);
+      _chunk466ED5WNcjs.loggers.intent.info(`Action plan: ${actionPlan}`);
+      finalResponseText !== "" && thread.messages.push({
+        messageId: _crypto.randomUUID.call(void 0, ),
+        role: "MODEL" /* MODEL */,
+        timestamp: Date.now(),
+        content: { type: "text", parts: [finalResponseText] },
+        metadata: { isThinking: true }
+      });
+      await this.addToThreadMessages(thread, {
+        role: "MODEL" /* MODEL */,
+        content: subquery,
+        metadata: {
+          subquery,
+          isThinking: true,
+          actionPlan
+        }
+      });
+      finalResponseText = await this.intentFulfilling(subquery, thread, intent);
+    }
+    await this.addToThreadMessages(thread, {
+      role: "MODEL" /* MODEL */,
+      content: finalResponseText
+    });
+    return finalResponseText;
+  }
+};
+
+
+
+exports.IntentFulfillService = IntentFulfillService;
+//# sourceMappingURL=chunk-X5INT5HE.cjs.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-X5INT5HE.cjs","../../src/services/intents/fulfill.service.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACA;ACPA,gCAA2B;AAkBpB,IAAM,qBAAA,EAAN,MAA2B;AAAA,EACzB;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EAER,WAAA,CACC,WAAA,EACA,SAAA,EACA,SAAA,EACA,YAAA,EACC;AACD,IAAA,IAAA,CAAK,YAAA,EAAc,WAAA;AACnB,IAAA,IAAA,CAAK,UAAA,EAAY,SAAA;AACjB,IAAA,IAAA,CAAK,UAAA,EAAY,SAAA;AACjB,IAAA,IAAA,CAAK,aAAA,EAAe,YAAA;AAAA,EACrB;AAAA,EAEA,MAAc,mBAAA,CACb,MAAA,EACA,MAAA,EACC;AACD,IAAA,IAAI;AACH,MAAA,MAAM,aAAA,kBAAe,IAAA,mBAAK,YAAA,6BAAc,eAAA,mBAAgB,GAAA;AACxD,MAAA,MAAM,EAAE,MAAA,EAAQ,SAAS,EAAA,EAAI,MAAA;AAC7B,MAAA,MAAM,WAAA,EAA4B;AAAA,QACjC,SAAA,EAAW,gCAAA,CAAW;AAAA,QACtB,IAAA,EAAM,MAAA,CAAO,IAAA;AAAA,QACb,SAAA,EAAW,IAAA,CAAK,GAAA,CAAI,CAAA;AAAA,QACpB,OAAA,EAAS,EAAE,IAAA,EAAM,MAAA,EAAQ,KAAA,EAAO,CAAC,MAAA,CAAO,OAAO,EAAE,CAAA;AAAA,QACjD,QAAA,EAAU,MAAA,CAAO;AAAA,MAClB,CAAA;AACA,MAAA,MAAA,CAAO,QAAA,CAAS,IAAA,CAAK,UAAU,CAAA;AAC/B,MAAA,sBAAM,YAAA,6BAAc,mBAAA,mBAAoB,MAAA,EAAQ,QAAA,EAAU,CAAC,UAAU,CAAC,GAAA;AAAA,IACvE,EAAA,MAAA,CAAS,KAAA,EAAO;AACf,MAAA,yBAAA,CAAQ,YAAA,CAAa,KAAA,CAAM,gCAAA,EAAkC,KAAK,CAAA;AAAA,IACnE;AAAA,EACD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAkBA,MAAc,gBAAA,CACb,KAAA,EACA,MAAA,EACA,MAAA,EACkB;AAClB,IAAA,MAAM,YAAA,kBAAc,IAAA,qBAAK,YAAA,6BAAc,cAAA,mBAAe,GAAA;AACtD,IAAA,MAAM,cAAA,EAAgB,MAAM,mDAAA,WAAoB,EAAa,MAAM,CAAA;AAEnE,IAAA,MAAM,cAAA,EAAgB,IAAA,CAAK,WAAA,CAAY,QAAA,CAAS,CAAA;AAChD,IAAA,MAAM,aAAA,EAAe,IAAA,CAAK,WAAA,CAAY,eAAA,CAAgB,CAAA;AACtD,IAAA,MAAM,SAAA,EAAW,aAAA,CAAc,gBAAA,CAAiB;AAAA,MAC/C,KAAA;AAAA,MACA,MAAA;AAAA,MACA,YAAA,EAAc,aAAA,CAAc,IAAA,CAAK;AAAA,IAClC,CAAC,CAAA;AAED,IAAA,yBAAA,CAAQ,MAAA,CAAO,KAAA,CAAM,0BAAA,EAA4B;AAAA,MAChD,QAAA,EAAU,MAAA,CAAO,QAAA;AAAA,MACjB;AAAA,IACD,CAAC,CAAA;AAED,IAAA,MAAM,MAAA,EAAyB,CAAC,CAAA;AAChC,IAAA,IAAA,CAAK,UAAA,GAAa,KAAA,CAAM,IAAA,CAAK,GAAG,IAAA,CAAK,SAAA,CAAU,QAAA,CAAS,CAAC,CAAA;AACzD,IAAA,IAAA,CAAK,UAAA,GAAa,KAAA,CAAM,IAAA,CAAK,GAAI,MAAM,IAAA,CAAK,SAAA,CAAU,QAAA,CAAS,CAAE,CAAA;AAEjE,IAAA,MAAM,UAAA,EAAY,aAAA,CAAc,uBAAA,CAAwB,KAAK,CAAA;AAE7D,IAAA,IAAI,aAAA,EAAe,EAAA;AACnB,IAAA,MAAA,CAAO,IAAA,EAAM;AACZ,MAAA,MAAM,SAAA,EAAW,MAAM,aAAA,CAAc,uBAAA;AAAA,QACpC,QAAA;AAAA,QACA,SAAA;AAAA,QACA;AAAA,MACD,CAAA;AAEA,MAAA,MAAM,EAAE,OAAA,EAAS,UAAU,EAAA,EAAI,QAAA;AAC/B,MAAA,yBAAA,CAAQ,MAAA,CAAO,KAAA,CAAM,YAAA,EAAc;AAAA,QAClC,QAAA,EAAU,MAAA,CAAO,QAAA;AAAA,QACjB,OAAA;AAAA,QACA;AAAA,MACD,CAAC,CAAA;AAED,MAAA,GAAA,CAAI,SAAA,EAAW;AACd,QAAA,IAAA,CAAA,MAAW,SAAA,GAAY,SAAA,EAAW;AACjC,UAAA,MAAM,SAAA,EAAW,QAAA,CAAS,IAAA;AAC1B,UAAA,MAAM,aAAA,EAAe,KAAA,CAAM,MAAA;AAAA,YAC1B,CAAC,IAAA,EAAA,GAAS,IAAA,CAAK,SAAA,IAAa;AAAA,UAC7B,CAAA,CAAE,CAAC,CAAA;AAEH,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,CAAQ,YAAA,EAAc,QAAQ,CAAA;AAAA,UACjE,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,KAAA;AAAA,cACA,MAAA,CAAO;AAAA,YACR,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;AAAoC,YAAA;AAClB,YAAA;AACjB,UAAA;AAGD,UAAA;AAAiD,QAAA;AAClD,MAAA;AAEA,QAAA;AACA,QAAA;AAAA,MAAA;AACD,IAAA;AAGD,IAAA;AAAO,EAAA;AACR;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAAA;AAaC,IAAA;AACA,IAAA;AACC,MAAA;AACA,MAAA;AACA,MAAA;AAGA,MAAA;AACsB,QAAA;AACE,QAAA;AACtB,QAAA;AACoB,QAAA;AACgC,QAAA;AACvB,MAAA;AAE/B,MAAA;AAAuC,QAAA;AACtC,QAAA;AACS,QAAA;AACC,UAAA;AACT,UAAA;AACY,UAAA;AACZ,QAAA;AACD,MAAA;AAGD,MAAA;AAAwE,IAAA;AAGzE,IAAA;AAAuC,MAAA;AACtC,MAAA;AACS,IAAA;AAGV,IAAA;AAAO,EAAA;AAET;ADrDA;AACA;AACA;AACA","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-X5INT5HE.cjs","sourcesContent":[null,"import { randomUUID } from \"node:crypto\";\nimport type {\n\tA2AModule,\n\tMCPModule,\n\tMemoryModule,\n\tModelModule,\n} from \"@/modules\";\nimport { CONNECTOR_PROTOCOL_TYPE, type ConnectorTool } from \"@/types/connector\";\nimport {\n\ttype Intent,\n\ttype MessageObject,\n\tMessageRole,\n\ttype ThreadObject,\n\ttype TriggeredIntent,\n} from \"@/types/memory\";\nimport { loggers } from \"@/utils/logger\";\nimport { createFulfillPrompt } from \"../utils/fulfill.common\";\n\nexport class IntentFulfillService {\n\tprivate modelModule: ModelModule;\n\tprivate a2aModule?: A2AModule;\n\tprivate mcpModule?: MCPModule;\n\tprivate memoryModule?: MemoryModule;\n\n\tconstructor(\n\t\tmodelModule: ModelModule,\n\t\ta2aModule?: A2AModule,\n\t\tmcpModule?: MCPModule,\n\t\tmemoryModule?: MemoryModule,\n\t) {\n\t\tthis.modelModule = modelModule;\n\t\tthis.a2aModule = a2aModule;\n\t\tthis.mcpModule = mcpModule;\n\t\tthis.memoryModule = memoryModule;\n\t}\n\n\tprivate async addToThreadMessages(\n\t\tthread: ThreadObject,\n\t\tparams: { role: MessageRole; content: string; metadata?: any },\n\t) {\n\t\ttry {\n\t\t\tconst threadMemory = this.memoryModule?.getThreadMemory();\n\t\t\tconst { userId, threadId } = thread;\n\t\t\tconst newMessage: MessageObject = {\n\t\t\t\tmessageId: randomUUID(),\n\t\t\t\trole: params.role,\n\t\t\t\ttimestamp: Date.now(),\n\t\t\t\tcontent: { type: \"text\", parts: [params.content] },\n\t\t\t\tmetadata: params.metadata,\n\t\t\t};\n\t\t\tthread.messages.push(newMessage);\n\t\t\tawait threadMemory?.addMessagesToThread(userId, threadId, [newMessage]);\n\t\t} catch (error) {\n\t\t\tloggers.intentStream.error(\"Error adding message to thread\", error);\n\t\t}\n\t}\n\n\t/**\n\t * Fulfills the detected intent by generating a streaming response.\n\t *\n\t * Manages the complete inference loop including:\n\t * - Loading prompts and conversation history\n\t * - Collecting available tools from modules\n\t * - Executing model inference with tool support\n\t * - Processing tool calls iteratively until completion\n\t * - Streaming results as Server-Sent Events\n\t *\n\t * @param query - The user's input query\n\t * @param threadId - Thread identifier for context\n\t * @param thread - Previous conversation history\n\t * @param intent - Optional detected intent with custom prompt\n\t * @returns AsyncGenerator yielding StreamEvent objects\n\t */\n\tprivate async intentFulfilling(\n\t\tquery: string,\n\t\tthread: ThreadObject,\n\t\tintent?: Intent,\n\t): Promise<string> {\n\t\tconst agentMemory = this.memoryModule?.getAgentMemory();\n\t\tconst fulfillPrompt = await createFulfillPrompt(agentMemory, intent);\n\n\t\tconst modelInstance = this.modelModule.getModel();\n\t\tconst modelOptions = this.modelModule.getModelOptions();\n\t\tconst messages = modelInstance.generateMessages({\n\t\t\tquery,\n\t\t\tthread,\n\t\t\tsystemPrompt: fulfillPrompt.trim(),\n\t\t});\n\n\t\tloggers.intent.debug(\"Intent fulfillment start\", {\n\t\t\tthreadId: thread.threadId,\n\t\t\tmessages,\n\t\t});\n\n\t\tconst tools: ConnectorTool[] = [];\n\t\tthis.mcpModule && tools.push(...this.mcpModule.getTools());\n\t\tthis.a2aModule && tools.push(...(await this.a2aModule.getTools()));\n\n\t\tconst functions = modelInstance.convertToolsToFunctions(tools);\n\n\t\tlet finalMessage = \"\";\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\tmodelOptions,\n\t\t\t);\n\n\t\t\tconst { content, toolCalls } = response;\n\t\t\tloggers.intent.debug(\"Tool calls\", {\n\t\t\t\tthreadId: thread.threadId,\n\t\t\t\tcontent,\n\t\t\t\ttoolCalls,\n\t\t\t});\n\n\t\t\tif (toolCalls) {\n\t\t\t\tfor (const toolCall of toolCalls) {\n\t\t\t\t\tconst toolName = toolCall.name;\n\t\t\t\t\tconst selectedTool = tools.filter(\n\t\t\t\t\t\t(tool) => tool.toolName === toolName,\n\t\t\t\t\t)[0];\n\n\t\t\t\t\tlet toolResult = \"\";\n\t\t\t\t\tif (\n\t\t\t\t\t\tthis.mcpModule &&\n\t\t\t\t\t\tselectedTool.protocol === CONNECTOR_PROTOCOL_TYPE.MCP\n\t\t\t\t\t) {\n\t\t\t\t\t\tconst toolArgs = 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(selectedTool, toolArgs);\n\t\t\t\t\t} else if (\n\t\t\t\t\t\tthis.a2aModule &&\n\t\t\t\t\t\tselectedTool.protocol === CONNECTOR_PROTOCOL_TYPE.A2A\n\t\t\t\t\t) {\n\t\t\t\t\t\ttoolResult = await this.a2aModule.useTool(\n\t\t\t\t\t\t\tselectedTool,\n\t\t\t\t\t\t\tquery,\n\t\t\t\t\t\t\tthread.threadId,\n\t\t\t\t\t\t);\n\t\t\t\t\t} else {\n\t\t\t\t\t\t// Unrecognized tool type. It cannot be happened...\n\t\t\t\t\t\tloggers.intent.warn(\n\t\t\t\t\t\t\t`Unrecognized tool type: ${selectedTool.protocol}`,\n\t\t\t\t\t\t);\n\t\t\t\t\t\tcontinue;\n\t\t\t\t\t}\n\n\t\t\t\t\tloggers.intent.debug(\"Tool Result\", {\n\t\t\t\t\t\tthreadId: thread.threadId,\n\t\t\t\t\t\ttoolResult,\n\t\t\t\t\t});\n\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\tfinalMessage = content;\n\t\t\t\tbreak;\n\t\t\t}\n\t\t}\n\n\t\treturn finalMessage;\n\t}\n\n\t/**\n\t * Detects the intent from context.\n\t *\n\t * @param intents - The user's input query\n\t * @param thread - The thread history\n\t * @returns The detected intent\n\t */\n\tpublic async intentFulfill(\n\t\tintents: Array<TriggeredIntent>,\n\t\tthread: ThreadObject,\n\t): Promise<string> {\n\t\tlet finalResponseText = \"\";\n\t\tfor (let i = 0; i < intents.length; i++) {\n\t\t\tconst { subquery, intent, actionPlan } = intents[i];\n\t\t\tloggers.intent.info(`Process query: ${subquery}, ${intent?.name}`);\n\t\t\tloggers.intent.info(`Action plan: ${actionPlan}`);\n\n\t\t\t// only use for inference, not stored in memory\n\t\t\tfinalResponseText !== \"\" &&\n\t\t\t\tthread.messages.push({\n\t\t\t\t\tmessageId: randomUUID(),\n\t\t\t\t\trole: MessageRole.MODEL,\n\t\t\t\t\ttimestamp: Date.now(),\n\t\t\t\t\tcontent: { type: \"text\", parts: [finalResponseText] },\n\t\t\t\t\tmetadata: { isThinking: true },\n\t\t\t\t});\n\t\t\tawait this.addToThreadMessages(thread, {\n\t\t\t\trole: MessageRole.MODEL,\n\t\t\t\tcontent: subquery,\n\t\t\t\tmetadata: {\n\t\t\t\t\tsubquery,\n\t\t\t\t\tisThinking: true,\n\t\t\t\t\tactionPlan: actionPlan,\n\t\t\t\t},\n\t\t\t});\n\n\t\t\tfinalResponseText = await this.intentFulfilling(subquery, thread, intent);\n\t\t}\n\n\t\tawait this.addToThreadMessages(thread, {\n\t\t\trole: MessageRole.MODEL,\n\t\t\tcontent: finalResponseText,\n\t\t});\n\n\t\treturn finalResponseText;\n\t}\n}\n"]}

package/dist/cjs/chunk-X6R53KCL.cjs

@@ -0,0 +1,134 @@
+"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 _chunkTOVCM7I7cjs = require('./chunk-TOVCM7I7.cjs');
+
+
+var _chunkOU5UB2HFcjs = require('./chunk-OU5UB2HF.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, _chunkOU5UB2HFcjs.IntentTriggerService)(
+      modelModule,
+      memoryModule
+    );
+    this.intentFulfillStreamService = new (0, _chunkTOVCM7I7cjs.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: `Today's date: ${(/* @__PURE__ */ new Date()).toISOString().split("T")[0]} (YYYY-MM-DD format).
+	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 } };
+    }
+    const triggeredIntent = await this.intentTriggerService.intentTriggering(query, thread);
+    _chunk466ED5WNcjs.loggers.intent.debug("Triggered intents", { triggeredIntent });
+    await _optionalChain([threadMemory, 'optionalAccess', _8 => _8.addMessagesToThread, 'call', _9 => _9(userId, threadId, [
+      {
+        messageId: _crypto.randomUUID.call(void 0, ),
+        role: "USER" /* USER */,
+        timestamp: Date.now(),
+        content: { type: "text", parts: [query] },
+        metadata: {
+          intents: triggeredIntent.filter((intent) => !!intent.intent).map((intent) => ({
+            id: _optionalChain([intent, 'access', _10 => _10.intent, 'optionalAccess', _11 => _11.id]),
+            subquery: intent.subquery
+          }))
+        }
+      }
+    ])]);
+    const stream = this.intentFulfillStreamService.intentFulfillStream(
+      triggeredIntent,
+      thread
+    );
+    for await (const event of stream) {
+      yield event;
+    }
+  }
+};
+
+
+
+exports.QueryStreamService = QueryStreamService;
+//# sourceMappingURL=chunk-X6R53KCL.cjs.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-X6R53KCL.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,cAAA,EAAA,iBAAiB,IAAI,IAAA,CAAK,CAAA,CAAA,CAAE,WAAA,CAAY,CAAA,CAAE,KAAA,CAAM,GAAG,CAAA,CAAE,CAAC,CAAC,CAAA;AAAA;AAAA;AAAA;AAAA,oFAAA;AAAA,MAKtE,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;AAIY,IAAA;AACgC,IAAA;AAGI,IAAA;AAC/C,MAAA;AACuB,QAAA;AACtB,QAAA;AACoB,QAAA;AACoB,QAAA;AAC9B,QAAA;AAEc,UAAA;AAEF,YAAA;AACF,YAAA;AAChB,UAAA;AACJ,QAAA;AACD,MAAA;AACA,IAAA;AAG8C,IAAA;AAC9C,MAAA;AACA,MAAA;AACD,IAAA;AAEkC,IAAA;AAC3B,MAAA;AACP,IAAA;AACD,EAAA;AACD;AD1CyD;AACA;AACA;AACA","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-X6R53KCL.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: `Today's date: ${new Date().toISOString().split(\"T\")[0]} (YYYY-MM-DD format).\n\tYou 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// 2. intent triggering\n\t\tconst triggeredIntent: Array<TriggeredIntent> =\n\t\t\tawait this.intentTriggerService.intentTriggering(query, thread);\n\t\tloggers.intent.debug(\"Triggered intents\", { triggeredIntent });\n\n\t\t// only add for storage, not for inference\n\t\tawait 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\tmetadata: {\n\t\t\t\t\tintents: triggeredIntent\n\t\t\t\t\t\t.filter((intent) => !!intent.intent)\n\t\t\t\t\t\t.map((intent) => ({\n\t\t\t\t\t\t\tid: intent.intent?.id,\n\t\t\t\t\t\t\tsubquery: intent.subquery,\n\t\t\t\t\t\t})),\n\t\t\t\t},\n\t\t\t},\n\t\t]);\n\n\t\t// 3. intent fulfillment\n\t\tconst stream = this.intentFulfillStreamService.intentFulfillStream(\n\t\t\ttriggeredIntent,\n\t\t\tthread,\n\t\t);\n\n\t\tfor await (const event of stream) {\n\t\t\tyield event;\n\t\t}\n\t}\n}\n"]}