@ainetwork/adk 0.0.2 → 0.1.1

package/README.md

@@ -80,7 +80,10 @@ await mcpModule.addMCPConfig({
 });
 
 // Pass to agent constructor
-const agent = new AINAgent(manifest, { modelModule, mcpModule, memoryModule });
+const agent = new AINAgent(
+  manifest,
+  { modelModule, mcpModule, memoryModule }
+);
 ```
 
 ### Adding A2A Support
@@ -155,30 +158,18 @@ Each module can be independently configured and passed to the agent constructor.
 ```bash
 # Build commands
 yarn build          # Build both ESM and CJS distributions
-yarn build:esm      # Build ESM format only  
-yarn build:cjs      # Build CJS format only
 
 # Development
 yarn dev            # Run TypeScript directly with tsx
 
 # Code quality
-yarn lint           # Run linting with Biome
-yarn format         # Format code with Biome
-yarn check          # Check code with Biome
+yarn biome          # Check code with Biome
 yarn check:write    # Check and auto-fix with Biome
 
 # Testing
 yarn test           # Run Jest tests
 ```
 
-### Examples
-
-```bash
-# Run example applications
-npx tsx examples/simpleAgent.ts
-npx tsx examples/a2aClientAgent.ts
-```
-
 ## Logging System
 
 The library uses Winston for structured logging with service-specific loggers:
@@ -236,27 +227,6 @@ The project supports dual build output:
 - **ESM** (`dist/esm/`): ES Module format with `{"type": "module"}`
 - **CJS** (`dist/cjs/`): CommonJS format with `{"type": "commonjs"}`
 
-Import paths use `@/` alias for `src/` directory.
-
-## Models Support
-
-### Built-in Models
-- **OpenAI/Azure OpenAI**: Full support for GPT models
-- **Google Gemini**: Support for Gemini models
-
-### Custom Models
-Extend the `BaseModel` class to add support for other AI models:
-
-```typescript
-import { BaseModel } from '@ainetwork/adk/modules/models/base';
-
-class MyCustomModel extends BaseModel {
-  async generateResponse(messages: any[], tools?: any[]): Promise<any> {
-    // Implement your model logic
-  }
-}
-```
-
 ## Error Handling
 
 The library includes comprehensive error handling:
@@ -276,16 +246,16 @@ All errors are logged with appropriate context for debugging.
 The library supports optional authentication middleware:
 
 ```typescript
-import { BaseAuth } from '@ainetwork/adk/middlewares/auth/base';
+import { BaseAuth } from '@ainetwork/adk/modules';
 
 class MyAuth extends BaseAuth {
-  async authenticate(req: Request): Promise<boolean> {
+  async authenticate(req: Request, res: Response): Promise<boolean> {
     // Implement your auth logic
     return true;
   }
 }
 
-const agent = new AINAgent(manifest, modules, false, new MyAuth());
+const agent = new AINAgent(manifest, modules, new MyAuth());
 ```
 
 ## Contributing
@@ -293,7 +263,7 @@ const agent = new AINAgent(manifest, modules, false, new MyAuth());
 1. Follow the established code conventions
 2. Use TypeScript strict mode
 3. Add appropriate service-specific logging
-4. Run `yarn check:write` and `yarn test` before submitting
+4. Run `yarn biome:write` and `yarn test` before submitting
 5. Maintain the modular architecture
 
 ## License

package/dist/cjs/chunk-5VIAH4HE.cjs

@@ -0,0 +1,34 @@
+"use strict";Object.defineProperty(exports, "__esModule", {value: true});// src/modules/memory/memory.module.ts
+var MemoryModule = class {
+  /** The memory storage implementation */
+  
+  /**
+   * Creates a new MemoryModule with the specified storage implementation.
+   *
+   * @param memory - The memory storage implementation to use
+   */
+  constructor(memory) {
+    this.memoryMap = /* @__PURE__ */ new Map();
+    this.memoryMap.set("_DEFAULT" /* _DEFAULT */, memory);
+  }
+  /**
+   * Returns the current memory storage implementation.
+   *
+   * @returns The active memory storage instance
+   */
+  getMemory(type) {
+    if (!type) {
+      return this.memoryMap.get("_DEFAULT" /* _DEFAULT */);
+    }
+    const memory = this.memoryMap.get(type);
+    if (!memory) {
+      return this.memoryMap.get("_DEFAULT" /* _DEFAULT */);
+    }
+    return memory;
+  }
+};
+
+
+
+exports.MemoryModule = MemoryModule;
+//# sourceMappingURL=chunk-5VIAH4HE.cjs.map

package/dist/cjs/chunk-5VIAH4HE.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-5VIAH4HE.cjs","../../src/modules/memory/memory.module.ts"],"names":[],"mappings":"AAAA;ACkBO,IAAM,aAAA,EAAN,MAAmB;AAAA;AAAA,EAEjB;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOR,WAAA,CAAY,MAAA,EAAoB;AAC/B,IAAA,IAAA,CAAK,UAAA,kBAAY,IAAI,GAAA,CAAI,CAAA;AACzB,IAAA,IAAA,CAAK,SAAA,CAAU,GAAA,CAAA,yBAAA,EAAyB,MAAM,CAAA;AAAA,EAC/C;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOO,SAAA,CAAU,IAAA,EAA2C;AAC3D,IAAA,GAAA,CAAI,CAAC,IAAA,EAAM;AACV,MAAA,OAAO,IAAA,CAAK,SAAA,CAAU,GAAA,CAAA,yBAAuB,CAAA;AAAA,IAC9C;AAEA,IAAA,MAAM,OAAA,EAAS,IAAA,CAAK,SAAA,CAAU,GAAA,CAAI,IAAI,CAAA;AACtC,IAAA,GAAA,CAAI,CAAC,MAAA,EAAQ;AACZ,MAAA,OAAO,IAAA,CAAK,SAAA,CAAU,GAAA,CAAA,yBAAuB,CAAA;AAAA,IAC9C;AACA,IAAA,OAAO,MAAA;AAAA,EACR;AACD,CAAA;ADnBA;AACA;AACE;AACF,oCAAC","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-5VIAH4HE.cjs","sourcesContent":[null,"import { MemoryType } from \"@/types/memory.js\";\nimport type { BaseMemory } from \"./base.memory.js\";\n/**\n * Module wrapper for memory storage implementations.\n *\n * Provides a consistent interface for accessing memory storage\n * functionality throughout the application. Currently supports\n * a single memory implementation but designed for future extensibility.\n *\n * @example\n * ```typescript\n * const inMemoryStorage = new InMemoryStorage();\n * const memoryModule = new MemoryModule(inMemoryStorage);\n *\n * const memory = memoryModule.getMemory();\n * await memory.updateSessionHistory(\"session-123\", chatMessage);\n * ```\n */\nexport class MemoryModule {\n\t/** The memory storage implementation */\n\tprivate memoryMap: Map<MemoryType, BaseMemory>;\n\n\t/**\n\t * Creates a new MemoryModule with the specified storage implementation.\n\t *\n\t * @param memory - The memory storage implementation to use\n\t */\n\tconstructor(memory: BaseMemory) {\n\t\tthis.memoryMap = new Map();\n\t\tthis.memoryMap.set(MemoryType._DEFAULT, memory);\n\t}\n\n\t/**\n\t * Returns the current memory storage implementation.\n\t *\n\t * @returns The active memory storage instance\n\t */\n\tpublic getMemory(type?: MemoryType): BaseMemory | undefined {\n\t\tif (!type) {\n\t\t\treturn this.memoryMap.get(MemoryType._DEFAULT);\n\t\t}\n\n\t\tconst memory = this.memoryMap.get(type);\n\t\tif (!memory) {\n\t\t\treturn this.memoryMap.get(MemoryType._DEFAULT);\n\t\t}\n\t\treturn memory;\n\t}\n}\n"]}

package/dist/cjs/chunk-6YJFC7BY.cjs

@@ -0,0 +1,22 @@
+"use strict";Object.defineProperty(exports, "__esModule", {value: true});
+
+var _chunkN7NGEMKLcjs = require('./chunk-N7NGEMKL.cjs');
+
+
+var _chunkD525HP3Rcjs = require('./chunk-D525HP3R.cjs');
+
+// src/routes/api.routes.ts
+var _express = require('express');
+var createApiRouter = (agent) => {
+  const router = _express.Router.call(void 0, );
+  router.use("/api/models", _chunkN7NGEMKLcjs.createModelApiRouter.call(void 0, agent.modelModule));
+  if (agent.memoryModule) {
+    router.use("/api/sessions", _chunkD525HP3Rcjs.createSessionApiRouter.call(void 0, agent.memoryModule));
+  }
+  return router;
+};
+
+
+
+exports.createApiRouter = createApiRouter;
+//# sourceMappingURL=chunk-6YJFC7BY.cjs.map

package/dist/cjs/chunk-6YJFC7BY.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-6YJFC7BY.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,aAAA,EAAe,oDAAA,KAAqB,CAAM,WAAW,CAAC,CAAA;AACjE,EAAA,GAAA,CAAI,KAAA,CAAM,YAAA,EAAc;AACvB,IAAA,MAAA,CAAO,GAAA,CAAI,eAAA,EAAiB,sDAAA,KAAuB,CAAM,YAAY,CAAC,CAAA;AAAA,EACvE;AAEA,EAAA,OAAO,MAAA;AACR,CAAA;ADEA;AACA;AACE;AACF,0CAAC","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-6YJFC7BY.cjs","sourcesContent":[null,"import { Router } from \"express\";\nimport type { AINAgent } from \"@/index.js\";\n\nimport { createModelApiRouter } from \"./api/models.routes.js\";\nimport { createSessionApiRouter } from \"./api/sessions.routes.js\";\n\nexport const createApiRouter = (agent: AINAgent): Router => {\n\tconst router = Router();\n\n\trouter.use(\"/api/models\", createModelApiRouter(agent.modelModule));\n\tif (agent.memoryModule) {\n\t\trouter.use(\"/api/sessions\", createSessionApiRouter(agent.memoryModule));\n\t}\n\n\treturn router;\n};\n"]}

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

@@ -0,0 +1,111 @@
+"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 _chunkJTAXZJQ2cjs = require('./chunk-JTAXZJQ2.cjs');
+
+// src/services/a2a.service.ts
+var _crypto = require('crypto');
+var A2AService = (_class = class {
+  
+  __init() {this.canceledTasks = /* @__PURE__ */ new Set()}
+  constructor(queryService) {;_class.prototype.__init.call(this);_class.prototype.__init2.call(this);_class.prototype.__init3.call(this);
+    this.queryService = queryService;
+  }
+  __init2() {this.cancelTask = async (taskId, _eventBus) => {
+    this.canceledTasks.add(taskId);
+  }}
+  __init3() {this.createTaskStatusUpdateEvent = (taskId, contextId, state, message) => {
+    return {
+      kind: "status-update",
+      taskId,
+      contextId,
+      status: {
+        state,
+        message: message ? {
+          kind: "message",
+          role: "agent",
+          messageId: _crypto.randomUUID.call(void 0, ),
+          parts: [{ kind: "text", text: message }],
+          taskId,
+          contextId
+        } : void 0,
+        timestamp: (/* @__PURE__ */ new Date()).toISOString()
+      },
+      final: state !== "working"
+    };
+  }}
+  async execute(requestContext, eventBus) {
+    const userMessage = requestContext.userMessage;
+    const { sessionId } = userMessage.metadata;
+    const existingTask = requestContext.task;
+    const taskId = _optionalChain([existingTask, 'optionalAccess', _ => _.id]) || _crypto.randomUUID.call(void 0, );
+    const contextId = userMessage.contextId || _optionalChain([existingTask, 'optionalAccess', _2 => _2.contextId]) || _crypto.randomUUID.call(void 0, );
+    if (!existingTask) {
+      const initialTask = {
+        kind: "task",
+        id: taskId,
+        contextId,
+        status: {
+          state: "submitted",
+          timestamp: (/* @__PURE__ */ new Date()).toISOString()
+        },
+        history: [userMessage],
+        metadata: userMessage.metadata,
+        artifacts: []
+      };
+      eventBus.publish(initialTask);
+    }
+    const workingStatusUpdate = this.createTaskStatusUpdateEvent(
+      taskId,
+      contextId,
+      "working"
+    );
+    eventBus.publish(workingStatusUpdate);
+    const message = userMessage.parts.filter((part) => part.kind === "text").map((part) => part.text).join("\n");
+    if (message.length === 0) {
+      _chunkJTAXZJQ2cjs.loggers.server.warn(`Empty message received for task ${taskId}.`);
+      const failureUpdate = this.createTaskStatusUpdateEvent(
+        taskId,
+        contextId,
+        "failed",
+        "No message found to process."
+      );
+      eventBus.publish(failureUpdate);
+      return;
+    }
+    try {
+      const response = await this.queryService.handleQuery(message, sessionId);
+      if (this.canceledTasks.has(taskId)) {
+        _chunkJTAXZJQ2cjs.loggers.server.info(`Task ${taskId} was canceled.`);
+        const canceledUpdate = this.createTaskStatusUpdateEvent(
+          taskId,
+          contextId,
+          "canceled"
+        );
+        eventBus.publish(canceledUpdate);
+        return;
+      }
+      const finalUpdate = this.createTaskStatusUpdateEvent(
+        taskId,
+        contextId,
+        "completed",
+        response.content
+      );
+      eventBus.publish(finalUpdate);
+      _chunkJTAXZJQ2cjs.loggers.server.info(`Task ${taskId} completed successfully.`);
+    } catch (error) {
+      _chunkJTAXZJQ2cjs.loggers.server.error(`Error processing task ${taskId}:`, error);
+      const errorUpdate = this.createTaskStatusUpdateEvent(
+        taskId,
+        contextId,
+        "failed",
+        `Agent error: ${error.message}`
+      );
+      eventBus.publish(errorUpdate);
+    }
+  }
+}, _class);
+
+
+
+exports.A2AService = A2AService;
+//# sourceMappingURL=chunk-7BZO57TG.cjs.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-7BZO57TG.cjs","../../src/services/a2a.service.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACA;ACJA,gCAA2B;AAcpB,IAAM,WAAA,YAAN,MAA0C;AAAA,EACxC;AAAA,iBACA,cAAA,kBAA6B,IAAI,GAAA,CAAY,EAAA;AAAA,EAErD,WAAA,CAAY,YAAA,EAA4B;AACvC,IAAA,IAAA,CAAK,aAAA,EAAe,YAAA;AAAA,EACrB;AAAA,kBAEO,WAAA,EAAa,MAAA,CACnB,MAAA,EACA,SAAA,EAAA,GACmB;AACnB,IAAA,IAAA,CAAK,aAAA,CAAc,GAAA,CAAI,MAAM,CAAA;AAAA,EAC9B,EAAA;AAAA,kBAEQ,4BAAA,EAA8B,CACrC,MAAA,EACA,SAAA,EACA,KAAA,EACA,OAAA,EAAA,GAC2B;AAC3B,IAAA,OAAO;AAAA,MACN,IAAA,EAAM,eAAA;AAAA,MACN,MAAA;AAAA,MACA,SAAA;AAAA,MACA,MAAA,EAAQ;AAAA,QACP,KAAA;AAAA,QACA,OAAA,EAAS,QAAA,EACN;AAAA,UACA,IAAA,EAAM,SAAA;AAAA,UACN,IAAA,EAAM,OAAA;AAAA,UACN,SAAA,EAAW,gCAAA,CAAW;AAAA,UACtB,KAAA,EAAO,CAAC,EAAE,IAAA,EAAM,MAAA,EAAQ,IAAA,EAAM,QAAQ,CAAC,CAAA;AAAA,UACvC,MAAA;AAAA,UACA;AAAA,QACD,EAAA,EACC,KAAA,CAAA;AAAA,QACH,SAAA,EAAA,iBAAW,IAAI,IAAA,CAAK,CAAA,CAAA,CAAE,WAAA,CAAY;AAAA,MACnC,CAAA;AAAA,MACA,KAAA,EAAO,MAAA,IAAU;AAAA,IAClB,CAAA;AAAA,EACD,EAAA;AAAA,EAEA,MAAM,OAAA,CACL,cAAA,EACA,QAAA,EACgB;AAChB,IAAA,MAAM,YAAA,EAAc,cAAA,CAAe,WAAA;AACnC,IAAA,MAAM,EAAE,UAAU,EAAA,EAAI,WAAA,CAAY,QAAA;AAClC,IAAA,MAAM,aAAA,EAAe,cAAA,CAAe,IAAA;AAEpC,IAAA,MAAM,OAAA,kBAAS,YAAA,2BAAc,KAAA,GAAM,gCAAA,CAAW;AAC9C,IAAA,MAAM,UAAA,EACL,WAAA,CAAY,UAAA,mBAAa,YAAA,6BAAc,YAAA,GAAa,gCAAA,CAAW;AAEhE,IAAA,GAAA,CAAI,CAAC,YAAA,EAAc;AAClB,MAAA,MAAM,YAAA,EAAoB;AAAA,QACzB,IAAA,EAAM,MAAA;AAAA,QACN,EAAA,EAAI,MAAA;AAAA,QACJ,SAAA;AAAA,QACA,MAAA,EAAQ;AAAA,UACP,KAAA,EAAO,WAAA;AAAA,UACP,SAAA,EAAA,iBAAW,IAAI,IAAA,CAAK,CAAA,CAAA,CAAE,WAAA,CAAY;AAAA,QACnC,CAAA;AAAA,QACA,OAAA,EAAS,CAAC,WAAW,CAAA;AAAA,QACrB,QAAA,EAAU,WAAA,CAAY,QAAA;AAAA,QACtB,SAAA,EAAW,CAAC;AAAA,MACb,CAAA;AACA,MAAA,QAAA,CAAS,OAAA,CAAQ,WAAW,CAAA;AAAA,IAC7B;AAEA,IAAA,MAAM,oBAAA,EAAsB,IAAA,CAAK,2BAAA;AAAA,MAChC,MAAA;AAAA,MACA,SAAA;AAAA,MACA;AAAA,IACD,CAAA;AACA,IAAA,QAAA,CAAS,OAAA,CAAQ,mBAAmB,CAAA;AAEpC,IAAA,MAAM,QAAA,EAAkB,WAAA,CAAY,KAAA,CAClC,MAAA,CAAO,CAAC,IAAA,EAAA,GAAS,IAAA,CAAK,KAAA,IAAS,MAAM,CAAA,CACrC,GAAA,CAAI,CAAC,IAAA,EAAA,GAAS,IAAA,CAAK,IAAI,CAAA,CACvB,IAAA,CAAK,IAAI,CAAA;AACX,IAAA,GAAA,CAAI,OAAA,CAAQ,OAAA,IAAW,CAAA,EAAG;AACzB,MAAA,yBAAA,CAAQ,MAAA,CAAO,IAAA,CAAK,CAAA,gCAAA,EAAmC,MAAM,CAAA,CAAA,CAAG,CAAA;AAChE,MAAA,MAAM,cAAA,EAAgB,IAAA,CAAK,2BAAA;AAAA,QAC1B,MAAA;AAAA,QACA,SAAA;AAAA,QACA,QAAA;AAAA,QACA;AAAA,MACD,CAAA;AACA,MAAA,QAAA,CAAS,OAAA,CAAQ,aAAa,CAAA;AAC9B,MAAA,MAAA;AAAA,IACD;AAEA,IAAA,IAAI;AACH,MAAA,MAAM,SAAA,EAAW,MAAM,IAAA,CAAK,YAAA,CAAa,WAAA,CAAY,OAAA,EAAS,SAAS,CAAA;AAEvE,MAAA,GAAA,CAAI,IAAA,CAAK,aAAA,CAAc,GAAA,CAAI,MAAM,CAAA,EAAG;AACnC,QAAA,yBAAA,CAAQ,MAAA,CAAO,IAAA,CAAK,CAAA,KAAA,EAAQ,MAAM,CAAA,cAAA,CAAgB,CAAA;AAClD,QAAA,MAAM,eAAA,EAAiB,IAAA,CAAK,2BAAA;AAAA,UAC3B,MAAA;AAAA,UACA,SAAA;AAAA,UACA;AAAA,QACD,CAAA;AACA,QAAA,QAAA,CAAS,OAAA,CAAQ,cAAc,CAAA;AAC/B,QAAA,MAAA;AAAA,MACD;AAEA,MAAA,MAAM,YAAA,EAAc,IAAA,CAAK,2BAAA;AAAA,QACxB,MAAA;AAAA,QACA,SAAA;AAAA,QACA,WAAA;AAAA,QACA,QAAA,CAAS;AAAA,MACV,CAAA;AACA,MAAA,QAAA,CAAS,OAAA,CAAQ,WAAW,CAAA;AAC5B,MAAA,yBAAA,CAAQ,MAAA,CAAO,IAAA,CAAK,CAAA,KAAA,EAAQ,MAAM,CAAA,wBAAA,CAA0B,CAAA;AAAA,IAC7D,EAAA,MAAA,CAAS,KAAA,EAAY;AACpB,MAAA,yBAAA,CAAQ,MAAA,CAAO,KAAA,CAAM,CAAA,sBAAA,EAAyB,MAAM,CAAA,CAAA,CAAA,EAAK,KAAK,CAAA;AAC9D,MAAA,MAAM,YAAA,EAAc,IAAA,CAAK,2BAAA;AAAA,QACxB,MAAA;AAAA,QACA,SAAA;AAAA,QACA,QAAA;AAAA,QACA,CAAA,aAAA,EAAgB,KAAA,CAAM,OAAO,CAAA;AAAA,MAAA;AAE9B,MAAA;AAA4B,IAAA;AAC7B,EAAA;AAEF;ADnCA;AACA;AACA;AACA","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-7BZO57TG.cjs","sourcesContent":[null,"import { randomUUID } from \"node:crypto\";\nimport type { Task, TaskStatusUpdateEvent } from \"@a2a-js/sdk\";\nimport type {\n\tAgentExecutor,\n\tExecutionEventBus,\n\tRequestContext,\n} from \"@a2a-js/sdk/server\";\nimport { loggers } from \"@/utils/logger.js\";\nimport type { QueryService } from \"./query.service.js\";\n\n/**\n * Implements the AgentExecutor interface from the a2a-js-sdk.\n * This service is responsible for the core business logic of executing an A2A task.\n */\nexport class A2AService implements AgentExecutor {\n\tprivate queryService: QueryService;\n\tprivate canceledTasks: Set<string> = new Set<string>();\n\n\tconstructor(queryService: QueryService) {\n\t\tthis.queryService = queryService;\n\t}\n\n\tpublic cancelTask = async (\n\t\ttaskId: string,\n\t\t_eventBus: ExecutionEventBus,\n\t): Promise<void> => {\n\t\tthis.canceledTasks.add(taskId);\n\t};\n\n\tprivate createTaskStatusUpdateEvent = (\n\t\ttaskId: string,\n\t\tcontextId: string,\n\t\tstate: \"working\" | \"failed\" | \"canceled\" | \"completed\",\n\t\tmessage?: string,\n\t): TaskStatusUpdateEvent => {\n\t\treturn {\n\t\t\tkind: \"status-update\",\n\t\t\ttaskId: taskId,\n\t\t\tcontextId: contextId,\n\t\t\tstatus: {\n\t\t\t\tstate: state,\n\t\t\t\tmessage: message\n\t\t\t\t\t? {\n\t\t\t\t\t\t\tkind: \"message\",\n\t\t\t\t\t\t\trole: \"agent\",\n\t\t\t\t\t\t\tmessageId: randomUUID(),\n\t\t\t\t\t\t\tparts: [{ kind: \"text\", text: message }],\n\t\t\t\t\t\t\ttaskId: taskId,\n\t\t\t\t\t\t\tcontextId: contextId,\n\t\t\t\t\t\t}\n\t\t\t\t\t: undefined,\n\t\t\t\ttimestamp: new Date().toISOString(),\n\t\t\t},\n\t\t\tfinal: state !== \"working\",\n\t\t};\n\t};\n\n\tasync execute(\n\t\trequestContext: RequestContext,\n\t\teventBus: ExecutionEventBus,\n\t): Promise<void> {\n\t\tconst userMessage = requestContext.userMessage;\n\t\tconst { sessionId } = userMessage.metadata as { sessionId: string };\n\t\tconst existingTask = requestContext.task;\n\n\t\tconst taskId = existingTask?.id || randomUUID();\n\t\tconst contextId =\n\t\t\tuserMessage.contextId || existingTask?.contextId || randomUUID();\n\n\t\tif (!existingTask) {\n\t\t\tconst initialTask: Task = {\n\t\t\t\tkind: \"task\",\n\t\t\t\tid: taskId,\n\t\t\t\tcontextId: contextId,\n\t\t\t\tstatus: {\n\t\t\t\t\tstate: \"submitted\",\n\t\t\t\t\ttimestamp: new Date().toISOString(),\n\t\t\t\t},\n\t\t\t\thistory: [userMessage],\n\t\t\t\tmetadata: userMessage.metadata,\n\t\t\t\tartifacts: [],\n\t\t\t};\n\t\t\teventBus.publish(initialTask);\n\t\t}\n\n\t\tconst workingStatusUpdate = this.createTaskStatusUpdateEvent(\n\t\t\ttaskId,\n\t\t\tcontextId,\n\t\t\t\"working\",\n\t\t);\n\t\teventBus.publish(workingStatusUpdate);\n\n\t\tconst message: string = userMessage.parts\n\t\t\t.filter((part) => part.kind === \"text\")\n\t\t\t.map((part) => part.text)\n\t\t\t.join(\"\\n\");\n\t\tif (message.length === 0) {\n\t\t\tloggers.server.warn(`Empty message received for task ${taskId}.`);\n\t\t\tconst failureUpdate = this.createTaskStatusUpdateEvent(\n\t\t\t\ttaskId,\n\t\t\t\tcontextId,\n\t\t\t\t\"failed\",\n\t\t\t\t\"No message found to process.\",\n\t\t\t);\n\t\t\teventBus.publish(failureUpdate);\n\t\t\treturn;\n\t\t}\n\n\t\ttry {\n\t\t\tconst response = await this.queryService.handleQuery(message, sessionId);\n\n\t\t\tif (this.canceledTasks.has(taskId)) {\n\t\t\t\tloggers.server.info(`Task ${taskId} was canceled.`);\n\t\t\t\tconst canceledUpdate = this.createTaskStatusUpdateEvent(\n\t\t\t\t\ttaskId,\n\t\t\t\t\tcontextId,\n\t\t\t\t\t\"canceled\",\n\t\t\t\t);\n\t\t\t\teventBus.publish(canceledUpdate);\n\t\t\t\treturn;\n\t\t\t}\n\n\t\t\tconst finalUpdate = this.createTaskStatusUpdateEvent(\n\t\t\t\ttaskId,\n\t\t\t\tcontextId,\n\t\t\t\t\"completed\",\n\t\t\t\tresponse.content,\n\t\t\t);\n\t\t\teventBus.publish(finalUpdate);\n\t\t\tloggers.server.info(`Task ${taskId} completed successfully.`);\n\t\t} catch (error: any) {\n\t\t\tloggers.server.error(`Error processing task ${taskId}:`, error);\n\t\t\tconst errorUpdate = this.createTaskStatusUpdateEvent(\n\t\t\t\ttaskId,\n\t\t\t\tcontextId,\n\t\t\t\t\"failed\",\n\t\t\t\t`Agent error: ${error.message}`,\n\t\t\t);\n\t\t\teventBus.publish(errorUpdate);\n\t\t}\n\t}\n}\n"]}

package/dist/cjs/chunk-AHMW3HCU.cjs

@@ -0,0 +1,168 @@
+"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;
+
+var _chunkF7I4G6JGcjs = require('./chunk-F7I4G6JG.cjs');
+
+
+var _chunkJTAXZJQ2cjs = require('./chunk-JTAXZJQ2.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); }
+  /** 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()}
+  /**
+   * Registers a new A2A peer server URL for connection.
+   *
+   * @param url - The URL of the A2A-compatible agent to connect to
+   */
+  async addA2APeerServer(url) {
+    this.a2aPeerServers.set(url, null);
+  }
+  /**
+   * Retrieves tools from all registered A2A peer servers.
+   *
+   * Attempts to connect to each registered server, fetch their agent cards,
+   * and create tool instances. Disables tools for unreachable servers.
+   *
+   * @returns Promise resolving to array of available A2A tools
+   */
+  async getTools() {
+    const tools = [];
+    for (const url of [...this.a2aPeerServers.keys()]) {
+      const tool = this.a2aPeerServers.get(url);
+      if (!tool || !tool.enabled) {
+        try {
+          const client = new (0, _client.A2AClient)(url);
+          const card = await client.getAgentCard();
+          const toolName = card.name.replaceAll(" ", "-");
+          const a2aTool = new (0, _chunkF7I4G6JGcjs.A2ATool)(toolName, client, card);
+          tools.push(a2aTool);
+        } catch (_error) {
+          if (tool) {
+            tool.disable();
+          }
+        }
+      } else {
+        tools.push(tool);
+      }
+    }
+    return tools;
+  }
+  /**
+   * Gets or creates an A2A session for the given session ID.
+   *
+   * @param sessionId - The session identifier
+   * @returns A2ASession object with task and context IDs
+   */
+  __init3() {this.getA2ASessionWithId = (sessionId) => {
+    const a2aSession = _nullishCoalesce(this.a2aSessions.get(sessionId), () => ( {
+      taskId: void 0,
+      contextId: void 0
+    }));
+    if (!this.a2aSessions.has(sessionId)) {
+      this.a2aSessions.set(sessionId, a2aSession);
+    }
+    return a2aSession;
+  }}
+  /**
+   * Constructs a message payload for A2A communication.
+   *
+   * Includes session context (task ID and context ID) if available
+   * for maintaining conversation continuity.
+   *
+   * @param query - The message content to send
+   * @param sessionId - The session identifier
+   * @returns Formatted Message object for A2A protocol
+   */
+  getMessagePayload(query, sessionId) {
+    const messagePayload = {
+      messageId: _crypto.randomUUID.call(void 0, ),
+      kind: "message",
+      role: "user",
+      // FIXME: it could be 'agent'
+      metadata: {
+        sessionId
+      },
+      parts: [
+        {
+          kind: "text",
+          text: query
+        }
+      ]
+    };
+    const a2aSession = this.getA2ASessionWithId(sessionId);
+    if (a2aSession.taskId) {
+      messagePayload.taskId = a2aSession.taskId;
+    }
+    if (a2aSession.contextId) {
+      messagePayload.contextId = a2aSession.contextId;
+    }
+    return messagePayload;
+  }
+  /**
+   * Executes an A2A tool by sending a message to the remote agent.
+   *
+   * Handles streaming responses, maintains session state, and extracts
+   * text content from various event types in the response stream.
+   *
+   * @param tool - The A2ATool instance to use
+   * @param messagePayload - The message to send to the agent
+   * @param sessionId - The session identifier for context tracking
+   * @returns Promise resolving to array of text responses from the agent
+   */
+  async useTool(tool, messagePayload, sessionId) {
+    const finalText = [];
+    const client = tool.client;
+    const params = {
+      message: messagePayload
+    };
+    const a2aSession = this.getA2ASessionWithId(sessionId);
+    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;
+          }
+          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) {
+            finalText.push(texts);
+          }
+        } 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;
+          }
+        } 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;
+          }
+        } else {
+          _chunkJTAXZJQ2cjs.loggers.a2a.warn("Received unknown event structure from stream:", {
+            event
+          });
+        }
+      }
+    } catch (error) {
+      _chunkJTAXZJQ2cjs.loggers.a2a.error("Error communicating with agent:", { error });
+      tool.disable();
+    }
+    return finalText;
+  }
+}, _class);
+
+
+
+exports.A2AModule = A2AModule;
+//# sourceMappingURL=chunk-AHMW3HCU.cjs.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-AHMW3HCU.cjs","../../src/modules/a2a/a2a.module.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACA;ACPA,gCAA2B;AAS3B,4CAA0B;AA+BnB,IAAM,UAAA,YAAN,MAAgB;AAAA;AAAA,iBAEd,eAAA,kBAA8C,IAAI,GAAA,CAAI,EAAA;AAAA;AAAA,kBAEtD,YAAA,kBAAuC,IAAI,GAAA,CAAI,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOvD,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,kBAQQ,oBAAA,EAAsB,CAAC,SAAA,EAAA,GAAkC;AAChE,IAAA,MAAM,WAAA,mBAAa,IAAA,CAAK,WAAA,CAAY,GAAA,CAAI,SAAS,CAAA,UAAK;AAAA,MACrD,MAAA,EAAQ,KAAA,CAAA;AAAA,MACR,SAAA,EAAW,KAAA;AAAA,IACZ,GAAA;AACA,IAAA,GAAA,CAAI,CAAC,IAAA,CAAK,WAAA,CAAY,GAAA,CAAI,SAAS,CAAA,EAAG;AACrC,MAAA,IAAA,CAAK,WAAA,CAAY,GAAA,CAAI,SAAA,EAAW,UAAU,CAAA;AAAA,IAC3C;AAEA,IAAA,OAAO,UAAA;AAAA,EACR,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAYO,iBAAA,CAAkB,KAAA,EAAe,SAAA,EAA4B;AACnE,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;AAAA,MACD,CAAA;AAAA,MACA,KAAA,EAAO;AAAA,QACN;AAAA,UACC,IAAA,EAAM,MAAA;AAAA,UACN,IAAA,EAAM;AAAA,QACP;AAAA,MACD;AAAA,IACD,CAAA;AAEA,IAAA,MAAM,WAAA,EAAa,IAAA,CAAK,mBAAA,CAAoB,SAAS,CAAA;AACrD,IAAA,GAAA,CAAI,UAAA,CAAW,MAAA,EAAQ;AACtB,MAAA,cAAA,CAAe,OAAA,EAAS,UAAA,CAAW,MAAA;AAAA,IACpC;AACA,IAAA,GAAA,CAAI,UAAA,CAAW,SAAA,EAAW;AACzB,MAAA,cAAA,CAAe,UAAA,EAAY,UAAA,CAAW,SAAA;AAAA,IACvC;AAEA,IAAA,OAAO,cAAA;AAAA,EACR;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAaA,MAAa,OAAA,CACZ,IAAA,EACA,cAAA,EACA,SAAA,EACoB;AACpB,IAAA,MAAM,UAAA,EAAsB,CAAC,CAAA;AAC7B,IAAA,MAAM,OAAA,EAAS,IAAA,CAAK,MAAA;AACpB,IAAA,MAAM,OAAA,EAA4B;AAAA,MACjC,OAAA,EAAS;AAAA,IACV,CAAA;AACA,IAAA,MAAM,WAAA,EAAa,IAAA,CAAK,mBAAA,CAAoB,SAAS,CAAA;AAErD,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,UAAA,CAAW,OAAA,EAAS,KAAA,CAAA;AAAA,UACrB;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,GAAA,CAAI,GAAA,CAAI,OAAA,GAAU,GAAA,CAAI,OAAA,IAAW,UAAA,CAAW,MAAA,EAAQ;AACnD,YAAA,UAAA,CAAW,OAAA,EAAS,GAAA,CAAI,MAAA;AAAA,UACzB;AACA,UAAA,GAAA,CAAI,GAAA,CAAI,UAAA,GAAa,GAAA,CAAI,UAAA,IAAc,UAAA,CAAW,SAAA,EAAW;AAC5D,YAAA,UAAA,CAAW,UAAA,EAAY,GAAA,CAAI,SAAA;AAAA,UAC5B;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,UAAA,CAAW,MAAA,EAAQ;AAClC,YAAA,UAAA,CAAW,OAAA,EAAS,IAAA,CAAK,EAAA;AAAA,UAC1B;AACA,UAAA,GAAA,CAAI,IAAA,CAAK,UAAA,GAAa,IAAA,CAAK,UAAA,IAAc,UAAA,CAAW,SAAA,EAAW;AAC9D,YAAA,UAAA,CAAW,UAAA,EAAY,IAAA,CAAK,SAAA;AAAA,UAC7B;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,EAAgB;AACxB,MAAA,yBAAA,CAAQ,GAAA,CAAI,KAAA,CAAM,iCAAA,EAAmC,EAAE,MAAM,CAAC,CAAA;AAC9D,MAAA,IAAA,CAAK,OAAA,CAAQ,CAAA;AAAA,IAEd;AAEA,IAAA,OAAO,SAAA;AAAA,EACR;AACD,UAAA;ADtDA;AACA;AACE;AACF,8BAAC","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-AHMW3HCU.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 { loggers } from \"@/utils/logger.js\";\nimport { A2ATool } from \"./a2a.tool.js\";\n\n/**\n * Represents an active A2A communication session.\n */\ninterface A2ASession {\n\t/** Current task ID for multi-turn conversations */\n\ttaskId: string | undefined;\n\t/** Context ID for maintaining conversation state */\n\tcontextId: string | undefined;\n}\n\n/**\n * Module for managing Agent-to-Agent (A2A) protocol connections.\n *\n * This module handles connections to other A2A-compatible agents, manages\n * conversation sessions, and provides an interface for inter-agent communication.\n * Supports multi-turn conversations with task and context tracking.\n *\n * @example\n * ```typescript\n * const a2aModule = new A2AModule();\n * await a2aModule.addA2APeerServer(\"https://api.example.com/agent\");\n *\n * const tools = await a2aModule.getTools();\n * const message = a2aModule.getMessagePayload(\"Hello\", \"session-123\");\n * const response = await a2aModule.useTool(tools[0], message, \"session-123\");\n * ```\n */\nexport class A2AModule {\n\t/** Map of A2A server URLs to their corresponding tool instances */\n\tprivate a2aPeerServers: Map<string, A2ATool | null> = new Map();\n\t/** Map of session IDs to their A2A session state */\n\tprivate a2aSessions: Map<string, A2ASession> = new Map();\n\n\t/**\n\t * Registers a new A2A peer server URL for connection.\n\t *\n\t * @param url - The URL of the A2A-compatible agent to connect to\n\t */\n\tpublic async addA2APeerServer(url: string): Promise<void> {\n\t\tthis.a2aPeerServers.set(url, null);\n\t}\n\n\t/**\n\t * Retrieves tools from all registered A2A peer servers.\n\t *\n\t * Attempts to connect to each registered server, fetch their agent cards,\n\t * and create tool instances. Disables tools for unreachable servers.\n\t *\n\t * @returns Promise resolving to array of available A2A tools\n\t */\n\tpublic async getTools(): Promise<A2ATool[]> {\n\t\tconst tools: A2ATool[] = [];\n\t\tfor (const url of [...this.a2aPeerServers.keys()]) {\n\t\t\tconst tool = this.a2aPeerServers.get(url);\n\t\t\tif (!tool || !tool.enabled) {\n\t\t\t\ttry {\n\t\t\t\t\tconst client = new A2AClient(url);\n\t\t\t\t\tconst card: AgentCard = await client.getAgentCard();\n\t\t\t\t\tconst toolName = card.name.replaceAll(\" \", \"-\");\n\t\t\t\t\tconst a2aTool = new A2ATool(toolName, client, card);\n\n\t\t\t\t\ttools.push(a2aTool);\n\t\t\t\t} catch (_error: any) {\n\t\t\t\t\t// Agent not responded\n\t\t\t\t\tif (tool) {\n\t\t\t\t\t\ttool.disable();\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\ttools.push(tool);\n\t\t\t}\n\t\t}\n\t\treturn tools;\n\t}\n\n\t/**\n\t * Gets or creates an A2A session for the given session ID.\n\t *\n\t * @param sessionId - The session identifier\n\t * @returns A2ASession object with task and context IDs\n\t */\n\tprivate getA2ASessionWithId = (sessionId: string): A2ASession => {\n\t\tconst a2aSession = this.a2aSessions.get(sessionId) ?? {\n\t\t\ttaskId: undefined,\n\t\t\tcontextId: undefined,\n\t\t};\n\t\tif (!this.a2aSessions.has(sessionId)) {\n\t\t\tthis.a2aSessions.set(sessionId, a2aSession);\n\t\t}\n\n\t\treturn a2aSession;\n\t};\n\n\t/**\n\t * Constructs a message payload for A2A communication.\n\t *\n\t * Includes session context (task ID and context ID) if available\n\t * for maintaining conversation continuity.\n\t *\n\t * @param query - The message content to send\n\t * @param sessionId - The session identifier\n\t * @returns Formatted Message object for A2A protocol\n\t */\n\tpublic getMessagePayload(query: string, sessionId: string): Message {\n\t\tconst messagePayload: Message = {\n\t\t\tmessageId: randomUUID(),\n\t\t\tkind: \"message\",\n\t\t\trole: \"user\", // FIXME: it could be 'agent'\n\t\t\tmetadata: {\n\t\t\t\tsessionId,\n\t\t\t},\n\t\t\tparts: [\n\t\t\t\t{\n\t\t\t\t\tkind: \"text\",\n\t\t\t\t\ttext: query,\n\t\t\t\t},\n\t\t\t],\n\t\t};\n\n\t\tconst a2aSession = this.getA2ASessionWithId(sessionId);\n\t\tif (a2aSession.taskId) {\n\t\t\tmessagePayload.taskId = a2aSession.taskId;\n\t\t}\n\t\tif (a2aSession.contextId) {\n\t\t\tmessagePayload.contextId = a2aSession.contextId;\n\t\t}\n\n\t\treturn messagePayload;\n\t}\n\n\t/**\n\t * Executes an A2A tool by sending a message to the remote agent.\n\t *\n\t * Handles streaming responses, maintains session state, and extracts\n\t * text content from various event types in the response stream.\n\t *\n\t * @param tool - The A2ATool instance to use\n\t * @param messagePayload - The message to send to the agent\n\t * @param sessionId - The session identifier for context tracking\n\t * @returns Promise resolving to array of text responses from the agent\n\t */\n\tpublic async useTool(\n\t\ttool: A2ATool,\n\t\tmessagePayload: Message,\n\t\tsessionId: string,\n\t): Promise<string[]> {\n\t\tconst finalText: string[] = [];\n\t\tconst client = tool.client;\n\t\tconst params: MessageSendParams = {\n\t\t\tmessage: messagePayload,\n\t\t};\n\t\tconst a2aSession = this.getA2ASessionWithId(sessionId);\n\n\t\ttry {\n\t\t\tconst stream = client.sendMessageStream(params);\n\t\t\tfor await (const event of stream) {\n\t\t\t\tif (event.kind === \"status-update\") {\n\t\t\t\t\tconst typedEvent = event as TaskStatusUpdateEvent;\n\t\t\t\t\tif (\n\t\t\t\t\t\ttypedEvent.final &&\n\t\t\t\t\t\ttypedEvent.status.state !== \"input-required\"\n\t\t\t\t\t) {\n\t\t\t\t\t\ta2aSession.taskId = undefined;\n\t\t\t\t\t}\n\t\t\t\t\t// TODO: handle 'file', 'data' parts\n\t\t\t\t\tconst texts = typedEvent.status.message?.parts\n\t\t\t\t\t\t.filter((part) => part.kind === \"text\")\n\t\t\t\t\t\t.map((part: TextPart) => part.text)\n\t\t\t\t\t\t.join(\"\\n\");\n\t\t\t\t\tif (texts) {\n\t\t\t\t\t\tfinalText.push(texts);\n\t\t\t\t\t}\n\t\t\t\t} else if (event.kind === \"message\") {\n\t\t\t\t\t// FIXME: handling text in 'message'?\n\t\t\t\t\tconst msg = event as Message;\n\t\t\t\t\tif (msg.taskId && msg.taskId !== a2aSession.taskId) {\n\t\t\t\t\t\ta2aSession.taskId = msg.taskId;\n\t\t\t\t\t}\n\t\t\t\t\tif (msg.contextId && msg.contextId !== a2aSession.contextId) {\n\t\t\t\t\t\ta2aSession.contextId = msg.contextId;\n\t\t\t\t\t}\n\t\t\t\t} else if (event.kind === \"task\") {\n\t\t\t\t\t// FIXME: handling text in 'task'?\n\t\t\t\t\tconst task = event as Task;\n\t\t\t\t\tif (task.id !== a2aSession.taskId) {\n\t\t\t\t\t\ta2aSession.taskId = task.id;\n\t\t\t\t\t}\n\t\t\t\t\tif (task.contextId && task.contextId !== a2aSession.contextId) {\n\t\t\t\t\t\ta2aSession.contextId = task.contextId;\n\t\t\t\t\t}\n\t\t\t\t} else {\n\t\t\t\t\tloggers.a2a.warn(\"Received unknown event structure from stream:\", {\n\t\t\t\t\t\tevent,\n\t\t\t\t\t});\n\t\t\t\t}\n\t\t\t}\n\t\t} catch (error: unknown) {\n\t\t\tloggers.a2a.error(\"Error communicating with agent:\", { error });\n\t\t\ttool.disable();\n\t\t\t// TODO: add failed & disabled text for next inference?\n\t\t}\n\n\t\treturn finalText;\n\t}\n}\n"]}

package/dist/cjs/chunk-CR53JLN7.cjs

@@ -0,0 +1,37 @@
+"use strict";Object.defineProperty(exports, "__esModule", {value: true});
+
+var _chunk7BZO57TGcjs = require('./chunk-7BZO57TG.cjs');
+
+
+var _chunkZMYG5HDTcjs = require('./chunk-ZMYG5HDT.cjs');
+
+
+var _chunkILGIEJCXcjs = require('./chunk-ILGIEJCX.cjs');
+
+// src/routes/a2a.routes.ts
+var _server = require('@a2a-js/sdk/server');
+var _express = require('express');
+var createA2ARouter = (agent) => {
+  const router = _express.Router.call(void 0, );
+  const taskStore = new (0, _server.InMemoryTaskStore)();
+  const queryService = new (0, _chunkZMYG5HDTcjs.QueryService)(
+    agent.modelModule,
+    agent.a2aModule,
+    agent.mcpModule,
+    agent.memoryModule,
+    agent.manifest.prompts
+  );
+  const a2aService = new (0, _chunk7BZO57TGcjs.A2AService)(queryService);
+  const a2aController = new (0, _chunkILGIEJCXcjs.A2AController)(
+    a2aService,
+    taskStore,
+    agent.generateAgentCard
+  );
+  router.post("/a2a", a2aController.handleA2ARequest);
+  return router;
+};
+
+
+
+exports.createA2ARouter = createA2ARouter;
+//# sourceMappingURL=chunk-CR53JLN7.cjs.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-CR53JLN7.cjs","../../src/routes/a2a.routes.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACA;ACVA,4CAAkC;AAClC,kCAAuB;AAchB,IAAM,gBAAA,EAAkB,CAAC,KAAA,EAAA,GAA4B;AAC3D,EAAA,MAAM,OAAA,EAAS,6BAAA,CAAO;AAGtB,EAAA,MAAM,UAAA,EAAY,IAAI,8BAAA,CAAkB,CAAA;AACxC,EAAA,MAAM,aAAA,EAAe,IAAI,mCAAA;AAAA,IACxB,KAAA,CAAM,WAAA;AAAA,IACN,KAAA,CAAM,SAAA;AAAA,IACN,KAAA,CAAM,SAAA;AAAA,IACN,KAAA,CAAM,YAAA;AAAA,IACN,KAAA,CAAM,QAAA,CAAS;AAAA,EAChB,CAAA;AACA,EAAA,MAAM,WAAA,EAAa,IAAI,iCAAA,CAAW,YAAY,CAAA;AAC9C,EAAA,MAAM,cAAA,EAAgB,IAAI,oCAAA;AAAA,IACzB,UAAA;AAAA,IACA,SAAA;AAAA,IACA,KAAA,CAAM;AAAA,EACP,CAAA;AAGA,EAAA,MAAA,CAAO,IAAA,CAAK,MAAA,EAAQ,aAAA,CAAc,gBAAgB,CAAA;AAElD,EAAA,OAAO,MAAA;AACR,CAAA;ADNA;AACA;AACE;AACF,0CAAC","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-CR53JLN7.cjs","sourcesContent":[null,"import { InMemoryTaskStore } from \"@a2a-js/sdk/server\";\nimport { Router } from \"express\";\nimport type { AINAgent } from \"@/index.js\";\nimport { QueryService } from \"@/services/query.service.js\";\nimport { A2AController } from \"../controllers/a2a.controller.js\";\nimport { A2AService } from \"../services/a2a.service.js\";\n\n/**\n * Creates and configures the A2A router.\n * This function is a \"composition root\" for the A2A feature,\n * creating and injecting all necessary dependencies.\n * @param intentAnalyzer The core intent analyzer.\n * @param agentCard The agent's card.\n * @returns An Express Router instance.\n */\nexport const createA2ARouter = (agent: AINAgent): Router => {\n\tconst router = Router();\n\n\t// 1. Create dependencies for the A2A feature\n\tconst taskStore = new InMemoryTaskStore();\n\tconst queryService = new QueryService(\n\t\tagent.modelModule,\n\t\tagent.a2aModule,\n\t\tagent.mcpModule,\n\t\tagent.memoryModule,\n\t\tagent.manifest.prompts,\n\t);\n\tconst a2aService = new A2AService(queryService);\n\tconst a2aController = new A2AController(\n\t\ta2aService,\n\t\ttaskStore,\n\t\tagent.generateAgentCard,\n\t);\n\n\t// 2. Define the route\n\trouter.post(\"/a2a\", a2aController.handleA2ARequest);\n\n\treturn router;\n};\n"]}

package/dist/cjs/chunk-D525HP3R.cjs

@@ -0,0 +1,17 @@
+"use strict";Object.defineProperty(exports, "__esModule", {value: true});
+
+var _chunkYRNIZTPFcjs = require('./chunk-YRNIZTPF.cjs');
+
+// src/routes/api/sessions.routes.ts
+var _express = require('express');
+var createSessionApiRouter = (memoryModule) => {
+  const router = _express.Router.call(void 0, );
+  const sessionApiController = new (0, _chunkYRNIZTPFcjs.SessionApiController)(memoryModule);
+  router.get("/:id", sessionApiController.handleSessionHistory);
+  return router;
+};
+
+
+
+exports.createSessionApiRouter = createSessionApiRouter;
+//# sourceMappingURL=chunk-D525HP3R.cjs.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-D525HP3R.cjs","../../src/routes/api/sessions.routes.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACA;ACJA,kCAAuB;AAIhB,IAAM,uBAAA,EAAyB,CAAC,YAAA,EAAA,GAAuC;AAC7E,EAAA,MAAM,OAAA,EAAS,6BAAA,CAAO;AACtB,EAAA,MAAM,qBAAA,EAAuB,IAAI,2CAAA,CAAqB,YAAY,CAAA;AAGlE,EAAA,MAAA,CAAO,GAAA,CAAI,MAAA,EAAQ,oBAAA,CAAqB,oBAAoB,CAAA;AAE5D,EAAA,OAAO,MAAA;AACR,CAAA;ADAA;AACA;AACE;AACF,wDAAC","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-D525HP3R.cjs","sourcesContent":[null,"import { Router } from \"express\";\nimport { SessionApiController } from \"@/controllers/api/session.api.controller.js\";\nimport type { MemoryModule } from \"@/modules/index.js\";\n\nexport const createSessionApiRouter = (memoryModule: MemoryModule): Router => {\n\tconst router = Router();\n\tconst sessionApiController = new SessionApiController(memoryModule);\n\n\t// APIs (prefix: /api/sessions)\n\trouter.get(\"/:id\", sessionApiController.handleSessionHistory);\n\n\treturn router;\n};\n"]}

package/dist/cjs/chunk-DOFTYGJR.cjs

@@ -0,0 +1,21 @@
+"use strict";Object.defineProperty(exports, "__esModule", {value: true}); var _class;// src/controllers/query.controller.ts
+var QueryController = (_class = class {
+  
+  constructor(queryService) {;_class.prototype.__init.call(this);
+    this.queryService = queryService;
+  }
+  __init() {this.handleQueryRequest = async (req, res, next) => {
+    try {
+      const { message, sessionId } = req.body;
+      const result = await this.queryService.handleQuery(message, sessionId);
+      res.status(200).json(result);
+    } catch (error) {
+      next(error);
+    }
+  }}
+}, _class);
+
+
+
+exports.QueryController = QueryController;
+//# sourceMappingURL=chunk-DOFTYGJR.cjs.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-DOFTYGJR.cjs","../../src/controllers/query.controller.ts"],"names":[],"mappings":"AAAA;ACGO,IAAM,gBAAA,YAAN,MAAsB;AAAA,EACpB;AAAA,EAER,WAAA,CAAY,YAAA,EAA4B;AACvC,IAAA,IAAA,CAAK,aAAA,EAAe,YAAA;AAAA,EACrB;AAAA,iBAEO,mBAAA,EAAqB,MAAA,CAC3B,GAAA,EACA,GAAA,EACA,IAAA,EAAA,GACI;AACJ,IAAA,IAAI;AACH,MAAA,MAAM,EAAE,OAAA,EAAS,UAAU,EAAA,EAAI,GAAA,CAAI,IAAA;AAEnC,MAAA,MAAM,OAAA,EAAS,MAAM,IAAA,CAAK,YAAA,CAAa,WAAA,CAAY,OAAA,EAAS,SAAS,CAAA;AAErE,MAAA,GAAA,CAAI,MAAA,CAAO,GAAG,CAAA,CAAE,IAAA,CAAK,MAAM,CAAA;AAAA,IAC5B,EAAA,MAAA,CAAS,KAAA,EAAO;AACf,MAAA,IAAA,CAAK,KAAK,CAAA;AAAA,IACX;AAAA,EACD,EAAA;AACD,UAAA;ADTA;AACA;AACE;AACF,0CAAC","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-DOFTYGJR.cjs","sourcesContent":[null,"import type { NextFunction, Request, Response } from \"express\";\nimport type { QueryService } from \"@/services\";\n\nexport class QueryController {\n\tprivate queryService;\n\n\tconstructor(queryService: QueryService) {\n\t\tthis.queryService = queryService;\n\t}\n\n\tpublic handleQueryRequest = async (\n\t\treq: Request,\n\t\tres: Response,\n\t\tnext: NextFunction,\n\t) => {\n\t\ttry {\n\t\t\tconst { message, sessionId } = req.body;\n\n\t\t\tconst result = await this.queryService.handleQuery(message, sessionId);\n\n\t\t\tres.status(200).json(result);\n\t\t} catch (error) {\n\t\t\tnext(error);\n\t\t}\n\t};\n}\n"]}

package/dist/cjs/chunk-DX2JH3FJ.cjs

@@ -0,0 +1,13 @@
+"use strict";Object.defineProperty(exports, "__esModule", {value: true});// src/types/agent.ts
+var AinHttpError = class extends Error {
+  
+  constructor(status, message) {
+    super(message);
+    this.status = status;
+  }
+};
+
+
+
+exports.AinHttpError = AinHttpError;
+//# sourceMappingURL=chunk-DX2JH3FJ.cjs.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-DX2JH3FJ.cjs","../../src/types/agent.ts"],"names":[],"mappings":"AAAA;AC+CO,IAAM,aAAA,EAAN,MAAA,QAA2B,MAAM;AAAA,EAChC;AAAA,EAEP,WAAA,CAAY,MAAA,EAAqB,OAAA,EAAiB;AACjD,IAAA,KAAA,CAAM,OAAO,CAAA;AACb,IAAA,IAAA,CAAK,OAAA,EAAS,MAAA;AAAA,EACf;AACD,CAAA;AD9CA;AACA;AACE;AACF,oCAAC","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-DX2JH3FJ.cjs","sourcesContent":[null,"import type { StatusCodes } from \"http-status-codes\";\n\n/**\n * Custom prompts configuration for the agent.\n *\n * Allows customization of agent behavior through system and agent-level prompts.\n */\nexport type AinAgentPrompts = {\n\t/** Agent-level prompt that defines the agent's personality and behavior */\n\tagent?: string;\n\t/** System-level prompt that provides context and instructions */\n\tsystem?: string;\n};\n\n/**\n * Agent manifest containing essential metadata and configuration.\n *\n * The manifest defines the agent's identity, version, and optional configuration\n * such as the public URL for A2A protocol support and custom prompts.\n *\n * @example\n * ```typescript\n * const manifest: AinAgentManifest = {\n *   name: \"CustomerSupportAgent\",\n *   description: \"AI agent for handling customer support queries\",\n *   version: \"1.0.0\",\n *   url: \"https://api.example.com/agent\",\n *   prompts: {\n *     system: \"You are a helpful customer support assistant.\",\n *     agent: \"Always be polite and provide detailed solutions.\"\n *   }\n * };\n * ```\n */\nexport type AinAgentManifest = {\n\t/** Unique name identifier for the agent */\n\tname: string;\n\t/** Human-readable description of the agent's purpose and capabilities */\n\tdescription: string;\n\t/** Semantic version of the agent (e.g., \"1.0.0\") */\n\tversion: string;\n\t/** Optional public URL for A2A protocol discovery and communication */\n\turl?: string;\n\t/** Optional custom prompts to configure agent behavior */\n\tprompts?: AinAgentPrompts;\n};\n\nexport class AinHttpError extends Error {\n\tpublic status?: StatusCodes;\n\n\tconstructor(status: StatusCodes, message: string) {\n\t\tsuper(message);\n\t\tthis.status = status;\n\t}\n}\n"]}

package/dist/cjs/chunk-DYRSSHDI.cjs

@@ -0,0 +1,22 @@
+"use strict";Object.defineProperty(exports, "__esModule", {value: true});
+
+var _chunkJTAXZJQ2cjs = require('./chunk-JTAXZJQ2.cjs');
+
+// src/middlewares/error.middleware.ts
+var errorMiddleware = (error, req, res, next) => {
+  try {
+    const status = error.status || 500;
+    const message = error.message || "Something went wrong";
+    _chunkJTAXZJQ2cjs.logger.error(
+      `[${req.method}] ${req.path} >> StatusCode:: ${status}, Message:: ${message}`
+    );
+    res.status(status).json({ message });
+  } catch (error2) {
+    next(error2);
+  }
+};
+
+
+
+exports.errorMiddleware = errorMiddleware;
+//# sourceMappingURL=chunk-DYRSSHDI.cjs.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-DYRSSHDI.cjs","../../src/middlewares/error.middleware.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACA;ACAO,IAAM,gBAAA,EAAkB,CAC9B,KAAA,EACA,GAAA,EACA,GAAA,EACA,IAAA,EAAA,GACU;AACV,EAAA,IAAI;AACH,IAAA,MAAM,OAAA,EAAiB,KAAA,CAAM,OAAA,GAAU,GAAA;AACvC,IAAA,MAAM,QAAA,EAAkB,KAAA,CAAM,QAAA,GAAW,sBAAA;AAEzC,IAAA,wBAAA,CAAO,KAAA;AAAA,MACN,CAAA,CAAA,EAAI,GAAA,CAAI,MAAM,CAAA,EAAA,EAAK,GAAA,CAAI,IAAI,CAAA,iBAAA,EAAoB,MAAM,CAAA,YAAA,EAAe,OAAO,CAAA;AAAA,IAAA;AAE5E,IAAA;AAAmC,EAAA;AAEnC,IAAA;AAAU,EAAA;AAEZ;ADJA;AACA;AACA;AACA","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-DYRSSHDI.cjs","sourcesContent":[null,"import type { NextFunction, Request, Response } from \"express\";\nimport type { AinHttpError } from \"@/types/agent\";\nimport { logger } from \"@/utils/logger\";\n\nexport const errorMiddleware = (\n\terror: AinHttpError,\n\treq: Request,\n\tres: Response,\n\tnext: NextFunction,\n): void => {\n\ttry {\n\t\tconst status: number = error.status || 500;\n\t\tconst message: string = error.message || \"Something went wrong\";\n\n\t\tlogger.error(\n\t\t\t`[${req.method}] ${req.path} >> StatusCode:: ${status}, Message:: ${message}`,\n\t\t);\n\t\tres.status(status).json({ message });\n\t} catch (error) {\n\t\tnext(error);\n\t}\n};\n"]}