@ainetwork/adk 0.2.7 → 0.2.8

package/dist/esm/chunk-EYZ5Z7FG.js

@@ -0,0 +1,156 @@
+import {
+  loggers
+} from "./chunk-N7VYM2R5.js";
+
+// src/services/intents/fulfill.service.ts
+import { randomUUID } from "crypto";
+var IntentFulfillService = 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;
+  }
+  async addToThreadMessages(thread, params) {
+    try {
+      const threadMemory = this.memoryModule?.getThreadMemory();
+      const { userId, threadId } = thread;
+      const newMessage = {
+        messageId: randomUUID(),
+        role: params.role,
+        timestamp: Date.now(),
+        content: { type: "text", parts: [params.content] },
+        metadata: params.metadata
+      };
+      thread.messages.push(newMessage);
+      await threadMemory?.addMessagesToThread(userId, threadId, [newMessage]);
+    } catch (error) {
+      loggers.intentStream.error("Error adding message to thread", error);
+    }
+  }
+  /**
+   * Fulfills the detected intent by generating a streaming response.
+   *
+   * Manages the complete inference loop including:
+   * - Loading prompts and conversation history
+   * - Collecting available tools from modules
+   * - Executing model inference with tool support
+   * - Processing tool calls iteratively until completion
+   * - Streaming results as Server-Sent Events
+   *
+   * @param query - The user's input query
+   * @param threadId - Thread identifier for context
+   * @param thread - Previous conversation history
+   * @param intent - Optional detected intent with custom prompt
+   * @returns AsyncGenerator yielding StreamEvent objects
+   */
+  async intentFulfilling(query, thread, intent) {
+    const systemPrompt = `
+Today is ${(/* @__PURE__ */ new Date()).toLocaleDateString()}.
+
+${this.prompts?.agent || ""}
+
+${this.prompts?.system || ""}
+
+${intent?.prompt || ""}
+	`.trim();
+    const modelInstance = this.modelModule.getModel();
+    const messages = modelInstance.generateMessages({
+      query,
+      thread,
+      systemPrompt: systemPrompt.trim()
+    });
+    const tools = [];
+    this.mcpModule && tools.push(...this.mcpModule.getTools());
+    this.a2aModule && tools.push(...await this.a2aModule.getTools());
+    const functions = modelInstance.convertToolsToFunctions(tools);
+    let finalMessage = "";
+    while (true) {
+      const response = await modelInstance.fetchWithContextMessage(
+        messages,
+        functions
+      );
+      loggers.intentStream.info("messages", { messages });
+      const { content, toolCalls } = response;
+      if (toolCalls) {
+        for (const toolCall of toolCalls) {
+          const toolName = toolCall.name;
+          const selectedTool = tools.filter(
+            (tool) => tool.toolName === toolName
+          )[0];
+          let toolResult = "";
+          if (this.mcpModule && selectedTool.protocol === "MCP" /* MCP */) {
+            const toolArgs = toolCall.arguments;
+            loggers.intent.debug("MCP tool call", { toolName, toolArgs });
+            toolResult = await this.mcpModule.useTool(selectedTool, toolArgs);
+          } else if (this.a2aModule && selectedTool.protocol === "A2A" /* A2A */) {
+            toolResult = await this.a2aModule.useTool(
+              selectedTool,
+              query,
+              thread.threadId
+            );
+          } else {
+            loggers.intent.warn(
+              `Unrecognized tool type: ${selectedTool.protocol}`
+            );
+            continue;
+          }
+          loggers.intent.debug("toolResult", { toolResult });
+          modelInstance.appendMessages(messages, toolResult);
+        }
+      } else if (content) {
+        finalMessage = content;
+        break;
+      }
+    }
+    return finalMessage;
+  }
+  /**
+   * Detects the intent from context.
+   *
+   * @param intents - The user's input query
+   * @param thread - The thread history
+   * @returns The detected intent
+   */
+  async intentFulfill(intents, thread) {
+    let finalResponseText = "";
+    for (let i = 0; i < intents.length; i++) {
+      const { subquery, intent, actionPlan } = intents[i];
+      loggers.intent.info(`Process query: ${subquery}, ${intent?.name}`);
+      loggers.intent.info(`Action plan: ${actionPlan}`);
+      finalResponseText !== "" && thread.messages.push({
+        messageId: randomUUID(),
+        role: "MODEL" /* MODEL */,
+        timestamp: Date.now(),
+        content: { type: "text", parts: [finalResponseText] },
+        metadata: { isThinking: true }
+      });
+      await this.addToThreadMessages(thread, {
+        role: "MODEL" /* MODEL */,
+        content: subquery,
+        metadata: {
+          subquery,
+          isThinking: true,
+          actionPlan
+        }
+      });
+      finalResponseText = await this.intentFulfilling(subquery, thread, intent);
+    }
+    await this.addToThreadMessages(thread, {
+      role: "MODEL" /* MODEL */,
+      content: finalResponseText
+    });
+    return finalResponseText;
+  }
+};
+
+export {
+  IntentFulfillService
+};
+//# sourceMappingURL=chunk-EYZ5Z7FG.js.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/services/intents/fulfill.service.ts"],"sourcesContent":["import { randomUUID } from \"node:crypto\";\nimport type {\n\tA2AModule,\n\tMCPModule,\n\tMemoryModule,\n\tModelModule,\n} from \"@/modules\";\nimport type { AinAgentPrompts } from \"@/types/agent\";\nimport { CONNECTOR_PROTOCOL_TYPE, type ConnectorTool } from \"@/types/connector\";\nimport {\n\ttype Intent,\n\ttype MessageObject,\n\tMessageRole,\n\ttype ThreadObject,\n\ttype TriggeredIntent,\n} from \"@/types/memory\";\nimport { StreamEvent } from \"@/types/stream\";\nimport { loggers } from \"@/utils/logger\";\n\nexport class IntentFulfillService {\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\tprivate async addToThreadMessages(\n\t\tthread: ThreadObject,\n\t\tparams: { role: MessageRole; content: string; metadata?: any },\n\t) {\n\t\ttry {\n\t\t\tconst threadMemory = this.memoryModule?.getThreadMemory();\n\t\t\tconst { userId, threadId } = thread;\n\t\t\tconst newMessage: MessageObject = {\n\t\t\t\tmessageId: randomUUID(),\n\t\t\t\trole: params.role,\n\t\t\t\ttimestamp: Date.now(),\n\t\t\t\tcontent: { type: \"text\", parts: [params.content] },\n\t\t\t\tmetadata: params.metadata,\n\t\t\t};\n\t\t\tthread.messages.push(newMessage);\n\t\t\tawait threadMemory?.addMessagesToThread(userId, threadId, [newMessage]);\n\t\t} catch (error) {\n\t\t\tloggers.intentStream.error(\"Error adding message to thread\", error);\n\t\t}\n\t}\n\n\t/**\n\t * Fulfills the detected intent by generating a streaming response.\n\t *\n\t * Manages the complete inference loop including:\n\t * - Loading prompts and conversation history\n\t * - Collecting available tools from modules\n\t * - Executing model inference with tool support\n\t * - Processing tool calls iteratively until completion\n\t * - Streaming results as Server-Sent Events\n\t *\n\t * @param query - The user's input query\n\t * @param threadId - Thread identifier for context\n\t * @param thread - Previous conversation history\n\t * @param intent - Optional detected intent with custom prompt\n\t * @returns AsyncGenerator yielding StreamEvent objects\n\t */\n\tpublic async intentFulfilling(\n\t\tquery: string,\n\t\tthread: ThreadObject,\n\t\tintent?: Intent,\n\t): Promise<string> {\n\t\tconst systemPrompt = `\nToday is ${new Date().toLocaleDateString()}.\n\n${this.prompts?.agent || \"\"}\n\n${this.prompts?.system || \"\"}\n\n${intent?.prompt || \"\"}\n\t`.trim();\n\n\t\tconst modelInstance = this.modelModule.getModel();\n\t\tconst messages = modelInstance.generateMessages({\n\t\t\tquery,\n\t\t\tthread,\n\t\t\tsystemPrompt: systemPrompt.trim(),\n\t\t});\n\n\t\tconst tools: ConnectorTool[] = [];\n\t\tthis.mcpModule && tools.push(...this.mcpModule.getTools());\n\t\tthis.a2aModule && tools.push(...(await this.a2aModule.getTools()));\n\n\t\tconst functions = modelInstance.convertToolsToFunctions(tools);\n\t\tlet finalMessage = \"\";\n\n\t\twhile (true) {\n\t\t\tconst response = await modelInstance.fetchWithContextMessage(\n\t\t\t\tmessages,\n\t\t\t\tfunctions,\n\t\t\t);\n\n\t\t\tloggers.intentStream.info(\"messages\", { messages });\n\n\t\t\tconst { content, toolCalls } = response;\n\n\t\t\tif (toolCalls) {\n\t\t\t\tfor (const toolCall of toolCalls) {\n\t\t\t\t\tconst toolName = toolCall.name;\n\t\t\t\t\tconst selectedTool = tools.filter(\n\t\t\t\t\t\t(tool) => tool.toolName === toolName,\n\t\t\t\t\t)[0];\n\n\t\t\t\t\tlet toolResult = \"\";\n\t\t\t\t\tif (\n\t\t\t\t\t\tthis.mcpModule &&\n\t\t\t\t\t\tselectedTool.protocol === CONNECTOR_PROTOCOL_TYPE.MCP\n\t\t\t\t\t) {\n\t\t\t\t\t\tconst toolArgs = toolCall.arguments as\n\t\t\t\t\t\t\t| { [x: string]: unknown }\n\t\t\t\t\t\t\t| undefined;\n\t\t\t\t\t\tloggers.intent.debug(\"MCP tool call\", { toolName, toolArgs });\n\t\t\t\t\t\ttoolResult = await this.mcpModule.useTool(selectedTool, toolArgs);\n\t\t\t\t\t} else if (\n\t\t\t\t\t\tthis.a2aModule &&\n\t\t\t\t\t\tselectedTool.protocol === CONNECTOR_PROTOCOL_TYPE.A2A\n\t\t\t\t\t) {\n\t\t\t\t\t\ttoolResult = await this.a2aModule.useTool(\n\t\t\t\t\t\t\tselectedTool,\n\t\t\t\t\t\t\tquery,\n\t\t\t\t\t\t\tthread.threadId,\n\t\t\t\t\t\t);\n\t\t\t\t\t} else {\n\t\t\t\t\t\t// Unrecognized tool type. It cannot be happened...\n\t\t\t\t\t\tloggers.intent.warn(\n\t\t\t\t\t\t\t`Unrecognized tool type: ${selectedTool.protocol}`,\n\t\t\t\t\t\t);\n\t\t\t\t\t\tcontinue;\n\t\t\t\t\t}\n\n\t\t\t\t\tloggers.intent.debug(\"toolResult\", { toolResult });\n\n\t\t\t\t\tmodelInstance.appendMessages(messages, toolResult);\n\t\t\t\t}\n\t\t\t} else if (content) {\n\t\t\t\tfinalMessage = content;\n\t\t\t\tbreak;\n\t\t\t}\n\t\t}\n\n\t\treturn finalMessage;\n\t}\n\n\t/**\n\t * Detects the intent from context.\n\t *\n\t * @param intents - The user's input query\n\t * @param thread - The thread history\n\t * @returns The detected intent\n\t */\n\tpublic async intentFulfill(\n\t\tintents: Array<TriggeredIntent>,\n\t\tthread: ThreadObject,\n\t): Promise<string> {\n\t\tlet finalResponseText = \"\";\n\t\tfor (let i = 0; i < intents.length; i++) {\n\t\t\tconst { subquery, intent, actionPlan } = intents[i];\n\t\t\tloggers.intent.info(`Process query: ${subquery}, ${intent?.name}`);\n\t\t\tloggers.intent.info(`Action plan: ${actionPlan}`);\n\n\t\t\t// only use for inference, not stored in memory\n\t\t\tfinalResponseText !== \"\" &&\n\t\t\t\tthread.messages.push({\n\t\t\t\t\tmessageId: randomUUID(),\n\t\t\t\t\trole: MessageRole.MODEL,\n\t\t\t\t\ttimestamp: Date.now(),\n\t\t\t\t\tcontent: { type: \"text\", parts: [finalResponseText] },\n\t\t\t\t\tmetadata: { isThinking: true },\n\t\t\t\t});\n\t\t\tawait this.addToThreadMessages(thread, {\n\t\t\t\trole: MessageRole.MODEL,\n\t\t\t\tcontent: subquery,\n\t\t\t\tmetadata: {\n\t\t\t\t\tsubquery,\n\t\t\t\t\tisThinking: true,\n\t\t\t\t\tactionPlan: actionPlan,\n\t\t\t\t},\n\t\t\t});\n\n\t\t\tfinalResponseText = await this.intentFulfilling(subquery, thread, intent);\n\t\t}\n\n\t\tawait this.addToThreadMessages(thread, {\n\t\t\trole: MessageRole.MODEL,\n\t\t\tcontent: finalResponseText,\n\t\t});\n\n\t\treturn finalResponseText;\n\t}\n}\n"],"mappings":";;;;;AAAA,SAAS,kBAAkB;AAmBpB,IAAM,uBAAN,MAA2B;AAAA,EACzB;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,EAEA,MAAc,oBACb,QACA,QACC;AACD,QAAI;AACH,YAAM,eAAe,KAAK,cAAc,gBAAgB;AACxD,YAAM,EAAE,QAAQ,SAAS,IAAI;AAC7B,YAAM,aAA4B;AAAA,QACjC,WAAW,WAAW;AAAA,QACtB,MAAM,OAAO;AAAA,QACb,WAAW,KAAK,IAAI;AAAA,QACpB,SAAS,EAAE,MAAM,QAAQ,OAAO,CAAC,OAAO,OAAO,EAAE;AAAA,QACjD,UAAU,OAAO;AAAA,MAClB;AACA,aAAO,SAAS,KAAK,UAAU;AAC/B,YAAM,cAAc,oBAAoB,QAAQ,UAAU,CAAC,UAAU,CAAC;AAAA,IACvE,SAAS,OAAO;AACf,cAAQ,aAAa,MAAM,kCAAkC,KAAK;AAAA,IACnE;AAAA,EACD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAkBA,MAAa,iBACZ,OACA,QACA,QACkB;AAClB,UAAM,eAAe;AAAA,YACZ,oBAAI,KAAK,GAAE,mBAAmB,CAAC;AAAA;AAAA,EAExC,KAAK,SAAS,SAAS,EAAE;AAAA;AAAA,EAEzB,KAAK,SAAS,UAAU,EAAE;AAAA;AAAA,EAE1B,QAAQ,UAAU,EAAE;AAAA,GACnB,KAAK;AAEN,UAAM,gBAAgB,KAAK,YAAY,SAAS;AAChD,UAAM,WAAW,cAAc,iBAAiB;AAAA,MAC/C;AAAA,MACA;AAAA,MACA,cAAc,aAAa,KAAK;AAAA,IACjC,CAAC;AAED,UAAM,QAAyB,CAAC;AAChC,SAAK,aAAa,MAAM,KAAK,GAAG,KAAK,UAAU,SAAS,CAAC;AACzD,SAAK,aAAa,MAAM,KAAK,GAAI,MAAM,KAAK,UAAU,SAAS,CAAE;AAEjE,UAAM,YAAY,cAAc,wBAAwB,KAAK;AAC7D,QAAI,eAAe;AAEnB,WAAO,MAAM;AACZ,YAAM,WAAW,MAAM,cAAc;AAAA,QACpC;AAAA,QACA;AAAA,MACD;AAEA,cAAQ,aAAa,KAAK,YAAY,EAAE,SAAS,CAAC;AAElD,YAAM,EAAE,SAAS,UAAU,IAAI;AAE/B,UAAI,WAAW;AACd,mBAAW,YAAY,WAAW;AACjC,gBAAM,WAAW,SAAS;AAC1B,gBAAM,eAAe,MAAM;AAAA,YAC1B,CAAC,SAAS,KAAK,aAAa;AAAA,UAC7B,EAAE,CAAC;AAEH,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,QAAQ,cAAc,QAAQ;AAAA,UACjE,WACC,KAAK,aACL,aAAa,8BACZ;AACD,yBAAa,MAAM,KAAK,UAAU;AAAA,cACjC;AAAA,cACA;AAAA,cACA,OAAO;AAAA,YACR;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,wBAAc,eAAe,UAAU,UAAU;AAAA,QAClD;AAAA,MACD,WAAW,SAAS;AACnB,uBAAe;AACf;AAAA,MACD;AAAA,IACD;AAEA,WAAO;AAAA,EACR;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EASA,MAAa,cACZ,SACA,QACkB;AAClB,QAAI,oBAAoB;AACxB,aAAS,IAAI,GAAG,IAAI,QAAQ,QAAQ,KAAK;AACxC,YAAM,EAAE,UAAU,QAAQ,WAAW,IAAI,QAAQ,CAAC;AAClD,cAAQ,OAAO,KAAK,kBAAkB,QAAQ,KAAK,QAAQ,IAAI,EAAE;AACjE,cAAQ,OAAO,KAAK,gBAAgB,UAAU,EAAE;AAGhD,4BAAsB,MACrB,OAAO,SAAS,KAAK;AAAA,QACpB,WAAW,WAAW;AAAA,QACtB;AAAA,QACA,WAAW,KAAK,IAAI;AAAA,QACpB,SAAS,EAAE,MAAM,QAAQ,OAAO,CAAC,iBAAiB,EAAE;AAAA,QACpD,UAAU,EAAE,YAAY,KAAK;AAAA,MAC9B,CAAC;AACF,YAAM,KAAK,oBAAoB,QAAQ;AAAA,QACtC;AAAA,QACA,SAAS;AAAA,QACT,UAAU;AAAA,UACT;AAAA,UACA,YAAY;AAAA,UACZ;AAAA,QACD;AAAA,MACD,CAAC;AAED,0BAAoB,MAAM,KAAK,iBAAiB,UAAU,QAAQ,MAAM;AAAA,IACzE;AAEA,UAAM,KAAK,oBAAoB,QAAQ;AAAA,MACtC;AAAA,MACA,SAAS;AAAA,IACV,CAAC;AAED,WAAO;AAAA,EACR;AACD;","names":[]}

package/dist/esm/chunk-IRGR2OHS.js

@@ -0,0 +1,123 @@
+import {
+  IntentFulfillService
+} from "./chunk-EYZ5Z7FG.js";
+import {
+  IntentTriggerService
+} from "./chunk-OCONOO5Z.js";
+import {
+  AinHttpError
+} from "./chunk-ZBUAT73P.js";
+import {
+  loggers
+} from "./chunk-N7VYM2R5.js";
+
+// src/services/query.service.ts
+import { randomUUID } from "crypto";
+import { StatusCodes } from "http-status-codes";
+var QueryService = class {
+  modelModule;
+  memoryModule;
+  intentTriggerService;
+  intentFulfillService;
+  constructor(modelModule, a2aModule, mcpModule, memoryModule, prompts) {
+    this.modelModule = modelModule;
+    this.memoryModule = memoryModule;
+    this.intentTriggerService = new IntentTriggerService(
+      modelModule,
+      memoryModule
+    );
+    this.intentFulfillService = new IntentFulfillService(
+      modelModule,
+      a2aModule,
+      mcpModule,
+      memoryModule,
+      prompts
+    );
+  }
+  /**
+   * Generates a title for the conversation based on the query.
+   *
+   * @param query - The user's input query
+   * @returns Promise resolving to a generated title
+   */
+  async generateTitle(query) {
+    const DEFAULT_TITLE = "New Chat";
+    try {
+      const modelInstance = this.modelModule.getModel();
+      const messages = modelInstance.generateMessages({
+        query,
+        systemPrompt: `You are a helpful assistant that generates titles for conversations.
+  Please analyze the user's query and create a concise title that accurately reflects the conversation's core topic.
+  The title must be no more than 5 words long.
+  Respond with only the title. Do not include any punctuation or extra explanations.`
+      });
+      const response = await modelInstance.fetch(messages);
+      return response.content || DEFAULT_TITLE;
+    } catch (error) {
+      loggers.intent.error("Error generating title", {
+        error,
+        query
+      });
+      return DEFAULT_TITLE;
+    }
+  }
+  /**
+   * Main entry point for processing user queries.
+   *
+   * Handles the complete query lifecycle:
+   * 1. Loads or creates thread from memory
+   * 2. Detects intent from the query
+   * 3. Fulfills the intent with AI response
+   * 4. Updates conversation history
+   *
+   * @param threadMetadata - Metadata containing type, userId, and optional threadId
+   * @param threadMetadata.type - The type of thread (e.g., chat, workflow)
+   * @param threadMetadata.userId - The user's unique identifier
+   * @param threadMetadata.threadId - Optional thread identifier
+   * @param query - The user's input query
+   * @returns Promise resolving to object with content property containing the response
+   */
+  async handleQuery(threadMetadata, query, isA2A) {
+    const { type, userId } = threadMetadata;
+    const threadMemory = this.memoryModule?.getThreadMemory();
+    let threadId = threadMetadata.threadId;
+    let thread;
+    if (threadId) {
+      thread = await threadMemory?.getThread(userId, threadId);
+      if (!thread && !isA2A) {
+        throw new AinHttpError(StatusCodes.NOT_FOUND, "Thread not found");
+      }
+    }
+    threadId ??= randomUUID();
+    if (!thread) {
+      const title = await this.generateTitle(query);
+      const metadata = await threadMemory?.createThread(
+        type,
+        userId,
+        threadId,
+        title
+      ) || { type, userId, threadId, title };
+      thread = { ...metadata, messages: [] };
+      loggers.intent.info(`Create new thread: ${threadId}`);
+    }
+    await threadMemory?.addMessagesToThread(userId, threadId, [
+      {
+        messageId: randomUUID(),
+        role: "USER" /* USER */,
+        timestamp: Date.now(),
+        content: { type: "text", parts: [query] }
+      }
+    ]);
+    const triggeredIntent = await this.intentTriggerService.intentTriggering(query, thread);
+    const result = await this.intentFulfillService.intentFulfill(
+      triggeredIntent,
+      thread
+    );
+    return { content: result };
+  }
+};
+
+export {
+  QueryService
+};
+//# sourceMappingURL=chunk-IRGR2OHS.js.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/services/query.service.ts"],"sourcesContent":["import { randomUUID } from \"node:crypto\";\nimport { StatusCodes } from \"http-status-codes\";\nimport type {\n\tA2AModule,\n\tMCPModule,\n\tMemoryModule,\n\tModelModule,\n} from \"@/modules/index.js\";\nimport { type AinAgentPrompts, AinHttpError } from \"@/types/agent.js\";\nimport {\n\tMessageRole,\n\ttype ThreadMetadata,\n\ttype ThreadObject,\n\ttype ThreadType,\n\ttype TriggeredIntent,\n} from \"@/types/memory.js\";\nimport { loggers } from \"@/utils/logger.js\";\nimport { IntentFulfillService } from \"./intents/fulfill.service\";\nimport { IntentTriggerService } from \"./intents/trigger.service\";\n\n/**\n * Service for processing user queries through the agent's AI pipeline.\n *\n * Orchestrates the query processing workflow including intent detection,\n * model inference, tool execution, and response generation. Manages\n * conversation context and coordinates between different modules.\n */\nexport class QueryService {\n\tprivate modelModule: ModelModule;\n\tprivate memoryModule?: MemoryModule;\n\tprivate intentTriggerService: IntentTriggerService;\n\tprivate intentFulfillService: IntentFulfillService;\n\n\tconstructor(\n\t\tmodelModule: ModelModule,\n\t\ta2aModule?: A2AModule,\n\t\tmcpModule?: MCPModule,\n\t\tmemoryModule?: MemoryModule,\n\t\tprompts?: AinAgentPrompts,\n\t) {\n\t\tthis.modelModule = modelModule;\n\t\tthis.memoryModule = memoryModule;\n\t\tthis.intentTriggerService = new IntentTriggerService(\n\t\t\tmodelModule,\n\t\t\tmemoryModule,\n\t\t);\n\t\tthis.intentFulfillService = new IntentFulfillService(\n\t\t\tmodelModule,\n\t\t\ta2aModule,\n\t\t\tmcpModule,\n\t\t\tmemoryModule,\n\t\t\tprompts,\n\t\t);\n\t}\n\n\t/**\n\t * Generates a title for the conversation based on the query.\n\t *\n\t * @param query - The user's input query\n\t * @returns Promise resolving to a generated title\n\t */\n\n\tprivate async generateTitle(query: string): Promise<string> {\n\t\tconst DEFAULT_TITLE = \"New Chat\";\n\t\ttry {\n\t\t\tconst modelInstance = this.modelModule.getModel();\n\t\t\tconst messages = modelInstance.generateMessages({\n\t\t\t\tquery,\n\t\t\t\tsystemPrompt: `You are a helpful assistant that generates titles for conversations.\n  Please analyze the user's query and create a concise title that accurately reflects the conversation's core topic.\n  The title must be no more than 5 words long.\n  Respond with only the title. Do not include any punctuation or extra explanations.`,\n\t\t\t});\n\t\t\tconst response = await modelInstance.fetch(messages);\n\t\t\treturn response.content || DEFAULT_TITLE;\n\t\t} catch (error) {\n\t\t\tloggers.intent.error(\"Error generating title\", {\n\t\t\t\terror,\n\t\t\t\tquery,\n\t\t\t});\n\t\t\treturn DEFAULT_TITLE;\n\t\t}\n\t}\n\n\t/**\n\t * Main entry point for processing user queries.\n\t *\n\t * Handles the complete query lifecycle:\n\t * 1. Loads or creates thread from memory\n\t * 2. Detects intent from the query\n\t * 3. Fulfills the intent with AI response\n\t * 4. Updates conversation history\n\t *\n\t * @param threadMetadata - Metadata containing type, userId, and optional threadId\n\t * @param threadMetadata.type - The type of thread (e.g., chat, workflow)\n\t * @param threadMetadata.userId - The user's unique identifier\n\t * @param threadMetadata.threadId - Optional thread identifier\n\t * @param query - The user's input query\n\t * @returns Promise resolving to object with content property containing the response\n\t */\n\tpublic async handleQuery(\n\t\tthreadMetadata: {\n\t\t\ttype: ThreadType;\n\t\t\tuserId: string;\n\t\t\tthreadId?: string;\n\t\t},\n\t\tquery: string,\n\t\tisA2A?: boolean,\n\t) {\n\t\tconst { type, userId } = threadMetadata;\n\t\tconst threadMemory = this.memoryModule?.getThreadMemory();\n\n\t\t// 1. Load or create thread\n\t\tlet threadId = threadMetadata.threadId;\n\t\tlet thread: ThreadObject | undefined;\n\t\tif (threadId) {\n\t\t\tthread = await threadMemory?.getThread(userId, threadId);\n\t\t\tif (!thread && !isA2A) {\n\t\t\t\tthrow new AinHttpError(StatusCodes.NOT_FOUND, \"Thread not found\");\n\t\t\t}\n\t\t}\n\n\t\tthreadId ??= randomUUID();\n\t\tif (!thread) {\n\t\t\tconst title = await this.generateTitle(query);\n\t\t\tconst metadata: ThreadMetadata = (await threadMemory?.createThread(\n\t\t\t\ttype,\n\t\t\t\tuserId,\n\t\t\t\tthreadId,\n\t\t\t\ttitle,\n\t\t\t)) || { type, userId, threadId, title };\n\t\t\tthread = { ...metadata, messages: [] };\n\t\t\tloggers.intent.info(`Create new thread: ${threadId}`);\n\t\t}\n\n\t\t// only add for storage, not for inference\n\t\tawait threadMemory?.addMessagesToThread(userId, threadId, [\n\t\t\t{\n\t\t\t\tmessageId: randomUUID(),\n\t\t\t\trole: MessageRole.USER,\n\t\t\t\ttimestamp: Date.now(),\n\t\t\t\tcontent: { type: \"text\", parts: [query] },\n\t\t\t},\n\t\t]);\n\n\t\t// 2. intent triggering\n\t\tconst triggeredIntent: Array<TriggeredIntent> =\n\t\t\tawait this.intentTriggerService.intentTriggering(query, thread);\n\n\t\t// 3. intent fulfillment\n\t\tconst result = await this.intentFulfillService.intentFulfill(\n\t\t\ttriggeredIntent,\n\t\t\tthread,\n\t\t);\n\n\t\treturn { content: result };\n\t}\n}\n"],"mappings":";;;;;;;;;;;;;;AAAA,SAAS,kBAAkB;AAC3B,SAAS,mBAAmB;AA0BrB,IAAM,eAAN,MAAmB;AAAA,EACjB;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EAER,YACC,aACA,WACA,WACA,cACA,SACC;AACD,SAAK,cAAc;AACnB,SAAK,eAAe;AACpB,SAAK,uBAAuB,IAAI;AAAA,MAC/B;AAAA,MACA;AAAA,IACD;AACA,SAAK,uBAAuB,IAAI;AAAA,MAC/B;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IACD;AAAA,EACD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EASA,MAAc,cAAc,OAAgC;AAC3D,UAAM,gBAAgB;AACtB,QAAI;AACH,YAAM,gBAAgB,KAAK,YAAY,SAAS;AAChD,YAAM,WAAW,cAAc,iBAAiB;AAAA,QAC/C;AAAA,QACA,cAAc;AAAA;AAAA;AAAA;AAAA,MAIf,CAAC;AACD,YAAM,WAAW,MAAM,cAAc,MAAM,QAAQ;AACnD,aAAO,SAAS,WAAW;AAAA,IAC5B,SAAS,OAAO;AACf,cAAQ,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;AAAA;AAAA,EAkBA,MAAa,YACZ,gBAKA,OACA,OACC;AACD,UAAM,EAAE,MAAM,OAAO,IAAI;AACzB,UAAM,eAAe,KAAK,cAAc,gBAAgB;AAGxD,QAAI,WAAW,eAAe;AAC9B,QAAI;AACJ,QAAI,UAAU;AACb,eAAS,MAAM,cAAc,UAAU,QAAQ,QAAQ;AACvD,UAAI,CAAC,UAAU,CAAC,OAAO;AACtB,cAAM,IAAI,aAAa,YAAY,WAAW,kBAAkB;AAAA,MACjE;AAAA,IACD;AAEA,iBAAa,WAAW;AACxB,QAAI,CAAC,QAAQ;AACZ,YAAM,QAAQ,MAAM,KAAK,cAAc,KAAK;AAC5C,YAAM,WAA4B,MAAM,cAAc;AAAA,QACrD;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,MACD,KAAM,EAAE,MAAM,QAAQ,UAAU,MAAM;AACtC,eAAS,EAAE,GAAG,UAAU,UAAU,CAAC,EAAE;AACrC,cAAQ,OAAO,KAAK,sBAAsB,QAAQ,EAAE;AAAA,IACrD;AAGA,UAAM,cAAc,oBAAoB,QAAQ,UAAU;AAAA,MACzD;AAAA,QACC,WAAW,WAAW;AAAA,QACtB;AAAA,QACA,WAAW,KAAK,IAAI;AAAA,QACpB,SAAS,EAAE,MAAM,QAAQ,OAAO,CAAC,KAAK,EAAE;AAAA,MACzC;AAAA,IACD,CAAC;AAGD,UAAM,kBACL,MAAM,KAAK,qBAAqB,iBAAiB,OAAO,MAAM;AAG/D,UAAM,SAAS,MAAM,KAAK,qBAAqB;AAAA,MAC9C;AAAA,MACA;AAAA,IACD;AAEA,WAAO,EAAE,SAAS,OAAO;AAAA,EAC1B;AACD;","names":[]}

package/dist/esm/chunk-OCONOO5Z.js

@@ -0,0 +1,103 @@
+import {
+  loggers
+} from "./chunk-N7VYM2R5.js";
+
+// src/services/intents/trigger.service.ts
+var IntentTriggerService = class {
+  modelModule;
+  memoryModule;
+  constructor(modelModule, memoryModule) {
+    this.modelModule = modelModule;
+    this.memoryModule = memoryModule;
+  }
+  /**
+   * 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 [{ subquery: query }];
+    }
+    const intents = await intentMemory.listIntents();
+    if (intents.length === 0) {
+      loggers.intentStream.warn("No intent found");
+      return [{ subquery: query }];
+    }
+    const intentList = intents.map((intent) => `- ${intent.name}: ${intent.description}`).join("\n");
+    const threadMessages = !thread ? "" : thread.messages.sort((a, b) => a.timestamp - b.timestamp).map((message) => {
+      const role = message.role === "USER" ? "User" : message.role === "MODEL" ? "Assistant" : "System";
+      const content = Array.isArray(message.content.parts) ? message.content.parts.join(" ") : String(message.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:
+
+${threadMessages}
+
+Last user question: "${query}"
+
+Based on the above conversation history, analyze the last user question and identify all relevant intents.
+
+Instructions:
+1. First, decompose the last user question into action-based subqueries (each representing a distinct action or task)
+2. Then, map each subquery to its corresponding intent from the available intent list
+3. For each subquery, provide a 2-3 sentence summary of what actions will be performed
+4. Multiple intents can be identified if the question covers various topics or actions
+5. Maintain the logical sequence of the original question when splitting into subqueries
+6. If any subquery doesn't match any intent in the available list, map it to "default" intent
+
+Output Format:
+Return the results as a JSON array with the following structure:
+[
+  {
+    "subquery": "<subquery_1>",
+    "intentName": "<intent_name_1>",
+		"actionPlan": "<2-3 sentence description of what will be done for this subquery>"
+	},
+  {
+    "subquery": "<subquery_2>",
+    "intentName": "<intent_name_2>",
+		"actionPlan": "<2-3 sentence description of what will be done for this subquery>"
+  },
+  ...
+]
+
+Requirements:
+- Each subquery should represent a single, actionable task or request
+- Preserve the original meaning and context when splitting queries
+- Select only from the provided intent list
+- Use "default" as the intent value for any subquery that doesn't match available intents.`;
+    const messages = modelInstance.generateMessages({
+      query: userMessage,
+      systemPrompt
+    });
+    const response = await modelInstance.fetch(messages);
+    if (!response.content) {
+      loggers.intent.warn("Cannot extract intent from query");
+      return [{ subquery: query }];
+    }
+    const subqueries = JSON.parse(response.content);
+    const triggeredIntent = [];
+    for (const { subquery, intentName, actionPlan } of subqueries) {
+      const intent = await intentMemory.getIntentByName(intentName);
+      triggeredIntent.push({ subquery, intent, actionPlan });
+    }
+    return triggeredIntent;
+  }
+};
+
+export {
+  IntentTriggerService
+};
+//# sourceMappingURL=chunk-OCONOO5Z.js.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/services/intents/trigger.service.ts"],"sourcesContent":["import type { MemoryModule, ModelModule } from \"@/modules\";\nimport type {\n\tMessageObject,\n\tThreadObject,\n\tTriggeredIntent,\n} from \"@/types/memory\";\nimport { loggers } from \"@/utils/logger\";\n\nexport class IntentTriggerService {\n\tprivate modelModule: ModelModule;\n\tprivate memoryModule?: MemoryModule;\n\n\tconstructor(modelModule: ModelModule, memoryModule?: MemoryModule) {\n\t\tthis.modelModule = modelModule;\n\t\tthis.memoryModule = memoryModule;\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\tpublic async intentTriggering(\n\t\tquery: string,\n\t\tthread: ThreadObject | undefined,\n\t): Promise<Array<TriggeredIntent>> {\n\t\tconst modelInstance = this.modelModule.getModel();\n\t\tconst intentMemory = this.memoryModule?.getIntentMemory();\n\t\tif (!intentMemory) {\n\t\t\treturn [{ subquery: query }];\n\t\t}\n\n\t\t// 인텐트 목록 가져오기\n\t\tconst intents = await intentMemory.listIntents();\n\n\t\tif (intents.length === 0) {\n\t\t\tloggers.intentStream.warn(\"No intent found\");\n\t\t\treturn [{ subquery: query }];\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 thread messages to a string\n\t\tconst threadMessages = !thread\n\t\t\t? \"\"\n\t\t\t: thread.messages\n\t\t\t\t\t.sort((a, b) => a.timestamp - b.timestamp)\n\t\t\t\t\t.map((message: MessageObject) => {\n\t\t\t\t\t\tconst role =\n\t\t\t\t\t\t\tmessage.role === \"USER\"\n\t\t\t\t\t\t\t\t? \"User\"\n\t\t\t\t\t\t\t\t: message.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(message.content.parts)\n\t\t\t\t\t\t\t? message.content.parts.join(\" \")\n\t\t\t\t\t\t\t: String(message.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${threadMessages}\n\nLast user question: \"${query}\"\n\nBased on the above conversation history, analyze the last user question and identify all relevant intents.\n\nInstructions:\n1. First, decompose the last user question into action-based subqueries (each representing a distinct action or task)\n2. Then, map each subquery to its corresponding intent from the available intent list\n3. For each subquery, provide a 2-3 sentence summary of what actions will be performed\n4. Multiple intents can be identified if the question covers various topics or actions\n5. Maintain the logical sequence of the original question when splitting into subqueries\n6. If any subquery doesn't match any intent in the available list, map it to \"default\" intent\n\nOutput Format:\nReturn the results as a JSON array with the following structure:\n[\n  {\n    \"subquery\": \"<subquery_1>\",\n    \"intentName\": \"<intent_name_1>\",\n\t\t\"actionPlan\": \"<2-3 sentence description of what will be done for this subquery>\"\n\t},\n  {\n    \"subquery\": \"<subquery_2>\",\n    \"intentName\": \"<intent_name_2>\",\n\t\t\"actionPlan\": \"<2-3 sentence description of what will be done for this subquery>\"\n  },\n  ...\n]\n\nRequirements:\n- Each subquery should represent a single, actionable task or request\n- Preserve the original meaning and context when splitting queries\n- Select only from the provided intent list\n- Use \"default\" as the intent value for any subquery that doesn't match available intents.`;\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\tloggers.intent.warn(\"Cannot extract intent from query\");\n\t\t\treturn [{ subquery: query }];\n\t\t}\n\n\t\tconst subqueries = JSON.parse(response.content);\n\t\tconst triggeredIntent: Array<TriggeredIntent> = [];\n\t\tfor (const { subquery, intentName, actionPlan } of subqueries) {\n\t\t\tconst intent = await intentMemory.getIntentByName(intentName);\n\t\t\ttriggeredIntent.push({ subquery, intent, actionPlan });\n\t\t}\n\n\t\treturn triggeredIntent;\n\t}\n}\n"],"mappings":";;;;;AAQO,IAAM,uBAAN,MAA2B;AAAA,EACzB;AAAA,EACA;AAAA,EAER,YAAY,aAA0B,cAA6B;AAClE,SAAK,cAAc;AACnB,SAAK,eAAe;AAAA,EACrB;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EASA,MAAa,iBACZ,OACA,QACkC;AAClC,UAAM,gBAAgB,KAAK,YAAY,SAAS;AAChD,UAAM,eAAe,KAAK,cAAc,gBAAgB;AACxD,QAAI,CAAC,cAAc;AAClB,aAAO,CAAC,EAAE,UAAU,MAAM,CAAC;AAAA,IAC5B;AAGA,UAAM,UAAU,MAAM,aAAa,YAAY;AAE/C,QAAI,QAAQ,WAAW,GAAG;AACzB,cAAQ,aAAa,KAAK,iBAAiB;AAC3C,aAAO,CAAC,EAAE,UAAU,MAAM,CAAC;AAAA,IAC5B;AAEA,UAAM,aAAa,QACjB,IAAI,CAAC,WAAW,KAAK,OAAO,IAAI,KAAK,OAAO,WAAW,EAAE,EACzD,KAAK,IAAI;AAGX,UAAM,iBAAiB,CAAC,SACrB,KACA,OAAO,SACN,KAAK,CAAC,GAAG,MAAM,EAAE,YAAY,EAAE,SAAS,EACxC,IAAI,CAAC,YAA2B;AAChC,YAAM,OACL,QAAQ,SAAS,SACd,SACA,QAAQ,SAAS,UAChB,cACA;AACL,YAAM,UAAU,MAAM,QAAQ,QAAQ,QAAQ,KAAK,IAChD,QAAQ,QAAQ,MAAM,KAAK,GAAG,IAC9B,OAAO,QAAQ,QAAQ,KAAK;AAC/B,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,cAAc;AAAA;AAAA,uBAEO,KAAK;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAkC1B,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,cAAQ,OAAO,KAAK,kCAAkC;AACtD,aAAO,CAAC,EAAE,UAAU,MAAM,CAAC;AAAA,IAC5B;AAEA,UAAM,aAAa,KAAK,MAAM,SAAS,OAAO;AAC9C,UAAM,kBAA0C,CAAC;AACjD,eAAW,EAAE,UAAU,YAAY,WAAW,KAAK,YAAY;AAC9D,YAAM,SAAS,MAAM,aAAa,gBAAgB,UAAU;AAC5D,sBAAgB,KAAK,EAAE,UAAU,QAAQ,WAAW,CAAC;AAAA,IACtD;AAEA,WAAO;AAAA,EACR;AACD;","names":[]}

package/dist/esm/{chunk-NWL3BMUH.js → chunk-SHQWRBWS.js}

@@ -34,7 +34,10 @@ var A2AModule = class {
     return connectors;
   }
   getOrCreateClient(connector) {
-    connector.client ??= new A2AClient(connector.url);
+    if (!connector.client) {
+      console.log(`Creating A2AClient for ${connector.name} with URL: ${connector.url}`);
+      connector.client = new A2AClient(connector.url);
+    }
     return connector.client;
   }
   /**
@@ -61,7 +64,8 @@ var A2AModule = class {
           description: card.description
         };
         tools.push(tool);
-      } catch (_error) {
+      } catch (error) {
+        console.error(`Failed to get agent card for ${name}:`, error.message);
       }
     }
     return tools;
@@ -161,4 +165,4 @@ ${finalText.join("\n")}`;
 export {
   A2AModule
 };
-//# sourceMappingURL=chunk-NWL3BMUH.js.map
+//# sourceMappingURL=chunk-SHQWRBWS.js.map

package/dist/esm/chunk-SHQWRBWS.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 {\n\tCONNECTOR_PROTOCOL_TYPE,\n\ttype ConnectorTool,\n} from \"@/types/connector.js\";\nimport { ThreadType } from \"@/types/memory.js\";\nimport { loggers } from \"@/utils/logger.js\";\nimport { A2AConnector } from \"./a2a.connector.js\";\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 a2aConnectors: Map<string, A2AConnector> = new Map();\n\t/** Map of session IDs to their A2A session state */\n\tprivate a2aTasks: Map<string, string> = 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 conns - Set of name, url pair\n\t */\n\tpublic async addA2AConnector(conns: {\n\t\t[name: string]: string;\n\t}): Promise<void> {\n\t\tfor (const [name, url] of Object.entries(conns)) {\n\t\t\tconst conn = new A2AConnector(name, url);\n\t\t\tthis.a2aConnectors.set(name, conn);\n\t\t}\n\t}\n\n\tpublic getA2AConnectors(): Array<{ name: string; url: string }> {\n\t\tconst connectors: Array<{ name: string; url: string }> = [];\n\t\tfor (const [name, connector] of this.a2aConnectors.entries()) {\n\t\t\tconnectors.push({ name, url: connector.url });\n\t\t}\n\t\treturn connectors;\n\t}\n\n\tprivate getOrCreateClient(connector: A2AConnector): A2AClient {\n\t\tif (!connector.client) {\n\t\t\tconsole.log(`Creating A2AClient for ${connector.name} with URL: ${connector.url}`);\n\t\t\tconnector.client = new A2AClient(connector.url);\n\t\t}\n\t\treturn connector.client;\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<ConnectorTool[]> {\n\t\tconst tools: ConnectorTool[] = [];\n\t\tfor (const [name, conn] of this.a2aConnectors.entries()) {\n\t\t\tif (!conn.enabled) {\n\t\t\t\tcontinue; // skip disabled agent\n\t\t\t}\n\n\t\t\ttry {\n\t\t\t\tconst client = this.getOrCreateClient(conn);\n\t\t\t\tconst card: AgentCard = await client.getAgentCard();\n\t\t\t\t/* TODO: add each skill as tool? */\n\t\t\t\tconst tool: ConnectorTool = {\n\t\t\t\t\ttoolName: card.name.replaceAll(\" \", \"-\"),\n\t\t\t\t\tconnectorName: name,\n\t\t\t\t\tprotocol: CONNECTOR_PROTOCOL_TYPE.A2A,\n\t\t\t\t\tdescription: card.description,\n\t\t\t\t};\n\n\t\t\t\ttools.push(tool);\n\t\t\t} catch (error: any) {\n\t\t\t\tconsole.error(`Failed to get agent card for ${name}:`, error.message);\n\t\t\t\t// Agent not responded, just skip\n\t\t\t}\n\t\t}\n\t\treturn tools;\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},\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\tcontextId: threadId,\n\t\t};\n\n\t\tif (this.a2aTasks.has(threadId)) {\n\t\t\tmessagePayload.taskId = this.a2aTasks.get(threadId);\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 query - 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: ConnectorTool,\n\t\tquery: string,\n\t\tthreadId: string,\n\t): Promise<string> {\n\t\tconst finalText: string[] = [];\n\t\tconst connector = this.a2aConnectors.get(tool.connectorName);\n\t\tconst messagePayload = this.getMessagePayload(query, threadId);\n\t\tconst params: MessageSendParams = {\n\t\t\tmessage: messagePayload,\n\t\t};\n\n\t\ttry {\n\t\t\tconst client = this.getOrCreateClient(connector!);\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\tthis.a2aTasks.delete(threadId);\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\tconst taskId = this.a2aTasks.get(threadId);\n\t\t\t\t\tif (msg.taskId && msg.taskId !== taskId) {\n\t\t\t\t\t\tthis.a2aTasks.set(threadId, msg.taskId);\n\t\t\t\t\t}\n\t\t\t\t} else if (event.kind === \"task\") {\n\t\t\t\t\t// establishing the Task ID\n\t\t\t\t\tconst task = event as Task;\n\t\t\t\t\tif (task.id !== this.a2aTasks.get(threadId)) {\n\t\t\t\t\t\tthis.a2aTasks.set(threadId, task.id);\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\tconst toolResult = `[Bot Called A2A Tool ${tool.toolName}]\\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.toolName}]\\n${finalText.join(\"\\n\")}`;\n\t}\n}\n"],"mappings":";;;;;;;;AAAA,SAAS,kBAAkB;AAS3B,SAAS,iBAAiB;AAgBnB,IAAM,YAAN,MAAgB;AAAA;AAAA,EAEd,gBAA2C,oBAAI,IAAI;AAAA;AAAA,EAEnD,WAAgC,oBAAI,IAAI;AAAA,EACxC,UAAkB,WAAW;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOrC,MAAa,gBAAgB,OAEX;AACjB,eAAW,CAAC,MAAM,GAAG,KAAK,OAAO,QAAQ,KAAK,GAAG;AAChD,YAAM,OAAO,IAAI,aAAa,MAAM,GAAG;AACvC,WAAK,cAAc,IAAI,MAAM,IAAI;AAAA,IAClC;AAAA,EACD;AAAA,EAEO,mBAAyD;AAC/D,UAAM,aAAmD,CAAC;AAC1D,eAAW,CAAC,MAAM,SAAS,KAAK,KAAK,cAAc,QAAQ,GAAG;AAC7D,iBAAW,KAAK,EAAE,MAAM,KAAK,UAAU,IAAI,CAAC;AAAA,IAC7C;AACA,WAAO;AAAA,EACR;AAAA,EAEQ,kBAAkB,WAAoC;AAC7D,QAAI,CAAC,UAAU,QAAQ;AACtB,cAAQ,IAAI,0BAA0B,UAAU,IAAI,cAAc,UAAU,GAAG,EAAE;AACjF,gBAAU,SAAS,IAAI,UAAU,UAAU,GAAG;AAAA,IAC/C;AACA,WAAO,UAAU;AAAA,EAClB;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAUA,MAAa,WAAqC;AACjD,UAAM,QAAyB,CAAC;AAChC,eAAW,CAAC,MAAM,IAAI,KAAK,KAAK,cAAc,QAAQ,GAAG;AACxD,UAAI,CAAC,KAAK,SAAS;AAClB;AAAA,MACD;AAEA,UAAI;AACH,cAAM,SAAS,KAAK,kBAAkB,IAAI;AAC1C,cAAM,OAAkB,MAAM,OAAO,aAAa;AAElD,cAAM,OAAsB;AAAA,UAC3B,UAAU,KAAK,KAAK,WAAW,KAAK,GAAG;AAAA,UACvC,eAAe;AAAA,UACf;AAAA,UACA,aAAa,KAAK;AAAA,QACnB;AAEA,cAAM,KAAK,IAAI;AAAA,MAChB,SAAS,OAAY;AACpB,gBAAQ,MAAM,gCAAgC,IAAI,KAAK,MAAM,OAAO;AAAA,MAErE;AAAA,IACD;AACA,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,MACD;AAAA,MACA,OAAO;AAAA,QACN;AAAA,UACC,MAAM;AAAA,UACN,MAAM;AAAA,QACP;AAAA,MACD;AAAA,MACA,WAAW;AAAA,IACZ;AAEA,QAAI,KAAK,SAAS,IAAI,QAAQ,GAAG;AAChC,qBAAe,SAAS,KAAK,SAAS,IAAI,QAAQ;AAAA,IACnD;AAEA,WAAO;AAAA,EACR;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAaA,MAAa,QACZ,MACA,OACA,UACkB;AAClB,UAAM,YAAsB,CAAC;AAC7B,UAAM,YAAY,KAAK,cAAc,IAAI,KAAK,aAAa;AAC3D,UAAM,iBAAiB,KAAK,kBAAkB,OAAO,QAAQ;AAC7D,UAAM,SAA4B;AAAA,MACjC,SAAS;AAAA,IACV;AAEA,QAAI;AACH,YAAM,SAAS,KAAK,kBAAkB,SAAU;AAChD,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,iBAAK,SAAS,OAAO,QAAQ;AAAA,UAC9B;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,gBAAM,SAAS,KAAK,SAAS,IAAI,QAAQ;AACzC,cAAI,IAAI,UAAU,IAAI,WAAW,QAAQ;AACxC,iBAAK,SAAS,IAAI,UAAU,IAAI,MAAM;AAAA,UACvC;AAAA,QACD,WAAW,MAAM,SAAS,QAAQ;AAEjC,gBAAM,OAAO;AACb,cAAI,KAAK,OAAO,KAAK,SAAS,IAAI,QAAQ,GAAG;AAC5C,iBAAK,SAAS,IAAI,UAAU,KAAK,EAAE;AAAA,UACpC;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,YAAM,aAAa,wBAAwB,KAAK,QAAQ;AAAA,EAAM,OAAO,UAAU,WAAW,QAAQ,KAAK,UAAU,OAAO,MAAM,CAAC,CAAC;AAChI,aAAO;AAAA,IACR;AAEA,WAAO,wBAAwB,KAAK,QAAQ;AAAA,EAAM,UAAU,KAAK,IAAI,CAAC;AAAA,EACvE;AACD;","names":[]}

package/dist/esm/{chunk-UQK33CJX.js → chunk-X6JNM3XR.js}

@@ -15,4 +15,4 @@ export {
   MessageRole,
   ThreadType
 };
-//# sourceMappingURL=chunk-UQK33CJX.js.map
+//# sourceMappingURL=chunk-X6JNM3XR.js.map

package/dist/esm/{chunk-UQK33CJX.js.map → chunk-X6JNM3XR.js.map}

@@ -1 +1 @@
-{"version":3,"sources":["../../src/types/memory.ts"],"sourcesContent":["/**\n * Roles for participants in a message.\n */\nexport enum MessageRole {\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 message content.\n *\n * Supports multi-part content with different types (text, images, etc.).\n */\nexport type MessageContentObject = {\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 thread.\n *\n * @example\n * ```typescript\n * const message: MessageObject = {\n *   role: MessageRole.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 MessageObject = {\n\tmessageId: string;\n\t/** Role of the message sender */\n\trole: MessageRole;\n\t/** Message content with type and parts */\n\tcontent: MessageContentObject;\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 enum ThreadType {\n\tWORKFLOW = \"WORKFLOW\",\n\tCHAT = \"CHAT\",\n}\n\nexport type ThreadMetadata = {\n\ttype: ThreadType;\n\ttitle: string;\n\tuserId: string;\n\tthreadId: string;\n};\n\n/**\n * Represents a conversation thread containing multiple messages.\n *\n * Messages are stored in a key-value structure where keys are unique message IDs\n * and values are the corresponding message objects.\n *\n * @example\n * ```typescript\n * const thread: ThreadObject = {\n * \t title: \"New conversation\",\n *   messages: [\n *     { messageId: <UUID_1>, role: MessageRole.USER, content: {...}, timestamp: 1234567890 },\n *     { messageId: <UUID_2> ,role: MessageRole.MODEL, content: {...}, timestamp: 1234567891 }\n *   ]\n * };\n * ```\n */\nexport type ThreadObject = {\n\tuserId: string;\n\tthreadId: string;\n\ttype: ThreadType;\n\ttitle: string;\n\tmessages: Array<MessageObject>;\n};\n\nexport interface Intent {\n\tid: string;\n\tname: string;\n\tdescription: string;\n\tprompt?: string;\n\ttriggeringSentences?: Array<string>;\n\ttags?: Array<string>;\n}\n"],"mappings":";AAGO,IAAK,cAAL,kBAAKA,iBAAL;AAEN,EAAAA,aAAA,UAAO;AAEP,EAAAA,aAAA,YAAS;AAET,EAAAA,aAAA,WAAQ;AANG,SAAAA;AAAA,GAAA;AAiDL,IAAK,aAAL,kBAAKC,gBAAL;AACN,EAAAA,YAAA,cAAW;AACX,EAAAA,YAAA,UAAO;AAFI,SAAAA;AAAA,GAAA;","names":["MessageRole","ThreadType"]}
+{"version":3,"sources":["../../src/types/memory.ts"],"sourcesContent":["/**\n * Roles for participants in a message.\n */\nexport enum MessageRole {\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 message content.\n *\n * Supports multi-part content with different types (text, images, etc.).\n */\nexport type MessageContentObject = {\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 thread.\n *\n * @example\n * ```typescript\n * const message: MessageObject = {\n *   role: MessageRole.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 MessageObject = {\n\tmessageId: string;\n\t/** Role of the message sender */\n\trole: MessageRole;\n\t/** Message content with type and parts */\n\tcontent: MessageContentObject;\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 enum ThreadType {\n\tWORKFLOW = \"WORKFLOW\",\n\tCHAT = \"CHAT\",\n}\n\nexport type ThreadMetadata = {\n\ttype: ThreadType;\n\ttitle: string;\n\tuserId: string;\n\tthreadId: string;\n};\n\n/**\n * Represents a conversation thread containing multiple messages.\n *\n * Messages are stored in a key-value structure where keys are unique message IDs\n * and values are the corresponding message objects.\n *\n * @example\n * ```typescript\n * const thread: ThreadObject = {\n * \t title: \"New conversation\",\n *   messages: [\n *     { messageId: <UUID_1>, role: MessageRole.USER, content: {...}, timestamp: 1234567890 },\n *     { messageId: <UUID_2> ,role: MessageRole.MODEL, content: {...}, timestamp: 1234567891 }\n *   ]\n * };\n * ```\n */\nexport type ThreadObject = {\n\tuserId: string;\n\tthreadId: string;\n\ttype: ThreadType;\n\ttitle: string;\n\tmessages: Array<MessageObject>;\n};\n\nexport interface Intent {\n\tid: string;\n\tname: string;\n\tdescription: string;\n\tprompt?: string;\n\ttriggeringSentences?: Array<string>;\n\ttags?: Array<string>;\n}\n\nexport type TriggeredIntent = {\n\tsubquery: string;\n\tintent?: Intent;\n\tactionPlan?: string;\n};\n"],"mappings":";AAGO,IAAK,cAAL,kBAAKA,iBAAL;AAEN,EAAAA,aAAA,UAAO;AAEP,EAAAA,aAAA,YAAS;AAET,EAAAA,aAAA,WAAQ;AANG,SAAAA;AAAA,GAAA;AAiDL,IAAK,aAAL,kBAAKC,gBAAL;AACN,EAAAA,YAAA,cAAW;AACX,EAAAA,YAAA,UAAO;AAFI,SAAAA;AAAA,GAAA;","names":["MessageRole","ThreadType"]}

package/dist/esm/chunk-XTPDMHUI.js

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

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

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/services/query-stream.service.ts"],"sourcesContent":["import { randomUUID } from \"node:crypto\";\nimport { StatusCodes } from \"http-status-codes\";\nimport type {\n\tA2AModule,\n\tMCPModule,\n\tMemoryModule,\n\tModelModule,\n} from \"@/modules/index.js\";\nimport { type AinAgentPrompts, AinHttpError } from \"@/types/agent.js\";\nimport {\n\tMessageRole,\n\ttype ThreadMetadata,\n\ttype ThreadObject,\n\ttype ThreadType,\n\ttype TriggeredIntent,\n} from \"@/types/memory.js\";\nimport type { StreamEvent } from \"@/types/stream\";\nimport { loggers } from \"@/utils/logger.js\";\nimport { IntentFulfillStreamService } from \"./intents/fulfill-stream.service\";\nimport { IntentTriggerService } from \"./intents/trigger.service\";\n\n/**\n * Service for processing user queries through the agent's AI pipeline.\n *\n * Orchestrates the query processing workflow including intent detection,\n * model inference, tool execution, and response generation. Manages\n * conversation context and coordinates between different modules.\n */\nexport class QueryStreamService {\n\tprivate modelModule: ModelModule;\n\tprivate memoryModule?: MemoryModule;\n\tprivate intentTriggerService: IntentTriggerService;\n\tprivate intentFulfillStreamService: IntentFulfillStreamService;\n\n\tconstructor(\n\t\tmodelModule: ModelModule,\n\t\ta2aModule?: A2AModule,\n\t\tmcpModule?: MCPModule,\n\t\tmemoryModule?: MemoryModule,\n\t\tprompts?: AinAgentPrompts,\n\t) {\n\t\tthis.modelModule = modelModule;\n\t\tthis.memoryModule = memoryModule;\n\t\tthis.intentTriggerService = new IntentTriggerService(\n\t\t\tmodelModule,\n\t\t\tmemoryModule,\n\t\t);\n\t\tthis.intentFulfillStreamService = new IntentFulfillStreamService(\n\t\t\tmodelModule,\n\t\t\ta2aModule,\n\t\t\tmcpModule,\n\t\t\tmemoryModule,\n\t\t\tprompts,\n\t\t);\n\t}\n\n\t/**\n\t * Generates a title for the conversation based on the query.\n\t *\n\t * @param query - The user's input query\n\t * @returns Promise resolving to a generated title\n\t */\n\n\tprivate async generateTitle(query: string): Promise<string> {\n\t\tconst DEFAULT_TITLE = \"New Chat\";\n\t\ttry {\n\t\t\tconst modelInstance = this.modelModule.getModel();\n\t\t\tconst messages = modelInstance.generateMessages({\n\t\t\t\tquery,\n\t\t\t\tsystemPrompt: `You are a helpful assistant that generates titles for conversations.\n  Please analyze the user's query and create a concise title that accurately reflects the conversation's core topic.\n  The title must be no more than 5 words long.\n  Respond with only the title. Do not include any punctuation or extra explanations.`,\n\t\t\t});\n\t\t\tconst response = await modelInstance.fetch(messages);\n\t\t\treturn response.content || DEFAULT_TITLE;\n\t\t} catch (error) {\n\t\t\tloggers.intentStream.error(\"Error generating title\", {\n\t\t\t\terror,\n\t\t\t\tquery,\n\t\t\t});\n\t\t\treturn DEFAULT_TITLE;\n\t\t}\n\t}\n\n\t/**\n\t * Main entry point for processing streaming user queries.\n\t *\n\t * Handles the complete query lifecycle:\n\t * 1. Loads or creates thread from memory\n\t * 2. Detects intent from the query\n\t * 3. Fulfills the intent with streaming AI response\n\t * 4. Updates conversation history in real-time\n\t *\n\t * @param threadMetadata - Metadata containing type, userId, and optional threadId\n\t * @param threadMetadata.type - The type of thread (e.g., chat, workflow)\n\t * @param threadMetadata.userId - The user's unique identifier\n\t * @param threadMetadata.threadId - Optional thread identifier\n\t * @param query - The user's input query\n\t * @returns AsyncGenerator yielding StreamEvent objects for SSE\n\t */\n\tpublic async *handleQueryStream(\n\t\tthreadMetadata: {\n\t\t\ttype: ThreadType;\n\t\t\tuserId: string;\n\t\t\tthreadId?: string;\n\t\t},\n\t\tquery: string,\n\t\tisA2A?: boolean,\n\t): AsyncGenerator<StreamEvent> {\n\t\tconst { type, userId } = threadMetadata;\n\t\tconst threadMemory = this.memoryModule?.getThreadMemory();\n\n\t\t// 1. Load or create thread\n\t\tlet threadId = threadMetadata.threadId;\n\t\tlet thread: ThreadObject | undefined;\n\t\tif (threadId) {\n\t\t\tthread = await threadMemory?.getThread(userId, threadId);\n\t\t\tif (!thread && !isA2A) {\n\t\t\t\tthrow new AinHttpError(StatusCodes.NOT_FOUND, \"Thread not found\");\n\t\t\t}\n\t\t}\n\n\t\tthreadId ??= randomUUID();\n\t\tif (!thread) {\n\t\t\tconst title = await this.generateTitle(query);\n\t\t\tconst metadata: ThreadMetadata = (await threadMemory?.createThread(\n\t\t\t\ttype,\n\t\t\t\tuserId,\n\t\t\t\tthreadId,\n\t\t\t\ttitle,\n\t\t\t)) || { type, userId, threadId, title };\n\t\t\tthread = { ...metadata, messages: [] };\n\t\t\tloggers.intent.info(`Create new thread: ${threadId}`);\n\t\t\tyield { event: \"thread_id\", data: { type, userId, threadId, title } };\n\t\t}\n\n\t\t// only add for storage, not for inference\n\t\tawait threadMemory?.addMessagesToThread(userId, threadId, [\n\t\t\t{\n\t\t\t\tmessageId: randomUUID(),\n\t\t\t\trole: MessageRole.USER,\n\t\t\t\ttimestamp: Date.now(),\n\t\t\t\tcontent: { type: \"text\", parts: [query] },\n\t\t\t},\n\t\t]);\n\n\t\t// 2. intent triggering\n\t\tconst triggeredIntent: Array<TriggeredIntent> =\n\t\t\tawait this.intentTriggerService.intentTriggering(query, thread);\n\n\t\t// 3. intent fulfillment\n\t\tconst stream = this.intentFulfillStreamService.intentFulfillStream(\n\t\t\ttriggeredIntent,\n\t\t\tthread,\n\t\t);\n\n\t\tfor await (const event of stream) {\n\t\t\tyield event;\n\t\t}\n\t}\n}\n"],"mappings":";;;;;;;;;;;;;;AAAA,SAAS,kBAAkB;AAC3B,SAAS,mBAAmB;AA2BrB,IAAM,qBAAN,MAAyB;AAAA,EACvB;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EAER,YACC,aACA,WACA,WACA,cACA,SACC;AACD,SAAK,cAAc;AACnB,SAAK,eAAe;AACpB,SAAK,uBAAuB,IAAI;AAAA,MAC/B;AAAA,MACA;AAAA,IACD;AACA,SAAK,6BAA6B,IAAI;AAAA,MACrC;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IACD;AAAA,EACD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EASA,MAAc,cAAc,OAAgC;AAC3D,UAAM,gBAAgB;AACtB,QAAI;AACH,YAAM,gBAAgB,KAAK,YAAY,SAAS;AAChD,YAAM,WAAW,cAAc,iBAAiB;AAAA,QAC/C;AAAA,QACA,cAAc;AAAA;AAAA;AAAA;AAAA,MAIf,CAAC;AACD,YAAM,WAAW,MAAM,cAAc,MAAM,QAAQ;AACnD,aAAO,SAAS,WAAW;AAAA,IAC5B,SAAS,OAAO;AACf,cAAQ,aAAa,MAAM,0BAA0B;AAAA,QACpD;AAAA,QACA;AAAA,MACD,CAAC;AACD,aAAO;AAAA,IACR;AAAA,EACD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAkBA,OAAc,kBACb,gBAKA,OACA,OAC8B;AAC9B,UAAM,EAAE,MAAM,OAAO,IAAI;AACzB,UAAM,eAAe,KAAK,cAAc,gBAAgB;AAGxD,QAAI,WAAW,eAAe;AAC9B,QAAI;AACJ,QAAI,UAAU;AACb,eAAS,MAAM,cAAc,UAAU,QAAQ,QAAQ;AACvD,UAAI,CAAC,UAAU,CAAC,OAAO;AACtB,cAAM,IAAI,aAAa,YAAY,WAAW,kBAAkB;AAAA,MACjE;AAAA,IACD;AAEA,iBAAa,WAAW;AACxB,QAAI,CAAC,QAAQ;AACZ,YAAM,QAAQ,MAAM,KAAK,cAAc,KAAK;AAC5C,YAAM,WAA4B,MAAM,cAAc;AAAA,QACrD;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,MACD,KAAM,EAAE,MAAM,QAAQ,UAAU,MAAM;AACtC,eAAS,EAAE,GAAG,UAAU,UAAU,CAAC,EAAE;AACrC,cAAQ,OAAO,KAAK,sBAAsB,QAAQ,EAAE;AACpD,YAAM,EAAE,OAAO,aAAa,MAAM,EAAE,MAAM,QAAQ,UAAU,MAAM,EAAE;AAAA,IACrE;AAGA,UAAM,cAAc,oBAAoB,QAAQ,UAAU;AAAA,MACzD;AAAA,QACC,WAAW,WAAW;AAAA,QACtB;AAAA,QACA,WAAW,KAAK,IAAI;AAAA,QACpB,SAAS,EAAE,MAAM,QAAQ,OAAO,CAAC,KAAK,EAAE;AAAA,MACzC;AAAA,IACD,CAAC;AAGD,UAAM,kBACL,MAAM,KAAK,qBAAqB,iBAAiB,OAAO,MAAM;AAG/D,UAAM,SAAS,KAAK,2BAA2B;AAAA,MAC9C;AAAA,MACA;AAAA,IACD;AAEA,qBAAiB,SAAS,QAAQ;AACjC,YAAM;AAAA,IACP;AAAA,EACD;AACD;","names":[]}