@ainetwork/adk 0.1.5 → 0.1.7

package/dist/esm/{chunk-B3R4A2NL.js → chunk-A3W52LKL.js}

@@ -1,12 +1,12 @@
 import {
   A2AService
-} from "./chunk-AQ5YZERE.js";
+} from "./chunk-Y6E73I7I.js";
 import {
   QueryService
-} from "./chunk-356AV44C.js";
+} from "./chunk-MC7CPQ6W.js";
 import {
   A2AController
-} from "./chunk-EODMRV4H.js";
+} from "./chunk-HAE333YO.js";
 
 // src/routes/a2a.routes.ts
 import { InMemoryTaskStore } from "@a2a-js/sdk/server";
@@ -34,4 +34,4 @@ var createA2ARouter = (agent) => {
 export {
   createA2ARouter
 };
-//# sourceMappingURL=chunk-B3R4A2NL.js.map
+//# sourceMappingURL=chunk-A3W52LKL.js.map

package/dist/esm/{chunk-SQWZXP4P.js → chunk-DJRRMFKH.js}

@@ -3,7 +3,7 @@ import {
 } from "./chunk-DGH3GB5I.js";
 import {
   loggers
-} from "./chunk-64GEXWSE.js";
+} from "./chunk-N7VYM2R5.js";
 
 // src/modules/a2a/a2a.module.ts
 import { randomUUID } from "crypto";
@@ -165,4 +165,4 @@ var A2AModule = class {
 export {
   A2AModule
 };
-//# sourceMappingURL=chunk-SQWZXP4P.js.map
+//# sourceMappingURL=chunk-DJRRMFKH.js.map

package/dist/esm/chunk-EFMGNCWG.js

@@ -0,0 +1,225 @@
+import {
+  loggers
+} from "./chunk-N7VYM2R5.js";
+
+// src/services/query-stream.service.ts
+var QueryStreamService = 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 a user query.
+   *
+   * @param query - The user's input query
+   * @returns The detected intent (currently returns the query as-is)
+   * @todo Implement actual intent detection logic
+   */
+  async intentTriggering(query) {
+    return query;
+  }
+  /**
+   * Fulfills the detected intent by generating a response.
+   *
+   * Manages the complete inference loop including:
+   * - Loading prompts and conversation history
+   * - Collecting available tools from modules
+   * - Executing model inference with tool support
+   * - Processing tool calls iteratively until completion
+   *
+   * @param query - The user's input query
+   * @param sessionId - Session identifier for context
+   * @param sessionHistory - Previous conversation history
+   * @returns Object containing process steps and final response
+   */
+  async *intentFulfilling(query, sessionId, sessionHistory) {
+    try {
+      const systemPrompt = `
+Today is ${(/* @__PURE__ */ new Date()).toLocaleDateString()}.
+
+${this.prompts?.agent || ""}
+
+${this.prompts?.system || ""}
+    `;
+      const modelInstance = this.modelModule.getModel();
+      const messages = modelInstance.generateMessages({
+        query,
+        sessionHistory,
+        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 = [];
+      const finalMessage = "";
+      let didCallTool = false;
+      while (true) {
+        const responseStream = await modelInstance.fetchStreamWithContextMessage(
+          messages,
+          functions
+        );
+        didCallTool = false;
+        const assembledToolCalls = [];
+        loggers.intentStream.debug("messages", { messages });
+        for await (const chunk of responseStream) {
+          const delta = chunk.delta;
+          if (delta?.tool_calls) {
+            didCallTool = true;
+            for (const toolCallDelta of delta.tool_calls) {
+              const index = toolCallDelta.index;
+              if (!assembledToolCalls[index]) {
+                assembledToolCalls[index] = {
+                  id: "",
+                  type: "function",
+                  function: { name: "", arguments: "" }
+                };
+              }
+              if (toolCallDelta.id) {
+                assembledToolCalls[index].id = toolCallDelta.id;
+              }
+              if (toolCallDelta.function?.name) {
+                assembledToolCalls[index].function.name = toolCallDelta.function.name;
+              }
+              if (toolCallDelta.function?.arguments) {
+                assembledToolCalls[index].function.arguments += toolCallDelta.function.arguments;
+              }
+            }
+          } else if (chunk.delta?.content) {
+            yield {
+              event: "text_chunk",
+              data: { delta: chunk.delta.content }
+            };
+          }
+        }
+        loggers.intentStream.debug("assembledToolCalls", {
+          assembledToolCalls
+        });
+        if (didCallTool && assembledToolCalls.length > 0) {
+          const messagePayload = this.a2aModule?.getMessagePayload(
+            query,
+            sessionId
+          );
+          for (const toolCall of assembledToolCalls) {
+            const toolName = toolCall.function.name;
+            const toolArgs = JSON.parse(toolCall.function.arguments);
+            const selectedTool = tools.filter(
+              (tool) => tool.id === toolName
+            )[0];
+            yield { event: "tool_start", data: { toolName, toolArgs } };
+            let toolResult = "";
+            if (this.mcpModule && selectedTool.protocol === "MCP" /* MCP */) {
+              const toolArgs2 = JSON.parse(toolCall.function.arguments);
+              loggers.intent.debug("MCP tool call", { toolName, toolArgs: toolArgs2 });
+              const result = await this.mcpModule.useTool(
+                selectedTool,
+                toolArgs2
+              );
+              toolResult = `[Bot Called MCP Tool ${toolName} with args ${JSON.stringify(toolArgs2)}]
+` + JSON.stringify(result.content, null, 2);
+            } else if (this.a2aModule && selectedTool.protocol === "A2A" /* A2A */) {
+              const result = await this.a2aModule.useTool(
+                selectedTool,
+                messagePayload,
+                sessionId
+              );
+              toolResult = `[Bot Called A2A Tool ${toolName}]
+${result.join("\n")}`;
+            } else {
+              loggers.intent.warn(
+                `Unrecognized tool type: ${selectedTool.protocol}`
+              );
+              continue;
+            }
+            yield {
+              event: "tool_output",
+              data: { toolName, result: toolResult }
+            };
+            loggers.intent.debug("toolResult", { toolResult });
+            processList.push(toolResult);
+            modelInstance.appendMessages(messages, toolResult);
+          }
+        }
+        if (!didCallTool) break;
+      }
+    } catch (error) {
+      loggers.intent.error("Error in intentFulfilling generator", { error });
+      if (error instanceof Error) {
+        yield {
+          event: "error",
+          data: { message: error.message || "An unknown error occurred." }
+        };
+      } else {
+        yield {
+          event: "error",
+          data: { message: "An unknown error occurred." }
+        };
+      }
+    }
+  }
+  /**
+   * Main entry point for processing user queries.
+   *
+   * Handles the complete query lifecycle:
+   * 1. Loads session history from memory
+   * 2. Detects intent from the query
+   * 3. Fulfills the intent with AI response
+   * 4. Updates conversation history
+   *
+   * @param query - The user's input query
+   * @param sessionId - Unique session identifier
+   * @returns Object containing the AI-generated response
+   */
+  async handleQueryStream(query, sessionId, res, userId) {
+    const queryStartAt = Date.now();
+    const sessionMemory = this.memoryModule?.getSessionMemory();
+    const session = !userId ? void 0 : await sessionMemory?.getSession(sessionId, userId);
+    const intent = this.intentTriggering(query);
+    try {
+      const stream = await this.intentFulfilling(
+        query,
+        sessionId,
+        session || { chats: {} }
+      );
+      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;
+        }
+        const sseFormattedEvent = `event: ${event.event}
+data: ${JSON.stringify(event.data)}
+
+`;
+        res.write(sseFormattedEvent);
+      }
+    } catch (error) {
+      loggers.intent.error("Error in handleQuery", { error });
+      res.write(
+        `event: error
+data: ${JSON.stringify({ message: "Stream failed" })}
+
+`
+      );
+    } finally {
+      res.end();
+    }
+  }
+};
+
+export {
+  QueryStreamService
+};
+//# sourceMappingURL=chunk-EFMGNCWG.js.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/services/query-stream.service.ts"],"sourcesContent":["import type { Response } from \"express\";\nimport type {\n\tA2AModule,\n\tMCPModule,\n\tMemoryModule,\n\tModelModule,\n} from \"@/modules/index.js\";\nimport type { AinAgentPrompts } from \"@/types/agent.js\";\nimport type { SessionObject } 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 a user query.\n\t *\n\t * @param query - The user's input query\n\t * @returns The detected intent (currently returns the query as-is)\n\t * @todo Implement actual intent detection logic\n\t */\n\tprivate async intentTriggering(query: string) {\n\t\t/* TODO */\n\t\treturn query;\n\t}\n\n\t/**\n\t * Fulfills the detected intent by generating a response.\n\t *\n\t * Manages the complete inference loop including:\n\t * - Loading prompts and conversation history\n\t * - Collecting available tools from modules\n\t * - Executing model inference with tool support\n\t * - Processing tool calls iteratively until completion\n\t *\n\t * @param query - The user's input query\n\t * @param sessionId - Session identifier for context\n\t * @param sessionHistory - 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\tsessionId: string,\n\t\tsessionHistory: SessionObject,\n\t): AsyncGenerator<StreamEvent> {\n\t\ttry {\n\t\t\tconst systemPrompt = `\nToday is ${new Date().toLocaleDateString()}.\n\n${this.prompts?.agent || \"\"}\n\n${this.prompts?.system || \"\"}\n    `;\n\n\t\t\tconst modelInstance = this.modelModule.getModel();\n\t\t\tconst messages = modelInstance.generateMessages({\n\t\t\t\tquery,\n\t\t\t\tsessionHistory,\n\t\t\t\tsystemPrompt: systemPrompt.trim(),\n\t\t\t});\n\n\t\t\tconst tools: IAgentTool[] = [];\n\t\t\tif (this.mcpModule) {\n\t\t\t\ttools.push(...this.mcpModule.getTools());\n\t\t\t}\n\t\t\tif (this.a2aModule) {\n\t\t\t\ttools.push(...(await this.a2aModule.getTools()));\n\t\t\t}\n\t\t\tconst functions = modelInstance.convertToolsToFunctions(tools);\n\n\t\t\tconst processList: string[] = [];\n\t\t\tconst finalMessage = \"\";\n\t\t\tlet didCallTool = false;\n\n\t\t\twhile (true) {\n\t\t\t\tconst responseStream =\n\t\t\t\t\tawait modelInstance.fetchStreamWithContextMessage(\n\t\t\t\t\t\tmessages,\n\t\t\t\t\t\tfunctions,\n\t\t\t\t\t);\n\t\t\t\tdidCallTool = false;\n\n\t\t\t\tconst assembledToolCalls: {\n\t\t\t\t\tid: string;\n\t\t\t\t\ttype: \"function\";\n\t\t\t\t\tfunction: { name: string; arguments: string };\n\t\t\t\t}[] = [];\n\n\t\t\t\tloggers.intentStream.debug(\"messages\", { messages });\n\n\t\t\t\tfor await (const chunk of responseStream) {\n\t\t\t\t\tconst delta = chunk.delta;\n\t\t\t\t\tif (delta?.tool_calls) {\n\t\t\t\t\t\tdidCallTool = true;\n\t\t\t\t\t\tfor (const toolCallDelta of delta.tool_calls) {\n\t\t\t\t\t\t\tconst index = toolCallDelta.index;\n\n\t\t\t\t\t\t\tif (!assembledToolCalls[index]) {\n\t\t\t\t\t\t\t\tassembledToolCalls[index] = {\n\t\t\t\t\t\t\t\t\tid: \"\",\n\t\t\t\t\t\t\t\t\ttype: \"function\",\n\t\t\t\t\t\t\t\t\tfunction: { name: \"\", arguments: \"\" },\n\t\t\t\t\t\t\t\t};\n\t\t\t\t\t\t\t}\n\n\t\t\t\t\t\t\tif (toolCallDelta.id) {\n\t\t\t\t\t\t\t\tassembledToolCalls[index].id = toolCallDelta.id;\n\t\t\t\t\t\t\t}\n\n\t\t\t\t\t\t\tif (toolCallDelta.function?.name) {\n\t\t\t\t\t\t\t\tassembledToolCalls[index].function.name =\n\t\t\t\t\t\t\t\t\ttoolCallDelta.function.name;\n\t\t\t\t\t\t\t}\n\n\t\t\t\t\t\t\tif (toolCallDelta.function?.arguments) {\n\t\t\t\t\t\t\t\tassembledToolCalls[index].function.arguments +=\n\t\t\t\t\t\t\t\t\ttoolCallDelta.function.arguments;\n\t\t\t\t\t\t\t}\n\t\t\t\t\t\t}\n\t\t\t\t\t} else if (chunk.delta?.content) {\n\t\t\t\t\t\tyield {\n\t\t\t\t\t\t\tevent: \"text_chunk\",\n\t\t\t\t\t\t\tdata: { delta: chunk.delta.content },\n\t\t\t\t\t\t};\n\t\t\t\t\t}\n\t\t\t\t}\n\n\t\t\t\tloggers.intentStream.debug(\"assembledToolCalls\", {\n\t\t\t\t\tassembledToolCalls,\n\t\t\t\t});\n\n\t\t\t\tif (didCallTool && assembledToolCalls.length > 0) {\n\t\t\t\t\tconst messagePayload = this.a2aModule?.getMessagePayload(\n\t\t\t\t\t\tquery,\n\t\t\t\t\t\tsessionId,\n\t\t\t\t\t);\n\t\t\t\t\tfor (const toolCall of assembledToolCalls) {\n\t\t\t\t\t\tconst toolName = toolCall.function.name;\n\t\t\t\t\t\tconst toolArgs = JSON.parse(toolCall.function.arguments);\n\t\t\t\t\t\tconst selectedTool = tools.filter(\n\t\t\t\t\t\t\t(tool) => tool.id === toolName,\n\t\t\t\t\t\t)[0];\n\n\t\t\t\t\t\tyield { event: \"tool_start\", data: { toolName, toolArgs } };\n\n\t\t\t\t\t\tlet toolResult = \"\";\n\n\t\t\t\t\t\tif (\n\t\t\t\t\t\t\tthis.mcpModule &&\n\t\t\t\t\t\t\tselectedTool.protocol === TOOL_PROTOCOL_TYPE.MCP\n\t\t\t\t\t\t) {\n\t\t\t\t\t\t\tconst toolArgs = JSON.parse(toolCall.function.arguments) as\n\t\t\t\t\t\t\t\t| { [x: string]: unknown }\n\t\t\t\t\t\t\t\t| undefined;\n\t\t\t\t\t\t\tloggers.intent.debug(\"MCP tool call\", { toolName, toolArgs });\n\t\t\t\t\t\t\tconst result = await this.mcpModule.useTool(\n\t\t\t\t\t\t\t\tselectedTool as IMCPTool,\n\t\t\t\t\t\t\t\ttoolArgs,\n\t\t\t\t\t\t\t);\n\t\t\t\t\t\t\ttoolResult =\n\t\t\t\t\t\t\t\t`[Bot Called MCP Tool ${toolName} with args ${JSON.stringify(toolArgs)}]\\n` +\n\t\t\t\t\t\t\t\tJSON.stringify(result.content, null, 2);\n\t\t\t\t\t\t} else if (\n\t\t\t\t\t\t\tthis.a2aModule &&\n\t\t\t\t\t\t\tselectedTool.protocol === TOOL_PROTOCOL_TYPE.A2A\n\t\t\t\t\t\t) {\n\t\t\t\t\t\t\tconst result = await this.a2aModule.useTool(\n\t\t\t\t\t\t\t\tselectedTool as IA2ATool,\n\t\t\t\t\t\t\t\tmessagePayload!,\n\t\t\t\t\t\t\t\tsessionId,\n\t\t\t\t\t\t\t);\n\t\t\t\t\t\t\ttoolResult = `[Bot Called A2A Tool ${toolName}]\\n${result.join(\"\\n\")}`;\n\t\t\t\t\t\t} else {\n\t\t\t\t\t\t\t// Unrecognized tool type. It cannot be happened...\n\t\t\t\t\t\t\tloggers.intent.warn(\n\t\t\t\t\t\t\t\t`Unrecognized tool type: ${selectedTool.protocol}`,\n\t\t\t\t\t\t\t);\n\t\t\t\t\t\t\tcontinue;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tyield {\n\t\t\t\t\t\t\tevent: \"tool_output\",\n\t\t\t\t\t\t\tdata: { toolName, result: toolResult },\n\t\t\t\t\t\t};\n\t\t\t\t\t\tloggers.intent.debug(\"toolResult\", { toolResult });\n\n\t\t\t\t\t\tprocessList.push(toolResult);\n\t\t\t\t\t\tmodelInstance.appendMessages(messages, toolResult);\n\t\t\t\t\t}\n\t\t\t\t}\n\n\t\t\t\tif (!didCallTool) break;\n\t\t\t}\n\t\t} catch (error) {\n\t\t\tloggers.intent.error(\"Error in intentFulfilling generator\", { error });\n\t\t\tif (error instanceof Error) {\n\t\t\t\tyield {\n\t\t\t\t\tevent: \"error\",\n\t\t\t\t\tdata: { message: error.message || \"An unknown error occurred.\" },\n\t\t\t\t};\n\t\t\t} else {\n\t\t\t\tyield {\n\t\t\t\t\tevent: \"error\",\n\t\t\t\t\tdata: { message: \"An unknown error occurred.\" },\n\t\t\t\t};\n\t\t\t}\n\t\t}\n\t}\n\t/**\n\t * Main entry point for processing user queries.\n\t *\n\t * Handles the complete query lifecycle:\n\t * 1. Loads session 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 query - The user's input query\n\t * @param sessionId - Unique session identifier\n\t * @returns Object containing the AI-generated response\n\t */\n\tpublic async handleQueryStream(\n\t\tquery: string,\n\t\tsessionId: string,\n\t\tres: Response,\n\t\tuserId?: string,\n\t) {\n\t\t// 1. Load session history with sessionId\n\t\tconst queryStartAt = Date.now();\n\t\tconst sessionMemory = this.memoryModule?.getSessionMemory();\n\t\tconst session = !userId\n\t\t\t? undefined\n\t\t\t: await sessionMemory?.getSession(sessionId, userId);\n\n\t\t// 2. intent triggering\n\t\tconst intent = this.intentTriggering(query);\n\n\t\ttry {\n\t\t\t// 3. intent fulfillment\n\t\t\tconst stream = await this.intentFulfilling(\n\t\t\t\tquery,\n\t\t\t\tsessionId,\n\t\t\t\tsession || { chats: {} },\n\t\t\t);\n\n\t\t\tlet finalResponseText = \"\";\n\t\t\tfor await (const event of stream) {\n\t\t\t\tif (event.event === \"text_chunk\" && event.data.delta) {\n\t\t\t\t\tloggers.intentStream.debug(\"text_chunk\", { event });\n\t\t\t\t\tfinalResponseText += event.data.delta;\n\t\t\t\t}\n\n\t\t\t\tconst sseFormattedEvent = `event: ${event.event}\\ndata: ${JSON.stringify(event.data)}\\n\\n`;\n\t\t\t\tres.write(sseFormattedEvent);\n\t\t\t}\n\t\t} catch (error) {\n\t\t\tloggers.intent.error(\"Error in handleQuery\", { error });\n\t\t\tres.write(\n\t\t\t\t`event: error\\ndata: ${JSON.stringify({ message: \"Stream failed\" })}\\n\\n`,\n\t\t\t);\n\t\t} finally {\n\t\t\tres.end();\n\t\t}\n\t}\n}\n"],"mappings":";;;;;AAyBO,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,iBAAiB,OAAe;AAE7C,WAAO;AAAA,EACR;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAgBA,OAAc,iBACb,OACA,WACA,gBAC8B;AAC9B,QAAI;AACH,YAAM,eAAe;AAAA,YACb,oBAAI,KAAK,GAAE,mBAAmB,CAAC;AAAA;AAAA,EAExC,KAAK,SAAS,SAAS,EAAE;AAAA;AAAA,EAEzB,KAAK,SAAS,UAAU,EAAE;AAAA;AAGzB,YAAM,gBAAgB,KAAK,YAAY,SAAS;AAChD,YAAM,WAAW,cAAc,iBAAiB;AAAA,QAC/C;AAAA,QACA;AAAA,QACA,cAAc,aAAa,KAAK;AAAA,MACjC,CAAC;AAED,YAAM,QAAsB,CAAC;AAC7B,UAAI,KAAK,WAAW;AACnB,cAAM,KAAK,GAAG,KAAK,UAAU,SAAS,CAAC;AAAA,MACxC;AACA,UAAI,KAAK,WAAW;AACnB,cAAM,KAAK,GAAI,MAAM,KAAK,UAAU,SAAS,CAAE;AAAA,MAChD;AACA,YAAM,YAAY,cAAc,wBAAwB,KAAK;AAE7D,YAAM,cAAwB,CAAC;AAC/B,YAAM,eAAe;AACrB,UAAI,cAAc;AAElB,aAAO,MAAM;AACZ,cAAM,iBACL,MAAM,cAAc;AAAA,UACnB;AAAA,UACA;AAAA,QACD;AACD,sBAAc;AAEd,cAAM,qBAIA,CAAC;AAEP,gBAAQ,aAAa,MAAM,YAAY,EAAE,SAAS,CAAC;AAEnD,yBAAiB,SAAS,gBAAgB;AACzC,gBAAM,QAAQ,MAAM;AACpB,cAAI,OAAO,YAAY;AACtB,0BAAc;AACd,uBAAW,iBAAiB,MAAM,YAAY;AAC7C,oBAAM,QAAQ,cAAc;AAE5B,kBAAI,CAAC,mBAAmB,KAAK,GAAG;AAC/B,mCAAmB,KAAK,IAAI;AAAA,kBAC3B,IAAI;AAAA,kBACJ,MAAM;AAAA,kBACN,UAAU,EAAE,MAAM,IAAI,WAAW,GAAG;AAAA,gBACrC;AAAA,cACD;AAEA,kBAAI,cAAc,IAAI;AACrB,mCAAmB,KAAK,EAAE,KAAK,cAAc;AAAA,cAC9C;AAEA,kBAAI,cAAc,UAAU,MAAM;AACjC,mCAAmB,KAAK,EAAE,SAAS,OAClC,cAAc,SAAS;AAAA,cACzB;AAEA,kBAAI,cAAc,UAAU,WAAW;AACtC,mCAAmB,KAAK,EAAE,SAAS,aAClC,cAAc,SAAS;AAAA,cACzB;AAAA,YACD;AAAA,UACD,WAAW,MAAM,OAAO,SAAS;AAChC,kBAAM;AAAA,cACL,OAAO;AAAA,cACP,MAAM,EAAE,OAAO,MAAM,MAAM,QAAQ;AAAA,YACpC;AAAA,UACD;AAAA,QACD;AAEA,gBAAQ,aAAa,MAAM,sBAAsB;AAAA,UAChD;AAAA,QACD,CAAC;AAED,YAAI,eAAe,mBAAmB,SAAS,GAAG;AACjD,gBAAM,iBAAiB,KAAK,WAAW;AAAA,YACtC;AAAA,YACA;AAAA,UACD;AACA,qBAAW,YAAY,oBAAoB;AAC1C,kBAAM,WAAW,SAAS,SAAS;AACnC,kBAAM,WAAW,KAAK,MAAM,SAAS,SAAS,SAAS;AACvD,kBAAM,eAAe,MAAM;AAAA,cAC1B,CAAC,SAAS,KAAK,OAAO;AAAA,YACvB,EAAE,CAAC;AAEH,kBAAM,EAAE,OAAO,cAAc,MAAM,EAAE,UAAU,SAAS,EAAE;AAE1D,gBAAI,aAAa;AAEjB,gBACC,KAAK,aACL,aAAa,8BACZ;AACD,oBAAMA,YAAW,KAAK,MAAM,SAAS,SAAS,SAAS;AAGvD,sBAAQ,OAAO,MAAM,iBAAiB,EAAE,UAAU,UAAAA,UAAS,CAAC;AAC5D,oBAAM,SAAS,MAAM,KAAK,UAAU;AAAA,gBACnC;AAAA,gBACAA;AAAA,cACD;AACA,2BACC,wBAAwB,QAAQ,cAAc,KAAK,UAAUA,SAAQ,CAAC;AAAA,IACtE,KAAK,UAAU,OAAO,SAAS,MAAM,CAAC;AAAA,YACxC,WACC,KAAK,aACL,aAAa,8BACZ;AACD,oBAAM,SAAS,MAAM,KAAK,UAAU;AAAA,gBACnC;AAAA,gBACA;AAAA,gBACA;AAAA,cACD;AACA,2BAAa,wBAAwB,QAAQ;AAAA,EAAM,OAAO,KAAK,IAAI,CAAC;AAAA,YACrE,OAAO;AAEN,sBAAQ,OAAO;AAAA,gBACd,2BAA2B,aAAa,QAAQ;AAAA,cACjD;AACA;AAAA,YACD;AACA,kBAAM;AAAA,cACL,OAAO;AAAA,cACP,MAAM,EAAE,UAAU,QAAQ,WAAW;AAAA,YACtC;AACA,oBAAQ,OAAO,MAAM,cAAc,EAAE,WAAW,CAAC;AAEjD,wBAAY,KAAK,UAAU;AAC3B,0BAAc,eAAe,UAAU,UAAU;AAAA,UAClD;AAAA,QACD;AAEA,YAAI,CAAC,YAAa;AAAA,MACnB;AAAA,IACD,SAAS,OAAO;AACf,cAAQ,OAAO,MAAM,uCAAuC,EAAE,MAAM,CAAC;AACrE,UAAI,iBAAiB,OAAO;AAC3B,cAAM;AAAA,UACL,OAAO;AAAA,UACP,MAAM,EAAE,SAAS,MAAM,WAAW,6BAA6B;AAAA,QAChE;AAAA,MACD,OAAO;AACN,cAAM;AAAA,UACL,OAAO;AAAA,UACP,MAAM,EAAE,SAAS,6BAA6B;AAAA,QAC/C;AAAA,MACD;AAAA,IACD;AAAA,EACD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAcA,MAAa,kBACZ,OACA,WACA,KACA,QACC;AAED,UAAM,eAAe,KAAK,IAAI;AAC9B,UAAM,gBAAgB,KAAK,cAAc,iBAAiB;AAC1D,UAAM,UAAU,CAAC,SACd,SACA,MAAM,eAAe,WAAW,WAAW,MAAM;AAGpD,UAAM,SAAS,KAAK,iBAAiB,KAAK;AAE1C,QAAI;AAEH,YAAM,SAAS,MAAM,KAAK;AAAA,QACzB;AAAA,QACA;AAAA,QACA,WAAW,EAAE,OAAO,CAAC,EAAE;AAAA,MACxB;AAEA,UAAI,oBAAoB;AACxB,uBAAiB,SAAS,QAAQ;AACjC,YAAI,MAAM,UAAU,gBAAgB,MAAM,KAAK,OAAO;AACrD,kBAAQ,aAAa,MAAM,cAAc,EAAE,MAAM,CAAC;AAClD,+BAAqB,MAAM,KAAK;AAAA,QACjC;AAEA,cAAM,oBAAoB,UAAU,MAAM,KAAK;AAAA,QAAW,KAAK,UAAU,MAAM,IAAI,CAAC;AAAA;AAAA;AACpF,YAAI,MAAM,iBAAiB;AAAA,MAC5B;AAAA,IACD,SAAS,OAAO;AACf,cAAQ,OAAO,MAAM,wBAAwB,EAAE,MAAM,CAAC;AACtD,UAAI;AAAA,QACH;AAAA,QAAuB,KAAK,UAAU,EAAE,SAAS,gBAAgB,CAAC,CAAC;AAAA;AAAA;AAAA,MACpE;AAAA,IACD,UAAE;AACD,UAAI,IAAI;AAAA,IACT;AAAA,EACD;AACD;","names":["toolArgs"]}

package/dist/esm/{chunk-EODMRV4H.js → chunk-HAE333YO.js}

@@ -1,6 +1,6 @@
 import {
   loggers
-} from "./chunk-64GEXWSE.js";
+} from "./chunk-N7VYM2R5.js";
 
 // src/controllers/a2a.controller.ts
 import {
@@ -108,4 +108,4 @@ var A2AController = class {
 export {
   A2AController
 };
-//# sourceMappingURL=chunk-EODMRV4H.js.map
+//# sourceMappingURL=chunk-HAE333YO.js.map

package/dist/esm/{chunk-ILFWNK3W.js → chunk-MAU2MAK2.js}

@@ -1,6 +1,6 @@
 import {
   logger
-} from "./chunk-64GEXWSE.js";
+} from "./chunk-N7VYM2R5.js";
 
 // src/middlewares/error.middleware.ts
 var errorMiddleware = (error, req, res, next) => {
@@ -19,4 +19,4 @@ var errorMiddleware = (error, req, res, next) => {
 export {
   errorMiddleware
 };
-//# sourceMappingURL=chunk-ILFWNK3W.js.map
+//# sourceMappingURL=chunk-MAU2MAK2.js.map

package/dist/esm/{chunk-356AV44C.js → chunk-MC7CPQ6W.js}

@@ -1,6 +1,6 @@
 import {
   loggers
-} from "./chunk-64GEXWSE.js";
+} from "./chunk-N7VYM2R5.js";
 
 // src/services/query.service.ts
 var QueryService = class {
@@ -141,28 +141,20 @@ ${result.join("\n")}`;
   async handleQuery(query, sessionId, userId) {
     const queryStartAt = Date.now();
     const sessionMemory = this.memoryModule?.getSessionMemory();
-    const session = !userId ? void 0 : await sessionMemory?.getSession(sessionId, userId);
+    const session = !userId ? void 0 : await sessionMemory?.getSession(userId, sessionId);
     const intent = this.intentTriggering(query);
     const result = await this.intentFulfilling(query, sessionId, session);
     if (userId) {
-      await sessionMemory?.addChatToSession(
-        sessionId,
-        {
-          role: "USER" /* USER */,
-          timestamp: queryStartAt,
-          content: { type: "text", parts: [query] }
-        },
-        userId
-      );
-      await sessionMemory?.addChatToSession(
-        sessionId,
-        {
-          role: "MODEL" /* MODEL */,
-          timestamp: Date.now(),
-          content: { type: "text", parts: [result.response] }
-        },
-        userId
-      );
+      await sessionMemory?.addChatToSession(userId, sessionId, {
+        role: "USER" /* USER */,
+        timestamp: queryStartAt,
+        content: { type: "text", parts: [query] }
+      });
+      await sessionMemory?.addChatToSession(userId, sessionId, {
+        role: "MODEL" /* MODEL */,
+        timestamp: Date.now(),
+        content: { type: "text", parts: [result.response] }
+      });
     }
     return { content: result.response };
   }
@@ -171,4 +163,4 @@ ${result.join("\n")}`;
 export {
   QueryService
 };
-//# sourceMappingURL=chunk-356AV44C.js.map
+//# sourceMappingURL=chunk-MC7CPQ6W.js.map

package/dist/esm/{chunk-356AV44C.js.map → chunk-MC7CPQ6W.js.map}

@@ -1 +1 @@
-{"version":3,"sources":["../../src/services/query.service.ts"],"sourcesContent":["import type {\n\tA2AModule,\n\tMCPModule,\n\tMemoryModule,\n\tModelModule,\n} from \"@/modules/index.js\";\nimport type { AinAgentPrompts } from \"@/types/agent.js\";\nimport { ChatRole, type SessionObject } from \"@/types/memory.js\";\nimport {\n\ttype IA2ATool,\n\ttype IAgentTool,\n\ttype IMCPTool,\n\tTOOL_PROTOCOL_TYPE,\n} from \"@/types/tool.js\";\nimport { loggers } from \"@/utils/logger.js\";\n\n/**\n * Service for processing user queries through the agent's AI pipeline.\n *\n * Orchestrates the query processing workflow including intent detection,\n * model inference, tool execution, and response generation. Manages\n * conversation context and coordinates between different modules.\n */\nexport class QueryService {\n\tprivate modelModule: ModelModule;\n\tprivate a2aModule?: A2AModule;\n\tprivate mcpModule?: MCPModule;\n\tprivate memoryModule?: MemoryModule;\n\tprivate prompts?: AinAgentPrompts;\n\n\tconstructor(\n\t\tmodelModule: ModelModule,\n\t\ta2aModule?: A2AModule,\n\t\tmcpModule?: MCPModule,\n\t\tmemoryModule?: MemoryModule,\n\t\tprompts?: AinAgentPrompts,\n\t) {\n\t\tthis.modelModule = modelModule;\n\t\tthis.a2aModule = a2aModule;\n\t\tthis.mcpModule = mcpModule;\n\t\tthis.memoryModule = memoryModule;\n\t\tthis.prompts = prompts;\n\t}\n\n\t/**\n\t * Detects the intent from a user query.\n\t *\n\t * @param query - The user's input query\n\t * @returns The detected intent (currently returns the query as-is)\n\t * @todo Implement actual intent detection logic\n\t */\n\tprivate async intentTriggering(query: string) {\n\t\t/* TODO */\n\t\treturn query;\n\t}\n\n\t/**\n\t * Fulfills the detected intent by generating a response.\n\t *\n\t * Manages the complete inference loop including:\n\t * - Loading prompts and conversation history\n\t * - Collecting available tools from modules\n\t * - Executing model inference with tool support\n\t * - Processing tool calls iteratively until completion\n\t *\n\t * @param query - The user's input query\n\t * @param sessionId - Session identifier for context\n\t * @param sessionHistory - 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\tsessionId: string,\n\t\tsessionHistory: SessionObject | undefined,\n\t) {\n\t\t// 1. Load agent / system prompt from memory\n\t\tconst systemPrompt = `\nToday is ${new Date().toLocaleDateString()}.\n\n${this.prompts?.agent || \"\"}\n\n${this.prompts?.system || \"\"}\n    `;\n\n\t\tconst modelInstance = this.modelModule.getModel();\n\t\tconst messages = modelInstance.generateMessages({\n\t\t\tquery,\n\t\t\tsessionHistory,\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\tsessionId,\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\tconst result = 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\ttoolResult =\n\t\t\t\t\t\t\t`[Bot Called MCP Tool ${toolName} with args ${JSON.stringify(toolArgs)}]\\n` +\n\t\t\t\t\t\t\tJSON.stringify(result.content, null, 2);\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\tconst result = 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\tsessionId,\n\t\t\t\t\t\t);\n\t\t\t\t\t\ttoolResult = `[Bot Called A2A Tool ${toolName}]\\n${result.join(\"\\n\")}`;\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 * Main entry point for processing user queries.\n\t *\n\t * Handles the complete query lifecycle:\n\t * 1. Loads session 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 query - The user's input query\n\t * @param sessionId - Unique session identifier\n\t * @param userId - Unique user identifier\n\t * @returns Object containing the AI-generated response\n\t */\n\tpublic async handleQuery(query: string, sessionId: string, userId?: string) {\n\t\t// 1. Load session history with sessionId\n\t\tconst queryStartAt = Date.now();\n\t\tconst sessionMemory = this.memoryModule?.getSessionMemory();\n\t\tconst session = !userId\n\t\t\t? undefined\n\t\t\t: await sessionMemory?.getSession(sessionId, userId);\n\n\t\t// 2. intent triggering\n\t\tconst intent = this.intentTriggering(query);\n\n\t\t// 3. intent fulfillment\n\t\tconst result = await this.intentFulfilling(query, sessionId, session);\n\t\tif (userId) {\n\t\t\tawait sessionMemory?.addChatToSession(\n\t\t\t\tsessionId,\n\t\t\t\t{\n\t\t\t\t\trole: ChatRole.USER,\n\t\t\t\t\ttimestamp: queryStartAt,\n\t\t\t\t\tcontent: { type: \"text\", parts: [query] },\n\t\t\t\t},\n\t\t\t\tuserId,\n\t\t\t);\n\t\t\tawait sessionMemory?.addChatToSession(\n\t\t\t\tsessionId,\n\t\t\t\t{\n\t\t\t\t\trole: ChatRole.MODEL,\n\t\t\t\t\ttimestamp: Date.now(),\n\t\t\t\t\tcontent: { type: \"text\", parts: [result.response] },\n\t\t\t\t},\n\t\t\t\tuserId,\n\t\t\t);\n\t\t}\n\n\t\treturn { content: result.response };\n\t}\n}\n"],"mappings":";;;;;AAuBO,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,iBAAiB,OAAe;AAE7C,WAAO;AAAA,EACR;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAgBA,MAAc,iBACb,OACA,WACA,gBACC;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;AAG1B,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,kBAAM,SAAS,MAAM,KAAK,UAAU;AAAA,cACnC;AAAA,cACA;AAAA,YACD;AACA,yBACC,wBAAwB,QAAQ,cAAc,KAAK,UAAU,QAAQ,CAAC;AAAA,IACtE,KAAK,UAAU,OAAO,SAAS,MAAM,CAAC;AAAA,UACxC,WACC,KAAK,aACL,aAAa,8BACZ;AACD,kBAAM,SAAS,MAAM,KAAK,UAAU;AAAA,cACnC;AAAA,cACA;AAAA,cACA;AAAA,YACD;AACA,yBAAa,wBAAwB,QAAQ;AAAA,EAAM,OAAO,KAAK,IAAI,CAAC;AAAA,UACrE,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;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAgBA,MAAa,YAAY,OAAe,WAAmB,QAAiB;AAE3E,UAAM,eAAe,KAAK,IAAI;AAC9B,UAAM,gBAAgB,KAAK,cAAc,iBAAiB;AAC1D,UAAM,UAAU,CAAC,SACd,SACA,MAAM,eAAe,WAAW,WAAW,MAAM;AAGpD,UAAM,SAAS,KAAK,iBAAiB,KAAK;AAG1C,UAAM,SAAS,MAAM,KAAK,iBAAiB,OAAO,WAAW,OAAO;AACpE,QAAI,QAAQ;AACX,YAAM,eAAe;AAAA,QACpB;AAAA,QACA;AAAA,UACC;AAAA,UACA,WAAW;AAAA,UACX,SAAS,EAAE,MAAM,QAAQ,OAAO,CAAC,KAAK,EAAE;AAAA,QACzC;AAAA,QACA;AAAA,MACD;AACA,YAAM,eAAe;AAAA,QACpB;AAAA,QACA;AAAA,UACC;AAAA,UACA,WAAW,KAAK,IAAI;AAAA,UACpB,SAAS,EAAE,MAAM,QAAQ,OAAO,CAAC,OAAO,QAAQ,EAAE;AAAA,QACnD;AAAA,QACA;AAAA,MACD;AAAA,IACD;AAEA,WAAO,EAAE,SAAS,OAAO,SAAS;AAAA,EACnC;AACD;","names":[]}
+{"version":3,"sources":["../../src/services/query.service.ts"],"sourcesContent":["import type {\n\tA2AModule,\n\tMCPModule,\n\tMemoryModule,\n\tModelModule,\n} from \"@/modules/index.js\";\nimport type { AinAgentPrompts } from \"@/types/agent.js\";\nimport { ChatRole, type SessionObject } from \"@/types/memory.js\";\nimport {\n\ttype IA2ATool,\n\ttype IAgentTool,\n\ttype IMCPTool,\n\tTOOL_PROTOCOL_TYPE,\n} from \"@/types/tool.js\";\nimport { loggers } from \"@/utils/logger.js\";\n\n/**\n * Service for processing user queries through the agent's AI pipeline.\n *\n * Orchestrates the query processing workflow including intent detection,\n * model inference, tool execution, and response generation. Manages\n * conversation context and coordinates between different modules.\n */\nexport class QueryService {\n\tprivate modelModule: ModelModule;\n\tprivate a2aModule?: A2AModule;\n\tprivate mcpModule?: MCPModule;\n\tprivate memoryModule?: MemoryModule;\n\tprivate prompts?: AinAgentPrompts;\n\n\tconstructor(\n\t\tmodelModule: ModelModule,\n\t\ta2aModule?: A2AModule,\n\t\tmcpModule?: MCPModule,\n\t\tmemoryModule?: MemoryModule,\n\t\tprompts?: AinAgentPrompts,\n\t) {\n\t\tthis.modelModule = modelModule;\n\t\tthis.a2aModule = a2aModule;\n\t\tthis.mcpModule = mcpModule;\n\t\tthis.memoryModule = memoryModule;\n\t\tthis.prompts = prompts;\n\t}\n\n\t/**\n\t * Detects the intent from a user query.\n\t *\n\t * @param query - The user's input query\n\t * @returns The detected intent (currently returns the query as-is)\n\t * @todo Implement actual intent detection logic\n\t */\n\tprivate async intentTriggering(query: string) {\n\t\t/* TODO */\n\t\treturn query;\n\t}\n\n\t/**\n\t * Fulfills the detected intent by generating a response.\n\t *\n\t * Manages the complete inference loop including:\n\t * - Loading prompts and conversation history\n\t * - Collecting available tools from modules\n\t * - Executing model inference with tool support\n\t * - Processing tool calls iteratively until completion\n\t *\n\t * @param query - The user's input query\n\t * @param sessionId - Session identifier for context\n\t * @param sessionHistory - 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\tsessionId: string,\n\t\tsessionHistory: SessionObject | undefined,\n\t) {\n\t\t// 1. Load agent / system prompt from memory\n\t\tconst systemPrompt = `\nToday is ${new Date().toLocaleDateString()}.\n\n${this.prompts?.agent || \"\"}\n\n${this.prompts?.system || \"\"}\n    `;\n\n\t\tconst modelInstance = this.modelModule.getModel();\n\t\tconst messages = modelInstance.generateMessages({\n\t\t\tquery,\n\t\t\tsessionHistory,\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\tsessionId,\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\tconst result = 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\ttoolResult =\n\t\t\t\t\t\t\t`[Bot Called MCP Tool ${toolName} with args ${JSON.stringify(toolArgs)}]\\n` +\n\t\t\t\t\t\t\tJSON.stringify(result.content, null, 2);\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\tconst result = 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\tsessionId,\n\t\t\t\t\t\t);\n\t\t\t\t\t\ttoolResult = `[Bot Called A2A Tool ${toolName}]\\n${result.join(\"\\n\")}`;\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 * Main entry point for processing user queries.\n\t *\n\t * Handles the complete query lifecycle:\n\t * 1. Loads session 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 query - The user's input query\n\t * @param sessionId - Unique session identifier\n\t * @param userId - Unique user identifier\n\t * @returns Object containing the AI-generated response\n\t */\n\tpublic async handleQuery(query: string, sessionId: string, userId?: string) {\n\t\t// 1. Load session history with sessionId\n\t\tconst queryStartAt = Date.now();\n\t\tconst sessionMemory = this.memoryModule?.getSessionMemory();\n\t\tconst session = !userId\n\t\t\t? undefined\n\t\t\t: await sessionMemory?.getSession(userId, sessionId);\n\n\t\t// 2. intent triggering\n\t\tconst intent = this.intentTriggering(query);\n\n\t\t// 3. intent fulfillment\n\t\tconst result = await this.intentFulfilling(query, sessionId, session);\n\t\tif (userId) {\n\t\t\tawait sessionMemory?.addChatToSession(userId, sessionId, {\n\t\t\t\trole: ChatRole.USER,\n\t\t\t\ttimestamp: queryStartAt,\n\t\t\t\tcontent: { type: \"text\", parts: [query] },\n\t\t\t});\n\t\t\tawait sessionMemory?.addChatToSession(userId, sessionId, {\n\t\t\t\trole: ChatRole.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":";;;;;AAuBO,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,iBAAiB,OAAe;AAE7C,WAAO;AAAA,EACR;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAgBA,MAAc,iBACb,OACA,WACA,gBACC;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;AAG1B,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,kBAAM,SAAS,MAAM,KAAK,UAAU;AAAA,cACnC;AAAA,cACA;AAAA,YACD;AACA,yBACC,wBAAwB,QAAQ,cAAc,KAAK,UAAU,QAAQ,CAAC;AAAA,IACtE,KAAK,UAAU,OAAO,SAAS,MAAM,CAAC;AAAA,UACxC,WACC,KAAK,aACL,aAAa,8BACZ;AACD,kBAAM,SAAS,MAAM,KAAK,UAAU;AAAA,cACnC;AAAA,cACA;AAAA,cACA;AAAA,YACD;AACA,yBAAa,wBAAwB,QAAQ;AAAA,EAAM,OAAO,KAAK,IAAI,CAAC;AAAA,UACrE,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;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAgBA,MAAa,YAAY,OAAe,WAAmB,QAAiB;AAE3E,UAAM,eAAe,KAAK,IAAI;AAC9B,UAAM,gBAAgB,KAAK,cAAc,iBAAiB;AAC1D,UAAM,UAAU,CAAC,SACd,SACA,MAAM,eAAe,WAAW,QAAQ,SAAS;AAGpD,UAAM,SAAS,KAAK,iBAAiB,KAAK;AAG1C,UAAM,SAAS,MAAM,KAAK,iBAAiB,OAAO,WAAW,OAAO;AACpE,QAAI,QAAQ;AACX,YAAM,eAAe,iBAAiB,QAAQ,WAAW;AAAA,QACxD;AAAA,QACA,WAAW;AAAA,QACX,SAAS,EAAE,MAAM,QAAQ,OAAO,CAAC,KAAK,EAAE;AAAA,MACzC,CAAC;AACD,YAAM,eAAe,iBAAiB,QAAQ,WAAW;AAAA,QACxD;AAAA,QACA,WAAW,KAAK,IAAI;AAAA,QACpB,SAAS,EAAE,MAAM,QAAQ,OAAO,CAAC,OAAO,QAAQ,EAAE;AAAA,MACnD,CAAC;AAAA,IACF;AAEA,WAAO,EAAE,SAAS,OAAO,SAAS;AAAA,EACnC;AACD;","names":[]}

package/dist/esm/{chunk-64GEXWSE.js → chunk-N7VYM2R5.js}

@@ -25,6 +25,7 @@ var getLogger = (service) => createLogger(service);
 var loggers = {
   agent: getLogger("AINAgent"),
   intent: getLogger("Intent"),
+  intentStream: getLogger("IntentStream"),
   mcp: getLogger("MCPModule"),
   a2a: getLogger("A2AModule"),
   model: getLogger("Model"),
@@ -37,4 +38,4 @@ export {
   getLogger,
   loggers
 };
-//# sourceMappingURL=chunk-64GEXWSE.js.map
+//# sourceMappingURL=chunk-N7VYM2R5.js.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/utils/logger.ts"],"sourcesContent":["import winston from \"winston\";\n\nconst { combine, timestamp, printf, colorize, errors } = winston.format;\n\nconst createLogger = (service: string) => {\n\tconst logger = winston.createLogger({\n\t\tlevel: (process.env.LOG_LEVEL || \"info\").toLowerCase(),\n\t\tformat: combine(\n\t\t\terrors({ stack: true }),\n\t\t\tcolorize(),\n\t\t\ttimestamp({ format: \"HH:mm:ss\" }),\n\t\t\tprintf(({ level, message, timestamp, service, stack, ...meta }) => {\n\t\t\t\tconst metaStr = Object.keys(meta).length\n\t\t\t\t\t? ` | ${JSON.stringify(meta)}`\n\t\t\t\t\t: \"\";\n\t\t\t\tconst errorStack = stack ? `\\n${stack}` : \"\";\n\t\t\t\treturn `${timestamp} [${service}] ${level}: ${message}${metaStr}${errorStack}`;\n\t\t\t}),\n\t\t),\n\t\tdefaultMeta: { service },\n\t\ttransports: [new winston.transports.Console()],\n\t});\n\n\treturn logger;\n};\n\n// Default logger instance\nexport const logger = createLogger(\"ain-adk\");\n\n// Factory function for creating service-specific loggers\nexport const getLogger = (service: string) => createLogger(service);\n\n// Convenience methods for different components\nexport const loggers = {\n\tagent: getLogger(\"AINAgent\"),\n\tintent: getLogger(\"Intent\"),\n\tintentStream: getLogger(\"IntentStream\"),\n\tmcp: getLogger(\"MCPModule\"),\n\ta2a: getLogger(\"A2AModule\"),\n\tmodel: getLogger(\"Model\"),\n\tserver: getLogger(\"A2AServer\"),\n\tfol: getLogger(\"FOL\"),\n} as const;\n"],"mappings":";AAAA,OAAO,aAAa;AAEpB,IAAM,EAAE,SAAS,WAAW,QAAQ,UAAU,OAAO,IAAI,QAAQ;AAEjE,IAAM,eAAe,CAAC,YAAoB;AACzC,QAAMA,UAAS,QAAQ,aAAa;AAAA,IACnC,QAAQ,QAAQ,IAAI,aAAa,QAAQ,YAAY;AAAA,IACrD,QAAQ;AAAA,MACP,OAAO,EAAE,OAAO,KAAK,CAAC;AAAA,MACtB,SAAS;AAAA,MACT,UAAU,EAAE,QAAQ,WAAW,CAAC;AAAA,MAChC,OAAO,CAAC,EAAE,OAAO,SAAS,WAAAC,YAAW,SAAAC,UAAS,OAAO,GAAG,KAAK,MAAM;AAClE,cAAM,UAAU,OAAO,KAAK,IAAI,EAAE,SAC/B,MAAM,KAAK,UAAU,IAAI,CAAC,KAC1B;AACH,cAAM,aAAa,QAAQ;AAAA,EAAK,KAAK,KAAK;AAC1C,eAAO,GAAGD,UAAS,KAAKC,QAAO,KAAK,KAAK,KAAK,OAAO,GAAG,OAAO,GAAG,UAAU;AAAA,MAC7E,CAAC;AAAA,IACF;AAAA,IACA,aAAa,EAAE,QAAQ;AAAA,IACvB,YAAY,CAAC,IAAI,QAAQ,WAAW,QAAQ,CAAC;AAAA,EAC9C,CAAC;AAED,SAAOF;AACR;AAGO,IAAM,SAAS,aAAa,SAAS;AAGrC,IAAM,YAAY,CAAC,YAAoB,aAAa,OAAO;AAG3D,IAAM,UAAU;AAAA,EACtB,OAAO,UAAU,UAAU;AAAA,EAC3B,QAAQ,UAAU,QAAQ;AAAA,EAC1B,cAAc,UAAU,cAAc;AAAA,EACtC,KAAK,UAAU,WAAW;AAAA,EAC1B,KAAK,UAAU,WAAW;AAAA,EAC1B,OAAO,UAAU,OAAO;AAAA,EACxB,QAAQ,UAAU,WAAW;AAAA,EAC7B,KAAK,UAAU,KAAK;AACrB;","names":["logger","timestamp","service"]}

package/dist/esm/{chunk-6BK7LDIA.js → chunk-PN6NMQBM.js}

@@ -40,4 +40,4 @@ var AuthMiddleware = class {
 export {
   AuthMiddleware
 };
-//# sourceMappingURL=chunk-6BK7LDIA.js.map
+//# sourceMappingURL=chunk-PN6NMQBM.js.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/middlewares/auth.middleware.ts"],"sourcesContent":["import type { NextFunction, Request, RequestHandler, Response } from \"express\";\nimport { StatusCodes } from \"http-status-codes\";\nimport type { BaseAuth } from \"@/modules/auth/base.auth\";\nimport { AinHttpError } from \"@/types/agent\";\nimport type { AuthResponse } from \"@/types/auth\";\nimport { loggers } from \"@/utils/logger\";\n\nexport class AuthMiddleware {\n\tprivate auth: BaseAuth;\n\tconstructor(auth: BaseAuth) {\n\t\tthis.auth = auth;\n\t}\n\n\tpublic middleware(): RequestHandler {\n\t\treturn async (req: Request, res: Response, next: NextFunction) => {\n\t\t\ttry {\n\t\t\t\tconst authRes: AuthResponse = await this.auth.authenticate(req, res);\n\t\t\t\tif (authRes.isAuthenticated) {\n\t\t\t\t\tres.locals.userId = authRes.userId;\n\t\t\t\t\tnext();\n\t\t\t\t} else {\n\t\t\t\t\tconst error: AinHttpError = new AinHttpError(\n\t\t\t\t\t\tStatusCodes.UNAUTHORIZED,\n\t\t\t\t\t\t\"Unauthorized\",\n\t\t\t\t\t);\n\t\t\t\t\tthrow error;\n\t\t\t\t}\n\t\t\t} catch (e: any) {\n\t\t\t\tif (!e.status) {\n\t\t\t\t\tconst error: AinHttpError = new AinHttpError(\n\t\t\t\t\t\tStatusCodes.INTERNAL_SERVER_ERROR,\n\t\t\t\t\t\t`Authentication error: ${JSON.stringify(e)}`,\n\t\t\t\t\t);\n\t\t\t\t\tthrow error;\n\t\t\t\t}\n\t\t\t\tthrow e;\n\t\t\t}\n\t\t};\n\t}\n}\n"],"mappings":";;;;;AACA,SAAS,mBAAmB;AAMrB,IAAM,iBAAN,MAAqB;AAAA,EACnB;AAAA,EACR,YAAY,MAAgB;AAC3B,SAAK,OAAO;AAAA,EACb;AAAA,EAEO,aAA6B;AACnC,WAAO,OAAO,KAAc,KAAe,SAAuB;AACjE,UAAI;AACH,cAAM,UAAwB,MAAM,KAAK,KAAK,aAAa,KAAK,GAAG;AACnE,YAAI,QAAQ,iBAAiB;AAC5B,cAAI,OAAO,SAAS,QAAQ;AAC5B,eAAK;AAAA,QACN,OAAO;AACN,gBAAM,QAAsB,IAAI;AAAA,YAC/B,YAAY;AAAA,YACZ;AAAA,UACD;AACA,gBAAM;AAAA,QACP;AAAA,MACD,SAAS,GAAQ;AAChB,YAAI,CAAC,EAAE,QAAQ;AACd,gBAAM,QAAsB,IAAI;AAAA,YAC/B,YAAY;AAAA,YACZ,yBAAyB,KAAK,UAAU,CAAC,CAAC;AAAA,UAC3C;AACA,gBAAM;AAAA,QACP;AACA,cAAM;AAAA,MACP;AAAA,IACD;AAAA,EACD;AACD;","names":[]}

package/dist/esm/{chunk-RSSLUA2N.js → chunk-QROUPWWO.js}

@@ -1,6 +1,6 @@
 import {
   SessionApiController
-} from "./chunk-RB4IFZJA.js";
+} from "./chunk-5EPK37PE.js";
 
 // src/routes/api/sessions.routes.ts
 import { Router } from "express";
@@ -14,4 +14,4 @@ var createSessionApiRouter = (memoryModule) => {
 export {
   createSessionApiRouter
 };
-//# sourceMappingURL=chunk-RSSLUA2N.js.map
+//# sourceMappingURL=chunk-QROUPWWO.js.map

package/dist/esm/{chunk-F727XG4N.js → chunk-VAZWF5IK.js}

@@ -17,4 +17,4 @@ export {
   MemoryType,
   ChatRole
 };
-//# sourceMappingURL=chunk-F727XG4N.js.map
+//# sourceMappingURL=chunk-VAZWF5IK.js.map

package/dist/esm/{chunk-F727XG4N.js.map → chunk-VAZWF5IK.js.map}

@@ -1 +1 @@
-{"version":3,"sources":["../../src/types/memory.ts"],"sourcesContent":["export enum MemoryType {\n\tSESSION = \"SESSION\",\n\tINTENT = \"INTENT\",\n\tAGENT = \"AGENT\",\n\tUSER = \"USER\",\n}\n\n/**\n * Roles for participants in a chat conversation.\n */\nexport enum ChatRole {\n\t/** User/human participant */\n\tUSER = \"USER\",\n\t/** System-generated messages or instructions */\n\tSYSTEM = \"SYSTEM\",\n\t/** AI model responses */\n\tMODEL = \"MODEL\",\n}\n\n/**\n * Content structure for chat messages.\n *\n * Supports multi-part content with different types (text, images, etc.).\n */\nexport type ChatContentObject = {\n\t/** Content type (e.g., \"text\", \"image\", \"tool_use\") */\n\ttype: string;\n\t/** Array of content parts, structure depends on content type */\n\tparts: any[];\n};\n\n/**\n * Represents a single message in a chat conversation.\n *\n * @example\n * ```typescript\n * const message: ChatObject = {\n *   role: ChatRole.USER,\n *   content: {\n *     type: \"text\",\n *     parts: [\"Hello, how can you help me?\"]\n *   },\n *   timestamp: Date.now(),\n *   metadata: { source: \"web-ui\" }\n * };\n * ```\n */\nexport type ChatObject = {\n\t/** Role of the message sender */\n\trole: ChatRole;\n\t/** Message content with type and parts */\n\tcontent: ChatContentObject;\n\t/** Unix timestamp when the message was created */\n\ttimestamp: number;\n\t/** Optional metadata for additional context */\n\tmetadata?: { [key: string]: unknown };\n};\n\n/**\n * Represents a conversation session containing multiple chat messages.\n *\n * Messages are stored in a key-value structure where keys are unique chat IDs\n * and values are the corresponding chat objects.\n *\n * @example\n * ```typescript\n * const session: SessionObject = {\n *   chats: {\n *     \"<UUID_1>\": { role: ChatRole.USER, content: {...}, timestamp: 1234567890 },\n *     \"<UUID_2>\": { role: ChatRole.MODEL, content: {...}, timestamp: 1234567891 }\n *   }\n * };\n * ```\n */\nexport type SessionObject = {\n\t/** Collection of chat messages indexed by unique chat ID */\n\tchats: {\n\t\t[chatId: string]: ChatObject;\n\t};\n};\n\nexport interface Intent {\n\tname: string;\n\tdescription: string;\n}\n"],"mappings":";AAAO,IAAK,aAAL,kBAAKA,gBAAL;AACN,EAAAA,YAAA,aAAU;AACV,EAAAA,YAAA,YAAS;AACT,EAAAA,YAAA,WAAQ;AACR,EAAAA,YAAA,UAAO;AAJI,SAAAA;AAAA,GAAA;AAUL,IAAK,WAAL,kBAAKC,cAAL;AAEN,EAAAA,UAAA,UAAO;AAEP,EAAAA,UAAA,YAAS;AAET,EAAAA,UAAA,WAAQ;AANG,SAAAA;AAAA,GAAA;","names":["MemoryType","ChatRole"]}
+{"version":3,"sources":["../../src/types/memory.ts"],"sourcesContent":["export enum MemoryType {\n\tSESSION = \"SESSION\",\n\tINTENT = \"INTENT\",\n\tAGENT = \"AGENT\",\n\tUSER = \"USER\",\n}\n\n/**\n * Roles for participants in a chat conversation.\n */\nexport enum ChatRole {\n\t/** User/human participant */\n\tUSER = \"USER\",\n\t/** System-generated messages or instructions */\n\tSYSTEM = \"SYSTEM\",\n\t/** AI model responses */\n\tMODEL = \"MODEL\",\n}\n\n/**\n * Content structure for chat messages.\n *\n * Supports multi-part content with different types (text, images, etc.).\n */\nexport type ChatContentObject = {\n\t/** Content type (e.g., \"text\", \"image\", \"tool_use\") */\n\ttype: string;\n\t/** Array of content parts, structure depends on content type */\n\tparts: any[];\n};\n\n/**\n * Represents a single message in a chat conversation.\n *\n * @example\n * ```typescript\n * const message: ChatObject = {\n *   role: ChatRole.USER,\n *   content: {\n *     type: \"text\",\n *     parts: [\"Hello, how can you help me?\"]\n *   },\n *   timestamp: Date.now(),\n *   metadata: { source: \"web-ui\" }\n * };\n * ```\n */\nexport type ChatObject = {\n\t/** Role of the message sender */\n\trole: ChatRole;\n\t/** Message content with type and parts */\n\tcontent: ChatContentObject;\n\t/** Unix timestamp when the message was created */\n\ttimestamp: number;\n\t/** Optional metadata for additional context */\n\tmetadata?: { [key: string]: unknown };\n};\n\nexport type SessionMetadata = {\n\ttitle?: string;\n\tsessionId: string;\n\tupdatedAt: number;\n};\n\n/**\n * Represents a conversation session containing multiple chat messages.\n *\n * Messages are stored in a key-value structure where keys are unique chat IDs\n * and values are the corresponding chat objects.\n *\n * @example\n * ```typescript\n * const session: SessionObject = {\n *   chats: {\n *     \"<UUID_1>\": { role: ChatRole.USER, content: {...}, timestamp: 1234567890 },\n *     \"<UUID_2>\": { role: ChatRole.MODEL, content: {...}, timestamp: 1234567891 }\n *   }\n * };\n * ```\n */\nexport type SessionObject = {\n\ttitle?: string;\n\t/** Collection of chat messages indexed by unique chat ID */\n\tchats: {\n\t\t[chatId: string]: ChatObject;\n\t};\n};\n\nexport interface Intent {\n\tname: string;\n\tdescription: string;\n\tprompt?: string;\n\tllm?: string;\n}\n"],"mappings":";AAAO,IAAK,aAAL,kBAAKA,gBAAL;AACN,EAAAA,YAAA,aAAU;AACV,EAAAA,YAAA,YAAS;AACT,EAAAA,YAAA,WAAQ;AACR,EAAAA,YAAA,UAAO;AAJI,SAAAA;AAAA,GAAA;AAUL,IAAK,WAAL,kBAAKC,cAAL;AAEN,EAAAA,UAAA,UAAO;AAEP,EAAAA,UAAA,YAAS;AAET,EAAAA,UAAA,WAAQ;AANG,SAAAA;AAAA,GAAA;","names":["MemoryType","ChatRole"]}

package/dist/esm/chunk-WGHAZ4HG.js

@@ -0,0 +1,44 @@
+// src/controllers/query.controller.ts
+var QueryController = class {
+  queryService;
+  queryStreamService;
+  constructor(queryService, queryStreamService) {
+    this.queryService = queryService;
+    this.queryStreamService = queryStreamService;
+  }
+  handleQueryRequest = async (req, res, next) => {
+    try {
+      const { message, sessionId } = req.body;
+      const userId = res.locals.userId;
+      const result = await this.queryService.handleQuery(
+        message,
+        sessionId,
+        userId
+      );
+      res.status(200).json(result);
+    } catch (error) {
+      next(error);
+    }
+  };
+  handleQueryStreamRequest = async (req, res, next) => {
+    const { message, sessionId, userId } = req.body;
+    if (!this.queryStreamService) {
+      throw new Error("This Agent does not support stream query");
+    }
+    try {
+      await this.queryStreamService.handleQueryStream(
+        message,
+        sessionId,
+        res,
+        userId
+      );
+    } catch (error) {
+      next(error);
+    }
+  };
+};
+
+export {
+  QueryController
+};
+//# sourceMappingURL=chunk-WGHAZ4HG.js.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/controllers/query.controller.ts"],"sourcesContent":["import type { NextFunction, Request, Response } from \"express\";\nimport type { QueryService, QueryStreamService } from \"@/services\";\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\ttry {\n\t\t\tconst { message, sessionId } = req.body;\n\t\t\tconst userId = res.locals.userId;\n\n\t\t\tconst result = await this.queryService.handleQuery(\n\t\t\t\tmessage,\n\t\t\t\tsessionId,\n\t\t\t\tuserId,\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 { message, sessionId, userId } = req.body;\n\n\t\tif (!this.queryStreamService) {\n\t\t\tthrow new Error(\"This Agent does not support stream query\");\n\t\t}\n\t\ttry {\n\t\t\tawait this.queryStreamService.handleQueryStream(\n\t\t\t\tmessage,\n\t\t\t\tsessionId,\n\t\t\t\tres,\n\t\t\t\tuserId,\n\t\t\t);\n\t\t} catch (error) {\n\t\t\tnext(error);\n\t\t}\n\t};\n}\n"],"mappings":";AAGO,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,QAAI;AACH,YAAM,EAAE,SAAS,UAAU,IAAI,IAAI;AACnC,YAAM,SAAS,IAAI,OAAO;AAE1B,YAAM,SAAS,MAAM,KAAK,aAAa;AAAA,QACtC;AAAA,QACA;AAAA,QACA;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,SAAS,WAAW,OAAO,IAAI,IAAI;AAE3C,QAAI,CAAC,KAAK,oBAAoB;AAC7B,YAAM,IAAI,MAAM,0CAA0C;AAAA,IAC3D;AACA,QAAI;AACH,YAAM,KAAK,mBAAmB;AAAA,QAC7B;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,MACD;AAAA,IACD,SAAS,OAAO;AACf,WAAK,KAAK;AAAA,IACX;AAAA,EACD;AACD;","names":[]}

package/dist/esm/{chunk-AQ5YZERE.js → chunk-Y6E73I7I.js}

@@ -1,6 +1,6 @@
 import {
   loggers
-} from "./chunk-64GEXWSE.js";
+} from "./chunk-N7VYM2R5.js";
 
 // src/services/a2a.service.ts
 import { randomUUID } from "crypto";
@@ -108,4 +108,4 @@ var A2AService = class {
 export {
   A2AService
 };
-//# sourceMappingURL=chunk-AQ5YZERE.js.map
+//# sourceMappingURL=chunk-Y6E73I7I.js.map

package/dist/esm/{chunk-NSKXMJWE.js → chunk-YT6PD6UF.js}

@@ -1,9 +1,9 @@
-import {
-  createSessionApiRouter
-} from "./chunk-RSSLUA2N.js";
 import {
   createModelApiRouter
 } from "./chunk-2Z7GJVJJ.js";
+import {
+  createSessionApiRouter
+} from "./chunk-QROUPWWO.js";
 
 // src/routes/api.routes.ts
 import { Router } from "express";
@@ -19,4 +19,4 @@ var createApiRouter = (agent) => {
 export {
   createApiRouter
 };
-//# sourceMappingURL=chunk-NSKXMJWE.js.map
+//# sourceMappingURL=chunk-YT6PD6UF.js.map

package/dist/esm/chunk-ZFSIP5KN.js

@@ -0,0 +1,43 @@
+import {
+  QueryStreamService
+} from "./chunk-EFMGNCWG.js";
+import {
+  QueryController
+} from "./chunk-WGHAZ4HG.js";
+import {
+  QueryService
+} from "./chunk-MC7CPQ6W.js";
+
+// src/routes/query.routes.ts
+import { Router } from "express";
+var createQueryRouter = (agent, allowStream = false) => {
+  const router = Router();
+  const queryService = new QueryService(
+    agent.modelModule,
+    agent.a2aModule,
+    agent.mcpModule,
+    agent.memoryModule,
+    agent.manifest.prompts
+  );
+  let queryStreamService;
+  if (allowStream) {
+    queryStreamService = new QueryStreamService(
+      agent.modelModule,
+      agent.a2aModule,
+      agent.mcpModule,
+      agent.memoryModule,
+      agent.manifest.prompts
+    );
+  }
+  const queryController = new QueryController(queryService, queryStreamService);
+  router.post("/query", queryController.handleQueryRequest);
+  if (allowStream) {
+    router.post("/query-stream", queryController.handleQueryStreamRequest);
+  }
+  return router;
+};
+
+export {
+  createQueryRouter
+};
+//# sourceMappingURL=chunk-ZFSIP5KN.js.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/routes/query.routes.ts"],"sourcesContent":["import { Router } from \"express\";\nimport { QueryController } from \"@/controllers/query.controller.js\";\nimport type { AINAgent } from \"@/index.js\";\nimport { QueryService } from \"@/services/query.service.js\";\nimport { QueryStreamService } from \"@/services/query-stream.service.js\";\n\nexport const createQueryRouter = (\n\tagent: AINAgent,\n\tallowStream = false,\n): Router => {\n\tconst router = Router();\n\n\tconst queryService = new QueryService(\n\t\tagent.modelModule,\n\t\tagent.a2aModule,\n\t\tagent.mcpModule,\n\t\tagent.memoryModule,\n\t\tagent.manifest.prompts,\n\t);\n\n\tlet queryStreamService: QueryStreamService | undefined;\n\n\tif (allowStream) {\n\t\tqueryStreamService = new QueryStreamService(\n\t\t\tagent.modelModule,\n\t\t\tagent.a2aModule,\n\t\t\tagent.mcpModule,\n\t\t\tagent.memoryModule,\n\t\t\tagent.manifest.prompts,\n\t\t);\n\t}\n\n\tconst queryController = new QueryController(queryService, queryStreamService);\n\trouter.post(\"/query\", queryController.handleQueryRequest);\n\n\tif (allowStream) {\n\t\trouter.post(\"/query-stream\", queryController.handleQueryStreamRequest);\n\t}\n\n\treturn router;\n};\n"],"mappings":";;;;;;;;;;;AAAA,SAAS,cAAc;AAMhB,IAAM,oBAAoB,CAChC,OACA,cAAc,UACF;AACZ,QAAM,SAAS,OAAO;AAEtB,QAAM,eAAe,IAAI;AAAA,IACxB,MAAM;AAAA,IACN,MAAM;AAAA,IACN,MAAM;AAAA,IACN,MAAM;AAAA,IACN,MAAM,SAAS;AAAA,EAChB;AAEA,MAAI;AAEJ,MAAI,aAAa;AAChB,yBAAqB,IAAI;AAAA,MACxB,MAAM;AAAA,MACN,MAAM;AAAA,MACN,MAAM;AAAA,MACN,MAAM;AAAA,MACN,MAAM,SAAS;AAAA,IAChB;AAAA,EACD;AAEA,QAAM,kBAAkB,IAAI,gBAAgB,cAAc,kBAAkB;AAC5E,SAAO,KAAK,UAAU,gBAAgB,kBAAkB;AAExD,MAAI,aAAa;AAChB,WAAO,KAAK,iBAAiB,gBAAgB,wBAAwB;AAAA,EACtE;AAEA,SAAO;AACR;","names":[]}

package/dist/esm/controllers/a2a.controller.d.ts

@@ -17,6 +17,7 @@ import '../modules/memory/base.memory.js';
 import '../types/memory.js';
 import '../modules/models/model.module.js';
 import '../modules/models/base.model.js';
+import '../types/stream.js';
 import '../types/agent.js';
 import 'http-status-codes';