@ainetwork/adk 0.1.12 → 0.1.13
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +9 -4
- package/dist/cjs/chunk-36ECGO5C.cjs +269 -0
- package/dist/cjs/chunk-36ECGO5C.cjs.map +1 -0
- package/dist/cjs/chunk-3J3ORQOG.cjs +65 -0
- package/dist/cjs/chunk-3J3ORQOG.cjs.map +1 -0
- package/dist/cjs/chunk-3QNRWPMM.cjs +37 -0
- package/dist/cjs/chunk-3QNRWPMM.cjs.map +1 -0
- package/dist/cjs/chunk-4MCEFDVX.cjs +43 -0
- package/dist/cjs/chunk-4MCEFDVX.cjs.map +1 -0
- package/dist/cjs/chunk-55IME4BG.cjs +114 -0
- package/dist/cjs/chunk-55IME4BG.cjs.map +1 -0
- package/dist/cjs/chunk-5T53A5X4.cjs +12 -0
- package/dist/cjs/chunk-5T53A5X4.cjs.map +1 -0
- package/dist/cjs/chunk-65M7DC3L.cjs +172 -0
- package/dist/cjs/chunk-65M7DC3L.cjs.map +1 -0
- package/dist/cjs/chunk-6BKOR3ZH.cjs +111 -0
- package/dist/cjs/chunk-6BKOR3ZH.cjs.map +1 -0
- package/dist/cjs/chunk-6YVTJEMQ.cjs +176 -0
- package/dist/cjs/chunk-6YVTJEMQ.cjs.map +1 -0
- package/dist/cjs/chunk-7EGJAURF.cjs +205 -0
- package/dist/cjs/chunk-7EGJAURF.cjs.map +1 -0
- package/dist/cjs/chunk-7EOUPEHU.cjs +243 -0
- package/dist/cjs/chunk-7EOUPEHU.cjs.map +1 -0
- package/dist/cjs/chunk-7K2R2UKB.cjs +8 -0
- package/dist/cjs/chunk-7K2R2UKB.cjs.map +1 -0
- package/dist/cjs/chunk-7TEDNNSL.cjs +18 -0
- package/dist/cjs/chunk-7TEDNNSL.cjs.map +1 -0
- package/dist/cjs/chunk-7XDYJNTA.cjs +35 -0
- package/dist/cjs/chunk-7XDYJNTA.cjs.map +1 -0
- package/dist/cjs/chunk-BBCZU5VZ.cjs +17 -0
- package/dist/cjs/chunk-BBCZU5VZ.cjs.map +1 -0
- package/dist/cjs/chunk-BNY3RLYC.cjs +43 -0
- package/dist/cjs/chunk-BNY3RLYC.cjs.map +1 -0
- package/dist/cjs/chunk-BYD5RS6L.cjs +12 -0
- package/dist/cjs/chunk-BYD5RS6L.cjs.map +1 -0
- package/dist/cjs/chunk-EAK7MP6X.cjs +22 -0
- package/dist/cjs/chunk-EAK7MP6X.cjs.map +1 -0
- package/dist/cjs/chunk-H65RVLWK.cjs +267 -0
- package/dist/cjs/chunk-H65RVLWK.cjs.map +1 -0
- package/dist/cjs/chunk-IS2IVXXA.cjs +176 -0
- package/dist/cjs/chunk-IS2IVXXA.cjs.map +1 -0
- package/dist/{esm/chunk-3NEV3O2X.js → cjs/chunk-J4Z7ZJUJ.cjs} +57 -55
- package/dist/cjs/chunk-J4Z7ZJUJ.cjs.map +1 -0
- package/dist/cjs/chunk-J6PDRTNM.cjs +37 -0
- package/dist/cjs/chunk-J6PDRTNM.cjs.map +1 -0
- package/dist/cjs/chunk-JAPFZA2K.cjs +205 -0
- package/dist/cjs/chunk-JAPFZA2K.cjs.map +1 -0
- package/dist/cjs/chunk-K4P2MYZF.cjs +43 -0
- package/dist/cjs/chunk-K4P2MYZF.cjs.map +1 -0
- package/dist/cjs/chunk-KDNNRPBV.cjs +43 -0
- package/dist/cjs/chunk-KDNNRPBV.cjs.map +1 -0
- package/dist/cjs/chunk-KRLUWTLN.cjs +35 -0
- package/dist/cjs/chunk-KRLUWTLN.cjs.map +1 -0
- package/dist/cjs/chunk-M4MYRCSF.cjs +18 -0
- package/dist/cjs/chunk-M4MYRCSF.cjs.map +1 -0
- package/dist/cjs/chunk-MVN5XJGZ.cjs +66 -0
- package/dist/cjs/chunk-MVN5XJGZ.cjs.map +1 -0
- package/dist/cjs/chunk-O6XQRAWR.cjs +45 -0
- package/dist/cjs/chunk-O6XQRAWR.cjs.map +1 -0
- package/dist/cjs/chunk-PDF6B6A3.cjs +37 -0
- package/dist/cjs/chunk-PDF6B6A3.cjs.map +1 -0
- package/dist/cjs/chunk-PT2CVI52.cjs +162 -0
- package/dist/cjs/chunk-PT2CVI52.cjs.map +1 -0
- package/dist/cjs/chunk-QUSBMECP.cjs +37 -0
- package/dist/cjs/chunk-QUSBMECP.cjs.map +1 -0
- package/dist/cjs/chunk-SAXFX7U2.cjs +37 -0
- package/dist/cjs/chunk-SAXFX7U2.cjs.map +1 -0
- package/dist/cjs/chunk-SCOMNXDA.cjs +51 -0
- package/dist/cjs/chunk-SCOMNXDA.cjs.map +1 -0
- package/dist/cjs/chunk-SGW3CSPD.cjs +43 -0
- package/dist/cjs/chunk-SGW3CSPD.cjs.map +1 -0
- package/dist/cjs/chunk-SJHSHDJL.cjs +65 -0
- package/dist/cjs/chunk-SJHSHDJL.cjs.map +1 -0
- package/dist/cjs/chunk-UAWGHNZD.cjs +22 -0
- package/dist/cjs/chunk-UAWGHNZD.cjs.map +1 -0
- package/dist/cjs/chunk-UBEB55WC.cjs +43 -0
- package/dist/cjs/chunk-UBEB55WC.cjs.map +1 -0
- package/dist/cjs/chunk-UPNSAV5A.cjs +18 -0
- package/dist/cjs/chunk-UPNSAV5A.cjs.map +1 -0
- package/dist/cjs/chunk-UW43BI5Z.cjs +17 -0
- package/dist/cjs/chunk-UW43BI5Z.cjs.map +1 -0
- package/dist/cjs/chunk-WNDA2G4Q.cjs +43 -0
- package/dist/cjs/chunk-WNDA2G4Q.cjs.map +1 -0
- package/dist/cjs/chunk-X46GJS7X.cjs +17 -0
- package/dist/cjs/chunk-X46GJS7X.cjs.map +1 -0
- package/dist/cjs/chunk-XPD3ANDG.cjs +267 -0
- package/dist/cjs/chunk-XPD3ANDG.cjs.map +1 -0
- package/dist/cjs/chunk-XPTRG6IC.cjs +43 -0
- package/dist/cjs/chunk-XPTRG6IC.cjs.map +1 -0
- package/dist/cjs/chunk-XVMXTCJI.cjs +64 -0
- package/dist/cjs/chunk-XVMXTCJI.cjs.map +1 -0
- package/dist/cjs/chunk-YQ5P2K3X.cjs +37 -0
- package/dist/cjs/chunk-YQ5P2K3X.cjs.map +1 -0
- package/dist/cjs/controllers/api/thread.api.controller.cjs +8 -0
- package/dist/cjs/controllers/api/thread.api.controller.cjs.map +1 -0
- package/dist/cjs/controllers/index.cjs +3 -3
- package/dist/cjs/controllers/index.cjs.map +1 -1
- package/dist/cjs/controllers/query.controller.cjs +3 -2
- package/dist/cjs/controllers/query.controller.cjs.map +1 -1
- package/dist/cjs/index.cjs +17 -16
- package/dist/cjs/index.cjs.map +1 -1
- package/dist/cjs/modules/a2a/a2a.module.cjs +3 -2
- package/dist/cjs/modules/a2a/a2a.module.cjs.map +1 -1
- package/dist/cjs/modules/index.cjs +5 -4
- package/dist/cjs/modules/index.cjs.map +1 -1
- package/dist/cjs/modules/memory/memory.module.cjs +2 -2
- package/dist/cjs/modules/models/base.model.cjs +2 -2
- package/dist/cjs/routes/a2a.routes.cjs +5 -5
- package/dist/cjs/routes/api/model.routes.cjs +8 -0
- package/dist/cjs/routes/api/model.routes.cjs.map +1 -0
- package/dist/cjs/routes/api/thread.routes.cjs +9 -0
- package/dist/cjs/routes/api/thread.routes.cjs.map +1 -0
- package/dist/cjs/routes/api.routes.cjs +5 -5
- package/dist/cjs/routes/index.cjs +12 -12
- package/dist/cjs/routes/index.cjs.map +1 -1
- package/dist/cjs/routes/query.routes.cjs +7 -6
- package/dist/cjs/routes/query.routes.cjs.map +1 -1
- package/dist/cjs/services/a2a.service.cjs +2 -2
- package/dist/cjs/services/index.cjs +5 -5
- package/dist/cjs/services/query-stream.service.cjs +3 -3
- package/dist/cjs/services/query.service.cjs +3 -3
- package/dist/cjs/types/memory.cjs +2 -2
- package/dist/cjs/types/memory.cjs.map +1 -1
- package/dist/esm/chunk-2BYUKYVB.js +243 -0
- package/dist/esm/chunk-2BYUKYVB.js.map +1 -0
- package/dist/esm/{chunk-Y6E73I7I.js → chunk-2RW77MJB.js} +6 -3
- package/dist/esm/chunk-2RW77MJB.js.map +1 -0
- package/dist/esm/chunk-7D4SKPIS.js +65 -0
- package/dist/esm/chunk-7D4SKPIS.js.map +1 -0
- package/dist/esm/{chunk-5EPK37PE.js → chunk-ADIAIRNH.js} +10 -10
- package/dist/esm/chunk-ADIAIRNH.js.map +1 -0
- package/dist/esm/{chunk-2Z7GJVJJ.js → chunk-AMRKK46V.js} +2 -2
- package/dist/esm/chunk-AMRKK46V.js.map +1 -0
- package/dist/esm/{chunk-2BITOA2R.js → chunk-CDFLKADH.js} +1 -1
- package/dist/esm/chunk-CDFLKADH.js.map +1 -0
- package/dist/esm/{chunk-DRUCJYPJ.js → chunk-EGR6OTTN.js} +65 -22
- package/dist/esm/chunk-EGR6OTTN.js.map +1 -0
- package/dist/esm/{chunk-5N4X25IK.js → chunk-G6VR5RSV.js} +5 -5
- package/dist/esm/chunk-HOFAK5J2.js +22 -0
- package/dist/esm/chunk-HOFAK5J2.js.map +1 -0
- package/dist/esm/{chunk-GNWF6QG4.js → chunk-JCWH7LIC.js} +6 -6
- package/dist/esm/{chunk-C6SGV7JV.js → chunk-LJVVSUF7.js} +17 -13
- package/dist/esm/chunk-LJVVSUF7.js.map +1 -0
- package/dist/esm/chunk-MNAWSBYP.js +18 -0
- package/dist/esm/chunk-MNAWSBYP.js.map +1 -0
- package/dist/esm/chunk-VA3XOZBV.js +17 -0
- package/dist/esm/chunk-VA3XOZBV.js.map +1 -0
- package/dist/esm/{chunk-CKTURH34.js → chunk-XBMWWTSX.js} +9 -9
- package/dist/esm/chunk-XBMWWTSX.js.map +1 -0
- package/dist/esm/controllers/a2a.controller.d.ts +1 -1
- package/dist/esm/controllers/api/{session.api.controller.d.ts → thread.api.controller.d.ts} +3 -3
- package/dist/esm/controllers/api/thread.api.controller.js +8 -0
- package/dist/esm/controllers/index.d.ts +2 -2
- package/dist/esm/controllers/index.js +7 -7
- package/dist/esm/controllers/query.controller.d.ts +1 -1
- package/dist/esm/controllers/query.controller.js +2 -1
- package/dist/esm/index.d.ts +1 -0
- package/dist/esm/index.js +17 -16
- package/dist/esm/index.js.map +1 -1
- package/dist/esm/modules/a2a/a2a.module.d.ts +6 -15
- package/dist/esm/modules/a2a/a2a.module.js +2 -1
- package/dist/esm/modules/index.d.ts +1 -1
- package/dist/esm/modules/index.js +8 -7
- package/dist/esm/modules/memory/base.memory.d.ts +9 -9
- package/dist/esm/modules/memory/memory.module.d.ts +4 -4
- package/dist/esm/modules/memory/memory.module.js +1 -1
- package/dist/esm/modules/models/base.model.d.ts +4 -4
- package/dist/esm/modules/models/base.model.js +1 -1
- package/dist/esm/routes/a2a.routes.js +4 -4
- package/dist/esm/routes/api/{models.routes.js → model.routes.js} +2 -2
- package/dist/esm/routes/api/{sessions.routes.d.ts → thread.routes.d.ts} +2 -2
- package/dist/esm/routes/api/thread.routes.js +9 -0
- package/dist/esm/routes/api.routes.js +4 -4
- package/dist/esm/routes/index.d.ts +2 -2
- package/dist/esm/routes/index.js +16 -16
- package/dist/esm/routes/query.routes.js +6 -5
- package/dist/esm/services/a2a.service.d.ts +1 -1
- package/dist/esm/services/a2a.service.js +1 -1
- package/dist/esm/services/index.d.ts +1 -2
- package/dist/esm/services/index.js +6 -6
- package/dist/esm/services/query-stream.service.d.ts +13 -8
- package/dist/esm/services/query-stream.service.js +2 -2
- package/dist/esm/services/query.service.d.ts +19 -8
- package/dist/esm/services/query.service.js +2 -2
- package/dist/esm/types/memory.d.ts +32 -32
- package/dist/esm/types/memory.js +5 -5
- package/dist/esm/types/stream.d.ts +32 -2
- package/package.json +1 -1
- package/dist/esm/chunk-2BITOA2R.js.map +0 -1
- package/dist/esm/chunk-2Z7GJVJJ.js.map +0 -1
- package/dist/esm/chunk-3NEV3O2X.js.map +0 -1
- package/dist/esm/chunk-5EPK37PE.js.map +0 -1
- package/dist/esm/chunk-C6SGV7JV.js.map +0 -1
- package/dist/esm/chunk-CKTURH34.js.map +0 -1
- package/dist/esm/chunk-DRUCJYPJ.js.map +0 -1
- package/dist/esm/chunk-Q6HHVAO6.js +0 -45
- package/dist/esm/chunk-Q6HHVAO6.js.map +0 -1
- package/dist/esm/chunk-QROUPWWO.js +0 -17
- package/dist/esm/chunk-QROUPWWO.js.map +0 -1
- package/dist/esm/chunk-VAZWF5IK.js +0 -20
- package/dist/esm/chunk-VAZWF5IK.js.map +0 -1
- package/dist/esm/chunk-Y6E73I7I.js.map +0 -1
- package/dist/esm/chunk-YT6PD6UF.js +0 -22
- package/dist/esm/chunk-YT6PD6UF.js.map +0 -1
- package/dist/esm/controllers/api/session.api.controller.js +0 -8
- package/dist/esm/routes/api/sessions.routes.js +0 -9
- /package/dist/esm/{chunk-5N4X25IK.js.map → chunk-G6VR5RSV.js.map} +0 -0
- /package/dist/esm/{chunk-GNWF6QG4.js.map → chunk-JCWH7LIC.js.map} +0 -0
- /package/dist/esm/controllers/api/{session.api.controller.js.map → thread.api.controller.js.map} +0 -0
- /package/dist/esm/routes/api/{models.routes.d.ts → model.routes.d.ts} +0 -0
- /package/dist/esm/routes/api/{models.routes.js.map → model.routes.js.map} +0 -0
- /package/dist/esm/routes/api/{sessions.routes.js.map → thread.routes.js.map} +0 -0
|
@@ -0,0 +1,43 @@
|
|
|
1
|
+
"use strict";Object.defineProperty(exports, "__esModule", {value: true});
|
|
2
|
+
|
|
3
|
+
var _chunkXPD3ANDGcjs = require('./chunk-XPD3ANDG.cjs');
|
|
4
|
+
|
|
5
|
+
|
|
6
|
+
var _chunkSJHSHDJLcjs = require('./chunk-SJHSHDJL.cjs');
|
|
7
|
+
|
|
8
|
+
|
|
9
|
+
var _chunkJAPFZA2Kcjs = require('./chunk-JAPFZA2K.cjs');
|
|
10
|
+
|
|
11
|
+
// src/routes/query.routes.ts
|
|
12
|
+
var _express = require('express');
|
|
13
|
+
var createQueryRouter = (agent, allowStream = false) => {
|
|
14
|
+
const router = _express.Router.call(void 0, );
|
|
15
|
+
const queryService = new (0, _chunkJAPFZA2Kcjs.QueryService)(
|
|
16
|
+
agent.modelModule,
|
|
17
|
+
agent.a2aModule,
|
|
18
|
+
agent.mcpModule,
|
|
19
|
+
agent.memoryModule,
|
|
20
|
+
agent.manifest.prompts
|
|
21
|
+
);
|
|
22
|
+
let queryStreamService;
|
|
23
|
+
if (allowStream) {
|
|
24
|
+
queryStreamService = new (0, _chunkXPD3ANDGcjs.QueryStreamService)(
|
|
25
|
+
agent.modelModule,
|
|
26
|
+
agent.a2aModule,
|
|
27
|
+
agent.mcpModule,
|
|
28
|
+
agent.memoryModule,
|
|
29
|
+
agent.manifest.prompts
|
|
30
|
+
);
|
|
31
|
+
}
|
|
32
|
+
const queryController = new (0, _chunkSJHSHDJLcjs.QueryController)(queryService, queryStreamService);
|
|
33
|
+
router.post("/", queryController.handleQueryRequest);
|
|
34
|
+
if (allowStream) {
|
|
35
|
+
router.post("/stream", queryController.handleQueryStreamRequest);
|
|
36
|
+
}
|
|
37
|
+
return router;
|
|
38
|
+
};
|
|
39
|
+
|
|
40
|
+
|
|
41
|
+
|
|
42
|
+
exports.createQueryRouter = createQueryRouter;
|
|
43
|
+
//# sourceMappingURL=chunk-KDNNRPBV.cjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-KDNNRPBV.cjs","../../src/routes/query.routes.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACA;ACVA,kCAAuB;AAMhB,IAAM,kBAAA,EAAoB,CAChC,KAAA,EACA,YAAA,EAAc,KAAA,EAAA,GACF;AACZ,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;AAEA,EAAA,IAAI,kBAAA;AAEJ,EAAA,GAAA,CAAI,WAAA,EAAa;AAChB,IAAA,mBAAA,EAAqB,IAAI,yCAAA;AAAA,MACxB,KAAA,CAAM,WAAA;AAAA,MACN,KAAA,CAAM,SAAA;AAAA,MACN,KAAA,CAAM,SAAA;AAAA,MACN,KAAA,CAAM,YAAA;AAAA,MACN,KAAA,CAAM,QAAA,CAAS;AAAA,IAChB,CAAA;AAAA,EACD;AAEA,EAAA,MAAM,gBAAA,EAAkB,IAAI,sCAAA,CAAgB,YAAA,EAAc,kBAAkB,CAAA;AAC5E,EAAA,MAAA,CAAO,IAAA,CAAK,GAAA,EAAK,eAAA,CAAgB,kBAAkB,CAAA;AAEnD,EAAA,GAAA,CAAI,WAAA,EAAa;AAChB,IAAA,MAAA,CAAO,IAAA,CAAK,SAAA,EAAW,eAAA,CAAgB,wBAAwB,CAAA;AAAA,EAChE;AAEA,EAAA,OAAO,MAAA;AACR,CAAA;ADFA;AACA;AACE;AACF,8CAAC","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-KDNNRPBV.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\";\nimport { QueryStreamService } from \"@/services/query-stream.service.js\";\n\nexport const createQueryRouter = (\n\tagent: AINAgent,\n\tallowStream = false,\n): 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\n\tlet queryStreamService: QueryStreamService | undefined;\n\n\tif (allowStream) {\n\t\tqueryStreamService = new QueryStreamService(\n\t\t\tagent.modelModule,\n\t\t\tagent.a2aModule,\n\t\t\tagent.mcpModule,\n\t\t\tagent.memoryModule,\n\t\t\tagent.manifest.prompts,\n\t\t);\n\t}\n\n\tconst queryController = new QueryController(queryService, queryStreamService);\n\trouter.post(\"/\", queryController.handleQueryRequest);\n\n\tif (allowStream) {\n\t\trouter.post(\"/stream\", queryController.handleQueryStreamRequest);\n\t}\n\n\treturn router;\n};\n"]}
|
|
@@ -0,0 +1,35 @@
|
|
|
1
|
+
"use strict";Object.defineProperty(exports, "__esModule", {value: true}); var _class;
|
|
2
|
+
|
|
3
|
+
var _chunkDX2JH3FJcjs = require('./chunk-DX2JH3FJ.cjs');
|
|
4
|
+
|
|
5
|
+
// src/controllers/api/thread.api.controller.ts
|
|
6
|
+
var _httpstatuscodes = require('http-status-codes');
|
|
7
|
+
var ThreadApiController = (_class = class {
|
|
8
|
+
|
|
9
|
+
constructor(memoryModule) {;_class.prototype.__init.call(this);
|
|
10
|
+
this.memoryModule = memoryModule;
|
|
11
|
+
}
|
|
12
|
+
__init() {this.handleGetThread = async (req, res, next) => {
|
|
13
|
+
try {
|
|
14
|
+
const { id: threadId, type } = req.params;
|
|
15
|
+
const userId = res.locals.userId || "";
|
|
16
|
+
const threadMemory = this.memoryModule.getThreadMemory();
|
|
17
|
+
if (!threadMemory) {
|
|
18
|
+
const error = new (0, _chunkDX2JH3FJcjs.AinHttpError)(
|
|
19
|
+
_httpstatuscodes.StatusCodes.SERVICE_UNAVAILABLE,
|
|
20
|
+
"Memory module is not initialized"
|
|
21
|
+
);
|
|
22
|
+
throw error;
|
|
23
|
+
}
|
|
24
|
+
const thread = await threadMemory.getThread(type, userId, threadId);
|
|
25
|
+
res.json(thread);
|
|
26
|
+
} catch (error) {
|
|
27
|
+
next(error);
|
|
28
|
+
}
|
|
29
|
+
}}
|
|
30
|
+
}, _class);
|
|
31
|
+
|
|
32
|
+
|
|
33
|
+
|
|
34
|
+
exports.ThreadApiController = ThreadApiController;
|
|
35
|
+
//# sourceMappingURL=chunk-KRLUWTLN.cjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-KRLUWTLN.cjs","../../src/controllers/api/thread.api.controller.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACA;ACHA,oDAA4B;AAKrB,IAAM,oBAAA,YAAN,MAA0B;AAAA,EACxB;AAAA,EAER,WAAA,CAAY,YAAA,EAA4B;AACvC,IAAA,IAAA,CAAK,aAAA,EAAe,YAAA;AAAA,EACrB;AAAA,iBAEO,gBAAA,EAAkB,MAAA,CACxB,GAAA,EACA,GAAA,EACA,IAAA,EAAA,GACI;AACJ,IAAA,IAAI;AACH,MAAA,MAAM,EAAE,EAAA,EAAI,QAAA,EAAU,KAAK,EAAA,EAAI,GAAA,CAAI,MAAA;AAInC,MAAA,MAAM,OAAA,EAAS,GAAA,CAAI,MAAA,CAAO,OAAA,GAAU,EAAA;AACpC,MAAA,MAAM,aAAA,EAAe,IAAA,CAAK,YAAA,CAAa,eAAA,CAAgB,CAAA;AACvD,MAAA,GAAA,CAAI,CAAC,YAAA,EAAc;AAClB,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,OAAA,EAAS,MAAM,YAAA,CAAa,SAAA,CAAU,IAAA,EAAM,MAAA,EAAQ,QAAQ,CAAA;AAClE,MAAA,GAAA,CAAI,IAAA,CAAK,MAAM,CAAA;AAAA,IAChB,EAAA,MAAA,CAAS,KAAA,EAAO;AACf,MAAA,IAAA,CAAK,KAAK,CAAA;AAAA,IACX;AAAA,EACD,EAAA;AACD,UAAA;ADRA;AACA;AACE;AACF,kDAAC","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-KRLUWTLN.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/agent.js\";\nimport type { ThreadType } from \"@/types/memory\";\n\nexport class ThreadApiController {\n\tprivate memoryModule: MemoryModule;\n\n\tconstructor(memoryModule: MemoryModule) {\n\t\tthis.memoryModule = memoryModule;\n\t}\n\n\tpublic handleGetThread = async (\n\t\treq: Request,\n\t\tres: Response,\n\t\tnext: NextFunction,\n\t) => {\n\t\ttry {\n\t\t\tconst { id: threadId, type } = req.params as {\n\t\t\t\tid: string;\n\t\t\t\ttype: ThreadType;\n\t\t\t};\n\t\t\tconst userId = res.locals.userId || \"\";\n\t\t\tconst threadMemory = this.memoryModule.getThreadMemory();\n\t\t\tif (!threadMemory) {\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 thread = await threadMemory.getThread(type, userId, threadId);\n\t\t\tres.json(thread);\n\t\t} catch (error) {\n\t\t\tnext(error);\n\t\t}\n\t};\n}\n"]}
|
|
@@ -0,0 +1,18 @@
|
|
|
1
|
+
"use strict";Object.defineProperty(exports, "__esModule", {value: true});// src/types/memory.ts
|
|
2
|
+
var MessageRole = /* @__PURE__ */ ((MessageRole2) => {
|
|
3
|
+
MessageRole2["USER"] = "USER";
|
|
4
|
+
MessageRole2["SYSTEM"] = "SYSTEM";
|
|
5
|
+
MessageRole2["MODEL"] = "MODEL";
|
|
6
|
+
return MessageRole2;
|
|
7
|
+
})(MessageRole || {});
|
|
8
|
+
var ThreadType = /* @__PURE__ */ ((ThreadType2) => {
|
|
9
|
+
ThreadType2["WORKFLOW"] = "WORKFLOW";
|
|
10
|
+
ThreadType2["CHAT"] = "CHAT";
|
|
11
|
+
return ThreadType2;
|
|
12
|
+
})(ThreadType || {});
|
|
13
|
+
|
|
14
|
+
|
|
15
|
+
|
|
16
|
+
|
|
17
|
+
exports.MessageRole = MessageRole; exports.ThreadType = ThreadType;
|
|
18
|
+
//# sourceMappingURL=chunk-M4MYRCSF.cjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-M4MYRCSF.cjs","../../src/types/memory.ts"],"names":["MessageRole","ThreadType"],"mappings":"AAAA;ACGO,IAAK,YAAA,kBAAL,CAAA,CAAKA,YAAAA,EAAAA,GAAL;AAEN,EAAAA,YAAAA,CAAA,MAAA,EAAA,EAAO,MAAA;AAEP,EAAAA,YAAAA,CAAA,QAAA,EAAA,EAAS,QAAA;AAET,EAAAA,YAAAA,CAAA,OAAA,EAAA,EAAQ,OAAA;AANG,EAAA,OAAAA,YAAAA;AAAA,CAAA,CAAA,CAAA,YAAA,GAAA,CAAA,CAAA,CAAA;AAgDL,IAAK,WAAA,kBAAL,CAAA,CAAKC,WAAAA,EAAAA,GAAL;AACN,EAAAA,WAAAA,CAAA,UAAA,EAAA,EAAW,UAAA;AACX,EAAAA,WAAAA,CAAA,MAAA,EAAA,EAAO,MAAA;AAFI,EAAA,OAAAA,WAAAA;AAAA,CAAA,CAAA,CAAA,WAAA,GAAA,CAAA,CAAA,CAAA;ADvCZ;AACA;AACE;AACA;AACF,mEAAC","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-M4MYRCSF.cjs","sourcesContent":[null,"/**\n * Roles for participants in a message.\n */\nexport enum MessageRole {\n\t/** User/human participant */\n\tUSER = \"USER\",\n\t/** System-generated messages or instructions */\n\tSYSTEM = \"SYSTEM\",\n\t/** AI model responses */\n\tMODEL = \"MODEL\",\n}\n\n/**\n * Content structure for message content.\n *\n * Supports multi-part content with different types (text, images, etc.).\n */\nexport type MessageContentObject = {\n\t/** Content type (e.g., \"text\", \"image\", \"tool_use\") */\n\ttype: string;\n\t/** Array of content parts, structure depends on content type */\n\tparts: any[];\n};\n\n/**\n * Represents a single message in a thread.\n *\n * @example\n * ```typescript\n * const message: MessageObject = {\n * role: MessageRole.USER,\n * content: {\n * type: \"text\",\n * parts: [\"Hello, how can you help me?\"]\n * },\n * timestamp: Date.now(),\n * metadata: { source: \"web-ui\" }\n * };\n * ```\n */\nexport type MessageObject = {\n\t/** Role of the message sender */\n\trole: MessageRole;\n\t/** Message content with type and parts */\n\tcontent: MessageContentObject;\n\t/** Unix timestamp when the message was created */\n\ttimestamp: number;\n\t/** Optional metadata for additional context */\n\tmetadata?: { [key: string]: unknown };\n};\n\nexport enum ThreadType {\n\tWORKFLOW = \"WORKFLOW\",\n\tCHAT = \"CHAT\",\n}\n\nexport type ThreadMetadata = {\n\ttype: ThreadType;\n\ttitle: string;\n\tthreadId: string;\n\tupdatedAt: number;\n};\n\n/**\n * Represents a conversation thread containing multiple messages.\n *\n * Messages are stored in a key-value structure where keys are unique message IDs\n * and values are the corresponding message objects.\n *\n * @example\n * ```typescript\n * const thread: ThreadObject = {\n * \t title: \"New conversation\",\n * messages: {\n * \"<UUID_1>\": { role: MessageRole.USER, content: {...}, timestamp: 1234567890 },\n * \"<UUID_2>\": { role: MessageRole.MODEL, content: {...}, timestamp: 1234567891 }\n * }\n * };\n * ```\n */\nexport type ThreadObject = {\n\ttype: ThreadType;\n\ttitle: string;\n\t/* Collection of messages indexed by unique message ID */\n\tmessages: {\n\t\t[messageId: string]: MessageObject;\n\t};\n};\n\nexport interface Intent {\n\tname: string;\n\tdescription: string;\n\tprompt?: string;\n\tllm?: string;\n}\n"]}
|
|
@@ -0,0 +1,66 @@
|
|
|
1
|
+
"use strict";Object.defineProperty(exports, "__esModule", {value: true}); var _class;
|
|
2
|
+
|
|
3
|
+
var _chunkDX2JH3FJcjs = require('./chunk-DX2JH3FJ.cjs');
|
|
4
|
+
|
|
5
|
+
// src/controllers/query.controller.ts
|
|
6
|
+
var _httpstatuscodes = require('http-status-codes');
|
|
7
|
+
var QueryController = (_class = class {
|
|
8
|
+
|
|
9
|
+
|
|
10
|
+
constructor(queryService, queryStreamService) {;_class.prototype.__init.call(this);_class.prototype.__init2.call(this);
|
|
11
|
+
this.queryService = queryService;
|
|
12
|
+
this.queryStreamService = queryStreamService;
|
|
13
|
+
}
|
|
14
|
+
__init() {this.handleQueryRequest = async (req, res, next) => {
|
|
15
|
+
const { type, message, threadId } = req.body;
|
|
16
|
+
const userId = res.locals.userId;
|
|
17
|
+
try {
|
|
18
|
+
const result = await this.queryService.handleQuery(
|
|
19
|
+
{ type, userId, threadId },
|
|
20
|
+
message
|
|
21
|
+
);
|
|
22
|
+
res.status(200).json(result);
|
|
23
|
+
} catch (error) {
|
|
24
|
+
next(error);
|
|
25
|
+
}
|
|
26
|
+
}}
|
|
27
|
+
__init2() {this.handleQueryStreamRequest = async (req, res, next) => {
|
|
28
|
+
const { type, threadId, message } = req.body;
|
|
29
|
+
const userId = res.locals.userId;
|
|
30
|
+
console.log(type, threadId, message);
|
|
31
|
+
if (!this.queryStreamService) {
|
|
32
|
+
const error = new (0, _chunkDX2JH3FJcjs.AinHttpError)(
|
|
33
|
+
_httpstatuscodes.StatusCodes.NOT_IMPLEMENTED,
|
|
34
|
+
"Stream query not supported"
|
|
35
|
+
);
|
|
36
|
+
return next(error);
|
|
37
|
+
}
|
|
38
|
+
const stream = this.queryStreamService.handleQueryStream(
|
|
39
|
+
{ type, userId, threadId },
|
|
40
|
+
message
|
|
41
|
+
);
|
|
42
|
+
try {
|
|
43
|
+
for await (const event of stream) {
|
|
44
|
+
console.log("Event:", event);
|
|
45
|
+
res.write(
|
|
46
|
+
`event: ${event.event}
|
|
47
|
+
data: ${JSON.stringify(event.data)}
|
|
48
|
+
|
|
49
|
+
`
|
|
50
|
+
);
|
|
51
|
+
}
|
|
52
|
+
} catch (error) {
|
|
53
|
+
res.write(`event: error
|
|
54
|
+
data: ${JSON.stringify(error)}
|
|
55
|
+
|
|
56
|
+
`);
|
|
57
|
+
} finally {
|
|
58
|
+
res.end();
|
|
59
|
+
}
|
|
60
|
+
}}
|
|
61
|
+
}, _class);
|
|
62
|
+
|
|
63
|
+
|
|
64
|
+
|
|
65
|
+
exports.QueryController = QueryController;
|
|
66
|
+
//# sourceMappingURL=chunk-MVN5XJGZ.cjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-MVN5XJGZ.cjs","../../src/controllers/query.controller.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACA;ACHA,oDAA4B;AAIrB,IAAM,gBAAA,YAAN,MAAsB;AAAA,EACpB;AAAA,EACA;AAAA,EAER,WAAA,CACC,YAAA,EACA,kBAAA,EACC;AACD,IAAA,IAAA,CAAK,aAAA,EAAe,YAAA;AACpB,IAAA,IAAA,CAAK,mBAAA,EAAqB,kBAAA;AAAA,EAC3B;AAAA,iBAEO,mBAAA,EAAqB,MAAA,CAC3B,GAAA,EACA,GAAA,EACA,IAAA,EAAA,GACI;AACJ,IAAA,MAAM,EAAE,IAAA,EAAM,OAAA,EAAS,SAAS,EAAA,EAAI,GAAA,CAAI,IAAA;AACxC,IAAA,MAAM,OAAA,EAAS,GAAA,CAAI,MAAA,CAAO,MAAA;AAE1B,IAAA,IAAI;AACH,MAAA,MAAM,OAAA,EAAS,MAAM,IAAA,CAAK,YAAA,CAAa,WAAA;AAAA,QACtC,EAAE,IAAA,EAAM,MAAA,EAAQ,SAAS,CAAA;AAAA,QACzB;AAAA,MACD,CAAA;AAEA,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;AAAA,kBAEO,yBAAA,EAA2B,MAAA,CACjC,GAAA,EACA,GAAA,EACA,IAAA,EAAA,GACI;AACJ,IAAA,MAAM,EAAE,IAAA,EAAM,QAAA,EAAU,QAAQ,EAAA,EAAI,GAAA,CAAI,IAAA;AACxC,IAAA,MAAM,OAAA,EAAS,GAAA,CAAI,MAAA,CAAO,MAAA;AAE1B,IAAA,OAAA,CAAQ,GAAA,CAAI,IAAA,EAAM,QAAA,EAAU,OAAO,CAAA;AAEnC,IAAA,GAAA,CAAI,CAAC,IAAA,CAAK,kBAAA,EAAoB;AAC7B,MAAA,MAAM,MAAA,EAAQ,IAAI,mCAAA;AAAA,QACjB,4BAAA,CAAY,eAAA;AAAA,QACZ;AAAA,MACD,CAAA;AACA,MAAA,OAAO,IAAA,CAAK,KAAK,CAAA;AAAA,IAClB;AAEA,IAAA,MAAM,OAAA,EAAS,IAAA,CAAK,kBAAA,CAAmB,iBAAA;AAAA,MACtC,EAAE,IAAA,EAAM,MAAA,EAAQ,SAAS,CAAA;AAAA,MACzB;AAAA,IACD,CAAA;AAEA,IAAA,IAAI;AACH,MAAA,IAAA,MAAA,CAAA,MAAiB,MAAA,GAAS,MAAA,EAAQ;AACjC,QAAA,OAAA,CAAQ,GAAA,CAAI,QAAA,EAAU,KAAK,CAAA;AAC3B,QAAA,GAAA,CAAI,KAAA;AAAA,UACH,CAAA,OAAA,EAAU,KAAA,CAAM,KAAK,CAAA;AAAA,MAAA,EAAW,IAAA,CAAK,SAAA,CAAU,KAAA,CAAM,IAAI,CAAC,CAAA;AAAA;AAAA;AAAA,QAC3D,CAAA;AAAA,MACD;AAAA,IACD,EAAA,MAAA,CAAS,KAAA,EAAO;AACf,MAAA,GAAA,CAAI,KAAA,CAAM,CAAA;AAAA,MAAA,EAAuB,IAAA,CAAK,SAAA,CAAU,KAAK,CAAC,CAAA;AAAA;AAAA,CAAM,CAAA;AAAA,IAC7D,EAAA,QAAE;AACD,MAAA,GAAA,CAAI,GAAA,CAAI,CAAA;AAAA,IACT;AAAA,EACD,EAAA;AACD,UAAA;ADZA;AACA;AACE;AACF,0CAAC","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-MVN5XJGZ.cjs","sourcesContent":[null,"import type { NextFunction, Request, Response } from \"express\";\nimport { StatusCodes } from \"http-status-codes\";\nimport type { QueryService, QueryStreamService } from \"@/services\";\nimport { AinHttpError } from \"@/types/agent\";\n\nexport class QueryController {\n\tprivate queryService;\n\tprivate queryStreamService;\n\n\tconstructor(\n\t\tqueryService: QueryService,\n\t\tqueryStreamService?: QueryStreamService,\n\t) {\n\t\tthis.queryService = queryService;\n\t\tthis.queryStreamService = queryStreamService;\n\t}\n\n\tpublic handleQueryRequest = async (\n\t\treq: Request,\n\t\tres: Response,\n\t\tnext: NextFunction,\n\t) => {\n\t\tconst { type, message, threadId } = req.body;\n\t\tconst userId = res.locals.userId;\n\n\t\ttry {\n\t\t\tconst result = await this.queryService.handleQuery(\n\t\t\t\t{ type, userId, threadId },\n\t\t\t\tmessage,\n\t\t\t);\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\tpublic handleQueryStreamRequest = async (\n\t\treq: Request,\n\t\tres: Response,\n\t\tnext: NextFunction,\n\t) => {\n\t\tconst { type, threadId, message } = req.body;\n\t\tconst userId = res.locals.userId;\n\n\t\tconsole.log(type, threadId, message);\n\n\t\tif (!this.queryStreamService) {\n\t\t\tconst error = new AinHttpError(\n\t\t\t\tStatusCodes.NOT_IMPLEMENTED,\n\t\t\t\t\"Stream query not supported\",\n\t\t\t);\n\t\t\treturn next(error);\n\t\t}\n\n\t\tconst stream = this.queryStreamService.handleQueryStream(\n\t\t\t{ type, userId, threadId },\n\t\t\tmessage,\n\t\t);\n\n\t\ttry {\n\t\t\tfor await (const event of stream) {\n\t\t\t\tconsole.log(\"Event:\", event);\n\t\t\t\tres.write(\n\t\t\t\t\t`event: ${event.event}\\ndata: ${JSON.stringify(event.data)}\\n\\n`,\n\t\t\t\t);\n\t\t\t}\n\t\t} catch (error) {\n\t\t\tres.write(`event: error\\ndata: ${JSON.stringify(error)}\\n\\n`);\n\t\t} finally {\n\t\t\tres.end();\n\t\t}\n\t};\n}\n"]}
|
|
@@ -0,0 +1,45 @@
|
|
|
1
|
+
"use strict";Object.defineProperty(exports, "__esModule", {value: true}); var _class;// src/controllers/query.controller.ts
|
|
2
|
+
var QueryController = (_class = class {
|
|
3
|
+
|
|
4
|
+
|
|
5
|
+
constructor(queryService, queryStreamService) {;_class.prototype.__init.call(this);_class.prototype.__init2.call(this);
|
|
6
|
+
this.queryService = queryService;
|
|
7
|
+
this.queryStreamService = queryStreamService;
|
|
8
|
+
}
|
|
9
|
+
__init() {this.handleQueryRequest = async (req, res, next) => {
|
|
10
|
+
const { message, threadId } = req.body;
|
|
11
|
+
const userId = res.locals.userId;
|
|
12
|
+
try {
|
|
13
|
+
const result = await this.queryService.handleQuery(
|
|
14
|
+
message,
|
|
15
|
+
threadId,
|
|
16
|
+
userId
|
|
17
|
+
);
|
|
18
|
+
res.status(200).json(result);
|
|
19
|
+
} catch (error) {
|
|
20
|
+
next(error);
|
|
21
|
+
}
|
|
22
|
+
}}
|
|
23
|
+
__init2() {this.handleQueryStreamRequest = async (req, res, next) => {
|
|
24
|
+
const { message, threadId } = req.body;
|
|
25
|
+
const userId = res.locals.userId;
|
|
26
|
+
if (!this.queryStreamService) {
|
|
27
|
+
throw new Error("This Agent does not support stream query");
|
|
28
|
+
}
|
|
29
|
+
try {
|
|
30
|
+
await this.queryStreamService.handleQueryStream(
|
|
31
|
+
message,
|
|
32
|
+
userId,
|
|
33
|
+
res,
|
|
34
|
+
threadId
|
|
35
|
+
);
|
|
36
|
+
} catch (error) {
|
|
37
|
+
next(error);
|
|
38
|
+
}
|
|
39
|
+
}}
|
|
40
|
+
}, _class);
|
|
41
|
+
|
|
42
|
+
|
|
43
|
+
|
|
44
|
+
exports.QueryController = QueryController;
|
|
45
|
+
//# sourceMappingURL=chunk-O6XQRAWR.cjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-O6XQRAWR.cjs","../../src/controllers/query.controller.ts"],"names":[],"mappings":"AAAA;ACGO,IAAM,gBAAA,YAAN,MAAsB;AAAA,EACpB;AAAA,EACA;AAAA,EAER,WAAA,CACC,YAAA,EACA,kBAAA,EACC;AACD,IAAA,IAAA,CAAK,aAAA,EAAe,YAAA;AACpB,IAAA,IAAA,CAAK,mBAAA,EAAqB,kBAAA;AAAA,EAC3B;AAAA,iBAEO,mBAAA,EAAqB,MAAA,CAC3B,GAAA,EACA,GAAA,EACA,IAAA,EAAA,GACI;AACJ,IAAA,MAAM,EAAE,OAAA,EAAS,SAAS,EAAA,EAAI,GAAA,CAAI,IAAA;AAClC,IAAA,MAAM,OAAA,EAAS,GAAA,CAAI,MAAA,CAAO,MAAA;AAE1B,IAAA,IAAI;AACH,MAAA,MAAM,OAAA,EAAS,MAAM,IAAA,CAAK,YAAA,CAAa,WAAA;AAAA,QACtC,OAAA;AAAA,QACA,QAAA;AAAA,QACA;AAAA,MACD,CAAA;AAEA,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;AAAA,kBAEO,yBAAA,EAA2B,MAAA,CACjC,GAAA,EACA,GAAA,EACA,IAAA,EAAA,GACI;AACJ,IAAA,MAAM,EAAE,OAAA,EAAS,SAAS,EAAA,EAAI,GAAA,CAAI,IAAA;AAClC,IAAA,MAAM,OAAA,EAAS,GAAA,CAAI,MAAA,CAAO,MAAA;AAE1B,IAAA,GAAA,CAAI,CAAC,IAAA,CAAK,kBAAA,EAAoB;AAC7B,MAAA,MAAM,IAAI,KAAA,CAAM,0CAA0C,CAAA;AAAA,IAC3D;AAEA,IAAA,IAAI;AACH,MAAA,MAAM,IAAA,CAAK,kBAAA,CAAmB,iBAAA;AAAA,QAC7B,OAAA;AAAA,QACA,MAAA;AAAA,QACA,GAAA;AAAA,QACA;AAAA,MACD,CAAA;AAAA,IACD,EAAA,MAAA,CAAS,KAAA,EAAO;AACf,MAAA,IAAA,CAAK,KAAK,CAAA;AAAA,IACX;AAAA,EACD,EAAA;AACD,UAAA;ADnBA;AACA;AACE;AACF,0CAAC","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-O6XQRAWR.cjs","sourcesContent":[null,"import type { NextFunction, Request, Response } from \"express\";\nimport type { QueryService, QueryStreamService } from \"@/services\";\n\nexport class QueryController {\n\tprivate queryService;\n\tprivate queryStreamService;\n\n\tconstructor(\n\t\tqueryService: QueryService,\n\t\tqueryStreamService?: QueryStreamService,\n\t) {\n\t\tthis.queryService = queryService;\n\t\tthis.queryStreamService = queryStreamService;\n\t}\n\n\tpublic handleQueryRequest = async (\n\t\treq: Request,\n\t\tres: Response,\n\t\tnext: NextFunction,\n\t) => {\n\t\tconst { message, threadId } = req.body;\n\t\tconst userId = res.locals.userId;\n\n\t\ttry {\n\t\t\tconst result = await this.queryService.handleQuery(\n\t\t\t\tmessage,\n\t\t\t\tthreadId,\n\t\t\t\tuserId,\n\t\t\t);\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\tpublic handleQueryStreamRequest = async (\n\t\treq: Request,\n\t\tres: Response,\n\t\tnext: NextFunction,\n\t) => {\n\t\tconst { message, threadId } = req.body;\n\t\tconst userId = res.locals.userId;\n\n\t\tif (!this.queryStreamService) {\n\t\t\tthrow new Error(\"This Agent does not support stream query\");\n\t\t}\n\n\t\ttry {\n\t\t\tawait this.queryStreamService.handleQueryStream(\n\t\t\t\tmessage,\n\t\t\t\tuserId,\n\t\t\t\tres,\n\t\t\t\tthreadId,\n\t\t\t);\n\t\t} catch (error) {\n\t\t\tnext(error);\n\t\t}\n\t};\n}\n"]}
|
|
@@ -0,0 +1,37 @@
|
|
|
1
|
+
"use strict";Object.defineProperty(exports, "__esModule", {value: true});
|
|
2
|
+
|
|
3
|
+
var _chunk55IME4BGcjs = require('./chunk-55IME4BG.cjs');
|
|
4
|
+
|
|
5
|
+
|
|
6
|
+
var _chunk7EGJAURFcjs = require('./chunk-7EGJAURF.cjs');
|
|
7
|
+
|
|
8
|
+
|
|
9
|
+
var _chunkICFTUZP5cjs = require('./chunk-ICFTUZP5.cjs');
|
|
10
|
+
|
|
11
|
+
// src/routes/a2a.routes.ts
|
|
12
|
+
var _server = require('@a2a-js/sdk/server');
|
|
13
|
+
var _express = require('express');
|
|
14
|
+
var createA2ARouter = (agent) => {
|
|
15
|
+
const router = _express.Router.call(void 0, );
|
|
16
|
+
const taskStore = new (0, _server.InMemoryTaskStore)();
|
|
17
|
+
const queryService = new (0, _chunk7EGJAURFcjs.QueryService)(
|
|
18
|
+
agent.modelModule,
|
|
19
|
+
agent.a2aModule,
|
|
20
|
+
agent.mcpModule,
|
|
21
|
+
agent.memoryModule,
|
|
22
|
+
agent.manifest.prompts
|
|
23
|
+
);
|
|
24
|
+
const a2aService = new (0, _chunk55IME4BGcjs.A2AService)(queryService);
|
|
25
|
+
const a2aController = new (0, _chunkICFTUZP5cjs.A2AController)(
|
|
26
|
+
a2aService,
|
|
27
|
+
taskStore,
|
|
28
|
+
agent.generateAgentCard
|
|
29
|
+
);
|
|
30
|
+
router.post("/", a2aController.handleA2ARequest);
|
|
31
|
+
return router;
|
|
32
|
+
};
|
|
33
|
+
|
|
34
|
+
|
|
35
|
+
|
|
36
|
+
exports.createA2ARouter = createA2ARouter;
|
|
37
|
+
//# sourceMappingURL=chunk-PDF6B6A3.cjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-PDF6B6A3.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,GAAA,EAAK,aAAA,CAAc,gBAAgB,CAAA;AAE/C,EAAA,OAAO,MAAA;AACR,CAAA;ADNA;AACA;AACE;AACF,0CAAC","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-PDF6B6A3.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(\"/\", a2aController.handleA2ARequest);\n\n\treturn router;\n};\n"]}
|
|
@@ -0,0 +1,162 @@
|
|
|
1
|
+
"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; }
|
|
2
|
+
|
|
3
|
+
var _chunk466ED5WNcjs = require('./chunk-466ED5WN.cjs');
|
|
4
|
+
|
|
5
|
+
// src/services/query.service.ts
|
|
6
|
+
var QueryService = class {
|
|
7
|
+
|
|
8
|
+
|
|
9
|
+
|
|
10
|
+
|
|
11
|
+
|
|
12
|
+
constructor(modelModule, a2aModule, mcpModule, memoryModule, prompts) {
|
|
13
|
+
this.modelModule = modelModule;
|
|
14
|
+
this.a2aModule = a2aModule;
|
|
15
|
+
this.mcpModule = mcpModule;
|
|
16
|
+
this.memoryModule = memoryModule;
|
|
17
|
+
this.prompts = prompts;
|
|
18
|
+
}
|
|
19
|
+
/**
|
|
20
|
+
* Detects the intent from a user query.
|
|
21
|
+
*
|
|
22
|
+
* @param query - The user's input query
|
|
23
|
+
* @returns The detected intent (currently returns the query as-is)
|
|
24
|
+
* @todo Implement actual intent detection logic
|
|
25
|
+
*/
|
|
26
|
+
async intentTriggering(query) {
|
|
27
|
+
return query;
|
|
28
|
+
}
|
|
29
|
+
/**
|
|
30
|
+
* Fulfills the detected intent by generating a response.
|
|
31
|
+
*
|
|
32
|
+
* Manages the complete inference loop including:
|
|
33
|
+
* - Loading prompts and conversation history
|
|
34
|
+
* - Collecting available tools from modules
|
|
35
|
+
* - Executing model inference with tool support
|
|
36
|
+
* - Processing tool calls iteratively until completion
|
|
37
|
+
*
|
|
38
|
+
* @param query - The user's input query
|
|
39
|
+
* @param threadId - Thread identifier for context
|
|
40
|
+
* @param thread - Previous conversation history
|
|
41
|
+
* @returns Object containing process steps and final response
|
|
42
|
+
*/
|
|
43
|
+
async intentFulfilling(query, threadId, thread) {
|
|
44
|
+
const systemPrompt = `
|
|
45
|
+
Today is ${(/* @__PURE__ */ new Date()).toLocaleDateString()}.
|
|
46
|
+
|
|
47
|
+
${_optionalChain([this, 'access', _ => _.prompts, 'optionalAccess', _2 => _2.agent]) || ""}
|
|
48
|
+
|
|
49
|
+
${_optionalChain([this, 'access', _3 => _3.prompts, 'optionalAccess', _4 => _4.system]) || ""}
|
|
50
|
+
`;
|
|
51
|
+
const modelInstance = this.modelModule.getModel();
|
|
52
|
+
const messages = modelInstance.generateMessages({
|
|
53
|
+
query,
|
|
54
|
+
thread,
|
|
55
|
+
systemPrompt: systemPrompt.trim()
|
|
56
|
+
});
|
|
57
|
+
const tools = [];
|
|
58
|
+
if (this.mcpModule) {
|
|
59
|
+
tools.push(...this.mcpModule.getTools());
|
|
60
|
+
}
|
|
61
|
+
if (this.a2aModule) {
|
|
62
|
+
tools.push(...await this.a2aModule.getTools());
|
|
63
|
+
}
|
|
64
|
+
const functions = modelInstance.convertToolsToFunctions(tools);
|
|
65
|
+
const processList = [];
|
|
66
|
+
let finalMessage = "";
|
|
67
|
+
let didCallTool = false;
|
|
68
|
+
while (true) {
|
|
69
|
+
const response = await modelInstance.fetchWithContextMessage(
|
|
70
|
+
messages,
|
|
71
|
+
functions
|
|
72
|
+
);
|
|
73
|
+
didCallTool = false;
|
|
74
|
+
_chunk466ED5WNcjs.loggers.intent.debug("messages", { messages });
|
|
75
|
+
const { content, toolCalls } = response;
|
|
76
|
+
_chunk466ED5WNcjs.loggers.intent.debug("content", { content });
|
|
77
|
+
_chunk466ED5WNcjs.loggers.intent.debug("tool_calls", { ...toolCalls });
|
|
78
|
+
if (toolCalls) {
|
|
79
|
+
const messagePayload = _optionalChain([this, 'access', _5 => _5.a2aModule, 'optionalAccess', _6 => _6.getMessagePayload, 'call', _7 => _7(
|
|
80
|
+
query,
|
|
81
|
+
threadId
|
|
82
|
+
)]);
|
|
83
|
+
for (const toolCall of toolCalls) {
|
|
84
|
+
const toolName = toolCall.name;
|
|
85
|
+
didCallTool = true;
|
|
86
|
+
const selectedTool = tools.filter((tool) => tool.id === toolName)[0];
|
|
87
|
+
let toolResult = "";
|
|
88
|
+
if (this.mcpModule && selectedTool.protocol === "MCP" /* MCP */) {
|
|
89
|
+
const toolArgs = toolCall.arguments;
|
|
90
|
+
_chunk466ED5WNcjs.loggers.intent.debug("MCP tool call", { toolName, toolArgs });
|
|
91
|
+
toolResult = await this.mcpModule.useTool(
|
|
92
|
+
selectedTool,
|
|
93
|
+
toolArgs
|
|
94
|
+
);
|
|
95
|
+
} else if (this.a2aModule && selectedTool.protocol === "A2A" /* A2A */) {
|
|
96
|
+
toolResult = await this.a2aModule.useTool(
|
|
97
|
+
selectedTool,
|
|
98
|
+
messagePayload,
|
|
99
|
+
threadId
|
|
100
|
+
);
|
|
101
|
+
} else {
|
|
102
|
+
_chunk466ED5WNcjs.loggers.intent.warn(
|
|
103
|
+
`Unrecognized tool type: ${selectedTool.protocol}`
|
|
104
|
+
);
|
|
105
|
+
continue;
|
|
106
|
+
}
|
|
107
|
+
_chunk466ED5WNcjs.loggers.intent.debug("toolResult", { toolResult });
|
|
108
|
+
processList.push(toolResult);
|
|
109
|
+
modelInstance.appendMessages(messages, toolResult);
|
|
110
|
+
}
|
|
111
|
+
} else if (content) {
|
|
112
|
+
processList.push(content);
|
|
113
|
+
finalMessage = content;
|
|
114
|
+
}
|
|
115
|
+
if (!didCallTool) break;
|
|
116
|
+
}
|
|
117
|
+
const botResponse = {
|
|
118
|
+
process: processList.join("\n"),
|
|
119
|
+
response: finalMessage
|
|
120
|
+
};
|
|
121
|
+
return botResponse;
|
|
122
|
+
}
|
|
123
|
+
/**
|
|
124
|
+
* Main entry point for processing user queries.
|
|
125
|
+
*
|
|
126
|
+
* Handles the complete query lifecycle:
|
|
127
|
+
* 1. Loads thread history from memory
|
|
128
|
+
* 2. Detects intent from the query
|
|
129
|
+
* 3. Fulfills the intent with AI response
|
|
130
|
+
* 4. Updates conversation history
|
|
131
|
+
*
|
|
132
|
+
* @param query - The user's input query
|
|
133
|
+
* @param threadId - Unique thread identifier
|
|
134
|
+
* @param userId - Unique user identifier
|
|
135
|
+
* @returns Object containing the AI-generated response
|
|
136
|
+
*/
|
|
137
|
+
async handleQuery(query, threadId, userId) {
|
|
138
|
+
const queryStartAt = Date.now();
|
|
139
|
+
const threadMemory = _optionalChain([this, 'access', _8 => _8.memoryModule, 'optionalAccess', _9 => _9.getThreadMemory, 'call', _10 => _10()]);
|
|
140
|
+
const thread = !userId ? void 0 : await _optionalChain([threadMemory, 'optionalAccess', _11 => _11.getThread, 'call', _12 => _12(userId, threadId)]);
|
|
141
|
+
const intent = this.intentTriggering(query);
|
|
142
|
+
const result = await this.intentFulfilling(query, threadId, thread);
|
|
143
|
+
if (userId) {
|
|
144
|
+
await _optionalChain([threadMemory, 'optionalAccess', _13 => _13.addMessageToThread, 'call', _14 => _14(userId, threadId, {
|
|
145
|
+
role: "USER" /* USER */,
|
|
146
|
+
timestamp: queryStartAt,
|
|
147
|
+
content: { type: "text", parts: [query] }
|
|
148
|
+
})]);
|
|
149
|
+
await _optionalChain([threadMemory, 'optionalAccess', _15 => _15.addMessageToThread, 'call', _16 => _16(userId, threadId, {
|
|
150
|
+
role: "MODEL" /* MODEL */,
|
|
151
|
+
timestamp: Date.now(),
|
|
152
|
+
content: { type: "text", parts: [result.response] }
|
|
153
|
+
})]);
|
|
154
|
+
}
|
|
155
|
+
return { content: result.response };
|
|
156
|
+
}
|
|
157
|
+
};
|
|
158
|
+
|
|
159
|
+
|
|
160
|
+
|
|
161
|
+
exports.QueryService = QueryService;
|
|
162
|
+
//# sourceMappingURL=chunk-PT2CVI52.cjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-PT2CVI52.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,QAAA,EACA,MAAA,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,MAAA;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,WAAA,EAAa,MAAM,IAAA,CAAK,SAAA,CAAU,OAAA;AAAA,cACjC,YAAA;AAAA,cACA;AAAA,YACD,CAAA;AAAA,UACD,EAAA,KAAA,GAAA,CACC,IAAA,CAAK,UAAA,GACL,YAAA,CAAa,SAAA,IAAA,eAAA,EACZ;AACD,YAAA,WAAA,EAAa,MAAM,IAAA,CAAK,SAAA,CAAU,OAAA;AAAA,cACjC,YAAA;AAAA,cACA,cAAA;AAAA,cACA;AAAA,YACD,CAAA;AAAA,UACD,EAAA,KAAO;AAEN,YAAA,yBAAA,CAAQ,MAAA,CAAO,IAAA;AAAA,cACd,CAAA,wBAAA,EAA2B,YAAA,CAAa,QAAQ,CAAA;AAAA,YAAA;AAEjD,YAAA;AAAA,UAAA;AAGD,UAAA;AAEA,UAAA;AACA,UAAA;AAAiD,QAAA;AAClD,MAAA;AAEA,QAAA;AACA,QAAA;AAAe,MAAA;AAGhB,MAAA;AAAkB,IAAA;AAGnB,IAAA;AAAoB,MAAA;AACW,MAAA;AACpB,IAAA;AAGX,IAAA;AAAO,EAAA;AACR;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAAA;AAkBC,IAAA;AACA,IAAA;AACA,IAAA;AAKA,IAAA;AAGA,IAAA;AACA,IAAA;AACC,MAAA;AAAyD,QAAA;AACxD,QAAA;AACW,QAAA;AAC6B,MAAA;AAEzC,MAAA;AAAyD,QAAA;AACxD,QAAA;AACoB,QAAA;AAC8B,MAAA;AAClD,IAAA;AAGF,IAAA;AAAkC,EAAA;AAEpC;ADjEA;AACA;AACA;AACA","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-PT2CVI52.cjs","sourcesContent":[null,"import type {\n\tA2AModule,\n\tMCPModule,\n\tMemoryModule,\n\tModelModule,\n} from \"@/modules/index.js\";\nimport type { AinAgentPrompts } from \"@/types/agent.js\";\nimport { MessageRole, type ThreadObject } from \"@/types/memory.js\";\nimport {\n\ttype IA2ATool,\n\ttype IAgentTool,\n\ttype IMCPTool,\n\tTOOL_PROTOCOL_TYPE,\n} from \"@/types/tool.js\";\nimport { loggers } from \"@/utils/logger.js\";\n\n/**\n * Service for processing user queries through the agent's AI pipeline.\n *\n * Orchestrates the query processing workflow including intent detection,\n * model inference, tool execution, and response generation. Manages\n * conversation context and coordinates between different modules.\n */\nexport class QueryService {\n\tprivate modelModule: ModelModule;\n\tprivate a2aModule?: A2AModule;\n\tprivate mcpModule?: MCPModule;\n\tprivate memoryModule?: MemoryModule;\n\tprivate prompts?: AinAgentPrompts;\n\n\tconstructor(\n\t\tmodelModule: ModelModule,\n\t\ta2aModule?: A2AModule,\n\t\tmcpModule?: MCPModule,\n\t\tmemoryModule?: MemoryModule,\n\t\tprompts?: AinAgentPrompts,\n\t) {\n\t\tthis.modelModule = modelModule;\n\t\tthis.a2aModule = a2aModule;\n\t\tthis.mcpModule = mcpModule;\n\t\tthis.memoryModule = memoryModule;\n\t\tthis.prompts = prompts;\n\t}\n\n\t/**\n\t * Detects the intent from a user query.\n\t *\n\t * @param query - The user's input query\n\t * @returns The detected intent (currently returns the query as-is)\n\t * @todo Implement actual intent detection logic\n\t */\n\tprivate async intentTriggering(query: string) {\n\t\t/* TODO */\n\t\treturn query;\n\t}\n\n\t/**\n\t * Fulfills the detected intent by generating a response.\n\t *\n\t * Manages the complete inference loop including:\n\t * - Loading prompts and conversation history\n\t * - Collecting available tools from modules\n\t * - Executing model inference with tool support\n\t * - Processing tool calls iteratively until completion\n\t *\n\t * @param query - The user's input query\n\t * @param threadId - Thread identifier for context\n\t * @param thread - Previous conversation history\n\t * @returns Object containing process steps and final response\n\t */\n\tprivate async intentFulfilling(\n\t\tquery: string,\n\t\tthreadId: string,\n\t\tthread?: ThreadObject,\n\t) {\n\t\t// 1. Load agent / system prompt from memory\n\t\tconst systemPrompt = `\nToday is ${new Date().toLocaleDateString()}.\n\n${this.prompts?.agent || \"\"}\n\n${this.prompts?.system || \"\"}\n `;\n\n\t\tconst modelInstance = this.modelModule.getModel();\n\t\tconst messages = modelInstance.generateMessages({\n\t\t\tquery,\n\t\t\tthread,\n\t\t\tsystemPrompt: systemPrompt.trim(),\n\t\t});\n\n\t\tconst tools: IAgentTool[] = [];\n\t\tif (this.mcpModule) {\n\t\t\ttools.push(...this.mcpModule.getTools());\n\t\t}\n\t\tif (this.a2aModule) {\n\t\t\ttools.push(...(await this.a2aModule.getTools()));\n\t\t}\n\t\tconst functions = modelInstance.convertToolsToFunctions(tools);\n\n\t\tconst processList: string[] = [];\n\t\tlet finalMessage = \"\";\n\t\tlet didCallTool = false;\n\n\t\twhile (true) {\n\t\t\tconst response = await modelInstance.fetchWithContextMessage(\n\t\t\t\tmessages,\n\t\t\t\tfunctions,\n\t\t\t);\n\t\t\tdidCallTool = false;\n\n\t\t\tloggers.intent.debug(\"messages\", { messages });\n\n\t\t\tconst { content, toolCalls } = response;\n\n\t\t\tloggers.intent.debug(\"content\", { content });\n\t\t\tloggers.intent.debug(\"tool_calls\", { ...toolCalls });\n\n\t\t\tif (toolCalls) {\n\t\t\t\tconst messagePayload = this.a2aModule?.getMessagePayload(\n\t\t\t\t\tquery,\n\t\t\t\t\tthreadId,\n\t\t\t\t);\n\n\t\t\t\tfor (const toolCall of toolCalls) {\n\t\t\t\t\tconst toolName = toolCall.name;\n\t\t\t\t\tdidCallTool = true;\n\t\t\t\t\tconst selectedTool = tools.filter((tool) => tool.id === toolName)[0];\n\n\t\t\t\t\tlet toolResult = \"\";\n\t\t\t\t\tif (\n\t\t\t\t\t\tthis.mcpModule &&\n\t\t\t\t\t\tselectedTool.protocol === TOOL_PROTOCOL_TYPE.MCP\n\t\t\t\t\t) {\n\t\t\t\t\t\tconst toolArgs = toolCall.arguments as\n\t\t\t\t\t\t\t| { [x: string]: unknown }\n\t\t\t\t\t\t\t| undefined;\n\t\t\t\t\t\tloggers.intent.debug(\"MCP tool call\", { toolName, toolArgs });\n\t\t\t\t\t\ttoolResult = await this.mcpModule.useTool(\n\t\t\t\t\t\t\tselectedTool as IMCPTool,\n\t\t\t\t\t\t\ttoolArgs,\n\t\t\t\t\t\t);\n\t\t\t\t\t} else if (\n\t\t\t\t\t\tthis.a2aModule &&\n\t\t\t\t\t\tselectedTool.protocol === TOOL_PROTOCOL_TYPE.A2A\n\t\t\t\t\t) {\n\t\t\t\t\t\ttoolResult = await this.a2aModule.useTool(\n\t\t\t\t\t\t\tselectedTool as IA2ATool,\n\t\t\t\t\t\t\tmessagePayload!,\n\t\t\t\t\t\t\tthreadId,\n\t\t\t\t\t\t);\n\t\t\t\t\t} else {\n\t\t\t\t\t\t// Unrecognized tool type. It cannot be happened...\n\t\t\t\t\t\tloggers.intent.warn(\n\t\t\t\t\t\t\t`Unrecognized tool type: ${selectedTool.protocol}`,\n\t\t\t\t\t\t);\n\t\t\t\t\t\tcontinue;\n\t\t\t\t\t}\n\n\t\t\t\t\tloggers.intent.debug(\"toolResult\", { toolResult });\n\n\t\t\t\t\tprocessList.push(toolResult);\n\t\t\t\t\tmodelInstance.appendMessages(messages, toolResult);\n\t\t\t\t}\n\t\t\t} else if (content) {\n\t\t\t\tprocessList.push(content);\n\t\t\t\tfinalMessage = content;\n\t\t\t}\n\n\t\t\tif (!didCallTool) break;\n\t\t}\n\n\t\tconst botResponse = {\n\t\t\tprocess: processList.join(\"\\n\"),\n\t\t\tresponse: finalMessage,\n\t\t};\n\n\t\treturn botResponse;\n\t}\n\n\t/**\n\t * Main entry point for processing user queries.\n\t *\n\t * Handles the complete query lifecycle:\n\t * 1. Loads thread history from memory\n\t * 2. Detects intent from the query\n\t * 3. Fulfills the intent with AI response\n\t * 4. Updates conversation history\n\t *\n\t * @param query - The user's input query\n\t * @param threadId - Unique thread identifier\n\t * @param userId - Unique user identifier\n\t * @returns Object containing the AI-generated response\n\t */\n\tpublic async handleQuery(query: string, threadId: string, userId?: string) {\n\t\t// 1. Load thread with threadId\n\t\tconst queryStartAt = Date.now();\n\t\tconst threadMemory = this.memoryModule?.getThreadMemory();\n\t\tconst thread = !userId\n\t\t\t? undefined\n\t\t\t: await threadMemory?.getThread(userId, threadId);\n\n\t\t// 2. intent triggering\n\t\tconst intent = this.intentTriggering(query);\n\n\t\t// 3. intent fulfillment\n\t\tconst result = await this.intentFulfilling(query, threadId, thread);\n\t\tif (userId) {\n\t\t\tawait threadMemory?.addMessageToThread(userId, threadId, {\n\t\t\t\trole: MessageRole.USER,\n\t\t\t\ttimestamp: queryStartAt,\n\t\t\t\tcontent: { type: \"text\", parts: [query] },\n\t\t\t});\n\t\t\tawait threadMemory?.addMessageToThread(userId, threadId, {\n\t\t\t\trole: MessageRole.MODEL,\n\t\t\t\ttimestamp: Date.now(),\n\t\t\t\tcontent: { type: \"text\", parts: [result.response] },\n\t\t\t});\n\t\t}\n\n\t\treturn { content: result.response };\n\t}\n}\n"]}
|
|
@@ -0,0 +1,37 @@
|
|
|
1
|
+
"use strict";Object.defineProperty(exports, "__esModule", {value: true});
|
|
2
|
+
|
|
3
|
+
var _chunk55IME4BGcjs = require('./chunk-55IME4BG.cjs');
|
|
4
|
+
|
|
5
|
+
|
|
6
|
+
var _chunkJAPFZA2Kcjs = require('./chunk-JAPFZA2K.cjs');
|
|
7
|
+
|
|
8
|
+
|
|
9
|
+
var _chunkICFTUZP5cjs = require('./chunk-ICFTUZP5.cjs');
|
|
10
|
+
|
|
11
|
+
// src/routes/a2a.routes.ts
|
|
12
|
+
var _server = require('@a2a-js/sdk/server');
|
|
13
|
+
var _express = require('express');
|
|
14
|
+
var createA2ARouter = (agent) => {
|
|
15
|
+
const router = _express.Router.call(void 0, );
|
|
16
|
+
const taskStore = new (0, _server.InMemoryTaskStore)();
|
|
17
|
+
const queryService = new (0, _chunkJAPFZA2Kcjs.QueryService)(
|
|
18
|
+
agent.modelModule,
|
|
19
|
+
agent.a2aModule,
|
|
20
|
+
agent.mcpModule,
|
|
21
|
+
agent.memoryModule,
|
|
22
|
+
agent.manifest.prompts
|
|
23
|
+
);
|
|
24
|
+
const a2aService = new (0, _chunk55IME4BGcjs.A2AService)(queryService);
|
|
25
|
+
const a2aController = new (0, _chunkICFTUZP5cjs.A2AController)(
|
|
26
|
+
a2aService,
|
|
27
|
+
taskStore,
|
|
28
|
+
agent.generateAgentCard
|
|
29
|
+
);
|
|
30
|
+
router.post("/", a2aController.handleA2ARequest);
|
|
31
|
+
return router;
|
|
32
|
+
};
|
|
33
|
+
|
|
34
|
+
|
|
35
|
+
|
|
36
|
+
exports.createA2ARouter = createA2ARouter;
|
|
37
|
+
//# sourceMappingURL=chunk-QUSBMECP.cjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-QUSBMECP.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,GAAA,EAAK,aAAA,CAAc,gBAAgB,CAAA;AAE/C,EAAA,OAAO,MAAA;AACR,CAAA;ADNA;AACA;AACE;AACF,0CAAC","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-QUSBMECP.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(\"/\", a2aController.handleA2ARequest);\n\n\treturn router;\n};\n"]}
|
|
@@ -0,0 +1,37 @@
|
|
|
1
|
+
"use strict";Object.defineProperty(exports, "__esModule", {value: true});
|
|
2
|
+
|
|
3
|
+
var _chunk6BKOR3ZHcjs = require('./chunk-6BKOR3ZH.cjs');
|
|
4
|
+
|
|
5
|
+
|
|
6
|
+
var _chunkPT2CVI52cjs = require('./chunk-PT2CVI52.cjs');
|
|
7
|
+
|
|
8
|
+
|
|
9
|
+
var _chunkICFTUZP5cjs = require('./chunk-ICFTUZP5.cjs');
|
|
10
|
+
|
|
11
|
+
// src/routes/a2a.routes.ts
|
|
12
|
+
var _server = require('@a2a-js/sdk/server');
|
|
13
|
+
var _express = require('express');
|
|
14
|
+
var createA2ARouter = (agent) => {
|
|
15
|
+
const router = _express.Router.call(void 0, );
|
|
16
|
+
const taskStore = new (0, _server.InMemoryTaskStore)();
|
|
17
|
+
const queryService = new (0, _chunkPT2CVI52cjs.QueryService)(
|
|
18
|
+
agent.modelModule,
|
|
19
|
+
agent.a2aModule,
|
|
20
|
+
agent.mcpModule,
|
|
21
|
+
agent.memoryModule,
|
|
22
|
+
agent.manifest.prompts
|
|
23
|
+
);
|
|
24
|
+
const a2aService = new (0, _chunk6BKOR3ZHcjs.A2AService)(queryService);
|
|
25
|
+
const a2aController = new (0, _chunkICFTUZP5cjs.A2AController)(
|
|
26
|
+
a2aService,
|
|
27
|
+
taskStore,
|
|
28
|
+
agent.generateAgentCard
|
|
29
|
+
);
|
|
30
|
+
router.post("/", a2aController.handleA2ARequest);
|
|
31
|
+
return router;
|
|
32
|
+
};
|
|
33
|
+
|
|
34
|
+
|
|
35
|
+
|
|
36
|
+
exports.createA2ARouter = createA2ARouter;
|
|
37
|
+
//# sourceMappingURL=chunk-SAXFX7U2.cjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-SAXFX7U2.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,GAAA,EAAK,aAAA,CAAc,gBAAgB,CAAA;AAE/C,EAAA,OAAO,MAAA;AACR,CAAA;ADNA;AACA;AACE;AACF,0CAAC","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-SAXFX7U2.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(\"/\", a2aController.handleA2ARequest);\n\n\treturn router;\n};\n"]}
|
|
@@ -0,0 +1,51 @@
|
|
|
1
|
+
"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; }// src/modules/memory/memory.module.ts
|
|
2
|
+
var MemoryModule = class {
|
|
3
|
+
|
|
4
|
+
|
|
5
|
+
|
|
6
|
+
constructor(config) {
|
|
7
|
+
this.agentMemory = config.agent;
|
|
8
|
+
this.threadMemory = config.thread;
|
|
9
|
+
this.intentMemory = config.intent;
|
|
10
|
+
}
|
|
11
|
+
async initialize() {
|
|
12
|
+
const connectPromises = [];
|
|
13
|
+
if (this.agentMemory) {
|
|
14
|
+
connectPromises.push(this.agentMemory.connect());
|
|
15
|
+
}
|
|
16
|
+
if (this.threadMemory) {
|
|
17
|
+
connectPromises.push(this.threadMemory.connect());
|
|
18
|
+
}
|
|
19
|
+
if (this.intentMemory) {
|
|
20
|
+
connectPromises.push(this.intentMemory.connect());
|
|
21
|
+
}
|
|
22
|
+
await Promise.all(connectPromises);
|
|
23
|
+
}
|
|
24
|
+
async shutdown() {
|
|
25
|
+
const disconnectPromises = [];
|
|
26
|
+
if (_optionalChain([this, 'access', _ => _.agentMemory, 'optionalAccess', _2 => _2.isConnected, 'call', _3 => _3()])) {
|
|
27
|
+
disconnectPromises.push(this.agentMemory.disconnect());
|
|
28
|
+
}
|
|
29
|
+
if (_optionalChain([this, 'access', _4 => _4.threadMemory, 'optionalAccess', _5 => _5.isConnected, 'call', _6 => _6()])) {
|
|
30
|
+
disconnectPromises.push(this.threadMemory.disconnect());
|
|
31
|
+
}
|
|
32
|
+
if (_optionalChain([this, 'access', _7 => _7.intentMemory, 'optionalAccess', _8 => _8.isConnected, 'call', _9 => _9()])) {
|
|
33
|
+
disconnectPromises.push(this.intentMemory.disconnect());
|
|
34
|
+
}
|
|
35
|
+
await Promise.all(disconnectPromises);
|
|
36
|
+
}
|
|
37
|
+
getAgentMemory() {
|
|
38
|
+
return this.agentMemory;
|
|
39
|
+
}
|
|
40
|
+
getThreadMemory() {
|
|
41
|
+
return this.threadMemory;
|
|
42
|
+
}
|
|
43
|
+
getIntentMemory() {
|
|
44
|
+
return this.intentMemory;
|
|
45
|
+
}
|
|
46
|
+
};
|
|
47
|
+
|
|
48
|
+
|
|
49
|
+
|
|
50
|
+
exports.MemoryModule = MemoryModule;
|
|
51
|
+
//# sourceMappingURL=chunk-SCOMNXDA.cjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-SCOMNXDA.cjs","../../src/modules/memory/memory.module.ts"],"names":[],"mappings":"AAAA;ACYO,IAAM,aAAA,EAAN,MAAmB;AAAA,EACjB;AAAA,EACA;AAAA,EACA;AAAA,EAER,WAAA,CAAY,MAAA,EAAsB;AACjC,IAAA,IAAA,CAAK,YAAA,EAAc,MAAA,CAAO,KAAA;AAC1B,IAAA,IAAA,CAAK,aAAA,EAAe,MAAA,CAAO,MAAA;AAC3B,IAAA,IAAA,CAAK,aAAA,EAAe,MAAA,CAAO,MAAA;AAAA,EAC5B;AAAA,EAEA,MAAM,UAAA,CAAA,EAA4B;AACjC,IAAA,MAAM,gBAAA,EAAmC,CAAC,CAAA;AAE1C,IAAA,GAAA,CAAI,IAAA,CAAK,WAAA,EAAa;AACrB,MAAA,eAAA,CAAgB,IAAA,CAAK,IAAA,CAAK,WAAA,CAAY,OAAA,CAAQ,CAAC,CAAA;AAAA,IAChD;AACA,IAAA,GAAA,CAAI,IAAA,CAAK,YAAA,EAAc;AACtB,MAAA,eAAA,CAAgB,IAAA,CAAK,IAAA,CAAK,YAAA,CAAa,OAAA,CAAQ,CAAC,CAAA;AAAA,IACjD;AACA,IAAA,GAAA,CAAI,IAAA,CAAK,YAAA,EAAc;AACtB,MAAA,eAAA,CAAgB,IAAA,CAAK,IAAA,CAAK,YAAA,CAAa,OAAA,CAAQ,CAAC,CAAA;AAAA,IACjD;AAEA,IAAA,MAAM,OAAA,CAAQ,GAAA,CAAI,eAAe,CAAA;AAAA,EAClC;AAAA,EAEA,MAAM,QAAA,CAAA,EAA0B;AAC/B,IAAA,MAAM,mBAAA,EAAsC,CAAC,CAAA;AAE7C,IAAA,GAAA,iBAAI,IAAA,mBAAK,WAAA,6BAAa,WAAA,mBAAY,GAAA,EAAG;AACpC,MAAA,kBAAA,CAAmB,IAAA,CAAK,IAAA,CAAK,WAAA,CAAY,UAAA,CAAW,CAAC,CAAA;AAAA,IACtD;AACA,IAAA,GAAA,iBAAI,IAAA,qBAAK,YAAA,6BAAc,WAAA,mBAAY,GAAA,EAAG;AACrC,MAAA,kBAAA,CAAmB,IAAA,CAAK,IAAA,CAAK,YAAA,CAAa,UAAA,CAAW,CAAC,CAAA;AAAA,IACvD;AACA,IAAA,GAAA,iBAAI,IAAA,qBAAK,YAAA,6BAAc,WAAA,mBAAY,GAAA,EAAG;AACrC,MAAA,kBAAA,CAAmB,IAAA,CAAK,IAAA,CAAK,YAAA,CAAa,UAAA,CAAW,CAAC,CAAA;AAAA,IACvD;AAEA,IAAA,MAAM,OAAA,CAAQ,GAAA,CAAI,kBAAkB,CAAA;AAAA,EACrC;AAAA,EAEO,cAAA,CAAA,EAA2C;AACjD,IAAA,OAAO,IAAA,CAAK,WAAA;AAAA,EACb;AAAA,EAEO,eAAA,CAAA,EAA6C;AACnD,IAAA,OAAO,IAAA,CAAK,YAAA;AAAA,EACb;AAAA,EAEO,eAAA,CAAA,EAA6C;AACnD,IAAA,OAAO,IAAA,CAAK,YAAA;AAAA,EACb;AACD,CAAA;ADpBA;AACA;AACE;AACF,oCAAC","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-SCOMNXDA.cjs","sourcesContent":[null,"import type {\n\tIAgentMemory,\n\tIIntentMemory,\n\tIThreadMemory,\n} from \"./base.memory.js\";\n\nexport interface MemoryConfig {\n\tagent?: IAgentMemory;\n\tthread?: IThreadMemory;\n\tintent?: IIntentMemory;\n}\n\nexport class MemoryModule {\n\tprivate agentMemory?: IAgentMemory;\n\tprivate threadMemory?: IThreadMemory;\n\tprivate intentMemory?: IIntentMemory;\n\n\tconstructor(config: MemoryConfig) {\n\t\tthis.agentMemory = config.agent;\n\t\tthis.threadMemory = config.thread;\n\t\tthis.intentMemory = config.intent;\n\t}\n\n\tasync initialize(): Promise<void> {\n\t\tconst connectPromises: Promise<void>[] = [];\n\n\t\tif (this.agentMemory) {\n\t\t\tconnectPromises.push(this.agentMemory.connect());\n\t\t}\n\t\tif (this.threadMemory) {\n\t\t\tconnectPromises.push(this.threadMemory.connect());\n\t\t}\n\t\tif (this.intentMemory) {\n\t\t\tconnectPromises.push(this.intentMemory.connect());\n\t\t}\n\n\t\tawait Promise.all(connectPromises);\n\t}\n\n\tasync shutdown(): Promise<void> {\n\t\tconst disconnectPromises: Promise<void>[] = [];\n\n\t\tif (this.agentMemory?.isConnected()) {\n\t\t\tdisconnectPromises.push(this.agentMemory.disconnect());\n\t\t}\n\t\tif (this.threadMemory?.isConnected()) {\n\t\t\tdisconnectPromises.push(this.threadMemory.disconnect());\n\t\t}\n\t\tif (this.intentMemory?.isConnected()) {\n\t\t\tdisconnectPromises.push(this.intentMemory.disconnect());\n\t\t}\n\n\t\tawait Promise.all(disconnectPromises);\n\t}\n\n\tpublic getAgentMemory(): IAgentMemory | undefined {\n\t\treturn this.agentMemory;\n\t}\n\n\tpublic getThreadMemory(): IThreadMemory | undefined {\n\t\treturn this.threadMemory;\n\t}\n\n\tpublic getIntentMemory(): IIntentMemory | undefined {\n\t\treturn this.intentMemory;\n\t}\n}\n"]}
|