@ainetwork/adk 0.1.7 → 0.1.9

package/dist/esm/chunk-3NEV3O2X.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/services/query-stream.service.ts"],"sourcesContent":["import { randomUUID } from \"node:crypto\";\nimport 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 {\n\tChatRole,\n\ttype SessionMetadata,\n\ttype SessionObject,\n} from \"@/types/memory.js\";\nimport type { StreamEvent } from \"@/types/stream\";\nimport {\n\ttype IA2ATool,\n\ttype IAgentTool,\n\ttype IMCPTool,\n\tTOOL_PROTOCOL_TYPE,\n} from \"@/types/tool.js\";\nimport { loggers } from \"@/utils/logger.js\";\n\n/**\n * Service for processing user queries through the agent's AI pipeline.\n *\n * Orchestrates the query processing workflow including intent detection,\n * model inference, tool execution, and response generation. Manages\n * conversation context and coordinates between different modules.\n */\nexport class QueryStreamService {\n\tprivate modelModule: ModelModule;\n\tprivate a2aModule?: A2AModule;\n\tprivate mcpModule?: MCPModule;\n\tprivate memoryModule?: MemoryModule;\n\tprivate prompts?: AinAgentPrompts;\n\n\tconstructor(\n\t\tmodelModule: ModelModule,\n\t\ta2aModule?: A2AModule,\n\t\tmcpModule?: MCPModule,\n\t\tmemoryModule?: MemoryModule,\n\t\tprompts?: AinAgentPrompts,\n\t) {\n\t\tthis.modelModule = modelModule;\n\t\tthis.a2aModule = a2aModule;\n\t\tthis.mcpModule = mcpModule;\n\t\tthis.memoryModule = memoryModule;\n\t\tthis.prompts = prompts;\n\t}\n\n\t/**\n\t * Detects the intent from 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\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\ttoolResult = 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} 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\ttoolResult = 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} 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\n\t/**\n\t * Generates a title for the conversation based on the query.\n\t *\n\t * @param query - The user's input query\n\t * @returns Promise resolving to a generated title\n\t */\n\n\tprivate async generateTitle(query: string): Promise<string> {\n\t\tconst DEFAULT_TITLE = \"New Chat\";\n\t\ttry {\n\t\t\tconst modelInstance = this.modelModule.getModel();\n\t\t\tconst messages = modelInstance.generateMessages({\n\t\t\t\tquery,\n\t\t\t\tsystemPrompt: `You are a helpful assistant that generates titles for conversations.\n  Please analyze the user's query and create a concise title that accurately reflects the conversation's core topic.\n  The title must be no more than 5 words long.\n  Respond with only the title. Do not include any punctuation or extra explanations.`,\n\t\t\t});\n\t\t\tconst response = await modelInstance.fetch(messages);\n\t\t\treturn response.content || DEFAULT_TITLE;\n\t\t} catch (error) {\n\t\t\tloggers.intentStream.error(\"Error generating title\", {\n\t\t\t\terror,\n\t\t\t\tquery,\n\t\t\t});\n\t\t\treturn DEFAULT_TITLE;\n\t\t}\n\t}\n\n\t/**\n\t * Main entry point for processing user queries.\n\t *\n\t * Handles the complete query lifecycle:\n\t * 1. Loads 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\tuserId: string,\n\t\tres: Response,\n\t\t_sessionId?: string,\n\t) {\n\t\t// 1. Load session history with sessionId\n\t\tlet sessionId = _sessionId;\n\t\tconst queryStartAt = Date.now();\n\t\tconst sessionMemory = this.memoryModule?.getSessionMemory();\n\t\tconst session =\n\t\t\t!userId || !sessionId\n\t\t\t\t? undefined\n\t\t\t\t: await sessionMemory?.getSession(userId, sessionId);\n\n\t\ttry {\n\t\t\tif (!sessionId) {\n\t\t\t\tsessionId = randomUUID();\n\t\t\t\tconst title = await this.generateTitle(query);\n\n\t\t\t\tconst metadata =\n\t\t\t\t\t(await sessionMemory?.createSession(userId, sessionId, title)) ||\n\t\t\t\t\t({\n\t\t\t\t\t\tsessionId,\n\t\t\t\t\t\ttitle,\n\t\t\t\t\t\tupdatedAt: Date.now(),\n\t\t\t\t\t} as SessionMetadata);\n\t\t\t\tloggers.intentStream.info(\"Create new session\", { metadata });\n\t\t\t\tres.write(`event: session_id\\ndata: ${JSON.stringify(metadata)}\\n\\n`);\n\t\t\t}\n\t\t} catch (error) {\n\t\t\tthrow new Error(\"Failed to create new session\");\n\t\t}\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(query, sessionId, session);\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\n\t\t\tif (userId) {\n\t\t\t\tawait sessionMemory?.addChatToSession(userId, sessionId, {\n\t\t\t\t\trole: ChatRole.USER,\n\t\t\t\t\ttimestamp: queryStartAt,\n\t\t\t\t\tcontent: { type: \"text\", parts: [query] },\n\t\t\t\t});\n\t\t\t\tawait sessionMemory?.addChatToSession(userId, sessionId, {\n\t\t\t\t\trole: ChatRole.MODEL,\n\t\t\t\t\ttimestamp: Date.now(),\n\t\t\t\t\tcontent: { type: \"text\", parts: [finalResponseText] },\n\t\t\t\t});\n\t\t\t}\n\t\t} catch (error) {\n\t\t\tloggers.intent.error(\"Error in handleQuery\", { error });\n\t\t\tres.write(\n\t\t\t\t`event: error\\ndata: ${JSON.stringify({ message: \"Stream failed\" })}\\n\\n`,\n\t\t\t);\n\t\t} finally {\n\t\t\tres.end();\n\t\t}\n\t}\n}\n"],"mappings":";;;;;AAAA,SAAS,kBAAkB;AA8BpB,IAAM,qBAAN,MAAyB;AAAA,EACvB;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EAER,YACC,aACA,WACA,WACA,cACA,SACC;AACD,SAAK,cAAc;AACnB,SAAK,YAAY;AACjB,SAAK,YAAY;AACjB,SAAK,eAAe;AACpB,SAAK,UAAU;AAAA,EAChB;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EASA,MAAc,iBAAiB,OAAe;AAE7C,WAAO;AAAA,EACR;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAgBA,OAAc,iBACb,OACA,WACA,gBAC8B;AAC9B,QAAI;AACH,YAAM,eAAe;AAAA,YACb,oBAAI,KAAK,GAAE,mBAAmB,CAAC;AAAA;AAAA,EAExC,KAAK,SAAS,SAAS,EAAE;AAAA;AAAA,EAEzB,KAAK,SAAS,UAAU,EAAE;AAAA;AAGzB,YAAM,gBAAgB,KAAK,YAAY,SAAS;AAChD,YAAM,WAAW,cAAc,iBAAiB;AAAA,QAC/C;AAAA,QACA;AAAA,QACA,cAAc,aAAa,KAAK;AAAA,MACjC,CAAC;AAED,YAAM,QAAsB,CAAC;AAC7B,UAAI,KAAK,WAAW;AACnB,cAAM,KAAK,GAAG,KAAK,UAAU,SAAS,CAAC;AAAA,MACxC;AACA,UAAI,KAAK,WAAW;AACnB,cAAM,KAAK,GAAI,MAAM,KAAK,UAAU,SAAS,CAAE;AAAA,MAChD;AACA,YAAM,YAAY,cAAc,wBAAwB,KAAK;AAE7D,YAAM,cAAwB,CAAC;AAC/B,UAAI,cAAc;AAElB,aAAO,MAAM;AACZ,cAAM,iBACL,MAAM,cAAc;AAAA,UACnB;AAAA,UACA;AAAA,QACD;AACD,sBAAc;AAEd,cAAM,qBAIA,CAAC;AAEP,gBAAQ,aAAa,MAAM,YAAY,EAAE,SAAS,CAAC;AAEnD,yBAAiB,SAAS,gBAAgB;AACzC,gBAAM,QAAQ,MAAM;AACpB,cAAI,OAAO,YAAY;AACtB,0BAAc;AACd,uBAAW,iBAAiB,MAAM,YAAY;AAC7C,oBAAM,QAAQ,cAAc;AAE5B,kBAAI,CAAC,mBAAmB,KAAK,GAAG;AAC/B,mCAAmB,KAAK,IAAI;AAAA,kBAC3B,IAAI;AAAA,kBACJ,MAAM;AAAA,kBACN,UAAU,EAAE,MAAM,IAAI,WAAW,GAAG;AAAA,gBACrC;AAAA,cACD;AAEA,kBAAI,cAAc,IAAI;AACrB,mCAAmB,KAAK,EAAE,KAAK,cAAc;AAAA,cAC9C;AAEA,kBAAI,cAAc,UAAU,MAAM;AACjC,mCAAmB,KAAK,EAAE,SAAS,OAClC,cAAc,SAAS;AAAA,cACzB;AAEA,kBAAI,cAAc,UAAU,WAAW;AACtC,mCAAmB,KAAK,EAAE,SAAS,aAClC,cAAc,SAAS;AAAA,cACzB;AAAA,YACD;AAAA,UACD,WAAW,MAAM,OAAO,SAAS;AAChC,kBAAM;AAAA,cACL,OAAO;AAAA,cACP,MAAM,EAAE,OAAO,MAAM,MAAM,QAAQ;AAAA,YACpC;AAAA,UACD;AAAA,QACD;AAEA,gBAAQ,aAAa,MAAM,sBAAsB;AAAA,UAChD;AAAA,QACD,CAAC;AAED,YAAI,eAAe,mBAAmB,SAAS,GAAG;AACjD,gBAAM,iBAAiB,KAAK,WAAW;AAAA,YACtC;AAAA,YACA;AAAA,UACD;AACA,qBAAW,YAAY,oBAAoB;AAC1C,kBAAM,WAAW,SAAS,SAAS;AACnC,kBAAM,WAAW,KAAK,MAAM,SAAS,SAAS,SAAS;AACvD,kBAAM,eAAe,MAAM;AAAA,cAC1B,CAAC,SAAS,KAAK,OAAO;AAAA,YACvB,EAAE,CAAC;AAEH,kBAAM,EAAE,OAAO,cAAc,MAAM,EAAE,UAAU,SAAS,EAAE;AAE1D,gBAAI,aAAa;AAEjB,gBACC,KAAK,aACL,aAAa,8BACZ;AACD,oBAAMA,YAAW,KAAK,MAAM,SAAS,SAAS,SAAS;AAGvD,sBAAQ,OAAO,MAAM,iBAAiB,EAAE,UAAU,UAAAA,UAAS,CAAC;AAC5D,2BAAa,MAAM,KAAK,UAAU;AAAA,gBACjC;AAAA,gBACAA;AAAA,cACD;AAAA,YACD,WACC,KAAK,aACL,aAAa,8BACZ;AACD,2BAAa,MAAM,KAAK,UAAU;AAAA,gBACjC;AAAA,gBACA;AAAA,gBACA;AAAA,cACD;AAAA,YACD,OAAO;AAEN,sBAAQ,OAAO;AAAA,gBACd,2BAA2B,aAAa,QAAQ;AAAA,cACjD;AACA;AAAA,YACD;AACA,kBAAM;AAAA,cACL,OAAO;AAAA,cACP,MAAM,EAAE,UAAU,QAAQ,WAAW;AAAA,YACtC;AACA,oBAAQ,OAAO,MAAM,cAAc,EAAE,WAAW,CAAC;AAEjD,wBAAY,KAAK,UAAU;AAC3B,0BAAc,eAAe,UAAU,UAAU;AAAA,UAClD;AAAA,QACD;AAEA,YAAI,CAAC,YAAa;AAAA,MACnB;AAAA,IACD,SAAS,OAAO;AACf,cAAQ,OAAO,MAAM,uCAAuC,EAAE,MAAM,CAAC;AACrE,UAAI,iBAAiB,OAAO;AAC3B,cAAM;AAAA,UACL,OAAO;AAAA,UACP,MAAM,EAAE,SAAS,MAAM,WAAW,6BAA6B;AAAA,QAChE;AAAA,MACD,OAAO;AACN,cAAM;AAAA,UACL,OAAO;AAAA,UACP,MAAM,EAAE,SAAS,6BAA6B;AAAA,QAC/C;AAAA,MACD;AAAA,IACD;AAAA,EACD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,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,EAeA,MAAa,kBACZ,OACA,QACA,KACA,YACC;AAED,QAAI,YAAY;AAChB,UAAM,eAAe,KAAK,IAAI;AAC9B,UAAM,gBAAgB,KAAK,cAAc,iBAAiB;AAC1D,UAAM,UACL,CAAC,UAAU,CAAC,YACT,SACA,MAAM,eAAe,WAAW,QAAQ,SAAS;AAErD,QAAI;AACH,UAAI,CAAC,WAAW;AACf,oBAAY,WAAW;AACvB,cAAM,QAAQ,MAAM,KAAK,cAAc,KAAK;AAE5C,cAAM,WACJ,MAAM,eAAe,cAAc,QAAQ,WAAW,KAAK,KAC3D;AAAA,UACA;AAAA,UACA;AAAA,UACA,WAAW,KAAK,IAAI;AAAA,QACrB;AACD,gBAAQ,aAAa,KAAK,sBAAsB,EAAE,SAAS,CAAC;AAC5D,YAAI,MAAM;AAAA,QAA4B,KAAK,UAAU,QAAQ,CAAC;AAAA;AAAA,CAAM;AAAA,MACrE;AAAA,IACD,SAAS,OAAO;AACf,YAAM,IAAI,MAAM,8BAA8B;AAAA,IAC/C;AAGA,UAAM,SAAS,KAAK,iBAAiB,KAAK;AAE1C,QAAI;AAEH,YAAM,SAAS,MAAM,KAAK,iBAAiB,OAAO,WAAW,OAAO;AAEpE,UAAI,oBAAoB;AACxB,uBAAiB,SAAS,QAAQ;AACjC,YAAI,MAAM,UAAU,gBAAgB,MAAM,KAAK,OAAO;AACrD,kBAAQ,aAAa,MAAM,cAAc,EAAE,MAAM,CAAC;AAClD,+BAAqB,MAAM,KAAK;AAAA,QACjC;AAEA,cAAM,oBAAoB,UAAU,MAAM,KAAK;AAAA,QAAW,KAAK,UAAU,MAAM,IAAI,CAAC;AAAA;AAAA;AACpF,YAAI,MAAM,iBAAiB;AAAA,MAC5B;AAEA,UAAI,QAAQ;AACX,cAAM,eAAe,iBAAiB,QAAQ,WAAW;AAAA,UACxD;AAAA,UACA,WAAW;AAAA,UACX,SAAS,EAAE,MAAM,QAAQ,OAAO,CAAC,KAAK,EAAE;AAAA,QACzC,CAAC;AACD,cAAM,eAAe,iBAAiB,QAAQ,WAAW;AAAA,UACxD;AAAA,UACA,WAAW,KAAK,IAAI;AAAA,UACpB,SAAS,EAAE,MAAM,QAAQ,OAAO,CAAC,iBAAiB,EAAE;AAAA,QACrD,CAAC;AAAA,MACF;AAAA,IACD,SAAS,OAAO;AACf,cAAQ,OAAO,MAAM,wBAAwB,EAAE,MAAM,CAAC;AACtD,UAAI;AAAA,QACH;AAAA,QAAuB,KAAK,UAAU,EAAE,SAAS,gBAAgB,CAAC,CAAC;AAAA;AAAA;AAAA,MACpE;AAAA,IACD,UAAE;AACD,UAAI,IAAI;AAAA,IACT;AAAA,EACD;AACD;","names":["toolArgs"]}

package/dist/esm/{chunk-624FPEEV.js → chunk-BDMVIJPF.js}

@@ -69,17 +69,23 @@ var MCPModule = class {
     const { serverName, mcpTool } = tool;
     const toolName = mcpTool.name;
     const mcp = this.mcpMap.get(serverName);
-    if (!mcp) {
-      throw new Error(`Invalid MCP Tool ${serverName}-${mcpTool.name}`);
-    }
-    const result = await mcp.callTool({
-      name: toolName,
-      arguments: _args
-    });
-    const toolResult = `[Bot Called Tool ${toolName} with args ${JSON.stringify(_args)}]
+    try {
+      if (!mcp) {
+        throw new Error(`Invalid MCP Tool ${serverName}-${mcpTool.name}`);
+      }
+      const result = await mcp.callTool({
+        name: toolName,
+        arguments: _args
+      });
+      const toolResult = `[Bot Called Tool ${toolName} with args ${JSON.stringify(_args)}]
 ` + JSON.stringify(result.content, null, 2);
-    loggers.mcp.debug("MCP useTool result:", toolResult);
-    return result;
+      return toolResult;
+    } catch (error) {
+      loggers.mcp.error("Failed to call tool", { error });
+      const toolResult = `[Bot Called Tool ${toolName} with args ${JSON.stringify(_args)}]
+${typeof error === "string" ? error : JSON.stringify(error, null, 2)}`;
+      return toolResult;
+    }
   }
   /**
    * Closes all MCP client connections.
@@ -97,4 +103,4 @@ var MCPModule = class {
 export {
   MCPModule
 };
-//# sourceMappingURL=chunk-624FPEEV.js.map
+//# sourceMappingURL=chunk-BDMVIJPF.js.map

package/dist/esm/{chunk-624FPEEV.js.map → chunk-BDMVIJPF.js.map}

@@ -1 +1 @@
-{"version":3,"sources":["../../src/modules/mcp/mcp.module.ts"],"sourcesContent":["import { Client } from \"@modelcontextprotocol/sdk/client/index.js\";\nimport { StdioClientTransport } from \"@modelcontextprotocol/sdk/client/stdio.js\";\nimport type { MCPConfig } from \"@/types/mcp.js\";\nimport { loggers } from \"@/utils/logger.js\";\nimport { MCPTool } from \"./mcp.tool.js\";\n\n/**\n * Module for managing Model Context Protocol (MCP) server connections.\n *\n * This module handles the lifecycle of MCP client connections, discovers\n * available tools from connected servers, and provides an interface for\n * executing those tools. Multiple MCP servers can be connected simultaneously.\n *\n * @example\n * ```typescript\n * const mcpModule = new MCPModule();\n * await mcpModule.addMCPConfig({\n *   \"filesystem\": {\n *     command: \"npx\",\n *     args: [\"@modelcontextprotocol/server-filesystem\", \"/path/to/files\"]\n *   }\n * });\n *\n * const tools = mcpModule.getTools();\n * const result = await mcpModule.useTool(tools[0], { path: \"/example.txt\" });\n * ```\n */\nexport class MCPModule {\n\t/** Map of MCP server names to their client instances */\n\tprivate mcpMap: Map<string, Client> = new Map();\n\t/** Map of MCP server names to their transport instances */\n\tprivate transportMap: Map<string, StdioClientTransport> = new Map();\n\t/** Array of all discovered tools from connected MCP servers */\n\tprivate tools: MCPTool[] = [];\n\n\t/**\n\t * Connects to MCP servers based on the provided configuration.\n\t *\n\t * For each server in the config, establishes a connection, discovers\n\t * available tools, and adds them to the module's tool collection.\n\t * Skips servers that are already connected.\n\t *\n\t * @param mcpConfig - Configuration object mapping server names to connection details\n\t * @throws Error if connection to any MCP server fails\n\t */\n\tasync addMCPConfig(mcpConfig: MCPConfig) {\n\t\ttry {\n\t\t\tfor (const [name, conf] of Object.entries(mcpConfig)) {\n\t\t\t\t// FIXME(yoojin): Need strict duplication check.\n\t\t\t\tif (this.mcpMap.get(name) && this.transportMap.get(name)) continue; // Duplicated mcp: skip\n\n\t\t\t\tconst transport = new StdioClientTransport(conf);\n\t\t\t\tthis.transportMap.set(name, transport);\n\t\t\t\tconst mcp = new Client({ name: \"mcp-client-cli\", version: \"1.0.0\" });\n\t\t\t\tawait mcp.connect(transport);\n\t\t\t\tthis.mcpMap.set(name, mcp);\n\n\t\t\t\tconst toolsResult = await mcp.listTools();\n\t\t\t\tthis.tools.push(\n\t\t\t\t\t...toolsResult.tools.map((tool) => {\n\t\t\t\t\t\treturn new MCPTool(name, tool);\n\t\t\t\t\t}),\n\t\t\t\t);\n\t\t\t}\n\t\t\tloggers.mcp.info(\"Connected to MCP server with tools:\", {\n\t\t\t\ttools: this.tools.map((tool) => tool.id),\n\t\t\t});\n\t\t} catch (error: unknown) {\n\t\t\tloggers.mcp.error(\"Failed to connect to MCP server:\", { error });\n\t\t\tthrow error;\n\t\t}\n\t}\n\n\t/**\n\t * Returns all available tools from connected MCP servers.\n\t *\n\t * @returns Array of MCPTool instances representing available tools\n\t */\n\tgetTools() {\n\t\treturn this.tools;\n\t}\n\n\t/**\n\t * Executes a tool on its corresponding MCP server.\n\t *\n\t * @param tool - The MCPTool instance to execute\n\t * @param _args - Arguments to pass to the tool\n\t * @returns Promise resolving to the tool's execution result\n\t * @throws Error if the MCP server for the tool is not found\n\t */\n\tasync useTool(tool: MCPTool, _args?: any): Promise<any> {\n\t\tconst { serverName, mcpTool } = tool;\n\t\tconst toolName = mcpTool.name;\n\t\tconst mcp = this.mcpMap.get(serverName);\n\n\t\tif (!mcp) {\n\t\t\tthrow new Error(`Invalid MCP Tool ${serverName}-${mcpTool.name}`);\n\t\t}\n\n\t\tconst result = await mcp.callTool({\n\t\t\tname: toolName,\n\t\t\targuments: _args,\n\t\t});\n\t\tconst toolResult =\n\t\t\t`[Bot Called Tool ${toolName} with args ${JSON.stringify(_args)}]\\n` +\n\t\t\tJSON.stringify(result.content, null, 2);\n\n\t\tloggers.mcp.debug(\"MCP useTool result:\", toolResult);\n\t\treturn result;\n\t}\n\n\t/**\n\t * Closes all MCP client connections.\n\t *\n\t * Should be called when shutting down the application to ensure\n\t * all MCP connections are properly closed.\n\t */\n\tasync cleanup() {\n\t\tthis.mcpMap.forEach((mcp: Client) => {\n\t\t\tmcp.close();\n\t\t});\n\t}\n}\n"],"mappings":";;;;;;;;AAAA,SAAS,cAAc;AACvB,SAAS,4BAA4B;AA0B9B,IAAM,YAAN,MAAgB;AAAA;AAAA,EAEd,SAA8B,oBAAI,IAAI;AAAA;AAAA,EAEtC,eAAkD,oBAAI,IAAI;AAAA;AAAA,EAE1D,QAAmB,CAAC;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAY5B,MAAM,aAAa,WAAsB;AACxC,QAAI;AACH,iBAAW,CAAC,MAAM,IAAI,KAAK,OAAO,QAAQ,SAAS,GAAG;AAErD,YAAI,KAAK,OAAO,IAAI,IAAI,KAAK,KAAK,aAAa,IAAI,IAAI,EAAG;AAE1D,cAAM,YAAY,IAAI,qBAAqB,IAAI;AAC/C,aAAK,aAAa,IAAI,MAAM,SAAS;AACrC,cAAM,MAAM,IAAI,OAAO,EAAE,MAAM,kBAAkB,SAAS,QAAQ,CAAC;AACnE,cAAM,IAAI,QAAQ,SAAS;AAC3B,aAAK,OAAO,IAAI,MAAM,GAAG;AAEzB,cAAM,cAAc,MAAM,IAAI,UAAU;AACxC,aAAK,MAAM;AAAA,UACV,GAAG,YAAY,MAAM,IAAI,CAAC,SAAS;AAClC,mBAAO,IAAI,QAAQ,MAAM,IAAI;AAAA,UAC9B,CAAC;AAAA,QACF;AAAA,MACD;AACA,cAAQ,IAAI,KAAK,uCAAuC;AAAA,QACvD,OAAO,KAAK,MAAM,IAAI,CAAC,SAAS,KAAK,EAAE;AAAA,MACxC,CAAC;AAAA,IACF,SAAS,OAAgB;AACxB,cAAQ,IAAI,MAAM,oCAAoC,EAAE,MAAM,CAAC;AAC/D,YAAM;AAAA,IACP;AAAA,EACD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,WAAW;AACV,WAAO,KAAK;AAAA,EACb;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAUA,MAAM,QAAQ,MAAe,OAA2B;AACvD,UAAM,EAAE,YAAY,QAAQ,IAAI;AAChC,UAAM,WAAW,QAAQ;AACzB,UAAM,MAAM,KAAK,OAAO,IAAI,UAAU;AAEtC,QAAI,CAAC,KAAK;AACT,YAAM,IAAI,MAAM,oBAAoB,UAAU,IAAI,QAAQ,IAAI,EAAE;AAAA,IACjE;AAEA,UAAM,SAAS,MAAM,IAAI,SAAS;AAAA,MACjC,MAAM;AAAA,MACN,WAAW;AAAA,IACZ,CAAC;AACD,UAAM,aACL,oBAAoB,QAAQ,cAAc,KAAK,UAAU,KAAK,CAAC;AAAA,IAC/D,KAAK,UAAU,OAAO,SAAS,MAAM,CAAC;AAEvC,YAAQ,IAAI,MAAM,uBAAuB,UAAU;AACnD,WAAO;AAAA,EACR;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQA,MAAM,UAAU;AACf,SAAK,OAAO,QAAQ,CAAC,QAAgB;AACpC,UAAI,MAAM;AAAA,IACX,CAAC;AAAA,EACF;AACD;","names":[]}
+{"version":3,"sources":["../../src/modules/mcp/mcp.module.ts"],"sourcesContent":["import { Client } from \"@modelcontextprotocol/sdk/client/index.js\";\nimport { StdioClientTransport } from \"@modelcontextprotocol/sdk/client/stdio.js\";\nimport type { MCPConfig } from \"@/types/mcp.js\";\nimport { loggers } from \"@/utils/logger.js\";\nimport { MCPTool } from \"./mcp.tool.js\";\n\n/**\n * Module for managing Model Context Protocol (MCP) server connections.\n *\n * This module handles the lifecycle of MCP client connections, discovers\n * available tools from connected servers, and provides an interface for\n * executing those tools. Multiple MCP servers can be connected simultaneously.\n *\n * @example\n * ```typescript\n * const mcpModule = new MCPModule();\n * await mcpModule.addMCPConfig({\n *   \"filesystem\": {\n *     command: \"npx\",\n *     args: [\"@modelcontextprotocol/server-filesystem\", \"/path/to/files\"]\n *   }\n * });\n *\n * const tools = mcpModule.getTools();\n * const result = await mcpModule.useTool(tools[0], { path: \"/example.txt\" });\n * ```\n */\nexport class MCPModule {\n\t/** Map of MCP server names to their client instances */\n\tprivate mcpMap: Map<string, Client> = new Map();\n\t/** Map of MCP server names to their transport instances */\n\tprivate transportMap: Map<string, StdioClientTransport> = new Map();\n\t/** Array of all discovered tools from connected MCP servers */\n\tprivate tools: MCPTool[] = [];\n\n\t/**\n\t * Connects to MCP servers based on the provided configuration.\n\t *\n\t * For each server in the config, establishes a connection, discovers\n\t * available tools, and adds them to the module's tool collection.\n\t * Skips servers that are already connected.\n\t *\n\t * @param mcpConfig - Configuration object mapping server names to connection details\n\t * @throws Error if connection to any MCP server fails\n\t */\n\tasync addMCPConfig(mcpConfig: MCPConfig) {\n\t\ttry {\n\t\t\tfor (const [name, conf] of Object.entries(mcpConfig)) {\n\t\t\t\t// FIXME(yoojin): Need strict duplication check.\n\t\t\t\tif (this.mcpMap.get(name) && this.transportMap.get(name)) continue; // Duplicated mcp: skip\n\n\t\t\t\tconst transport = new StdioClientTransport(conf);\n\t\t\t\tthis.transportMap.set(name, transport);\n\t\t\t\tconst mcp = new Client({ name: \"mcp-client-cli\", version: \"1.0.0\" });\n\t\t\t\tawait mcp.connect(transport);\n\t\t\t\tthis.mcpMap.set(name, mcp);\n\n\t\t\t\tconst toolsResult = await mcp.listTools();\n\t\t\t\tthis.tools.push(\n\t\t\t\t\t...toolsResult.tools.map((tool) => {\n\t\t\t\t\t\treturn new MCPTool(name, tool);\n\t\t\t\t\t}),\n\t\t\t\t);\n\t\t\t}\n\t\t\tloggers.mcp.info(\"Connected to MCP server with tools:\", {\n\t\t\t\ttools: this.tools.map((tool) => tool.id),\n\t\t\t});\n\t\t} catch (error: unknown) {\n\t\t\tloggers.mcp.error(\"Failed to connect to MCP server:\", { error });\n\t\t\tthrow error;\n\t\t}\n\t}\n\n\t/**\n\t * Returns all available tools from connected MCP servers.\n\t *\n\t * @returns Array of MCPTool instances representing available tools\n\t */\n\tgetTools() {\n\t\treturn this.tools;\n\t}\n\n\t/**\n\t * Executes a tool on its corresponding MCP server.\n\t *\n\t * @param tool - The MCPTool instance to execute\n\t * @param _args - Arguments to pass to the tool\n\t * @returns Promise resolving to the tool's execution result\n\t * @throws Error if the MCP server for the tool is not found\n\t */\n\tasync useTool(tool: MCPTool, _args?: any): Promise<string> {\n\t\tconst { serverName, mcpTool } = tool;\n\t\tconst toolName = mcpTool.name;\n\t\tconst mcp = this.mcpMap.get(serverName);\n\n\t\ttry {\n\t\t\tif (!mcp) {\n\t\t\t\tthrow new Error(`Invalid MCP Tool ${serverName}-${mcpTool.name}`);\n\t\t\t}\n\n\t\t\tconst result = await mcp.callTool({\n\t\t\t\tname: toolName,\n\t\t\t\targuments: _args,\n\t\t\t});\n\t\t\tconst toolResult =\n\t\t\t\t`[Bot Called Tool ${toolName} with args ${JSON.stringify(_args)}]\\n` +\n\t\t\t\tJSON.stringify(result.content, null, 2);\n\t\t\treturn toolResult;\n\t\t} catch (error) {\n\t\t\tloggers.mcp.error(\"Failed to call tool\", { error });\n\t\t\tconst toolResult = `[Bot Called Tool ${toolName} with args ${JSON.stringify(_args)}]\\n${typeof error === \"string\" ? error : JSON.stringify(error, null, 2)}`;\n\t\t\treturn toolResult;\n\t\t}\n\t}\n\n\t/**\n\t * Closes all MCP client connections.\n\t *\n\t * Should be called when shutting down the application to ensure\n\t * all MCP connections are properly closed.\n\t */\n\tasync cleanup() {\n\t\tthis.mcpMap.forEach((mcp: Client) => {\n\t\t\tmcp.close();\n\t\t});\n\t}\n}\n"],"mappings":";;;;;;;;AAAA,SAAS,cAAc;AACvB,SAAS,4BAA4B;AA0B9B,IAAM,YAAN,MAAgB;AAAA;AAAA,EAEd,SAA8B,oBAAI,IAAI;AAAA;AAAA,EAEtC,eAAkD,oBAAI,IAAI;AAAA;AAAA,EAE1D,QAAmB,CAAC;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAY5B,MAAM,aAAa,WAAsB;AACxC,QAAI;AACH,iBAAW,CAAC,MAAM,IAAI,KAAK,OAAO,QAAQ,SAAS,GAAG;AAErD,YAAI,KAAK,OAAO,IAAI,IAAI,KAAK,KAAK,aAAa,IAAI,IAAI,EAAG;AAE1D,cAAM,YAAY,IAAI,qBAAqB,IAAI;AAC/C,aAAK,aAAa,IAAI,MAAM,SAAS;AACrC,cAAM,MAAM,IAAI,OAAO,EAAE,MAAM,kBAAkB,SAAS,QAAQ,CAAC;AACnE,cAAM,IAAI,QAAQ,SAAS;AAC3B,aAAK,OAAO,IAAI,MAAM,GAAG;AAEzB,cAAM,cAAc,MAAM,IAAI,UAAU;AACxC,aAAK,MAAM;AAAA,UACV,GAAG,YAAY,MAAM,IAAI,CAAC,SAAS;AAClC,mBAAO,IAAI,QAAQ,MAAM,IAAI;AAAA,UAC9B,CAAC;AAAA,QACF;AAAA,MACD;AACA,cAAQ,IAAI,KAAK,uCAAuC;AAAA,QACvD,OAAO,KAAK,MAAM,IAAI,CAAC,SAAS,KAAK,EAAE;AAAA,MACxC,CAAC;AAAA,IACF,SAAS,OAAgB;AACxB,cAAQ,IAAI,MAAM,oCAAoC,EAAE,MAAM,CAAC;AAC/D,YAAM;AAAA,IACP;AAAA,EACD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,WAAW;AACV,WAAO,KAAK;AAAA,EACb;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAUA,MAAM,QAAQ,MAAe,OAA8B;AAC1D,UAAM,EAAE,YAAY,QAAQ,IAAI;AAChC,UAAM,WAAW,QAAQ;AACzB,UAAM,MAAM,KAAK,OAAO,IAAI,UAAU;AAEtC,QAAI;AACH,UAAI,CAAC,KAAK;AACT,cAAM,IAAI,MAAM,oBAAoB,UAAU,IAAI,QAAQ,IAAI,EAAE;AAAA,MACjE;AAEA,YAAM,SAAS,MAAM,IAAI,SAAS;AAAA,QACjC,MAAM;AAAA,QACN,WAAW;AAAA,MACZ,CAAC;AACD,YAAM,aACL,oBAAoB,QAAQ,cAAc,KAAK,UAAU,KAAK,CAAC;AAAA,IAC/D,KAAK,UAAU,OAAO,SAAS,MAAM,CAAC;AACvC,aAAO;AAAA,IACR,SAAS,OAAO;AACf,cAAQ,IAAI,MAAM,uBAAuB,EAAE,MAAM,CAAC;AAClD,YAAM,aAAa,oBAAoB,QAAQ,cAAc,KAAK,UAAU,KAAK,CAAC;AAAA,EAAM,OAAO,UAAU,WAAW,QAAQ,KAAK,UAAU,OAAO,MAAM,CAAC,CAAC;AAC1J,aAAO;AAAA,IACR;AAAA,EACD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQA,MAAM,UAAU;AACf,SAAK,OAAO,QAAQ,CAAC,QAAgB;AACpC,UAAI,MAAM;AAAA,IACX,CAAC;AAAA,EACF;AACD;","names":[]}

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

@@ -157,12 +157,16 @@ var A2AModule = class {
     } catch (error) {
       loggers.a2a.error("Error communicating with agent:", { error });
       tool.disable();
+      const toolResult = `[Bot Called A2A Tool ${tool.card.name}]
+${typeof error === "string" ? error : JSON.stringify(error, null, 2)}`;
+      return toolResult;
     }
-    return finalText;
+    return `[Bot Called A2A Tool ${tool.card.name}]
+${finalText.join("\n")}`;
   }
 };
 
 export {
   A2AModule
 };
-//# sourceMappingURL=chunk-DJRRMFKH.js.map
+//# sourceMappingURL=chunk-C6SGV7JV.js.map

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

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

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

@@ -88,20 +88,16 @@ ${this.prompts?.system || ""}
           if (this.mcpModule && selectedTool.protocol === "MCP" /* MCP */) {
             const toolArgs = toolCall.arguments;
             loggers.intent.debug("MCP tool call", { toolName, toolArgs });
-            const result = await this.mcpModule.useTool(
+            toolResult = await this.mcpModule.useTool(
               selectedTool,
               toolArgs
             );
-            toolResult = `[Bot Called MCP Tool ${toolName} with args ${JSON.stringify(toolArgs)}]
-` + JSON.stringify(result.content, null, 2);
           } else if (this.a2aModule && selectedTool.protocol === "A2A" /* A2A */) {
-            const result = await this.a2aModule.useTool(
+            toolResult = await this.a2aModule.useTool(
               selectedTool,
               messagePayload,
               sessionId
             );
-            toolResult = `[Bot Called A2A Tool ${toolName}]
-${result.join("\n")}`;
           } else {
             loggers.intent.warn(
               `Unrecognized tool type: ${selectedTool.protocol}`
@@ -163,4 +159,4 @@ ${result.join("\n")}`;
 export {
   QueryService
 };
-//# sourceMappingURL=chunk-MC7CPQ6W.js.map
+//# sourceMappingURL=chunk-DRUCJYPJ.js.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/services/query.service.ts"],"sourcesContent":["import type {\n\tA2AModule,\n\tMCPModule,\n\tMemoryModule,\n\tModelModule,\n} from \"@/modules/index.js\";\nimport type { AinAgentPrompts } from \"@/types/agent.js\";\nimport { ChatRole, type SessionObject } from \"@/types/memory.js\";\nimport {\n\ttype IA2ATool,\n\ttype IAgentTool,\n\ttype IMCPTool,\n\tTOOL_PROTOCOL_TYPE,\n} from \"@/types/tool.js\";\nimport { loggers } from \"@/utils/logger.js\";\n\n/**\n * Service for processing user queries through the agent's AI pipeline.\n *\n * Orchestrates the query processing workflow including intent detection,\n * model inference, tool execution, and response generation. Manages\n * conversation context and coordinates between different modules.\n */\nexport class QueryService {\n\tprivate modelModule: ModelModule;\n\tprivate a2aModule?: A2AModule;\n\tprivate mcpModule?: MCPModule;\n\tprivate memoryModule?: MemoryModule;\n\tprivate prompts?: AinAgentPrompts;\n\n\tconstructor(\n\t\tmodelModule: ModelModule,\n\t\ta2aModule?: A2AModule,\n\t\tmcpModule?: MCPModule,\n\t\tmemoryModule?: MemoryModule,\n\t\tprompts?: AinAgentPrompts,\n\t) {\n\t\tthis.modelModule = modelModule;\n\t\tthis.a2aModule = a2aModule;\n\t\tthis.mcpModule = mcpModule;\n\t\tthis.memoryModule = memoryModule;\n\t\tthis.prompts = prompts;\n\t}\n\n\t/**\n\t * Detects the intent from a user query.\n\t *\n\t * @param query - The user's input query\n\t * @returns The detected intent (currently returns the query as-is)\n\t * @todo Implement actual intent detection logic\n\t */\n\tprivate async intentTriggering(query: string) {\n\t\t/* TODO */\n\t\treturn query;\n\t}\n\n\t/**\n\t * Fulfills the detected intent by generating a response.\n\t *\n\t * Manages the complete inference loop including:\n\t * - Loading prompts and conversation history\n\t * - Collecting available tools from modules\n\t * - Executing model inference with tool support\n\t * - Processing tool calls iteratively until completion\n\t *\n\t * @param query - The user's input query\n\t * @param sessionId - Session identifier for context\n\t * @param sessionHistory - Previous conversation history\n\t * @returns Object containing process steps and final response\n\t */\n\tprivate async intentFulfilling(\n\t\tquery: string,\n\t\tsessionId: string,\n\t\tsessionHistory?: SessionObject,\n\t) {\n\t\t// 1. Load agent / system prompt from memory\n\t\tconst systemPrompt = `\nToday is ${new Date().toLocaleDateString()}.\n\n${this.prompts?.agent || \"\"}\n\n${this.prompts?.system || \"\"}\n    `;\n\n\t\tconst modelInstance = this.modelModule.getModel();\n\t\tconst messages = modelInstance.generateMessages({\n\t\t\tquery,\n\t\t\tsessionHistory,\n\t\t\tsystemPrompt: systemPrompt.trim(),\n\t\t});\n\n\t\tconst tools: IAgentTool[] = [];\n\t\tif (this.mcpModule) {\n\t\t\ttools.push(...this.mcpModule.getTools());\n\t\t}\n\t\tif (this.a2aModule) {\n\t\t\ttools.push(...(await this.a2aModule.getTools()));\n\t\t}\n\t\tconst functions = modelInstance.convertToolsToFunctions(tools);\n\n\t\tconst processList: string[] = [];\n\t\tlet finalMessage = \"\";\n\t\tlet didCallTool = false;\n\n\t\twhile (true) {\n\t\t\tconst response = await modelInstance.fetchWithContextMessage(\n\t\t\t\tmessages,\n\t\t\t\tfunctions,\n\t\t\t);\n\t\t\tdidCallTool = false;\n\n\t\t\tloggers.intent.debug(\"messages\", { messages });\n\n\t\t\tconst { content, toolCalls } = response;\n\n\t\t\tloggers.intent.debug(\"content\", { content });\n\t\t\tloggers.intent.debug(\"tool_calls\", { ...toolCalls });\n\n\t\t\tif (toolCalls) {\n\t\t\t\tconst messagePayload = this.a2aModule?.getMessagePayload(\n\t\t\t\t\tquery,\n\t\t\t\t\tsessionId,\n\t\t\t\t);\n\n\t\t\t\tfor (const toolCall of toolCalls) {\n\t\t\t\t\tconst toolName = toolCall.name;\n\t\t\t\t\tdidCallTool = true;\n\t\t\t\t\tconst selectedTool = tools.filter((tool) => tool.id === toolName)[0];\n\n\t\t\t\t\tlet toolResult = \"\";\n\t\t\t\t\tif (\n\t\t\t\t\t\tthis.mcpModule &&\n\t\t\t\t\t\tselectedTool.protocol === TOOL_PROTOCOL_TYPE.MCP\n\t\t\t\t\t) {\n\t\t\t\t\t\tconst toolArgs = toolCall.arguments as\n\t\t\t\t\t\t\t| { [x: string]: unknown }\n\t\t\t\t\t\t\t| undefined;\n\t\t\t\t\t\tloggers.intent.debug(\"MCP tool call\", { toolName, toolArgs });\n\t\t\t\t\t\ttoolResult = await this.mcpModule.useTool(\n\t\t\t\t\t\t\tselectedTool as IMCPTool,\n\t\t\t\t\t\t\ttoolArgs,\n\t\t\t\t\t\t);\n\t\t\t\t\t} else if (\n\t\t\t\t\t\tthis.a2aModule &&\n\t\t\t\t\t\tselectedTool.protocol === TOOL_PROTOCOL_TYPE.A2A\n\t\t\t\t\t) {\n\t\t\t\t\t\ttoolResult = await this.a2aModule.useTool(\n\t\t\t\t\t\t\tselectedTool as IA2ATool,\n\t\t\t\t\t\t\tmessagePayload!,\n\t\t\t\t\t\t\tsessionId,\n\t\t\t\t\t\t);\n\t\t\t\t\t} else {\n\t\t\t\t\t\t// Unrecognized tool type. It cannot be happened...\n\t\t\t\t\t\tloggers.intent.warn(\n\t\t\t\t\t\t\t`Unrecognized tool type: ${selectedTool.protocol}`,\n\t\t\t\t\t\t);\n\t\t\t\t\t\tcontinue;\n\t\t\t\t\t}\n\n\t\t\t\t\tloggers.intent.debug(\"toolResult\", { toolResult });\n\n\t\t\t\t\tprocessList.push(toolResult);\n\t\t\t\t\tmodelInstance.appendMessages(messages, toolResult);\n\t\t\t\t}\n\t\t\t} else if (content) {\n\t\t\t\tprocessList.push(content);\n\t\t\t\tfinalMessage = content;\n\t\t\t}\n\n\t\t\tif (!didCallTool) break;\n\t\t}\n\n\t\tconst botResponse = {\n\t\t\tprocess: processList.join(\"\\n\"),\n\t\t\tresponse: finalMessage,\n\t\t};\n\n\t\treturn botResponse;\n\t}\n\n\t/**\n\t * Main entry point for processing user queries.\n\t *\n\t * Handles the complete query lifecycle:\n\t * 1. Loads session history from memory\n\t * 2. Detects intent from the query\n\t * 3. Fulfills the intent with AI response\n\t * 4. Updates conversation history\n\t *\n\t * @param query - The user's input query\n\t * @param sessionId - Unique session identifier\n\t * @param userId - Unique user identifier\n\t * @returns Object containing the AI-generated response\n\t */\n\tpublic async handleQuery(query: string, sessionId: string, userId?: string) {\n\t\t// 1. Load session history with sessionId\n\t\tconst queryStartAt = Date.now();\n\t\tconst sessionMemory = this.memoryModule?.getSessionMemory();\n\t\tconst session = !userId\n\t\t\t? undefined\n\t\t\t: await sessionMemory?.getSession(userId, sessionId);\n\n\t\t// 2. intent triggering\n\t\tconst intent = this.intentTriggering(query);\n\n\t\t// 3. intent fulfillment\n\t\tconst result = await this.intentFulfilling(query, sessionId, session);\n\t\tif (userId) {\n\t\t\tawait sessionMemory?.addChatToSession(userId, sessionId, {\n\t\t\t\trole: ChatRole.USER,\n\t\t\t\ttimestamp: queryStartAt,\n\t\t\t\tcontent: { type: \"text\", parts: [query] },\n\t\t\t});\n\t\t\tawait sessionMemory?.addChatToSession(userId, sessionId, {\n\t\t\t\trole: ChatRole.MODEL,\n\t\t\t\ttimestamp: Date.now(),\n\t\t\t\tcontent: { type: \"text\", parts: [result.response] },\n\t\t\t});\n\t\t}\n\n\t\treturn { content: result.response };\n\t}\n}\n"],"mappings":";;;;;AAuBO,IAAM,eAAN,MAAmB;AAAA,EACjB;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EAER,YACC,aACA,WACA,WACA,cACA,SACC;AACD,SAAK,cAAc;AACnB,SAAK,YAAY;AACjB,SAAK,YAAY;AACjB,SAAK,eAAe;AACpB,SAAK,UAAU;AAAA,EAChB;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EASA,MAAc,iBAAiB,OAAe;AAE7C,WAAO;AAAA,EACR;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAgBA,MAAc,iBACb,OACA,WACA,gBACC;AAED,UAAM,eAAe;AAAA,YACZ,oBAAI,KAAK,GAAE,mBAAmB,CAAC;AAAA;AAAA,EAExC,KAAK,SAAS,SAAS,EAAE;AAAA;AAAA,EAEzB,KAAK,SAAS,UAAU,EAAE;AAAA;AAG1B,UAAM,gBAAgB,KAAK,YAAY,SAAS;AAChD,UAAM,WAAW,cAAc,iBAAiB;AAAA,MAC/C;AAAA,MACA;AAAA,MACA,cAAc,aAAa,KAAK;AAAA,IACjC,CAAC;AAED,UAAM,QAAsB,CAAC;AAC7B,QAAI,KAAK,WAAW;AACnB,YAAM,KAAK,GAAG,KAAK,UAAU,SAAS,CAAC;AAAA,IACxC;AACA,QAAI,KAAK,WAAW;AACnB,YAAM,KAAK,GAAI,MAAM,KAAK,UAAU,SAAS,CAAE;AAAA,IAChD;AACA,UAAM,YAAY,cAAc,wBAAwB,KAAK;AAE7D,UAAM,cAAwB,CAAC;AAC/B,QAAI,eAAe;AACnB,QAAI,cAAc;AAElB,WAAO,MAAM;AACZ,YAAM,WAAW,MAAM,cAAc;AAAA,QACpC;AAAA,QACA;AAAA,MACD;AACA,oBAAc;AAEd,cAAQ,OAAO,MAAM,YAAY,EAAE,SAAS,CAAC;AAE7C,YAAM,EAAE,SAAS,UAAU,IAAI;AAE/B,cAAQ,OAAO,MAAM,WAAW,EAAE,QAAQ,CAAC;AAC3C,cAAQ,OAAO,MAAM,cAAc,EAAE,GAAG,UAAU,CAAC;AAEnD,UAAI,WAAW;AACd,cAAM,iBAAiB,KAAK,WAAW;AAAA,UACtC;AAAA,UACA;AAAA,QACD;AAEA,mBAAW,YAAY,WAAW;AACjC,gBAAM,WAAW,SAAS;AAC1B,wBAAc;AACd,gBAAM,eAAe,MAAM,OAAO,CAAC,SAAS,KAAK,OAAO,QAAQ,EAAE,CAAC;AAEnE,cAAI,aAAa;AACjB,cACC,KAAK,aACL,aAAa,8BACZ;AACD,kBAAM,WAAW,SAAS;AAG1B,oBAAQ,OAAO,MAAM,iBAAiB,EAAE,UAAU,SAAS,CAAC;AAC5D,yBAAa,MAAM,KAAK,UAAU;AAAA,cACjC;AAAA,cACA;AAAA,YACD;AAAA,UACD,WACC,KAAK,aACL,aAAa,8BACZ;AACD,yBAAa,MAAM,KAAK,UAAU;AAAA,cACjC;AAAA,cACA;AAAA,cACA;AAAA,YACD;AAAA,UACD,OAAO;AAEN,oBAAQ,OAAO;AAAA,cACd,2BAA2B,aAAa,QAAQ;AAAA,YACjD;AACA;AAAA,UACD;AAEA,kBAAQ,OAAO,MAAM,cAAc,EAAE,WAAW,CAAC;AAEjD,sBAAY,KAAK,UAAU;AAC3B,wBAAc,eAAe,UAAU,UAAU;AAAA,QAClD;AAAA,MACD,WAAW,SAAS;AACnB,oBAAY,KAAK,OAAO;AACxB,uBAAe;AAAA,MAChB;AAEA,UAAI,CAAC,YAAa;AAAA,IACnB;AAEA,UAAM,cAAc;AAAA,MACnB,SAAS,YAAY,KAAK,IAAI;AAAA,MAC9B,UAAU;AAAA,IACX;AAEA,WAAO;AAAA,EACR;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAgBA,MAAa,YAAY,OAAe,WAAmB,QAAiB;AAE3E,UAAM,eAAe,KAAK,IAAI;AAC9B,UAAM,gBAAgB,KAAK,cAAc,iBAAiB;AAC1D,UAAM,UAAU,CAAC,SACd,SACA,MAAM,eAAe,WAAW,QAAQ,SAAS;AAGpD,UAAM,SAAS,KAAK,iBAAiB,KAAK;AAG1C,UAAM,SAAS,MAAM,KAAK,iBAAiB,OAAO,WAAW,OAAO;AACpE,QAAI,QAAQ;AACX,YAAM,eAAe,iBAAiB,QAAQ,WAAW;AAAA,QACxD;AAAA,QACA,WAAW;AAAA,QACX,SAAS,EAAE,MAAM,QAAQ,OAAO,CAAC,KAAK,EAAE;AAAA,MACzC,CAAC;AACD,YAAM,eAAe,iBAAiB,QAAQ,WAAW;AAAA,QACxD;AAAA,QACA,WAAW,KAAK,IAAI;AAAA,QACpB,SAAS,EAAE,MAAM,QAAQ,OAAO,CAAC,OAAO,QAAQ,EAAE;AAAA,MACnD,CAAC;AAAA,IACF;AAEA,WAAO,EAAE,SAAS,OAAO,SAAS;AAAA,EACnC;AACD;","names":[]}

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

@@ -3,7 +3,7 @@ import {
 } from "./chunk-Y6E73I7I.js";
 import {
   QueryService
-} from "./chunk-MC7CPQ6W.js";
+} from "./chunk-DRUCJYPJ.js";
 import {
   A2AController
 } from "./chunk-HAE333YO.js";
@@ -34,4 +34,4 @@ var createA2ARouter = (agent) => {
 export {
   createA2ARouter
 };
-//# sourceMappingURL=chunk-A3W52LKL.js.map
+//# sourceMappingURL=chunk-JOE4VYJJ.js.map

package/dist/esm/{chunk-WGHAZ4HG.js → chunk-Q6HHVAO6.js}

@@ -7,9 +7,9 @@ var QueryController = class {
     this.queryStreamService = queryStreamService;
   }
   handleQueryRequest = async (req, res, next) => {
+    const { message, sessionId } = req.body;
+    const userId = res.locals.userId;
     try {
-      const { message, sessionId } = req.body;
-      const userId = res.locals.userId;
       const result = await this.queryService.handleQuery(
         message,
         sessionId,
@@ -21,16 +21,17 @@ var QueryController = class {
     }
   };
   handleQueryStreamRequest = async (req, res, next) => {
-    const { message, sessionId, userId } = req.body;
+    const { message, sessionId } = req.body;
+    const userId = res.locals.userId;
     if (!this.queryStreamService) {
       throw new Error("This Agent does not support stream query");
     }
     try {
       await this.queryStreamService.handleQueryStream(
         message,
-        sessionId,
+        userId,
         res,
-        userId
+        sessionId
       );
     } catch (error) {
       next(error);
@@ -41,4 +42,4 @@ var QueryController = class {
 export {
   QueryController
 };
-//# sourceMappingURL=chunk-WGHAZ4HG.js.map
+//# sourceMappingURL=chunk-Q6HHVAO6.js.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/controllers/query.controller.ts"],"sourcesContent":["import type { NextFunction, Request, Response } from \"express\";\nimport type { QueryService, QueryStreamService } from \"@/services\";\n\nexport class QueryController {\n\tprivate queryService;\n\tprivate queryStreamService;\n\n\tconstructor(\n\t\tqueryService: QueryService,\n\t\tqueryStreamService?: QueryStreamService,\n\t) {\n\t\tthis.queryService = queryService;\n\t\tthis.queryStreamService = queryStreamService;\n\t}\n\n\tpublic handleQueryRequest = async (\n\t\treq: Request,\n\t\tres: Response,\n\t\tnext: NextFunction,\n\t) => {\n\t\tconst { message, sessionId } = req.body;\n\t\tconst userId = res.locals.userId;\n\n\t\ttry {\n\t\t\tconst result = await this.queryService.handleQuery(\n\t\t\t\tmessage,\n\t\t\t\tsessionId,\n\t\t\t\tuserId,\n\t\t\t);\n\n\t\t\tres.status(200).json(result);\n\t\t} catch (error) {\n\t\t\tnext(error);\n\t\t}\n\t};\n\n\tpublic handleQueryStreamRequest = async (\n\t\treq: Request,\n\t\tres: Response,\n\t\tnext: NextFunction,\n\t) => {\n\t\tconst { message, sessionId } = req.body;\n\t\tconst userId = res.locals.userId;\n\n\t\tif (!this.queryStreamService) {\n\t\t\tthrow new Error(\"This Agent does not support stream query\");\n\t\t}\n\n\t\ttry {\n\t\t\tawait this.queryStreamService.handleQueryStream(\n\t\t\t\tmessage,\n\t\t\t\tuserId,\n\t\t\t\tres,\n\t\t\t\tsessionId,\n\t\t\t);\n\t\t} catch (error) {\n\t\t\tnext(error);\n\t\t}\n\t};\n}\n"],"mappings":";AAGO,IAAM,kBAAN,MAAsB;AAAA,EACpB;AAAA,EACA;AAAA,EAER,YACC,cACA,oBACC;AACD,SAAK,eAAe;AACpB,SAAK,qBAAqB;AAAA,EAC3B;AAAA,EAEO,qBAAqB,OAC3B,KACA,KACA,SACI;AACJ,UAAM,EAAE,SAAS,UAAU,IAAI,IAAI;AACnC,UAAM,SAAS,IAAI,OAAO;AAE1B,QAAI;AACH,YAAM,SAAS,MAAM,KAAK,aAAa;AAAA,QACtC;AAAA,QACA;AAAA,QACA;AAAA,MACD;AAEA,UAAI,OAAO,GAAG,EAAE,KAAK,MAAM;AAAA,IAC5B,SAAS,OAAO;AACf,WAAK,KAAK;AAAA,IACX;AAAA,EACD;AAAA,EAEO,2BAA2B,OACjC,KACA,KACA,SACI;AACJ,UAAM,EAAE,SAAS,UAAU,IAAI,IAAI;AACnC,UAAM,SAAS,IAAI,OAAO;AAE1B,QAAI,CAAC,KAAK,oBAAoB;AAC7B,YAAM,IAAI,MAAM,0CAA0C;AAAA,IAC3D;AAEA,QAAI;AACH,YAAM,KAAK,mBAAmB;AAAA,QAC7B;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,MACD;AAAA,IACD,SAAS,OAAO;AACf,WAAK,KAAK;AAAA,IACX;AAAA,EACD;AACD;","names":[]}

package/dist/esm/{chunk-ZFSIP5KN.js → chunk-UUXE3TQE.js}

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

package/dist/esm/controllers/index.js

@@ -1,17 +1,17 @@
+import {
+  QueryController
+} from "../chunk-Q6HHVAO6.js";
+import {
+  A2AController
+} from "../chunk-HAE333YO.js";
+import "../chunk-N7VYM2R5.js";
 import {
   ModelApiController
 } from "../chunk-UA44VKFE.js";
 import {
   SessionApiController
 } from "../chunk-5EPK37PE.js";
-import {
-  QueryController
-} from "../chunk-WGHAZ4HG.js";
 import "../chunk-FL3JRQM4.js";
-import {
-  A2AController
-} from "../chunk-HAE333YO.js";
-import "../chunk-N7VYM2R5.js";
 export {
   A2AController,
   ModelApiController,

package/dist/esm/controllers/query.controller.js

@@ -1,6 +1,6 @@
 import {
   QueryController
-} from "../chunk-WGHAZ4HG.js";
+} from "../chunk-Q6HHVAO6.js";
 export {
   QueryController
 };

package/dist/esm/index.js

@@ -1,34 +1,34 @@
 import "./chunk-XYCWK2ZM.js";
-import {
-  createApiRouter
-} from "./chunk-YT6PD6UF.js";
-import "./chunk-2Z7GJVJJ.js";
-import "./chunk-QROUPWWO.js";
 import {
   createQueryRouter
-} from "./chunk-ZFSIP5KN.js";
-import "./chunk-EFMGNCWG.js";
-import "./chunk-UA44VKFE.js";
-import "./chunk-5EPK37PE.js";
-import "./chunk-WGHAZ4HG.js";
+} from "./chunk-UUXE3TQE.js";
+import "./chunk-3NEV3O2X.js";
+import "./chunk-Q6HHVAO6.js";
 import {
   AuthMiddleware
 } from "./chunk-PN6NMQBM.js";
-import "./chunk-FL3JRQM4.js";
 import {
   errorMiddleware
 } from "./chunk-MAU2MAK2.js";
 import {
   createA2ARouter
-} from "./chunk-A3W52LKL.js";
+} from "./chunk-JOE4VYJJ.js";
 import "./chunk-Y6E73I7I.js";
-import "./chunk-MC7CPQ6W.js";
+import "./chunk-DRUCJYPJ.js";
 import "./chunk-VAZWF5IK.js";
-import "./chunk-HAE333YO.js";
 import "./chunk-FI2UQLVC.js";
+import "./chunk-HAE333YO.js";
 import {
   loggers
 } from "./chunk-N7VYM2R5.js";
+import {
+  createApiRouter
+} from "./chunk-YT6PD6UF.js";
+import "./chunk-2Z7GJVJJ.js";
+import "./chunk-QROUPWWO.js";
+import "./chunk-UA44VKFE.js";
+import "./chunk-5EPK37PE.js";
+import "./chunk-FL3JRQM4.js";
 
 // src/index.ts
 import cors from "cors";

package/dist/esm/modules/a2a/a2a.module.d.ts

@@ -70,7 +70,7 @@ declare class A2AModule {
      * @param sessionId - The session identifier for context tracking
      * @returns Promise resolving to array of text responses from the agent
      */
-    useTool(tool: A2ATool, messagePayload: Message, sessionId: string): Promise<string[]>;
+    useTool(tool: A2ATool, messagePayload: Message, sessionId: string): Promise<string>;
 }
 
 export { A2AModule };

package/dist/esm/modules/a2a/a2a.module.js

@@ -1,6 +1,6 @@
 import {
   A2AModule
-} from "../../chunk-DJRRMFKH.js";
+} from "../../chunk-C6SGV7JV.js";
 import "../../chunk-DGH3GB5I.js";
 import "../../chunk-FI2UQLVC.js";
 import "../../chunk-N7VYM2R5.js";

package/dist/esm/modules/index.js

@@ -1,23 +1,23 @@
+import {
+  MemoryModule
+} from "../chunk-CKTURH34.js";
+import {
+  BaseModel
+} from "../chunk-2BITOA2R.js";
 import {
   ModelModule
 } from "../chunk-PHKK5NHK.js";
 import {
   A2AModule
-} from "../chunk-DJRRMFKH.js";
+} from "../chunk-C6SGV7JV.js";
 import "../chunk-DGH3GB5I.js";
 import {
   BaseAuth
 } from "../chunk-FF5RJ4L2.js";
 import {
   MCPModule
-} from "../chunk-624FPEEV.js";
+} from "../chunk-BDMVIJPF.js";
 import "../chunk-3LLHEUY2.js";
-import {
-  MemoryModule
-} from "../chunk-CKTURH34.js";
-import {
-  BaseModel
-} from "../chunk-2BITOA2R.js";
 import "../chunk-FI2UQLVC.js";
 import "../chunk-N7VYM2R5.js";
 export {

package/dist/esm/modules/mcp/mcp.module.d.ts

@@ -59,7 +59,7 @@ declare class MCPModule {
      * @returns Promise resolving to the tool's execution result
      * @throws Error if the MCP server for the tool is not found
      */
-    useTool(tool: MCPTool, _args?: any): Promise<any>;
+    useTool(tool: MCPTool, _args?: any): Promise<string>;
     /**
      * Closes all MCP client connections.
      *

package/dist/esm/modules/mcp/mcp.module.js

@@ -1,6 +1,6 @@
 import {
   MCPModule
-} from "../../chunk-624FPEEV.js";
+} from "../../chunk-BDMVIJPF.js";
 import "../../chunk-3LLHEUY2.js";
 import "../../chunk-FI2UQLVC.js";
 import "../../chunk-N7VYM2R5.js";

package/dist/esm/modules/memory/base.memory.d.ts

@@ -1,4 +1,4 @@
-import { SessionObject, ChatObject, SessionMetadata, Intent } from '../../types/memory.js';
+import { SessionObject, SessionMetadata, ChatObject, Intent } from '../../types/memory.js';
 
 /**
  * Base interface for all memory implementations
@@ -13,7 +13,7 @@ interface IMemory {
  */
 interface ISessionMemory extends IMemory {
     getSession(userId: string, sessionId: string): Promise<SessionObject | undefined>;
-    createSession(userId: string, sessionId: string, title?: string): Promise<void>;
+    createSession(userId: string, sessionId: string, title: string): Promise<SessionMetadata>;
     addChatToSession(userId: string, sessionId: string, chat: ChatObject): Promise<void>;
     deleteSession(userId: string, sessionId: string): Promise<void>;
     listSessions(userId: string): Promise<SessionMetadata[]>;

package/dist/esm/routes/a2a.routes.js

@@ -1,11 +1,11 @@
 import {
   createA2ARouter
-} from "../chunk-A3W52LKL.js";
+} from "../chunk-JOE4VYJJ.js";
 import "../chunk-Y6E73I7I.js";
-import "../chunk-MC7CPQ6W.js";
+import "../chunk-DRUCJYPJ.js";
 import "../chunk-VAZWF5IK.js";
-import "../chunk-HAE333YO.js";
 import "../chunk-FI2UQLVC.js";
+import "../chunk-HAE333YO.js";
 import "../chunk-N7VYM2R5.js";
 export {
   createA2ARouter

package/dist/esm/routes/index.js

@@ -1,4 +1,18 @@
 import "../chunk-XYCWK2ZM.js";
+import {
+  createQueryRouter
+} from "../chunk-UUXE3TQE.js";
+import "../chunk-3NEV3O2X.js";
+import "../chunk-Q6HHVAO6.js";
+import {
+  createA2ARouter
+} from "../chunk-JOE4VYJJ.js";
+import "../chunk-Y6E73I7I.js";
+import "../chunk-DRUCJYPJ.js";
+import "../chunk-VAZWF5IK.js";
+import "../chunk-FI2UQLVC.js";
+import "../chunk-HAE333YO.js";
+import "../chunk-N7VYM2R5.js";
 import {
   createApiRouter
 } from "../chunk-YT6PD6UF.js";
@@ -8,23 +22,9 @@ import {
 import {
   createSessionApiRouter
 } from "../chunk-QROUPWWO.js";
-import {
-  createQueryRouter
-} from "../chunk-ZFSIP5KN.js";
-import "../chunk-EFMGNCWG.js";
 import "../chunk-UA44VKFE.js";
 import "../chunk-5EPK37PE.js";
-import "../chunk-WGHAZ4HG.js";
 import "../chunk-FL3JRQM4.js";
-import {
-  createA2ARouter
-} from "../chunk-A3W52LKL.js";
-import "../chunk-Y6E73I7I.js";
-import "../chunk-MC7CPQ6W.js";
-import "../chunk-VAZWF5IK.js";
-import "../chunk-HAE333YO.js";
-import "../chunk-FI2UQLVC.js";
-import "../chunk-N7VYM2R5.js";
 export {
   createA2ARouter,
   createApiRouter,

package/dist/esm/routes/query.routes.js

@@ -1,9 +1,9 @@
 import {
   createQueryRouter
-} from "../chunk-ZFSIP5KN.js";
-import "../chunk-EFMGNCWG.js";
-import "../chunk-WGHAZ4HG.js";
-import "../chunk-MC7CPQ6W.js";
+} from "../chunk-UUXE3TQE.js";
+import "../chunk-3NEV3O2X.js";
+import "../chunk-Q6HHVAO6.js";
+import "../chunk-DRUCJYPJ.js";
 import "../chunk-VAZWF5IK.js";
 import "../chunk-FI2UQLVC.js";
 import "../chunk-N7VYM2R5.js";

package/dist/esm/services/index.js

@@ -1,12 +1,12 @@
 import {
   QueryStreamService
-} from "../chunk-EFMGNCWG.js";
+} from "../chunk-3NEV3O2X.js";
 import {
   A2AService
 } from "../chunk-Y6E73I7I.js";
 import {
   QueryService
-} from "../chunk-MC7CPQ6W.js";
+} from "../chunk-DRUCJYPJ.js";
 import "../chunk-VAZWF5IK.js";
 import "../chunk-FI2UQLVC.js";
 import "../chunk-N7VYM2R5.js";

package/dist/esm/services/query-stream.service.d.ts

@@ -54,7 +54,14 @@ declare class QueryStreamService {
      * @param sessionHistory - Previous conversation history
      * @returns Object containing process steps and final response
      */
-    intentFulfilling(query: string, sessionId: string, sessionHistory: SessionObject): AsyncGenerator<StreamEvent>;
+    intentFulfilling(query: string, sessionId: string, sessionHistory?: SessionObject): AsyncGenerator<StreamEvent>;
+    /**
+     * Generates a title for the conversation based on the query.
+     *
+     * @param query - The user's input query
+     * @returns Promise resolving to a generated title
+     */
+    private generateTitle;
     /**
      * Main entry point for processing user queries.
      *
@@ -68,7 +75,7 @@ declare class QueryStreamService {
      * @param sessionId - Unique session identifier
      * @returns Object containing the AI-generated response
      */
-    handleQueryStream(query: string, sessionId: string, res: Response, userId?: string): Promise<void>;
+    handleQueryStream(query: string, userId: string, res: Response, _sessionId?: string): Promise<void>;
 }
 
 export { QueryStreamService };

package/dist/esm/services/query-stream.service.js

@@ -1,6 +1,7 @@
 import {
   QueryStreamService
-} from "../chunk-EFMGNCWG.js";
+} from "../chunk-3NEV3O2X.js";
+import "../chunk-VAZWF5IK.js";
 import "../chunk-FI2UQLVC.js";
 import "../chunk-N7VYM2R5.js";
 export {

package/dist/esm/services/query.service.js

@@ -1,6 +1,6 @@
 import {
   QueryService
-} from "../chunk-MC7CPQ6W.js";
+} from "../chunk-DRUCJYPJ.js";
 import "../chunk-VAZWF5IK.js";
 import "../chunk-FI2UQLVC.js";
 import "../chunk-N7VYM2R5.js";