@ainetwork/adk 0.1.12 → 0.1.13

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

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/routes/api/model.routes.ts"],"sourcesContent":["import { Router } from \"express\";\nimport { ModelApiController } from \"@/controllers/api/model.api.controller.js\";\nimport type { ModelModule } from \"@/modules/index.js\";\n\nexport const createModelApiRouter = (modelModule: ModelModule): Router => {\n\tconst router = Router();\n\tconst modelApiController = new ModelApiController(modelModule);\n\n\t// APIs (prefix: /api/model)\n\trouter.get(\"/\", modelApiController.handleModelList);\n\n\treturn router;\n};\n"],"mappings":";;;;;AAAA,SAAS,cAAc;AAIhB,IAAM,uBAAuB,CAAC,gBAAqC;AACzE,QAAM,SAAS,OAAO;AACtB,QAAM,qBAAqB,IAAI,mBAAmB,WAAW;AAG7D,SAAO,IAAI,KAAK,mBAAmB,eAAe;AAElD,SAAO;AACR;","names":[]}

package/dist/esm/{chunk-2BITOA2R.js → chunk-CDFLKADH.js}

@@ -5,4 +5,4 @@ var BaseModel = class {
 export {
   BaseModel
 };
-//# sourceMappingURL=chunk-2BITOA2R.js.map
+//# sourceMappingURL=chunk-CDFLKADH.js.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/modules/models/base.model.ts"],"sourcesContent":["import type { ThreadObject } from \"@/types/memory.js\";\nimport type { LLMStream } from \"@/types/stream.js\";\nimport type { FetchResponse, IAgentTool } from \"@/types/tool.js\";\n\n/**\n * Abstract base class for AI model implementations.\n *\n * Provides a common interface for different AI model providers (OpenAI, Gemini, etc.)\n * to integrate with the AIN-ADK framework. Each model implementation must handle\n * message formatting, tool conversion, and API communication.\n *\n * @typeParam MessageType - The message format used by the specific model API\n * @typeParam FunctionType - The function/tool format used by the specific model API\n */\nexport abstract class BaseModel<MessageType, FunctionType> {\n\t/**\n\t * Generates an array of messages from thread and current query.\n\t *\n\t * @param query - Current user query\n\t * @param thread - Previous conversation history\n\t * @param systemPrompt - Optional system prompt to set context\n\t * @returns Array of messages formatted for the specific model API\n\t */\n\tabstract generateMessages(params: {\n\t\tquery: string;\n\t\tthread?: ThreadObject;\n\t\tsystemPrompt?: string;\n\t}): MessageType[];\n\n\t/**\n\t * Appends a new message to the existing message array.\n\t *\n\t * @param messages - Existing message array to expand\n\t * @param message - New message content to append\n\t */\n\tabstract appendMessages(messages: MessageType[], message: string): void;\n\n\t/**\n\t * Converts protocol-agnostic tools to model-specific function format.\n\t *\n\t * @param tools - Array of agent tools from MCP or A2A sources\n\t * @returns Array of functions in the format required by the model API\n\t */\n\tabstract convertToolsToFunctions(tools: IAgentTool[]): FunctionType[];\n\n\t/**\n\t * Fetches a response from the model API without tool support.\n\t *\n\t * @param messages - Array of messages to send to the model\n\t * @returns Promise resolving to the model's response\n\t */\n\tabstract fetch(messages: MessageType[]): Promise<FetchResponse>;\n\n\t/**\n\t * Fetches a response from the model API with tool/function support.\n\t *\n\t * @param messages - Array of messages to send to the model\n\t * @param functions - Array of available functions/tools the model can call\n\t * @returns Promise resolving to the model's response, possibly including tool calls\n\t */\n\tabstract fetchWithContextMessage(\n\t\tmessages: MessageType[],\n\t\tfunctions: FunctionType[],\n\t): Promise<FetchResponse>;\n\n\t/**\n\t * Fetches a streaming response from the model API with tool/function support.\n\t *\n\t * Returns a standardized LLMStream that can be used consistently across\n\t * different AI model providers. Each implementation should convert their\n\t * provider-specific stream format to the common StreamChunk interface.\n\t *\n\t * @param messages - Array of messages to send to the model\n\t * @param functions - Array of available functions/tools the model can call\n\t * @returns Promise resolving to an LLMStream for consistent iteration\n\t */\n\tabstract fetchStreamWithContextMessage(\n\t\tmessages: MessageType[],\n\t\tfunctions: FunctionType[],\n\t): Promise<LLMStream>;\n}\n"],"mappings":";AAcO,IAAe,YAAf,MAAoD;AAkE3D;","names":[]}

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

@@ -3,6 +3,7 @@ import {
 } from "./chunk-N7VYM2R5.js";
 
 // src/services/query.service.ts
+import { randomUUID } from "crypto";
 var QueryService = class {
   modelModule;
   a2aModule;
@@ -36,11 +37,11 @@ var QueryService = class {
    * - Processing tool calls iteratively until completion
    *
    * @param query - The user's input query
-   * @param sessionId - Session identifier for context
-   * @param sessionHistory - Previous conversation history
+   * @param threadId - Thread identifier for context
+   * @param thread - Previous conversation history
    * @returns Object containing process steps and final response
    */
-  async intentFulfilling(query, sessionId, sessionHistory) {
+  async intentFulfilling(query, threadId, thread) {
     const systemPrompt = `
 Today is ${(/* @__PURE__ */ new Date()).toLocaleDateString()}.
 
@@ -51,7 +52,7 @@ ${this.prompts?.system || ""}
     const modelInstance = this.modelModule.getModel();
     const messages = modelInstance.generateMessages({
       query,
-      sessionHistory,
+      thread,
       systemPrompt: systemPrompt.trim()
     });
     const tools = [];
@@ -78,7 +79,7 @@ ${this.prompts?.system || ""}
       if (toolCalls) {
         const messagePayload = this.a2aModule?.getMessagePayload(
           query,
-          sessionId
+          threadId
         );
         for (const toolCall of toolCalls) {
           const toolName = toolCall.name;
@@ -96,7 +97,7 @@ ${this.prompts?.system || ""}
             toolResult = await this.a2aModule.useTool(
               selectedTool,
               messagePayload,
-              sessionId
+              threadId
             );
           } else {
             loggers.intent.warn(
@@ -120,38 +121,80 @@ ${this.prompts?.system || ""}
     };
     return botResponse;
   }
+  /**
+   * Generates a title for the conversation based on the query.
+   *
+   * @param query - The user's input query
+   * @returns Promise resolving to a generated title
+   */
+  async generateTitle(query) {
+    const DEFAULT_TITLE = "New Chat";
+    try {
+      const modelInstance = this.modelModule.getModel();
+      const messages = modelInstance.generateMessages({
+        query,
+        systemPrompt: `You are a helpful assistant that generates titles for conversations.
+  Please analyze the user's query and create a concise title that accurately reflects the conversation's core topic.
+  The title must be no more than 5 words long.
+  Respond with only the title. Do not include any punctuation or extra explanations.`
+      });
+      const response = await modelInstance.fetch(messages);
+      return response.content || DEFAULT_TITLE;
+    } catch (error) {
+      loggers.intent.error("Error generating title", {
+        error,
+        query
+      });
+      return DEFAULT_TITLE;
+    }
+  }
   /**
    * Main entry point for processing user queries.
    *
    * Handles the complete query lifecycle:
-   * 1. Loads session history from memory
+   * 1. Loads thread history from memory
    * 2. Detects intent from the query
    * 3. Fulfills the intent with AI response
    * 4. Updates conversation history
    *
+   * @param type - The type of thread (e.g., chat, workflow)
+   * @param userId - The user's unique identifier
+   * @param threadId - Unique thread identifier
    * @param query - The user's input query
-   * @param sessionId - Unique session identifier
-   * @param userId - Unique user identifier
-   * @returns Object containing the AI-generated response
    */
-  async handleQuery(query, sessionId, userId) {
+  async handleQuery(threadMetadata, query) {
+    const { type, userId } = threadMetadata;
     const queryStartAt = Date.now();
-    const sessionMemory = this.memoryModule?.getSessionMemory();
-    const session = !userId ? void 0 : await sessionMemory?.getSession(userId, sessionId);
-    const intent = this.intentTriggering(query);
-    const result = await this.intentFulfilling(query, sessionId, session);
-    if (userId) {
-      await sessionMemory?.addChatToSession(userId, sessionId, {
+    const threadMemory = this.memoryModule?.getThreadMemory();
+    let threadId = threadMetadata.threadId;
+    let thread;
+    if (threadId) {
+      thread = await threadMemory?.getThread(type, userId, threadId);
+    } else {
+      threadId = randomUUID();
+      const title = await this.generateTitle(query);
+      const metadata = await threadMemory?.createThread(type, userId, threadId, title) || {
+        type,
+        threadId,
+        title,
+        updatedAt: Date.now()
+      };
+      loggers.intent.info("Create new thread", { metadata });
+    }
+    const _intent = this.intentTriggering(query);
+    const result = await this.intentFulfilling(query, threadId, thread);
+    await threadMemory?.addMessagesToThread(userId, threadId, [
+      {
         role: "USER" /* USER */,
         timestamp: queryStartAt,
         content: { type: "text", parts: [query] }
-      });
-      await sessionMemory?.addChatToSession(userId, sessionId, {
+      },
+      {
         role: "MODEL" /* MODEL */,
         timestamp: Date.now(),
         content: { type: "text", parts: [result.response] }
-      });
-    }
+      }
+    ]);
     return { content: result.response };
   }
 };
@@ -159,4 +202,4 @@ ${this.prompts?.system || ""}
 export {
   QueryService
 };
-//# sourceMappingURL=chunk-DRUCJYPJ.js.map
+//# sourceMappingURL=chunk-EGR6OTTN.js.map

package/dist/esm/chunk-EGR6OTTN.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\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 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 threadId - Thread identifier for context\n\t * @param thread - 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\tthreadId: string,\n\t\tthread?: ThreadObject,\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\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\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\tthreadId,\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\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}\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 * 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 thread 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 type - The type of thread (e.g., chat, workflow)\n\t * @param userId - The user's unique identifier\n\t * @param threadId - Unique thread identifier\n\t * @param query - The user's input query\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(type, 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 = this.intentTriggering(query);\n\n\t\t// 3. intent fulfillment\n\t\tconst result = await this.intentFulfilling(query, threadId, thread);\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;AA6BpB,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,UACA,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;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,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,EAgBA,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,MAAM,QAAQ,QAAQ;AAAA,IAC9D,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,UAAU,KAAK,iBAAiB,KAAK;AAG3C,UAAM,SAAS,MAAM,KAAK,iBAAiB,OAAO,UAAU,MAAM;AAClE,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":[]}

package/dist/esm/{chunk-5N4X25IK.js → chunk-G6VR5RSV.js}

@@ -1,12 +1,12 @@
-import {
-  A2AService
-} from "./chunk-Y6E73I7I.js";
 import {
   QueryService
-} from "./chunk-DRUCJYPJ.js";
+} from "./chunk-EGR6OTTN.js";
 import {
   A2AController
 } from "./chunk-HAE333YO.js";
+import {
+  A2AService
+} from "./chunk-2RW77MJB.js";
 
 // src/routes/a2a.routes.ts
 import { InMemoryTaskStore } from "@a2a-js/sdk/server";
@@ -34,4 +34,4 @@ var createA2ARouter = (agent) => {
 export {
   createA2ARouter
 };
-//# sourceMappingURL=chunk-5N4X25IK.js.map
+//# sourceMappingURL=chunk-G6VR5RSV.js.map

package/dist/esm/chunk-HOFAK5J2.js

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

package/dist/esm/chunk-HOFAK5J2.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/thread.routes.js\";\n\nexport const createApiRouter = (agent: AINAgent): Router => {\n\tconst router = Router();\n\n\trouter.use(\"/api/model\", createModelApiRouter(agent.modelModule));\n\tif (agent.memoryModule) {\n\t\trouter.use(\"/api/thread\", 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,cAAc,qBAAqB,MAAM,WAAW,CAAC;AAChE,MAAI,MAAM,cAAc;AACvB,WAAO,IAAI,eAAe,sBAAsB,MAAM,YAAY,CAAC;AAAA,EACpE;AAEA,SAAO;AACR;","names":[]}

package/dist/esm/{chunk-GNWF6QG4.js → chunk-JCWH7LIC.js}

@@ -1,12 +1,12 @@
 import {
   QueryStreamService
-} from "./chunk-3NEV3O2X.js";
-import {
-  QueryController
-} from "./chunk-Q6HHVAO6.js";
+} from "./chunk-2BYUKYVB.js";
 import {
   QueryService
-} from "./chunk-DRUCJYPJ.js";
+} from "./chunk-EGR6OTTN.js";
+import {
+  QueryController
+} from "./chunk-7D4SKPIS.js";
 
 // src/routes/query.routes.ts
 import { Router } from "express";
@@ -40,4 +40,4 @@ var createQueryRouter = (agent, allowStream = false) => {
 export {
   createQueryRouter
 };
-//# sourceMappingURL=chunk-GNWF6QG4.js.map
+//# sourceMappingURL=chunk-JCWH7LIC.js.map

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

@@ -13,6 +13,8 @@ var A2AModule = class {
   a2aPeerServers = /* @__PURE__ */ new Map();
   /** Map of session IDs to their A2A session state */
   a2aSessions = /* @__PURE__ */ new Map();
+  agentId = randomUUID();
+  /* FIXME */
   /**
    * Registers a new A2A peer server URL for connection.
    *
@@ -54,16 +56,16 @@ var A2AModule = class {
   /**
    * Gets or creates an A2A session for the given session ID.
    *
-   * @param sessionId - The session identifier
+   * @param threadId - The session identifier
    * @returns A2ASession object with task and context IDs
    */
-  getA2ASessionWithId = (sessionId) => {
-    const a2aSession = this.a2aSessions.get(sessionId) ?? {
+  getA2ASessionWithId = (threadId) => {
+    const a2aSession = this.a2aSessions.get(threadId) ?? {
       taskId: void 0,
       contextId: void 0
     };
-    if (!this.a2aSessions.has(sessionId)) {
-      this.a2aSessions.set(sessionId, a2aSession);
+    if (!this.a2aSessions.has(threadId)) {
+      this.a2aSessions.set(threadId, a2aSession);
     }
     return a2aSession;
   };
@@ -74,17 +76,19 @@ var A2AModule = class {
    * for maintaining conversation continuity.
    *
    * @param query - The message content to send
-   * @param sessionId - The session identifier
+   * @param threadId - The session identifier
    * @returns Formatted Message object for A2A protocol
    */
-  getMessagePayload(query, sessionId) {
+  getMessagePayload(query, threadId) {
     const messagePayload = {
       messageId: randomUUID(),
       kind: "message",
       role: "user",
       // FIXME: it could be 'agent'
       metadata: {
-        sessionId
+        agentId: this.agentId,
+        type: "CHAT" /* CHAT */,
+        threadId
       },
       parts: [
         {
@@ -93,7 +97,7 @@ var A2AModule = class {
         }
       ]
     };
-    const a2aSession = this.getA2ASessionWithId(sessionId);
+    const a2aSession = this.getA2ASessionWithId(threadId);
     if (a2aSession.taskId) {
       messagePayload.taskId = a2aSession.taskId;
     }
@@ -110,16 +114,16 @@ var A2AModule = class {
    *
    * @param tool - The A2ATool instance to use
    * @param messagePayload - The message to send to the agent
-   * @param sessionId - The session identifier for context tracking
+   * @param threadId - The session identifier for context tracking
    * @returns Promise resolving to array of text responses from the agent
    */
-  async useTool(tool, messagePayload, sessionId) {
+  async useTool(tool, messagePayload, threadId) {
     const finalText = [];
     const client = tool.client;
     const params = {
       message: messagePayload
     };
-    const a2aSession = this.getA2ASessionWithId(sessionId);
+    const a2aSession = this.getA2ASessionWithId(threadId);
     try {
       const stream = client.sendMessageStream(params);
       for await (const event of stream) {
@@ -169,4 +173,4 @@ ${finalText.join("\n")}`;
 export {
   A2AModule
 };
-//# sourceMappingURL=chunk-C6SGV7JV.js.map
+//# sourceMappingURL=chunk-LJVVSUF7.js.map

package/dist/esm/chunk-LJVVSUF7.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 * 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 */\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\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 * Gets or creates an A2A session for the given session ID.\n\t *\n\t * @param threadId - The session identifier\n\t * @returns A2ASession object with task and context IDs\n\t */\n\tprivate getA2ASessionWithId = (threadId: string): A2ASession => {\n\t\tconst a2aSession = this.a2aSessions.get(threadId) ?? {\n\t\t\ttaskId: undefined,\n\t\t\tcontextId: undefined,\n\t\t};\n\t\tif (!this.a2aSessions.has(threadId)) {\n\t\t\tthis.a2aSessions.set(threadId, 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 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\tthreadId,\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(threadId);\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 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\tmessagePayload: Message,\n\t\tthreadId: 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(threadId);\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;AAsBnB,IAAM,YAAN,MAAgB;AAAA;AAAA,EAEd,iBAA8C,oBAAI,IAAI;AAAA;AAAA,EAEtD,cAAuC,oBAAI,IAAI;AAAA,EAC/C,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,EAQQ,sBAAsB,CAAC,aAAiC;AAC/D,UAAM,aAAa,KAAK,YAAY,IAAI,QAAQ,KAAK;AAAA,MACpD,QAAQ;AAAA,MACR,WAAW;AAAA,IACZ;AACA,QAAI,CAAC,KAAK,YAAY,IAAI,QAAQ,GAAG;AACpC,WAAK,YAAY,IAAI,UAAU,UAAU;AAAA,IAC1C;AAEA,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,QACA;AAAA,MACD;AAAA,MACA,OAAO;AAAA,QACN;AAAA,UACC,MAAM;AAAA,UACN,MAAM;AAAA,QACP;AAAA,MACD;AAAA,IACD;AAEA,UAAM,aAAa,KAAK,oBAAoB,QAAQ;AACpD,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,UACkB;AAClB,UAAM,YAAsB,CAAC;AAC7B,UAAM,SAAS,KAAK;AACpB,UAAM,SAA4B;AAAA,MACjC,SAAS;AAAA,IACV;AACA,UAAM,aAAa,KAAK,oBAAoB,QAAQ;AAEpD,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-MNAWSBYP.js

@@ -0,0 +1,18 @@
+// src/types/memory.ts
+var MessageRole = /* @__PURE__ */ ((MessageRole2) => {
+  MessageRole2["USER"] = "USER";
+  MessageRole2["SYSTEM"] = "SYSTEM";
+  MessageRole2["MODEL"] = "MODEL";
+  return MessageRole2;
+})(MessageRole || {});
+var ThreadType = /* @__PURE__ */ ((ThreadType2) => {
+  ThreadType2["WORKFLOW"] = "WORKFLOW";
+  ThreadType2["CHAT"] = "CHAT";
+  return ThreadType2;
+})(ThreadType || {});
+
+export {
+  MessageRole,
+  ThreadType
+};
+//# sourceMappingURL=chunk-MNAWSBYP.js.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/types/memory.ts"],"sourcesContent":["/**\n * Roles for participants in a message.\n */\nexport enum MessageRole {\n\t/** User/human participant */\n\tUSER = \"USER\",\n\t/** System-generated messages or instructions */\n\tSYSTEM = \"SYSTEM\",\n\t/** AI model responses */\n\tMODEL = \"MODEL\",\n}\n\n/**\n * Content structure for message content.\n *\n * Supports multi-part content with different types (text, images, etc.).\n */\nexport type MessageContentObject = {\n\t/** Content type (e.g., \"text\", \"image\", \"tool_use\") */\n\ttype: string;\n\t/** Array of content parts, structure depends on content type */\n\tparts: any[];\n};\n\n/**\n * Represents a single message in a thread.\n *\n * @example\n * ```typescript\n * const message: MessageObject = {\n *   role: MessageRole.USER,\n *   content: {\n *     type: \"text\",\n *     parts: [\"Hello, how can you help me?\"]\n *   },\n *   timestamp: Date.now(),\n *   metadata: { source: \"web-ui\" }\n * };\n * ```\n */\nexport type MessageObject = {\n\t/** Role of the message sender */\n\trole: MessageRole;\n\t/** Message content with type and parts */\n\tcontent: MessageContentObject;\n\t/** Unix timestamp when the message was created */\n\ttimestamp: number;\n\t/** Optional metadata for additional context */\n\tmetadata?: { [key: string]: unknown };\n};\n\nexport enum ThreadType {\n\tWORKFLOW = \"WORKFLOW\",\n\tCHAT = \"CHAT\",\n}\n\nexport type ThreadMetadata = {\n\ttype: ThreadType;\n\ttitle: string;\n\tthreadId: string;\n\tupdatedAt: number;\n};\n\n/**\n * Represents a conversation thread containing multiple messages.\n *\n * Messages are stored in a key-value structure where keys are unique message IDs\n * and values are the corresponding message objects.\n *\n * @example\n * ```typescript\n * const thread: ThreadObject = {\n * \t title: \"New conversation\",\n *   messages: {\n *     \"<UUID_1>\": { role: MessageRole.USER, content: {...}, timestamp: 1234567890 },\n *     \"<UUID_2>\": { role: MessageRole.MODEL, content: {...}, timestamp: 1234567891 }\n *   }\n * };\n * ```\n */\nexport type ThreadObject = {\n\ttype: ThreadType;\n\ttitle: string;\n\t/* Collection of messages indexed by unique message ID */\n\tmessages: {\n\t\t[messageId: string]: MessageObject;\n\t};\n};\n\nexport interface Intent {\n\tname: string;\n\tdescription: string;\n\tprompt?: string;\n\tllm?: string;\n}\n"],"mappings":";AAGO,IAAK,cAAL,kBAAKA,iBAAL;AAEN,EAAAA,aAAA,UAAO;AAEP,EAAAA,aAAA,YAAS;AAET,EAAAA,aAAA,WAAQ;AANG,SAAAA;AAAA,GAAA;AAgDL,IAAK,aAAL,kBAAKC,gBAAL;AACN,EAAAA,YAAA,cAAW;AACX,EAAAA,YAAA,UAAO;AAFI,SAAAA;AAAA,GAAA;","names":["MessageRole","ThreadType"]}

package/dist/esm/chunk-VA3XOZBV.js

@@ -0,0 +1,17 @@
+import {
+  ThreadApiController
+} from "./chunk-ADIAIRNH.js";
+
+// src/routes/api/thread.routes.ts
+import { Router } from "express";
+var createThreadApiRouter = (memoryModule) => {
+  const router = Router();
+  const threadApiController = new ThreadApiController(memoryModule);
+  router.get("/:id", threadApiController.handleGetThread);
+  return router;
+};
+
+export {
+  createThreadApiRouter
+};
+//# sourceMappingURL=chunk-VA3XOZBV.js.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/routes/api/thread.routes.ts"],"sourcesContent":["import { Router } from \"express\";\nimport { ThreadApiController } from \"@/controllers/api/thread.api.controller.js\";\nimport type { MemoryModule } from \"@/modules/index.js\";\n\nexport const createThreadApiRouter = (memoryModule: MemoryModule): Router => {\n\tconst router = Router();\n\tconst threadApiController = new ThreadApiController(memoryModule);\n\n\t// APIs (prefix: /api/thread)\n\trouter.get(\"/:id\", threadApiController.handleGetThread);\n\n\treturn router;\n};\n"],"mappings":";;;;;AAAA,SAAS,cAAc;AAIhB,IAAM,wBAAwB,CAAC,iBAAuC;AAC5E,QAAM,SAAS,OAAO;AACtB,QAAM,sBAAsB,IAAI,oBAAoB,YAAY;AAGhE,SAAO,IAAI,QAAQ,oBAAoB,eAAe;AAEtD,SAAO;AACR;","names":[]}

package/dist/esm/{chunk-CKTURH34.js → chunk-XBMWWTSX.js}

@@ -1,11 +1,11 @@
 // src/modules/memory/memory.module.ts
 var MemoryModule = class {
   agentMemory;
-  sessionMemory;
+  threadMemory;
   intentMemory;
   constructor(config) {
     this.agentMemory = config.agent;
-    this.sessionMemory = config.session;
+    this.threadMemory = config.thread;
     this.intentMemory = config.intent;
   }
   async initialize() {
@@ -13,8 +13,8 @@ var MemoryModule = class {
     if (this.agentMemory) {
       connectPromises.push(this.agentMemory.connect());
     }
-    if (this.sessionMemory) {
-      connectPromises.push(this.sessionMemory.connect());
+    if (this.threadMemory) {
+      connectPromises.push(this.threadMemory.connect());
     }
     if (this.intentMemory) {
       connectPromises.push(this.intentMemory.connect());
@@ -26,8 +26,8 @@ var MemoryModule = class {
     if (this.agentMemory?.isConnected()) {
       disconnectPromises.push(this.agentMemory.disconnect());
     }
-    if (this.sessionMemory?.isConnected()) {
-      disconnectPromises.push(this.sessionMemory.disconnect());
+    if (this.threadMemory?.isConnected()) {
+      disconnectPromises.push(this.threadMemory.disconnect());
     }
     if (this.intentMemory?.isConnected()) {
       disconnectPromises.push(this.intentMemory.disconnect());
@@ -37,8 +37,8 @@ var MemoryModule = class {
   getAgentMemory() {
     return this.agentMemory;
   }
-  getSessionMemory() {
-    return this.sessionMemory;
+  getThreadMemory() {
+    return this.threadMemory;
   }
   getIntentMemory() {
     return this.intentMemory;
@@ -48,4 +48,4 @@ var MemoryModule = class {
 export {
   MemoryModule
 };
-//# sourceMappingURL=chunk-CKTURH34.js.map
+//# sourceMappingURL=chunk-XBMWWTSX.js.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/modules/memory/memory.module.ts"],"sourcesContent":["import type {\n\tIAgentMemory,\n\tIIntentMemory,\n\tIThreadMemory,\n} from \"./base.memory.js\";\n\nexport interface MemoryConfig {\n\tagent?: IAgentMemory;\n\tthread?: IThreadMemory;\n\tintent?: IIntentMemory;\n}\n\nexport class MemoryModule {\n\tprivate agentMemory?: IAgentMemory;\n\tprivate threadMemory?: IThreadMemory;\n\tprivate intentMemory?: IIntentMemory;\n\n\tconstructor(config: MemoryConfig) {\n\t\tthis.agentMemory = config.agent;\n\t\tthis.threadMemory = config.thread;\n\t\tthis.intentMemory = config.intent;\n\t}\n\n\tasync initialize(): Promise<void> {\n\t\tconst connectPromises: Promise<void>[] = [];\n\n\t\tif (this.agentMemory) {\n\t\t\tconnectPromises.push(this.agentMemory.connect());\n\t\t}\n\t\tif (this.threadMemory) {\n\t\t\tconnectPromises.push(this.threadMemory.connect());\n\t\t}\n\t\tif (this.intentMemory) {\n\t\t\tconnectPromises.push(this.intentMemory.connect());\n\t\t}\n\n\t\tawait Promise.all(connectPromises);\n\t}\n\n\tasync shutdown(): Promise<void> {\n\t\tconst disconnectPromises: Promise<void>[] = [];\n\n\t\tif (this.agentMemory?.isConnected()) {\n\t\t\tdisconnectPromises.push(this.agentMemory.disconnect());\n\t\t}\n\t\tif (this.threadMemory?.isConnected()) {\n\t\t\tdisconnectPromises.push(this.threadMemory.disconnect());\n\t\t}\n\t\tif (this.intentMemory?.isConnected()) {\n\t\t\tdisconnectPromises.push(this.intentMemory.disconnect());\n\t\t}\n\n\t\tawait Promise.all(disconnectPromises);\n\t}\n\n\tpublic getAgentMemory(): IAgentMemory | undefined {\n\t\treturn this.agentMemory;\n\t}\n\n\tpublic getThreadMemory(): IThreadMemory | undefined {\n\t\treturn this.threadMemory;\n\t}\n\n\tpublic getIntentMemory(): IIntentMemory | undefined {\n\t\treturn this.intentMemory;\n\t}\n}\n"],"mappings":";AAYO,IAAM,eAAN,MAAmB;AAAA,EACjB;AAAA,EACA;AAAA,EACA;AAAA,EAER,YAAY,QAAsB;AACjC,SAAK,cAAc,OAAO;AAC1B,SAAK,eAAe,OAAO;AAC3B,SAAK,eAAe,OAAO;AAAA,EAC5B;AAAA,EAEA,MAAM,aAA4B;AACjC,UAAM,kBAAmC,CAAC;AAE1C,QAAI,KAAK,aAAa;AACrB,sBAAgB,KAAK,KAAK,YAAY,QAAQ,CAAC;AAAA,IAChD;AACA,QAAI,KAAK,cAAc;AACtB,sBAAgB,KAAK,KAAK,aAAa,QAAQ,CAAC;AAAA,IACjD;AACA,QAAI,KAAK,cAAc;AACtB,sBAAgB,KAAK,KAAK,aAAa,QAAQ,CAAC;AAAA,IACjD;AAEA,UAAM,QAAQ,IAAI,eAAe;AAAA,EAClC;AAAA,EAEA,MAAM,WAA0B;AAC/B,UAAM,qBAAsC,CAAC;AAE7C,QAAI,KAAK,aAAa,YAAY,GAAG;AACpC,yBAAmB,KAAK,KAAK,YAAY,WAAW,CAAC;AAAA,IACtD;AACA,QAAI,KAAK,cAAc,YAAY,GAAG;AACrC,yBAAmB,KAAK,KAAK,aAAa,WAAW,CAAC;AAAA,IACvD;AACA,QAAI,KAAK,cAAc,YAAY,GAAG;AACrC,yBAAmB,KAAK,KAAK,aAAa,WAAW,CAAC;AAAA,IACvD;AAEA,UAAM,QAAQ,IAAI,kBAAkB;AAAA,EACrC;AAAA,EAEO,iBAA2C;AACjD,WAAO,KAAK;AAAA,EACb;AAAA,EAEO,kBAA6C;AACnD,WAAO,KAAK;AAAA,EACb;AAAA,EAEO,kBAA6C;AACnD,WAAO,KAAK;AAAA,EACb;AACD;","names":[]}

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

@@ -12,9 +12,9 @@ import '../modules/mcp/mcp.module.js';
 import '../types/mcp.js';
 import '@modelcontextprotocol/sdk/client/stdio.js';
 import '../modules/mcp/mcp.tool.js';
+import '../types/memory.js';
 import '../modules/memory/memory.module.js';
 import '../modules/memory/base.memory.js';
-import '../types/memory.js';
 import '../modules/models/model.module.js';
 import '../modules/models/base.model.js';
 import '../types/stream.js';

package/dist/esm/controllers/api/{session.api.controller.d.ts → thread.api.controller.d.ts}

@@ -3,10 +3,10 @@ import { MemoryModule } from '../../modules/memory/memory.module.js';
 import '../../modules/memory/base.memory.js';
 import '../../types/memory.js';
 
-declare class SessionApiController {
+declare class ThreadApiController {
     private memoryModule;
     constructor(memoryModule: MemoryModule);
-    handleSessionHistory: (req: Request, res: Response, next: NextFunction) => Promise<void>;
+    handleGetThread: (req: Request, res: Response, next: NextFunction) => Promise<void>;
 }
 
-export { SessionApiController };
+export { ThreadApiController };

package/dist/esm/controllers/api/thread.api.controller.js

@@ -0,0 +1,8 @@
+import {
+  ThreadApiController
+} from "../../chunk-ADIAIRNH.js";
+import "../../chunk-FL3JRQM4.js";
+export {
+  ThreadApiController
+};
+//# sourceMappingURL=thread.api.controller.js.map

package/dist/esm/controllers/index.d.ts

@@ -1,6 +1,6 @@
 export { A2AController } from './a2a.controller.js';
 export { ModelApiController } from './api/model.api.controller.js';
-export { SessionApiController } from './api/session.api.controller.js';
+export { ThreadApiController } from './api/thread.api.controller.js';
 export { QueryController } from './query.controller.js';
 import '@a2a-js/sdk';
 import '@a2a-js/sdk/server';
@@ -16,9 +16,9 @@ import '../modules/mcp/mcp.module.js';
 import '../types/mcp.js';
 import '@modelcontextprotocol/sdk/client/stdio.js';
 import '../modules/mcp/mcp.tool.js';
+import '../types/memory.js';
 import '../modules/memory/memory.module.js';
 import '../modules/memory/base.memory.js';
-import '../types/memory.js';
 import '../modules/models/model.module.js';
 import '../modules/models/base.model.js';
 import '../types/stream.js';

package/dist/esm/controllers/index.js

@@ -2,20 +2,20 @@ import {
   ModelApiController
 } from "../chunk-UA44VKFE.js";
 import {
-  SessionApiController
-} from "../chunk-5EPK37PE.js";
-import {
-  QueryController
-} from "../chunk-Q6HHVAO6.js";
-import "../chunk-FL3JRQM4.js";
+  ThreadApiController
+} from "../chunk-ADIAIRNH.js";
 import {
   A2AController
 } from "../chunk-HAE333YO.js";
+import {
+  QueryController
+} from "../chunk-7D4SKPIS.js";
+import "../chunk-FL3JRQM4.js";
 import "../chunk-N7VYM2R5.js";
 export {
   A2AController,
   ModelApiController,
   QueryController,
-  SessionApiController
+  ThreadApiController
 };
 //# sourceMappingURL=index.js.map

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

@@ -11,9 +11,9 @@ import '../modules/mcp/mcp.module.js';
 import '../types/mcp.js';
 import '@modelcontextprotocol/sdk/client/stdio.js';
 import '../modules/mcp/mcp.tool.js';
+import '../types/memory.js';
 import '../modules/memory/memory.module.js';
 import '../modules/memory/base.memory.js';
-import '../types/memory.js';
 import '../modules/models/model.module.js';
 import '../modules/models/base.model.js';
 import '../types/stream.js';

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

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

package/dist/esm/index.d.ts

@@ -103,6 +103,7 @@ declare class AINAgent {
      * - GET / - Health check endpoint
      * - GET /.well-known/agent.json - Agent card discovery endpoint
      * - /query/* - Query processing endpoints
+     * - /api/* - API endpoints for agent management
      * - /a2a/* - A2A protocol endpoints (only if valid URL is configured)
      */
     private initializeRoutes;