@ainetwork/adk 0.1.13 → 0.2.1

package/dist/esm/{chunk-EGR6OTTN.js → chunk-CWFIK2EK.js}

@@ -18,14 +18,58 @@ var QueryService = class {
     this.prompts = prompts;
   }
   /**
-   * Detects the intent from a user query.
+   * Detects the intent from context.
    *
    * @param query - The user's input query
-   * @returns The detected intent (currently returns the query as-is)
-   * @todo Implement actual intent detection logic
+   * @param thread - The thread history
+   * @returns The detected intent
    */
-  async intentTriggering(query) {
-    return query;
+  async intentTriggering(query, thread) {
+    const modelInstance = this.modelModule.getModel();
+    const intentMemory = this.memoryModule?.getIntentMemory();
+    if (!intentMemory) {
+      return void 0;
+    }
+    const intents = await intentMemory.listIntents();
+    if (intents.length === 0) {
+      loggers.intent.warn("No intent found");
+      return void 0;
+    }
+    const intentList = intents.map((intent2) => `- ${intent2.name}: ${intent2.description}`).join("\n");
+    const historyMessages = !thread ? "" : Object.entries(thread.messages).sort(([, a], [, b]) => a.timestamp - b.timestamp).map(([chatId, chat]) => {
+      const role = chat.role === "USER" ? "User" : chat.role === "MODEL" ? "Assistant" : "System";
+      const content = Array.isArray(chat.content.parts) ? chat.content.parts.join(" ") : String(chat.content.parts);
+      return `${role}: """${content}"""`;
+    }).join("\n");
+    const systemPrompt = `You are an expert in accurately identifying user intentions.
+
+Available intent list:
+${intentList}
+
+Please select and answer only from the above intent list. 
+Please return only the exact intent name without any additional explanations or text.`;
+    const userMessage = `The following is the conversation history with the user:
+
+${historyMessages}
+
+Last user question: "${query}"
+
+Based on the above conversation history, please determine what the intention of the last user question is. 
+Please select and answer the most appropriate intent name from the available intent list.`;
+    const messages = modelInstance.generateMessages({
+      query: userMessage,
+      systemPrompt
+    });
+    const response = await modelInstance.fetch(messages);
+    if (!response.content) {
+      throw new Error("No intent detected");
+    }
+    const intentName = response.content.trim();
+    const intent = await intentMemory.getIntentByName(intentName);
+    if (!intent) {
+      throw new Error(`No intent found: ${intentName}`);
+    }
+    return intent;
   }
   /**
    * Fulfills the detected intent by generating a response.
@@ -41,13 +85,15 @@ var QueryService = class {
    * @param thread - Previous conversation history
    * @returns Object containing process steps and final response
    */
-  async intentFulfilling(query, threadId, thread) {
+  async intentFulfilling(query, threadId, thread, intent) {
     const systemPrompt = `
 Today is ${(/* @__PURE__ */ new Date()).toLocaleDateString()}.
 
 ${this.prompts?.agent || ""}
 
 ${this.prompts?.system || ""}
+
+${intent?.prompt || ""}
     `;
     const modelInstance = this.modelModule.getModel();
     const messages = modelInstance.generateMessages({
@@ -65,13 +111,11 @@ ${this.prompts?.system || ""}
     const functions = modelInstance.convertToolsToFunctions(tools);
     const processList = [];
     let finalMessage = "";
-    let didCallTool = false;
     while (true) {
       const response = await modelInstance.fetchWithContextMessage(
         messages,
         functions
       );
-      didCallTool = false;
       loggers.intent.debug("messages", { messages });
       const { content, toolCalls } = response;
       loggers.intent.debug("content", { content });
@@ -83,7 +127,6 @@ ${this.prompts?.system || ""}
         );
         for (const toolCall of toolCalls) {
           const toolName = toolCall.name;
-          didCallTool = true;
           const selectedTool = tools.filter((tool) => tool.id === toolName)[0];
           let toolResult = "";
           if (this.mcpModule && selectedTool.protocol === "MCP" /* MCP */) {
@@ -112,8 +155,8 @@ ${this.prompts?.system || ""}
       } else if (content) {
         processList.push(content);
         finalMessage = content;
+        break;
       }
-      if (!didCallTool) break;
     }
     const botResponse = {
       process: processList.join("\n"),
@@ -169,7 +212,7 @@ ${this.prompts?.system || ""}
     let threadId = threadMetadata.threadId;
     let thread;
     if (threadId) {
-      thread = await threadMemory?.getThread(type, userId, threadId);
+      thread = await threadMemory?.getThread(userId, threadId);
     } else {
       threadId = randomUUID();
       const title = await this.generateTitle(query);
@@ -181,8 +224,8 @@ ${this.prompts?.system || ""}
       };
       loggers.intent.info("Create new thread", { metadata });
     }
-    const _intent = this.intentTriggering(query);
-    const result = await this.intentFulfilling(query, threadId, thread);
+    const intent = await this.intentTriggering(query, thread);
+    const result = await this.intentFulfilling(query, threadId, thread, intent);
     await threadMemory?.addMessagesToThread(userId, threadId, [
       {
         role: "USER" /* USER */,
@@ -202,4 +245,4 @@ ${this.prompts?.system || ""}
 export {
   QueryService
 };
-//# sourceMappingURL=chunk-EGR6OTTN.js.map
+//# sourceMappingURL=chunk-CWFIK2EK.js.map

package/dist/esm/chunk-CWFIK2EK.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/services/query.service.ts"],"sourcesContent":["import { randomUUID } from \"node:crypto\";\nimport type {\n\tA2AModule,\n\tMCPModule,\n\tMemoryModule,\n\tModelModule,\n} from \"@/modules/index.js\";\nimport type { AinAgentPrompts } from \"@/types/agent.js\";\nimport {\n\ttype Intent,\n\tMessageRole,\n\ttype ThreadMetadata,\n\ttype ThreadObject,\n\ttype ThreadType,\n} from \"@/types/memory.js\";\nimport {\n\ttype IA2ATool,\n\ttype IAgentTool,\n\ttype IMCPTool,\n\tTOOL_PROTOCOL_TYPE,\n} from \"@/types/tool.js\";\nimport { loggers } from \"@/utils/logger.js\";\n\n/**\n * Service for processing user queries through the agent's AI pipeline.\n *\n * Orchestrates the query processing workflow including intent detection,\n * model inference, tool execution, and response generation. Manages\n * conversation context and coordinates between different modules.\n */\nexport class QueryService {\n\tprivate modelModule: ModelModule;\n\tprivate a2aModule?: A2AModule;\n\tprivate mcpModule?: MCPModule;\n\tprivate memoryModule?: MemoryModule;\n\tprivate prompts?: AinAgentPrompts;\n\n\tconstructor(\n\t\tmodelModule: ModelModule,\n\t\ta2aModule?: A2AModule,\n\t\tmcpModule?: MCPModule,\n\t\tmemoryModule?: MemoryModule,\n\t\tprompts?: AinAgentPrompts,\n\t) {\n\t\tthis.modelModule = modelModule;\n\t\tthis.a2aModule = a2aModule;\n\t\tthis.mcpModule = mcpModule;\n\t\tthis.memoryModule = memoryModule;\n\t\tthis.prompts = prompts;\n\t}\n\n\t/**\n\t * Detects the intent from context.\n\t *\n\t * @param query - The user's input query\n\t * @param thread - The thread history\n\t * @returns The detected intent\n\t */\n\tprivate async intentTriggering(\n\t\tquery: string,\n\t\tthread: ThreadObject | undefined,\n\t): Promise<Intent | undefined> {\n\t\tconst modelInstance = this.modelModule.getModel();\n\t\tconst intentMemory = this.memoryModule?.getIntentMemory();\n\t\tif (!intentMemory) {\n\t\t\treturn undefined;\n\t\t}\n\n\t\t// 인텐트 목록 가져오기\n\t\tconst intents = await intentMemory.listIntents();\n\n\t\tif (intents.length === 0) {\n\t\t\tloggers.intent.warn(\"No intent found\");\n\t\t\treturn undefined;\n\t\t}\n\n\t\tconst intentList = intents\n\t\t\t.map((intent) => `- ${intent.name}: ${intent.description}`)\n\t\t\t.join(\"\\n\");\n\n\t\t// Convert session history to a string\n\t\tconst historyMessages = !thread\n\t\t\t? \"\"\n\t\t\t: Object.entries(thread.messages)\n\t\t\t\t\t.sort(([, a], [, b]) => a.timestamp - b.timestamp)\n\t\t\t\t\t.map(([chatId, chat]) => {\n\t\t\t\t\t\tconst role =\n\t\t\t\t\t\t\tchat.role === \"USER\"\n\t\t\t\t\t\t\t\t? \"User\"\n\t\t\t\t\t\t\t\t: chat.role === \"MODEL\"\n\t\t\t\t\t\t\t\t\t? \"Assistant\"\n\t\t\t\t\t\t\t\t\t: \"System\";\n\t\t\t\t\t\tconst content = Array.isArray(chat.content.parts)\n\t\t\t\t\t\t\t? chat.content.parts.join(\" \")\n\t\t\t\t\t\t\t: String(chat.content.parts);\n\t\t\t\t\t\treturn `${role}: \"\"\"${content}\"\"\"`;\n\t\t\t\t\t})\n\t\t\t\t\t.join(\"\\n\");\n\n\t\tconst systemPrompt = `You are an expert in accurately identifying user intentions.\n\nAvailable intent list:\n${intentList}\n\nPlease select and answer only from the above intent list. \nPlease return only the exact intent name without any additional explanations or text.`;\n\n\t\tconst userMessage = `The following is the conversation history with the user:\n\n${historyMessages}\n\nLast user question: \"${query}\"\n\nBased on the above conversation history, please determine what the intention of the last user question is. \nPlease select and answer the most appropriate intent name from the available intent list.`;\n\n\t\tconst messages = modelInstance.generateMessages({\n\t\t\tquery: userMessage,\n\t\t\tsystemPrompt,\n\t\t});\n\n\t\tconst response = await modelInstance.fetch(messages);\n\t\tif (!response.content) {\n\t\t\tthrow new Error(\"No intent detected\");\n\t\t}\n\t\tconst intentName = response.content.trim();\n\t\tconst intent = await intentMemory.getIntentByName(intentName);\n\t\tif (!intent) {\n\t\t\tthrow new Error(`No intent found: ${intentName}`);\n\t\t}\n\t\treturn intent;\n\t}\n\n\t/**\n\t * Fulfills the detected intent by generating a response.\n\t *\n\t * Manages the complete inference loop including:\n\t * - Loading prompts and conversation history\n\t * - Collecting available tools from modules\n\t * - Executing model inference with tool support\n\t * - Processing tool calls iteratively until completion\n\t *\n\t * @param query - The user's input query\n\t * @param threadId - Thread identifier for context\n\t * @param thread - Previous conversation history\n\t * @returns Object containing process steps and final response\n\t */\n\tprivate async intentFulfilling(\n\t\tquery: string,\n\t\tthreadId: string,\n\t\tthread?: ThreadObject,\n\t\tintent?: Intent,\n\t) {\n\t\t// 1. Load agent / system prompt from memory\n\t\tconst systemPrompt = `\nToday is ${new Date().toLocaleDateString()}.\n\n${this.prompts?.agent || \"\"}\n\n${this.prompts?.system || \"\"}\n\n${intent?.prompt || \"\"}\n    `;\n\t\t// NOTE(haechan@comcom.ai):\n\t\t// When the `intent.llm` is guaranteed to be consistent, it will be used as a parameter for getModel\n\t\t// const model_name = intent?.llm || \"gpt-4o\";\n\t\tconst modelInstance = this.modelModule.getModel();\n\t\tconst messages = modelInstance.generateMessages({\n\t\t\tquery,\n\t\t\tthread,\n\t\t\tsystemPrompt: systemPrompt.trim(),\n\t\t});\n\n\t\tconst tools: IAgentTool[] = [];\n\t\tif (this.mcpModule) {\n\t\t\ttools.push(...this.mcpModule.getTools());\n\t\t}\n\t\tif (this.a2aModule) {\n\t\t\ttools.push(...(await this.a2aModule.getTools()));\n\t\t}\n\t\tconst functions = modelInstance.convertToolsToFunctions(tools);\n\n\t\tconst processList: string[] = [];\n\t\tlet finalMessage = \"\";\n\n\t\twhile (true) {\n\t\t\tconst response = await modelInstance.fetchWithContextMessage(\n\t\t\t\tmessages,\n\t\t\t\tfunctions,\n\t\t\t);\n\n\t\t\tloggers.intent.debug(\"messages\", { messages });\n\n\t\t\tconst { content, toolCalls } = response;\n\n\t\t\tloggers.intent.debug(\"content\", { content });\n\t\t\tloggers.intent.debug(\"tool_calls\", { ...toolCalls });\n\n\t\t\tif (toolCalls) {\n\t\t\t\tconst messagePayload = this.a2aModule?.getMessagePayload(\n\t\t\t\t\tquery,\n\t\t\t\t\tthreadId,\n\t\t\t\t);\n\n\t\t\t\tfor (const toolCall of toolCalls) {\n\t\t\t\t\tconst toolName = toolCall.name;\n\t\t\t\t\tconst selectedTool = tools.filter((tool) => tool.id === toolName)[0];\n\n\t\t\t\t\tlet toolResult = \"\";\n\t\t\t\t\tif (\n\t\t\t\t\t\tthis.mcpModule &&\n\t\t\t\t\t\tselectedTool.protocol === TOOL_PROTOCOL_TYPE.MCP\n\t\t\t\t\t) {\n\t\t\t\t\t\tconst toolArgs = toolCall.arguments as\n\t\t\t\t\t\t\t| { [x: string]: unknown }\n\t\t\t\t\t\t\t| undefined;\n\t\t\t\t\t\tloggers.intent.debug(\"MCP tool call\", { toolName, toolArgs });\n\t\t\t\t\t\ttoolResult = await this.mcpModule.useTool(\n\t\t\t\t\t\t\tselectedTool as IMCPTool,\n\t\t\t\t\t\t\ttoolArgs,\n\t\t\t\t\t\t);\n\t\t\t\t\t} else if (\n\t\t\t\t\t\tthis.a2aModule &&\n\t\t\t\t\t\tselectedTool.protocol === TOOL_PROTOCOL_TYPE.A2A\n\t\t\t\t\t) {\n\t\t\t\t\t\ttoolResult = await this.a2aModule.useTool(\n\t\t\t\t\t\t\tselectedTool as IA2ATool,\n\t\t\t\t\t\t\tmessagePayload!,\n\t\t\t\t\t\t\tthreadId,\n\t\t\t\t\t\t);\n\t\t\t\t\t} else {\n\t\t\t\t\t\t// Unrecognized tool type. It cannot be happened...\n\t\t\t\t\t\tloggers.intent.warn(\n\t\t\t\t\t\t\t`Unrecognized tool type: ${selectedTool.protocol}`,\n\t\t\t\t\t\t);\n\t\t\t\t\t\tcontinue;\n\t\t\t\t\t}\n\n\t\t\t\t\tloggers.intent.debug(\"toolResult\", { toolResult });\n\n\t\t\t\t\tprocessList.push(toolResult);\n\t\t\t\t\tmodelInstance.appendMessages(messages, toolResult);\n\t\t\t\t}\n\t\t\t} else if (content) {\n\t\t\t\tprocessList.push(content);\n\t\t\t\tfinalMessage = content;\n\t\t\t\tbreak;\n\t\t\t}\n\t\t}\n\n\t\tconst botResponse = {\n\t\t\tprocess: processList.join(\"\\n\"),\n\t\t\tresponse: finalMessage,\n\t\t};\n\n\t\treturn botResponse;\n\t}\n\n\t/**\n\t * Generates a title for the conversation based on the query.\n\t *\n\t * @param query - The user's input query\n\t * @returns Promise resolving to a generated title\n\t */\n\n\tprivate async generateTitle(query: string): Promise<string> {\n\t\tconst DEFAULT_TITLE = \"New Chat\";\n\t\ttry {\n\t\t\tconst modelInstance = this.modelModule.getModel();\n\t\t\tconst messages = modelInstance.generateMessages({\n\t\t\t\tquery,\n\t\t\t\tsystemPrompt: `You are a helpful assistant that generates titles for conversations.\n  Please analyze the user's query and create a concise title that accurately reflects the conversation's core topic.\n  The title must be no more than 5 words long.\n  Respond with only the title. Do not include any punctuation or extra explanations.`,\n\t\t\t});\n\t\t\tconst response = await modelInstance.fetch(messages);\n\t\t\treturn response.content || DEFAULT_TITLE;\n\t\t} catch (error) {\n\t\t\tloggers.intent.error(\"Error generating title\", {\n\t\t\t\terror,\n\t\t\t\tquery,\n\t\t\t});\n\t\t\treturn DEFAULT_TITLE;\n\t\t}\n\t}\n\n\t/**\n\t * Main entry point for processing user queries.\n\t *\n\t * Handles the complete query lifecycle:\n\t * 1. Loads thread history from memory\n\t * 2. Detects intent from the query\n\t * 3. Fulfills the intent with AI response\n\t * 4. Updates conversation history\n\t *\n\t * @param type - The type of thread (e.g., chat, workflow)\n\t * @param userId - The user's unique identifier\n\t * @param threadId - Unique thread identifier\n\t * @param query - The user's input query\n\t */\n\tpublic async handleQuery(\n\t\tthreadMetadata: {\n\t\t\ttype: ThreadType;\n\t\t\tuserId: string;\n\t\t\tthreadId?: string;\n\t\t},\n\t\tquery: string,\n\t) {\n\t\t// 1. Load thread with threadId\n\t\tconst { type, userId } = threadMetadata;\n\t\tconst queryStartAt = Date.now();\n\t\tconst threadMemory = this.memoryModule?.getThreadMemory();\n\n\t\tlet threadId = threadMetadata.threadId;\n\t\tlet thread: ThreadObject | undefined;\n\n\t\tif (threadId) {\n\t\t\tthread = await threadMemory?.getThread(userId, threadId);\n\t\t} else {\n\t\t\tthreadId = randomUUID();\n\t\t\tconst title = await this.generateTitle(query);\n\n\t\t\tconst metadata =\n\t\t\t\t(await threadMemory?.createThread(type, userId, threadId, title)) ||\n\t\t\t\t({\n\t\t\t\t\ttype,\n\t\t\t\t\tthreadId,\n\t\t\t\t\ttitle,\n\t\t\t\t\tupdatedAt: Date.now(),\n\t\t\t\t} as ThreadMetadata);\n\t\t\tloggers.intent.info(\"Create new thread\", { metadata });\n\t\t}\n\n\t\t// 2. intent triggering\n\t\tconst intent = await this.intentTriggering(query, thread);\n\n\t\t// 3. intent fulfillment\n\t\tconst result = await this.intentFulfilling(query, threadId, thread, intent);\n\t\tawait threadMemory?.addMessagesToThread(userId, threadId, [\n\t\t\t{\n\t\t\t\trole: MessageRole.USER,\n\t\t\t\ttimestamp: queryStartAt,\n\t\t\t\tcontent: { type: \"text\", parts: [query] },\n\t\t\t},\n\t\t\t{\n\t\t\t\trole: MessageRole.MODEL,\n\t\t\t\ttimestamp: Date.now(),\n\t\t\t\tcontent: { type: \"text\", parts: [result.response] },\n\t\t\t},\n\t\t]);\n\n\t\treturn { content: result.response };\n\t}\n}\n"],"mappings":";;;;;AAAA,SAAS,kBAAkB;AA8BpB,IAAM,eAAN,MAAmB;AAAA,EACjB;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EAER,YACC,aACA,WACA,WACA,cACA,SACC;AACD,SAAK,cAAc;AACnB,SAAK,YAAY;AACjB,SAAK,YAAY;AACjB,SAAK,eAAe;AACpB,SAAK,UAAU;AAAA,EAChB;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EASA,MAAc,iBACb,OACA,QAC8B;AAC9B,UAAM,gBAAgB,KAAK,YAAY,SAAS;AAChD,UAAM,eAAe,KAAK,cAAc,gBAAgB;AACxD,QAAI,CAAC,cAAc;AAClB,aAAO;AAAA,IACR;AAGA,UAAM,UAAU,MAAM,aAAa,YAAY;AAE/C,QAAI,QAAQ,WAAW,GAAG;AACzB,cAAQ,OAAO,KAAK,iBAAiB;AACrC,aAAO;AAAA,IACR;AAEA,UAAM,aAAa,QACjB,IAAI,CAACA,YAAW,KAAKA,QAAO,IAAI,KAAKA,QAAO,WAAW,EAAE,EACzD,KAAK,IAAI;AAGX,UAAM,kBAAkB,CAAC,SACtB,KACA,OAAO,QAAQ,OAAO,QAAQ,EAC7B,KAAK,CAAC,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,MAAM,EAAE,YAAY,EAAE,SAAS,EAChD,IAAI,CAAC,CAAC,QAAQ,IAAI,MAAM;AACxB,YAAM,OACL,KAAK,SAAS,SACX,SACA,KAAK,SAAS,UACb,cACA;AACL,YAAM,UAAU,MAAM,QAAQ,KAAK,QAAQ,KAAK,IAC7C,KAAK,QAAQ,MAAM,KAAK,GAAG,IAC3B,OAAO,KAAK,QAAQ,KAAK;AAC5B,aAAO,GAAG,IAAI,QAAQ,OAAO;AAAA,IAC9B,CAAC,EACA,KAAK,IAAI;AAEb,UAAM,eAAe;AAAA;AAAA;AAAA,EAGrB,UAAU;AAAA;AAAA;AAAA;AAKV,UAAM,cAAc;AAAA;AAAA,EAEpB,eAAe;AAAA;AAAA,uBAEM,KAAK;AAAA;AAAA;AAAA;AAK1B,UAAM,WAAW,cAAc,iBAAiB;AAAA,MAC/C,OAAO;AAAA,MACP;AAAA,IACD,CAAC;AAED,UAAM,WAAW,MAAM,cAAc,MAAM,QAAQ;AACnD,QAAI,CAAC,SAAS,SAAS;AACtB,YAAM,IAAI,MAAM,oBAAoB;AAAA,IACrC;AACA,UAAM,aAAa,SAAS,QAAQ,KAAK;AACzC,UAAM,SAAS,MAAM,aAAa,gBAAgB,UAAU;AAC5D,QAAI,CAAC,QAAQ;AACZ,YAAM,IAAI,MAAM,oBAAoB,UAAU,EAAE;AAAA,IACjD;AACA,WAAO;AAAA,EACR;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAgBA,MAAc,iBACb,OACA,UACA,QACA,QACC;AAED,UAAM,eAAe;AAAA,YACZ,oBAAI,KAAK,GAAE,mBAAmB,CAAC;AAAA;AAAA,EAExC,KAAK,SAAS,SAAS,EAAE;AAAA;AAAA,EAEzB,KAAK,SAAS,UAAU,EAAE;AAAA;AAAA,EAE1B,QAAQ,UAAU,EAAE;AAAA;AAKpB,UAAM,gBAAgB,KAAK,YAAY,SAAS;AAChD,UAAM,WAAW,cAAc,iBAAiB;AAAA,MAC/C;AAAA,MACA;AAAA,MACA,cAAc,aAAa,KAAK;AAAA,IACjC,CAAC;AAED,UAAM,QAAsB,CAAC;AAC7B,QAAI,KAAK,WAAW;AACnB,YAAM,KAAK,GAAG,KAAK,UAAU,SAAS,CAAC;AAAA,IACxC;AACA,QAAI,KAAK,WAAW;AACnB,YAAM,KAAK,GAAI,MAAM,KAAK,UAAU,SAAS,CAAE;AAAA,IAChD;AACA,UAAM,YAAY,cAAc,wBAAwB,KAAK;AAE7D,UAAM,cAAwB,CAAC;AAC/B,QAAI,eAAe;AAEnB,WAAO,MAAM;AACZ,YAAM,WAAW,MAAM,cAAc;AAAA,QACpC;AAAA,QACA;AAAA,MACD;AAEA,cAAQ,OAAO,MAAM,YAAY,EAAE,SAAS,CAAC;AAE7C,YAAM,EAAE,SAAS,UAAU,IAAI;AAE/B,cAAQ,OAAO,MAAM,WAAW,EAAE,QAAQ,CAAC;AAC3C,cAAQ,OAAO,MAAM,cAAc,EAAE,GAAG,UAAU,CAAC;AAEnD,UAAI,WAAW;AACd,cAAM,iBAAiB,KAAK,WAAW;AAAA,UACtC;AAAA,UACA;AAAA,QACD;AAEA,mBAAW,YAAY,WAAW;AACjC,gBAAM,WAAW,SAAS;AAC1B,gBAAM,eAAe,MAAM,OAAO,CAAC,SAAS,KAAK,OAAO,QAAQ,EAAE,CAAC;AAEnE,cAAI,aAAa;AACjB,cACC,KAAK,aACL,aAAa,8BACZ;AACD,kBAAM,WAAW,SAAS;AAG1B,oBAAQ,OAAO,MAAM,iBAAiB,EAAE,UAAU,SAAS,CAAC;AAC5D,yBAAa,MAAM,KAAK,UAAU;AAAA,cACjC;AAAA,cACA;AAAA,YACD;AAAA,UACD,WACC,KAAK,aACL,aAAa,8BACZ;AACD,yBAAa,MAAM,KAAK,UAAU;AAAA,cACjC;AAAA,cACA;AAAA,cACA;AAAA,YACD;AAAA,UACD,OAAO;AAEN,oBAAQ,OAAO;AAAA,cACd,2BAA2B,aAAa,QAAQ;AAAA,YACjD;AACA;AAAA,UACD;AAEA,kBAAQ,OAAO,MAAM,cAAc,EAAE,WAAW,CAAC;AAEjD,sBAAY,KAAK,UAAU;AAC3B,wBAAc,eAAe,UAAU,UAAU;AAAA,QAClD;AAAA,MACD,WAAW,SAAS;AACnB,oBAAY,KAAK,OAAO;AACxB,uBAAe;AACf;AAAA,MACD;AAAA,IACD;AAEA,UAAM,cAAc;AAAA,MACnB,SAAS,YAAY,KAAK,IAAI;AAAA,MAC9B,UAAU;AAAA,IACX;AAEA,WAAO;AAAA,EACR;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EASA,MAAc,cAAc,OAAgC;AAC3D,UAAM,gBAAgB;AACtB,QAAI;AACH,YAAM,gBAAgB,KAAK,YAAY,SAAS;AAChD,YAAM,WAAW,cAAc,iBAAiB;AAAA,QAC/C;AAAA,QACA,cAAc;AAAA;AAAA;AAAA;AAAA,MAIf,CAAC;AACD,YAAM,WAAW,MAAM,cAAc,MAAM,QAAQ;AACnD,aAAO,SAAS,WAAW;AAAA,IAC5B,SAAS,OAAO;AACf,cAAQ,OAAO,MAAM,0BAA0B;AAAA,QAC9C;AAAA,QACA;AAAA,MACD,CAAC;AACD,aAAO;AAAA,IACR;AAAA,EACD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAgBA,MAAa,YACZ,gBAKA,OACC;AAED,UAAM,EAAE,MAAM,OAAO,IAAI;AACzB,UAAM,eAAe,KAAK,IAAI;AAC9B,UAAM,eAAe,KAAK,cAAc,gBAAgB;AAExD,QAAI,WAAW,eAAe;AAC9B,QAAI;AAEJ,QAAI,UAAU;AACb,eAAS,MAAM,cAAc,UAAU,QAAQ,QAAQ;AAAA,IACxD,OAAO;AACN,iBAAW,WAAW;AACtB,YAAM,QAAQ,MAAM,KAAK,cAAc,KAAK;AAE5C,YAAM,WACJ,MAAM,cAAc,aAAa,MAAM,QAAQ,UAAU,KAAK,KAC9D;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA,WAAW,KAAK,IAAI;AAAA,MACrB;AACD,cAAQ,OAAO,KAAK,qBAAqB,EAAE,SAAS,CAAC;AAAA,IACtD;AAGA,UAAM,SAAS,MAAM,KAAK,iBAAiB,OAAO,MAAM;AAGxD,UAAM,SAAS,MAAM,KAAK,iBAAiB,OAAO,UAAU,QAAQ,MAAM;AAC1E,UAAM,cAAc,oBAAoB,QAAQ,UAAU;AAAA,MACzD;AAAA,QACC;AAAA,QACA,WAAW;AAAA,QACX,SAAS,EAAE,MAAM,QAAQ,OAAO,CAAC,KAAK,EAAE;AAAA,MACzC;AAAA,MACA;AAAA,QACC;AAAA,QACA,WAAW,KAAK,IAAI;AAAA,QACpB,SAAS,EAAE,MAAM,QAAQ,OAAO,CAAC,OAAO,QAAQ,EAAE;AAAA,MACnD;AAAA,IACD,CAAC;AAED,WAAO,EAAE,SAAS,OAAO,SAAS;AAAA,EACnC;AACD;","names":["intent"]}

package/dist/esm/{chunk-JCWH7LIC.js → chunk-EIHZ4S5V.js}

@@ -1,12 +1,12 @@
 import {
   QueryStreamService
-} from "./chunk-2BYUKYVB.js";
-import {
-  QueryService
-} from "./chunk-EGR6OTTN.js";
+} from "./chunk-ILKEX4ZV.js";
 import {
   QueryController
 } from "./chunk-7D4SKPIS.js";
+import {
+  QueryService
+} from "./chunk-CWFIK2EK.js";
 
 // src/routes/query.routes.ts
 import { Router } from "express";
@@ -40,4 +40,4 @@ var createQueryRouter = (agent, allowStream = false) => {
 export {
   createQueryRouter
 };
-//# sourceMappingURL=chunk-JCWH7LIC.js.map
+//# sourceMappingURL=chunk-EIHZ4S5V.js.map

package/dist/esm/{chunk-2BYUKYVB.js → chunk-ILKEX4ZV.js}

@@ -18,14 +18,58 @@ var QueryStreamService = class {
     this.prompts = prompts;
   }
   /**
-   * Detects the intent from a user query.
+   * Detects the intent from context.
    *
    * @param query - The user's input query
-   * @returns The detected intent (currently returns the query as-is)
-   * @todo Implement actual intent detection logic
+   * @param thread - The thread history
+   * @returns The detected intent
    */
-  async intentTriggering(query) {
-    return query;
+  async intentTriggering(query, thread) {
+    const modelInstance = this.modelModule.getModel();
+    const intentMemory = this.memoryModule?.getIntentMemory();
+    if (!intentMemory) {
+      return void 0;
+    }
+    const intents = await intentMemory.listIntents();
+    if (intents.length === 0) {
+      loggers.intentStream.warn("No intent found");
+      return void 0;
+    }
+    const intentList = intents.map((intent2) => `- ${intent2.name}: ${intent2.description}`).join("\n");
+    const historyMessages = !thread ? "" : Object.entries(thread.messages).sort(([, a], [, b]) => a.timestamp - b.timestamp).map(([_chatId, chat]) => {
+      const role = chat.role === "USER" ? "User" : chat.role === "MODEL" ? "Assistant" : "System";
+      const content = Array.isArray(chat.content.parts) ? chat.content.parts.join(" ") : String(chat.content.parts);
+      return `${role}: """${content}"""`;
+    }).join("\n");
+    const systemPrompt = `You are an expert in accurately identifying user intentions.
+
+Available intent list:
+${intentList}
+
+Please select and answer only from the above intent list. 
+Please return only the exact intent name without any additional explanations or text.`;
+    const userMessage = `The following is the conversation history with the user:
+
+${historyMessages}
+
+Last user question: "${query}"
+
+Based on the above conversation history, please determine what the intention of the last user question is. 
+Please select and answer the most appropriate intent name from the available intent list.`;
+    const messages = modelInstance.generateMessages({
+      query: userMessage,
+      systemPrompt
+    });
+    const response = await modelInstance.fetch(messages);
+    if (!response.content) {
+      throw new Error("No intent detected");
+    }
+    const intentName = response.content.trim();
+    const intent = await intentMemory.getIntentByName(intentName);
+    if (!intent) {
+      throw new Error(`No intent found: ${intentName}`);
+    }
+    return intent;
   }
   /**
    * Fulfills the detected intent by generating a response.
@@ -41,14 +85,16 @@ var QueryStreamService = class {
    * @param thread - Previous conversation history
    * @returns Object containing process steps and final response
    */
-  async *intentFulfilling(query, threadId, thread) {
+  async *intentFulfilling(query, threadId, thread, intent) {
     const systemPrompt = `
 Today is ${(/* @__PURE__ */ new Date()).toLocaleDateString()}.
 
 ${this.prompts?.agent || ""}
 
 ${this.prompts?.system || ""}
-	`;
+
+${intent?.prompt || ""}
+	`.trim();
     const modelInstance = this.modelModule.getModel();
     const messages = modelInstance.generateMessages({
       query,
@@ -97,6 +143,7 @@ ${this.prompts?.system || ""}
           threadId
         );
         for (const toolCall of assembledToolCalls) {
+          const toolCallId = randomUUID();
           const toolName = toolCall.function.name;
           const selectedTool = tools.filter((tool) => tool.id === toolName)[0];
           let toolResult = "";
@@ -104,7 +151,12 @@ ${this.prompts?.system || ""}
             const toolArgs = JSON.parse(toolCall.function.arguments);
             yield {
               event: "tool_start",
-              data: { protocol: "MCP" /* MCP */, toolName, toolArgs }
+              data: {
+                toolCallId,
+                protocol: "MCP" /* MCP */,
+                toolName,
+                toolArgs
+              }
             };
             loggers.intent.debug("MCP tool call", { toolName, toolArgs });
             toolResult = await this.mcpModule.useTool(
@@ -115,6 +167,7 @@ ${this.prompts?.system || ""}
             yield {
               event: "tool_start",
               data: {
+                toolCallId,
                 protocol: "A2A" /* A2A */,
                 toolName,
                 toolArgs: null
@@ -136,6 +189,7 @@ ${this.prompts?.system || ""}
           yield {
             event: "tool_output",
             data: {
+              toolCallId,
               protocol: selectedTool.protocol,
               toolName,
               result: toolResult
@@ -199,7 +253,7 @@ ${this.prompts?.system || ""}
     let threadId = threadMetadata.threadId;
     let thread;
     if (threadId) {
-      thread = await threadMemory?.getThread(type, userId, threadId);
+      thread = await threadMemory?.getThread(userId, threadId);
     } else {
       threadId = randomUUID();
       const title = await this.generateTitle(query);
@@ -212,22 +266,53 @@ ${this.prompts?.system || ""}
       loggers.intentStream.info("Create new thread", { metadata });
       yield { event: "thread_id", data: metadata };
     }
-    const _intent = await this.intentTriggering(query);
-    const stream = this.intentFulfilling(query, threadId, thread);
+    await threadMemory?.addMessagesToThread(userId, threadId, [
+      {
+        role: "USER" /* USER */,
+        timestamp: queryStartAt,
+        content: { type: "text", parts: [query] }
+      }
+    ]);
+    const intent = await this.intentTriggering(query, thread);
+    const stream = this.intentFulfilling(query, threadId, thread, intent);
     let finalResponseText = "";
     for await (const event of stream) {
       if (event.event === "text_chunk" && event.data.delta) {
         loggers.intentStream.debug("text_chunk", { event });
         finalResponseText += event.data.delta;
+      } else if (event.event === "tool_start") {
+        await threadMemory?.addMessagesToThread(userId, threadId, [
+          {
+            role: "MODEL" /* MODEL */,
+            timestamp: Date.now(),
+            content: {
+              type: "text",
+              parts: [JSON.stringify(event.data.toolArgs)]
+            },
+            metadata: {
+              toolCallId: event.data.toolCallId,
+              toolName: event.data.toolName,
+              protocol: event.data.protocol
+            }
+          }
+        ]);
+      } else if (event.event === "tool_output") {
+        await threadMemory?.addMessagesToThread(userId, threadId, [
+          {
+            role: "MODEL" /* MODEL */,
+            timestamp: Date.now(),
+            content: { type: "text", parts: [event.data.result] },
+            metadata: {
+              toolCallId: event.data.toolCallId,
+              toolName: event.data.toolName,
+              protocol: event.data.protocol
+            }
+          }
+        ]);
       }
       yield event;
     }
     await threadMemory?.addMessagesToThread(userId, threadId, [
-      {
-        role: "USER" /* USER */,
-        timestamp: queryStartAt,
-        content: { type: "text", parts: [query] }
-      },
       {
         role: "MODEL" /* MODEL */,
         timestamp: Date.now(),
@@ -240,4 +325,4 @@ ${this.prompts?.system || ""}
 export {
   QueryStreamService
 };
-//# sourceMappingURL=chunk-2BYUKYVB.js.map
+//# sourceMappingURL=chunk-ILKEX4ZV.js.map

package/dist/esm/chunk-ILKEX4ZV.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/services/query-stream.service.ts"],"sourcesContent":["import { randomUUID } from \"node:crypto\";\nimport type {\n\tA2AModule,\n\tMCPModule,\n\tMemoryModule,\n\tModelModule,\n} from \"@/modules/index.js\";\nimport type { AinAgentPrompts } from \"@/types/agent.js\";\nimport {\n\ttype Intent,\n\tMessageRole,\n\ttype ThreadMetadata,\n\ttype ThreadObject,\n\ttype ThreadType,\n} from \"@/types/memory.js\";\nimport type { StreamEvent } from \"@/types/stream\";\nimport {\n\ttype IA2ATool,\n\ttype IAgentTool,\n\ttype IMCPTool,\n\tTOOL_PROTOCOL_TYPE,\n} from \"@/types/tool.js\";\nimport { loggers } from \"@/utils/logger.js\";\n\n/**\n * Service for processing user queries through the agent's AI pipeline.\n *\n * Orchestrates the query processing workflow including intent detection,\n * model inference, tool execution, and response generation. Manages\n * conversation context and coordinates between different modules.\n */\nexport class QueryStreamService {\n\tprivate modelModule: ModelModule;\n\tprivate a2aModule?: A2AModule;\n\tprivate mcpModule?: MCPModule;\n\tprivate memoryModule?: MemoryModule;\n\tprivate prompts?: AinAgentPrompts;\n\n\tconstructor(\n\t\tmodelModule: ModelModule,\n\t\ta2aModule?: A2AModule,\n\t\tmcpModule?: MCPModule,\n\t\tmemoryModule?: MemoryModule,\n\t\tprompts?: AinAgentPrompts,\n\t) {\n\t\tthis.modelModule = modelModule;\n\t\tthis.a2aModule = a2aModule;\n\t\tthis.mcpModule = mcpModule;\n\t\tthis.memoryModule = memoryModule;\n\t\tthis.prompts = prompts;\n\t}\n\n\t/**\n\t * Detects the intent from context.\n\t *\n\t * @param query - The user's input query\n\t * @param thread - The thread history\n\t * @returns The detected intent\n\t */\n\tprivate async intentTriggering(\n\t\tquery: string,\n\t\tthread: ThreadObject | undefined,\n\t): Promise<Intent | undefined> {\n\t\tconst modelInstance = this.modelModule.getModel();\n\t\tconst intentMemory = this.memoryModule?.getIntentMemory();\n\t\tif (!intentMemory) {\n\t\t\treturn undefined;\n\t\t}\n\n\t\t// 인텐트 목록 가져오기\n\t\tconst intents = await intentMemory.listIntents();\n\n\t\tif (intents.length === 0) {\n\t\t\tloggers.intentStream.warn(\"No intent found\");\n\t\t\treturn undefined;\n\t\t}\n\n\t\tconst intentList = intents\n\t\t\t.map((intent) => `- ${intent.name}: ${intent.description}`)\n\t\t\t.join(\"\\n\");\n\n\t\t// Convert session history to a string\n\t\tconst historyMessages = !thread\n\t\t\t? \"\"\n\t\t\t: Object.entries(thread.messages)\n\t\t\t\t\t.sort(([, a], [, b]) => a.timestamp - b.timestamp)\n\t\t\t\t\t.map(([_chatId, chat]) => {\n\t\t\t\t\t\tconst role =\n\t\t\t\t\t\t\tchat.role === \"USER\"\n\t\t\t\t\t\t\t\t? \"User\"\n\t\t\t\t\t\t\t\t: chat.role === \"MODEL\"\n\t\t\t\t\t\t\t\t\t? \"Assistant\"\n\t\t\t\t\t\t\t\t\t: \"System\";\n\t\t\t\t\t\tconst content = Array.isArray(chat.content.parts)\n\t\t\t\t\t\t\t? chat.content.parts.join(\" \")\n\t\t\t\t\t\t\t: String(chat.content.parts);\n\t\t\t\t\t\treturn `${role}: \"\"\"${content}\"\"\"`;\n\t\t\t\t\t})\n\t\t\t\t\t.join(\"\\n\");\n\n\t\tconst systemPrompt = `You are an expert in accurately identifying user intentions.\n\nAvailable intent list:\n${intentList}\n\nPlease select and answer only from the above intent list. \nPlease return only the exact intent name without any additional explanations or text.`;\n\n\t\tconst userMessage = `The following is the conversation history with the user:\n\n${historyMessages}\n\nLast user question: \"${query}\"\n\nBased on the above conversation history, please determine what the intention of the last user question is. \nPlease select and answer the most appropriate intent name from the available intent list.`;\n\n\t\tconst messages = modelInstance.generateMessages({\n\t\t\tquery: userMessage,\n\t\t\tsystemPrompt,\n\t\t});\n\n\t\tconst response = await modelInstance.fetch(messages);\n\t\tif (!response.content) {\n\t\t\tthrow new Error(\"No intent detected\");\n\t\t}\n\t\tconst intentName = response.content.trim();\n\t\tconst intent = await intentMemory.getIntentByName(intentName);\n\t\tif (!intent) {\n\t\t\tthrow new Error(`No intent found: ${intentName}`);\n\t\t}\n\t\treturn intent;\n\t}\n\n\t/**\n\t * Fulfills the detected intent by generating a response.\n\t *\n\t * Manages the complete inference loop including:\n\t * - Loading prompts and conversation history\n\t * - Collecting available tools from modules\n\t * - Executing model inference with tool support\n\t * - Processing tool calls iteratively until completion\n\t *\n\t * @param query - The user's input query\n\t * @param threadId - Thread identifier for context\n\t * @param thread - Previous conversation history\n\t * @returns Object containing process steps and final response\n\t */\n\tpublic async *intentFulfilling(\n\t\tquery: string,\n\t\tthreadId: string,\n\t\tthread?: ThreadObject,\n\t\tintent?: Intent,\n\t): AsyncGenerator<StreamEvent> {\n\t\tconst systemPrompt = `\nToday is ${new Date().toLocaleDateString()}.\n\n${this.prompts?.agent || \"\"}\n\n${this.prompts?.system || \"\"}\n\n${intent?.prompt || \"\"}\n\t`.trim();\n\n\t\tconst modelInstance = this.modelModule.getModel();\n\t\tconst messages = modelInstance.generateMessages({\n\t\t\tquery,\n\t\t\tthread,\n\t\t\tsystemPrompt: systemPrompt.trim(),\n\t\t});\n\n\t\tconst tools: IAgentTool[] = [];\n\t\tthis.mcpModule && tools.push(...this.mcpModule.getTools());\n\t\tthis.a2aModule && tools.push(...(await this.a2aModule.getTools()));\n\n\t\tconst functions = modelInstance.convertToolsToFunctions(tools);\n\n\t\tconst processList: string[] = [];\n\n\t\twhile (true) {\n\t\t\tconst responseStream = await modelInstance.fetchStreamWithContextMessage(\n\t\t\t\tmessages,\n\t\t\t\tfunctions,\n\t\t\t);\n\n\t\t\tconst assembledToolCalls: {\n\t\t\t\tid: string;\n\t\t\t\ttype: \"function\";\n\t\t\t\tfunction: { name: string; arguments: string };\n\t\t\t}[] = [];\n\n\t\t\tloggers.intentStream.debug(\"messages\", { messages });\n\n\t\t\tfor await (const chunk of responseStream) {\n\t\t\t\tconst delta = chunk.delta;\n\t\t\t\tif (delta?.tool_calls) {\n\t\t\t\t\tfor (const { index, id, function: func } of delta.tool_calls) {\n\t\t\t\t\t\tassembledToolCalls[index] ??= {\n\t\t\t\t\t\t\tid: \"\",\n\t\t\t\t\t\t\ttype: \"function\",\n\t\t\t\t\t\t\tfunction: { name: \"\", arguments: \"\" },\n\t\t\t\t\t\t};\n\n\t\t\t\t\t\tif (id) assembledToolCalls[index].id = id;\n\t\t\t\t\t\tif (func?.name) assembledToolCalls[index].function.name = func.name;\n\t\t\t\t\t\tif (func?.arguments)\n\t\t\t\t\t\t\tassembledToolCalls[index].function.arguments += func.arguments;\n\t\t\t\t\t}\n\t\t\t\t} else if (chunk.delta?.content) {\n\t\t\t\t\tyield {\n\t\t\t\t\t\tevent: \"text_chunk\",\n\t\t\t\t\t\tdata: { delta: chunk.delta.content },\n\t\t\t\t\t};\n\t\t\t\t}\n\t\t\t}\n\n\t\t\tloggers.intentStream.debug(\"assembledToolCalls\", {\n\t\t\t\tassembledToolCalls,\n\t\t\t});\n\n\t\t\tif (assembledToolCalls.length > 0) {\n\t\t\t\tconst messagePayload = this.a2aModule?.getMessagePayload(\n\t\t\t\t\tquery,\n\t\t\t\t\tthreadId,\n\t\t\t\t);\n\t\t\t\tfor (const toolCall of assembledToolCalls) {\n\t\t\t\t\tconst toolCallId = randomUUID();\n\t\t\t\t\tconst toolName = toolCall.function.name;\n\t\t\t\t\tconst selectedTool = tools.filter((tool) => tool.id === toolName)[0];\n\n\t\t\t\t\tlet toolResult = \"\";\n\t\t\t\t\tif (\n\t\t\t\t\t\tthis.mcpModule &&\n\t\t\t\t\t\tselectedTool.protocol === TOOL_PROTOCOL_TYPE.MCP\n\t\t\t\t\t) {\n\t\t\t\t\t\tconst toolArgs = JSON.parse(toolCall.function.arguments) as\n\t\t\t\t\t\t\t| { [x: string]: unknown }\n\t\t\t\t\t\t\t| undefined;\n\t\t\t\t\t\tyield {\n\t\t\t\t\t\t\tevent: \"tool_start\",\n\t\t\t\t\t\t\tdata: {\n\t\t\t\t\t\t\t\ttoolCallId,\n\t\t\t\t\t\t\t\tprotocol: TOOL_PROTOCOL_TYPE.MCP,\n\t\t\t\t\t\t\t\ttoolName,\n\t\t\t\t\t\t\t\ttoolArgs,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t};\n\t\t\t\t\t\tloggers.intent.debug(\"MCP tool call\", { toolName, toolArgs });\n\t\t\t\t\t\ttoolResult = await this.mcpModule.useTool(\n\t\t\t\t\t\t\tselectedTool as IMCPTool,\n\t\t\t\t\t\t\ttoolArgs,\n\t\t\t\t\t\t);\n\t\t\t\t\t} else if (\n\t\t\t\t\t\tthis.a2aModule &&\n\t\t\t\t\t\tselectedTool.protocol === TOOL_PROTOCOL_TYPE.A2A\n\t\t\t\t\t) {\n\t\t\t\t\t\tyield {\n\t\t\t\t\t\t\tevent: \"tool_start\",\n\t\t\t\t\t\t\tdata: {\n\t\t\t\t\t\t\t\ttoolCallId,\n\t\t\t\t\t\t\t\tprotocol: TOOL_PROTOCOL_TYPE.A2A,\n\t\t\t\t\t\t\t\ttoolName,\n\t\t\t\t\t\t\t\ttoolArgs: null,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t};\n\t\t\t\t\t\tloggers.intent.debug(\"A2A tool call\", { toolName });\n\t\t\t\t\t\ttoolResult = await this.a2aModule.useTool(\n\t\t\t\t\t\t\tselectedTool as IA2ATool,\n\t\t\t\t\t\t\t// biome-ignore lint/style/noNonNullAssertion: <a2aModule is guaranteed to be defined>\n\t\t\t\t\t\t\tmessagePayload!,\n\t\t\t\t\t\t\tthreadId,\n\t\t\t\t\t\t);\n\t\t\t\t\t} else {\n\t\t\t\t\t\t// Unrecognized tool type. It cannot be happened...\n\t\t\t\t\t\tloggers.intent.warn(\n\t\t\t\t\t\t\t`Unrecognized tool type: ${selectedTool.protocol}`,\n\t\t\t\t\t\t);\n\t\t\t\t\t\tcontinue;\n\t\t\t\t\t}\n\t\t\t\t\tyield {\n\t\t\t\t\t\tevent: \"tool_output\",\n\t\t\t\t\t\tdata: {\n\t\t\t\t\t\t\ttoolCallId,\n\t\t\t\t\t\t\tprotocol: selectedTool.protocol,\n\t\t\t\t\t\t\ttoolName,\n\t\t\t\t\t\t\tresult: toolResult,\n\t\t\t\t\t\t},\n\t\t\t\t\t};\n\t\t\t\t\tloggers.intent.debug(\"toolResult\", { toolResult });\n\n\t\t\t\t\tprocessList.push(toolResult);\n\t\t\t\t\tmodelInstance.appendMessages(messages, toolResult);\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\tbreak;\n\t\t\t}\n\t\t}\n\t}\n\n\t/**\n\t * Generates a title for the conversation based on the query.\n\t *\n\t * @param query - The user's input query\n\t * @returns Promise resolving to a generated title\n\t */\n\n\tprivate async generateTitle(query: string): Promise<string> {\n\t\tconst DEFAULT_TITLE = \"New Chat\";\n\t\ttry {\n\t\t\tconst modelInstance = this.modelModule.getModel();\n\t\t\tconst messages = modelInstance.generateMessages({\n\t\t\t\tquery,\n\t\t\t\tsystemPrompt: `You are a helpful assistant that generates titles for conversations.\n  Please analyze the user's query and create a concise title that accurately reflects the conversation's core topic.\n  The title must be no more than 5 words long.\n  Respond with only the title. Do not include any punctuation or extra explanations.`,\n\t\t\t});\n\t\t\tconst response = await modelInstance.fetch(messages);\n\t\t\treturn response.content || DEFAULT_TITLE;\n\t\t} catch (error) {\n\t\t\tloggers.intentStream.error(\"Error generating title\", {\n\t\t\t\terror,\n\t\t\t\tquery,\n\t\t\t});\n\t\t\treturn DEFAULT_TITLE;\n\t\t}\n\t}\n\n\t/**\n\t * Main entry point for processing user queries.\n\t *\n\t * Handles the complete query lifecycle:\n\t * 1. Loads 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 type - The type of thread (e.g., chat, workflow)\n\t * @param userId - The user's unique identifier\n\t * @param threadId - Unique thread identifier\n\t * @param query - The user's input query\n\t * @returns Object containing the AI-generated response\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): AsyncGenerator<StreamEvent> {\n\t\tconst { type, userId } = threadMetadata;\n\t\tconst queryStartAt = Date.now();\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} else {\n\t\t\tthreadId = randomUUID();\n\t\t\tconst title = await this.generateTitle(query);\n\n\t\t\tconst metadata =\n\t\t\t\t(await threadMemory?.createThread(type, userId, threadId, title)) ||\n\t\t\t\t({\n\t\t\t\t\ttype,\n\t\t\t\t\tthreadId,\n\t\t\t\t\ttitle,\n\t\t\t\t\tupdatedAt: Date.now(),\n\t\t\t\t} as ThreadMetadata);\n\t\t\tloggers.intentStream.info(\"Create new thread\", { metadata });\n\t\t\tyield { event: \"thread_id\", data: metadata };\n\t\t}\n\n\t\tawait threadMemory?.addMessagesToThread(userId, threadId, [\n\t\t\t{\n\t\t\t\trole: MessageRole.USER,\n\t\t\t\ttimestamp: queryStartAt,\n\t\t\t\tcontent: { type: \"text\", parts: [query] },\n\t\t\t},\n\t\t]);\n\n\t\t// 2. intent triggering\n\t\tconst intent = await this.intentTriggering(query, thread);\n\n\t\t// 3. intent fulfillment\n\t\tconst stream = this.intentFulfilling(query, threadId, thread, intent);\n\n\t\tlet finalResponseText = \"\";\n\t\tfor await (const event of stream) {\n\t\t\tif (event.event === \"text_chunk\" && event.data.delta) {\n\t\t\t\tloggers.intentStream.debug(\"text_chunk\", { event });\n\t\t\t\tfinalResponseText += event.data.delta;\n\t\t\t} else if (event.event === \"tool_start\") {\n\t\t\t\tawait threadMemory?.addMessagesToThread(userId, threadId, [\n\t\t\t\t\t{\n\t\t\t\t\t\trole: MessageRole.MODEL,\n\t\t\t\t\t\ttimestamp: Date.now(),\n\t\t\t\t\t\tcontent: {\n\t\t\t\t\t\t\ttype: \"text\",\n\t\t\t\t\t\t\tparts: [JSON.stringify(event.data.toolArgs)],\n\t\t\t\t\t\t},\n\t\t\t\t\t\tmetadata: {\n\t\t\t\t\t\t\ttoolCallId: event.data.toolCallId,\n\t\t\t\t\t\t\ttoolName: event.data.toolName,\n\t\t\t\t\t\t\tprotocol: event.data.protocol,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t]);\n\t\t\t} else if (event.event === \"tool_output\") {\n\t\t\t\tawait threadMemory?.addMessagesToThread(userId, threadId, [\n\t\t\t\t\t{\n\t\t\t\t\t\trole: MessageRole.MODEL,\n\t\t\t\t\t\ttimestamp: Date.now(),\n\t\t\t\t\t\tcontent: { type: \"text\", parts: [event.data.result] },\n\t\t\t\t\t\tmetadata: {\n\t\t\t\t\t\t\ttoolCallId: event.data.toolCallId,\n\t\t\t\t\t\t\ttoolName: event.data.toolName,\n\t\t\t\t\t\t\tprotocol: event.data.protocol,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t]);\n\t\t\t}\n\t\t\tyield event;\n\t\t}\n\n\t\tawait threadMemory?.addMessagesToThread(userId, threadId, [\n\t\t\t{\n\t\t\t\trole: MessageRole.MODEL,\n\t\t\t\ttimestamp: Date.now(),\n\t\t\t\tcontent: { type: \"text\", parts: [finalResponseText] },\n\t\t\t},\n\t\t]);\n\t}\n}\n"],"mappings":";;;;;AAAA,SAAS,kBAAkB;AA+BpB,IAAM,qBAAN,MAAyB;AAAA,EACvB;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EAER,YACC,aACA,WACA,WACA,cACA,SACC;AACD,SAAK,cAAc;AACnB,SAAK,YAAY;AACjB,SAAK,YAAY;AACjB,SAAK,eAAe;AACpB,SAAK,UAAU;AAAA,EAChB;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EASA,MAAc,iBACb,OACA,QAC8B;AAC9B,UAAM,gBAAgB,KAAK,YAAY,SAAS;AAChD,UAAM,eAAe,KAAK,cAAc,gBAAgB;AACxD,QAAI,CAAC,cAAc;AAClB,aAAO;AAAA,IACR;AAGA,UAAM,UAAU,MAAM,aAAa,YAAY;AAE/C,QAAI,QAAQ,WAAW,GAAG;AACzB,cAAQ,aAAa,KAAK,iBAAiB;AAC3C,aAAO;AAAA,IACR;AAEA,UAAM,aAAa,QACjB,IAAI,CAACA,YAAW,KAAKA,QAAO,IAAI,KAAKA,QAAO,WAAW,EAAE,EACzD,KAAK,IAAI;AAGX,UAAM,kBAAkB,CAAC,SACtB,KACA,OAAO,QAAQ,OAAO,QAAQ,EAC7B,KAAK,CAAC,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,MAAM,EAAE,YAAY,EAAE,SAAS,EAChD,IAAI,CAAC,CAAC,SAAS,IAAI,MAAM;AACzB,YAAM,OACL,KAAK,SAAS,SACX,SACA,KAAK,SAAS,UACb,cACA;AACL,YAAM,UAAU,MAAM,QAAQ,KAAK,QAAQ,KAAK,IAC7C,KAAK,QAAQ,MAAM,KAAK,GAAG,IAC3B,OAAO,KAAK,QAAQ,KAAK;AAC5B,aAAO,GAAG,IAAI,QAAQ,OAAO;AAAA,IAC9B,CAAC,EACA,KAAK,IAAI;AAEb,UAAM,eAAe;AAAA;AAAA;AAAA,EAGrB,UAAU;AAAA;AAAA;AAAA;AAKV,UAAM,cAAc;AAAA;AAAA,EAEpB,eAAe;AAAA;AAAA,uBAEM,KAAK;AAAA;AAAA;AAAA;AAK1B,UAAM,WAAW,cAAc,iBAAiB;AAAA,MAC/C,OAAO;AAAA,MACP;AAAA,IACD,CAAC;AAED,UAAM,WAAW,MAAM,cAAc,MAAM,QAAQ;AACnD,QAAI,CAAC,SAAS,SAAS;AACtB,YAAM,IAAI,MAAM,oBAAoB;AAAA,IACrC;AACA,UAAM,aAAa,SAAS,QAAQ,KAAK;AACzC,UAAM,SAAS,MAAM,aAAa,gBAAgB,UAAU;AAC5D,QAAI,CAAC,QAAQ;AACZ,YAAM,IAAI,MAAM,oBAAoB,UAAU,EAAE;AAAA,IACjD;AACA,WAAO;AAAA,EACR;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAgBA,OAAc,iBACb,OACA,UACA,QACA,QAC8B;AAC9B,UAAM,eAAe;AAAA,YACZ,oBAAI,KAAK,GAAE,mBAAmB,CAAC;AAAA;AAAA,EAExC,KAAK,SAAS,SAAS,EAAE;AAAA;AAAA,EAEzB,KAAK,SAAS,UAAU,EAAE;AAAA;AAAA,EAE1B,QAAQ,UAAU,EAAE;AAAA,GACnB,KAAK;AAEN,UAAM,gBAAgB,KAAK,YAAY,SAAS;AAChD,UAAM,WAAW,cAAc,iBAAiB;AAAA,MAC/C;AAAA,MACA;AAAA,MACA,cAAc,aAAa,KAAK;AAAA,IACjC,CAAC;AAED,UAAM,QAAsB,CAAC;AAC7B,SAAK,aAAa,MAAM,KAAK,GAAG,KAAK,UAAU,SAAS,CAAC;AACzD,SAAK,aAAa,MAAM,KAAK,GAAI,MAAM,KAAK,UAAU,SAAS,CAAE;AAEjE,UAAM,YAAY,cAAc,wBAAwB,KAAK;AAE7D,UAAM,cAAwB,CAAC;AAE/B,WAAO,MAAM;AACZ,YAAM,iBAAiB,MAAM,cAAc;AAAA,QAC1C;AAAA,QACA;AAAA,MACD;AAEA,YAAM,qBAIA,CAAC;AAEP,cAAQ,aAAa,MAAM,YAAY,EAAE,SAAS,CAAC;AAEnD,uBAAiB,SAAS,gBAAgB;AACzC,cAAM,QAAQ,MAAM;AACpB,YAAI,OAAO,YAAY;AACtB,qBAAW,EAAE,OAAO,IAAI,UAAU,KAAK,KAAK,MAAM,YAAY;AAC7D,+BAAmB,KAAK,MAAM;AAAA,cAC7B,IAAI;AAAA,cACJ,MAAM;AAAA,cACN,UAAU,EAAE,MAAM,IAAI,WAAW,GAAG;AAAA,YACrC;AAEA,gBAAI,GAAI,oBAAmB,KAAK,EAAE,KAAK;AACvC,gBAAI,MAAM,KAAM,oBAAmB,KAAK,EAAE,SAAS,OAAO,KAAK;AAC/D,gBAAI,MAAM;AACT,iCAAmB,KAAK,EAAE,SAAS,aAAa,KAAK;AAAA,UACvD;AAAA,QACD,WAAW,MAAM,OAAO,SAAS;AAChC,gBAAM;AAAA,YACL,OAAO;AAAA,YACP,MAAM,EAAE,OAAO,MAAM,MAAM,QAAQ;AAAA,UACpC;AAAA,QACD;AAAA,MACD;AAEA,cAAQ,aAAa,MAAM,sBAAsB;AAAA,QAChD;AAAA,MACD,CAAC;AAED,UAAI,mBAAmB,SAAS,GAAG;AAClC,cAAM,iBAAiB,KAAK,WAAW;AAAA,UACtC;AAAA,UACA;AAAA,QACD;AACA,mBAAW,YAAY,oBAAoB;AAC1C,gBAAM,aAAa,WAAW;AAC9B,gBAAM,WAAW,SAAS,SAAS;AACnC,gBAAM,eAAe,MAAM,OAAO,CAAC,SAAS,KAAK,OAAO,QAAQ,EAAE,CAAC;AAEnE,cAAI,aAAa;AACjB,cACC,KAAK,aACL,aAAa,8BACZ;AACD,kBAAM,WAAW,KAAK,MAAM,SAAS,SAAS,SAAS;AAGvD,kBAAM;AAAA,cACL,OAAO;AAAA,cACP,MAAM;AAAA,gBACL;AAAA,gBACA;AAAA,gBACA;AAAA,gBACA;AAAA,cACD;AAAA,YACD;AACA,oBAAQ,OAAO,MAAM,iBAAiB,EAAE,UAAU,SAAS,CAAC;AAC5D,yBAAa,MAAM,KAAK,UAAU;AAAA,cACjC;AAAA,cACA;AAAA,YACD;AAAA,UACD,WACC,KAAK,aACL,aAAa,8BACZ;AACD,kBAAM;AAAA,cACL,OAAO;AAAA,cACP,MAAM;AAAA,gBACL;AAAA,gBACA;AAAA,gBACA;AAAA,gBACA,UAAU;AAAA,cACX;AAAA,YACD;AACA,oBAAQ,OAAO,MAAM,iBAAiB,EAAE,SAAS,CAAC;AAClD,yBAAa,MAAM,KAAK,UAAU;AAAA,cACjC;AAAA;AAAA,cAEA;AAAA,cACA;AAAA,YACD;AAAA,UACD,OAAO;AAEN,oBAAQ,OAAO;AAAA,cACd,2BAA2B,aAAa,QAAQ;AAAA,YACjD;AACA;AAAA,UACD;AACA,gBAAM;AAAA,YACL,OAAO;AAAA,YACP,MAAM;AAAA,cACL;AAAA,cACA,UAAU,aAAa;AAAA,cACvB;AAAA,cACA,QAAQ;AAAA,YACT;AAAA,UACD;AACA,kBAAQ,OAAO,MAAM,cAAc,EAAE,WAAW,CAAC;AAEjD,sBAAY,KAAK,UAAU;AAC3B,wBAAc,eAAe,UAAU,UAAU;AAAA,QAClD;AAAA,MACD,OAAO;AACN;AAAA,MACD;AAAA,IACD;AAAA,EACD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EASA,MAAc,cAAc,OAAgC;AAC3D,UAAM,gBAAgB;AACtB,QAAI;AACH,YAAM,gBAAgB,KAAK,YAAY,SAAS;AAChD,YAAM,WAAW,cAAc,iBAAiB;AAAA,QAC/C;AAAA,QACA,cAAc;AAAA;AAAA;AAAA;AAAA,MAIf,CAAC;AACD,YAAM,WAAW,MAAM,cAAc,MAAM,QAAQ;AACnD,aAAO,SAAS,WAAW;AAAA,IAC5B,SAAS,OAAO;AACf,cAAQ,aAAa,MAAM,0BAA0B;AAAA,QACpD;AAAA,QACA;AAAA,MACD,CAAC;AACD,aAAO;AAAA,IACR;AAAA,EACD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAiBA,OAAc,kBACb,gBAKA,OAC8B;AAC9B,UAAM,EAAE,MAAM,OAAO,IAAI;AACzB,UAAM,eAAe,KAAK,IAAI;AAC9B,UAAM,eAAe,KAAK,cAAc,gBAAgB;AAGxD,QAAI,WAAW,eAAe;AAC9B,QAAI;AACJ,QAAI,UAAU;AACb,eAAS,MAAM,cAAc,UAAU,QAAQ,QAAQ;AAAA,IACxD,OAAO;AACN,iBAAW,WAAW;AACtB,YAAM,QAAQ,MAAM,KAAK,cAAc,KAAK;AAE5C,YAAM,WACJ,MAAM,cAAc,aAAa,MAAM,QAAQ,UAAU,KAAK,KAC9D;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA,WAAW,KAAK,IAAI;AAAA,MACrB;AACD,cAAQ,aAAa,KAAK,qBAAqB,EAAE,SAAS,CAAC;AAC3D,YAAM,EAAE,OAAO,aAAa,MAAM,SAAS;AAAA,IAC5C;AAEA,UAAM,cAAc,oBAAoB,QAAQ,UAAU;AAAA,MACzD;AAAA,QACC;AAAA,QACA,WAAW;AAAA,QACX,SAAS,EAAE,MAAM,QAAQ,OAAO,CAAC,KAAK,EAAE;AAAA,MACzC;AAAA,IACD,CAAC;AAGD,UAAM,SAAS,MAAM,KAAK,iBAAiB,OAAO,MAAM;AAGxD,UAAM,SAAS,KAAK,iBAAiB,OAAO,UAAU,QAAQ,MAAM;AAEpE,QAAI,oBAAoB;AACxB,qBAAiB,SAAS,QAAQ;AACjC,UAAI,MAAM,UAAU,gBAAgB,MAAM,KAAK,OAAO;AACrD,gBAAQ,aAAa,MAAM,cAAc,EAAE,MAAM,CAAC;AAClD,6BAAqB,MAAM,KAAK;AAAA,MACjC,WAAW,MAAM,UAAU,cAAc;AACxC,cAAM,cAAc,oBAAoB,QAAQ,UAAU;AAAA,UACzD;AAAA,YACC;AAAA,YACA,WAAW,KAAK,IAAI;AAAA,YACpB,SAAS;AAAA,cACR,MAAM;AAAA,cACN,OAAO,CAAC,KAAK,UAAU,MAAM,KAAK,QAAQ,CAAC;AAAA,YAC5C;AAAA,YACA,UAAU;AAAA,cACT,YAAY,MAAM,KAAK;AAAA,cACvB,UAAU,MAAM,KAAK;AAAA,cACrB,UAAU,MAAM,KAAK;AAAA,YACtB;AAAA,UACD;AAAA,QACD,CAAC;AAAA,MACF,WAAW,MAAM,UAAU,eAAe;AACzC,cAAM,cAAc,oBAAoB,QAAQ,UAAU;AAAA,UACzD;AAAA,YACC;AAAA,YACA,WAAW,KAAK,IAAI;AAAA,YACpB,SAAS,EAAE,MAAM,QAAQ,OAAO,CAAC,MAAM,KAAK,MAAM,EAAE;AAAA,YACpD,UAAU;AAAA,cACT,YAAY,MAAM,KAAK;AAAA,cACvB,UAAU,MAAM,KAAK;AAAA,cACrB,UAAU,MAAM,KAAK;AAAA,YACtB;AAAA,UACD;AAAA,QACD,CAAC;AAAA,MACF;AACA,YAAM;AAAA,IACP;AAEA,UAAM,cAAc,oBAAoB,QAAQ,UAAU;AAAA,MACzD;AAAA,QACC;AAAA,QACA,WAAW,KAAK,IAAI;AAAA,QACpB,SAAS,EAAE,MAAM,QAAQ,OAAO,CAAC,iBAAiB,EAAE;AAAA,MACrD;AAAA,IACD,CAAC;AAAA,EACF;AACD;","names":["intent"]}

package/dist/esm/chunk-RNA32QWV.js

@@ -0,0 +1,128 @@
+import {
+  MCPTool
+} from "./chunk-3LLHEUY2.js";
+import {
+  loggers
+} from "./chunk-N7VYM2R5.js";
+
+// src/modules/mcp/mcp.module.ts
+import { Client } from "@modelcontextprotocol/sdk/client/index.js";
+import { SSEClientTransport } from "@modelcontextprotocol/sdk/client/sse.js";
+import { StdioClientTransport } from "@modelcontextprotocol/sdk/client/stdio.js";
+import { StreamableHTTPClientTransport } from "@modelcontextprotocol/sdk/client/streamableHttp.js";
+import { WebSocketClientTransport } from "@modelcontextprotocol/sdk/client/websocket.js";
+var MCPModule = class {
+  mcpConfigs = /* @__PURE__ */ new Map();
+  /** Map of MCP server names to their client instances */
+  mcpClientMap = /* @__PURE__ */ new Map();
+  /** Array of all discovered tools from connected MCP servers */
+  tools = [];
+  addMCPServerConfig(configs) {
+    for (const [name, config] of Object.entries(configs)) {
+      this.mcpConfigs.set(name, config);
+    }
+  }
+  async connectToServers() {
+    for (const [name, config] of this.mcpConfigs.entries()) {
+      try {
+        const mcpClient = new Client({ name, version: "1.0.0" });
+        switch (config.type) {
+          case "stdio": {
+            const transport = new StdioClientTransport(config.params);
+            await mcpClient.connect(transport);
+            break;
+          }
+          case "websocket": {
+            const transport = new WebSocketClientTransport(config.url);
+            await mcpClient.connect(transport);
+            break;
+          }
+          case "sse": {
+            const transport = new SSEClientTransport(
+              config.url,
+              config.options
+            );
+            await mcpClient.connect(transport);
+            break;
+          }
+          case "streamableHttp": {
+            const transport = new StreamableHTTPClientTransport(
+              config.url,
+              config.options
+            );
+            await mcpClient.connect(transport);
+            break;
+          }
+          default:
+            loggers.mcp.error("Unsupported MCP config type");
+            break;
+        }
+        this.mcpClientMap.set(name, mcpClient);
+        const toolList = await mcpClient.listTools();
+        const newToolList = toolList.tools.map((tool) => {
+          return new MCPTool(name, tool);
+        });
+        this.tools.push(...newToolList);
+        loggers.mcp.info("Connected to MCP server with tools:", {
+          tools: newToolList.map((tool) => tool.id)
+        });
+      } catch (error) {
+        loggers.mcp.error(`Failed to connect to MCP server ${name}`, { error });
+      }
+    }
+  }
+  /**
+   * Returns all available tools from connected MCP servers.
+   *
+   * @returns Array of MCPTool instances representing available tools
+   */
+  getTools() {
+    return this.tools;
+  }
+  /**
+   * Executes a tool on its corresponding MCP server.
+   *
+   * @param tool - The MCPTool instance to execute
+   * @param _args - Arguments to pass to the tool
+   * @returns Promise resolving to the tool's execution result
+   * @throws Error if the MCP server for the tool is not found
+   */
+  async useTool(tool, _args) {
+    const { serverName, mcpTool } = tool;
+    const toolName = mcpTool.name;
+    const mcp = this.mcpClientMap.get(serverName);
+    try {
+      if (!mcp) {
+        throw new Error(`Invalid MCP Tool ${serverName}-${mcpTool.name}`);
+      }
+      const result = await mcp.callTool({
+        name: toolName,
+        arguments: _args
+      });
+      const toolResult = `[Bot Called Tool ${toolName} with args ${JSON.stringify(_args)}]
+` + JSON.stringify(result.content, null, 2);
+      return toolResult;
+    } catch (error) {
+      loggers.mcp.error("Failed to call tool", { error });
+      const toolResult = `[Bot Called Tool ${toolName} with args ${JSON.stringify(_args)}]
+${typeof error === "string" ? error : JSON.stringify(error, null, 2)}`;
+      return toolResult;
+    }
+  }
+  /**
+   * Closes all MCP client connections.
+   *
+   * Should be called when shutting down the application to ensure
+   * all MCP connections are properly closed.
+   */
+  async cleanup() {
+    this.mcpClientMap.forEach((mcpClient) => {
+      mcpClient.close();
+    });
+  }
+};
+
+export {
+  MCPModule
+};
+//# sourceMappingURL=chunk-RNA32QWV.js.map

package/dist/esm/chunk-RNA32QWV.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/modules/mcp/mcp.module.ts"],"sourcesContent":["import { Client } from \"@modelcontextprotocol/sdk/client/index.js\";\nimport { SSEClientTransport } from \"@modelcontextprotocol/sdk/client/sse.js\";\nimport { StdioClientTransport } from \"@modelcontextprotocol/sdk/client/stdio.js\";\nimport { StreamableHTTPClientTransport } from \"@modelcontextprotocol/sdk/client/streamableHttp.js\";\nimport { WebSocketClientTransport } from \"@modelcontextprotocol/sdk/client/websocket.js\";\nimport type { MCPConfig } from \"@/types/mcp.js\";\nimport { loggers } from \"@/utils/logger.js\";\nimport { MCPTool } from \"./mcp.tool.js\";\n\n/**\n * Module for managing Model Context Protocol (MCP) server connections.\n *\n * This module handles the lifecycle of MCP client connections, discovers\n * available tools from connected servers, and provides an interface for\n * executing those tools. Multiple MCP servers can be connected simultaneously.\n *\n * @example\n * ```typescript\n * const mcpModule = new MCPModule();\n * await mcpModule.addMCPConfig({\n *   \"filesystem\": {\n *     command: \"npx\",\n *     args: [\"@modelcontextprotocol/server-filesystem\", \"/path/to/files\"]\n *   }\n * });\n *\n * const tools = mcpModule.getTools();\n * const result = await mcpModule.useTool(tools[0], { path: \"/example.txt\" });\n * ```\n */\nexport class MCPModule {\n\tprivate mcpConfigs: Map<string, MCPConfig> = new Map();\n\t/** Map of MCP server names to their client instances */\n\tprivate mcpClientMap: Map<string, Client> = new Map();\n\t/** Array of all discovered tools from connected MCP servers */\n\tprivate tools: MCPTool[] = [];\n\n\taddMCPServerConfig(configs: { [name: string]: MCPConfig }): void {\n\t\tfor (const [name, config] of Object.entries(configs)) {\n\t\t\tthis.mcpConfigs.set(name, config);\n\t\t}\n\t}\n\n\tasync connectToServers(): Promise<void> {\n\t\tfor (const [name, config] of this.mcpConfigs.entries()) {\n\t\t\ttry {\n\t\t\t\tconst mcpClient = new Client({ name, version: \"1.0.0\" });\n\t\t\t\tswitch (config.type) {\n\t\t\t\t\tcase \"stdio\": {\n\t\t\t\t\t\tconst transport = new StdioClientTransport(config.params);\n\t\t\t\t\t\tawait mcpClient.connect(transport);\n\t\t\t\t\t\tbreak;\n\t\t\t\t\t}\n\t\t\t\t\tcase \"websocket\": {\n\t\t\t\t\t\tconst transport = new WebSocketClientTransport(config.url);\n\t\t\t\t\t\tawait mcpClient.connect(transport);\n\t\t\t\t\t\tbreak;\n\t\t\t\t\t}\n\t\t\t\t\tcase \"sse\": {\n\t\t\t\t\t\tconst transport = new SSEClientTransport(\n\t\t\t\t\t\t\tconfig.url,\n\t\t\t\t\t\t\tconfig.options,\n\t\t\t\t\t\t);\n\t\t\t\t\t\tawait mcpClient.connect(transport);\n\t\t\t\t\t\tbreak;\n\t\t\t\t\t}\n\t\t\t\t\tcase \"streamableHttp\": {\n\t\t\t\t\t\tconst transport = new StreamableHTTPClientTransport(\n\t\t\t\t\t\t\tconfig.url,\n\t\t\t\t\t\t\tconfig.options,\n\t\t\t\t\t\t);\n\t\t\t\t\t\tawait mcpClient.connect(transport);\n\t\t\t\t\t\tbreak;\n\t\t\t\t\t}\n\t\t\t\t\tdefault:\n\t\t\t\t\t\t// This cannot happen.\n\t\t\t\t\t\tloggers.mcp.error(\"Unsupported MCP config type\");\n\t\t\t\t\t\tbreak;\n\t\t\t\t}\n\n\t\t\t\tthis.mcpClientMap.set(name, mcpClient);\n\t\t\t\tconst toolList = await mcpClient.listTools();\n\t\t\t\tconst newToolList = toolList.tools.map((tool) => {\n\t\t\t\t\treturn new MCPTool(name, tool);\n\t\t\t\t});\n\t\t\t\tthis.tools.push(...newToolList);\n\t\t\t\tloggers.mcp.info(\"Connected to MCP server with tools:\", {\n\t\t\t\t\ttools: newToolList.map((tool) => tool.id),\n\t\t\t\t});\n\t\t\t} catch (error) {\n\t\t\t\tloggers.mcp.error(`Failed to connect to MCP server ${name}`, { error });\n\t\t\t}\n\t\t}\n\t}\n\n\t/**\n\t * Returns all available tools from connected MCP servers.\n\t *\n\t * @returns Array of MCPTool instances representing available tools\n\t */\n\tgetTools() {\n\t\treturn this.tools;\n\t}\n\n\t/**\n\t * Executes a tool on its corresponding MCP server.\n\t *\n\t * @param tool - The MCPTool instance to execute\n\t * @param _args - Arguments to pass to the tool\n\t * @returns Promise resolving to the tool's execution result\n\t * @throws Error if the MCP server for the tool is not found\n\t */\n\tasync useTool(tool: MCPTool, _args?: any): Promise<string> {\n\t\tconst { serverName, mcpTool } = tool;\n\t\tconst toolName = mcpTool.name;\n\t\tconst mcp = this.mcpClientMap.get(serverName);\n\n\t\ttry {\n\t\t\tif (!mcp) {\n\t\t\t\tthrow new Error(`Invalid MCP Tool ${serverName}-${mcpTool.name}`);\n\t\t\t}\n\n\t\t\tconst result = await mcp.callTool({\n\t\t\t\tname: toolName,\n\t\t\t\targuments: _args,\n\t\t\t});\n\t\t\tconst toolResult =\n\t\t\t\t`[Bot Called Tool ${toolName} with args ${JSON.stringify(_args)}]\\n` +\n\t\t\t\tJSON.stringify(result.content, null, 2);\n\t\t\treturn toolResult;\n\t\t} catch (error) {\n\t\t\tloggers.mcp.error(\"Failed to call tool\", { error });\n\t\t\tconst toolResult = `[Bot Called Tool ${toolName} with args ${JSON.stringify(_args)}]\\n${typeof error === \"string\" ? error : JSON.stringify(error, null, 2)}`;\n\t\t\treturn toolResult;\n\t\t}\n\t}\n\n\t/**\n\t * Closes all MCP client connections.\n\t *\n\t * Should be called when shutting down the application to ensure\n\t * all MCP connections are properly closed.\n\t */\n\tasync cleanup() {\n\t\tthis.mcpClientMap.forEach((mcpClient: Client) => {\n\t\t\tmcpClient.close();\n\t\t});\n\t}\n}\n"],"mappings":";;;;;;;;AAAA,SAAS,cAAc;AACvB,SAAS,0BAA0B;AACnC,SAAS,4BAA4B;AACrC,SAAS,qCAAqC;AAC9C,SAAS,gCAAgC;AA0BlC,IAAM,YAAN,MAAgB;AAAA,EACd,aAAqC,oBAAI,IAAI;AAAA;AAAA,EAE7C,eAAoC,oBAAI,IAAI;AAAA;AAAA,EAE5C,QAAmB,CAAC;AAAA,EAE5B,mBAAmB,SAA8C;AAChE,eAAW,CAAC,MAAM,MAAM,KAAK,OAAO,QAAQ,OAAO,GAAG;AACrD,WAAK,WAAW,IAAI,MAAM,MAAM;AAAA,IACjC;AAAA,EACD;AAAA,EAEA,MAAM,mBAAkC;AACvC,eAAW,CAAC,MAAM,MAAM,KAAK,KAAK,WAAW,QAAQ,GAAG;AACvD,UAAI;AACH,cAAM,YAAY,IAAI,OAAO,EAAE,MAAM,SAAS,QAAQ,CAAC;AACvD,gBAAQ,OAAO,MAAM;AAAA,UACpB,KAAK,SAAS;AACb,kBAAM,YAAY,IAAI,qBAAqB,OAAO,MAAM;AACxD,kBAAM,UAAU,QAAQ,SAAS;AACjC;AAAA,UACD;AAAA,UACA,KAAK,aAAa;AACjB,kBAAM,YAAY,IAAI,yBAAyB,OAAO,GAAG;AACzD,kBAAM,UAAU,QAAQ,SAAS;AACjC;AAAA,UACD;AAAA,UACA,KAAK,OAAO;AACX,kBAAM,YAAY,IAAI;AAAA,cACrB,OAAO;AAAA,cACP,OAAO;AAAA,YACR;AACA,kBAAM,UAAU,QAAQ,SAAS;AACjC;AAAA,UACD;AAAA,UACA,KAAK,kBAAkB;AACtB,kBAAM,YAAY,IAAI;AAAA,cACrB,OAAO;AAAA,cACP,OAAO;AAAA,YACR;AACA,kBAAM,UAAU,QAAQ,SAAS;AACjC;AAAA,UACD;AAAA,UACA;AAEC,oBAAQ,IAAI,MAAM,6BAA6B;AAC/C;AAAA,QACF;AAEA,aAAK,aAAa,IAAI,MAAM,SAAS;AACrC,cAAM,WAAW,MAAM,UAAU,UAAU;AAC3C,cAAM,cAAc,SAAS,MAAM,IAAI,CAAC,SAAS;AAChD,iBAAO,IAAI,QAAQ,MAAM,IAAI;AAAA,QAC9B,CAAC;AACD,aAAK,MAAM,KAAK,GAAG,WAAW;AAC9B,gBAAQ,IAAI,KAAK,uCAAuC;AAAA,UACvD,OAAO,YAAY,IAAI,CAAC,SAAS,KAAK,EAAE;AAAA,QACzC,CAAC;AAAA,MACF,SAAS,OAAO;AACf,gBAAQ,IAAI,MAAM,mCAAmC,IAAI,IAAI,EAAE,MAAM,CAAC;AAAA,MACvE;AAAA,IACD;AAAA,EACD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,WAAW;AACV,WAAO,KAAK;AAAA,EACb;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAUA,MAAM,QAAQ,MAAe,OAA8B;AAC1D,UAAM,EAAE,YAAY,QAAQ,IAAI;AAChC,UAAM,WAAW,QAAQ;AACzB,UAAM,MAAM,KAAK,aAAa,IAAI,UAAU;AAE5C,QAAI;AACH,UAAI,CAAC,KAAK;AACT,cAAM,IAAI,MAAM,oBAAoB,UAAU,IAAI,QAAQ,IAAI,EAAE;AAAA,MACjE;AAEA,YAAM,SAAS,MAAM,IAAI,SAAS;AAAA,QACjC,MAAM;AAAA,QACN,WAAW;AAAA,MACZ,CAAC;AACD,YAAM,aACL,oBAAoB,QAAQ,cAAc,KAAK,UAAU,KAAK,CAAC;AAAA,IAC/D,KAAK,UAAU,OAAO,SAAS,MAAM,CAAC;AACvC,aAAO;AAAA,IACR,SAAS,OAAO;AACf,cAAQ,IAAI,MAAM,uBAAuB,EAAE,MAAM,CAAC;AAClD,YAAM,aAAa,oBAAoB,QAAQ,cAAc,KAAK,UAAU,KAAK,CAAC;AAAA,EAAM,OAAO,UAAU,WAAW,QAAQ,KAAK,UAAU,OAAO,MAAM,CAAC,CAAC;AAC1J,aAAO;AAAA,IACR;AAAA,EACD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQA,MAAM,UAAU;AACf,SAAK,aAAa,QAAQ,CAAC,cAAsB;AAChD,gBAAU,MAAM;AAAA,IACjB,CAAC;AAAA,EACF;AACD;","names":[]}

package/dist/esm/chunk-SYCFBEUA.js

@@ -0,0 +1,40 @@
+import {
+  ThreadApiController
+} from "./chunk-WLT2B2SZ.js";
+import {
+  AinHttpError
+} from "./chunk-FL3JRQM4.js";
+
+// src/routes/api/threads.routes.ts
+import {
+  Router
+} from "express";
+import { StatusCodes } from "http-status-codes";
+var createThreadApiRouter = (memoryModule) => {
+  const router = Router();
+  const threadApiController = new ThreadApiController(memoryModule);
+  const checkThreadMemory = (req, res, next) => {
+    const threadMemory = memoryModule.getThreadMemory();
+    if (!threadMemory) {
+      const error = new AinHttpError(
+        StatusCodes.SERVICE_UNAVAILABLE,
+        "Memory module is not initialized"
+      );
+      throw error;
+    }
+    next();
+  };
+  router.get("/", checkThreadMemory, threadApiController.handleGetUserThreads);
+  router.get("/:id", checkThreadMemory, threadApiController.handleGetThread);
+  router.delete(
+    "/:id",
+    checkThreadMemory,
+    threadApiController.handleDeleteThread
+  );
+  return router;
+};
+
+export {
+  createThreadApiRouter
+};
+//# sourceMappingURL=chunk-SYCFBEUA.js.map

package/dist/esm/chunk-SYCFBEUA.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/routes/api/threads.routes.ts"],"sourcesContent":["import {\n\ttype NextFunction,\n\ttype Request,\n\ttype Response,\n\tRouter,\n} from \"express\";\nimport { StatusCodes } from \"http-status-codes\";\nimport { ThreadApiController } from \"@/controllers/api/threads.api.controller.js\";\nimport type { MemoryModule } from \"@/modules/index.js\";\nimport { AinHttpError } from \"@/types/agent\";\n\nexport const createThreadApiRouter = (memoryModule: MemoryModule): Router => {\n\tconst router = Router();\n\tconst threadApiController = new ThreadApiController(memoryModule);\n\n\tconst checkThreadMemory = (\n\t\treq: Request,\n\t\tres: Response,\n\t\tnext: NextFunction,\n\t) => {\n\t\tconst threadMemory = memoryModule.getThreadMemory();\n\t\tif (!threadMemory) {\n\t\t\tconst error = new AinHttpError(\n\t\t\t\tStatusCodes.SERVICE_UNAVAILABLE,\n\t\t\t\t\"Memory module is not initialized\",\n\t\t\t);\n\t\t\tthrow error;\n\t\t}\n\t\tnext();\n\t};\n\n\t// APIs (prefix: /api/threads)\n\trouter.get(\"/\", checkThreadMemory, threadApiController.handleGetUserThreads);\n\trouter.get(\"/:id\", checkThreadMemory, threadApiController.handleGetThread);\n\trouter.delete(\n\t\t\"/:id\",\n\t\tcheckThreadMemory,\n\t\tthreadApiController.handleDeleteThread,\n\t);\n\n\treturn router;\n};\n"],"mappings":";;;;;;;;AAAA;AAAA,EAIC;AAAA,OACM;AACP,SAAS,mBAAmB;AAKrB,IAAM,wBAAwB,CAAC,iBAAuC;AAC5E,QAAM,SAAS,OAAO;AACtB,QAAM,sBAAsB,IAAI,oBAAoB,YAAY;AAEhE,QAAM,oBAAoB,CACzB,KACA,KACA,SACI;AACJ,UAAM,eAAe,aAAa,gBAAgB;AAClD,QAAI,CAAC,cAAc;AAClB,YAAM,QAAQ,IAAI;AAAA,QACjB,YAAY;AAAA,QACZ;AAAA,MACD;AACA,YAAM;AAAA,IACP;AACA,SAAK;AAAA,EACN;AAGA,SAAO,IAAI,KAAK,mBAAmB,oBAAoB,oBAAoB;AAC3E,SAAO,IAAI,QAAQ,mBAAmB,oBAAoB,eAAe;AACzE,SAAO;AAAA,IACN;AAAA,IACA;AAAA,IACA,oBAAoB;AAAA,EACrB;AAEA,SAAO;AACR;","names":[]}

package/dist/esm/{chunk-HOFAK5J2.js → chunk-VDXWZ4NK.js}

@@ -3,15 +3,15 @@ import {
 } from "./chunk-AMRKK46V.js";
 import {
   createThreadApiRouter
-} from "./chunk-VA3XOZBV.js";
+} from "./chunk-SYCFBEUA.js";
 
 // src/routes/api.routes.ts
 import { Router } from "express";
 var createApiRouter = (agent) => {
   const router = Router();
-  router.use("/api/model", createModelApiRouter(agent.modelModule));
+  router.use("/model", createModelApiRouter(agent.modelModule));
   if (agent.memoryModule) {
-    router.use("/api/thread", createThreadApiRouter(agent.memoryModule));
+    router.use("/threads", createThreadApiRouter(agent.memoryModule));
   }
   return router;
 };
@@ -19,4 +19,4 @@ var createApiRouter = (agent) => {
 export {
   createApiRouter
 };
-//# sourceMappingURL=chunk-HOFAK5J2.js.map
+//# sourceMappingURL=chunk-VDXWZ4NK.js.map