@ainetwork/adk 0.2.1 → 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-GB3ON7UV.cjs → chunk-7PZP7YSA.cjs}

@@ -1,9 +1,9 @@
 "use strict";Object.defineProperty(exports, "__esModule", {value: true});
 
-var _chunkMIWLK57Kcjs = require('./chunk-MIWLK57K.cjs');
+var _chunkA2I7D6JWcjs = require('./chunk-A2I7D6JW.cjs');
 
 
-var _chunkDX2JH3FJcjs = require('./chunk-DX2JH3FJ.cjs');
+var _chunkQHQLU7BIcjs = require('./chunk-QHQLU7BI.cjs');
 
 // src/routes/api/threads.routes.ts
 
@@ -12,11 +12,11 @@ var _express = require('express');
 var _httpstatuscodes = require('http-status-codes');
 var createThreadApiRouter = (memoryModule) => {
   const router = _express.Router.call(void 0, );
-  const threadApiController = new (0, _chunkMIWLK57Kcjs.ThreadApiController)(memoryModule);
+  const threadApiController = new (0, _chunkA2I7D6JWcjs.ThreadApiController)(memoryModule);
   const checkThreadMemory = (req, res, next) => {
     const threadMemory = memoryModule.getThreadMemory();
     if (!threadMemory) {
-      const error = new (0, _chunkDX2JH3FJcjs.AinHttpError)(
+      const error = new (0, _chunkQHQLU7BIcjs.AinHttpError)(
         _httpstatuscodes.StatusCodes.SERVICE_UNAVAILABLE,
         "Memory module is not initialized"
       );
@@ -37,4 +37,4 @@ var createThreadApiRouter = (memoryModule) => {
 
 
 exports.createThreadApiRouter = createThreadApiRouter;
-//# sourceMappingURL=chunk-GB3ON7UV.cjs.map
+//# sourceMappingURL=chunk-7PZP7YSA.cjs.map

package/dist/cjs/{chunk-GB3ON7UV.cjs.map → chunk-7PZP7YSA.cjs.map}

@@ -1 +1 @@
-{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-GB3ON7UV.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-GB3ON7UV.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"]}
+{"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-MIWLK57K.cjs → chunk-A2I7D6JW.cjs}

@@ -42,4 +42,4 @@ var ThreadApiController = (_class = class {
 
 
 exports.ThreadApiController = ThreadApiController;
-//# sourceMappingURL=chunk-MIWLK57K.cjs.map
+//# 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-45G7LDK7.cjs → chunk-FRWC3FQ4.cjs}

@@ -3,7 +3,7 @@
 var _chunkX46GJS7Xcjs = require('./chunk-X46GJS7X.cjs');
 
 
-var _chunkGB3ON7UVcjs = require('./chunk-GB3ON7UV.cjs');
+var _chunk7PZP7YSAcjs = require('./chunk-7PZP7YSA.cjs');
 
 // src/routes/api.routes.ts
 var _express = require('express');
@@ -11,7 +11,7 @@ var createApiRouter = (agent) => {
   const router = _express.Router.call(void 0, );
   router.use("/model", _chunkX46GJS7Xcjs.createModelApiRouter.call(void 0, agent.modelModule));
   if (agent.memoryModule) {
-    router.use("/threads", _chunkGB3ON7UVcjs.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-45G7LDK7.cjs.map
+//# sourceMappingURL=chunk-FRWC3FQ4.cjs.map

package/dist/cjs/{chunk-45G7LDK7.cjs.map → chunk-FRWC3FQ4.cjs.map}

@@ -1 +1 @@
-{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-45G7LDK7.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-45G7LDK7.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"]}
+{"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"]}

package/dist/cjs/{chunk-AIE3SVG7.cjs → chunk-KHD7UAFW.cjs}

@@ -72,18 +72,20 @@ Please select and answer the most appropriate intent name from the available int
     return intent;
   }
   /**
-   * Fulfills the detected intent by generating a response.
+   * Fulfills the detected intent by generating a streaming response.
    *
    * Manages the complete inference loop including:
    * - Loading prompts and conversation history
    * - Collecting available tools from modules
    * - Executing model inference with tool support
    * - Processing tool calls iteratively until completion
+   * - Streaming results as Server-Sent Events
    *
    * @param query - The user's input query
    * @param threadId - Thread identifier for context
    * @param thread - Previous conversation history
-   * @returns Object containing process steps and final response
+   * @param intent - Optional detected intent with custom prompt
+   * @returns AsyncGenerator yielding StreamEvent objects
    */
   async *intentFulfilling(query, threadId, thread, intent) {
     const systemPrompt = `
@@ -138,10 +140,6 @@ ${_optionalChain([intent, 'optionalAccess', _8 => _8.prompt]) || ""}
         assembledToolCalls
       });
       if (assembledToolCalls.length > 0) {
-        const messagePayload = _optionalChain([this, 'access', _14 => _14.a2aModule, 'optionalAccess', _15 => _15.getMessagePayload, 'call', _16 => _16(
-          query,
-          threadId
-        )]);
         for (const toolCall of assembledToolCalls) {
           const toolCallId = _crypto.randomUUID.call(void 0, );
           const toolName = toolCall.function.name;
@@ -176,8 +174,7 @@ ${_optionalChain([intent, 'optionalAccess', _8 => _8.prompt]) || ""}
             _chunk466ED5WNcjs.loggers.intent.debug("A2A tool call", { toolName });
             toolResult = await this.a2aModule.useTool(
               selectedTool,
-              // biome-ignore lint/style/noNonNullAssertion: <a2aModule is guaranteed to be defined>
-              messagePayload,
+              query,
               threadId
             );
           } else {
@@ -232,32 +229,33 @@ ${_optionalChain([intent, 'optionalAccess', _8 => _8.prompt]) || ""}
     }
   }
   /**
-   * Main entry point for processing user queries.
+   * Main entry point for processing streaming user queries.
    *
    * Handles the complete query lifecycle:
-   * 1. Loads thread from memory
+   * 1. Loads or creates thread from memory
    * 2. Detects intent from the query
-   * 3. Fulfills the intent with AI response
-   * 4. Updates conversation history
+   * 3. Fulfills the intent with streaming AI response
+   * 4. Updates conversation history in real-time
    *
-   * @param type - The type of thread (e.g., chat, workflow)
-   * @param userId - The user's unique identifier
-   * @param threadId - Unique thread identifier
+   * @param threadMetadata - Metadata containing type, userId, and optional threadId
+   * @param threadMetadata.type - The type of thread (e.g., chat, workflow)
+   * @param threadMetadata.userId - The user's unique identifier
+   * @param threadMetadata.threadId - Optional thread identifier
    * @param query - The user's input query
-   * @returns Object containing the AI-generated response
+   * @returns AsyncGenerator yielding StreamEvent objects for SSE
    */
   async *handleQueryStream(threadMetadata, query) {
     const { type, userId } = threadMetadata;
     const queryStartAt = Date.now();
-    const threadMemory = _optionalChain([this, 'access', _17 => _17.memoryModule, 'optionalAccess', _18 => _18.getThreadMemory, 'call', _19 => _19()]);
+    const threadMemory = _optionalChain([this, 'access', _14 => _14.memoryModule, 'optionalAccess', _15 => _15.getThreadMemory, 'call', _16 => _16()]);
     let threadId = threadMetadata.threadId;
     let thread;
     if (threadId) {
-      thread = await _optionalChain([threadMemory, 'optionalAccess', _20 => _20.getThread, 'call', _21 => _21(userId, threadId)]);
+      thread = await _optionalChain([threadMemory, 'optionalAccess', _17 => _17.getThread, 'call', _18 => _18(userId, threadId)]);
     } else {
       threadId = _crypto.randomUUID.call(void 0, );
       const title = await this.generateTitle(query);
-      const metadata = await _optionalChain([threadMemory, 'optionalAccess', _22 => _22.createThread, 'call', _23 => _23(type, userId, threadId, title)]) || {
+      const metadata = await _optionalChain([threadMemory, 'optionalAccess', _19 => _19.createThread, 'call', _20 => _20(type, userId, threadId, title)]) || {
         type,
         threadId,
         title,
@@ -266,7 +264,7 @@ ${_optionalChain([intent, 'optionalAccess', _8 => _8.prompt]) || ""}
       _chunk466ED5WNcjs.loggers.intentStream.info("Create new thread", { metadata });
       yield { event: "thread_id", data: metadata };
     }
-    await _optionalChain([threadMemory, 'optionalAccess', _24 => _24.addMessagesToThread, 'call', _25 => _25(userId, threadId, [
+    await _optionalChain([threadMemory, 'optionalAccess', _21 => _21.addMessagesToThread, 'call', _22 => _22(userId, threadId, [
       {
         role: "USER" /* USER */,
         timestamp: queryStartAt,
@@ -281,7 +279,7 @@ ${_optionalChain([intent, 'optionalAccess', _8 => _8.prompt]) || ""}
         _chunk466ED5WNcjs.loggers.intentStream.debug("text_chunk", { event });
         finalResponseText += event.data.delta;
       } else if (event.event === "tool_start") {
-        await _optionalChain([threadMemory, 'optionalAccess', _26 => _26.addMessagesToThread, 'call', _27 => _27(userId, threadId, [
+        await _optionalChain([threadMemory, 'optionalAccess', _23 => _23.addMessagesToThread, 'call', _24 => _24(userId, threadId, [
           {
             role: "MODEL" /* MODEL */,
             timestamp: Date.now(),
@@ -297,7 +295,7 @@ ${_optionalChain([intent, 'optionalAccess', _8 => _8.prompt]) || ""}
           }
         ])]);
       } else if (event.event === "tool_output") {
-        await _optionalChain([threadMemory, 'optionalAccess', _28 => _28.addMessagesToThread, 'call', _29 => _29(userId, threadId, [
+        await _optionalChain([threadMemory, 'optionalAccess', _25 => _25.addMessagesToThread, 'call', _26 => _26(userId, threadId, [
           {
             role: "MODEL" /* MODEL */,
             timestamp: Date.now(),
@@ -312,7 +310,7 @@ ${_optionalChain([intent, 'optionalAccess', _8 => _8.prompt]) || ""}
       }
       yield event;
     }
-    await _optionalChain([threadMemory, 'optionalAccess', _30 => _30.addMessagesToThread, 'call', _31 => _31(userId, threadId, [
+    await _optionalChain([threadMemory, 'optionalAccess', _27 => _27.addMessagesToThread, 'call', _28 => _28(userId, threadId, [
       {
         role: "MODEL" /* MODEL */,
         timestamp: Date.now(),
@@ -325,4 +323,4 @@ ${_optionalChain([intent, 'optionalAccess', _8 => _8.prompt]) || ""}
 
 
 exports.QueryStreamService = QueryStreamService;
-//# sourceMappingURL=chunk-AIE3SVG7.cjs.map
+//# sourceMappingURL=chunk-KHD7UAFW.cjs.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-KHD7UAFW.cjs","../../src/services/query-stream.service.ts"],"names":["intent"],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACA;ACJA,gCAA2B;AA+BpB,IAAM,mBAAA,EAAN,MAAyB;AAAA,EACvB;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EAER,WAAA,CACC,WAAA,EACA,SAAA,EACA,SAAA,EACA,YAAA,EACA,OAAA,EACC;AACD,IAAA,IAAA,CAAK,YAAA,EAAc,WAAA;AACnB,IAAA,IAAA,CAAK,UAAA,EAAY,SAAA;AACjB,IAAA,IAAA,CAAK,UAAA,EAAY,SAAA;AACjB,IAAA,IAAA,CAAK,aAAA,EAAe,YAAA;AACpB,IAAA,IAAA,CAAK,QAAA,EAAU,OAAA;AAAA,EAChB;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EASA,MAAc,gBAAA,CACb,KAAA,EACA,MAAA,EAC8B;AAC9B,IAAA,MAAM,cAAA,EAAgB,IAAA,CAAK,WAAA,CAAY,QAAA,CAAS,CAAA;AAChD,IAAA,MAAM,aAAA,kBAAe,IAAA,mBAAK,YAAA,6BAAc,eAAA,mBAAgB,GAAA;AACxD,IAAA,GAAA,CAAI,CAAC,YAAA,EAAc;AAClB,MAAA,OAAO,KAAA,CAAA;AAAA,IACR;AAGA,IAAA,MAAM,QAAA,EAAU,MAAM,YAAA,CAAa,WAAA,CAAY,CAAA;AAE/C,IAAA,GAAA,CAAI,OAAA,CAAQ,OAAA,IAAW,CAAA,EAAG;AACzB,MAAA,yBAAA,CAAQ,YAAA,CAAa,IAAA,CAAK,iBAAiB,CAAA;AAC3C,MAAA,OAAO,KAAA,CAAA;AAAA,IACR;AAEA,IAAA,MAAM,WAAA,EAAa,OAAA,CACjB,GAAA,CAAI,CAACA,OAAAA,EAAAA,GAAW,CAAA,EAAA,EAAKA,OAAAA,CAAO,IAAI,CAAA,EAAA,EAAKA,OAAAA,CAAO,WAAW,CAAA,CAAA;AAOvC,IAAA;AAOT,MAAA;AAGF,MAAA;AAC0B,MAAA;AAEpB,IAAA;AAEQ,IAAA;AAAA;AAAA;AAGX;AAAA;AAAA;AAAA,qFAAA;AAKU,IAAA;AAAA;AAEL;AAAA;AAEW,qBAAA;AAAA;AAAA;AAAA,yFAAA;AAKsB,IAAA;AACxC,MAAA;AACP,MAAA;AACA,IAAA;AAEkD,IAAA;AAC5B,IAAA;AACc,MAAA;AACrC,IAAA;AACyC,IAAA;AACmB,IAAA;AAC/C,IAAA;AACoC,MAAA;AACjD,IAAA;AACO,IAAA;AACR,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAuB+B,EAAA;AACT,IAAA;AACmB,SAAA;AAAA;AAEf;AAAA;AAEC;AAAA;AAEN;AACd,CAAA;AAE0C,IAAA;AACA,IAAA;AAC/C,MAAA;AACA,MAAA;AACgC,MAAA;AAChC,IAAA;AAE4B,IAAA;AAC4B,IAAA;AACQ,IAAA;AAEJ,IAAA;AAE9B,IAAA;AAElB,IAAA;AAC+B,MAAA;AAC1C,QAAA;AACA,QAAA;AACD,MAAA;AAMO,MAAA;AAE4C,MAAA;AAET,MAAA;AACrB,QAAA;AACG,QAAA;AACwC,UAAA;AAC/B,YAAA;AACzB,cAAA;AACE,cAAA;AAC8B,cAAA;AACrC,YAAA;AAEuC,YAAA;AACwB,YAAA;AACrD,YAAA;AAC4C,cAAA;AACvD,UAAA;AACgC,QAAA;AAC1B,UAAA;AACE,YAAA;AAC4B,YAAA;AACpC,UAAA;AACD,QAAA;AACD,MAAA;AAEiD,MAAA;AAChD,QAAA;AACA,MAAA;AAEkC,MAAA;AACS,QAAA;AACZ,UAAA;AACK,UAAA;AACgC,UAAA;AAElD,UAAA;AAIf,UAAA;AACsD,YAAA;AAGjD,YAAA;AACE,cAAA;AACD,cAAA;AACL,gBAAA;AACA,gBAAA;AACA,gBAAA;AACA,gBAAA;AACD,cAAA;AACD,YAAA;AAC4D,YAAA;AAC1B,YAAA;AACjC,cAAA;AACA,cAAA;AACD,YAAA;AAIC,UAAA;AACK,YAAA;AACE,cAAA;AACD,cAAA;AACL,gBAAA;AACA,gBAAA;AACA,gBAAA;AACU,gBAAA;AACX,cAAA;AACD,YAAA;AACkD,YAAA;AAChB,YAAA;AACjC,cAAA;AACA,cAAA;AACA,cAAA;AACD,YAAA;AACM,UAAA;AAES,YAAA;AACkC,cAAA;AACjD,YAAA;AACA,YAAA;AACD,UAAA;AACM,UAAA;AACE,YAAA;AACD,YAAA;AACL,cAAA;AACuB,cAAA;AACvB,cAAA;AACQ,cAAA;AACT,YAAA;AACD,UAAA;AACiD,UAAA;AAEtB,UAAA;AACsB,UAAA;AAClD,QAAA;AACM,MAAA;AACN,QAAA;AACD,MAAA;AACD,IAAA;AACD,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAS4D,EAAA;AACrC,IAAA;AAClB,IAAA;AAC6C,MAAA;AACA,MAAA;AAC/C,QAAA;AACc,QAAA;AAAA;AAAA;AAAA,oFAAA;AAId,MAAA;AACkD,MAAA;AACxB,MAAA;AACZ,IAAA;AACsC,MAAA;AACpD,QAAA;AACA,QAAA;AACA,MAAA;AACM,MAAA;AACR,IAAA;AACD,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAyB+B,EAAA;AACL,IAAA;AACK,IAAA;AAC0B,IAAA;AAG1B,IAAA;AAC1B,IAAA;AACU,IAAA;AAC0C,MAAA;AACjD,IAAA;AACgB,MAAA;AACsB,MAAA;AAI1C,MAAA;AACA,QAAA;AACA,QAAA;AACA,QAAA;AACoB,QAAA;AACrB,MAAA;AAC0D,MAAA;AAChB,MAAA;AAC5C,IAAA;AAE0D,IAAA;AACzD,MAAA;AACC,QAAA;AACW,QAAA;AAC6B,QAAA;AACzC,MAAA;AACA,IAAA;AAGuD,IAAA;AAGY,IAAA;AAE5C,IAAA;AACU,IAAA;AACqB,MAAA;AACH,QAAA;AAClB,QAAA;AACQ,MAAA;AACkB,QAAA;AACzD,UAAA;AACC,YAAA;AACoB,YAAA;AACX,YAAA;AACF,cAAA;AACqC,cAAA;AAC5C,YAAA;AACU,YAAA;AACc,cAAA;AACF,cAAA;AACA,cAAA;AACtB,YAAA;AACD,UAAA;AACA,QAAA;AACwC,MAAA;AACiB,QAAA;AACzD,UAAA;AACC,YAAA;AACoB,YAAA;AACgC,YAAA;AAC1C,YAAA;AACc,cAAA;AACF,cAAA;AACA,cAAA;AACtB,YAAA;AACD,UAAA;AACA,QAAA;AACF,MAAA;AACM,MAAA;AACP,IAAA;AAE0D,IAAA;AACzD,MAAA;AACC,QAAA;AACoB,QAAA;AACgC,QAAA;AACrD,MAAA;AACA,IAAA;AACF,EAAA;AACD;ADjH0F;AACA;AACA;AACA","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-KHD7UAFW.cjs","sourcesContent":[null,"import { randomUUID } from \"node:crypto\";\nimport type {\n\tA2AModule,\n\tMCPModule,\n\tMemoryModule,\n\tModelModule,\n} from \"@/modules/index.js\";\nimport type { AinAgentPrompts } from \"@/types/agent.js\";\nimport {\n\ttype Intent,\n\tMessageRole,\n\ttype ThreadMetadata,\n\ttype ThreadObject,\n\ttype ThreadType,\n} from \"@/types/memory.js\";\nimport type { StreamEvent } from \"@/types/stream\";\nimport {\n\ttype IA2ATool,\n\ttype IAgentTool,\n\ttype IMCPTool,\n\tTOOL_PROTOCOL_TYPE,\n} from \"@/types/tool.js\";\nimport { loggers } from \"@/utils/logger.js\";\n\n/**\n * Service for processing user queries through the agent's AI pipeline.\n *\n * Orchestrates the query processing workflow including intent detection,\n * model inference, tool execution, and response generation. Manages\n * conversation context and coordinates between different modules.\n */\nexport class QueryStreamService {\n\tprivate modelModule: ModelModule;\n\tprivate a2aModule?: A2AModule;\n\tprivate mcpModule?: MCPModule;\n\tprivate memoryModule?: MemoryModule;\n\tprivate prompts?: AinAgentPrompts;\n\n\tconstructor(\n\t\tmodelModule: ModelModule,\n\t\ta2aModule?: A2AModule,\n\t\tmcpModule?: MCPModule,\n\t\tmemoryModule?: MemoryModule,\n\t\tprompts?: AinAgentPrompts,\n\t) {\n\t\tthis.modelModule = modelModule;\n\t\tthis.a2aModule = a2aModule;\n\t\tthis.mcpModule = mcpModule;\n\t\tthis.memoryModule = memoryModule;\n\t\tthis.prompts = prompts;\n\t}\n\n\t/**\n\t * Detects the intent from context.\n\t *\n\t * @param query - The user's input query\n\t * @param thread - The thread history\n\t * @returns The detected intent\n\t */\n\tprivate async intentTriggering(\n\t\tquery: string,\n\t\tthread: ThreadObject | undefined,\n\t): Promise<Intent | undefined> {\n\t\tconst modelInstance = this.modelModule.getModel();\n\t\tconst intentMemory = this.memoryModule?.getIntentMemory();\n\t\tif (!intentMemory) {\n\t\t\treturn undefined;\n\t\t}\n\n\t\t// 인텐트 목록 가져오기\n\t\tconst intents = await intentMemory.listIntents();\n\n\t\tif (intents.length === 0) {\n\t\t\tloggers.intentStream.warn(\"No intent found\");\n\t\t\treturn undefined;\n\t\t}\n\n\t\tconst intentList = intents\n\t\t\t.map((intent) => `- ${intent.name}: ${intent.description}`)\n\t\t\t.join(\"\\n\");\n\n\t\t// Convert session history to a string\n\t\tconst historyMessages = !thread\n\t\t\t? \"\"\n\t\t\t: Object.entries(thread.messages)\n\t\t\t\t\t.sort(([, a], [, b]) => a.timestamp - b.timestamp)\n\t\t\t\t\t.map(([_chatId, chat]) => {\n\t\t\t\t\t\tconst role =\n\t\t\t\t\t\t\tchat.role === \"USER\"\n\t\t\t\t\t\t\t\t? \"User\"\n\t\t\t\t\t\t\t\t: chat.role === \"MODEL\"\n\t\t\t\t\t\t\t\t\t? \"Assistant\"\n\t\t\t\t\t\t\t\t\t: \"System\";\n\t\t\t\t\t\tconst content = Array.isArray(chat.content.parts)\n\t\t\t\t\t\t\t? chat.content.parts.join(\" \")\n\t\t\t\t\t\t\t: String(chat.content.parts);\n\t\t\t\t\t\treturn `${role}: \"\"\"${content}\"\"\"`;\n\t\t\t\t\t})\n\t\t\t\t\t.join(\"\\n\");\n\n\t\tconst systemPrompt = `You are an expert in accurately identifying user intentions.\n\nAvailable intent list:\n${intentList}\n\nPlease select and answer only from the above intent list. \nPlease return only the exact intent name without any additional explanations or text.`;\n\n\t\tconst userMessage = `The following is the conversation history with the user:\n\n${historyMessages}\n\nLast user question: \"${query}\"\n\nBased on the above conversation history, please determine what the intention of the last user question is. \nPlease select and answer the most appropriate intent name from the available intent list.`;\n\n\t\tconst messages = modelInstance.generateMessages({\n\t\t\tquery: userMessage,\n\t\t\tsystemPrompt,\n\t\t});\n\n\t\tconst response = await modelInstance.fetch(messages);\n\t\tif (!response.content) {\n\t\t\tthrow new Error(\"No intent detected\");\n\t\t}\n\t\tconst intentName = response.content.trim();\n\t\tconst intent = await intentMemory.getIntentByName(intentName);\n\t\tif (!intent) {\n\t\t\tthrow new Error(`No intent found: ${intentName}`);\n\t\t}\n\t\treturn intent;\n\t}\n\n\t/**\n\t * Fulfills the detected intent by generating a streaming 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 * - Streaming results as Server-Sent Events\n\t *\n\t * @param query - The user's input query\n\t * @param threadId - Thread identifier for context\n\t * @param thread - Previous conversation history\n\t * @param intent - Optional detected intent with custom prompt\n\t * @returns AsyncGenerator yielding StreamEvent objects\n\t */\n\tpublic async *intentFulfilling(\n\t\tquery: string,\n\t\tthreadId: string,\n\t\tthread?: ThreadObject,\n\t\tintent?: Intent,\n\t): AsyncGenerator<StreamEvent> {\n\t\tconst systemPrompt = `\nToday is ${new Date().toLocaleDateString()}.\n\n${this.prompts?.agent || \"\"}\n\n${this.prompts?.system || \"\"}\n\n${intent?.prompt || \"\"}\n\t`.trim();\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\tthis.mcpModule && tools.push(...this.mcpModule.getTools());\n\t\tthis.a2aModule && tools.push(...(await this.a2aModule.getTools()));\n\n\t\tconst functions = modelInstance.convertToolsToFunctions(tools);\n\n\t\tconst processList: string[] = [];\n\n\t\twhile (true) {\n\t\t\tconst responseStream = await modelInstance.fetchStreamWithContextMessage(\n\t\t\t\tmessages,\n\t\t\t\tfunctions,\n\t\t\t);\n\n\t\t\tconst assembledToolCalls: {\n\t\t\t\tid: string;\n\t\t\t\ttype: \"function\";\n\t\t\t\tfunction: { name: string; arguments: string };\n\t\t\t}[] = [];\n\n\t\t\tloggers.intentStream.debug(\"messages\", { messages });\n\n\t\t\tfor await (const chunk of responseStream) {\n\t\t\t\tconst delta = chunk.delta;\n\t\t\t\tif (delta?.tool_calls) {\n\t\t\t\t\tfor (const { index, id, function: func } of delta.tool_calls) {\n\t\t\t\t\t\tassembledToolCalls[index] ??= {\n\t\t\t\t\t\t\tid: \"\",\n\t\t\t\t\t\t\ttype: \"function\",\n\t\t\t\t\t\t\tfunction: { name: \"\", arguments: \"\" },\n\t\t\t\t\t\t};\n\n\t\t\t\t\t\tif (id) assembledToolCalls[index].id = id;\n\t\t\t\t\t\tif (func?.name) assembledToolCalls[index].function.name = func.name;\n\t\t\t\t\t\tif (func?.arguments)\n\t\t\t\t\t\t\tassembledToolCalls[index].function.arguments += func.arguments;\n\t\t\t\t\t}\n\t\t\t\t} else if (chunk.delta?.content) {\n\t\t\t\t\tyield {\n\t\t\t\t\t\tevent: \"text_chunk\",\n\t\t\t\t\t\tdata: { delta: chunk.delta.content },\n\t\t\t\t\t};\n\t\t\t\t}\n\t\t\t}\n\n\t\t\tloggers.intentStream.debug(\"assembledToolCalls\", {\n\t\t\t\tassembledToolCalls,\n\t\t\t});\n\n\t\t\tif (assembledToolCalls.length > 0) {\n\t\t\t\tfor (const toolCall of assembledToolCalls) {\n\t\t\t\t\tconst toolCallId = randomUUID();\n\t\t\t\t\tconst toolName = toolCall.function.name;\n\t\t\t\t\tconst selectedTool = tools.filter((tool) => tool.id === toolName)[0];\n\n\t\t\t\t\tlet toolResult = \"\";\n\t\t\t\t\tif (\n\t\t\t\t\t\tthis.mcpModule &&\n\t\t\t\t\t\tselectedTool.protocol === TOOL_PROTOCOL_TYPE.MCP\n\t\t\t\t\t) {\n\t\t\t\t\t\tconst toolArgs = JSON.parse(toolCall.function.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\tyield {\n\t\t\t\t\t\t\tevent: \"tool_start\",\n\t\t\t\t\t\t\tdata: {\n\t\t\t\t\t\t\t\ttoolCallId,\n\t\t\t\t\t\t\t\tprotocol: TOOL_PROTOCOL_TYPE.MCP,\n\t\t\t\t\t\t\t\ttoolName,\n\t\t\t\t\t\t\t\ttoolArgs,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t};\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\tyield {\n\t\t\t\t\t\t\tevent: \"tool_start\",\n\t\t\t\t\t\t\tdata: {\n\t\t\t\t\t\t\t\ttoolCallId,\n\t\t\t\t\t\t\t\tprotocol: TOOL_PROTOCOL_TYPE.A2A,\n\t\t\t\t\t\t\t\ttoolName,\n\t\t\t\t\t\t\t\ttoolArgs: null,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t};\n\t\t\t\t\t\tloggers.intent.debug(\"A2A tool call\", { toolName });\n\t\t\t\t\t\ttoolResult = await this.a2aModule.useTool(\n\t\t\t\t\t\t\tselectedTool as IA2ATool,\n\t\t\t\t\t\t\tquery,\n\t\t\t\t\t\t\tthreadId,\n\t\t\t\t\t\t);\n\t\t\t\t\t} else {\n\t\t\t\t\t\t// Unrecognized tool type. It cannot be happened...\n\t\t\t\t\t\tloggers.intent.warn(\n\t\t\t\t\t\t\t`Unrecognized tool type: ${selectedTool.protocol}`,\n\t\t\t\t\t\t);\n\t\t\t\t\t\tcontinue;\n\t\t\t\t\t}\n\t\t\t\t\tyield {\n\t\t\t\t\t\tevent: \"tool_output\",\n\t\t\t\t\t\tdata: {\n\t\t\t\t\t\t\ttoolCallId,\n\t\t\t\t\t\t\tprotocol: selectedTool.protocol,\n\t\t\t\t\t\t\ttoolName,\n\t\t\t\t\t\t\tresult: toolResult,\n\t\t\t\t\t\t},\n\t\t\t\t\t};\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 {\n\t\t\t\tbreak;\n\t\t\t}\n\t\t}\n\t}\n\n\t/**\n\t * Generates a title for the conversation based on the query.\n\t *\n\t * @param query - The user's input query\n\t * @returns Promise resolving to a generated title\n\t */\n\n\tprivate async generateTitle(query: string): Promise<string> {\n\t\tconst DEFAULT_TITLE = \"New Chat\";\n\t\ttry {\n\t\t\tconst modelInstance = this.modelModule.getModel();\n\t\t\tconst messages = modelInstance.generateMessages({\n\t\t\t\tquery,\n\t\t\t\tsystemPrompt: `You are a helpful assistant that generates titles for conversations.\n  Please analyze the user's query and create a concise title that accurately reflects the conversation's core topic.\n  The title must be no more than 5 words long.\n  Respond with only the title. Do not include any punctuation or extra explanations.`,\n\t\t\t});\n\t\t\tconst response = await modelInstance.fetch(messages);\n\t\t\treturn response.content || DEFAULT_TITLE;\n\t\t} catch (error) {\n\t\t\tloggers.intentStream.error(\"Error generating title\", {\n\t\t\t\terror,\n\t\t\t\tquery,\n\t\t\t});\n\t\t\treturn DEFAULT_TITLE;\n\t\t}\n\t}\n\n\t/**\n\t * Main entry point for processing streaming user queries.\n\t *\n\t * Handles the complete query lifecycle:\n\t * 1. Loads or creates thread from memory\n\t * 2. Detects intent from the query\n\t * 3. Fulfills the intent with streaming AI response\n\t * 4. Updates conversation history in real-time\n\t *\n\t * @param threadMetadata - Metadata containing type, userId, and optional threadId\n\t * @param threadMetadata.type - The type of thread (e.g., chat, workflow)\n\t * @param threadMetadata.userId - The user's unique identifier\n\t * @param threadMetadata.threadId - Optional thread identifier\n\t * @param query - The user's input query\n\t * @returns AsyncGenerator yielding StreamEvent objects for SSE\n\t */\n\tpublic async *handleQueryStream(\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): AsyncGenerator<StreamEvent> {\n\t\tconst { type, userId } = threadMetadata;\n\t\tconst queryStartAt = Date.now();\n\t\tconst threadMemory = this.memoryModule?.getThreadMemory();\n\n\t\t// 1. Load or create thread\n\t\tlet threadId = threadMetadata.threadId;\n\t\tlet thread: ThreadObject | undefined;\n\t\tif (threadId) {\n\t\t\tthread = await threadMemory?.getThread(userId, threadId);\n\t\t} else {\n\t\t\tthreadId = randomUUID();\n\t\t\tconst title = await this.generateTitle(query);\n\n\t\t\tconst metadata =\n\t\t\t\t(await threadMemory?.createThread(type, userId, threadId, title)) ||\n\t\t\t\t({\n\t\t\t\t\ttype,\n\t\t\t\t\tthreadId,\n\t\t\t\t\ttitle,\n\t\t\t\t\tupdatedAt: Date.now(),\n\t\t\t\t} as ThreadMetadata);\n\t\t\tloggers.intentStream.info(\"Create new thread\", { metadata });\n\t\t\tyield { event: \"thread_id\", data: metadata };\n\t\t}\n\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]);\n\n\t\t// 2. intent triggering\n\t\tconst intent = await this.intentTriggering(query, thread);\n\n\t\t// 3. intent fulfillment\n\t\tconst stream = this.intentFulfilling(query, threadId, thread, intent);\n\n\t\tlet finalResponseText = \"\";\n\t\tfor await (const event of stream) {\n\t\t\tif (event.event === \"text_chunk\" && event.data.delta) {\n\t\t\t\tloggers.intentStream.debug(\"text_chunk\", { event });\n\t\t\t\tfinalResponseText += event.data.delta;\n\t\t\t} else if (event.event === \"tool_start\") {\n\t\t\t\tawait threadMemory?.addMessagesToThread(userId, threadId, [\n\t\t\t\t\t{\n\t\t\t\t\t\trole: MessageRole.MODEL,\n\t\t\t\t\t\ttimestamp: Date.now(),\n\t\t\t\t\t\tcontent: {\n\t\t\t\t\t\t\ttype: \"text\",\n\t\t\t\t\t\t\tparts: [JSON.stringify(event.data.toolArgs)],\n\t\t\t\t\t\t},\n\t\t\t\t\t\tmetadata: {\n\t\t\t\t\t\t\ttoolCallId: event.data.toolCallId,\n\t\t\t\t\t\t\ttoolName: event.data.toolName,\n\t\t\t\t\t\t\tprotocol: event.data.protocol,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t]);\n\t\t\t} else if (event.event === \"tool_output\") {\n\t\t\t\tawait threadMemory?.addMessagesToThread(userId, threadId, [\n\t\t\t\t\t{\n\t\t\t\t\t\trole: MessageRole.MODEL,\n\t\t\t\t\t\ttimestamp: Date.now(),\n\t\t\t\t\t\tcontent: { type: \"text\", parts: [event.data.result] },\n\t\t\t\t\t\tmetadata: {\n\t\t\t\t\t\t\ttoolCallId: event.data.toolCallId,\n\t\t\t\t\t\t\ttoolName: event.data.toolName,\n\t\t\t\t\t\t\tprotocol: event.data.protocol,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t]);\n\t\t\t}\n\t\t\tyield event;\n\t\t}\n\n\t\tawait threadMemory?.addMessagesToThread(userId, threadId, [\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: [finalResponseText] },\n\t\t\t},\n\t\t]);\n\t}\n}\n"]}