@ainetwork/adk 0.0.2 → 0.1.0

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-6BZKW7OJ.cjs

@@ -0,0 +1,27 @@
+"use strict";Object.defineProperty(exports, "__esModule", {value: true});
+
+var _chunkDOFTYGJRcjs = require('./chunk-DOFTYGJR.cjs');
+
+
+var _chunkAHF32UIXcjs = require('./chunk-AHF32UIX.cjs');
+
+// src/routes/query.routes.ts
+var _express = require('express');
+var createQueryRouter = (agent) => {
+  const router = _express.Router.call(void 0, );
+  const queryService = new (0, _chunkAHF32UIXcjs.QueryService)(
+    agent.modelModule,
+    agent.a2aModule,
+    agent.mcpModule,
+    agent.memoryModule,
+    agent.manifest.prompts
+  );
+  const queryController = new (0, _chunkDOFTYGJRcjs.QueryController)(queryService);
+  router.post("/query", queryController.handleQueryRequest);
+  return router;
+};
+
+
+
+exports.createQueryRouter = createQueryRouter;
+//# sourceMappingURL=chunk-6BZKW7OJ.cjs.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-6BZKW7OJ.cjs","../../src/routes/query.routes.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACA;ACPA,kCAAuB;AAKhB,IAAM,kBAAA,EAAoB,CAAC,KAAA,EAAA,GAA4B;AAC7D,EAAA,MAAM,OAAA,EAAS,6BAAA,CAAO;AAEtB,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,gBAAA,EAAkB,IAAI,sCAAA,CAAgB,YAAY,CAAA;AACxD,EAAA,MAAA,CAAO,IAAA,CAAK,QAAA,EAAU,eAAA,CAAgB,kBAAkB,CAAA;AAExD,EAAA,OAAO,MAAA;AACR,CAAA;ADGA;AACA;AACE;AACF,8CAAC","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-6BZKW7OJ.cjs","sourcesContent":[null,"import { Router } from \"express\";\nimport { QueryController } from \"@/controllers/query.controller.js\";\nimport type { AINAgent } from \"@/index.js\";\nimport { QueryService } from \"@/services/query.service.js\";\n\nexport const createQueryRouter = (agent: AINAgent): Router => {\n\tconst router = Router();\n\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 queryController = new QueryController(queryService);\n\trouter.post(\"/query\", queryController.handleQueryRequest);\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-A5UYNW2C.cjs

@@ -0,0 +1,34 @@
+"use strict";Object.defineProperty(exports, "__esModule", {value: true}); var _class;
+
+var _chunkPL7Q2WFScjs = require('./chunk-PL7Q2WFS.cjs');
+
+// src/controllers/api/session.api.controller.ts
+var _httpstatuscodes = require('http-status-codes');
+var SessionApiController = (_class = class {
+  
+  constructor(memoryModule) {;_class.prototype.__init.call(this);
+    this.memoryModule = memoryModule;
+  }
+  __init() {this.handleSessionHistory = async (req, res, next) => {
+    try {
+      const { id: sessionId } = req.params;
+      const memoryInstance = this.memoryModule.getMemory();
+      if (!memoryInstance) {
+        const error = new (0, _chunkPL7Q2WFScjs.AinHttpError)(
+          _httpstatuscodes.StatusCodes.SERVICE_UNAVAILABLE,
+          "Memory module is not initialized"
+        );
+        throw error;
+      }
+      const session = await memoryInstance.getSessionHistory(sessionId);
+      res.json(session);
+    } catch (error) {
+      next(error);
+    }
+  }}
+}, _class);
+
+
+
+exports.SessionApiController = SessionApiController;
+//# sourceMappingURL=chunk-A5UYNW2C.cjs.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-A5UYNW2C.cjs","../../src/controllers/api/session.api.controller.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACA;ACHA,oDAA4B;AAGrB,IAAM,qBAAA,YAAN,MAA2B;AAAA,EACzB;AAAA,EAER,WAAA,CAAY,YAAA,EAA4B;AACvC,IAAA,IAAA,CAAK,aAAA,EAAe,YAAA;AAAA,EACrB;AAAA,iBAEO,qBAAA,EAAuB,MAAA,CAC7B,GAAA,EACA,GAAA,EACA,IAAA,EAAA,GACI;AACJ,IAAA,IAAI;AACH,MAAA,MAAM,EAAE,EAAA,EAAI,UAAU,EAAA,EAAI,GAAA,CAAI,MAAA;AAC9B,MAAA,MAAM,eAAA,EAAiB,IAAA,CAAK,YAAA,CAAa,SAAA,CAAU,CAAA;AACnD,MAAA,GAAA,CAAI,CAAC,cAAA,EAAgB;AACpB,QAAA,MAAM,MAAA,EAAQ,IAAI,mCAAA;AAAA,UACjB,4BAAA,CAAY,mBAAA;AAAA,UACZ;AAAA,QACD,CAAA;AACA,QAAA,MAAM,KAAA;AAAA,MACP;AACA,MAAA,MAAM,QAAA,EAAU,MAAM,cAAA,CAAe,iBAAA,CAAkB,SAAS,CAAA;AAChE,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;ADHA;AACA;AACE;AACF,oDAAC","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-A5UYNW2C.cjs","sourcesContent":[null,"import type { NextFunction, Request, Response } from \"express\";\nimport { StatusCodes } from \"http-status-codes\";\nimport type { MemoryModule } from \"@/modules/index.js\";\nimport { AinHttpError } from \"@/types/index.js\";\nexport class SessionApiController {\n\tprivate memoryModule: MemoryModule;\n\n\tconstructor(memoryModule: MemoryModule) {\n\t\tthis.memoryModule = memoryModule;\n\t}\n\n\tpublic handleSessionHistory = async (\n\t\treq: Request,\n\t\tres: Response,\n\t\tnext: NextFunction,\n\t) => {\n\t\ttry {\n\t\t\tconst { id: sessionId } = req.params;\n\t\t\tconst memoryInstance = this.memoryModule.getMemory();\n\t\t\tif (!memoryInstance) {\n\t\t\t\tconst error = new AinHttpError(\n\t\t\t\t\tStatusCodes.SERVICE_UNAVAILABLE,\n\t\t\t\t\t\"Memory module is not initialized\",\n\t\t\t\t);\n\t\t\t\tthrow error;\n\t\t\t}\n\t\t\tconst session = await memoryInstance.getSessionHistory(sessionId);\n\t\t\tres.json(session);\n\t\t} catch (error) {\n\t\t\tnext(error);\n\t\t}\n\t};\n}\n"]}

package/dist/cjs/chunk-AHF32UIX.cjs

@@ -0,0 +1,171 @@
+"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 _chunkJTAXZJQ2cjs = require('./chunk-JTAXZJQ2.cjs');
+
+// src/services/query.service.ts
+var QueryService = class {
+  
+  
+  
+  
+  
+  constructor(modelModule, a2aModule, mcpModule, memoryModule, prompts) {
+    this.modelModule = modelModule;
+    this.a2aModule = a2aModule;
+    this.mcpModule = mcpModule;
+    this.memoryModule = memoryModule;
+    this.prompts = prompts;
+  }
+  /**
+   * Detects the intent from a user query.
+   *
+   * @param query - The user's input query
+   * @returns The detected intent (currently returns the query as-is)
+   * @todo Implement actual intent detection logic
+   */
+  async intentTriggering(query) {
+    return query;
+  }
+  /**
+   * Fulfills the detected intent by generating a 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
+   *
+   * @param query - The user's input query
+   * @param sessionId - Session identifier for context
+   * @param sessionHistory - Previous conversation history
+   * @returns Object containing process steps and final response
+   */
+  async intentFulfilling(query, sessionId, sessionHistory) {
+    const systemPrompt = `
+Today is ${(/* @__PURE__ */ new Date()).toLocaleDateString()}.
+
+${_optionalChain([this, 'access', _ => _.prompts, 'optionalAccess', _2 => _2.agent]) || ""}
+
+${_optionalChain([this, 'access', _3 => _3.prompts, 'optionalAccess', _4 => _4.system]) || ""}
+    `;
+    const modelInstance = this.modelModule.getModel();
+    const messages = modelInstance.generateMessages({
+      query,
+      sessionHistory,
+      systemPrompt: systemPrompt.trim()
+    });
+    const tools = [];
+    if (this.mcpModule) {
+      tools.push(...this.mcpModule.getTools());
+    }
+    if (this.a2aModule) {
+      tools.push(...await this.a2aModule.getTools());
+    }
+    const functions = modelInstance.convertToolsToFunctions(tools);
+    const processList = [];
+    let finalMessage = "";
+    let didCallTool = false;
+    while (true) {
+      const response = await modelInstance.fetchWithContextMessage(
+        messages,
+        functions
+      );
+      didCallTool = false;
+      _chunkJTAXZJQ2cjs.loggers.intent.debug("messages", { messages });
+      const { content, toolCalls } = response;
+      _chunkJTAXZJQ2cjs.loggers.intent.debug("content", { content });
+      _chunkJTAXZJQ2cjs.loggers.intent.debug("tool_calls", { ...toolCalls });
+      if (toolCalls) {
+        const messagePayload = _optionalChain([this, 'access', _5 => _5.a2aModule, 'optionalAccess', _6 => _6.getMessagePayload, 'call', _7 => _7(
+          query,
+          sessionId
+        )]);
+        for (const toolCall of toolCalls) {
+          const toolName = toolCall.name;
+          didCallTool = true;
+          const selectedTool = tools.filter((tool) => tool.id === toolName)[0];
+          let toolResult = "";
+          if (this.mcpModule && selectedTool.protocol === "MCP" /* MCP */) {
+            const toolArgs = toolCall.arguments;
+            _chunkJTAXZJQ2cjs.loggers.intent.debug("MCP tool call", { toolName, toolArgs });
+            const result = await this.mcpModule.useTool(
+              selectedTool,
+              toolArgs
+            );
+            toolResult = `[Bot Called MCP Tool ${toolName} with args ${JSON.stringify(toolArgs)}]
+` + JSON.stringify(result.content, null, 2);
+          } else if (this.a2aModule && selectedTool.protocol === "A2A" /* A2A */) {
+            const result = await this.a2aModule.useTool(
+              selectedTool,
+              messagePayload,
+              sessionId
+            );
+            toolResult = `[Bot Called A2A Tool ${toolName}]
+${result.join("\n")}`;
+          } else {
+            _chunkJTAXZJQ2cjs.loggers.intent.warn(
+              `Unrecognized tool type: ${selectedTool.protocol}`
+            );
+            continue;
+          }
+          _chunkJTAXZJQ2cjs.loggers.intent.debug("toolResult", { toolResult });
+          processList.push(toolResult);
+          modelInstance.appendMessages(messages, toolResult);
+        }
+      } else if (content) {
+        processList.push(content);
+        finalMessage = content;
+      }
+      if (!didCallTool) break;
+    }
+    const botResponse = {
+      process: processList.join("\n"),
+      response: finalMessage
+    };
+    return botResponse;
+  }
+  /**
+   * Main entry point for processing user queries.
+   *
+   * Handles the complete query lifecycle:
+   * 1. Loads session history from memory
+   * 2. Detects intent from the query
+   * 3. Fulfills the intent with AI response
+   * 4. Updates conversation history
+   *
+   * @param query - The user's input query
+   * @param sessionId - Unique session identifier
+   * @returns Object containing the AI-generated response
+   */
+  async handleQuery(query, sessionId) {
+    const queryStartAt = Date.now();
+    const memoryInstance = _optionalChain([this, 'access', _8 => _8.memoryModule, 'optionalAccess', _9 => _9.getMemory, 'call', _10 => _10()]);
+    const sessionHistory = await _optionalChain([memoryInstance, 'optionalAccess', _11 => _11.getSessionHistory, 'call', _12 => _12(
+      sessionId
+    )]) || { chats: {} };
+    const intent = this.intentTriggering(query);
+    const result = await this.intentFulfilling(
+      query,
+      sessionId,
+      sessionHistory
+    );
+    if (sessionId) {
+      await _optionalChain([memoryInstance, 'optionalAccess', _13 => _13.updateSessionHistory, 'call', _14 => _14(sessionId, {
+        role: "USER" /* USER */,
+        timestamp: queryStartAt,
+        content: { type: "text", parts: [query] }
+      })]);
+      await _optionalChain([memoryInstance, 'optionalAccess', _15 => _15.updateSessionHistory, 'call', _16 => _16(sessionId, {
+        role: "MODEL" /* MODEL */,
+        timestamp: Date.now(),
+        content: { type: "text", parts: [result.response] }
+      })]);
+    }
+    return { content: result.response };
+  }
+};
+
+
+
+exports.QueryService = QueryService;
+//# sourceMappingURL=chunk-AHF32UIX.cjs.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-AHF32UIX.cjs","../../src/services/query.service.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACA;ACmBO,IAAM,aAAA,EAAN,MAAmB;AAAA,EACjB;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,CAAiB,KAAA,EAAe;AAE7C,IAAA,OAAO,KAAA;AAAA,EACR;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAgBA,MAAc,gBAAA,CACb,KAAA,EACA,SAAA,EACA,cAAA,EACC;AAED,IAAA,MAAM,aAAA,EAAe,CAAA;AAAA,SAAA,EAAA,iBACZ,IAAI,IAAA,CAAK,CAAA,CAAA,CAAE,kBAAA,CAAmB,CAAC,CAAA;AAAA;AAAA,kBAExC,IAAA,mBAAK,OAAA,6BAAS,QAAA,GAAS,EAAE,CAAA;AAAA;AAAA,kBAEzB,IAAA,qBAAK,OAAA,6BAAS,SAAA,GAAU,EAAE,CAAA;AAAA,IAAA,CAAA;AAG1B,IAAA,MAAM,cAAA,EAAgB,IAAA,CAAK,WAAA,CAAY,QAAA,CAAS,CAAA;AAChD,IAAA,MAAM,SAAA,EAAW,aAAA,CAAc,gBAAA,CAAiB;AAAA,MAC/C,KAAA;AAAA,MACA,cAAA;AAAA,MACA,YAAA,EAAc,YAAA,CAAa,IAAA,CAAK;AAAA,IACjC,CAAC,CAAA;AAED,IAAA,MAAM,MAAA,EAAsB,CAAC,CAAA;AAC7B,IAAA,GAAA,CAAI,IAAA,CAAK,SAAA,EAAW;AACnB,MAAA,KAAA,CAAM,IAAA,CAAK,GAAG,IAAA,CAAK,SAAA,CAAU,QAAA,CAAS,CAAC,CAAA;AAAA,IACxC;AACA,IAAA,GAAA,CAAI,IAAA,CAAK,SAAA,EAAW;AACnB,MAAA,KAAA,CAAM,IAAA,CAAK,GAAI,MAAM,IAAA,CAAK,SAAA,CAAU,QAAA,CAAS,CAAE,CAAA;AAAA,IAChD;AACA,IAAA,MAAM,UAAA,EAAY,aAAA,CAAc,uBAAA,CAAwB,KAAK,CAAA;AAE7D,IAAA,MAAM,YAAA,EAAwB,CAAC,CAAA;AAC/B,IAAA,IAAI,aAAA,EAAe,EAAA;AACnB,IAAA,IAAI,YAAA,EAAc,KAAA;AAElB,IAAA,MAAA,CAAO,IAAA,EAAM;AACZ,MAAA,MAAM,SAAA,EAAW,MAAM,aAAA,CAAc,uBAAA;AAAA,QACpC,QAAA;AAAA,QACA;AAAA,MACD,CAAA;AACA,MAAA,YAAA,EAAc,KAAA;AAEd,MAAA,yBAAA,CAAQ,MAAA,CAAO,KAAA,CAAM,UAAA,EAAY,EAAE,SAAS,CAAC,CAAA;AAE7C,MAAA,MAAM,EAAE,OAAA,EAAS,UAAU,EAAA,EAAI,QAAA;AAE/B,MAAA,yBAAA,CAAQ,MAAA,CAAO,KAAA,CAAM,SAAA,EAAW,EAAE,QAAQ,CAAC,CAAA;AAC3C,MAAA,yBAAA,CAAQ,MAAA,CAAO,KAAA,CAAM,YAAA,EAAc,EAAE,GAAG,UAAU,CAAC,CAAA;AAEnD,MAAA,GAAA,CAAI,SAAA,EAAW;AACd,QAAA,MAAM,eAAA,kBAAiB,IAAA,qBAAK,SAAA,6BAAW,iBAAA;AAAA,UACtC,KAAA;AAAA,UACA;AAAA,QACD,GAAA;AAEA,QAAA,IAAA,CAAA,MAAW,SAAA,GAAY,SAAA,EAAW;AACjC,UAAA,MAAM,SAAA,EAAW,QAAA,CAAS,IAAA;AAC1B,UAAA,YAAA,EAAc,IAAA;AACd,UAAA,MAAM,aAAA,EAAe,KAAA,CAAM,MAAA,CAAO,CAAC,IAAA,EAAA,GAAS,IAAA,CAAK,GAAA,IAAO,QAAQ,CAAA,CAAE,CAAC,CAAA;AAEnE,UAAA,IAAI,WAAA,EAAa,EAAA;AACjB,UAAA,GAAA,CACC,IAAA,CAAK,UAAA,GACL,YAAA,CAAa,SAAA,IAAA,eAAA,EACZ;AACD,YAAA,MAAM,SAAA,EAAW,QAAA,CAAS,SAAA;AAG1B,YAAA,yBAAA,CAAQ,MAAA,CAAO,KAAA,CAAM,eAAA,EAAiB,EAAE,QAAA,EAAU,SAAS,CAAC,CAAA;AAC5D,YAAA,MAAM,OAAA,EAAS,MAAM,IAAA,CAAK,SAAA,CAAU,OAAA;AAAA,cACnC,YAAA;AAAA,cACA;AAAA,YACD,CAAA;AACA,YAAA,WAAA,EACC,CAAA,qBAAA,EAAwB,QAAQ,CAAA,WAAA,EAAc,IAAA,CAAK,SAAA,CAAU,QAAQ,CAAC,CAAA;AAAA,EAAA,EACtE,IAAA,CAAK,SAAA,CAAU,MAAA,CAAO,OAAA,EAAS,IAAA,EAAM,CAAC,CAAA;AAAA,UACxC,EAAA,KAAA,GAAA,CACC,IAAA,CAAK,UAAA,GACL,YAAA,CAAa,SAAA,IAAA,eAAA,EACZ;AACD,YAAA,MAAM,OAAA,EAAS,MAAM,IAAA,CAAK,SAAA,CAAU,OAAA;AAAA,cACnC,YAAA;AAAA,cACA,cAAA;AAAA,cACA;AAAA,YACD,CAAA;AACA,YAAA,WAAA,EAAa,CAAA,qBAAA,EAAwB,QAAQ,CAAA;AAAA,EAAM,MAAA,CAAO,IAAA,CAAK,IAAI,CAAC,CAAA,CAAA;AAC9D,UAAA;AAEE,YAAA;AACP,cAAA;AACD,YAAA;AACA,YAAA;AACD,UAAA;AAEQ,UAAA;AAER,UAAA;AACA,UAAA;AACD,QAAA;AACU,MAAA;AACE,QAAA;AACZ,QAAA;AACD,MAAA;AAEK,MAAA;AACN,IAAA;AAEM,IAAA;AACI,MAAA;AACC,MAAA;AACX,IAAA;AAEO,IAAA;AACR,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAeyB,EAAA;AAElB,IAAA;AACA,IAAA;AACA,IAAA;AACL,MAAA;AACe,IAAA;AAGD,IAAA;AAGA,IAAA;AACd,MAAA;AACA,MAAA;AACA,MAAA;AACD,IAAA;AACe,IAAA;AACR,MAAA;AACL,QAAA;AACW,QAAA;AACA,QAAA;AACX,MAAA;AACK,MAAA;AACL,QAAA;AACW,QAAA;AACA,QAAA;AACX,MAAA;AACF,IAAA;AAES,IAAA;AACV,EAAA;AACD;AD/DoB;AACA;AACA;AACA","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-AHF32UIX.cjs","sourcesContent":[null,"import type {\n\tA2AModule,\n\tMCPModule,\n\tMemoryModule,\n\tModelModule,\n} from \"@/modules/index.js\";\nimport type { AinAgentPrompts } from \"@/types/index.js\";\nimport { ChatRole, type SessionObject } from \"@/types/memory.js\";\nimport {\n\ttype IA2ATool,\n\ttype IAgentTool,\n\ttype IMCPTool,\n\tTOOL_PROTOCOL_TYPE,\n} from \"@/types/tool.js\";\nimport { loggers } from \"@/utils/logger.js\";\n\n/**\n * Service for processing user queries through the agent's AI pipeline.\n *\n * Orchestrates the query processing workflow including intent detection,\n * model inference, tool execution, and response generation. Manages\n * conversation context and coordinates between different modules.\n */\nexport class QueryService {\n\tprivate modelModule: ModelModule;\n\tprivate a2aModule?: A2AModule;\n\tprivate mcpModule?: MCPModule;\n\tprivate memoryModule?: MemoryModule;\n\tprivate prompts?: AinAgentPrompts;\n\n\tconstructor(\n\t\tmodelModule: ModelModule,\n\t\ta2aModule?: A2AModule,\n\t\tmcpModule?: MCPModule,\n\t\tmemoryModule?: MemoryModule,\n\t\tprompts?: AinAgentPrompts,\n\t) {\n\t\tthis.modelModule = modelModule;\n\t\tthis.a2aModule = a2aModule;\n\t\tthis.mcpModule = mcpModule;\n\t\tthis.memoryModule = memoryModule;\n\t\tthis.prompts = prompts;\n\t}\n\n\t/**\n\t * Detects the intent from a user query.\n\t *\n\t * @param query - The user's input query\n\t * @returns The detected intent (currently returns the query as-is)\n\t * @todo Implement actual intent detection logic\n\t */\n\tprivate async intentTriggering(query: string) {\n\t\t/* TODO */\n\t\treturn query;\n\t}\n\n\t/**\n\t * Fulfills the detected intent by generating a response.\n\t *\n\t * Manages the complete inference loop including:\n\t * - Loading prompts and conversation history\n\t * - Collecting available tools from modules\n\t * - Executing model inference with tool support\n\t * - Processing tool calls iteratively until completion\n\t *\n\t * @param query - The user's input query\n\t * @param sessionId - Session identifier for context\n\t * @param sessionHistory - Previous conversation history\n\t * @returns Object containing process steps and final response\n\t */\n\tprivate async intentFulfilling(\n\t\tquery: string,\n\t\tsessionId: string,\n\t\tsessionHistory: SessionObject,\n\t) {\n\t\t// 1. Load agent / system prompt from memory\n\t\tconst systemPrompt = `\nToday is ${new Date().toLocaleDateString()}.\n\n${this.prompts?.agent || \"\"}\n\n${this.prompts?.system || \"\"}\n    `;\n\n\t\tconst modelInstance = this.modelModule.getModel();\n\t\tconst messages = modelInstance.generateMessages({\n\t\t\tquery,\n\t\t\tsessionHistory,\n\t\t\tsystemPrompt: systemPrompt.trim(),\n\t\t});\n\n\t\tconst tools: IAgentTool[] = [];\n\t\tif (this.mcpModule) {\n\t\t\ttools.push(...this.mcpModule.getTools());\n\t\t}\n\t\tif (this.a2aModule) {\n\t\t\ttools.push(...(await this.a2aModule.getTools()));\n\t\t}\n\t\tconst functions = modelInstance.convertToolsToFunctions(tools);\n\n\t\tconst processList: string[] = [];\n\t\tlet finalMessage = \"\";\n\t\tlet didCallTool = false;\n\n\t\twhile (true) {\n\t\t\tconst response = await modelInstance.fetchWithContextMessage(\n\t\t\t\tmessages,\n\t\t\t\tfunctions,\n\t\t\t);\n\t\t\tdidCallTool = false;\n\n\t\t\tloggers.intent.debug(\"messages\", { messages });\n\n\t\t\tconst { content, toolCalls } = response;\n\n\t\t\tloggers.intent.debug(\"content\", { content });\n\t\t\tloggers.intent.debug(\"tool_calls\", { ...toolCalls });\n\n\t\t\tif (toolCalls) {\n\t\t\t\tconst messagePayload = this.a2aModule?.getMessagePayload(\n\t\t\t\t\tquery,\n\t\t\t\t\tsessionId,\n\t\t\t\t);\n\n\t\t\t\tfor (const toolCall of toolCalls) {\n\t\t\t\t\tconst toolName = toolCall.name;\n\t\t\t\t\tdidCallTool = true;\n\t\t\t\t\tconst selectedTool = tools.filter((tool) => tool.id === toolName)[0];\n\n\t\t\t\t\tlet toolResult = \"\";\n\t\t\t\t\tif (\n\t\t\t\t\t\tthis.mcpModule &&\n\t\t\t\t\t\tselectedTool.protocol === TOOL_PROTOCOL_TYPE.MCP\n\t\t\t\t\t) {\n\t\t\t\t\t\tconst toolArgs = toolCall.arguments as\n\t\t\t\t\t\t\t| { [x: string]: unknown }\n\t\t\t\t\t\t\t| undefined;\n\t\t\t\t\t\tloggers.intent.debug(\"MCP tool call\", { toolName, toolArgs });\n\t\t\t\t\t\tconst result = 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\ttoolResult =\n\t\t\t\t\t\t\t`[Bot Called MCP Tool ${toolName} with args ${JSON.stringify(toolArgs)}]\\n` +\n\t\t\t\t\t\t\tJSON.stringify(result.content, null, 2);\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\tconst result = await this.a2aModule.useTool(\n\t\t\t\t\t\t\tselectedTool as IA2ATool,\n\t\t\t\t\t\t\tmessagePayload!,\n\t\t\t\t\t\t\tsessionId,\n\t\t\t\t\t\t);\n\t\t\t\t\t\ttoolResult = `[Bot Called A2A Tool ${toolName}]\\n${result.join(\"\\n\")}`;\n\t\t\t\t\t} else {\n\t\t\t\t\t\t// Unrecognized tool type. It cannot be happened...\n\t\t\t\t\t\tloggers.intent.warn(\n\t\t\t\t\t\t\t`Unrecognized tool type: ${selectedTool.protocol}`,\n\t\t\t\t\t\t);\n\t\t\t\t\t\tcontinue;\n\t\t\t\t\t}\n\n\t\t\t\t\tloggers.intent.debug(\"toolResult\", { toolResult });\n\n\t\t\t\t\tprocessList.push(toolResult);\n\t\t\t\t\tmodelInstance.appendMessages(messages, toolResult);\n\t\t\t\t}\n\t\t\t} else if (content) {\n\t\t\t\tprocessList.push(content);\n\t\t\t\tfinalMessage = content;\n\t\t\t}\n\n\t\t\tif (!didCallTool) break;\n\t\t}\n\n\t\tconst botResponse = {\n\t\t\tprocess: processList.join(\"\\n\"),\n\t\t\tresponse: finalMessage,\n\t\t};\n\n\t\treturn botResponse;\n\t}\n\n\t/**\n\t * Main entry point for processing user queries.\n\t *\n\t * Handles the complete query lifecycle:\n\t * 1. Loads session history from memory\n\t * 2. Detects intent from the query\n\t * 3. Fulfills the intent with AI response\n\t * 4. Updates conversation history\n\t *\n\t * @param query - The user's input query\n\t * @param sessionId - Unique session identifier\n\t * @returns Object containing the AI-generated response\n\t */\n\tpublic async handleQuery(query: string, sessionId: string) {\n\t\t// 1. Load session history with sessionId\n\t\tconst queryStartAt = Date.now();\n\t\tconst memoryInstance = this.memoryModule?.getMemory();\n\t\tconst sessionHistory = (await memoryInstance?.getSessionHistory(\n\t\t\tsessionId,\n\t\t)) || { chats: {} } /* FIXME */;\n\n\t\t// 2. intent triggering\n\t\tconst intent = this.intentTriggering(query);\n\n\t\t// 3. intent fulfillment\n\t\tconst result = await this.intentFulfilling(\n\t\t\tquery,\n\t\t\tsessionId,\n\t\t\tsessionHistory,\n\t\t);\n\t\tif (sessionId) {\n\t\t\tawait memoryInstance?.updateSessionHistory(sessionId, {\n\t\t\t\trole: ChatRole.USER,\n\t\t\t\ttimestamp: queryStartAt,\n\t\t\t\tcontent: { type: \"text\", parts: [query] },\n\t\t\t});\n\t\t\tawait memoryInstance?.updateSessionHistory(sessionId, {\n\t\t\t\trole: ChatRole.MODEL,\n\t\t\t\ttimestamp: Date.now(),\n\t\t\t\tcontent: { type: \"text\", parts: [result.response] },\n\t\t\t});\n\t\t}\n\n\t\treturn { content: result.response };\n\t}\n}\n"]}