@ainetwork/adk 0.1.6 → 0.1.7

package/dist/cjs/chunk-UQ72NNHV.cjs

@@ -0,0 +1,225 @@
+"use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _optionalChain(ops) { let lastAccessLHS = undefined; let value = ops[0]; let i = 1; while (i < ops.length) { const op = ops[i]; const fn = ops[i + 1]; i += 2; if ((op === 'optionalAccess' || op === 'optionalCall') && value == null) { return undefined; } if (op === 'access' || op === 'optionalAccess') { lastAccessLHS = value; value = fn(value); } else if (op === 'call' || op === 'optionalCall') { value = fn((...args) => value.call(lastAccessLHS, ...args)); lastAccessLHS = undefined; } } return value; }
+
+var _chunk466ED5WNcjs = require('./chunk-466ED5WN.cjs');
+
+// src/services/query-stream.service.ts
+var QueryStreamService = class {
+  
+  
+  
+  
+  
+  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()}.
+
+${_optionalChain([this, 'access', _ => _.prompts, 'optionalAccess', _2 => _2.agent]) || ""}
+
+${_optionalChain([this, 'access', _3 => _3.prompts, 'optionalAccess', _4 => _4.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 = [];
+        _chunk466ED5WNcjs.loggers.intentStream.debug("messages", { messages });
+        for await (const chunk of responseStream) {
+          const delta = chunk.delta;
+          if (_optionalChain([delta, 'optionalAccess', _5 => _5.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 (_optionalChain([toolCallDelta, 'access', _6 => _6.function, 'optionalAccess', _7 => _7.name])) {
+                assembledToolCalls[index].function.name = toolCallDelta.function.name;
+              }
+              if (_optionalChain([toolCallDelta, 'access', _8 => _8.function, 'optionalAccess', _9 => _9.arguments])) {
+                assembledToolCalls[index].function.arguments += toolCallDelta.function.arguments;
+              }
+            }
+          } else if (_optionalChain([chunk, 'access', _10 => _10.delta, 'optionalAccess', _11 => _11.content])) {
+            yield {
+              event: "text_chunk",
+              data: { delta: chunk.delta.content }
+            };
+          }
+        }
+        _chunk466ED5WNcjs.loggers.intentStream.debug("assembledToolCalls", {
+          assembledToolCalls
+        });
+        if (didCallTool && assembledToolCalls.length > 0) {
+          const messagePayload = _optionalChain([this, 'access', _12 => _12.a2aModule, 'optionalAccess', _13 => _13.getMessagePayload, 'call', _14 => _14(
+            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);
+              _chunk466ED5WNcjs.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 {
+              _chunk466ED5WNcjs.loggers.intent.warn(
+                `Unrecognized tool type: ${selectedTool.protocol}`
+              );
+              continue;
+            }
+            yield {
+              event: "tool_output",
+              data: { toolName, result: toolResult }
+            };
+            _chunk466ED5WNcjs.loggers.intent.debug("toolResult", { toolResult });
+            processList.push(toolResult);
+            modelInstance.appendMessages(messages, toolResult);
+          }
+        }
+        if (!didCallTool) break;
+      }
+    } catch (error) {
+      _chunk466ED5WNcjs.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 = _optionalChain([this, 'access', _15 => _15.memoryModule, 'optionalAccess', _16 => _16.getSessionMemory, 'call', _17 => _17()]);
+    const session = !userId ? void 0 : await _optionalChain([sessionMemory, 'optionalAccess', _18 => _18.getSession, 'call', _19 => _19(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) {
+          _chunk466ED5WNcjs.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) {
+      _chunk466ED5WNcjs.loggers.intent.error("Error in handleQuery", { error });
+      res.write(
+        `event: error
+data: ${JSON.stringify({ message: "Stream failed" })}
+
+`
+      );
+    } finally {
+      res.end();
+    }
+  }
+};
+
+
+
+exports.QueryStreamService = QueryStreamService;
+//# sourceMappingURL=chunk-UQ72NNHV.cjs.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-UQ72NNHV.cjs","../../src/services/query-stream.service.ts"],"names":["toolArgs"],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACA;ACqBO,IAAM,mBAAA,EAAN,MAAyB;AAAA,EACvB;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EAER,WAAA,CACC,WAAA,EACA,SAAA,EACA,SAAA,EACA,YAAA,EACA,OAAA,EACC;AACD,IAAA,IAAA,CAAK,YAAA,EAAc,WAAA;AACnB,IAAA,IAAA,CAAK,UAAA,EAAY,SAAA;AACjB,IAAA,IAAA,CAAK,UAAA,EAAY,SAAA;AACjB,IAAA,IAAA,CAAK,aAAA,EAAe,YAAA;AACpB,IAAA,IAAA,CAAK,QAAA,EAAU,OAAA;AAAA,EAChB;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EASA,MAAc,gBAAA,CAAiB,KAAA,EAAe;AAE7C,IAAA,OAAO,KAAA;AAAA,EACR;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAgBA,MAAA,CAAc,gBAAA,CACb,KAAA,EACA,SAAA,EACA,cAAA,EAC8B;AAC9B,IAAA,IAAI;AACH,MAAA,MAAM,aAAA,EAAe,CAAA;AAAA,SAAA,EAAA,iBACb,IAAI,IAAA,CAAK,CAAA,CAAA,CAAE,kBAAA,CAAmB,CAAC,CAAA;AAAA;AAAA,kBAExC,IAAA,mBAAK,OAAA,6BAAS,QAAA,GAAS,EAAE,CAAA;AAAA;AAAA,kBAEzB,IAAA,qBAAK,OAAA,6BAAS,SAAA,GAAU,EAAE,CAAA;AAAA,IAAA,CAAA;AAGzB,MAAA,MAAM,cAAA,EAAgB,IAAA,CAAK,WAAA,CAAY,QAAA,CAAS,CAAA;AAChD,MAAA,MAAM,SAAA,EAAW,aAAA,CAAc,gBAAA,CAAiB;AAAA,QAC/C,KAAA;AAAA,QACA,cAAA;AAAA,QACA,YAAA,EAAc,YAAA,CAAa,IAAA,CAAK;AAAA,MACjC,CAAC,CAAA;AAED,MAAA,MAAM,MAAA,EAAsB,CAAC,CAAA;AAC7B,MAAA,GAAA,CAAI,IAAA,CAAK,SAAA,EAAW;AACnB,QAAA,KAAA,CAAM,IAAA,CAAK,GAAG,IAAA,CAAK,SAAA,CAAU,QAAA,CAAS,CAAC,CAAA;AAAA,MACxC;AACA,MAAA,GAAA,CAAI,IAAA,CAAK,SAAA,EAAW;AACnB,QAAA,KAAA,CAAM,IAAA,CAAK,GAAI,MAAM,IAAA,CAAK,SAAA,CAAU,QAAA,CAAS,CAAE,CAAA;AAAA,MAChD;AACA,MAAA,MAAM,UAAA,EAAY,aAAA,CAAc,uBAAA,CAAwB,KAAK,CAAA;AAE7D,MAAA,MAAM,YAAA,EAAwB,CAAC,CAAA;AAC/B,MAAA,MAAM,aAAA,EAAe,EAAA;AACrB,MAAA,IAAI,YAAA,EAAc,KAAA;AAElB,MAAA,MAAA,CAAO,IAAA,EAAM;AACZ,QAAA,MAAM,eAAA,EACL,MAAM,aAAA,CAAc,6BAAA;AAAA,UACnB,QAAA;AAAA,UACA;AAAA,QACD,CAAA;AACD,QAAA,YAAA,EAAc,KAAA;AAEd,QAAA,MAAM,mBAAA,EAIA,CAAC,CAAA;AAEP,QAAA,yBAAA,CAAQ,YAAA,CAAa,KAAA,CAAM,UAAA,EAAY,EAAE,SAAS,CAAC,CAAA;AAEnD,QAAA,IAAA,MAAA,CAAA,MAAiB,MAAA,GAAS,cAAA,EAAgB;AACzC,UAAA,MAAM,MAAA,EAAQ,KAAA,CAAM,KAAA;AACpB,UAAA,GAAA,iBAAI,KAAA,6BAAO,YAAA,EAAY;AACtB,YAAA,YAAA,EAAc,IAAA;AACd,YAAA,IAAA,CAAA,MAAW,cAAA,GAAiB,KAAA,CAAM,UAAA,EAAY;AAC7C,cAAA,MAAM,MAAA,EAAQ,aAAA,CAAc,KAAA;AAE5B,cAAA,GAAA,CAAI,CAAC,kBAAA,CAAmB,KAAK,CAAA,EAAG;AAC/B,gBAAA,kBAAA,CAAmB,KAAK,EAAA,EAAI;AAAA,kBAC3B,EAAA,EAAI,EAAA;AAAA,kBACJ,IAAA,EAAM,UAAA;AAAA,kBACN,QAAA,EAAU,EAAE,IAAA,EAAM,EAAA,EAAI,SAAA,EAAW,GAAG;AAAA,gBACrC,CAAA;AAAA,cACD;AAEA,cAAA,GAAA,CAAI,aAAA,CAAc,EAAA,EAAI;AACrB,gBAAA,kBAAA,CAAmB,KAAK,CAAA,CAAE,GAAA,EAAK,aAAA,CAAc,EAAA;AAAA,cAC9C;AAEA,cAAA,GAAA,iBAAI,aAAA,qBAAc,QAAA,6BAAU,MAAA,EAAM;AACjC,gBAAA,kBAAA,CAAmB,KAAK,CAAA,CAAE,QAAA,CAAS,KAAA,EAClC,aAAA,CAAc,QAAA,CAAS,IAAA;AAAA,cACzB;AAEA,cAAA,GAAA,iBAAI,aAAA,qBAAc,QAAA,6BAAU,WAAA,EAAW;AACtC,gBAAA,kBAAA,CAAmB,KAAK,CAAA,CAAE,QAAA,CAAS,UAAA,GAClC,aAAA,CAAc,QAAA,CAAS,SAAA;AAAA,cACzB;AAAA,YACD;AAAA,UACD,EAAA,KAAA,GAAA,iBAAW,KAAA,uBAAM,KAAA,+BAAO,SAAA,EAAS;AAChC,YAAA,MAAM;AAAA,cACL,KAAA,EAAO,YAAA;AAAA,cACP,IAAA,EAAM,EAAE,KAAA,EAAO,KAAA,CAAM,KAAA,CAAM,QAAQ;AAAA,YACpC,CAAA;AAAA,UACD;AAAA,QACD;AAEA,QAAA,yBAAA,CAAQ,YAAA,CAAa,KAAA,CAAM,oBAAA,EAAsB;AAAA,UAChD;AAAA,QACD,CAAC,CAAA;AAED,QAAA,GAAA,CAAI,YAAA,GAAe,kBAAA,CAAmB,OAAA,EAAS,CAAA,EAAG;AACjD,UAAA,MAAM,eAAA,kBAAiB,IAAA,uBAAK,SAAA,+BAAW,iBAAA;AAAA,YACtC,KAAA;AAAA,YACA;AAAA,UACD,GAAA;AACA,UAAA,IAAA,CAAA,MAAW,SAAA,GAAY,kBAAA,EAAoB;AAC1C,YAAA,MAAM,SAAA,EAAW,QAAA,CAAS,QAAA,CAAS,IAAA;AACnC,YAAA,MAAM,SAAA,EAAW,IAAA,CAAK,KAAA,CAAM,QAAA,CAAS,QAAA,CAAS,SAAS,CAAA;AACvD,YAAA,MAAM,aAAA,EAAe,KAAA,CAAM,MAAA;AAAA,cAC1B,CAAC,IAAA,EAAA,GAAS,IAAA,CAAK,GAAA,IAAO;AAAA,YACvB,CAAA,CAAE,CAAC,CAAA;AAEH,YAAA,MAAM,EAAE,KAAA,EAAO,YAAA,EAAc,IAAA,EAAM,EAAE,QAAA,EAAU,SAAS,EAAE,CAAA;AAE1D,YAAA,IAAI,WAAA,EAAa,EAAA;AAEjB,YAAA,GAAA,CACC,IAAA,CAAK,UAAA,GACL,YAAA,CAAa,SAAA,IAAA,eAAA,EACZ;AACD,cAAA,MAAMA,UAAAA,EAAW,IAAA,CAAK,KAAA,CAAM,QAAA,CAAS,QAAA,CAAS,SAAS,CAAA;AAGvD,cAAA,yBAAA,CAAQ,MAAA,CAAO,KAAA,CAAM,eAAA,EAAiB,EAAE,QAAA,EAAU,QAAA,EAAAA,UAAS,CAAC,CAAA;AAC5D,cAAA,MAAM,OAAA,EAAS,MAAM,IAAA,CAAK,SAAA,CAAU,OAAA;AAAA,gBACnC,YAAA;AAAA,gBACAA;AAAA,cACD,CAAA;AACA,cAAA,WAAA,EACC,CAAA,qBAAA,EAAwB,QAAQ,CAAA,WAAA,EAAc,IAAA,CAAK,SAAA,CAAUA,SAAQ,CAAC,CAAA;AAAA,EAAA,EACtE,IAAA,CAAK,SAAA,CAAU,MAAA,CAAO,OAAA,EAAS,IAAA,EAAM,CAAC,CAAA;AAAA,YACxC,EAAA,KAAA,GAAA,CACC,IAAA,CAAK,UAAA,GACL,YAAA,CAAa,SAAA,IAAA,eAAA,EACZ;AACD,cAAA,MAAM,OAAA,EAAS,MAAM,IAAA,CAAK,SAAA,CAAU,OAAA;AAAA,gBACnC,YAAA;AAAA,gBACA,cAAA;AAAA,gBACA;AAAA,cACD,CAAA;AACA,cAAA,WAAA,EAAa,CAAA,qBAAA,EAAwB,QAAQ,CAAA;AAAA,EAAM,MAAA,CAAO,IAAA,CAAK,IAAI,CAAC,CAAA,CAAA;AAC9D,YAAA;AAEN,cAAA;AACC,gBAAA;AACD,cAAA;AACA,cAAA;AACD,YAAA;AACM,YAAA;AACL,cAAA;AACM,cAAA;AACP,YAAA;AACQ,YAAA;AAER,YAAA;AACA,YAAA;AACD,UAAA;AACD,QAAA;AAEK,QAAA;AACN,MAAA;AACe,IAAA;AACP,MAAA;AACJ,MAAA;AACG,QAAA;AACE,UAAA;AACC,UAAA;AACT,QAAA;AACM,MAAA;AACA,QAAA;AACE,UAAA;AACC,UAAA;AACT,QAAA;AACD,MAAA;AACD,IAAA;AACD,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAca,EAAA;AAON,IAAA;AACA,IAAA;AACU,IAAA;AAKD,IAAA;AAEX,IAAA;AAEG,MAAA;AACL,QAAA;AACA,QAAA;AACW,QAAA;AACZ,MAAA;AAEI,MAAA;AACJ,MAAA;AACW,QAAA;AACD,UAAA;AACR,UAAA;AACD,QAAA;AAEM,QAAA;AAAyD,MAAA;AAAqB;AAAA;AAC1E,QAAA;AACX,MAAA;AACe,IAAA;AACP,MAAA;AACJ,MAAA;AACH,QAAA;AAA4B,MAAA;AAAuC;AAAA;AACpE,MAAA;AACC,IAAA;AACO,MAAA;AACT,IAAA;AACD,EAAA;AACD;AD3EoB;AACA;AACA;AACA","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-UQ72NNHV.cjs","sourcesContent":[null,"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"]}

package/dist/cjs/chunk-XZAKXJYO.cjs

@@ -0,0 +1,20 @@
+"use strict";Object.defineProperty(exports, "__esModule", {value: true});// src/types/memory.ts
+var MemoryType = /* @__PURE__ */ ((MemoryType2) => {
+  MemoryType2["SESSION"] = "SESSION";
+  MemoryType2["INTENT"] = "INTENT";
+  MemoryType2["AGENT"] = "AGENT";
+  MemoryType2["USER"] = "USER";
+  return MemoryType2;
+})(MemoryType || {});
+var ChatRole = /* @__PURE__ */ ((ChatRole2) => {
+  ChatRole2["USER"] = "USER";
+  ChatRole2["SYSTEM"] = "SYSTEM";
+  ChatRole2["MODEL"] = "MODEL";
+  return ChatRole2;
+})(ChatRole || {});
+
+
+
+
+exports.MemoryType = MemoryType; exports.ChatRole = ChatRole;
+//# sourceMappingURL=chunk-XZAKXJYO.cjs.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-XZAKXJYO.cjs","../../src/types/memory.ts"],"names":["MemoryType","ChatRole"],"mappings":"AAAA;ACAO,IAAK,WAAA,kBAAL,CAAA,CAAKA,WAAAA,EAAAA,GAAL;AACN,EAAAA,WAAAA,CAAA,SAAA,EAAA,EAAU,SAAA;AACV,EAAAA,WAAAA,CAAA,QAAA,EAAA,EAAS,QAAA;AACT,EAAAA,WAAAA,CAAA,OAAA,EAAA,EAAQ,OAAA;AACR,EAAAA,WAAAA,CAAA,MAAA,EAAA,EAAO,MAAA;AAJI,EAAA,OAAAA,WAAAA;AAAA,CAAA,CAAA,CAAA,WAAA,GAAA,CAAA,CAAA,CAAA;AAUL,IAAK,SAAA,kBAAL,CAAA,CAAKC,SAAAA,EAAAA,GAAL;AAEN,EAAAA,SAAAA,CAAA,MAAA,EAAA,EAAO,MAAA;AAEP,EAAAA,SAAAA,CAAA,QAAA,EAAA,EAAS,QAAA;AAET,EAAAA,SAAAA,CAAA,OAAA,EAAA,EAAQ,OAAA;AANG,EAAA,OAAAA,SAAAA;AAAA,CAAA,CAAA,CAAA,SAAA,GAAA,CAAA,CAAA,CAAA;ADIZ;AACA;AACE;AACA;AACF,6DAAC","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-XZAKXJYO.cjs","sourcesContent":[null,"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"]}

package/dist/cjs/chunk-ZAZ3KYP5.cjs

@@ -0,0 +1,43 @@
+"use strict";Object.defineProperty(exports, "__esModule", {value: true});
+
+var _chunkRPMERDHUcjs = require('./chunk-RPMERDHU.cjs');
+
+
+var _chunkUQ72NNHVcjs = require('./chunk-UQ72NNHV.cjs');
+
+
+var _chunkRWQDV2XYcjs = require('./chunk-RWQDV2XY.cjs');
+
+// src/routes/query.routes.ts
+var _express = require('express');
+var createQueryRouter = (agent, allowStream = false) => {
+  const router = _express.Router.call(void 0, );
+  const queryService = new (0, _chunkRWQDV2XYcjs.QueryService)(
+    agent.modelModule,
+    agent.a2aModule,
+    agent.mcpModule,
+    agent.memoryModule,
+    agent.manifest.prompts
+  );
+  let queryStreamService;
+  if (allowStream) {
+    queryStreamService = new (0, _chunkUQ72NNHVcjs.QueryStreamService)(
+      agent.modelModule,
+      agent.a2aModule,
+      agent.mcpModule,
+      agent.memoryModule,
+      agent.manifest.prompts
+    );
+  }
+  const queryController = new (0, _chunkRPMERDHUcjs.QueryController)(queryService, queryStreamService);
+  router.post("/query", queryController.handleQueryRequest);
+  if (allowStream) {
+    router.post("/query-stream", queryController.handleQueryStreamRequest);
+  }
+  return router;
+};
+
+
+
+exports.createQueryRouter = createQueryRouter;
+//# sourceMappingURL=chunk-ZAZ3KYP5.cjs.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-ZAZ3KYP5.cjs","../../src/routes/query.routes.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACA;ACVA,kCAAuB;AAMhB,IAAM,kBAAA,EAAoB,CAChC,KAAA,EACA,YAAA,EAAc,KAAA,EAAA,GACF;AACZ,EAAA,MAAM,OAAA,EAAS,6BAAA,CAAO;AAEtB,EAAA,MAAM,aAAA,EAAe,IAAI,mCAAA;AAAA,IACxB,KAAA,CAAM,WAAA;AAAA,IACN,KAAA,CAAM,SAAA;AAAA,IACN,KAAA,CAAM,SAAA;AAAA,IACN,KAAA,CAAM,YAAA;AAAA,IACN,KAAA,CAAM,QAAA,CAAS;AAAA,EAChB,CAAA;AAEA,EAAA,IAAI,kBAAA;AAEJ,EAAA,GAAA,CAAI,WAAA,EAAa;AAChB,IAAA,mBAAA,EAAqB,IAAI,yCAAA;AAAA,MACxB,KAAA,CAAM,WAAA;AAAA,MACN,KAAA,CAAM,SAAA;AAAA,MACN,KAAA,CAAM,SAAA;AAAA,MACN,KAAA,CAAM,YAAA;AAAA,MACN,KAAA,CAAM,QAAA,CAAS;AAAA,IAChB,CAAA;AAAA,EACD;AAEA,EAAA,MAAM,gBAAA,EAAkB,IAAI,sCAAA,CAAgB,YAAA,EAAc,kBAAkB,CAAA;AAC5E,EAAA,MAAA,CAAO,IAAA,CAAK,QAAA,EAAU,eAAA,CAAgB,kBAAkB,CAAA;AAExD,EAAA,GAAA,CAAI,WAAA,EAAa;AAChB,IAAA,MAAA,CAAO,IAAA,CAAK,eAAA,EAAiB,eAAA,CAAgB,wBAAwB,CAAA;AAAA,EACtE;AAEA,EAAA,OAAO,MAAA;AACR,CAAA;ADFA;AACA;AACE;AACF,8CAAC","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-ZAZ3KYP5.cjs","sourcesContent":[null,"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"]}

package/dist/cjs/controllers/a2a.controller.cjs

@@ -1,8 +1,8 @@
 "use strict";Object.defineProperty(exports, "__esModule", {value: true});
 
-var _chunkILGIEJCXcjs = require('../chunk-ILGIEJCX.cjs');
-require('../chunk-JTAXZJQ2.cjs');
+var _chunkICFTUZP5cjs = require('../chunk-ICFTUZP5.cjs');
+require('../chunk-466ED5WN.cjs');
 
 
-exports.A2AController = _chunkILGIEJCXcjs.A2AController;
+exports.A2AController = _chunkICFTUZP5cjs.A2AController;
 //# sourceMappingURL=a2a.controller.cjs.map

package/dist/cjs/controllers/index.cjs

@@ -1,21 +1,21 @@
 "use strict";Object.defineProperty(exports, "__esModule", {value: true});
 
-var _chunkRGKCBQGZcjs = require('../chunk-RGKCBQGZ.cjs');
+var _chunkV5RSN2DYcjs = require('../chunk-V5RSN2DY.cjs');
 
 
-var _chunkILGIEJCXcjs = require('../chunk-ILGIEJCX.cjs');
-require('../chunk-JTAXZJQ2.cjs');
+var _chunkNXKC2HVXcjs = require('../chunk-NXKC2HVX.cjs');
 
 
-var _chunkV5RSN2DYcjs = require('../chunk-V5RSN2DY.cjs');
+var _chunkICFTUZP5cjs = require('../chunk-ICFTUZP5.cjs');
 
 
-var _chunkNXKC2HVXcjs = require('../chunk-NXKC2HVX.cjs');
+var _chunkRPMERDHUcjs = require('../chunk-RPMERDHU.cjs');
 require('../chunk-DX2JH3FJ.cjs');
+require('../chunk-466ED5WN.cjs');
 
 
 
 
 
-exports.A2AController = _chunkILGIEJCXcjs.A2AController; exports.ModelApiController = _chunkV5RSN2DYcjs.ModelApiController; exports.QueryController = _chunkRGKCBQGZcjs.QueryController; exports.SessionApiController = _chunkNXKC2HVXcjs.SessionApiController;
+exports.A2AController = _chunkICFTUZP5cjs.A2AController; exports.ModelApiController = _chunkV5RSN2DYcjs.ModelApiController; exports.QueryController = _chunkRPMERDHUcjs.QueryController; exports.SessionApiController = _chunkNXKC2HVXcjs.SessionApiController;
 //# sourceMappingURL=index.cjs.map

package/dist/cjs/controllers/index.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/controllers/index.cjs"],"names":[],"mappings":"AAAA;AACE;AACF,yDAA8B;AAC9B;AACE;AACF,yDAA8B;AAC9B,iCAA8B;AAC9B;AACE;AACF,yDAA8B;AAC9B;AACE;AACF,yDAA8B;AAC9B,iCAA8B;AAC9B;AACE;AACA;AACA;AACA;AACF,+PAAC","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/controllers/index.cjs"}
+{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/controllers/index.cjs"],"names":[],"mappings":"AAAA;AACE;AACF,yDAA8B;AAC9B;AACE;AACF,yDAA8B;AAC9B;AACE;AACF,yDAA8B;AAC9B;AACE;AACF,yDAA8B;AAC9B,iCAA8B;AAC9B,iCAA8B;AAC9B;AACE;AACA;AACA;AACA;AACF,+PAAC","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/controllers/index.cjs"}

package/dist/cjs/controllers/query.controller.cjs

@@ -1,7 +1,7 @@
 "use strict";Object.defineProperty(exports, "__esModule", {value: true});
 
-var _chunkRGKCBQGZcjs = require('../chunk-RGKCBQGZ.cjs');
+var _chunkRPMERDHUcjs = require('../chunk-RPMERDHU.cjs');
 
 
-exports.QueryController = _chunkRGKCBQGZcjs.QueryController;
+exports.QueryController = _chunkRPMERDHUcjs.QueryController;
 //# sourceMappingURL=query.controller.cjs.map

package/dist/cjs/index.cjs

@@ -1,33 +1,34 @@
 "use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; } function _nullishCoalesce(lhs, rhsFn) { if (lhs != null) { return lhs; } else { return rhsFn(); } } function _optionalChain(ops) { let lastAccessLHS = undefined; let value = ops[0]; let i = 1; while (i < ops.length) { const op = ops[i]; const fn = ops[i + 1]; i += 2; if ((op === 'optionalAccess' || op === 'optionalCall') && value == null) { return undefined; } if (op === 'access' || op === 'optionalAccess') { lastAccessLHS = value; value = fn(value); } else if (op === 'call' || op === 'optionalCall') { value = fn((...args) => value.call(lastAccessLHS, ...args)); lastAccessLHS = undefined; } } return value; } var _class;require('./chunk-SSSDO7PA.cjs');
 
 
-var _chunkS3RQLMWYcjs = require('./chunk-S3RQLMWY.cjs');
-require('./chunk-RGKCBQGZ.cjs');
+var _chunk6S6GEY3Dcjs = require('./chunk-6S6GEY3D.cjs');
 
 
-var _chunkLRBRG2DScjs = require('./chunk-LRBRG2DS.cjs');
+var _chunkVBXOEQ3Ncjs = require('./chunk-VBXOEQ3N.cjs');
+require('./chunk-N7NGEMKL.cjs');
+require('./chunk-GBHMYJ6R.cjs');
 
 
-var _chunkDYRSSHDIcjs = require('./chunk-DYRSSHDI.cjs');
+var _chunkZAZ3KYP5cjs = require('./chunk-ZAZ3KYP5.cjs');
+require('./chunk-V5RSN2DY.cjs');
+require('./chunk-NXKC2HVX.cjs');
+require('./chunk-ICFTUZP5.cjs');
+require('./chunk-RPMERDHU.cjs');
 
 
-var _chunkQVIS5ELAcjs = require('./chunk-QVIS5ELA.cjs');
-require('./chunk-QUWZWKSZ.cjs');
-require('./chunk-3IGIYVGA.cjs');
-require('./chunk-LPXYZH22.cjs');
-require('./chunk-7BZO57TG.cjs');
-require('./chunk-ILGIEJCX.cjs');
+var _chunkLRBRG2DScjs = require('./chunk-LRBRG2DS.cjs');
+require('./chunk-DX2JH3FJ.cjs');
 
 
-var _chunkJTAXZJQ2cjs = require('./chunk-JTAXZJQ2.cjs');
+var _chunkNYYEZ4ERcjs = require('./chunk-NYYEZ4ER.cjs');
+require('./chunk-UQ72NNHV.cjs');
+require('./chunk-RWQDV2XY.cjs');
+require('./chunk-XZAKXJYO.cjs');
+require('./chunk-LPXYZH22.cjs');
+require('./chunk-3JKRJYJ6.cjs');
 
 
-var _chunkVBXOEQ3Ncjs = require('./chunk-VBXOEQ3N.cjs');
-require('./chunk-N7NGEMKL.cjs');
-require('./chunk-GBHMYJ6R.cjs');
-require('./chunk-V5RSN2DY.cjs');
-require('./chunk-NXKC2HVX.cjs');
-require('./chunk-DX2JH3FJ.cjs');
+var _chunk466ED5WNcjs = require('./chunk-466ED5WN.cjs');
 
 // src/index.ts
 var _cors = require('cors'); var _cors2 = _interopRequireDefault(_cors);
@@ -601,7 +602,7 @@ var AINAgent = (_class = class {
    * @param modules.folModule - Optional module for fol management
    * @param authScheme - Optional authentication middleware for securing endpoints
    */
-  constructor(manifest, modules, authScheme) {;_class.prototype.__init.call(this);_class.prototype.__init2.call(this);
+  constructor(manifest, modules, authScheme, allowStream = false) {;_class.prototype.__init.call(this);_class.prototype.__init2.call(this);
     this.app = _express2.default.call(void 0, );
     this.manifest = manifest;
     this.modelModule = modules.modelModule;
@@ -610,8 +611,8 @@ var AINAgent = (_class = class {
     this.memoryModule = modules.memoryModule;
     this.authScheme = authScheme;
     this.initializeMiddlewares();
-    this.initializeRoutes();
-    this.app.use(_chunkDYRSSHDIcjs.errorMiddleware);
+    this.initializeRoutes(allowStream);
+    this.app.use(_chunkNYYEZ4ERcjs.errorMiddleware);
   }
   /**
    * Initializes Express middlewares for security, CORS, and body parsing.
@@ -686,7 +687,7 @@ var AINAgent = (_class = class {
    * - /query/* - Query processing endpoints
    * - /a2a/* - A2A protocol endpoints (only if valid URL is configured)
    */
-  __init2() {this.initializeRoutes = () => {
+  __init2() {this.initializeRoutes = (allowStream = false) => {
     this.app.get("/", async (_, res) => {
       const { name, description, version } = this.manifest;
       res.status(200).send(
@@ -705,10 +706,10 @@ var AINAgent = (_class = class {
         res.status(_httpstatuscodes.StatusCodes.NOT_FOUND).send("No agent card");
       }
     });
-    this.app.use(_chunkS3RQLMWYcjs.createQueryRouter.call(void 0, this));
+    this.app.use(_chunkZAZ3KYP5cjs.createQueryRouter.call(void 0, this, allowStream));
     this.app.use(_chunkVBXOEQ3Ncjs.createApiRouter.call(void 0, this));
     if (this.isValidUrl(this.manifest.url)) {
-      this.app.use(_chunkQVIS5ELAcjs.createA2ARouter.call(void 0, this));
+      this.app.use(_chunk6S6GEY3Dcjs.createA2ARouter.call(void 0, this));
     }
   }}
   /**
@@ -719,7 +720,7 @@ var AINAgent = (_class = class {
   async start(port) {
     await _optionalChain([this, 'access', _4 => _4.memoryModule, 'optionalAccess', _5 => _5.initialize, 'call', _6 => _6()]);
     this.app.listen(port, () => {
-      _chunkJTAXZJQ2cjs.loggers.agent.info(`AINAgent is running on port ${port}`);
+      _chunk466ED5WNcjs.loggers.agent.info(`AINAgent is running on port ${port}`);
     });
   }
 }, _class);