@ainetwork/adk 0.2.0 → 0.2.2

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

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/services/query.service.ts"],"sourcesContent":["import { randomUUID } from \"node:crypto\";\nimport type {\n\tA2AModule,\n\tMCPModule,\n\tMemoryModule,\n\tModelModule,\n} from \"@/modules/index.js\";\nimport type { AinAgentPrompts } from \"@/types/agent.js\";\nimport {\n\ttype Intent,\n\tMessageRole,\n\ttype ThreadMetadata,\n\ttype ThreadObject,\n\ttype ThreadType,\n} from \"@/types/memory.js\";\nimport {\n\ttype IA2ATool,\n\ttype IAgentTool,\n\ttype IMCPTool,\n\tTOOL_PROTOCOL_TYPE,\n} from \"@/types/tool.js\";\nimport { loggers } from \"@/utils/logger.js\";\n\n/**\n * Service for processing user queries through the agent's AI pipeline.\n *\n * Orchestrates the query processing workflow including intent detection,\n * model inference, tool execution, and response generation. Manages\n * conversation context and coordinates between different modules.\n */\nexport class QueryService {\n\tprivate modelModule: ModelModule;\n\tprivate a2aModule?: A2AModule;\n\tprivate mcpModule?: MCPModule;\n\tprivate memoryModule?: MemoryModule;\n\tprivate prompts?: AinAgentPrompts;\n\n\tconstructor(\n\t\tmodelModule: ModelModule,\n\t\ta2aModule?: A2AModule,\n\t\tmcpModule?: MCPModule,\n\t\tmemoryModule?: MemoryModule,\n\t\tprompts?: AinAgentPrompts,\n\t) {\n\t\tthis.modelModule = modelModule;\n\t\tthis.a2aModule = a2aModule;\n\t\tthis.mcpModule = mcpModule;\n\t\tthis.memoryModule = memoryModule;\n\t\tthis.prompts = prompts;\n\t}\n\n\t/**\n\t * Detects the intent from context.\n\t *\n\t * @param query - The user's input query\n\t * @param thread - The thread history\n\t * @returns The detected intent\n\t */\n\tprivate async intentTriggering(\n\t\tquery: string,\n\t\tthread: ThreadObject | undefined,\n\t): Promise<Intent | undefined> {\n\t\tconst modelInstance = this.modelModule.getModel();\n\t\tconst intentMemory = this.memoryModule?.getIntentMemory();\n\t\tif (!intentMemory) {\n\t\t\treturn undefined;\n\t\t}\n\n\t\t// 인텐트 목록 가져오기\n\t\tconst intents = await intentMemory.listIntents();\n\n\t\tif (intents.length === 0) {\n\t\t\tloggers.intent.warn(\"No intent found\");\n\t\t\treturn undefined;\n\t\t}\n\n\t\tconst intentList = intents\n\t\t\t.map((intent) => `- ${intent.name}: ${intent.description}`)\n\t\t\t.join(\"\\n\");\n\n\t\t// Convert session history to a string\n\t\tconst historyMessages = !thread\n\t\t\t? \"\"\n\t\t\t: Object.entries(thread.messages)\n\t\t\t\t\t.sort(([, a], [, b]) => a.timestamp - b.timestamp)\n\t\t\t\t\t.map(([chatId, chat]) => {\n\t\t\t\t\t\tconst role =\n\t\t\t\t\t\t\tchat.role === \"USER\"\n\t\t\t\t\t\t\t\t? \"User\"\n\t\t\t\t\t\t\t\t: chat.role === \"MODEL\"\n\t\t\t\t\t\t\t\t\t? \"Assistant\"\n\t\t\t\t\t\t\t\t\t: \"System\";\n\t\t\t\t\t\tconst content = Array.isArray(chat.content.parts)\n\t\t\t\t\t\t\t? chat.content.parts.join(\" \")\n\t\t\t\t\t\t\t: String(chat.content.parts);\n\t\t\t\t\t\treturn `${role}: \"\"\"${content}\"\"\"`;\n\t\t\t\t\t})\n\t\t\t\t\t.join(\"\\n\");\n\n\t\tconst systemPrompt = `You are an expert in accurately identifying user intentions.\n\nAvailable intent list:\n${intentList}\n\nPlease select and answer only from the above intent list. \nPlease return only the exact intent name without any additional explanations or text.`;\n\n\t\tconst userMessage = `The following is the conversation history with the user:\n\n${historyMessages}\n\nLast user question: \"${query}\"\n\nBased on the above conversation history, please determine what the intention of the last user question is. \nPlease select and answer the most appropriate intent name from the available intent list.`;\n\n\t\tconst messages = modelInstance.generateMessages({\n\t\t\tquery: userMessage,\n\t\t\tsystemPrompt,\n\t\t});\n\n\t\tconst response = await modelInstance.fetch(messages);\n\t\tif (!response.content) {\n\t\t\tthrow new Error(\"No intent detected\");\n\t\t}\n\t\tconst intentName = response.content.trim();\n\t\tconst intent = await intentMemory.getIntentByName(intentName);\n\t\tif (!intent) {\n\t\t\tthrow new Error(`No intent found: ${intentName}`);\n\t\t}\n\t\treturn intent;\n\t}\n\n\t/**\n\t * Fulfills the detected intent by generating a response.\n\t *\n\t * Manages the complete inference loop including:\n\t * - Loading prompts and conversation history\n\t * - Collecting available tools from modules\n\t * - Executing model inference with tool support\n\t * - Processing tool calls iteratively until completion\n\t *\n\t * @param query - The user's input query\n\t * @param threadId - Thread identifier for context\n\t * @param thread - Previous conversation history\n\t * @param intent - Optional detected intent with custom prompt\n\t * @returns Object containing process steps and final response\n\t */\n\tprivate async intentFulfilling(\n\t\tquery: string,\n\t\tthreadId: string,\n\t\tthread?: ThreadObject,\n\t\tintent?: Intent,\n\t) {\n\t\t// 1. Load agent / system prompt from memory\n\t\tconst systemPrompt = `\nToday is ${new Date().toLocaleDateString()}.\n\n${this.prompts?.agent || \"\"}\n\n${this.prompts?.system || \"\"}\n\n${intent?.prompt || \"\"}\n    `;\n\t\t// NOTE(haechan@comcom.ai):\n\t\t// When the `intent.llm` is guaranteed to be consistent, it will be used as a parameter for getModel\n\t\t// const model_name = intent?.llm || \"gpt-4o\";\n\t\tconst modelInstance = this.modelModule.getModel();\n\t\tconst messages = modelInstance.generateMessages({\n\t\t\tquery,\n\t\t\tthread,\n\t\t\tsystemPrompt: systemPrompt.trim(),\n\t\t});\n\n\t\tconst tools: IAgentTool[] = [];\n\t\tif (this.mcpModule) {\n\t\t\ttools.push(...this.mcpModule.getTools());\n\t\t}\n\t\tif (this.a2aModule) {\n\t\t\ttools.push(...(await this.a2aModule.getTools()));\n\t\t}\n\t\tconst functions = modelInstance.convertToolsToFunctions(tools);\n\n\t\tconst processList: string[] = [];\n\t\tlet finalMessage = \"\";\n\n\t\twhile (true) {\n\t\t\tconst response = await modelInstance.fetchWithContextMessage(\n\t\t\t\tmessages,\n\t\t\t\tfunctions,\n\t\t\t);\n\n\t\t\tloggers.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\tfor (const toolCall of toolCalls) {\n\t\t\t\t\tconst toolName = toolCall.name;\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\tquery,\n\t\t\t\t\t\t\tthreadId,\n\t\t\t\t\t\t);\n\t\t\t\t\t} else {\n\t\t\t\t\t\t// Unrecognized tool type. It cannot be happened...\n\t\t\t\t\t\tloggers.intent.warn(\n\t\t\t\t\t\t\t`Unrecognized tool type: ${selectedTool.protocol}`,\n\t\t\t\t\t\t);\n\t\t\t\t\t\tcontinue;\n\t\t\t\t\t}\n\n\t\t\t\t\tloggers.intent.debug(\"toolResult\", { toolResult });\n\n\t\t\t\t\tprocessList.push(toolResult);\n\t\t\t\t\tmodelInstance.appendMessages(messages, toolResult);\n\t\t\t\t}\n\t\t\t} else if (content) {\n\t\t\t\tprocessList.push(content);\n\t\t\t\tfinalMessage = content;\n\t\t\t\tbreak;\n\t\t\t}\n\t\t}\n\n\t\tconst botResponse = {\n\t\t\tprocess: processList.join(\"\\n\"),\n\t\t\tresponse: finalMessage,\n\t\t};\n\n\t\treturn botResponse;\n\t}\n\n\t/**\n\t * Generates a title for the conversation based on the query.\n\t *\n\t * @param query - The user's input query\n\t * @returns Promise resolving to a generated title\n\t */\n\n\tprivate async generateTitle(query: string): Promise<string> {\n\t\tconst DEFAULT_TITLE = \"New Chat\";\n\t\ttry {\n\t\t\tconst modelInstance = this.modelModule.getModel();\n\t\t\tconst messages = modelInstance.generateMessages({\n\t\t\t\tquery,\n\t\t\t\tsystemPrompt: `You are a helpful assistant that generates titles for conversations.\n  Please analyze the user's query and create a concise title that accurately reflects the conversation's core topic.\n  The title must be no more than 5 words long.\n  Respond with only the title. Do not include any punctuation or extra explanations.`,\n\t\t\t});\n\t\t\tconst response = await modelInstance.fetch(messages);\n\t\t\treturn response.content || DEFAULT_TITLE;\n\t\t} catch (error) {\n\t\t\tloggers.intent.error(\"Error generating title\", {\n\t\t\t\terror,\n\t\t\t\tquery,\n\t\t\t});\n\t\t\treturn DEFAULT_TITLE;\n\t\t}\n\t}\n\n\t/**\n\t * Main entry point for processing user queries.\n\t *\n\t * Handles the complete query lifecycle:\n\t * 1. Loads or creates thread from memory\n\t * 2. Detects intent from the query\n\t * 3. Fulfills the intent with AI response\n\t * 4. Updates conversation history\n\t *\n\t * @param threadMetadata - Metadata containing type, userId, and optional threadId\n\t * @param threadMetadata.type - The type of thread (e.g., chat, workflow)\n\t * @param threadMetadata.userId - The user's unique identifier\n\t * @param threadMetadata.threadId - Optional thread identifier\n\t * @param query - The user's input query\n\t * @returns Promise resolving to object with content property containing the response\n\t */\n\tpublic async handleQuery(\n\t\tthreadMetadata: {\n\t\t\ttype: ThreadType;\n\t\t\tuserId: string;\n\t\t\tthreadId?: string;\n\t\t},\n\t\tquery: string,\n\t) {\n\t\t// 1. Load thread with threadId\n\t\tconst { type, userId } = threadMetadata;\n\t\tconst queryStartAt = Date.now();\n\t\tconst threadMemory = this.memoryModule?.getThreadMemory();\n\n\t\tlet threadId = threadMetadata.threadId;\n\t\tlet thread: ThreadObject | undefined;\n\n\t\tif (threadId) {\n\t\t\tthread = await threadMemory?.getThread(userId, threadId);\n\t\t} else {\n\t\t\tthreadId = randomUUID();\n\t\t\tconst title = await this.generateTitle(query);\n\n\t\t\tconst metadata =\n\t\t\t\t(await threadMemory?.createThread(type, userId, threadId, title)) ||\n\t\t\t\t({\n\t\t\t\t\ttype,\n\t\t\t\t\tthreadId,\n\t\t\t\t\ttitle,\n\t\t\t\t\tupdatedAt: Date.now(),\n\t\t\t\t} as ThreadMetadata);\n\t\t\tloggers.intent.info(\"Create new thread\", { metadata });\n\t\t}\n\n\t\t// 2. intent triggering\n\t\tconst intent = await this.intentTriggering(query, thread);\n\n\t\t// 3. intent fulfillment\n\t\tconst result = await this.intentFulfilling(query, threadId, thread, intent);\n\t\tawait threadMemory?.addMessagesToThread(userId, threadId, [\n\t\t\t{\n\t\t\t\trole: MessageRole.USER,\n\t\t\t\ttimestamp: queryStartAt,\n\t\t\t\tcontent: { type: \"text\", parts: [query] },\n\t\t\t},\n\t\t\t{\n\t\t\t\trole: MessageRole.MODEL,\n\t\t\t\ttimestamp: Date.now(),\n\t\t\t\tcontent: { type: \"text\", parts: [result.response] },\n\t\t\t},\n\t\t]);\n\n\t\treturn { content: result.response };\n\t}\n}\n"],"mappings":";;;;;AAAA,SAAS,kBAAkB;AA8BpB,IAAM,eAAN,MAAmB;AAAA,EACjB;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EAER,YACC,aACA,WACA,WACA,cACA,SACC;AACD,SAAK,cAAc;AACnB,SAAK,YAAY;AACjB,SAAK,YAAY;AACjB,SAAK,eAAe;AACpB,SAAK,UAAU;AAAA,EAChB;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EASA,MAAc,iBACb,OACA,QAC8B;AAC9B,UAAM,gBAAgB,KAAK,YAAY,SAAS;AAChD,UAAM,eAAe,KAAK,cAAc,gBAAgB;AACxD,QAAI,CAAC,cAAc;AAClB,aAAO;AAAA,IACR;AAGA,UAAM,UAAU,MAAM,aAAa,YAAY;AAE/C,QAAI,QAAQ,WAAW,GAAG;AACzB,cAAQ,OAAO,KAAK,iBAAiB;AACrC,aAAO;AAAA,IACR;AAEA,UAAM,aAAa,QACjB,IAAI,CAACA,YAAW,KAAKA,QAAO,IAAI,KAAKA,QAAO,WAAW,EAAE,EACzD,KAAK,IAAI;AAGX,UAAM,kBAAkB,CAAC,SACtB,KACA,OAAO,QAAQ,OAAO,QAAQ,EAC7B,KAAK,CAAC,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,MAAM,EAAE,YAAY,EAAE,SAAS,EAChD,IAAI,CAAC,CAAC,QAAQ,IAAI,MAAM;AACxB,YAAM,OACL,KAAK,SAAS,SACX,SACA,KAAK,SAAS,UACb,cACA;AACL,YAAM,UAAU,MAAM,QAAQ,KAAK,QAAQ,KAAK,IAC7C,KAAK,QAAQ,MAAM,KAAK,GAAG,IAC3B,OAAO,KAAK,QAAQ,KAAK;AAC5B,aAAO,GAAG,IAAI,QAAQ,OAAO;AAAA,IAC9B,CAAC,EACA,KAAK,IAAI;AAEb,UAAM,eAAe;AAAA;AAAA;AAAA,EAGrB,UAAU;AAAA;AAAA;AAAA;AAKV,UAAM,cAAc;AAAA;AAAA,EAEpB,eAAe;AAAA;AAAA,uBAEM,KAAK;AAAA;AAAA;AAAA;AAK1B,UAAM,WAAW,cAAc,iBAAiB;AAAA,MAC/C,OAAO;AAAA,MACP;AAAA,IACD,CAAC;AAED,UAAM,WAAW,MAAM,cAAc,MAAM,QAAQ;AACnD,QAAI,CAAC,SAAS,SAAS;AACtB,YAAM,IAAI,MAAM,oBAAoB;AAAA,IACrC;AACA,UAAM,aAAa,SAAS,QAAQ,KAAK;AACzC,UAAM,SAAS,MAAM,aAAa,gBAAgB,UAAU;AAC5D,QAAI,CAAC,QAAQ;AACZ,YAAM,IAAI,MAAM,oBAAoB,UAAU,EAAE;AAAA,IACjD;AACA,WAAO;AAAA,EACR;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAiBA,MAAc,iBACb,OACA,UACA,QACA,QACC;AAED,UAAM,eAAe;AAAA,YACZ,oBAAI,KAAK,GAAE,mBAAmB,CAAC;AAAA;AAAA,EAExC,KAAK,SAAS,SAAS,EAAE;AAAA;AAAA,EAEzB,KAAK,SAAS,UAAU,EAAE;AAAA;AAAA,EAE1B,QAAQ,UAAU,EAAE;AAAA;AAKpB,UAAM,gBAAgB,KAAK,YAAY,SAAS;AAChD,UAAM,WAAW,cAAc,iBAAiB;AAAA,MAC/C;AAAA,MACA;AAAA,MACA,cAAc,aAAa,KAAK;AAAA,IACjC,CAAC;AAED,UAAM,QAAsB,CAAC;AAC7B,QAAI,KAAK,WAAW;AACnB,YAAM,KAAK,GAAG,KAAK,UAAU,SAAS,CAAC;AAAA,IACxC;AACA,QAAI,KAAK,WAAW;AACnB,YAAM,KAAK,GAAI,MAAM,KAAK,UAAU,SAAS,CAAE;AAAA,IAChD;AACA,UAAM,YAAY,cAAc,wBAAwB,KAAK;AAE7D,UAAM,cAAwB,CAAC;AAC/B,QAAI,eAAe;AAEnB,WAAO,MAAM;AACZ,YAAM,WAAW,MAAM,cAAc;AAAA,QACpC;AAAA,QACA;AAAA,MACD;AAEA,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,mBAAW,YAAY,WAAW;AACjC,gBAAM,WAAW,SAAS;AAC1B,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;AACf;AAAA,MACD;AAAA,IACD;AAEA,UAAM,cAAc;AAAA,MACnB,SAAS,YAAY,KAAK,IAAI;AAAA,MAC9B,UAAU;AAAA,IACX;AAEA,WAAO;AAAA,EACR;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EASA,MAAc,cAAc,OAAgC;AAC3D,UAAM,gBAAgB;AACtB,QAAI;AACH,YAAM,gBAAgB,KAAK,YAAY,SAAS;AAChD,YAAM,WAAW,cAAc,iBAAiB;AAAA,QAC/C;AAAA,QACA,cAAc;AAAA;AAAA;AAAA;AAAA,MAIf,CAAC;AACD,YAAM,WAAW,MAAM,cAAc,MAAM,QAAQ;AACnD,aAAO,SAAS,WAAW;AAAA,IAC5B,SAAS,OAAO;AACf,cAAQ,OAAO,MAAM,0BAA0B;AAAA,QAC9C;AAAA,QACA;AAAA,MACD,CAAC;AACD,aAAO;AAAA,IACR;AAAA,EACD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAkBA,MAAa,YACZ,gBAKA,OACC;AAED,UAAM,EAAE,MAAM,OAAO,IAAI;AACzB,UAAM,eAAe,KAAK,IAAI;AAC9B,UAAM,eAAe,KAAK,cAAc,gBAAgB;AAExD,QAAI,WAAW,eAAe;AAC9B,QAAI;AAEJ,QAAI,UAAU;AACb,eAAS,MAAM,cAAc,UAAU,QAAQ,QAAQ;AAAA,IACxD,OAAO;AACN,iBAAW,WAAW;AACtB,YAAM,QAAQ,MAAM,KAAK,cAAc,KAAK;AAE5C,YAAM,WACJ,MAAM,cAAc,aAAa,MAAM,QAAQ,UAAU,KAAK,KAC9D;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA,WAAW,KAAK,IAAI;AAAA,MACrB;AACD,cAAQ,OAAO,KAAK,qBAAqB,EAAE,SAAS,CAAC;AAAA,IACtD;AAGA,UAAM,SAAS,MAAM,KAAK,iBAAiB,OAAO,MAAM;AAGxD,UAAM,SAAS,MAAM,KAAK,iBAAiB,OAAO,UAAU,QAAQ,MAAM;AAC1E,UAAM,cAAc,oBAAoB,QAAQ,UAAU;AAAA,MACzD;AAAA,QACC;AAAA,QACA,WAAW;AAAA,QACX,SAAS,EAAE,MAAM,QAAQ,OAAO,CAAC,KAAK,EAAE;AAAA,MACzC;AAAA,MACA;AAAA,QACC;AAAA,QACA,WAAW,KAAK,IAAI;AAAA,QACpB,SAAS,EAAE,MAAM,QAAQ,OAAO,CAAC,OAAO,QAAQ,EAAE;AAAA,MACnD;AAAA,IACD,CAAC;AAED,WAAO,EAAE,SAAS,OAAO,SAAS;AAAA,EACnC;AACD;","names":["intent"]}

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

@@ -12,7 +12,7 @@ var A2AModule = class {
   /** Map of A2A server URLs to their corresponding tool instances */
   a2aPeerServers = /* @__PURE__ */ new Map();
   /** Map of session IDs to their A2A session state */
-  a2aSessions = /* @__PURE__ */ new Map();
+  a2aTasks = /* @__PURE__ */ new Map();
   agentId = randomUUID();
   /* FIXME */
   /**
@@ -53,22 +53,6 @@ var A2AModule = class {
     }
     return tools;
   }
-  /**
-   * Gets or creates an A2A session for the given session ID.
-   *
-   * @param threadId - The session identifier
-   * @returns A2ASession object with task and context IDs
-   */
-  getA2ASessionWithId = (threadId) => {
-    const a2aSession = this.a2aSessions.get(threadId) ?? {
-      taskId: void 0,
-      contextId: void 0
-    };
-    if (!this.a2aSessions.has(threadId)) {
-      this.a2aSessions.set(threadId, a2aSession);
-    }
-    return a2aSession;
-  };
   /**
    * Constructs a message payload for A2A communication.
    *
@@ -87,22 +71,18 @@ var A2AModule = class {
       // FIXME: it could be 'agent'
       metadata: {
         agentId: this.agentId,
-        type: "CHAT" /* CHAT */,
-        threadId
+        type: "CHAT" /* CHAT */
       },
       parts: [
         {
           kind: "text",
           text: query
         }
-      ]
+      ],
+      contextId: threadId
     };
-    const a2aSession = this.getA2ASessionWithId(threadId);
-    if (a2aSession.taskId) {
-      messagePayload.taskId = a2aSession.taskId;
-    }
-    if (a2aSession.contextId) {
-      messagePayload.contextId = a2aSession.contextId;
+    if (this.a2aTasks.has(threadId)) {
+      messagePayload.taskId = this.a2aTasks.get(threadId);
     }
     return messagePayload;
   }
@@ -117,20 +97,20 @@ var A2AModule = class {
    * @param threadId - The session identifier for context tracking
    * @returns Promise resolving to array of text responses from the agent
    */
-  async useTool(tool, messagePayload, threadId) {
+  async useTool(tool, query, threadId) {
     const finalText = [];
     const client = tool.client;
+    const messagePayload = this.getMessagePayload(query, threadId);
     const params = {
       message: messagePayload
     };
-    const a2aSession = this.getA2ASessionWithId(threadId);
     try {
       const stream = client.sendMessageStream(params);
       for await (const event of stream) {
         if (event.kind === "status-update") {
           const typedEvent = event;
           if (typedEvent.final && typedEvent.status.state !== "input-required") {
-            a2aSession.taskId = void 0;
+            this.a2aTasks.delete(threadId);
           }
           const texts = typedEvent.status.message?.parts.filter((part) => part.kind === "text").map((part) => part.text).join("\n");
           if (texts) {
@@ -138,19 +118,14 @@ var A2AModule = class {
           }
         } else if (event.kind === "message") {
           const msg = event;
-          if (msg.taskId && msg.taskId !== a2aSession.taskId) {
-            a2aSession.taskId = msg.taskId;
-          }
-          if (msg.contextId && msg.contextId !== a2aSession.contextId) {
-            a2aSession.contextId = msg.contextId;
+          const taskId = this.a2aTasks.get(threadId);
+          if (msg.taskId && msg.taskId !== taskId) {
+            this.a2aTasks.set(threadId, msg.taskId);
           }
         } else if (event.kind === "task") {
           const task = event;
-          if (task.id !== a2aSession.taskId) {
-            a2aSession.taskId = task.id;
-          }
-          if (task.contextId && task.contextId !== a2aSession.contextId) {
-            a2aSession.contextId = task.contextId;
+          if (task.id !== this.a2aTasks.get(threadId)) {
+            this.a2aTasks.set(threadId, task.id);
           }
         } else {
           loggers.a2a.warn("Received unknown event structure from stream:", {
@@ -173,4 +148,4 @@ ${finalText.join("\n")}`;
 export {
   A2AModule
 };
-//# sourceMappingURL=chunk-LJVVSUF7.js.map
+//# sourceMappingURL=chunk-MJEPJ62P.js.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/modules/a2a/a2a.module.ts"],"sourcesContent":["import { randomUUID } from \"node:crypto\";\nimport type {\n\tAgentCard,\n\tMessage,\n\tMessageSendParams,\n\tTask,\n\tTaskStatusUpdateEvent,\n\tTextPart,\n} from \"@a2a-js/sdk\";\nimport { A2AClient } from \"@a2a-js/sdk/client\";\nimport { ThreadType } from \"@/types/memory.js\";\nimport { loggers } from \"@/utils/logger.js\";\nimport { A2ATool } from \"./a2a.tool.js\";\n\n/**\n * Module for managing Agent-to-Agent (A2A) protocol connections.\n *\n * This module handles connections to other A2A-compatible agents, manages\n * conversation sessions, and provides an interface for inter-agent communication.\n * Supports multi-turn conversations with task and context tracking.\n */\nexport class A2AModule {\n\t/** Map of A2A server URLs to their corresponding tool instances */\n\tprivate a2aPeerServers: Map<string, A2ATool | null> = new Map();\n\t/** Map of session IDs to their A2A session state */\n\tprivate a2aTasks: Map<string, string> = new Map();\n\tprivate agentId: string = randomUUID(); /* FIXME */\n\n\t/**\n\t * Registers a new A2A peer server URL for connection.\n\t *\n\t * @param 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 * Constructs a message payload for A2A communication.\n\t *\n\t * Includes session context (task ID and context ID) if available\n\t * for maintaining conversation continuity.\n\t *\n\t * @param query - The message content to send\n\t * @param threadId - The session identifier\n\t * @returns Formatted Message object for A2A protocol\n\t */\n\tpublic getMessagePayload(query: string, threadId: string): Message {\n\t\tconst messagePayload: Message = {\n\t\t\tmessageId: randomUUID(),\n\t\t\tkind: \"message\",\n\t\t\trole: \"user\", // FIXME: it could be 'agent'\n\t\t\tmetadata: {\n\t\t\t\tagentId: this.agentId,\n\t\t\t\ttype: ThreadType.CHAT,\n\t\t\t},\n\t\t\tparts: [\n\t\t\t\t{\n\t\t\t\t\tkind: \"text\",\n\t\t\t\t\ttext: query,\n\t\t\t\t},\n\t\t\t],\n\t\t\tcontextId: threadId,\n\t\t};\n\n\t\tif (this.a2aTasks.has(threadId)) {\n\t\t\tmessagePayload.taskId = this.a2aTasks.get(threadId);\n\t\t}\n\n\t\treturn messagePayload;\n\t}\n\n\t/**\n\t * Executes an A2A tool by sending a message to the remote agent.\n\t *\n\t * Handles streaming responses, maintains session state, and extracts\n\t * text content from various event types in the response stream.\n\t *\n\t * @param tool - The A2ATool instance to use\n\t * @param messagePayload - The message to send to the agent\n\t * @param threadId - The session identifier for context tracking\n\t * @returns Promise resolving to array of text responses from the agent\n\t */\n\tpublic async useTool(\n\t\ttool: A2ATool,\n\t\tquery: string,\n\t\tthreadId: string,\n\t): Promise<string> {\n\t\tconst finalText: string[] = [];\n\t\tconst client = tool.client;\n\t\tconst messagePayload = this.getMessagePayload(query, threadId);\n\t\tconst params: MessageSendParams = {\n\t\t\tmessage: messagePayload,\n\t\t};\n\n\t\ttry {\n\t\t\tconst stream = client.sendMessageStream(params);\n\t\t\tfor await (const event of stream) {\n\t\t\t\tif (event.kind === \"status-update\") {\n\t\t\t\t\tconst typedEvent = event as TaskStatusUpdateEvent;\n\t\t\t\t\tif (\n\t\t\t\t\t\ttypedEvent.final &&\n\t\t\t\t\t\ttypedEvent.status.state !== \"input-required\"\n\t\t\t\t\t) {\n\t\t\t\t\t\tthis.a2aTasks.delete(threadId);\n\t\t\t\t\t}\n\t\t\t\t\t// TODO: handle 'file', 'data' parts\n\t\t\t\t\tconst texts = typedEvent.status.message?.parts\n\t\t\t\t\t\t.filter((part) => part.kind === \"text\")\n\t\t\t\t\t\t.map((part: TextPart) => part.text)\n\t\t\t\t\t\t.join(\"\\n\");\n\t\t\t\t\tif (texts) {\n\t\t\t\t\t\tfinalText.push(texts);\n\t\t\t\t\t}\n\t\t\t\t} else if (event.kind === \"message\") {\n\t\t\t\t\t// FIXME: handling text in 'message'?\n\t\t\t\t\tconst msg = event as Message;\n\t\t\t\t\tconst taskId = this.a2aTasks.get(threadId);\n\t\t\t\t\tif (msg.taskId && msg.taskId !== taskId) {\n\t\t\t\t\t\tthis.a2aTasks.set(threadId, msg.taskId);\n\t\t\t\t\t}\n\t\t\t\t} else if (event.kind === \"task\") {\n\t\t\t\t\t// establishing the Task ID\n\t\t\t\t\tconst task = event as Task;\n\t\t\t\t\tif (task.id !== this.a2aTasks.get(threadId)) {\n\t\t\t\t\t\tthis.a2aTasks.set(threadId, task.id);\n\t\t\t\t\t}\n\t\t\t\t} else {\n\t\t\t\t\tloggers.a2a.warn(\"Received unknown event structure from stream:\", {\n\t\t\t\t\t\tevent,\n\t\t\t\t\t});\n\t\t\t\t}\n\t\t\t}\n\t\t} catch (error) {\n\t\t\tloggers.a2a.error(\"Error communicating with agent:\", { error });\n\t\t\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;AAYnB,IAAM,YAAN,MAAgB;AAAA;AAAA,EAEd,iBAA8C,oBAAI,IAAI;AAAA;AAAA,EAEtD,WAAgC,oBAAI,IAAI;AAAA,EACxC,UAAkB,WAAW;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOrC,MAAa,iBAAiB,KAA4B;AACzD,SAAK,eAAe,IAAI,KAAK,IAAI;AAAA,EAClC;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAUA,MAAa,WAA+B;AAC3C,UAAM,QAAmB,CAAC;AAC1B,eAAW,OAAO,CAAC,GAAG,KAAK,eAAe,KAAK,CAAC,GAAG;AAClD,YAAM,OAAO,KAAK,eAAe,IAAI,GAAG;AACxC,UAAI,CAAC,QAAQ,CAAC,KAAK,SAAS;AAC3B,YAAI;AACH,gBAAM,SAAS,IAAI,UAAU,GAAG;AAChC,gBAAM,OAAkB,MAAM,OAAO,aAAa;AAClD,gBAAM,WAAW,KAAK,KAAK,WAAW,KAAK,GAAG;AAC9C,gBAAM,UAAU,IAAI,QAAQ,UAAU,QAAQ,IAAI;AAElD,gBAAM,KAAK,OAAO;AAAA,QACnB,SAAS,QAAa;AAErB,cAAI,MAAM;AACT,iBAAK,QAAQ;AAAA,UACd;AAAA,QACD;AAAA,MACD,OAAO;AACN,cAAM,KAAK,IAAI;AAAA,MAChB;AAAA,IACD;AACA,WAAO;AAAA,EACR;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAYO,kBAAkB,OAAe,UAA2B;AAClE,UAAM,iBAA0B;AAAA,MAC/B,WAAW,WAAW;AAAA,MACtB,MAAM;AAAA,MACN,MAAM;AAAA;AAAA,MACN,UAAU;AAAA,QACT,SAAS,KAAK;AAAA,QACd;AAAA,MACD;AAAA,MACA,OAAO;AAAA,QACN;AAAA,UACC,MAAM;AAAA,UACN,MAAM;AAAA,QACP;AAAA,MACD;AAAA,MACA,WAAW;AAAA,IACZ;AAEA,QAAI,KAAK,SAAS,IAAI,QAAQ,GAAG;AAChC,qBAAe,SAAS,KAAK,SAAS,IAAI,QAAQ;AAAA,IACnD;AAEA,WAAO;AAAA,EACR;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAaA,MAAa,QACZ,MACA,OACA,UACkB;AAClB,UAAM,YAAsB,CAAC;AAC7B,UAAM,SAAS,KAAK;AACpB,UAAM,iBAAiB,KAAK,kBAAkB,OAAO,QAAQ;AAC7D,UAAM,SAA4B;AAAA,MACjC,SAAS;AAAA,IACV;AAEA,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,iBAAK,SAAS,OAAO,QAAQ;AAAA,UAC9B;AAEA,gBAAM,QAAQ,WAAW,OAAO,SAAS,MACvC,OAAO,CAAC,SAAS,KAAK,SAAS,MAAM,EACrC,IAAI,CAAC,SAAmB,KAAK,IAAI,EACjC,KAAK,IAAI;AACX,cAAI,OAAO;AACV,sBAAU,KAAK,KAAK;AAAA,UACrB;AAAA,QACD,WAAW,MAAM,SAAS,WAAW;AAEpC,gBAAM,MAAM;AACZ,gBAAM,SAAS,KAAK,SAAS,IAAI,QAAQ;AACzC,cAAI,IAAI,UAAU,IAAI,WAAW,QAAQ;AACxC,iBAAK,SAAS,IAAI,UAAU,IAAI,MAAM;AAAA,UACvC;AAAA,QACD,WAAW,MAAM,SAAS,QAAQ;AAEjC,gBAAM,OAAO;AACb,cAAI,KAAK,OAAO,KAAK,SAAS,IAAI,QAAQ,GAAG;AAC5C,iBAAK,SAAS,IAAI,UAAU,KAAK,EAAE;AAAA,UACpC;AAAA,QACD,OAAO;AACN,kBAAQ,IAAI,KAAK,iDAAiD;AAAA,YACjE;AAAA,UACD,CAAC;AAAA,QACF;AAAA,MACD;AAAA,IACD,SAAS,OAAO;AACf,cAAQ,IAAI,MAAM,mCAAmC,EAAE,MAAM,CAAC;AAC9D,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-2RW77MJB.js → chunk-PUUZRUE6.js}

@@ -5,10 +5,10 @@ import {
 // src/services/a2a.service.ts
 import { randomUUID } from "crypto";
 var A2AService = class {
-  queryService;
+  queryStreamService;
   canceledTasks = /* @__PURE__ */ new Set();
-  constructor(queryService) {
-    this.queryService = queryService;
+  constructor(queryStreamService) {
+    this.queryStreamService = queryStreamService;
   }
   cancelTask = async (taskId, _eventBus) => {
     this.canceledTasks.add(taskId);
@@ -35,15 +35,15 @@ var A2AService = class {
   };
   async execute(requestContext, eventBus) {
     const userMessage = requestContext.userMessage;
-    const { agentId, type, threadId } = userMessage.metadata;
+    const threadId = userMessage.contextId;
+    const { agentId, type } = userMessage.metadata;
     const existingTask = requestContext.task;
     const taskId = existingTask?.id || randomUUID();
-    const contextId = userMessage.contextId || existingTask?.contextId || randomUUID();
     if (!existingTask) {
       const initialTask = {
         kind: "task",
         id: taskId,
-        contextId,
+        contextId: threadId,
         status: {
           state: "submitted",
           timestamp: (/* @__PURE__ */ new Date()).toISOString()
@@ -56,7 +56,7 @@ var A2AService = class {
     }
     const workingStatusUpdate = this.createTaskStatusUpdateEvent(
       taskId,
-      contextId,
+      threadId,
       "working"
     );
     eventBus.publish(workingStatusUpdate);
@@ -65,33 +65,55 @@ var A2AService = class {
       loggers.server.warn(`Empty message received for task ${taskId}.`);
       const failureUpdate = this.createTaskStatusUpdateEvent(
         taskId,
-        contextId,
+        threadId,
         "failed",
         "No message found to process."
       );
       eventBus.publish(failureUpdate);
       return;
     }
+    const stream = this.queryStreamService.handleQueryStream(
+      { userId: agentId, type, threadId },
+      message
+    );
     try {
-      const response = await this.queryService.handleQuery(
-        { userId: agentId, type, threadId },
-        message
-      );
-      if (this.canceledTasks.has(taskId)) {
-        loggers.server.info(`Task ${taskId} was canceled.`);
-        const canceledUpdate = this.createTaskStatusUpdateEvent(
-          taskId,
-          contextId,
-          "canceled"
-        );
-        eventBus.publish(canceledUpdate);
-        return;
+      let finalResponseText = "";
+      for await (const event of stream) {
+        if (this.canceledTasks.has(taskId)) {
+          loggers.server.info(`Task ${taskId} was canceled.`);
+          const canceledUpdate = this.createTaskStatusUpdateEvent(
+            taskId,
+            threadId,
+            "canceled"
+          );
+          eventBus.publish(canceledUpdate);
+          return;
+        }
+        if (event.event === "text_chunk") {
+          finalResponseText += event.data.delta;
+        } else if (event.event === "tool_start") {
+          const toolStartUpdate = this.createTaskStatusUpdateEvent(
+            taskId,
+            threadId,
+            "working",
+            JSON.stringify(event.data)
+          );
+          eventBus.publish(toolStartUpdate);
+        } else if (event.event === "tool_output") {
+          const toolOutputUpdate = this.createTaskStatusUpdateEvent(
+            taskId,
+            threadId,
+            "working",
+            JSON.stringify(event.data)
+          );
+          eventBus.publish(toolOutputUpdate);
+        }
       }
       const finalUpdate = this.createTaskStatusUpdateEvent(
         taskId,
-        contextId,
+        threadId,
         "completed",
-        response.content
+        finalResponseText
       );
       eventBus.publish(finalUpdate);
       loggers.server.info(`Task ${taskId} completed successfully.`);
@@ -99,7 +121,7 @@ var A2AService = class {
       loggers.server.error(`Error processing task ${taskId}:`, error);
       const errorUpdate = this.createTaskStatusUpdateEvent(
         taskId,
-        contextId,
+        threadId,
         "failed",
         `Agent error: ${error.message}`
       );
@@ -111,4 +133,4 @@ var A2AService = class {
 export {
   A2AService
 };
-//# sourceMappingURL=chunk-2RW77MJB.js.map
+//# sourceMappingURL=chunk-PUUZRUE6.js.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/services/a2a.service.ts"],"sourcesContent":["import { randomUUID } from \"node:crypto\";\nimport type { Task, TaskStatusUpdateEvent } from \"@a2a-js/sdk\";\nimport type {\n\tAgentExecutor,\n\tExecutionEventBus,\n\tRequestContext,\n} from \"@a2a-js/sdk/server\";\nimport type { ThreadType } from \"@/types/memory.js\";\nimport { loggers } from \"@/utils/logger.js\";\nimport type { QueryStreamService } from \"./query-stream.service.js\";\n\n/**\n * Implements the AgentExecutor interface from the a2a-js-sdk.\n * This service is responsible for the core business logic of executing an A2A task.\n */\nexport class A2AService implements AgentExecutor {\n\tprivate queryStreamService: QueryStreamService;\n\tprivate canceledTasks: Set<string> = new Set<string>();\n\n\tconstructor(queryStreamService: QueryStreamService) {\n\t\tthis.queryStreamService = queryStreamService;\n\t}\n\n\tpublic cancelTask = async (\n\t\ttaskId: string,\n\t\t_eventBus: ExecutionEventBus,\n\t): Promise<void> => {\n\t\tthis.canceledTasks.add(taskId);\n\t};\n\n\tprivate createTaskStatusUpdateEvent = (\n\t\ttaskId: string,\n\t\tcontextId: string,\n\t\tstate: \"working\" | \"failed\" | \"canceled\" | \"completed\",\n\t\tmessage?: string,\n\t): TaskStatusUpdateEvent => {\n\t\treturn {\n\t\t\tkind: \"status-update\",\n\t\t\ttaskId: taskId,\n\t\t\tcontextId: contextId,\n\t\t\tstatus: {\n\t\t\t\tstate: state,\n\t\t\t\tmessage: message\n\t\t\t\t\t? {\n\t\t\t\t\t\t\tkind: \"message\",\n\t\t\t\t\t\t\trole: \"agent\",\n\t\t\t\t\t\t\tmessageId: randomUUID(),\n\t\t\t\t\t\t\tparts: [{ kind: \"text\", text: message }],\n\t\t\t\t\t\t\ttaskId: taskId,\n\t\t\t\t\t\t\tcontextId: contextId,\n\t\t\t\t\t\t}\n\t\t\t\t\t: undefined,\n\t\t\t\ttimestamp: new Date().toISOString(),\n\t\t\t},\n\t\t\tfinal: state !== \"working\",\n\t\t};\n\t};\n\n\tasync execute(\n\t\trequestContext: RequestContext,\n\t\teventBus: ExecutionEventBus,\n\t): Promise<void> {\n\t\tconst userMessage = requestContext.userMessage;\n\t\t// A2A context ID === AIN ADK thread ID\n\t\tconst threadId = userMessage.contextId!; // TODO: no context id case\n\n\t\tconst { agentId, type } = userMessage.metadata as {\n\t\t\tagentId: string;\n\t\t\ttype: ThreadType;\n\t\t};\n\t\tconst existingTask = requestContext.task;\n\n\t\tconst taskId = existingTask?.id || randomUUID();\n\n\t\tif (!existingTask) {\n\t\t\tconst initialTask: Task = {\n\t\t\t\tkind: \"task\",\n\t\t\t\tid: taskId,\n\t\t\t\tcontextId: threadId,\n\t\t\t\tstatus: {\n\t\t\t\t\tstate: \"submitted\",\n\t\t\t\t\ttimestamp: new Date().toISOString(),\n\t\t\t\t},\n\t\t\t\thistory: [userMessage],\n\t\t\t\tmetadata: userMessage.metadata,\n\t\t\t\tartifacts: [],\n\t\t\t};\n\t\t\teventBus.publish(initialTask);\n\t\t}\n\n\t\tconst workingStatusUpdate = this.createTaskStatusUpdateEvent(\n\t\t\ttaskId,\n\t\t\tthreadId,\n\t\t\t\"working\",\n\t\t);\n\t\teventBus.publish(workingStatusUpdate);\n\n\t\tconst message: string = userMessage.parts\n\t\t\t.filter((part) => part.kind === \"text\")\n\t\t\t.map((part) => part.text)\n\t\t\t.join(\"\\n\");\n\t\tif (message.length === 0) {\n\t\t\tloggers.server.warn(`Empty message received for task ${taskId}.`);\n\t\t\tconst failureUpdate = this.createTaskStatusUpdateEvent(\n\t\t\t\ttaskId,\n\t\t\t\tthreadId,\n\t\t\t\t\"failed\",\n\t\t\t\t\"No message found to process.\",\n\t\t\t);\n\t\t\teventBus.publish(failureUpdate);\n\t\t\treturn;\n\t\t}\n\n\t\tconst stream = this.queryStreamService.handleQueryStream(\n\t\t\t{ userId: agentId, type, threadId },\n\t\t\tmessage,\n\t\t);\n\n\t\ttry {\n\t\t\tlet finalResponseText = \"\";\n\t\t\tfor await (const event of stream) {\n\t\t\t\tif (this.canceledTasks.has(taskId)) {\n\t\t\t\t\tloggers.server.info(`Task ${taskId} was canceled.`);\n\t\t\t\t\tconst canceledUpdate = this.createTaskStatusUpdateEvent(\n\t\t\t\t\t\ttaskId,\n\t\t\t\t\t\tthreadId,\n\t\t\t\t\t\t\"canceled\",\n\t\t\t\t\t);\n\t\t\t\t\teventBus.publish(canceledUpdate);\n\t\t\t\t\treturn;\n\t\t\t\t}\n\n\t\t\t\tif (event.event === \"text_chunk\") {\n\t\t\t\t\tfinalResponseText += event.data.delta;\n\t\t\t\t} else if (event.event === \"tool_start\") {\n\t\t\t\t\tconst toolStartUpdate = this.createTaskStatusUpdateEvent(\n\t\t\t\t\t\ttaskId,\n\t\t\t\t\t\tthreadId,\n\t\t\t\t\t\t\"working\",\n\t\t\t\t\t\tJSON.stringify(event.data),\n\t\t\t\t\t);\n\t\t\t\t\teventBus.publish(toolStartUpdate);\n\t\t\t\t} else if (event.event === \"tool_output\") {\n\t\t\t\t\tconst toolOutputUpdate = this.createTaskStatusUpdateEvent(\n\t\t\t\t\t\ttaskId,\n\t\t\t\t\t\tthreadId,\n\t\t\t\t\t\t\"working\",\n\t\t\t\t\t\tJSON.stringify(event.data),\n\t\t\t\t\t);\n\t\t\t\t\teventBus.publish(toolOutputUpdate);\n\t\t\t\t}\n\t\t\t}\n\n\t\t\tconst finalUpdate = this.createTaskStatusUpdateEvent(\n\t\t\t\ttaskId,\n\t\t\t\tthreadId,\n\t\t\t\t\"completed\",\n\t\t\t\tfinalResponseText,\n\t\t\t);\n\t\t\teventBus.publish(finalUpdate);\n\t\t\tloggers.server.info(`Task ${taskId} completed successfully.`);\n\t\t} catch (error: any) {\n\t\t\tloggers.server.error(`Error processing task ${taskId}:`, error);\n\t\t\tconst errorUpdate = this.createTaskStatusUpdateEvent(\n\t\t\t\ttaskId,\n\t\t\t\tthreadId,\n\t\t\t\t\"failed\",\n\t\t\t\t`Agent error: ${error.message}`,\n\t\t\t);\n\t\t\teventBus.publish(errorUpdate);\n\t\t}\n\t}\n}\n"],"mappings":";;;;;AAAA,SAAS,kBAAkB;AAepB,IAAM,aAAN,MAA0C;AAAA,EACxC;AAAA,EACA,gBAA6B,oBAAI,IAAY;AAAA,EAErD,YAAY,oBAAwC;AACnD,SAAK,qBAAqB;AAAA,EAC3B;AAAA,EAEO,aAAa,OACnB,QACA,cACmB;AACnB,SAAK,cAAc,IAAI,MAAM;AAAA,EAC9B;AAAA,EAEQ,8BAA8B,CACrC,QACA,WACA,OACA,YAC2B;AAC3B,WAAO;AAAA,MACN,MAAM;AAAA,MACN;AAAA,MACA;AAAA,MACA,QAAQ;AAAA,QACP;AAAA,QACA,SAAS,UACN;AAAA,UACA,MAAM;AAAA,UACN,MAAM;AAAA,UACN,WAAW,WAAW;AAAA,UACtB,OAAO,CAAC,EAAE,MAAM,QAAQ,MAAM,QAAQ,CAAC;AAAA,UACvC;AAAA,UACA;AAAA,QACD,IACC;AAAA,QACH,YAAW,oBAAI,KAAK,GAAE,YAAY;AAAA,MACnC;AAAA,MACA,OAAO,UAAU;AAAA,IAClB;AAAA,EACD;AAAA,EAEA,MAAM,QACL,gBACA,UACgB;AAChB,UAAM,cAAc,eAAe;AAEnC,UAAM,WAAW,YAAY;AAE7B,UAAM,EAAE,SAAS,KAAK,IAAI,YAAY;AAItC,UAAM,eAAe,eAAe;AAEpC,UAAM,SAAS,cAAc,MAAM,WAAW;AAE9C,QAAI,CAAC,cAAc;AAClB,YAAM,cAAoB;AAAA,QACzB,MAAM;AAAA,QACN,IAAI;AAAA,QACJ,WAAW;AAAA,QACX,QAAQ;AAAA,UACP,OAAO;AAAA,UACP,YAAW,oBAAI,KAAK,GAAE,YAAY;AAAA,QACnC;AAAA,QACA,SAAS,CAAC,WAAW;AAAA,QACrB,UAAU,YAAY;AAAA,QACtB,WAAW,CAAC;AAAA,MACb;AACA,eAAS,QAAQ,WAAW;AAAA,IAC7B;AAEA,UAAM,sBAAsB,KAAK;AAAA,MAChC;AAAA,MACA;AAAA,MACA;AAAA,IACD;AACA,aAAS,QAAQ,mBAAmB;AAEpC,UAAM,UAAkB,YAAY,MAClC,OAAO,CAAC,SAAS,KAAK,SAAS,MAAM,EACrC,IAAI,CAAC,SAAS,KAAK,IAAI,EACvB,KAAK,IAAI;AACX,QAAI,QAAQ,WAAW,GAAG;AACzB,cAAQ,OAAO,KAAK,mCAAmC,MAAM,GAAG;AAChE,YAAM,gBAAgB,KAAK;AAAA,QAC1B;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,MACD;AACA,eAAS,QAAQ,aAAa;AAC9B;AAAA,IACD;AAEA,UAAM,SAAS,KAAK,mBAAmB;AAAA,MACtC,EAAE,QAAQ,SAAS,MAAM,SAAS;AAAA,MAClC;AAAA,IACD;AAEA,QAAI;AACH,UAAI,oBAAoB;AACxB,uBAAiB,SAAS,QAAQ;AACjC,YAAI,KAAK,cAAc,IAAI,MAAM,GAAG;AACnC,kBAAQ,OAAO,KAAK,QAAQ,MAAM,gBAAgB;AAClD,gBAAM,iBAAiB,KAAK;AAAA,YAC3B;AAAA,YACA;AAAA,YACA;AAAA,UACD;AACA,mBAAS,QAAQ,cAAc;AAC/B;AAAA,QACD;AAEA,YAAI,MAAM,UAAU,cAAc;AACjC,+BAAqB,MAAM,KAAK;AAAA,QACjC,WAAW,MAAM,UAAU,cAAc;AACxC,gBAAM,kBAAkB,KAAK;AAAA,YAC5B;AAAA,YACA;AAAA,YACA;AAAA,YACA,KAAK,UAAU,MAAM,IAAI;AAAA,UAC1B;AACA,mBAAS,QAAQ,eAAe;AAAA,QACjC,WAAW,MAAM,UAAU,eAAe;AACzC,gBAAM,mBAAmB,KAAK;AAAA,YAC7B;AAAA,YACA;AAAA,YACA;AAAA,YACA,KAAK,UAAU,MAAM,IAAI;AAAA,UAC1B;AACA,mBAAS,QAAQ,gBAAgB;AAAA,QAClC;AAAA,MACD;AAEA,YAAM,cAAc,KAAK;AAAA,QACxB;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,MACD;AACA,eAAS,QAAQ,WAAW;AAC5B,cAAQ,OAAO,KAAK,QAAQ,MAAM,0BAA0B;AAAA,IAC7D,SAAS,OAAY;AACpB,cAAQ,OAAO,MAAM,yBAAyB,MAAM,KAAK,KAAK;AAC9D,YAAM,cAAc,KAAK;AAAA,QACxB;AAAA,QACA;AAAA,QACA;AAAA,QACA,gBAAgB,MAAM,OAAO;AAAA,MAC9B;AACA,eAAS,QAAQ,WAAW;AAAA,IAC7B;AAAA,EACD;AACD;","names":[]}

package/dist/esm/chunk-QHWDMU43.js

@@ -0,0 +1,40 @@
+import {
+  ThreadApiController
+} from "./chunk-RL6SRSSK.js";
+import {
+  AinHttpError
+} from "./chunk-ZBUAT73P.js";
+
+// src/routes/api/threads.routes.ts
+import {
+  Router
+} from "express";
+import { StatusCodes } from "http-status-codes";
+var createThreadApiRouter = (memoryModule) => {
+  const router = Router();
+  const threadApiController = new ThreadApiController(memoryModule);
+  const checkThreadMemory = (req, res, next) => {
+    const threadMemory = memoryModule.getThreadMemory();
+    if (!threadMemory) {
+      const error = new AinHttpError(
+        StatusCodes.SERVICE_UNAVAILABLE,
+        "Memory module is not initialized"
+      );
+      throw error;
+    }
+    next();
+  };
+  router.get("/", checkThreadMemory, threadApiController.handleGetUserThreads);
+  router.get("/:id", checkThreadMemory, threadApiController.handleGetThread);
+  router.delete(
+    "/:id",
+    checkThreadMemory,
+    threadApiController.handleDeleteThread
+  );
+  return router;
+};
+
+export {
+  createThreadApiRouter
+};
+//# sourceMappingURL=chunk-QHWDMU43.js.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/routes/api/threads.routes.ts"],"sourcesContent":["import {\n\ttype NextFunction,\n\ttype Request,\n\ttype Response,\n\tRouter,\n} from \"express\";\nimport { StatusCodes } from \"http-status-codes\";\nimport { ThreadApiController } from \"@/controllers/api/threads.api.controller.js\";\nimport type { MemoryModule } from \"@/modules/index.js\";\nimport { AinHttpError } from \"@/types/agent\";\n\nexport const createThreadApiRouter = (memoryModule: MemoryModule): Router => {\n\tconst router = Router();\n\tconst threadApiController = new ThreadApiController(memoryModule);\n\n\tconst checkThreadMemory = (\n\t\treq: Request,\n\t\tres: Response,\n\t\tnext: NextFunction,\n\t) => {\n\t\tconst threadMemory = memoryModule.getThreadMemory();\n\t\tif (!threadMemory) {\n\t\t\tconst error = new AinHttpError(\n\t\t\t\tStatusCodes.SERVICE_UNAVAILABLE,\n\t\t\t\t\"Memory module is not initialized\",\n\t\t\t);\n\t\t\tthrow error;\n\t\t}\n\t\tnext();\n\t};\n\n\t// APIs (prefix: /api/threads)\n\trouter.get(\"/\", checkThreadMemory, threadApiController.handleGetUserThreads);\n\trouter.get(\"/:id\", checkThreadMemory, threadApiController.handleGetThread);\n\trouter.delete(\n\t\t\"/:id\",\n\t\tcheckThreadMemory,\n\t\tthreadApiController.handleDeleteThread,\n\t);\n\n\treturn router;\n};\n"],"mappings":";;;;;;;;AAAA;AAAA,EAIC;AAAA,OACM;AACP,SAAS,mBAAmB;AAKrB,IAAM,wBAAwB,CAAC,iBAAuC;AAC5E,QAAM,SAAS,OAAO;AACtB,QAAM,sBAAsB,IAAI,oBAAoB,YAAY;AAEhE,QAAM,oBAAoB,CACzB,KACA,KACA,SACI;AACJ,UAAM,eAAe,aAAa,gBAAgB;AAClD,QAAI,CAAC,cAAc;AAClB,YAAM,QAAQ,IAAI;AAAA,QACjB,YAAY;AAAA,QACZ;AAAA,MACD;AACA,YAAM;AAAA,IACP;AACA,SAAK;AAAA,EACN;AAGA,SAAO,IAAI,KAAK,mBAAmB,oBAAoB,oBAAoB;AAC3E,SAAO,IAAI,QAAQ,mBAAmB,oBAAoB,eAAe;AACzE,SAAO;AAAA,IACN;AAAA,IACA;AAAA,IACA,oBAAoB;AAAA,EACrB;AAEA,SAAO;AACR;","names":[]}

package/dist/esm/chunk-RL6SRSSK.js

@@ -0,0 +1,45 @@
+// src/controllers/api/threads.api.controller.ts
+import { StatusCodes } from "http-status-codes";
+var ThreadApiController = class {
+  memoryModule;
+  constructor(memoryModule) {
+    this.memoryModule = memoryModule;
+  }
+  handleGetThread = async (req, res, next) => {
+    try {
+      const { id: threadId } = req.params;
+      const userId = res.locals.userId || "";
+      const threadMemory = this.memoryModule.getThreadMemory();
+      const thread = await threadMemory?.getThread(userId, threadId);
+      res.json(thread);
+    } catch (error) {
+      next(error);
+    }
+  };
+  handleDeleteThread = async (req, res, next) => {
+    try {
+      const { id: threadId } = req.params;
+      const userId = res.locals.userId || "";
+      const threadMemory = this.memoryModule.getThreadMemory();
+      await threadMemory?.deleteThread(userId, threadId);
+      res.status(StatusCodes.OK).send();
+    } catch (error) {
+      next(error);
+    }
+  };
+  handleGetUserThreads = async (_req, res, next) => {
+    try {
+      const userId = res.locals.userId || "";
+      const threadMemory = this.memoryModule.getThreadMemory();
+      const threads = await threadMemory?.listThreads(userId);
+      res.json(threads);
+    } catch (error) {
+      next(error);
+    }
+  };
+};
+
+export {
+  ThreadApiController
+};
+//# sourceMappingURL=chunk-RL6SRSSK.js.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/controllers/api/threads.api.controller.ts"],"sourcesContent":["import type { NextFunction, Request, Response } from \"express\";\nimport { StatusCodes } from \"http-status-codes\";\nimport type { MemoryModule } from \"@/modules/index.js\";\n\nexport class ThreadApiController {\n\tprivate memoryModule: MemoryModule;\n\n\tconstructor(memoryModule: MemoryModule) {\n\t\tthis.memoryModule = memoryModule;\n\t}\n\n\tpublic handleGetThread = async (\n\t\treq: Request,\n\t\tres: Response,\n\t\tnext: NextFunction,\n\t) => {\n\t\ttry {\n\t\t\tconst { id: threadId } = req.params as {\n\t\t\t\tid: string;\n\t\t\t};\n\t\t\tconst userId = res.locals.userId || \"\";\n\t\t\tconst threadMemory = this.memoryModule.getThreadMemory();\n\t\t\tconst thread = await threadMemory?.getThread(userId, threadId);\n\t\t\tres.json(thread);\n\t\t} catch (error) {\n\t\t\tnext(error);\n\t\t}\n\t};\n\n\tpublic handleDeleteThread = async (\n\t\treq: Request,\n\t\tres: Response,\n\t\tnext: NextFunction,\n\t) => {\n\t\ttry {\n\t\t\tconst { id: threadId } = req.params;\n\t\t\tconst userId = res.locals.userId || \"\";\n\t\t\tconst threadMemory = this.memoryModule.getThreadMemory();\n\t\t\tawait threadMemory?.deleteThread(userId, threadId);\n\t\t\tres.status(StatusCodes.OK).send();\n\t\t} catch (error) {\n\t\t\tnext(error);\n\t\t}\n\t};\n\n\tpublic handleGetUserThreads = async (\n\t\t_req: Request,\n\t\tres: Response,\n\t\tnext: NextFunction,\n\t) => {\n\t\ttry {\n\t\t\tconst userId = res.locals.userId || \"\";\n\t\t\tconst threadMemory = this.memoryModule.getThreadMemory();\n\t\t\tconst threads = await threadMemory?.listThreads(userId);\n\t\t\tres.json(threads);\n\t\t} catch (error) {\n\t\t\tnext(error);\n\t\t}\n\t};\n}\n"],"mappings":";AACA,SAAS,mBAAmB;AAGrB,IAAM,sBAAN,MAA0B;AAAA,EACxB;AAAA,EAER,YAAY,cAA4B;AACvC,SAAK,eAAe;AAAA,EACrB;AAAA,EAEO,kBAAkB,OACxB,KACA,KACA,SACI;AACJ,QAAI;AACH,YAAM,EAAE,IAAI,SAAS,IAAI,IAAI;AAG7B,YAAM,SAAS,IAAI,OAAO,UAAU;AACpC,YAAM,eAAe,KAAK,aAAa,gBAAgB;AACvD,YAAM,SAAS,MAAM,cAAc,UAAU,QAAQ,QAAQ;AAC7D,UAAI,KAAK,MAAM;AAAA,IAChB,SAAS,OAAO;AACf,WAAK,KAAK;AAAA,IACX;AAAA,EACD;AAAA,EAEO,qBAAqB,OAC3B,KACA,KACA,SACI;AACJ,QAAI;AACH,YAAM,EAAE,IAAI,SAAS,IAAI,IAAI;AAC7B,YAAM,SAAS,IAAI,OAAO,UAAU;AACpC,YAAM,eAAe,KAAK,aAAa,gBAAgB;AACvD,YAAM,cAAc,aAAa,QAAQ,QAAQ;AACjD,UAAI,OAAO,YAAY,EAAE,EAAE,KAAK;AAAA,IACjC,SAAS,OAAO;AACf,WAAK,KAAK;AAAA,IACX;AAAA,EACD;AAAA,EAEO,uBAAuB,OAC7B,MACA,KACA,SACI;AACJ,QAAI;AACH,YAAM,SAAS,IAAI,OAAO,UAAU;AACpC,YAAM,eAAe,KAAK,aAAa,gBAAgB;AACvD,YAAM,UAAU,MAAM,cAAc,YAAY,MAAM;AACtD,UAAI,KAAK,OAAO;AAAA,IACjB,SAAS,OAAO;AACf,WAAK,KAAK;AAAA,IACX;AAAA,EACD;AACD;","names":[]}

package/dist/esm/chunk-RNA32QWV.js

@@ -0,0 +1,128 @@
+import {
+  MCPTool
+} from "./chunk-3LLHEUY2.js";
+import {
+  loggers
+} from "./chunk-N7VYM2R5.js";
+
+// src/modules/mcp/mcp.module.ts
+import { Client } from "@modelcontextprotocol/sdk/client/index.js";
+import { SSEClientTransport } from "@modelcontextprotocol/sdk/client/sse.js";
+import { StdioClientTransport } from "@modelcontextprotocol/sdk/client/stdio.js";
+import { StreamableHTTPClientTransport } from "@modelcontextprotocol/sdk/client/streamableHttp.js";
+import { WebSocketClientTransport } from "@modelcontextprotocol/sdk/client/websocket.js";
+var MCPModule = class {
+  mcpConfigs = /* @__PURE__ */ new Map();
+  /** Map of MCP server names to their client instances */
+  mcpClientMap = /* @__PURE__ */ new Map();
+  /** Array of all discovered tools from connected MCP servers */
+  tools = [];
+  addMCPServerConfig(configs) {
+    for (const [name, config] of Object.entries(configs)) {
+      this.mcpConfigs.set(name, config);
+    }
+  }
+  async connectToServers() {
+    for (const [name, config] of this.mcpConfigs.entries()) {
+      try {
+        const mcpClient = new Client({ name, version: "1.0.0" });
+        switch (config.type) {
+          case "stdio": {
+            const transport = new StdioClientTransport(config.params);
+            await mcpClient.connect(transport);
+            break;
+          }
+          case "websocket": {
+            const transport = new WebSocketClientTransport(config.url);
+            await mcpClient.connect(transport);
+            break;
+          }
+          case "sse": {
+            const transport = new SSEClientTransport(
+              config.url,
+              config.options
+            );
+            await mcpClient.connect(transport);
+            break;
+          }
+          case "streamableHttp": {
+            const transport = new StreamableHTTPClientTransport(
+              config.url,
+              config.options
+            );
+            await mcpClient.connect(transport);
+            break;
+          }
+          default:
+            loggers.mcp.error("Unsupported MCP config type");
+            break;
+        }
+        this.mcpClientMap.set(name, mcpClient);
+        const toolList = await mcpClient.listTools();
+        const newToolList = toolList.tools.map((tool) => {
+          return new MCPTool(name, tool);
+        });
+        this.tools.push(...newToolList);
+        loggers.mcp.info("Connected to MCP server with tools:", {
+          tools: newToolList.map((tool) => tool.id)
+        });
+      } catch (error) {
+        loggers.mcp.error(`Failed to connect to MCP server ${name}`, { error });
+      }
+    }
+  }
+  /**
+   * Returns all available tools from connected MCP servers.
+   *
+   * @returns Array of MCPTool instances representing available tools
+   */
+  getTools() {
+    return this.tools;
+  }
+  /**
+   * Executes a tool on its corresponding MCP server.
+   *
+   * @param tool - The MCPTool instance to execute
+   * @param _args - Arguments to pass to the tool
+   * @returns Promise resolving to the tool's execution result
+   * @throws Error if the MCP server for the tool is not found
+   */
+  async useTool(tool, _args) {
+    const { serverName, mcpTool } = tool;
+    const toolName = mcpTool.name;
+    const mcp = this.mcpClientMap.get(serverName);
+    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);
+      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.
+   *
+   * Should be called when shutting down the application to ensure
+   * all MCP connections are properly closed.
+   */
+  async cleanup() {
+    this.mcpClientMap.forEach((mcpClient) => {
+      mcpClient.close();
+    });
+  }
+};
+
+export {
+  MCPModule
+};
+//# sourceMappingURL=chunk-RNA32QWV.js.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/modules/mcp/mcp.module.ts"],"sourcesContent":["import { Client } from \"@modelcontextprotocol/sdk/client/index.js\";\nimport { SSEClientTransport } from \"@modelcontextprotocol/sdk/client/sse.js\";\nimport { StdioClientTransport } from \"@modelcontextprotocol/sdk/client/stdio.js\";\nimport { StreamableHTTPClientTransport } from \"@modelcontextprotocol/sdk/client/streamableHttp.js\";\nimport { WebSocketClientTransport } from \"@modelcontextprotocol/sdk/client/websocket.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\tprivate mcpConfigs: Map<string, MCPConfig> = new Map();\n\t/** Map of MCP server names to their client instances */\n\tprivate mcpClientMap: Map<string, Client> = new Map();\n\t/** Array of all discovered tools from connected MCP servers */\n\tprivate tools: MCPTool[] = [];\n\n\taddMCPServerConfig(configs: { [name: string]: MCPConfig }): void {\n\t\tfor (const [name, config] of Object.entries(configs)) {\n\t\t\tthis.mcpConfigs.set(name, config);\n\t\t}\n\t}\n\n\tasync connectToServers(): Promise<void> {\n\t\tfor (const [name, config] of this.mcpConfigs.entries()) {\n\t\t\ttry {\n\t\t\t\tconst mcpClient = new Client({ name, version: \"1.0.0\" });\n\t\t\t\tswitch (config.type) {\n\t\t\t\t\tcase \"stdio\": {\n\t\t\t\t\t\tconst transport = new StdioClientTransport(config.params);\n\t\t\t\t\t\tawait mcpClient.connect(transport);\n\t\t\t\t\t\tbreak;\n\t\t\t\t\t}\n\t\t\t\t\tcase \"websocket\": {\n\t\t\t\t\t\tconst transport = new WebSocketClientTransport(config.url);\n\t\t\t\t\t\tawait mcpClient.connect(transport);\n\t\t\t\t\t\tbreak;\n\t\t\t\t\t}\n\t\t\t\t\tcase \"sse\": {\n\t\t\t\t\t\tconst transport = new SSEClientTransport(\n\t\t\t\t\t\t\tconfig.url,\n\t\t\t\t\t\t\tconfig.options,\n\t\t\t\t\t\t);\n\t\t\t\t\t\tawait mcpClient.connect(transport);\n\t\t\t\t\t\tbreak;\n\t\t\t\t\t}\n\t\t\t\t\tcase \"streamableHttp\": {\n\t\t\t\t\t\tconst transport = new StreamableHTTPClientTransport(\n\t\t\t\t\t\t\tconfig.url,\n\t\t\t\t\t\t\tconfig.options,\n\t\t\t\t\t\t);\n\t\t\t\t\t\tawait mcpClient.connect(transport);\n\t\t\t\t\t\tbreak;\n\t\t\t\t\t}\n\t\t\t\t\tdefault:\n\t\t\t\t\t\t// This cannot happen.\n\t\t\t\t\t\tloggers.mcp.error(\"Unsupported MCP config type\");\n\t\t\t\t\t\tbreak;\n\t\t\t\t}\n\n\t\t\t\tthis.mcpClientMap.set(name, mcpClient);\n\t\t\t\tconst toolList = await mcpClient.listTools();\n\t\t\t\tconst newToolList = toolList.tools.map((tool) => {\n\t\t\t\t\treturn new MCPTool(name, tool);\n\t\t\t\t});\n\t\t\t\tthis.tools.push(...newToolList);\n\t\t\t\tloggers.mcp.info(\"Connected to MCP server with tools:\", {\n\t\t\t\t\ttools: newToolList.map((tool) => tool.id),\n\t\t\t\t});\n\t\t\t} catch (error) {\n\t\t\t\tloggers.mcp.error(`Failed to connect to MCP server ${name}`, { error });\n\t\t\t}\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.mcpClientMap.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.mcpClientMap.forEach((mcpClient: Client) => {\n\t\t\tmcpClient.close();\n\t\t});\n\t}\n}\n"],"mappings":";;;;;;;;AAAA,SAAS,cAAc;AACvB,SAAS,0BAA0B;AACnC,SAAS,4BAA4B;AACrC,SAAS,qCAAqC;AAC9C,SAAS,gCAAgC;AA0BlC,IAAM,YAAN,MAAgB;AAAA,EACd,aAAqC,oBAAI,IAAI;AAAA;AAAA,EAE7C,eAAoC,oBAAI,IAAI;AAAA;AAAA,EAE5C,QAAmB,CAAC;AAAA,EAE5B,mBAAmB,SAA8C;AAChE,eAAW,CAAC,MAAM,MAAM,KAAK,OAAO,QAAQ,OAAO,GAAG;AACrD,WAAK,WAAW,IAAI,MAAM,MAAM;AAAA,IACjC;AAAA,EACD;AAAA,EAEA,MAAM,mBAAkC;AACvC,eAAW,CAAC,MAAM,MAAM,KAAK,KAAK,WAAW,QAAQ,GAAG;AACvD,UAAI;AACH,cAAM,YAAY,IAAI,OAAO,EAAE,MAAM,SAAS,QAAQ,CAAC;AACvD,gBAAQ,OAAO,MAAM;AAAA,UACpB,KAAK,SAAS;AACb,kBAAM,YAAY,IAAI,qBAAqB,OAAO,MAAM;AACxD,kBAAM,UAAU,QAAQ,SAAS;AACjC;AAAA,UACD;AAAA,UACA,KAAK,aAAa;AACjB,kBAAM,YAAY,IAAI,yBAAyB,OAAO,GAAG;AACzD,kBAAM,UAAU,QAAQ,SAAS;AACjC;AAAA,UACD;AAAA,UACA,KAAK,OAAO;AACX,kBAAM,YAAY,IAAI;AAAA,cACrB,OAAO;AAAA,cACP,OAAO;AAAA,YACR;AACA,kBAAM,UAAU,QAAQ,SAAS;AACjC;AAAA,UACD;AAAA,UACA,KAAK,kBAAkB;AACtB,kBAAM,YAAY,IAAI;AAAA,cACrB,OAAO;AAAA,cACP,OAAO;AAAA,YACR;AACA,kBAAM,UAAU,QAAQ,SAAS;AACjC;AAAA,UACD;AAAA,UACA;AAEC,oBAAQ,IAAI,MAAM,6BAA6B;AAC/C;AAAA,QACF;AAEA,aAAK,aAAa,IAAI,MAAM,SAAS;AACrC,cAAM,WAAW,MAAM,UAAU,UAAU;AAC3C,cAAM,cAAc,SAAS,MAAM,IAAI,CAAC,SAAS;AAChD,iBAAO,IAAI,QAAQ,MAAM,IAAI;AAAA,QAC9B,CAAC;AACD,aAAK,MAAM,KAAK,GAAG,WAAW;AAC9B,gBAAQ,IAAI,KAAK,uCAAuC;AAAA,UACvD,OAAO,YAAY,IAAI,CAAC,SAAS,KAAK,EAAE;AAAA,QACzC,CAAC;AAAA,MACF,SAAS,OAAO;AACf,gBAAQ,IAAI,MAAM,mCAAmC,IAAI,IAAI,EAAE,MAAM,CAAC;AAAA,MACvE;AAAA,IACD;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,aAAa,IAAI,UAAU;AAE5C,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,aAAa,QAAQ,CAAC,cAAsB;AAChD,gBAAU,MAAM;AAAA,IACjB,CAAC;AAAA,EACF;AACD;","names":[]}

package/dist/esm/{chunk-HOFAK5J2.js → chunk-TV2GJAEH.js}

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

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

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

package/dist/esm/{chunk-PHKK5NHK.js → chunk-VNQCUGC3.js}

@@ -49,4 +49,4 @@ var ModelModule = class {
 export {
   ModelModule
 };
-//# sourceMappingURL=chunk-PHKK5NHK.js.map
+//# sourceMappingURL=chunk-VNQCUGC3.js.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/modules/models/model.module.ts"],"sourcesContent":["import type { BaseModel } from \"./base.model.js\";\n\n/**\n * Module for managing multiple AI model implementations.\n *\n * Allows registration of multiple models and provides a unified interface\n * for accessing them. Supports setting a default model for convenience.\n * All models must extend the BaseModel class for consistent API.\n *\n * @example\n * ```typescript\n * const modelModule = new ModelModule();\n * modelModule.addModel(\"gpt-4\", new OpenAIModel(\"gpt-4\"), true);\n * modelModule.addModel(\"gemini\", new GeminiModel(\"gemini-pro\"));\n *\n * const defaultModel = modelModule.getModel();\n * const specificModel = modelModule.getModel(\"gemini\");\n * ```\n */\nexport class ModelModule {\n\t/** Registry of available models indexed by name */\n\tprivate models: { [name: string]: BaseModel<unknown, unknown> } = {};\n\t/** Name of the default model to use when none specified */\n\tprivate defaultModelName?: string;\n\n\t/**\n\t * Registers a new model with the module.\n\t *\n\t * @param name - Unique identifier for the model\n\t * @param model - Model instance implementing BaseModel\n\t * @param isDefault - Whether to set this as the default model\n\t */\n\tpublic addModel(\n\t\tname: string,\n\t\tmodel: BaseModel<unknown, unknown>,\n\t\tisDefault?: boolean,\n\t) {\n\t\tthis.models[name] = model;\n\t\tif (isDefault || !this.defaultModelName) {\n\t\t\tthis.defaultModelName = name;\n\t\t}\n\t}\n\n\t/**\n\t * Retrieves a model by name or returns the default model.\n\t *\n\t * @param name - Optional model name to retrieve\n\t * @returns The requested model or default model if name not provided\n\t * @throws Error if no default model is set and name is not provided\n\t */\n\tpublic getModel(name?: string): BaseModel<unknown, unknown> {\n\t\tif (!this.defaultModelName) {\n\t\t\tthrow Error(\"No default model\");\n\t\t}\n\n\t\tif (!name || !this.models[name]) {\n\t\t\treturn this.models[this.defaultModelName];\n\t\t}\n\t\treturn this.models[name];\n\t}\n\n\t/**\n\t * Returns information about all registered models.\n\t *\n\t * @returns Object containing all models and the default model name\n\t */\n\tpublic getModelList() {\n\t\treturn {\n\t\t\tmodels: Object.keys(this.models),\n\t\t\tdefaultModelName: this.defaultModelName,\n\t\t};\n\t}\n}\n"],"mappings":";AAmBO,IAAM,cAAN,MAAkB;AAAA;AAAA,EAEhB,SAA0D,CAAC;AAAA;AAAA,EAE3D;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EASD,SACN,MACA,OACA,WACC;AACD,SAAK,OAAO,IAAI,IAAI;AACpB,QAAI,aAAa,CAAC,KAAK,kBAAkB;AACxC,WAAK,mBAAmB;AAAA,IACzB;AAAA,EACD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EASO,SAAS,MAA4C;AAC3D,QAAI,CAAC,KAAK,kBAAkB;AAC3B,YAAM,MAAM,kBAAkB;AAAA,IAC/B;AAEA,QAAI,CAAC,QAAQ,CAAC,KAAK,OAAO,IAAI,GAAG;AAChC,aAAO,KAAK,OAAO,KAAK,gBAAgB;AAAA,IACzC;AACA,WAAO,KAAK,OAAO,IAAI;AAAA,EACxB;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOO,eAAe;AACrB,WAAO;AAAA,MACN,QAAQ,OAAO,KAAK,KAAK,MAAM;AAAA,MAC/B,kBAAkB,KAAK;AAAA,IACxB;AAAA,EACD;AACD;","names":[]}

package/dist/esm/{chunk-FL3JRQM4.js → chunk-ZBUAT73P.js}

@@ -10,4 +10,4 @@ var AinHttpError = class extends Error {
 export {
   AinHttpError
 };
-//# sourceMappingURL=chunk-FL3JRQM4.js.map
+//# sourceMappingURL=chunk-ZBUAT73P.js.map

package/dist/esm/{chunk-FL3JRQM4.js.map → chunk-ZBUAT73P.js.map}

@@ -1 +1 @@
-{"version":3,"sources":["../../src/types/agent.ts"],"sourcesContent":["import type { StatusCodes } from \"http-status-codes\";\n\n/**\n * Custom prompts configuration for the agent.\n *\n * Allows customization of agent behavior through system and agent-level prompts.\n */\nexport type AinAgentPrompts = {\n\t/** Agent-level prompt that defines the agent's personality and behavior */\n\tagent?: string;\n\t/** System-level prompt that provides context and instructions */\n\tsystem?: string;\n};\n\n/**\n * Agent manifest containing essential metadata and configuration.\n *\n * The manifest defines the agent's identity, version, and optional configuration\n * such as the public URL for A2A protocol support and custom prompts.\n *\n * @example\n * ```typescript\n * const manifest: AinAgentManifest = {\n *   name: \"CustomerSupportAgent\",\n *   description: \"AI agent for handling customer support queries\",\n *   version: \"1.0.0\",\n *   url: \"https://api.example.com/agent\",\n *   prompts: {\n *     system: \"You are a helpful customer support assistant.\",\n *     agent: \"Always be polite and provide detailed solutions.\"\n *   }\n * };\n * ```\n */\nexport type AinAgentManifest = {\n\t/** Unique name identifier for the agent */\n\tname: string;\n\t/** Human-readable description of the agent's purpose and capabilities */\n\tdescription: string;\n\t/** Semantic version of the agent (e.g., \"1.0.0\") */\n\tversion: string;\n\t/** Optional public URL for A2A protocol discovery and communication */\n\turl?: string;\n\t/** Optional custom prompts to configure agent behavior */\n\tprompts?: AinAgentPrompts;\n};\n\nexport class AinHttpError extends Error {\n\tpublic status?: StatusCodes;\n\n\tconstructor(status: StatusCodes, message: string) {\n\t\tsuper(message);\n\t\tthis.status = status;\n\t}\n}\n"],"mappings":";AA+CO,IAAM,eAAN,cAA2B,MAAM;AAAA,EAChC;AAAA,EAEP,YAAY,QAAqB,SAAiB;AACjD,UAAM,OAAO;AACb,SAAK,SAAS;AAAA,EACf;AACD;","names":[]}
+{"version":3,"sources":["../../src/types/agent.ts"],"sourcesContent":["import type { StatusCodes } from \"http-status-codes\";\n\n/**\n * Custom prompts configuration for the agent.\n *\n * Allows customization of agent behavior through system and agent-level prompts.\n */\nexport type AinAgentPrompts = {\n\t/** Agent-level prompt that defines the agent's personality and behavior */\n\tagent?: string;\n\t/** System-level prompt that provides context and instructions */\n\tsystem?: string;\n};\n\n/**\n * Agent manifest containing essential metadata and configuration.\n *\n * The manifest defines the agent's identity, version, and optional configuration\n * such as the public URL for A2A protocol support and custom prompts.\n *\n * @example\n * ```typescript\n * const manifest: AinAgentManifest = {\n *   name: \"CustomerSupportAgent\",\n *   description: \"AI agent for handling customer support queries\",\n *   version: \"1.0.0\",\n *   url: \"https://api.example.com/agent\",\n *   prompts: {\n *     system: \"You are a helpful customer support assistant.\",\n *     agent: \"Always be polite and provide detailed solutions.\"\n *   }\n * };\n * ```\n */\nexport type AinAgentManifest = {\n\t/** Unique name identifier for the agent */\n\tname: string;\n\t/** Human-readable description of the agent's purpose and capabilities */\n\tdescription: string;\n\t/** Optional public URL for A2A protocol discovery and communication */\n\turl?: string;\n\t/** Optional custom prompts to configure agent behavior */\n\tprompts?: AinAgentPrompts;\n};\n\nexport class AinHttpError extends Error {\n\tpublic status?: StatusCodes;\n\n\tconstructor(status: StatusCodes, message: string) {\n\t\tsuper(message);\n\t\tthis.status = status;\n\t}\n}\n"],"mappings":";AA6CO,IAAM,eAAN,cAA2B,MAAM;AAAA,EAChC;AAAA,EAEP,YAAY,QAAqB,SAAiB;AACjD,UAAM,OAAO;AACb,SAAK,SAAS;AAAA,EACf;AACD;","names":[]}