@ainetwork/adk 0.2.0 → 0.2.2

package/README.md

@@ -41,11 +41,11 @@ To see how to use this package in your project, check out our comprehensive exam
 
 ### Core Components
 
-- **AINAgent** (`src/app.ts`): Main Express.js server class that orchestrates all modules
-- **ModelModule** (`src/modules/models/`): Manages AI model integrations (OpenAI, Gemini)
+- **AINAgent** (`src/index.ts`): Main Express.js server class that orchestrates all modules
+- **ModelModule** (`src/modules/models/`): Manages AI model integrations with streaming support
 - **MCPModule** (`src/modules/mcp/`): Handles Model Context Protocol connections
 - **A2AModule** (`src/modules/a2a/`): Manages agent-to-agent communication
-- **MemoryModule** (`src/modules/memory/`): Provides threads and conversation history
+- **MemoryModule** (`src/modules/memory/`): Provides threads, intents, and conversation history
 
 ### Module System
 
@@ -68,20 +68,25 @@ Each module can be independently configured and passed to the agent constructor.
 - Connects to external MCP servers via stdio
 - Automatic tool discovery and execution
 - Supports multiple concurrent MCP servers
+- Protocol-specific tool wrapping as `IMCPTool`
 
 #### A2A (Agent-to-Agent)  
 - RESTful API for inter-agent communication
-- Streaming response support
-- Agent discovery via well-known endpoints
-- Task delegation and context passing
+- Streaming response support via SSE
+- Agent discovery via well-known endpoints (`.well-known/agent-card.json`)
+- Task delegation with thread context passing
+- Protocol version 0.3.0 support
 
 ### Key Features
 
 - **Unified Tool Interface**: Protocol-agnostic `IAgentTool` interface
+- **Streaming Support**: Dual implementation for streaming and non-streaming queries
+- **Intent System**: Intent detection and fulfillment with custom prompts
 - **Service Layer**: Clean separation with controllers and services
-- **Type Safety**: Comprehensive TypeScript types throughout
+- **Type Safety**: Comprehensive TypeScript types with strict mode
 - **Error Handling**: Global error middleware with structured logging
-- **Authentication**: Optional auth middleware support
+- **Authentication**: Optional auth middleware via `BaseAuth` interface
+- **Graceful Shutdown**: Proper cleanup of modules and connections
 
 ## Development
 
@@ -96,10 +101,13 @@ yarn dev            # Run TypeScript directly with tsx
 
 # Code quality
 yarn biome          # Check code with Biome
-yarn check:write    # Check and auto-fix with Biome
+yarn biome:write    # Check and auto-fix with Biome
 
 # Testing
 yarn test           # Run Jest tests
+
+# Clean
+yarn clean          # Remove build artifacts
 ```
 
 ## Logging System
@@ -124,12 +132,13 @@ modelLogger.error('Model API error');
 
 ### Available Loggers
 - `agent`: Main agent operations
-- `intent`: Query processing and intent analysis
+- `intent`: Non-streaming query processing and intent analysis
+- `intentStream`: Streaming query processing
 - `mcp`: MCP server connections and tool execution
 - `a2a`: Agent-to-agent communication
 - `model`: AI model interactions
 - `server`: HTTP server operations
-- `memory`: Threads, Intents, Agent Data management
+- `memory`: Threads, intents, and agent data management
 
 ### Log Levels
 - `error`: Error conditions
@@ -140,22 +149,35 @@ modelLogger.error('Model API error');
 ## API Endpoints
 
 ### Standard Endpoints
-- `GET /` - Welcome message
-- `POST /query` - Process queries
-  - Request: `{ message: string, threadId: string }`
-  - Response: `{ content: string }`
-- `POST /query/stream` - Process queries with streaming
-  - Request: `{ message: string, threadId: string }`
+- `GET /` - Welcome message and health check
+- `POST /query` - Process queries (non-streaming)
+  - Request: `{ message: string, threadId?: string, type?: string }`
   - Response: `{ content: string }`
+- `POST /query/stream` - Process queries with streaming (SSE)
+  - Request: `{ message: string, threadId?: string, type?: string }`
+  - Response: Server-Sent Events stream with event types:
+    - `text_chunk`: Incremental text response
+    - `tool_start`: Tool execution started
+    - `tool_output`: Tool execution result
+    - `thread_id`: Thread metadata
+    - `error`: Error message
 
 ### Agent Management
+- `GET /api/threads/:userId` - List user threads
+- `POST /api/threads/:userId` - Create new thread
+- `GET /api/threads/:userId/:threadId` - Get thread details
+- `DELETE /api/threads/:userId/:threadId` - Delete thread
+- `GET /api/model` - Get model configuration
+- `POST /api/model` - Set default model
 
 ### A2A Server Endpoints (when `manifest.url` is configured)
-- `GET /agent-card` - Get agent card information
-- `GET /.well-known/agent.json` - Agent discovery endpoint  
+- `GET /.well-known/agent.json` - Agent discovery endpoint (A2A ~v0.2.0)
+- `GET /.well-known/agent-card.json` - Agent discovery endpoint (A2A v0.3.0~)
+  - Returns `AgentCard` with capabilities and supported modes
 - `POST /a2a` - A2A communication endpoint
-  - Supports streaming responses
-  - Request: `{ message: string, stream?: boolean }`
+  - Supports streaming responses via SSE
+  - Request: `{ message: string, threadId?: string, stream?: boolean }`
+  - Response: JSON or SSE stream based on `stream` parameter
 
 ## Build System
 
@@ -168,13 +190,15 @@ The project supports dual build output:
 
 The library includes comprehensive error handling:
 - Global error middleware for uncaught errors
-- Service-specific error logging
+- Custom `AinHttpError` class for HTTP-specific errors
+- Service-specific error logging with structured context
 - Graceful handling of:
   - MCP server connection failures
   - A2A agent communication errors  
-  - Model API failures
+  - Model API failures (rate limits, timeouts)
   - Tool execution errors
   - Invalid requests
+  - Streaming errors with proper cleanup
 
 All errors are logged with appropriate context for debugging.
 
@@ -197,11 +221,13 @@ const agent = new AINAgent(manifest, modules, new MyAuth());
 
 ## Contributing
 
-1. Follow the established code conventions
+1. Follow the established code conventions (tabs, double quotes)
 2. Use TypeScript strict mode
 3. Add appropriate service-specific logging
 4. Run `yarn biome:write` and `yarn test` before submitting
 5. Maintain the modular architecture
+6. Update JSDoc comments when changing function signatures
+7. Add streaming support when implementing new query handlers
 
 ## License
 

package/dist/cjs/chunk-3ZFV2K7O.cjs

@@ -0,0 +1,128 @@
+"use strict";Object.defineProperty(exports, "__esModule", {value: true}); var _class;
+
+var _chunkE433ONENcjs = require('./chunk-E433ONEN.cjs');
+
+
+var _chunk466ED5WNcjs = require('./chunk-466ED5WN.cjs');
+
+// src/modules/mcp/mcp.module.ts
+var _indexjs = require('@modelcontextprotocol/sdk/client/index.js');
+var _ssejs = require('@modelcontextprotocol/sdk/client/sse.js');
+var _stdiojs = require('@modelcontextprotocol/sdk/client/stdio.js');
+var _streamableHttpjs = require('@modelcontextprotocol/sdk/client/streamableHttp.js');
+var _websocketjs = require('@modelcontextprotocol/sdk/client/websocket.js');
+var MCPModule = (_class = class {constructor() { _class.prototype.__init.call(this);_class.prototype.__init2.call(this);_class.prototype.__init3.call(this); }
+  __init() {this.mcpConfigs = /* @__PURE__ */ new Map()}
+  /** Map of MCP server names to their client instances */
+  __init2() {this.mcpClientMap = /* @__PURE__ */ new Map()}
+  /** Array of all discovered tools from connected MCP servers */
+  __init3() {this.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 (0, _indexjs.Client)({ name, version: "1.0.0" });
+        switch (config.type) {
+          case "stdio": {
+            const transport = new (0, _stdiojs.StdioClientTransport)(config.params);
+            await mcpClient.connect(transport);
+            break;
+          }
+          case "websocket": {
+            const transport = new (0, _websocketjs.WebSocketClientTransport)(config.url);
+            await mcpClient.connect(transport);
+            break;
+          }
+          case "sse": {
+            const transport = new (0, _ssejs.SSEClientTransport)(
+              config.url,
+              config.options
+            );
+            await mcpClient.connect(transport);
+            break;
+          }
+          case "streamableHttp": {
+            const transport = new (0, _streamableHttpjs.StreamableHTTPClientTransport)(
+              config.url,
+              config.options
+            );
+            await mcpClient.connect(transport);
+            break;
+          }
+          default:
+            _chunk466ED5WNcjs.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 (0, _chunkE433ONENcjs.MCPTool)(name, tool);
+        });
+        this.tools.push(...newToolList);
+        _chunk466ED5WNcjs.loggers.mcp.info("Connected to MCP server with tools:", {
+          tools: newToolList.map((tool) => tool.id)
+        });
+      } catch (error) {
+        _chunk466ED5WNcjs.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) {
+      _chunk466ED5WNcjs.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();
+    });
+  }
+}, _class);
+
+
+
+exports.MCPModule = MCPModule;
+//# sourceMappingURL=chunk-3ZFV2K7O.cjs.map

package/dist/cjs/chunk-3ZFV2K7O.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-3ZFV2K7O.cjs","../../src/modules/mcp/mcp.module.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACA;ACPA,oEAAuB;AACvB,gEAAmC;AACnC,oEAAqC;AACrC,sFAA8C;AAC9C,4EAAyC;AA0BlC,IAAM,UAAA,YAAN,MAAgB;AAAA,iBACd,WAAA,kBAAqC,IAAI,GAAA,CAAI,EAAA;AAAA;AAAA,kBAE7C,aAAA,kBAAoC,IAAI,GAAA,CAAI,EAAA;AAAA;AAAA,kBAE5C,MAAA,EAAmB,CAAC,EAAA;AAAA,EAE5B,kBAAA,CAAmB,OAAA,EAA8C;AAChE,IAAA,IAAA,CAAA,MAAW,CAAC,IAAA,EAAM,MAAM,EAAA,GAAK,MAAA,CAAO,OAAA,CAAQ,OAAO,CAAA,EAAG;AACrD,MAAA,IAAA,CAAK,UAAA,CAAW,GAAA,CAAI,IAAA,EAAM,MAAM,CAAA;AAAA,IACjC;AAAA,EACD;AAAA,EAEA,MAAM,gBAAA,CAAA,EAAkC;AACvC,IAAA,IAAA,CAAA,MAAW,CAAC,IAAA,EAAM,MAAM,EAAA,GAAK,IAAA,CAAK,UAAA,CAAW,OAAA,CAAQ,CAAA,EAAG;AACvD,MAAA,IAAI;AACH,QAAA,MAAM,UAAA,EAAY,IAAI,oBAAA,CAAO,EAAE,IAAA,EAAM,OAAA,EAAS,QAAQ,CAAC,CAAA;AACvD,QAAA,OAAA,CAAQ,MAAA,CAAO,IAAA,EAAM;AAAA,UACpB,KAAK,OAAA,EAAS;AACb,YAAA,MAAM,UAAA,EAAY,IAAI,kCAAA,CAAqB,MAAA,CAAO,MAAM,CAAA;AACxD,YAAA,MAAM,SAAA,CAAU,OAAA,CAAQ,SAAS,CAAA;AACjC,YAAA,KAAA;AAAA,UACD;AAAA,UACA,KAAK,WAAA,EAAa;AACjB,YAAA,MAAM,UAAA,EAAY,IAAI,0CAAA,CAAyB,MAAA,CAAO,GAAG,CAAA;AACzD,YAAA,MAAM,SAAA,CAAU,OAAA,CAAQ,SAAS,CAAA;AACjC,YAAA,KAAA;AAAA,UACD;AAAA,UACA,KAAK,KAAA,EAAO;AACX,YAAA,MAAM,UAAA,EAAY,IAAI,8BAAA;AAAA,cACrB,MAAA,CAAO,GAAA;AAAA,cACP,MAAA,CAAO;AAAA,YACR,CAAA;AACA,YAAA,MAAM,SAAA,CAAU,OAAA,CAAQ,SAAS,CAAA;AACjC,YAAA,KAAA;AAAA,UACD;AAAA,UACA,KAAK,gBAAA,EAAkB;AACtB,YAAA,MAAM,UAAA,EAAY,IAAI,oDAAA;AAAA,cACrB,MAAA,CAAO,GAAA;AAAA,cACP,MAAA,CAAO;AAAA,YACR,CAAA;AACA,YAAA,MAAM,SAAA,CAAU,OAAA,CAAQ,SAAS,CAAA;AACjC,YAAA,KAAA;AAAA,UACD;AAAA,UACA,OAAA;AAEC,YAAA,yBAAA,CAAQ,GAAA,CAAI,KAAA,CAAM,6BAA6B,CAAA;AAC/C,YAAA,KAAA;AAAA,QACF;AAEA,QAAA,IAAA,CAAK,YAAA,CAAa,GAAA,CAAI,IAAA,EAAM,SAAS,CAAA;AACrC,QAAA,MAAM,SAAA,EAAW,MAAM,SAAA,CAAU,SAAA,CAAU,CAAA;AAC3C,QAAA,MAAM,YAAA,EAAc,QAAA,CAAS,KAAA,CAAM,GAAA,CAAI,CAAC,IAAA,EAAA,GAAS;AAChD,UAAA,OAAO,IAAI,8BAAA,CAAQ,IAAA,EAAM,IAAI,CAAA;AAAA,QAC9B,CAAC,CAAA;AACD,QAAA,IAAA,CAAK,KAAA,CAAM,IAAA,CAAK,GAAG,WAAW,CAAA;AAC9B,QAAA,yBAAA,CAAQ,GAAA,CAAI,IAAA,CAAK,qCAAA,EAAuC;AAAA,UACvD,KAAA,EAAO,WAAA,CAAY,GAAA,CAAI,CAAC,IAAA,EAAA,GAAS,IAAA,CAAK,EAAE;AAAA,QACzC,CAAC,CAAA;AAAA,MACF,EAAA,MAAA,CAAS,KAAA,EAAO;AACf,QAAA,yBAAA,CAAQ,GAAA,CAAI,KAAA,CAAM,CAAA,gCAAA,EAAmC,IAAI,CAAA,CAAA;AAC1D,MAAA;AACD,IAAA;AACD,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAOW,EAAA;AACE,IAAA;AACb,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAU2D,EAAA;AAC1B,IAAA;AACP,IAAA;AACmB,IAAA;AAExC,IAAA;AACO,MAAA;AACiD,QAAA;AAC3D,MAAA;AAEkC,MAAA;AAC3B,QAAA;AACK,QAAA;AACX,MAAA;AAE4B,MAAA;AACU;AAChC,MAAA;AACQ,IAAA;AACmC,MAAA;AACH,MAAA;AAAyG;AACjJ,MAAA;AACR,IAAA;AACD,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAQgB,EAAA;AACkC,IAAA;AAChC,MAAA;AAChB,IAAA;AACF,EAAA;AACD;ADzBkE;AACA;AACA;AACA","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-3ZFV2K7O.cjs","sourcesContent":[null,"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"]}

package/dist/cjs/chunk-7PZP7YSA.cjs

@@ -0,0 +1,40 @@
+"use strict";Object.defineProperty(exports, "__esModule", {value: true});
+
+var _chunkA2I7D6JWcjs = require('./chunk-A2I7D6JW.cjs');
+
+
+var _chunkQHQLU7BIcjs = require('./chunk-QHQLU7BI.cjs');
+
+// src/routes/api/threads.routes.ts
+
+
+var _express = require('express');
+var _httpstatuscodes = require('http-status-codes');
+var createThreadApiRouter = (memoryModule) => {
+  const router = _express.Router.call(void 0, );
+  const threadApiController = new (0, _chunkA2I7D6JWcjs.ThreadApiController)(memoryModule);
+  const checkThreadMemory = (req, res, next) => {
+    const threadMemory = memoryModule.getThreadMemory();
+    if (!threadMemory) {
+      const error = new (0, _chunkQHQLU7BIcjs.AinHttpError)(
+        _httpstatuscodes.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;
+};
+
+
+
+exports.createThreadApiRouter = createThreadApiRouter;
+//# sourceMappingURL=chunk-7PZP7YSA.cjs.map

package/dist/cjs/chunk-7PZP7YSA.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-7PZP7YSA.cjs","../../src/routes/api/threads.routes.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACA;ACPA;AAIC;AAAA,kCACM;AACP,oDAA4B;AAKrB,IAAM,sBAAA,EAAwB,CAAC,YAAA,EAAA,GAAuC;AAC5E,EAAA,MAAM,OAAA,EAAS,6BAAA,CAAO;AACtB,EAAA,MAAM,oBAAA,EAAsB,IAAI,0CAAA,CAAoB,YAAY,CAAA;AAEhE,EAAA,MAAM,kBAAA,EAAoB,CACzB,GAAA,EACA,GAAA,EACA,IAAA,EAAA,GACI;AACJ,IAAA,MAAM,aAAA,EAAe,YAAA,CAAa,eAAA,CAAgB,CAAA;AAClD,IAAA,GAAA,CAAI,CAAC,YAAA,EAAc;AAClB,MAAA,MAAM,MAAA,EAAQ,IAAI,mCAAA;AAAA,QACjB,4BAAA,CAAY,mBAAA;AAAA,QACZ;AAAA,MACD,CAAA;AACA,MAAA,MAAM,KAAA;AAAA,IACP;AACA,IAAA,IAAA,CAAK,CAAA;AAAA,EACN,CAAA;AAGA,EAAA,MAAA,CAAO,GAAA,CAAI,GAAA,EAAK,iBAAA,EAAmB,mBAAA,CAAoB,oBAAoB,CAAA;AAC3E,EAAA,MAAA,CAAO,GAAA,CAAI,MAAA,EAAQ,iBAAA,EAAmB,mBAAA,CAAoB,eAAe,CAAA;AACzE,EAAA,MAAA,CAAO,MAAA;AAAA,IACN,MAAA;AAAA,IACA,iBAAA;AAAA,IACA,mBAAA,CAAoB;AAAA,EACrB,CAAA;AAEA,EAAA,OAAO,MAAA;AACR,CAAA;ADNA;AACA;AACE;AACF,sDAAC","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-7PZP7YSA.cjs","sourcesContent":[null,"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"]}

package/dist/cjs/chunk-A2I7D6JW.cjs

@@ -0,0 +1,45 @@
+"use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _optionalChain(ops) { let lastAccessLHS = undefined; let value = ops[0]; let i = 1; while (i < ops.length) { const op = ops[i]; const fn = ops[i + 1]; i += 2; if ((op === 'optionalAccess' || op === 'optionalCall') && value == null) { return undefined; } if (op === 'access' || op === 'optionalAccess') { lastAccessLHS = value; value = fn(value); } else if (op === 'call' || op === 'optionalCall') { value = fn((...args) => value.call(lastAccessLHS, ...args)); lastAccessLHS = undefined; } } return value; } var _class;// src/controllers/api/threads.api.controller.ts
+var _httpstatuscodes = require('http-status-codes');
+var ThreadApiController = (_class = class {
+  
+  constructor(memoryModule) {;_class.prototype.__init.call(this);_class.prototype.__init2.call(this);_class.prototype.__init3.call(this);
+    this.memoryModule = memoryModule;
+  }
+  __init() {this.handleGetThread = async (req, res, next) => {
+    try {
+      const { id: threadId } = req.params;
+      const userId = res.locals.userId || "";
+      const threadMemory = this.memoryModule.getThreadMemory();
+      const thread = await _optionalChain([threadMemory, 'optionalAccess', _7 => _7.getThread, 'call', _8 => _8(userId, threadId)]);
+      res.json(thread);
+    } catch (error) {
+      next(error);
+    }
+  }}
+  __init2() {this.handleDeleteThread = async (req, res, next) => {
+    try {
+      const { id: threadId } = req.params;
+      const userId = res.locals.userId || "";
+      const threadMemory = this.memoryModule.getThreadMemory();
+      await _optionalChain([threadMemory, 'optionalAccess', _9 => _9.deleteThread, 'call', _10 => _10(userId, threadId)]);
+      res.status(_httpstatuscodes.StatusCodes.OK).send();
+    } catch (error) {
+      next(error);
+    }
+  }}
+  __init3() {this.handleGetUserThreads = async (_req, res, next) => {
+    try {
+      const userId = res.locals.userId || "";
+      const threadMemory = this.memoryModule.getThreadMemory();
+      const threads = await _optionalChain([threadMemory, 'optionalAccess', _11 => _11.listThreads, 'call', _12 => _12(userId)]);
+      res.json(threads);
+    } catch (error) {
+      next(error);
+    }
+  }}
+}, _class);
+
+
+
+exports.ThreadApiController = ThreadApiController;
+//# sourceMappingURL=chunk-A2I7D6JW.cjs.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-A2I7D6JW.cjs","../../src/controllers/api/threads.api.controller.ts"],"names":[],"mappings":"AAAA;ACCA,oDAA4B;AAGrB,IAAM,oBAAA,YAAN,MAA0B;AAAA,EACxB;AAAA,EAER,WAAA,CAAY,YAAA,EAA4B;AACvC,IAAA,IAAA,CAAK,aAAA,EAAe,YAAA;AAAA,EACrB;AAAA,iBAEO,gBAAA,EAAkB,MAAA,CACxB,GAAA,EACA,GAAA,EACA,IAAA,EAAA,GACI;AACJ,IAAA,IAAI;AACH,MAAA,MAAM,EAAE,EAAA,EAAI,SAAS,EAAA,EAAI,GAAA,CAAI,MAAA;AAG7B,MAAA,MAAM,OAAA,EAAS,GAAA,CAAI,MAAA,CAAO,OAAA,GAAU,EAAA;AACpC,MAAA,MAAM,aAAA,EAAe,IAAA,CAAK,YAAA,CAAa,eAAA,CAAgB,CAAA;AACvD,MAAA,MAAM,OAAA,EAAS,sBAAM,YAAA,6BAAc,SAAA,mBAAU,MAAA,EAAQ,QAAQ,GAAA;AAC7D,MAAA,GAAA,CAAI,IAAA,CAAK,MAAM,CAAA;AAAA,IAChB,EAAA,MAAA,CAAS,KAAA,EAAO;AACf,MAAA,IAAA,CAAK,KAAK,CAAA;AAAA,IACX;AAAA,EACD,EAAA;AAAA,kBAEO,mBAAA,EAAqB,MAAA,CAC3B,GAAA,EACA,GAAA,EACA,IAAA,EAAA,GACI;AACJ,IAAA,IAAI;AACH,MAAA,MAAM,EAAE,EAAA,EAAI,SAAS,EAAA,EAAI,GAAA,CAAI,MAAA;AAC7B,MAAA,MAAM,OAAA,EAAS,GAAA,CAAI,MAAA,CAAO,OAAA,GAAU,EAAA;AACpC,MAAA,MAAM,aAAA,EAAe,IAAA,CAAK,YAAA,CAAa,eAAA,CAAgB,CAAA;AACvD,MAAA,sBAAM,YAAA,6BAAc,YAAA,qBAAa,MAAA,EAAQ,QAAQ,GAAA;AACjD,MAAA,GAAA,CAAI,MAAA,CAAO,4BAAA,CAAY,EAAE,CAAA,CAAE,IAAA,CAAK,CAAA;AAAA,IACjC,EAAA,MAAA,CAAS,KAAA,EAAO;AACf,MAAA,IAAA,CAAK,KAAK,CAAA;AAAA,IACX;AAAA,EACD,EAAA;AAAA,kBAEO,qBAAA,EAAuB,MAAA,CAC7B,IAAA,EACA,GAAA,EACA,IAAA,EAAA,GACI;AACJ,IAAA,IAAI;AACH,MAAA,MAAM,OAAA,EAAS,GAAA,CAAI,MAAA,CAAO,OAAA,GAAU,EAAA;AACpC,MAAA,MAAM,aAAA,EAAe,IAAA,CAAK,YAAA,CAAa,eAAA,CAAgB,CAAA;AACvD,MAAA,MAAM,QAAA,EAAU,sBAAM,YAAA,+BAAc,WAAA,qBAAY,MAAM,GAAA;AACtD,MAAA,GAAA,CAAI,IAAA,CAAK,OAAO,CAAA;AAAA,IACjB,EAAA,MAAA,CAAS,KAAA,EAAO;AACf,MAAA,IAAA,CAAK,KAAK,CAAA;AAAA,IACX;AAAA,EACD,EAAA;AACD,UAAA;ADnBA;AACA;AACE;AACF,kDAAC","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-A2I7D6JW.cjs","sourcesContent":[null,"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"]}

package/dist/cjs/{chunk-6YVTJEMQ.cjs → chunk-BBFYVWE2.cjs}

@@ -1,4 +1,4 @@
-"use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _nullishCoalesce(lhs, rhsFn) { if (lhs != null) { return lhs; } else { return rhsFn(); } } function _optionalChain(ops) { let lastAccessLHS = undefined; let value = ops[0]; let i = 1; while (i < ops.length) { const op = ops[i]; const fn = ops[i + 1]; i += 2; if ((op === 'optionalAccess' || op === 'optionalCall') && value == null) { return undefined; } if (op === 'access' || op === 'optionalAccess') { lastAccessLHS = value; value = fn(value); } else if (op === 'call' || op === 'optionalCall') { value = fn((...args) => value.call(lastAccessLHS, ...args)); lastAccessLHS = undefined; } } return value; } var _class;
+"use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _optionalChain(ops) { let lastAccessLHS = undefined; let value = ops[0]; let i = 1; while (i < ops.length) { const op = ops[i]; const fn = ops[i + 1]; i += 2; if ((op === 'optionalAccess' || op === 'optionalCall') && value == null) { return undefined; } if (op === 'access' || op === 'optionalAccess') { lastAccessLHS = value; value = fn(value); } else if (op === 'call' || op === 'optionalCall') { value = fn((...args) => value.call(lastAccessLHS, ...args)); lastAccessLHS = undefined; } } return value; } var _class;
 
 var _chunkF7I4G6JGcjs = require('./chunk-F7I4G6JG.cjs');
 
@@ -8,11 +8,11 @@ var _chunk466ED5WNcjs = require('./chunk-466ED5WN.cjs');
 // src/modules/a2a/a2a.module.ts
 var _crypto = require('crypto');
 var _client = require('@a2a-js/sdk/client');
-var A2AModule = (_class = class {constructor() { _class.prototype.__init.call(this);_class.prototype.__init2.call(this);_class.prototype.__init3.call(this);_class.prototype.__init4.call(this); }
+var A2AModule = (_class = class {constructor() { _class.prototype.__init.call(this);_class.prototype.__init2.call(this);_class.prototype.__init3.call(this); }
   /** Map of A2A server URLs to their corresponding tool instances */
   __init() {this.a2aPeerServers = /* @__PURE__ */ new Map()}
   /** Map of session IDs to their A2A session state */
-  __init2() {this.a2aSessions = /* @__PURE__ */ new Map()}
+  __init2() {this.a2aTasks = /* @__PURE__ */ new Map()}
   __init3() {this.agentId = _crypto.randomUUID.call(void 0, )}
   /* FIXME */
   /**
@@ -53,22 +53,6 @@ var A2AModule = (_class = class {constructor() { _class.prototype.__init.call(th
     }
     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
-   */
-  __init4() {this.getA2ASessionWithId = (threadId) => {
-    const a2aSession = _nullishCoalesce(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 = class {constructor() { _class.prototype.__init.call(th
       // 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 = class {constructor() { _class.prototype.__init.call(th
    * @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 = _optionalChain([typedEvent, 'access', _ => _.status, 'access', _2 => _2.message, 'optionalAccess', _3 => _3.parts, 'access', _4 => _4.filter, 'call', _5 => _5((part) => part.kind === "text"), 'access', _6 => _6.map, 'call', _7 => _7((part) => part.text), 'access', _8 => _8.join, 'call', _9 => _9("\n")]);
           if (texts) {
@@ -138,19 +118,14 @@ var A2AModule = (_class = class {constructor() { _class.prototype.__init.call(th
           }
         } 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 {
           _chunk466ED5WNcjs.loggers.a2a.warn("Received unknown event structure from stream:", {
@@ -173,4 +148,4 @@ ${finalText.join("\n")}`;
 
 
 exports.A2AModule = A2AModule;
-//# sourceMappingURL=chunk-6YVTJEMQ.cjs.map
+//# sourceMappingURL=chunk-BBFYVWE2.cjs.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-BBFYVWE2.cjs","../../src/modules/a2a/a2a.module.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACA;ACPA,gCAA2B;AAS3B,4CAA0B;AAYnB,IAAM,UAAA,YAAN,MAAgB;AAAA;AAAA,iBAEd,eAAA,kBAA8C,IAAI,GAAA,CAAI,EAAA;AAAA;AAAA,kBAEtD,SAAA,kBAAgC,IAAI,GAAA,CAAI,EAAA;AAAA,kBACxC,QAAA,EAAkB,gCAAA,EAAW;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOrC,MAAa,gBAAA,CAAiB,GAAA,EAA4B;AACzD,IAAA,IAAA,CAAK,cAAA,CAAe,GAAA,CAAI,GAAA,EAAK,IAAI,CAAA;AAAA,EAClC;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAUA,MAAa,QAAA,CAAA,EAA+B;AAC3C,IAAA,MAAM,MAAA,EAAmB,CAAC,CAAA;AAC1B,IAAA,IAAA,CAAA,MAAW,IAAA,GAAO,CAAC,GAAG,IAAA,CAAK,cAAA,CAAe,IAAA,CAAK,CAAC,CAAA,EAAG;AAClD,MAAA,MAAM,KAAA,EAAO,IAAA,CAAK,cAAA,CAAe,GAAA,CAAI,GAAG,CAAA;AACxC,MAAA,GAAA,CAAI,CAAC,KAAA,GAAQ,CAAC,IAAA,CAAK,OAAA,EAAS;AAC3B,QAAA,IAAI;AACH,UAAA,MAAM,OAAA,EAAS,IAAI,sBAAA,CAAU,GAAG,CAAA;AAChC,UAAA,MAAM,KAAA,EAAkB,MAAM,MAAA,CAAO,YAAA,CAAa,CAAA;AAClD,UAAA,MAAM,SAAA,EAAW,IAAA,CAAK,IAAA,CAAK,UAAA,CAAW,GAAA,EAAK,GAAG,CAAA;AAC9C,UAAA,MAAM,QAAA,EAAU,IAAI,8BAAA,CAAQ,QAAA,EAAU,MAAA,EAAQ,IAAI,CAAA;AAElD,UAAA,KAAA,CAAM,IAAA,CAAK,OAAO,CAAA;AAAA,QACnB,EAAA,MAAA,CAAS,MAAA,EAAa;AAErB,UAAA,GAAA,CAAI,IAAA,EAAM;AACT,YAAA,IAAA,CAAK,OAAA,CAAQ,CAAA;AAAA,UACd;AAAA,QACD;AAAA,MACD,EAAA,KAAO;AACN,QAAA,KAAA,CAAM,IAAA,CAAK,IAAI,CAAA;AAAA,MAChB;AAAA,IACD;AACA,IAAA,OAAO,KAAA;AAAA,EACR;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAYO,iBAAA,CAAkB,KAAA,EAAe,QAAA,EAA2B;AAClE,IAAA,MAAM,eAAA,EAA0B;AAAA,MAC/B,SAAA,EAAW,gCAAA,CAAW;AAAA,MACtB,IAAA,EAAM,SAAA;AAAA,MACN,IAAA,EAAM,MAAA;AAAA;AAAA,MACN,QAAA,EAAU;AAAA,QACT,OAAA,EAAS,IAAA,CAAK,OAAA;AAAA,QACd,IAAA,EAAA;AAAA,MACD,CAAA;AAAA,MACA,KAAA,EAAO;AAAA,QACN;AAAA,UACC,IAAA,EAAM,MAAA;AAAA,UACN,IAAA,EAAM;AAAA,QACP;AAAA,MACD,CAAA;AAAA,MACA,SAAA,EAAW;AAAA,IACZ,CAAA;AAEA,IAAA,GAAA,CAAI,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,QAAQ,CAAA,EAAG;AAChC,MAAA,cAAA,CAAe,OAAA,EAAS,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,QAAQ,CAAA;AAAA,IACnD;AAEA,IAAA,OAAO,cAAA;AAAA,EACR;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAaA,MAAa,OAAA,CACZ,IAAA,EACA,KAAA,EACA,QAAA,EACkB;AAClB,IAAA,MAAM,UAAA,EAAsB,CAAC,CAAA;AAC7B,IAAA,MAAM,OAAA,EAAS,IAAA,CAAK,MAAA;AACpB,IAAA,MAAM,eAAA,EAAiB,IAAA,CAAK,iBAAA,CAAkB,KAAA,EAAO,QAAQ,CAAA;AAC7D,IAAA,MAAM,OAAA,EAA4B;AAAA,MACjC,OAAA,EAAS;AAAA,IACV,CAAA;AAEA,IAAA,IAAI;AACH,MAAA,MAAM,OAAA,EAAS,MAAA,CAAO,iBAAA,CAAkB,MAAM,CAAA;AAC9C,MAAA,IAAA,MAAA,CAAA,MAAiB,MAAA,GAAS,MAAA,EAAQ;AACjC,QAAA,GAAA,CAAI,KAAA,CAAM,KAAA,IAAS,eAAA,EAAiB;AACnC,UAAA,MAAM,WAAA,EAAa,KAAA;AACnB,UAAA,GAAA,CACC,UAAA,CAAW,MAAA,GACX,UAAA,CAAW,MAAA,CAAO,MAAA,IAAU,gBAAA,EAC3B;AACD,YAAA,IAAA,CAAK,QAAA,CAAS,MAAA,CAAO,QAAQ,CAAA;AAAA,UAC9B;AAEA,UAAA,MAAM,MAAA,kBAAQ,UAAA,mBAAW,MAAA,qBAAO,OAAA,6BAAS,KAAA,qBACvC,MAAA,mBAAO,CAAC,IAAA,EAAA,GAAS,IAAA,CAAK,KAAA,IAAS,MAAM,CAAA,qBACrC,GAAA,mBAAI,CAAC,IAAA,EAAA,GAAmB,IAAA,CAAK,IAAI,CAAA,qBACjC,IAAA,mBAAK,IAAI,GAAA;AACX,UAAA,GAAA,CAAI,KAAA,EAAO;AACV,YAAA,SAAA,CAAU,IAAA,CAAK,KAAK,CAAA;AAAA,UACrB;AAAA,QACD,EAAA,KAAA,GAAA,CAAW,KAAA,CAAM,KAAA,IAAS,SAAA,EAAW;AAEpC,UAAA,MAAM,IAAA,EAAM,KAAA;AACZ,UAAA,MAAM,OAAA,EAAS,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,QAAQ,CAAA;AACzC,UAAA,GAAA,CAAI,GAAA,CAAI,OAAA,GAAU,GAAA,CAAI,OAAA,IAAW,MAAA,EAAQ;AACxC,YAAA,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,QAAA,EAAU,GAAA,CAAI,MAAM,CAAA;AAAA,UACvC;AAAA,QACD,EAAA,KAAA,GAAA,CAAW,KAAA,CAAM,KAAA,IAAS,MAAA,EAAQ;AAEjC,UAAA,MAAM,KAAA,EAAO,KAAA;AACb,UAAA,GAAA,CAAI,IAAA,CAAK,GAAA,IAAO,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,QAAQ,CAAA,EAAG;AAC5C,YAAA,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,QAAA,EAAU,IAAA,CAAK,EAAE,CAAA;AAAA,UACpC;AAAA,QACD,EAAA,KAAO;AACN,UAAA,yBAAA,CAAQ,GAAA,CAAI,IAAA,CAAK,+CAAA,EAAiD;AAAA,YACjE;AAAA,UACD,CAAC,CAAA;AAAA,QACF;AAAA,MACD;AAAA,IACD,EAAA,MAAA,CAAS,KAAA,EAAO;AACf,MAAA,yBAAA,CAAQ,GAAA,CAAI,KAAA,CAAM,iCAAA,EAAmC,EAAE,MAAM,CAAC,CAAA;AAC9D,MAAA,IAAA,CAAK,OAAA,CAAQ,CAAA;AACb,MAAA,MAAM,WAAA,EAAa,CAAA,qBAAA,EAAwB,IAAA,CAAK,IAAA,CAAK,IAAI,CAAA;AAAA,EAAM,OAAO,MAAA,IAAU,SAAA,EAAW,MAAA,EAAQ,IAAA,CAAK,SAAA,CAAU,KAAA,EAAO,IAAA,EAAM,CAAC,CAAC,CAAA,CAAA;AAC1H,MAAA;AACR,IAAA;AAE6C,IAAA;AAA0B;AACxE,EAAA;AACD;AD7BqE;AACA;AACA;AACA","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-BBFYVWE2.cjs","sourcesContent":[null,"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"]}

package/dist/cjs/{chunk-LRBRG2DS.cjs → chunk-E35E4LQY.cjs}

@@ -1,6 +1,6 @@
 "use strict";Object.defineProperty(exports, "__esModule", {value: true});
 
-var _chunkDX2JH3FJcjs = require('./chunk-DX2JH3FJ.cjs');
+var _chunkQHQLU7BIcjs = require('./chunk-QHQLU7BI.cjs');
 
 // src/middlewares/auth.middleware.ts
 var _httpstatuscodes = require('http-status-codes');
@@ -17,7 +17,7 @@ var AuthMiddleware = class {
           res.locals.userId = authRes.userId;
           next();
         } else {
-          const error = new (0, _chunkDX2JH3FJcjs.AinHttpError)(
+          const error = new (0, _chunkQHQLU7BIcjs.AinHttpError)(
             _httpstatuscodes.StatusCodes.UNAUTHORIZED,
             "Unauthorized"
           );
@@ -25,7 +25,7 @@ var AuthMiddleware = class {
         }
       } catch (e) {
         if (!e.status) {
-          const error = new (0, _chunkDX2JH3FJcjs.AinHttpError)(
+          const error = new (0, _chunkQHQLU7BIcjs.AinHttpError)(
             _httpstatuscodes.StatusCodes.INTERNAL_SERVER_ERROR,
             `Authentication error: ${JSON.stringify(e)}`
           );
@@ -40,4 +40,4 @@ var AuthMiddleware = class {
 
 
 exports.AuthMiddleware = AuthMiddleware;
-//# sourceMappingURL=chunk-LRBRG2DS.cjs.map
+//# sourceMappingURL=chunk-E35E4LQY.cjs.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-E35E4LQY.cjs","../../src/middlewares/auth.middleware.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACA;ACHA,oDAA4B;AAKrB,IAAM,eAAA,EAAN,MAAqB;AAAA,EACnB;AAAA,EACR,WAAA,CAAY,IAAA,EAAgB;AAC3B,IAAA,IAAA,CAAK,KAAA,EAAO,IAAA;AAAA,EACb;AAAA,EAEO,UAAA,CAAA,EAA6B;AACnC,IAAA,OAAO,MAAA,CAAO,GAAA,EAAc,GAAA,EAAe,IAAA,EAAA,GAAuB;AACjE,MAAA,IAAI;AACH,QAAA,MAAM,QAAA,EAAwB,MAAM,IAAA,CAAK,IAAA,CAAK,YAAA,CAAa,GAAA,EAAK,GAAG,CAAA;AACnE,QAAA,GAAA,CAAI,OAAA,CAAQ,eAAA,EAAiB;AAC5B,UAAA,GAAA,CAAI,MAAA,CAAO,OAAA,EAAS,OAAA,CAAQ,MAAA;AAC5B,UAAA,IAAA,CAAK,CAAA;AAAA,QACN,EAAA,KAAO;AACN,UAAA,MAAM,MAAA,EAAsB,IAAI,mCAAA;AAAA,YAC/B,4BAAA,CAAY,YAAA;AAAA,YACZ;AAAA,UACD,CAAA;AACA,UAAA,MAAM,KAAA;AAAA,QACP;AAAA,MACD,EAAA,MAAA,CAAS,CAAA,EAAQ;AAChB,QAAA,GAAA,CAAI,CAAC,CAAA,CAAE,MAAA,EAAQ;AACd,UAAA,MAAM,MAAA,EAAsB,IAAI,mCAAA;AAAA,YAC/B,4BAAA,CAAY,qBAAA;AAAA,YACZ,CAAA,sBAAA,EAAyB,IAAA,CAAK,SAAA,CAAU,CAAC,CAAC,CAAA;AAAA,UAAA;AAE3C,UAAA;AAAM,QAAA;AAEP,QAAA;AAAM,MAAA;AACP,IAAA;AACD,EAAA;AAEF;ADAA;AACA;AACA;AACA","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-E35E4LQY.cjs","sourcesContent":[null,"import type { NextFunction, Request, RequestHandler, Response } from \"express\";\nimport { StatusCodes } from \"http-status-codes\";\nimport type { BaseAuth } from \"@/modules/auth/base.auth\";\nimport { AinHttpError } from \"@/types/agent\";\nimport type { AuthResponse } from \"@/types/auth\";\n\nexport class AuthMiddleware {\n\tprivate auth: BaseAuth;\n\tconstructor(auth: BaseAuth) {\n\t\tthis.auth = auth;\n\t}\n\n\tpublic middleware(): RequestHandler {\n\t\treturn async (req: Request, res: Response, next: NextFunction) => {\n\t\t\ttry {\n\t\t\t\tconst authRes: AuthResponse = await this.auth.authenticate(req, res);\n\t\t\t\tif (authRes.isAuthenticated) {\n\t\t\t\t\tres.locals.userId = authRes.userId;\n\t\t\t\t\tnext();\n\t\t\t\t} else {\n\t\t\t\t\tconst error: AinHttpError = new AinHttpError(\n\t\t\t\t\t\tStatusCodes.UNAUTHORIZED,\n\t\t\t\t\t\t\"Unauthorized\",\n\t\t\t\t\t);\n\t\t\t\t\tthrow error;\n\t\t\t\t}\n\t\t\t} catch (e: any) {\n\t\t\t\tif (!e.status) {\n\t\t\t\t\tconst error: AinHttpError = new AinHttpError(\n\t\t\t\t\t\tStatusCodes.INTERNAL_SERVER_ERROR,\n\t\t\t\t\t\t`Authentication error: ${JSON.stringify(e)}`,\n\t\t\t\t\t);\n\t\t\t\t\tthrow error;\n\t\t\t\t}\n\t\t\t\tthrow e;\n\t\t\t}\n\t\t};\n\t}\n}\n"]}

package/dist/cjs/{chunk-EAK7MP6X.cjs → chunk-FRWC3FQ4.cjs}

@@ -3,15 +3,15 @@
 var _chunkX46GJS7Xcjs = require('./chunk-X46GJS7X.cjs');
 
 
-var _chunkBBCZU5VZcjs = require('./chunk-BBCZU5VZ.cjs');
+var _chunk7PZP7YSAcjs = require('./chunk-7PZP7YSA.cjs');
 
 // src/routes/api.routes.ts
 var _express = require('express');
 var createApiRouter = (agent) => {
   const router = _express.Router.call(void 0, );
-  router.use("/api/model", _chunkX46GJS7Xcjs.createModelApiRouter.call(void 0, agent.modelModule));
+  router.use("/model", _chunkX46GJS7Xcjs.createModelApiRouter.call(void 0, agent.modelModule));
   if (agent.memoryModule) {
-    router.use("/api/thread", _chunkBBCZU5VZcjs.createThreadApiRouter.call(void 0, agent.memoryModule));
+    router.use("/threads", _chunk7PZP7YSAcjs.createThreadApiRouter.call(void 0, agent.memoryModule));
   }
   return router;
 };
@@ -19,4 +19,4 @@ var createApiRouter = (agent) => {
 
 
 exports.createApiRouter = createApiRouter;
-//# sourceMappingURL=chunk-EAK7MP6X.cjs.map
+//# sourceMappingURL=chunk-FRWC3FQ4.cjs.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-FRWC3FQ4.cjs","../../src/routes/api.routes.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACA;ACPA,kCAAuB;AAMhB,IAAM,gBAAA,EAAkB,CAAC,KAAA,EAAA,GAA4B;AAC3D,EAAA,MAAM,OAAA,EAAS,6BAAA,CAAO;AAEtB,EAAA,MAAA,CAAO,GAAA,CAAI,QAAA,EAAU,oDAAA,KAAqB,CAAM,WAAW,CAAC,CAAA;AAC5D,EAAA,GAAA,CAAI,KAAA,CAAM,YAAA,EAAc;AACvB,IAAA,MAAA,CAAO,GAAA,CAAI,UAAA,EAAY,qDAAA,KAAsB,CAAM,YAAY,CAAC,CAAA;AAAA,EACjE;AAEA,EAAA,OAAO,MAAA;AACR,CAAA;ADEA;AACA;AACE;AACF,0CAAC","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-FRWC3FQ4.cjs","sourcesContent":[null,"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"]}