@ainetwork/adk 0.1.12 → 0.2.0

package/dist/esm/chunk-7D4SKPIS.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/controllers/query.controller.ts"],"sourcesContent":["import type { NextFunction, Request, Response } from \"express\";\nimport { StatusCodes } from \"http-status-codes\";\nimport type { QueryService, QueryStreamService } from \"@/services\";\nimport { AinHttpError } from \"@/types/agent\";\n\nexport class QueryController {\n\tprivate queryService;\n\tprivate queryStreamService;\n\n\tconstructor(\n\t\tqueryService: QueryService,\n\t\tqueryStreamService?: QueryStreamService,\n\t) {\n\t\tthis.queryService = queryService;\n\t\tthis.queryStreamService = queryStreamService;\n\t}\n\n\tpublic handleQueryRequest = async (\n\t\treq: Request,\n\t\tres: Response,\n\t\tnext: NextFunction,\n\t) => {\n\t\tconst { type, message, threadId } = req.body;\n\t\tconst userId = res.locals.userId;\n\n\t\ttry {\n\t\t\tconst result = await this.queryService.handleQuery(\n\t\t\t\t{ type, userId, threadId },\n\t\t\t\tmessage,\n\t\t\t);\n\n\t\t\tres.status(200).json(result);\n\t\t} catch (error) {\n\t\t\tnext(error);\n\t\t}\n\t};\n\n\tpublic handleQueryStreamRequest = async (\n\t\treq: Request,\n\t\tres: Response,\n\t\tnext: NextFunction,\n\t) => {\n\t\tconst { type, threadId, message } = req.body;\n\t\tconst userId = res.locals.userId;\n\n\t\tif (!this.queryStreamService) {\n\t\t\tconst error = new AinHttpError(\n\t\t\t\tStatusCodes.NOT_IMPLEMENTED,\n\t\t\t\t\"Stream query not supported\",\n\t\t\t);\n\t\t\treturn next(error);\n\t\t}\n\n\t\tconst stream = this.queryStreamService.handleQueryStream(\n\t\t\t{ type, userId, threadId },\n\t\t\tmessage,\n\t\t);\n\n\t\ttry {\n\t\t\tfor await (const event of stream) {\n\t\t\t\tres.write(\n\t\t\t\t\t`event: ${event.event}\\ndata: ${JSON.stringify(event.data)}\\n\\n`,\n\t\t\t\t);\n\t\t\t}\n\t\t} catch (error: unknown) {\n\t\t\tconst errMsg =\n\t\t\t\t(error as Error)?.message || \"Failed to handle query stream\";\n\t\t\tres.write(`event: error\\ndata: ${errMsg}\\n\\n`);\n\t\t} finally {\n\t\t\tres.end();\n\t\t}\n\t};\n}\n"],"mappings":";;;;;AACA,SAAS,mBAAmB;AAIrB,IAAM,kBAAN,MAAsB;AAAA,EACpB;AAAA,EACA;AAAA,EAER,YACC,cACA,oBACC;AACD,SAAK,eAAe;AACpB,SAAK,qBAAqB;AAAA,EAC3B;AAAA,EAEO,qBAAqB,OAC3B,KACA,KACA,SACI;AACJ,UAAM,EAAE,MAAM,SAAS,SAAS,IAAI,IAAI;AACxC,UAAM,SAAS,IAAI,OAAO;AAE1B,QAAI;AACH,YAAM,SAAS,MAAM,KAAK,aAAa;AAAA,QACtC,EAAE,MAAM,QAAQ,SAAS;AAAA,QACzB;AAAA,MACD;AAEA,UAAI,OAAO,GAAG,EAAE,KAAK,MAAM;AAAA,IAC5B,SAAS,OAAO;AACf,WAAK,KAAK;AAAA,IACX;AAAA,EACD;AAAA,EAEO,2BAA2B,OACjC,KACA,KACA,SACI;AACJ,UAAM,EAAE,MAAM,UAAU,QAAQ,IAAI,IAAI;AACxC,UAAM,SAAS,IAAI,OAAO;AAE1B,QAAI,CAAC,KAAK,oBAAoB;AAC7B,YAAM,QAAQ,IAAI;AAAA,QACjB,YAAY;AAAA,QACZ;AAAA,MACD;AACA,aAAO,KAAK,KAAK;AAAA,IAClB;AAEA,UAAM,SAAS,KAAK,mBAAmB;AAAA,MACtC,EAAE,MAAM,QAAQ,SAAS;AAAA,MACzB;AAAA,IACD;AAEA,QAAI;AACH,uBAAiB,SAAS,QAAQ;AACjC,YAAI;AAAA,UACH,UAAU,MAAM,KAAK;AAAA,QAAW,KAAK,UAAU,MAAM,IAAI,CAAC;AAAA;AAAA;AAAA,QAC3D;AAAA,MACD;AAAA,IACD,SAAS,OAAgB;AACxB,YAAM,SACJ,OAAiB,WAAW;AAC9B,UAAI,MAAM;AAAA,QAAuB,MAAM;AAAA;AAAA,CAAM;AAAA,IAC9C,UAAE;AACD,UAAI,IAAI;AAAA,IACT;AAAA,EACD;AACD;","names":[]}

package/dist/esm/chunk-A65XRRHY.js

@@ -0,0 +1,251 @@
+import {
+  loggers
+} from "./chunk-N7VYM2R5.js";
+
+// src/services/query.service.ts
+import { randomUUID } from "crypto";
+var QueryService = class {
+  modelModule;
+  a2aModule;
+  mcpModule;
+  memoryModule;
+  prompts;
+  constructor(modelModule, a2aModule, mcpModule, memoryModule, prompts) {
+    this.modelModule = modelModule;
+    this.a2aModule = a2aModule;
+    this.mcpModule = mcpModule;
+    this.memoryModule = memoryModule;
+    this.prompts = prompts;
+  }
+  /**
+   * Detects the intent from context.
+   *
+   * @param query - The user's input query
+   * @param thread - The thread history
+   * @returns The detected intent
+   */
+  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.
+   *
+   * Manages the complete inference loop including:
+   * - Loading prompts and conversation history
+   * - Collecting available tools from modules
+   * - Executing model inference with tool support
+   * - Processing tool calls iteratively until completion
+   *
+   * @param query - The user's input query
+   * @param threadId - Thread identifier for context
+   * @param thread - Previous conversation history
+   * @returns Object containing process steps and final response
+   */
+  async intentFulfilling(query, threadId, thread, 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({
+      query,
+      thread,
+      systemPrompt: systemPrompt.trim()
+    });
+    const tools = [];
+    if (this.mcpModule) {
+      tools.push(...this.mcpModule.getTools());
+    }
+    if (this.a2aModule) {
+      tools.push(...await this.a2aModule.getTools());
+    }
+    const functions = modelInstance.convertToolsToFunctions(tools);
+    const processList = [];
+    let finalMessage = "";
+    let didCallTool = false;
+    while (true) {
+      const response = await modelInstance.fetchWithContextMessage(
+        messages,
+        functions
+      );
+      didCallTool = false;
+      loggers.intent.debug("messages", { messages });
+      const { content, toolCalls } = response;
+      loggers.intent.debug("content", { content });
+      loggers.intent.debug("tool_calls", { ...toolCalls });
+      if (toolCalls) {
+        const messagePayload = this.a2aModule?.getMessagePayload(
+          query,
+          threadId
+        );
+        for (const toolCall of toolCalls) {
+          const toolName = toolCall.name;
+          didCallTool = true;
+          const selectedTool = tools.filter((tool) => tool.id === toolName)[0];
+          let toolResult = "";
+          if (this.mcpModule && selectedTool.protocol === "MCP" /* MCP */) {
+            const toolArgs = toolCall.arguments;
+            loggers.intent.debug("MCP tool call", { toolName, toolArgs });
+            toolResult = await this.mcpModule.useTool(
+              selectedTool,
+              toolArgs
+            );
+          } else if (this.a2aModule && selectedTool.protocol === "A2A" /* A2A */) {
+            toolResult = await this.a2aModule.useTool(
+              selectedTool,
+              messagePayload,
+              threadId
+            );
+          } else {
+            loggers.intent.warn(
+              `Unrecognized tool type: ${selectedTool.protocol}`
+            );
+            continue;
+          }
+          loggers.intent.debug("toolResult", { toolResult });
+          processList.push(toolResult);
+          modelInstance.appendMessages(messages, toolResult);
+        }
+      } else if (content) {
+        processList.push(content);
+        finalMessage = content;
+      }
+      if (!didCallTool) break;
+    }
+    const botResponse = {
+      process: processList.join("\n"),
+      response: finalMessage
+    };
+    return botResponse;
+  }
+  /**
+   * Generates a title for the conversation based on the query.
+   *
+   * @param query - The user's input query
+   * @returns Promise resolving to a generated title
+   */
+  async generateTitle(query) {
+    const DEFAULT_TITLE = "New Chat";
+    try {
+      const modelInstance = this.modelModule.getModel();
+      const messages = modelInstance.generateMessages({
+        query,
+        systemPrompt: `You are a helpful assistant that generates titles for conversations.
+  Please analyze the user's query and create a concise title that accurately reflects the conversation's core topic.
+  The title must be no more than 5 words long.
+  Respond with only the title. Do not include any punctuation or extra explanations.`
+      });
+      const response = await modelInstance.fetch(messages);
+      return response.content || DEFAULT_TITLE;
+    } catch (error) {
+      loggers.intent.error("Error generating title", {
+        error,
+        query
+      });
+      return DEFAULT_TITLE;
+    }
+  }
+  /**
+   * Main entry point for processing user queries.
+   *
+   * Handles the complete query lifecycle:
+   * 1. Loads thread history from memory
+   * 2. Detects intent from the query
+   * 3. Fulfills the intent with AI response
+   * 4. Updates conversation history
+   *
+   * @param type - The type of thread (e.g., chat, workflow)
+   * @param userId - The user's unique identifier
+   * @param threadId - Unique thread identifier
+   * @param query - The user's input query
+   */
+  async handleQuery(threadMetadata, query) {
+    const { type, userId } = threadMetadata;
+    const queryStartAt = Date.now();
+    const threadMemory = this.memoryModule?.getThreadMemory();
+    let threadId = threadMetadata.threadId;
+    let thread;
+    if (threadId) {
+      thread = await threadMemory?.getThread(type, userId, threadId);
+    } else {
+      threadId = randomUUID();
+      const title = await this.generateTitle(query);
+      const metadata = await threadMemory?.createThread(type, userId, threadId, title) || {
+        type,
+        threadId,
+        title,
+        updatedAt: Date.now()
+      };
+      loggers.intent.info("Create new thread", { metadata });
+    }
+    const intent = await this.intentTriggering(query, thread);
+    const result = await this.intentFulfilling(query, threadId, thread, intent);
+    await threadMemory?.addMessagesToThread(userId, threadId, [
+      {
+        role: "USER" /* USER */,
+        timestamp: queryStartAt,
+        content: { type: "text", parts: [query] }
+      },
+      {
+        role: "MODEL" /* MODEL */,
+        timestamp: Date.now(),
+        content: { type: "text", parts: [result.response] }
+      }
+    ]);
+    return { content: result.response };
+  }
+};
+
+export {
+  QueryService
+};
+//# sourceMappingURL=chunk-A65XRRHY.js.map

package/dist/esm/chunk-A65XRRHY.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\t\tlet didCallTool = false;\n\n\t\twhile (true) {\n\t\t\tconst response = await modelInstance.fetchWithContextMessage(\n\t\t\t\tmessages,\n\t\t\t\tfunctions,\n\t\t\t);\n\t\t\tdidCallTool = false;\n\n\t\t\tloggers.intent.debug(\"messages\", { messages });\n\n\t\t\tconst { content, toolCalls } = response;\n\n\t\t\tloggers.intent.debug(\"content\", { content });\n\t\t\tloggers.intent.debug(\"tool_calls\", { ...toolCalls });\n\n\t\t\tif (toolCalls) {\n\t\t\t\tconst messagePayload = this.a2aModule?.getMessagePayload(\n\t\t\t\t\tquery,\n\t\t\t\t\tthreadId,\n\t\t\t\t);\n\n\t\t\t\tfor (const toolCall of toolCalls) {\n\t\t\t\t\tconst toolName = toolCall.name;\n\t\t\t\t\tdidCallTool = true;\n\t\t\t\t\tconst selectedTool = tools.filter((tool) => tool.id === toolName)[0];\n\n\t\t\t\t\tlet toolResult = \"\";\n\t\t\t\t\tif (\n\t\t\t\t\t\tthis.mcpModule &&\n\t\t\t\t\t\tselectedTool.protocol === TOOL_PROTOCOL_TYPE.MCP\n\t\t\t\t\t) {\n\t\t\t\t\t\tconst toolArgs = toolCall.arguments as\n\t\t\t\t\t\t\t| { [x: string]: unknown }\n\t\t\t\t\t\t\t| undefined;\n\t\t\t\t\t\tloggers.intent.debug(\"MCP tool call\", { toolName, toolArgs });\n\t\t\t\t\t\ttoolResult = await this.mcpModule.useTool(\n\t\t\t\t\t\t\tselectedTool as IMCPTool,\n\t\t\t\t\t\t\ttoolArgs,\n\t\t\t\t\t\t);\n\t\t\t\t\t} else if (\n\t\t\t\t\t\tthis.a2aModule &&\n\t\t\t\t\t\tselectedTool.protocol === TOOL_PROTOCOL_TYPE.A2A\n\t\t\t\t\t) {\n\t\t\t\t\t\ttoolResult = await this.a2aModule.useTool(\n\t\t\t\t\t\t\tselectedTool as IA2ATool,\n\t\t\t\t\t\t\tmessagePayload!,\n\t\t\t\t\t\t\tthreadId,\n\t\t\t\t\t\t);\n\t\t\t\t\t} else {\n\t\t\t\t\t\t// Unrecognized tool type. It cannot be happened...\n\t\t\t\t\t\tloggers.intent.warn(\n\t\t\t\t\t\t\t`Unrecognized tool type: ${selectedTool.protocol}`,\n\t\t\t\t\t\t);\n\t\t\t\t\t\tcontinue;\n\t\t\t\t\t}\n\n\t\t\t\t\tloggers.intent.debug(\"toolResult\", { toolResult });\n\n\t\t\t\t\tprocessList.push(toolResult);\n\t\t\t\t\tmodelInstance.appendMessages(messages, toolResult);\n\t\t\t\t}\n\t\t\t} else if (content) {\n\t\t\t\tprocessList.push(content);\n\t\t\t\tfinalMessage = content;\n\t\t\t}\n\n\t\t\tif (!didCallTool) break;\n\t\t}\n\n\t\tconst botResponse = {\n\t\t\tprocess: processList.join(\"\\n\"),\n\t\t\tresponse: finalMessage,\n\t\t};\n\n\t\treturn botResponse;\n\t}\n\n\t/**\n\t * Generates a title for the conversation based on the query.\n\t *\n\t * @param query - The user's input query\n\t * @returns Promise resolving to a generated title\n\t */\n\n\tprivate async generateTitle(query: string): Promise<string> {\n\t\tconst DEFAULT_TITLE = \"New Chat\";\n\t\ttry {\n\t\t\tconst modelInstance = this.modelModule.getModel();\n\t\t\tconst messages = modelInstance.generateMessages({\n\t\t\t\tquery,\n\t\t\t\tsystemPrompt: `You are a helpful assistant that generates titles for conversations.\n  Please analyze the user's query and create a concise title that accurately reflects the conversation's core topic.\n  The title must be no more than 5 words long.\n  Respond with only the title. Do not include any punctuation or extra explanations.`,\n\t\t\t});\n\t\t\tconst response = await modelInstance.fetch(messages);\n\t\t\treturn response.content || DEFAULT_TITLE;\n\t\t} catch (error) {\n\t\t\tloggers.intent.error(\"Error generating title\", {\n\t\t\t\terror,\n\t\t\t\tquery,\n\t\t\t});\n\t\t\treturn DEFAULT_TITLE;\n\t\t}\n\t}\n\n\t/**\n\t * Main entry point for processing user queries.\n\t *\n\t * Handles the complete query lifecycle:\n\t * 1. Loads thread history from memory\n\t * 2. Detects intent from the query\n\t * 3. Fulfills the intent with AI response\n\t * 4. Updates conversation history\n\t *\n\t * @param type - The type of thread (e.g., chat, workflow)\n\t * @param userId - The user's unique identifier\n\t * @param threadId - Unique thread identifier\n\t * @param query - The user's input query\n\t */\n\tpublic async handleQuery(\n\t\tthreadMetadata: {\n\t\t\ttype: ThreadType;\n\t\t\tuserId: string;\n\t\t\tthreadId?: string;\n\t\t},\n\t\tquery: string,\n\t) {\n\t\t// 1. Load thread with threadId\n\t\tconst { type, userId } = threadMetadata;\n\t\tconst queryStartAt = Date.now();\n\t\tconst threadMemory = this.memoryModule?.getThreadMemory();\n\n\t\tlet threadId = threadMetadata.threadId;\n\t\tlet thread: ThreadObject | undefined;\n\n\t\tif (threadId) {\n\t\t\tthread = await threadMemory?.getThread(type, userId, threadId);\n\t\t} else {\n\t\t\tthreadId = randomUUID();\n\t\t\tconst title = await this.generateTitle(query);\n\n\t\t\tconst metadata =\n\t\t\t\t(await threadMemory?.createThread(type, userId, threadId, title)) ||\n\t\t\t\t({\n\t\t\t\t\ttype,\n\t\t\t\t\tthreadId,\n\t\t\t\t\ttitle,\n\t\t\t\t\tupdatedAt: Date.now(),\n\t\t\t\t} as ThreadMetadata);\n\t\t\tloggers.intent.info(\"Create new thread\", { metadata });\n\t\t}\n\n\t\t// 2. intent triggering\n\t\tconst intent = 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;AACnB,QAAI,cAAc;AAElB,WAAO,MAAM;AACZ,YAAM,WAAW,MAAM,cAAc;AAAA,QACpC;AAAA,QACA;AAAA,MACD;AACA,oBAAc;AAEd,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,wBAAc;AACd,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;AAAA,MAChB;AAEA,UAAI,CAAC,YAAa;AAAA,IACnB;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,MAAM,QAAQ,QAAQ;AAAA,IAC9D,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-5EPK37PE.js → chunk-ADIAIRNH.js}

@@ -2,27 +2,27 @@ import {
   AinHttpError
 } from "./chunk-FL3JRQM4.js";
 
-// src/controllers/api/session.api.controller.ts
+// src/controllers/api/thread.api.controller.ts
 import { StatusCodes } from "http-status-codes";
-var SessionApiController = class {
+var ThreadApiController = class {
   memoryModule;
   constructor(memoryModule) {
     this.memoryModule = memoryModule;
   }
-  handleSessionHistory = async (req, res, next) => {
+  handleGetThread = async (req, res, next) => {
     try {
-      const { id: sessionId } = req.params;
+      const { id: threadId, type } = req.params;
       const userId = res.locals.userId || "";
-      const sessionMemory = this.memoryModule.getSessionMemory();
-      if (!sessionMemory) {
+      const threadMemory = this.memoryModule.getThreadMemory();
+      if (!threadMemory) {
         const error = new AinHttpError(
           StatusCodes.SERVICE_UNAVAILABLE,
           "Memory module is not initialized"
         );
         throw error;
       }
-      const session = await sessionMemory.getSession(userId, sessionId);
-      res.json(session);
+      const thread = await threadMemory.getThread(type, userId, threadId);
+      res.json(thread);
     } catch (error) {
       next(error);
     }
@@ -30,6 +30,6 @@ var SessionApiController = class {
 };
 
 export {
-  SessionApiController
+  ThreadApiController
 };
-//# sourceMappingURL=chunk-5EPK37PE.js.map
+//# sourceMappingURL=chunk-ADIAIRNH.js.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/controllers/api/thread.api.controller.ts"],"sourcesContent":["import type { NextFunction, Request, Response } from \"express\";\nimport { StatusCodes } from \"http-status-codes\";\nimport type { MemoryModule } from \"@/modules/index.js\";\nimport { AinHttpError } from \"@/types/agent.js\";\nimport type { ThreadType } from \"@/types/memory\";\n\nexport class ThreadApiController {\n\tprivate memoryModule: MemoryModule;\n\n\tconstructor(memoryModule: MemoryModule) {\n\t\tthis.memoryModule = memoryModule;\n\t}\n\n\tpublic handleGetThread = async (\n\t\treq: Request,\n\t\tres: Response,\n\t\tnext: NextFunction,\n\t) => {\n\t\ttry {\n\t\t\tconst { id: threadId, type } = req.params as {\n\t\t\t\tid: string;\n\t\t\t\ttype: ThreadType;\n\t\t\t};\n\t\t\tconst userId = res.locals.userId || \"\";\n\t\t\tconst threadMemory = this.memoryModule.getThreadMemory();\n\t\t\tif (!threadMemory) {\n\t\t\t\tconst error = new AinHttpError(\n\t\t\t\t\tStatusCodes.SERVICE_UNAVAILABLE,\n\t\t\t\t\t\"Memory module is not initialized\",\n\t\t\t\t);\n\t\t\t\tthrow error;\n\t\t\t}\n\t\t\tconst thread = await threadMemory.getThread(type, userId, threadId);\n\t\t\tres.json(thread);\n\t\t} catch (error) {\n\t\t\tnext(error);\n\t\t}\n\t};\n}\n"],"mappings":";;;;;AACA,SAAS,mBAAmB;AAKrB,IAAM,sBAAN,MAA0B;AAAA,EACxB;AAAA,EAER,YAAY,cAA4B;AACvC,SAAK,eAAe;AAAA,EACrB;AAAA,EAEO,kBAAkB,OACxB,KACA,KACA,SACI;AACJ,QAAI;AACH,YAAM,EAAE,IAAI,UAAU,KAAK,IAAI,IAAI;AAInC,YAAM,SAAS,IAAI,OAAO,UAAU;AACpC,YAAM,eAAe,KAAK,aAAa,gBAAgB;AACvD,UAAI,CAAC,cAAc;AAClB,cAAM,QAAQ,IAAI;AAAA,UACjB,YAAY;AAAA,UACZ;AAAA,QACD;AACA,cAAM;AAAA,MACP;AACA,YAAM,SAAS,MAAM,aAAa,UAAU,MAAM,QAAQ,QAAQ;AAClE,UAAI,KAAK,MAAM;AAAA,IAChB,SAAS,OAAO;AACf,WAAK,KAAK;AAAA,IACX;AAAA,EACD;AACD;","names":[]}

package/dist/esm/{chunk-2Z7GJVJJ.js → chunk-AMRKK46V.js}

@@ -2,7 +2,7 @@ import {
   ModelApiController
 } from "./chunk-UA44VKFE.js";
 
-// src/routes/api/models.routes.ts
+// src/routes/api/model.routes.ts
 import { Router } from "express";
 var createModelApiRouter = (modelModule) => {
   const router = Router();
@@ -14,4 +14,4 @@ var createModelApiRouter = (modelModule) => {
 export {
   createModelApiRouter
 };
-//# sourceMappingURL=chunk-2Z7GJVJJ.js.map
+//# sourceMappingURL=chunk-AMRKK46V.js.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/routes/api/model.routes.ts"],"sourcesContent":["import { Router } from \"express\";\nimport { ModelApiController } from \"@/controllers/api/model.api.controller.js\";\nimport type { ModelModule } from \"@/modules/index.js\";\n\nexport const createModelApiRouter = (modelModule: ModelModule): Router => {\n\tconst router = Router();\n\tconst modelApiController = new ModelApiController(modelModule);\n\n\t// APIs (prefix: /api/model)\n\trouter.get(\"/\", modelApiController.handleModelList);\n\n\treturn router;\n};\n"],"mappings":";;;;;AAAA,SAAS,cAAc;AAIhB,IAAM,uBAAuB,CAAC,gBAAqC;AACzE,QAAM,SAAS,OAAO;AACtB,QAAM,qBAAqB,IAAI,mBAAmB,WAAW;AAG7D,SAAO,IAAI,KAAK,mBAAmB,eAAe;AAElD,SAAO;AACR;","names":[]}

package/dist/esm/{chunk-2BITOA2R.js → chunk-CDFLKADH.js}

@@ -5,4 +5,4 @@ var BaseModel = class {
 export {
   BaseModel
 };
-//# sourceMappingURL=chunk-2BITOA2R.js.map
+//# sourceMappingURL=chunk-CDFLKADH.js.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/modules/models/base.model.ts"],"sourcesContent":["import type { ThreadObject } from \"@/types/memory.js\";\nimport type { LLMStream } from \"@/types/stream.js\";\nimport type { FetchResponse, IAgentTool } from \"@/types/tool.js\";\n\n/**\n * Abstract base class for AI model implementations.\n *\n * Provides a common interface for different AI model providers (OpenAI, Gemini, etc.)\n * to integrate with the AIN-ADK framework. Each model implementation must handle\n * message formatting, tool conversion, and API communication.\n *\n * @typeParam MessageType - The message format used by the specific model API\n * @typeParam FunctionType - The function/tool format used by the specific model API\n */\nexport abstract class BaseModel<MessageType, FunctionType> {\n\t/**\n\t * Generates an array of messages from thread and current query.\n\t *\n\t * @param query - Current user query\n\t * @param thread - Previous conversation history\n\t * @param systemPrompt - Optional system prompt to set context\n\t * @returns Array of messages formatted for the specific model API\n\t */\n\tabstract generateMessages(params: {\n\t\tquery: string;\n\t\tthread?: ThreadObject;\n\t\tsystemPrompt?: string;\n\t}): MessageType[];\n\n\t/**\n\t * Appends a new message to the existing message array.\n\t *\n\t * @param messages - Existing message array to expand\n\t * @param message - New message content to append\n\t */\n\tabstract appendMessages(messages: MessageType[], message: string): void;\n\n\t/**\n\t * Converts protocol-agnostic tools to model-specific function format.\n\t *\n\t * @param tools - Array of agent tools from MCP or A2A sources\n\t * @returns Array of functions in the format required by the model API\n\t */\n\tabstract convertToolsToFunctions(tools: IAgentTool[]): FunctionType[];\n\n\t/**\n\t * Fetches a response from the model API without tool support.\n\t *\n\t * @param messages - Array of messages to send to the model\n\t * @returns Promise resolving to the model's response\n\t */\n\tabstract fetch(messages: MessageType[]): Promise<FetchResponse>;\n\n\t/**\n\t * Fetches a response from the model API with tool/function support.\n\t *\n\t * @param messages - Array of messages to send to the model\n\t * @param functions - Array of available functions/tools the model can call\n\t * @returns Promise resolving to the model's response, possibly including tool calls\n\t */\n\tabstract fetchWithContextMessage(\n\t\tmessages: MessageType[],\n\t\tfunctions: FunctionType[],\n\t): Promise<FetchResponse>;\n\n\t/**\n\t * Fetches a streaming response from the model API with tool/function support.\n\t *\n\t * Returns a standardized LLMStream that can be used consistently across\n\t * different AI model providers. Each implementation should convert their\n\t * provider-specific stream format to the common StreamChunk interface.\n\t *\n\t * @param messages - Array of messages to send to the model\n\t * @param functions - Array of available functions/tools the model can call\n\t * @returns Promise resolving to an LLMStream for consistent iteration\n\t */\n\tabstract fetchStreamWithContextMessage(\n\t\tmessages: MessageType[],\n\t\tfunctions: FunctionType[],\n\t): Promise<LLMStream>;\n}\n"],"mappings":";AAcO,IAAe,YAAf,MAAoD;AAkE3D;","names":[]}

package/dist/esm/{chunk-GNWF6QG4.js → chunk-G37LGM6P.js}

@@ -1,12 +1,12 @@
 import {
   QueryStreamService
-} from "./chunk-3NEV3O2X.js";
+} from "./chunk-M4EJC7LZ.js";
 import {
   QueryController
-} from "./chunk-Q6HHVAO6.js";
+} from "./chunk-7D4SKPIS.js";
 import {
   QueryService
-} from "./chunk-DRUCJYPJ.js";
+} from "./chunk-A65XRRHY.js";
 
 // src/routes/query.routes.ts
 import { Router } from "express";
@@ -40,4 +40,4 @@ var createQueryRouter = (agent, allowStream = false) => {
 export {
   createQueryRouter
 };
-//# sourceMappingURL=chunk-GNWF6QG4.js.map
+//# sourceMappingURL=chunk-G37LGM6P.js.map

package/dist/esm/chunk-HOFAK5J2.js

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

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

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/routes/api.routes.ts"],"sourcesContent":["import { Router } from \"express\";\nimport type { AINAgent } from \"@/index.js\";\n\nimport { createModelApiRouter } from \"./api/model.routes.js\";\nimport { createThreadApiRouter } from \"./api/thread.routes.js\";\n\nexport const createApiRouter = (agent: AINAgent): Router => {\n\tconst router = Router();\n\n\trouter.use(\"/api/model\", createModelApiRouter(agent.modelModule));\n\tif (agent.memoryModule) {\n\t\trouter.use(\"/api/thread\", createThreadApiRouter(agent.memoryModule));\n\t}\n\n\treturn router;\n};\n"],"mappings":";;;;;;;;AAAA,SAAS,cAAc;AAMhB,IAAM,kBAAkB,CAAC,UAA4B;AAC3D,QAAM,SAAS,OAAO;AAEtB,SAAO,IAAI,cAAc,qBAAqB,MAAM,WAAW,CAAC;AAChE,MAAI,MAAM,cAAc;AACvB,WAAO,IAAI,eAAe,sBAAsB,MAAM,YAAY,CAAC;AAAA,EACpE;AAEA,SAAO;AACR;","names":[]}

package/dist/esm/{chunk-5N4X25IK.js → chunk-KMQX6T56.js}

@@ -1,9 +1,9 @@
 import {
   A2AService
-} from "./chunk-Y6E73I7I.js";
+} from "./chunk-2RW77MJB.js";
 import {
   QueryService
-} from "./chunk-DRUCJYPJ.js";
+} from "./chunk-A65XRRHY.js";
 import {
   A2AController
 } from "./chunk-HAE333YO.js";
@@ -34,4 +34,4 @@ var createA2ARouter = (agent) => {
 export {
   createA2ARouter
 };
-//# sourceMappingURL=chunk-5N4X25IK.js.map
+//# sourceMappingURL=chunk-KMQX6T56.js.map

package/dist/esm/{chunk-C6SGV7JV.js → chunk-LJVVSUF7.js}

@@ -13,6 +13,8 @@ var A2AModule = class {
   a2aPeerServers = /* @__PURE__ */ new Map();
   /** Map of session IDs to their A2A session state */
   a2aSessions = /* @__PURE__ */ new Map();
+  agentId = randomUUID();
+  /* FIXME */
   /**
    * Registers a new A2A peer server URL for connection.
    *
@@ -54,16 +56,16 @@ var A2AModule = class {
   /**
    * Gets or creates an A2A session for the given session ID.
    *
-   * @param sessionId - The session identifier
+   * @param threadId - The session identifier
    * @returns A2ASession object with task and context IDs
    */
-  getA2ASessionWithId = (sessionId) => {
-    const a2aSession = this.a2aSessions.get(sessionId) ?? {
+  getA2ASessionWithId = (threadId) => {
+    const a2aSession = this.a2aSessions.get(threadId) ?? {
       taskId: void 0,
       contextId: void 0
     };
-    if (!this.a2aSessions.has(sessionId)) {
-      this.a2aSessions.set(sessionId, a2aSession);
+    if (!this.a2aSessions.has(threadId)) {
+      this.a2aSessions.set(threadId, a2aSession);
     }
     return a2aSession;
   };
@@ -74,17 +76,19 @@ var A2AModule = class {
    * for maintaining conversation continuity.
    *
    * @param query - The message content to send
-   * @param sessionId - The session identifier
+   * @param threadId - The session identifier
    * @returns Formatted Message object for A2A protocol
    */
-  getMessagePayload(query, sessionId) {
+  getMessagePayload(query, threadId) {
     const messagePayload = {
       messageId: randomUUID(),
       kind: "message",
       role: "user",
       // FIXME: it could be 'agent'
       metadata: {
-        sessionId
+        agentId: this.agentId,
+        type: "CHAT" /* CHAT */,
+        threadId
       },
       parts: [
         {
@@ -93,7 +97,7 @@ var A2AModule = class {
         }
       ]
     };
-    const a2aSession = this.getA2ASessionWithId(sessionId);
+    const a2aSession = this.getA2ASessionWithId(threadId);
     if (a2aSession.taskId) {
       messagePayload.taskId = a2aSession.taskId;
     }
@@ -110,16 +114,16 @@ var A2AModule = class {
    *
    * @param tool - The A2ATool instance to use
    * @param messagePayload - The message to send to the agent
-   * @param sessionId - The session identifier for context tracking
+   * @param threadId - The session identifier for context tracking
    * @returns Promise resolving to array of text responses from the agent
    */
-  async useTool(tool, messagePayload, sessionId) {
+  async useTool(tool, messagePayload, threadId) {
     const finalText = [];
     const client = tool.client;
     const params = {
       message: messagePayload
     };
-    const a2aSession = this.getA2ASessionWithId(sessionId);
+    const a2aSession = this.getA2ASessionWithId(threadId);
     try {
       const stream = client.sendMessageStream(params);
       for await (const event of stream) {
@@ -169,4 +173,4 @@ ${finalText.join("\n")}`;
 export {
   A2AModule
 };
-//# sourceMappingURL=chunk-C6SGV7JV.js.map
+//# sourceMappingURL=chunk-LJVVSUF7.js.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/modules/a2a/a2a.module.ts"],"sourcesContent":["import { randomUUID } from \"node:crypto\";\nimport type {\n\tAgentCard,\n\tMessage,\n\tMessageSendParams,\n\tTask,\n\tTaskStatusUpdateEvent,\n\tTextPart,\n} from \"@a2a-js/sdk\";\nimport { A2AClient } from \"@a2a-js/sdk/client\";\nimport { ThreadType } from \"@/types/memory.js\";\nimport { loggers } from \"@/utils/logger.js\";\nimport { A2ATool } from \"./a2a.tool.js\";\n\n/**\n * Represents an active A2A communication session.\n */\ninterface A2ASession {\n\t/** Current task ID for multi-turn conversations */\n\ttaskId: string | undefined;\n\t/** Context ID for maintaining conversation state */\n\tcontextId: string | undefined;\n}\n\n/**\n * Module for managing Agent-to-Agent (A2A) protocol connections.\n *\n * This module handles connections to other A2A-compatible agents, manages\n * conversation sessions, and provides an interface for inter-agent communication.\n * Supports multi-turn conversations with task and context tracking.\n */\nexport class A2AModule {\n\t/** Map of A2A server URLs to their corresponding tool instances */\n\tprivate a2aPeerServers: Map<string, A2ATool | null> = new Map();\n\t/** Map of session IDs to their A2A session state */\n\tprivate a2aSessions: Map<string, A2ASession> = new Map();\n\tprivate agentId: string = randomUUID(); /* FIXME */\n\n\t/**\n\t * Registers a new A2A peer server URL for connection.\n\t *\n\t * @param url - The URL of the A2A-compatible agent to connect to\n\t */\n\tpublic async addA2APeerServer(url: string): Promise<void> {\n\t\tthis.a2aPeerServers.set(url, null);\n\t}\n\n\t/**\n\t * Retrieves tools from all registered A2A peer servers.\n\t *\n\t * Attempts to connect to each registered server, fetch their agent cards,\n\t * and create tool instances. Disables tools for unreachable servers.\n\t *\n\t * @returns Promise resolving to array of available A2A tools\n\t */\n\tpublic async getTools(): Promise<A2ATool[]> {\n\t\tconst tools: A2ATool[] = [];\n\t\tfor (const url of [...this.a2aPeerServers.keys()]) {\n\t\t\tconst tool = this.a2aPeerServers.get(url);\n\t\t\tif (!tool || !tool.enabled) {\n\t\t\t\ttry {\n\t\t\t\t\tconst client = new A2AClient(url);\n\t\t\t\t\tconst card: AgentCard = await client.getAgentCard();\n\t\t\t\t\tconst toolName = card.name.replaceAll(\" \", \"-\");\n\t\t\t\t\tconst a2aTool = new A2ATool(toolName, client, card);\n\n\t\t\t\t\ttools.push(a2aTool);\n\t\t\t\t} catch (_error: any) {\n\t\t\t\t\t// Agent not responded\n\t\t\t\t\tif (tool) {\n\t\t\t\t\t\ttool.disable();\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\ttools.push(tool);\n\t\t\t}\n\t\t}\n\t\treturn tools;\n\t}\n\n\t/**\n\t * Gets or creates an A2A session for the given session ID.\n\t *\n\t * @param threadId - The session identifier\n\t * @returns A2ASession object with task and context IDs\n\t */\n\tprivate getA2ASessionWithId = (threadId: string): A2ASession => {\n\t\tconst a2aSession = this.a2aSessions.get(threadId) ?? {\n\t\t\ttaskId: undefined,\n\t\t\tcontextId: undefined,\n\t\t};\n\t\tif (!this.a2aSessions.has(threadId)) {\n\t\t\tthis.a2aSessions.set(threadId, a2aSession);\n\t\t}\n\n\t\treturn a2aSession;\n\t};\n\n\t/**\n\t * Constructs a message payload for A2A communication.\n\t *\n\t * Includes session context (task ID and context ID) if available\n\t * for maintaining conversation continuity.\n\t *\n\t * @param query - The message content to send\n\t * @param threadId - The session identifier\n\t * @returns Formatted Message object for A2A protocol\n\t */\n\tpublic getMessagePayload(query: string, threadId: string): Message {\n\t\tconst messagePayload: Message = {\n\t\t\tmessageId: randomUUID(),\n\t\t\tkind: \"message\",\n\t\t\trole: \"user\", // FIXME: it could be 'agent'\n\t\t\tmetadata: {\n\t\t\t\tagentId: this.agentId,\n\t\t\t\ttype: ThreadType.CHAT,\n\t\t\t\tthreadId,\n\t\t\t},\n\t\t\tparts: [\n\t\t\t\t{\n\t\t\t\t\tkind: \"text\",\n\t\t\t\t\ttext: query,\n\t\t\t\t},\n\t\t\t],\n\t\t};\n\n\t\tconst a2aSession = this.getA2ASessionWithId(threadId);\n\t\tif (a2aSession.taskId) {\n\t\t\tmessagePayload.taskId = a2aSession.taskId;\n\t\t}\n\t\tif (a2aSession.contextId) {\n\t\t\tmessagePayload.contextId = a2aSession.contextId;\n\t\t}\n\n\t\treturn messagePayload;\n\t}\n\n\t/**\n\t * Executes an A2A tool by sending a message to the remote agent.\n\t *\n\t * Handles streaming responses, maintains session state, and extracts\n\t * text content from various event types in the response stream.\n\t *\n\t * @param tool - The A2ATool instance to use\n\t * @param messagePayload - The message to send to the agent\n\t * @param threadId - The session identifier for context tracking\n\t * @returns Promise resolving to array of text responses from the agent\n\t */\n\tpublic async useTool(\n\t\ttool: A2ATool,\n\t\tmessagePayload: Message,\n\t\tthreadId: string,\n\t): Promise<string> {\n\t\tconst finalText: string[] = [];\n\t\tconst client = tool.client;\n\t\tconst params: MessageSendParams = {\n\t\t\tmessage: messagePayload,\n\t\t};\n\t\tconst a2aSession = this.getA2ASessionWithId(threadId);\n\n\t\ttry {\n\t\t\tconst stream = client.sendMessageStream(params);\n\t\t\tfor await (const event of stream) {\n\t\t\t\tif (event.kind === \"status-update\") {\n\t\t\t\t\tconst typedEvent = event as TaskStatusUpdateEvent;\n\t\t\t\t\tif (\n\t\t\t\t\t\ttypedEvent.final &&\n\t\t\t\t\t\ttypedEvent.status.state !== \"input-required\"\n\t\t\t\t\t) {\n\t\t\t\t\t\ta2aSession.taskId = undefined;\n\t\t\t\t\t}\n\t\t\t\t\t// TODO: handle 'file', 'data' parts\n\t\t\t\t\tconst texts = typedEvent.status.message?.parts\n\t\t\t\t\t\t.filter((part) => part.kind === \"text\")\n\t\t\t\t\t\t.map((part: TextPart) => part.text)\n\t\t\t\t\t\t.join(\"\\n\");\n\t\t\t\t\tif (texts) {\n\t\t\t\t\t\tfinalText.push(texts);\n\t\t\t\t\t}\n\t\t\t\t} else if (event.kind === \"message\") {\n\t\t\t\t\t// FIXME: handling text in 'message'?\n\t\t\t\t\tconst msg = event as Message;\n\t\t\t\t\tif (msg.taskId && msg.taskId !== a2aSession.taskId) {\n\t\t\t\t\t\ta2aSession.taskId = msg.taskId;\n\t\t\t\t\t}\n\t\t\t\t\tif (msg.contextId && msg.contextId !== a2aSession.contextId) {\n\t\t\t\t\t\ta2aSession.contextId = msg.contextId;\n\t\t\t\t\t}\n\t\t\t\t} else if (event.kind === \"task\") {\n\t\t\t\t\t// FIXME: handling text in 'task'?\n\t\t\t\t\tconst task = event as Task;\n\t\t\t\t\tif (task.id !== a2aSession.taskId) {\n\t\t\t\t\t\ta2aSession.taskId = task.id;\n\t\t\t\t\t}\n\t\t\t\t\tif (task.contextId && task.contextId !== a2aSession.contextId) {\n\t\t\t\t\t\ta2aSession.contextId = task.contextId;\n\t\t\t\t\t}\n\t\t\t\t} else {\n\t\t\t\t\tloggers.a2a.warn(\"Received unknown event structure from stream:\", {\n\t\t\t\t\t\tevent,\n\t\t\t\t\t});\n\t\t\t\t}\n\t\t\t}\n\t\t} catch (error) {\n\t\t\tloggers.a2a.error(\"Error communicating with agent:\", { error });\n\t\t\ttool.disable();\n\t\t\tconst toolResult = `[Bot Called A2A Tool ${tool.card.name}]\\n${typeof error === \"string\" ? error : JSON.stringify(error, null, 2)}`;\n\t\t\treturn toolResult;\n\t\t}\n\n\t\treturn `[Bot Called A2A Tool ${tool.card.name}]\\n${finalText.join(\"\\n\")}`;\n\t}\n}\n"],"mappings":";;;;;;;;AAAA,SAAS,kBAAkB;AAS3B,SAAS,iBAAiB;AAsBnB,IAAM,YAAN,MAAgB;AAAA;AAAA,EAEd,iBAA8C,oBAAI,IAAI;AAAA;AAAA,EAEtD,cAAuC,oBAAI,IAAI;AAAA,EAC/C,UAAkB,WAAW;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOrC,MAAa,iBAAiB,KAA4B;AACzD,SAAK,eAAe,IAAI,KAAK,IAAI;AAAA,EAClC;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAUA,MAAa,WAA+B;AAC3C,UAAM,QAAmB,CAAC;AAC1B,eAAW,OAAO,CAAC,GAAG,KAAK,eAAe,KAAK,CAAC,GAAG;AAClD,YAAM,OAAO,KAAK,eAAe,IAAI,GAAG;AACxC,UAAI,CAAC,QAAQ,CAAC,KAAK,SAAS;AAC3B,YAAI;AACH,gBAAM,SAAS,IAAI,UAAU,GAAG;AAChC,gBAAM,OAAkB,MAAM,OAAO,aAAa;AAClD,gBAAM,WAAW,KAAK,KAAK,WAAW,KAAK,GAAG;AAC9C,gBAAM,UAAU,IAAI,QAAQ,UAAU,QAAQ,IAAI;AAElD,gBAAM,KAAK,OAAO;AAAA,QACnB,SAAS,QAAa;AAErB,cAAI,MAAM;AACT,iBAAK,QAAQ;AAAA,UACd;AAAA,QACD;AAAA,MACD,OAAO;AACN,cAAM,KAAK,IAAI;AAAA,MAChB;AAAA,IACD;AACA,WAAO;AAAA,EACR;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQQ,sBAAsB,CAAC,aAAiC;AAC/D,UAAM,aAAa,KAAK,YAAY,IAAI,QAAQ,KAAK;AAAA,MACpD,QAAQ;AAAA,MACR,WAAW;AAAA,IACZ;AACA,QAAI,CAAC,KAAK,YAAY,IAAI,QAAQ,GAAG;AACpC,WAAK,YAAY,IAAI,UAAU,UAAU;AAAA,IAC1C;AAEA,WAAO;AAAA,EACR;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAYO,kBAAkB,OAAe,UAA2B;AAClE,UAAM,iBAA0B;AAAA,MAC/B,WAAW,WAAW;AAAA,MACtB,MAAM;AAAA,MACN,MAAM;AAAA;AAAA,MACN,UAAU;AAAA,QACT,SAAS,KAAK;AAAA,QACd;AAAA,QACA;AAAA,MACD;AAAA,MACA,OAAO;AAAA,QACN;AAAA,UACC,MAAM;AAAA,UACN,MAAM;AAAA,QACP;AAAA,MACD;AAAA,IACD;AAEA,UAAM,aAAa,KAAK,oBAAoB,QAAQ;AACpD,QAAI,WAAW,QAAQ;AACtB,qBAAe,SAAS,WAAW;AAAA,IACpC;AACA,QAAI,WAAW,WAAW;AACzB,qBAAe,YAAY,WAAW;AAAA,IACvC;AAEA,WAAO;AAAA,EACR;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAaA,MAAa,QACZ,MACA,gBACA,UACkB;AAClB,UAAM,YAAsB,CAAC;AAC7B,UAAM,SAAS,KAAK;AACpB,UAAM,SAA4B;AAAA,MACjC,SAAS;AAAA,IACV;AACA,UAAM,aAAa,KAAK,oBAAoB,QAAQ;AAEpD,QAAI;AACH,YAAM,SAAS,OAAO,kBAAkB,MAAM;AAC9C,uBAAiB,SAAS,QAAQ;AACjC,YAAI,MAAM,SAAS,iBAAiB;AACnC,gBAAM,aAAa;AACnB,cACC,WAAW,SACX,WAAW,OAAO,UAAU,kBAC3B;AACD,uBAAW,SAAS;AAAA,UACrB;AAEA,gBAAM,QAAQ,WAAW,OAAO,SAAS,MACvC,OAAO,CAAC,SAAS,KAAK,SAAS,MAAM,EACrC,IAAI,CAAC,SAAmB,KAAK,IAAI,EACjC,KAAK,IAAI;AACX,cAAI,OAAO;AACV,sBAAU,KAAK,KAAK;AAAA,UACrB;AAAA,QACD,WAAW,MAAM,SAAS,WAAW;AAEpC,gBAAM,MAAM;AACZ,cAAI,IAAI,UAAU,IAAI,WAAW,WAAW,QAAQ;AACnD,uBAAW,SAAS,IAAI;AAAA,UACzB;AACA,cAAI,IAAI,aAAa,IAAI,cAAc,WAAW,WAAW;AAC5D,uBAAW,YAAY,IAAI;AAAA,UAC5B;AAAA,QACD,WAAW,MAAM,SAAS,QAAQ;AAEjC,gBAAM,OAAO;AACb,cAAI,KAAK,OAAO,WAAW,QAAQ;AAClC,uBAAW,SAAS,KAAK;AAAA,UAC1B;AACA,cAAI,KAAK,aAAa,KAAK,cAAc,WAAW,WAAW;AAC9D,uBAAW,YAAY,KAAK;AAAA,UAC7B;AAAA,QACD,OAAO;AACN,kBAAQ,IAAI,KAAK,iDAAiD;AAAA,YACjE;AAAA,UACD,CAAC;AAAA,QACF;AAAA,MACD;AAAA,IACD,SAAS,OAAO;AACf,cAAQ,IAAI,MAAM,mCAAmC,EAAE,MAAM,CAAC;AAC9D,WAAK,QAAQ;AACb,YAAM,aAAa,wBAAwB,KAAK,KAAK,IAAI;AAAA,EAAM,OAAO,UAAU,WAAW,QAAQ,KAAK,UAAU,OAAO,MAAM,CAAC,CAAC;AACjI,aAAO;AAAA,IACR;AAEA,WAAO,wBAAwB,KAAK,KAAK,IAAI;AAAA,EAAM,UAAU,KAAK,IAAI,CAAC;AAAA,EACxE;AACD;","names":[]}