@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 _chunk7EOUPEHUcjs = require('./chunk-7EOUPEHU.cjs');
|
|
4
|
+
|
|
5
|
+
|
|
6
|
+
var _chunk3J3ORQOGcjs = require('./chunk-3J3ORQOG.cjs');
|
|
7
|
+
|
|
8
|
+
|
|
9
|
+
var _chunk7EGJAURFcjs = require('./chunk-7EGJAURF.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, _chunk7EGJAURFcjs.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, _chunk7EOUPEHUcjs.QueryStreamService)(
|
|
25
|
+
agent.modelModule,
|
|
26
|
+
agent.a2aModule,
|
|
27
|
+
agent.mcpModule,
|
|
28
|
+
agent.memoryModule,
|
|
29
|
+
agent.manifest.prompts
|
|
30
|
+
);
|
|
31
|
+
}
|
|
32
|
+
const queryController = new (0, _chunk3J3ORQOGcjs.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-SGW3CSPD.cjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-SGW3CSPD.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-SGW3CSPD.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,65 @@
|
|
|
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
|
+
res.write(
|
|
45
|
+
`event: ${event.event}
|
|
46
|
+
data: ${JSON.stringify(event.data)}
|
|
47
|
+
|
|
48
|
+
`
|
|
49
|
+
);
|
|
50
|
+
}
|
|
51
|
+
} catch (error) {
|
|
52
|
+
res.write(`event: error
|
|
53
|
+
data: ${JSON.stringify(error)}
|
|
54
|
+
|
|
55
|
+
`);
|
|
56
|
+
} finally {
|
|
57
|
+
res.end();
|
|
58
|
+
}
|
|
59
|
+
}}
|
|
60
|
+
}, _class);
|
|
61
|
+
|
|
62
|
+
|
|
63
|
+
|
|
64
|
+
exports.QueryController = QueryController;
|
|
65
|
+
//# sourceMappingURL=chunk-SJHSHDJL.cjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-SJHSHDJL.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,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-SJHSHDJL.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\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,22 @@
|
|
|
1
|
+
"use strict";Object.defineProperty(exports, "__esModule", {value: true});
|
|
2
|
+
|
|
3
|
+
var _chunkX46GJS7Xcjs = require('./chunk-X46GJS7X.cjs');
|
|
4
|
+
|
|
5
|
+
|
|
6
|
+
var _chunkUW43BI5Zcjs = require('./chunk-UW43BI5Z.cjs');
|
|
7
|
+
|
|
8
|
+
// src/routes/api.routes.ts
|
|
9
|
+
var _express = require('express');
|
|
10
|
+
var createApiRouter = (agent) => {
|
|
11
|
+
const router = _express.Router.call(void 0, );
|
|
12
|
+
router.use("/api/model", _chunkX46GJS7Xcjs.createModelApiRouter.call(void 0, agent.modelModule));
|
|
13
|
+
if (agent.memoryModule) {
|
|
14
|
+
router.use("/api/thread", _chunkUW43BI5Zcjs.createThreadApiRouter.call(void 0, agent.memoryModule));
|
|
15
|
+
}
|
|
16
|
+
return router;
|
|
17
|
+
};
|
|
18
|
+
|
|
19
|
+
|
|
20
|
+
|
|
21
|
+
exports.createApiRouter = createApiRouter;
|
|
22
|
+
//# sourceMappingURL=chunk-UAWGHNZD.cjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-UAWGHNZD.cjs","../../src/routes/api.routes.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACA;ACPA,kCAAuB;AAMhB,IAAM,gBAAA,EAAkB,CAAC,KAAA,EAAA,GAA4B;AAC3D,EAAA,MAAM,OAAA,EAAS,6BAAA,CAAO;AAEtB,EAAA,MAAA,CAAO,GAAA,CAAI,YAAA,EAAc,oDAAA,KAAqB,CAAM,WAAW,CAAC,CAAA;AAChE,EAAA,GAAA,CAAI,KAAA,CAAM,YAAA,EAAc;AACvB,IAAA,MAAA,CAAO,GAAA,CAAI,aAAA,EAAe,qDAAA,KAAsB,CAAM,YAAY,CAAC,CAAA;AAAA,EACpE;AAEA,EAAA,OAAO,MAAA;AACR,CAAA;ADEA;AACA;AACE;AACF,0CAAC","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-UAWGHNZD.cjs","sourcesContent":[null,"import { Router } from \"express\";\nimport type { AINAgent } from \"@/index.js\";\n\nimport { createModelApiRouter } from \"./api/model.routes.js\";\nimport { createThreadApiRouter } from \"./api/thread.routes.js\";\n\nexport const createApiRouter = (agent: AINAgent): Router => {\n\tconst router = Router();\n\n\trouter.use(\"/api/model\", createModelApiRouter(agent.modelModule));\n\tif (agent.memoryModule) {\n\t\trouter.use(\"/api/thread\", createThreadApiRouter(agent.memoryModule));\n\t}\n\n\treturn router;\n};\n"]}
|
|
@@ -0,0 +1,43 @@
|
|
|
1
|
+
"use strict";Object.defineProperty(exports, "__esModule", {value: true});
|
|
2
|
+
|
|
3
|
+
var _chunkH65RVLWKcjs = require('./chunk-H65RVLWK.cjs');
|
|
4
|
+
|
|
5
|
+
|
|
6
|
+
var _chunkXVMXTCJIcjs = require('./chunk-XVMXTCJI.cjs');
|
|
7
|
+
|
|
8
|
+
|
|
9
|
+
var _chunk7EGJAURFcjs = require('./chunk-7EGJAURF.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, _chunk7EGJAURFcjs.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, _chunkH65RVLWKcjs.QueryStreamService)(
|
|
25
|
+
agent.modelModule,
|
|
26
|
+
agent.a2aModule,
|
|
27
|
+
agent.mcpModule,
|
|
28
|
+
agent.memoryModule,
|
|
29
|
+
agent.manifest.prompts
|
|
30
|
+
);
|
|
31
|
+
}
|
|
32
|
+
const queryController = new (0, _chunkXVMXTCJIcjs.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-UBEB55WC.cjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-UBEB55WC.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-UBEB55WC.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,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-UPNSAV5A.cjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-UPNSAV5A.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-UPNSAV5A.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: \"workflow\" | \"chat\";\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,17 @@
|
|
|
1
|
+
"use strict";Object.defineProperty(exports, "__esModule", {value: true});
|
|
2
|
+
|
|
3
|
+
var _chunk7XDYJNTAcjs = require('./chunk-7XDYJNTA.cjs');
|
|
4
|
+
|
|
5
|
+
// src/routes/api/thread.routes.ts
|
|
6
|
+
var _express = require('express');
|
|
7
|
+
var createThreadApiRouter = (memoryModule) => {
|
|
8
|
+
const router = _express.Router.call(void 0, );
|
|
9
|
+
const threadApiController = new (0, _chunk7XDYJNTAcjs.ThreadApiController)(memoryModule);
|
|
10
|
+
router.get("/:id", threadApiController.handleGetThread);
|
|
11
|
+
return router;
|
|
12
|
+
};
|
|
13
|
+
|
|
14
|
+
|
|
15
|
+
|
|
16
|
+
exports.createThreadApiRouter = createThreadApiRouter;
|
|
17
|
+
//# sourceMappingURL=chunk-UW43BI5Z.cjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-UW43BI5Z.cjs","../../src/routes/api/thread.routes.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACA;ACJA,kCAAuB;AAIhB,IAAM,sBAAA,EAAwB,CAAC,YAAA,EAAA,GAAuC;AAC5E,EAAA,MAAM,OAAA,EAAS,6BAAA,CAAO;AACtB,EAAA,MAAM,oBAAA,EAAsB,IAAI,0CAAA,CAAoB,YAAY,CAAA;AAGhE,EAAA,MAAA,CAAO,GAAA,CAAI,MAAA,EAAQ,mBAAA,CAAoB,eAAe,CAAA;AAEtD,EAAA,OAAO,MAAA;AACR,CAAA;ADAA;AACA;AACE;AACF,sDAAC","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-UW43BI5Z.cjs","sourcesContent":[null,"import { Router } from \"express\";\nimport { ThreadApiController } from \"@/controllers/api/thread.api.controller.js\";\nimport type { MemoryModule } from \"@/modules/index.js\";\n\nexport const createThreadApiRouter = (memoryModule: MemoryModule): Router => {\n\tconst router = Router();\n\tconst threadApiController = new ThreadApiController(memoryModule);\n\n\t// APIs (prefix: /api/thread)\n\trouter.get(\"/:id\", threadApiController.handleGetThread);\n\n\treturn router;\n};\n"]}
|
|
@@ -0,0 +1,43 @@
|
|
|
1
|
+
"use strict";Object.defineProperty(exports, "__esModule", {value: true});
|
|
2
|
+
|
|
3
|
+
var _chunkJ4Z7ZJUJcjs = require('./chunk-J4Z7ZJUJ.cjs');
|
|
4
|
+
|
|
5
|
+
|
|
6
|
+
var _chunkO6XQRAWRcjs = require('./chunk-O6XQRAWR.cjs');
|
|
7
|
+
|
|
8
|
+
|
|
9
|
+
var _chunkPT2CVI52cjs = require('./chunk-PT2CVI52.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, _chunkPT2CVI52cjs.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, _chunkJ4Z7ZJUJcjs.QueryStreamService)(
|
|
25
|
+
agent.modelModule,
|
|
26
|
+
agent.a2aModule,
|
|
27
|
+
agent.mcpModule,
|
|
28
|
+
agent.memoryModule,
|
|
29
|
+
agent.manifest.prompts
|
|
30
|
+
);
|
|
31
|
+
}
|
|
32
|
+
const queryController = new (0, _chunkO6XQRAWRcjs.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-WNDA2G4Q.cjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-WNDA2G4Q.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-WNDA2G4Q.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,17 @@
|
|
|
1
|
+
"use strict";Object.defineProperty(exports, "__esModule", {value: true});
|
|
2
|
+
|
|
3
|
+
var _chunkV5RSN2DYcjs = require('./chunk-V5RSN2DY.cjs');
|
|
4
|
+
|
|
5
|
+
// src/routes/api/model.routes.ts
|
|
6
|
+
var _express = require('express');
|
|
7
|
+
var createModelApiRouter = (modelModule) => {
|
|
8
|
+
const router = _express.Router.call(void 0, );
|
|
9
|
+
const modelApiController = new (0, _chunkV5RSN2DYcjs.ModelApiController)(modelModule);
|
|
10
|
+
router.get("/", modelApiController.handleModelList);
|
|
11
|
+
return router;
|
|
12
|
+
};
|
|
13
|
+
|
|
14
|
+
|
|
15
|
+
|
|
16
|
+
exports.createModelApiRouter = createModelApiRouter;
|
|
17
|
+
//# sourceMappingURL=chunk-X46GJS7X.cjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-X46GJS7X.cjs","../../src/routes/api/model.routes.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACA;ACJA,kCAAuB;AAIhB,IAAM,qBAAA,EAAuB,CAAC,WAAA,EAAA,GAAqC;AACzE,EAAA,MAAM,OAAA,EAAS,6BAAA,CAAO;AACtB,EAAA,MAAM,mBAAA,EAAqB,IAAI,yCAAA,CAAmB,WAAW,CAAA;AAG7D,EAAA,MAAA,CAAO,GAAA,CAAI,GAAA,EAAK,kBAAA,CAAmB,eAAe,CAAA;AAElD,EAAA,OAAO,MAAA;AACR,CAAA;ADAA;AACA;AACE;AACF,oDAAC","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-X46GJS7X.cjs","sourcesContent":[null,"import { Router } from \"express\";\nimport { ModelApiController } from \"@/controllers/api/model.api.controller.js\";\nimport type { ModelModule } from \"@/modules/index.js\";\n\nexport const createModelApiRouter = (modelModule: ModelModule): Router => {\n\tconst router = Router();\n\tconst modelApiController = new ModelApiController(modelModule);\n\n\t// APIs (prefix: /api/model)\n\trouter.get(\"/\", modelApiController.handleModelList);\n\n\treturn router;\n};\n"]}
|
|
@@ -0,0 +1,267 @@
|
|
|
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-stream.service.ts
|
|
6
|
+
var _crypto = require('crypto');
|
|
7
|
+
var QueryStreamService = class {
|
|
8
|
+
|
|
9
|
+
|
|
10
|
+
|
|
11
|
+
|
|
12
|
+
|
|
13
|
+
constructor(modelModule, a2aModule, mcpModule, memoryModule, prompts) {
|
|
14
|
+
this.modelModule = modelModule;
|
|
15
|
+
this.a2aModule = a2aModule;
|
|
16
|
+
this.mcpModule = mcpModule;
|
|
17
|
+
this.memoryModule = memoryModule;
|
|
18
|
+
this.prompts = prompts;
|
|
19
|
+
}
|
|
20
|
+
/**
|
|
21
|
+
* Detects the intent from a user query.
|
|
22
|
+
*
|
|
23
|
+
* @param query - The user's input query
|
|
24
|
+
* @returns The detected intent (currently returns the query as-is)
|
|
25
|
+
* @todo Implement actual intent detection logic
|
|
26
|
+
*/
|
|
27
|
+
async intentTriggering(query) {
|
|
28
|
+
return query;
|
|
29
|
+
}
|
|
30
|
+
/**
|
|
31
|
+
* Fulfills the detected intent by generating a response.
|
|
32
|
+
*
|
|
33
|
+
* Manages the complete inference loop including:
|
|
34
|
+
* - Loading prompts and conversation history
|
|
35
|
+
* - Collecting available tools from modules
|
|
36
|
+
* - Executing model inference with tool support
|
|
37
|
+
* - Processing tool calls iteratively until completion
|
|
38
|
+
*
|
|
39
|
+
* @param query - The user's input query
|
|
40
|
+
* @param threadId - Thread identifier for context
|
|
41
|
+
* @param thread - Previous conversation history
|
|
42
|
+
* @returns Object containing process steps and final response
|
|
43
|
+
*/
|
|
44
|
+
async *intentFulfilling(query, threadId, thread) {
|
|
45
|
+
try {
|
|
46
|
+
const systemPrompt = `
|
|
47
|
+
Today is ${(/* @__PURE__ */ new Date()).toLocaleDateString()}.
|
|
48
|
+
|
|
49
|
+
${_optionalChain([this, 'access', _ => _.prompts, 'optionalAccess', _2 => _2.agent]) || ""}
|
|
50
|
+
|
|
51
|
+
${_optionalChain([this, 'access', _3 => _3.prompts, 'optionalAccess', _4 => _4.system]) || ""}
|
|
52
|
+
`;
|
|
53
|
+
const modelInstance = this.modelModule.getModel();
|
|
54
|
+
const messages = modelInstance.generateMessages({
|
|
55
|
+
query,
|
|
56
|
+
thread,
|
|
57
|
+
systemPrompt: systemPrompt.trim()
|
|
58
|
+
});
|
|
59
|
+
const tools = [];
|
|
60
|
+
if (this.mcpModule) {
|
|
61
|
+
tools.push(...this.mcpModule.getTools());
|
|
62
|
+
}
|
|
63
|
+
if (this.a2aModule) {
|
|
64
|
+
tools.push(...await this.a2aModule.getTools());
|
|
65
|
+
}
|
|
66
|
+
const functions = modelInstance.convertToolsToFunctions(tools);
|
|
67
|
+
const processList = [];
|
|
68
|
+
let didCallTool = false;
|
|
69
|
+
while (true) {
|
|
70
|
+
const responseStream = await modelInstance.fetchStreamWithContextMessage(
|
|
71
|
+
messages,
|
|
72
|
+
functions
|
|
73
|
+
);
|
|
74
|
+
didCallTool = false;
|
|
75
|
+
const assembledToolCalls = [];
|
|
76
|
+
_chunk466ED5WNcjs.loggers.intentStream.debug("messages", { messages });
|
|
77
|
+
for await (const chunk of responseStream) {
|
|
78
|
+
const delta = chunk.delta;
|
|
79
|
+
if (_optionalChain([delta, 'optionalAccess', _5 => _5.tool_calls])) {
|
|
80
|
+
didCallTool = true;
|
|
81
|
+
for (const toolCallDelta of delta.tool_calls) {
|
|
82
|
+
const index = toolCallDelta.index;
|
|
83
|
+
if (!assembledToolCalls[index]) {
|
|
84
|
+
assembledToolCalls[index] = {
|
|
85
|
+
id: "",
|
|
86
|
+
type: "function",
|
|
87
|
+
function: { name: "", arguments: "" }
|
|
88
|
+
};
|
|
89
|
+
}
|
|
90
|
+
if (toolCallDelta.id) {
|
|
91
|
+
assembledToolCalls[index].id = toolCallDelta.id;
|
|
92
|
+
}
|
|
93
|
+
if (_optionalChain([toolCallDelta, 'access', _6 => _6.function, 'optionalAccess', _7 => _7.name])) {
|
|
94
|
+
assembledToolCalls[index].function.name = toolCallDelta.function.name;
|
|
95
|
+
}
|
|
96
|
+
if (_optionalChain([toolCallDelta, 'access', _8 => _8.function, 'optionalAccess', _9 => _9.arguments])) {
|
|
97
|
+
assembledToolCalls[index].function.arguments += toolCallDelta.function.arguments;
|
|
98
|
+
}
|
|
99
|
+
}
|
|
100
|
+
} else if (_optionalChain([chunk, 'access', _10 => _10.delta, 'optionalAccess', _11 => _11.content])) {
|
|
101
|
+
yield {
|
|
102
|
+
event: "text_chunk",
|
|
103
|
+
data: { delta: chunk.delta.content }
|
|
104
|
+
};
|
|
105
|
+
}
|
|
106
|
+
}
|
|
107
|
+
_chunk466ED5WNcjs.loggers.intentStream.debug("assembledToolCalls", {
|
|
108
|
+
assembledToolCalls
|
|
109
|
+
});
|
|
110
|
+
if (didCallTool && assembledToolCalls.length > 0) {
|
|
111
|
+
const messagePayload = _optionalChain([this, 'access', _12 => _12.a2aModule, 'optionalAccess', _13 => _13.getMessagePayload, 'call', _14 => _14(
|
|
112
|
+
query,
|
|
113
|
+
threadId
|
|
114
|
+
)]);
|
|
115
|
+
for (const toolCall of assembledToolCalls) {
|
|
116
|
+
const toolName = toolCall.function.name;
|
|
117
|
+
const toolArgs = JSON.parse(toolCall.function.arguments);
|
|
118
|
+
const selectedTool = tools.filter(
|
|
119
|
+
(tool) => tool.id === toolName
|
|
120
|
+
)[0];
|
|
121
|
+
yield { event: "tool_start", data: { toolName, toolArgs } };
|
|
122
|
+
let toolResult = "";
|
|
123
|
+
if (this.mcpModule && selectedTool.protocol === "MCP" /* MCP */) {
|
|
124
|
+
const toolArgs2 = JSON.parse(toolCall.function.arguments);
|
|
125
|
+
_chunk466ED5WNcjs.loggers.intent.debug("MCP tool call", { toolName, toolArgs: toolArgs2 });
|
|
126
|
+
toolResult = await this.mcpModule.useTool(
|
|
127
|
+
selectedTool,
|
|
128
|
+
toolArgs2
|
|
129
|
+
);
|
|
130
|
+
} else if (this.a2aModule && selectedTool.protocol === "A2A" /* A2A */) {
|
|
131
|
+
toolResult = await this.a2aModule.useTool(
|
|
132
|
+
selectedTool,
|
|
133
|
+
messagePayload,
|
|
134
|
+
threadId
|
|
135
|
+
);
|
|
136
|
+
} else {
|
|
137
|
+
_chunk466ED5WNcjs.loggers.intent.warn(
|
|
138
|
+
`Unrecognized tool type: ${selectedTool.protocol}`
|
|
139
|
+
);
|
|
140
|
+
continue;
|
|
141
|
+
}
|
|
142
|
+
yield {
|
|
143
|
+
event: "tool_output",
|
|
144
|
+
data: { toolName, result: toolResult }
|
|
145
|
+
};
|
|
146
|
+
_chunk466ED5WNcjs.loggers.intent.debug("toolResult", { toolResult });
|
|
147
|
+
processList.push(toolResult);
|
|
148
|
+
modelInstance.appendMessages(messages, toolResult);
|
|
149
|
+
}
|
|
150
|
+
}
|
|
151
|
+
if (!didCallTool) break;
|
|
152
|
+
}
|
|
153
|
+
} catch (error) {
|
|
154
|
+
_chunk466ED5WNcjs.loggers.intent.error("Error in intentFulfilling generator", { error });
|
|
155
|
+
if (error instanceof Error) {
|
|
156
|
+
yield {
|
|
157
|
+
event: "error",
|
|
158
|
+
data: { message: error.message || "An unknown error occurred." }
|
|
159
|
+
};
|
|
160
|
+
} else {
|
|
161
|
+
yield {
|
|
162
|
+
event: "error",
|
|
163
|
+
data: { message: "An unknown error occurred." }
|
|
164
|
+
};
|
|
165
|
+
}
|
|
166
|
+
}
|
|
167
|
+
}
|
|
168
|
+
/**
|
|
169
|
+
* Generates a title for the conversation based on the query.
|
|
170
|
+
*
|
|
171
|
+
* @param query - The user's input query
|
|
172
|
+
* @returns Promise resolving to a generated title
|
|
173
|
+
*/
|
|
174
|
+
async generateTitle(query) {
|
|
175
|
+
const DEFAULT_TITLE = "New Chat";
|
|
176
|
+
try {
|
|
177
|
+
const modelInstance = this.modelModule.getModel();
|
|
178
|
+
const messages = modelInstance.generateMessages({
|
|
179
|
+
query,
|
|
180
|
+
systemPrompt: `You are a helpful assistant that generates titles for conversations.
|
|
181
|
+
Please analyze the user's query and create a concise title that accurately reflects the conversation's core topic.
|
|
182
|
+
The title must be no more than 5 words long.
|
|
183
|
+
Respond with only the title. Do not include any punctuation or extra explanations.`
|
|
184
|
+
});
|
|
185
|
+
const response = await modelInstance.fetch(messages);
|
|
186
|
+
return response.content || DEFAULT_TITLE;
|
|
187
|
+
} catch (error) {
|
|
188
|
+
_chunk466ED5WNcjs.loggers.intentStream.error("Error generating title", {
|
|
189
|
+
error,
|
|
190
|
+
query
|
|
191
|
+
});
|
|
192
|
+
return DEFAULT_TITLE;
|
|
193
|
+
}
|
|
194
|
+
}
|
|
195
|
+
/**
|
|
196
|
+
* Main entry point for processing user queries.
|
|
197
|
+
*
|
|
198
|
+
* Handles the complete query lifecycle:
|
|
199
|
+
* 1. Loads thread from memory
|
|
200
|
+
* 2. Detects intent from the query
|
|
201
|
+
* 3. Fulfills the intent with AI response
|
|
202
|
+
* 4. Updates conversation history
|
|
203
|
+
*
|
|
204
|
+
* @param type - The type of thread (e.g., chat, workflow)
|
|
205
|
+
* @param userId - The user's unique identifier
|
|
206
|
+
* @param threadId - Unique thread identifier
|
|
207
|
+
* @param query - The user's input query
|
|
208
|
+
* @returns Object containing the AI-generated response
|
|
209
|
+
*/
|
|
210
|
+
async *handleQueryStream(threadMetadata, query) {
|
|
211
|
+
const { type, userId } = threadMetadata;
|
|
212
|
+
const queryStartAt = Date.now();
|
|
213
|
+
const threadMemory = _optionalChain([this, 'access', _15 => _15.memoryModule, 'optionalAccess', _16 => _16.getThreadMemory, 'call', _17 => _17()]);
|
|
214
|
+
let threadId = threadMetadata.threadId;
|
|
215
|
+
let thread;
|
|
216
|
+
if (threadId) {
|
|
217
|
+
thread = await _optionalChain([threadMemory, 'optionalAccess', _18 => _18.getThread, 'call', _19 => _19(type, userId, threadId)]);
|
|
218
|
+
} else {
|
|
219
|
+
threadId = _crypto.randomUUID.call(void 0, );
|
|
220
|
+
const title = await this.generateTitle(query);
|
|
221
|
+
const metadata = await _optionalChain([threadMemory, 'optionalAccess', _20 => _20.createThread, 'call', _21 => _21(type, userId, threadId, title)]) || {
|
|
222
|
+
type,
|
|
223
|
+
threadId,
|
|
224
|
+
title,
|
|
225
|
+
updatedAt: Date.now()
|
|
226
|
+
};
|
|
227
|
+
_chunk466ED5WNcjs.loggers.intentStream.info("Create new thread", { metadata });
|
|
228
|
+
yield { event: "thread_id", data: metadata };
|
|
229
|
+
}
|
|
230
|
+
const _intent = await this.intentTriggering(query);
|
|
231
|
+
try {
|
|
232
|
+
const stream = this.intentFulfilling(query, threadId, thread);
|
|
233
|
+
let finalResponseText = "";
|
|
234
|
+
for await (const event of stream) {
|
|
235
|
+
if (event.event === "text_chunk" && event.data.delta) {
|
|
236
|
+
_chunk466ED5WNcjs.loggers.intentStream.debug("text_chunk", { event });
|
|
237
|
+
finalResponseText += event.data.delta;
|
|
238
|
+
}
|
|
239
|
+
yield event;
|
|
240
|
+
}
|
|
241
|
+
if (userId) {
|
|
242
|
+
await _optionalChain([threadMemory, 'optionalAccess', _22 => _22.addMessageToThread, 'call', _23 => _23(userId, threadId, {
|
|
243
|
+
role: "USER" /* USER */,
|
|
244
|
+
timestamp: queryStartAt,
|
|
245
|
+
content: { type: "text", parts: [query] }
|
|
246
|
+
})]);
|
|
247
|
+
await _optionalChain([threadMemory, 'optionalAccess', _24 => _24.addMessageToThread, 'call', _25 => _25(userId, threadId, {
|
|
248
|
+
role: "MODEL" /* MODEL */,
|
|
249
|
+
timestamp: Date.now(),
|
|
250
|
+
content: { type: "text", parts: [finalResponseText] }
|
|
251
|
+
})]);
|
|
252
|
+
}
|
|
253
|
+
} catch (error) {
|
|
254
|
+
const message = error instanceof Error ? error.message : "Stream failed";
|
|
255
|
+
_chunk466ED5WNcjs.loggers.intentStream.error(message, { error });
|
|
256
|
+
yield {
|
|
257
|
+
event: "error",
|
|
258
|
+
data: { message }
|
|
259
|
+
};
|
|
260
|
+
}
|
|
261
|
+
}
|
|
262
|
+
};
|
|
263
|
+
|
|
264
|
+
|
|
265
|
+
|
|
266
|
+
exports.QueryStreamService = QueryStreamService;
|
|
267
|
+
//# sourceMappingURL=chunk-XPD3ANDG.cjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-XPD3ANDG.cjs","../../src/services/query-stream.service.ts"],"names":["toolArgs"],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACA;ACJA,gCAA2B;AA8BpB,IAAM,mBAAA,EAAN,MAAyB;AAAA,EACvB;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EAER,WAAA,CACC,WAAA,EACA,SAAA,EACA,SAAA,EACA,YAAA,EACA,OAAA,EACC;AACD,IAAA,IAAA,CAAK,YAAA,EAAc,WAAA;AACnB,IAAA,IAAA,CAAK,UAAA,EAAY,SAAA;AACjB,IAAA,IAAA,CAAK,UAAA,EAAY,SAAA;AACjB,IAAA,IAAA,CAAK,aAAA,EAAe,YAAA;AACpB,IAAA,IAAA,CAAK,QAAA,EAAU,OAAA;AAAA,EAChB;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EASA,MAAc,gBAAA,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,MAAA,CAAc,gBAAA,CACb,KAAA,EACA,QAAA,EACA,MAAA,EAC8B;AAC9B,IAAA,IAAI;AACH,MAAA,MAAM,aAAA,EAAe,CAAA;AAAA,SAAA,EAAA,iBACb,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;AAGzB,MAAA,MAAM,cAAA,EAAgB,IAAA,CAAK,WAAA,CAAY,QAAA,CAAS,CAAA;AAChD,MAAA,MAAM,SAAA,EAAW,aAAA,CAAc,gBAAA,CAAiB;AAAA,QAC/C,KAAA;AAAA,QACA,MAAA;AAAA,QACA,YAAA,EAAc,YAAA,CAAa,IAAA,CAAK;AAAA,MACjC,CAAC,CAAA;AAED,MAAA,MAAM,MAAA,EAAsB,CAAC,CAAA;AAC7B,MAAA,GAAA,CAAI,IAAA,CAAK,SAAA,EAAW;AACnB,QAAA,KAAA,CAAM,IAAA,CAAK,GAAG,IAAA,CAAK,SAAA,CAAU,QAAA,CAAS,CAAC,CAAA;AAAA,MACxC;AACA,MAAA,GAAA,CAAI,IAAA,CAAK,SAAA,EAAW;AACnB,QAAA,KAAA,CAAM,IAAA,CAAK,GAAI,MAAM,IAAA,CAAK,SAAA,CAAU,QAAA,CAAS,CAAE,CAAA;AAAA,MAChD;AACA,MAAA,MAAM,UAAA,EAAY,aAAA,CAAc,uBAAA,CAAwB,KAAK,CAAA;AAE7D,MAAA,MAAM,YAAA,EAAwB,CAAC,CAAA;AAC/B,MAAA,IAAI,YAAA,EAAc,KAAA;AAElB,MAAA,MAAA,CAAO,IAAA,EAAM;AACZ,QAAA,MAAM,eAAA,EACL,MAAM,aAAA,CAAc,6BAAA;AAAA,UACnB,QAAA;AAAA,UACA;AAAA,QACD,CAAA;AACD,QAAA,YAAA,EAAc,KAAA;AAEd,QAAA,MAAM,mBAAA,EAIA,CAAC,CAAA;AAEP,QAAA,yBAAA,CAAQ,YAAA,CAAa,KAAA,CAAM,UAAA,EAAY,EAAE,SAAS,CAAC,CAAA;AAEnD,QAAA,IAAA,MAAA,CAAA,MAAiB,MAAA,GAAS,cAAA,EAAgB;AACzC,UAAA,MAAM,MAAA,EAAQ,KAAA,CAAM,KAAA;AACpB,UAAA,GAAA,iBAAI,KAAA,6BAAO,YAAA,EAAY;AACtB,YAAA,YAAA,EAAc,IAAA;AACd,YAAA,IAAA,CAAA,MAAW,cAAA,GAAiB,KAAA,CAAM,UAAA,EAAY;AAC7C,cAAA,MAAM,MAAA,EAAQ,aAAA,CAAc,KAAA;AAE5B,cAAA,GAAA,CAAI,CAAC,kBAAA,CAAmB,KAAK,CAAA,EAAG;AAC/B,gBAAA,kBAAA,CAAmB,KAAK,EAAA,EAAI;AAAA,kBAC3B,EAAA,EAAI,EAAA;AAAA,kBACJ,IAAA,EAAM,UAAA;AAAA,kBACN,QAAA,EAAU,EAAE,IAAA,EAAM,EAAA,EAAI,SAAA,EAAW,GAAG;AAAA,gBACrC,CAAA;AAAA,cACD;AAEA,cAAA,GAAA,CAAI,aAAA,CAAc,EAAA,EAAI;AACrB,gBAAA,kBAAA,CAAmB,KAAK,CAAA,CAAE,GAAA,EAAK,aAAA,CAAc,EAAA;AAAA,cAC9C;AAEA,cAAA,GAAA,iBAAI,aAAA,qBAAc,QAAA,6BAAU,MAAA,EAAM;AACjC,gBAAA,kBAAA,CAAmB,KAAK,CAAA,CAAE,QAAA,CAAS,KAAA,EAClC,aAAA,CAAc,QAAA,CAAS,IAAA;AAAA,cACzB;AAEA,cAAA,GAAA,iBAAI,aAAA,qBAAc,QAAA,6BAAU,WAAA,EAAW;AACtC,gBAAA,kBAAA,CAAmB,KAAK,CAAA,CAAE,QAAA,CAAS,UAAA,GAClC,aAAA,CAAc,QAAA,CAAS,SAAA;AAAA,cACzB;AAAA,YACD;AAAA,UACD,EAAA,KAAA,GAAA,iBAAW,KAAA,uBAAM,KAAA,+BAAO,SAAA,EAAS;AAChC,YAAA,MAAM;AAAA,cACL,KAAA,EAAO,YAAA;AAAA,cACP,IAAA,EAAM,EAAE,KAAA,EAAO,KAAA,CAAM,KAAA,CAAM,QAAQ;AAAA,YACpC,CAAA;AAAA,UACD;AAAA,QACD;AAEA,QAAA,yBAAA,CAAQ,YAAA,CAAa,KAAA,CAAM,oBAAA,EAAsB;AAAA,UAChD;AAAA,QACD,CAAC,CAAA;AAED,QAAA,GAAA,CAAI,YAAA,GAAe,kBAAA,CAAmB,OAAA,EAAS,CAAA,EAAG;AACjD,UAAA,MAAM,eAAA,kBAAiB,IAAA,uBAAK,SAAA,+BAAW,iBAAA;AAAA,YACtC,KAAA;AAAA,YACA;AAAA,UACD,GAAA;AACA,UAAA,IAAA,CAAA,MAAW,SAAA,GAAY,kBAAA,EAAoB;AAC1C,YAAA,MAAM,SAAA,EAAW,QAAA,CAAS,QAAA,CAAS,IAAA;AACnC,YAAA,MAAM,SAAA,EAAW,IAAA,CAAK,KAAA,CAAM,QAAA,CAAS,QAAA,CAAS,SAAS,CAAA;AACvD,YAAA,MAAM,aAAA,EAAe,KAAA,CAAM,MAAA;AAAA,cAC1B,CAAC,IAAA,EAAA,GAAS,IAAA,CAAK,GAAA,IAAO;AAAA,YACvB,CAAA,CAAE,CAAC,CAAA;AAEH,YAAA,MAAM,EAAE,KAAA,EAAO,YAAA,EAAc,IAAA,EAAM,EAAE,QAAA,EAAU,SAAS,EAAE,CAAA;AAE1D,YAAA,IAAI,WAAA,EAAa,EAAA;AAEjB,YAAA,GAAA,CACC,IAAA,CAAK,UAAA,GACL,YAAA,CAAa,SAAA,IAAA,eAAA,EACZ;AACD,cAAA,MAAMA,UAAAA,EAAW,IAAA,CAAK,KAAA,CAAM,QAAA,CAAS,QAAA,CAAS,SAAS,CAAA;AAGvD,cAAA,yBAAA,CAAQ,MAAA,CAAO,KAAA,CAAM,eAAA,EAAiB,EAAE,QAAA,EAAU,QAAA,EAAAA,UAAS,CAAC,CAAA;AAC5D,cAAA,WAAA,EAAa,MAAM,IAAA,CAAK,SAAA,CAAU,OAAA;AAAA,gBACjC,YAAA;AAAA,gBACAA;AAAA,cACD,CAAA;AAAA,YACD,EAAA,KAAA,GAAA,CACC,IAAA,CAAK,UAAA,GACL,YAAA,CAAa,SAAA,IAAA,eAAA,EACZ;AACD,cAAA,WAAA,EAAa,MAAM,IAAA,CAAK,SAAA,CAAU,OAAA;AAAA,gBACjC,YAAA;AAAA,gBACA,cAAA;AAAA,gBACA;AAAA,cACD,CAAA;AAAA,YACD,EAAA,KAAO;AAEN,cAAA,yBAAA,CAAQ,MAAA,CAAO,IAAA;AAAA,gBACd,CAAA,wBAAA,EAA2B,YAAA,CAAa,QAAQ,CAAA;AAAA,cAAA;AAEjD,cAAA;AAAA,YAAA;AAED,YAAA;AAAM,cAAA;AACE,cAAA;AAC8B,YAAA;AAEtC,YAAA;AAEA,YAAA;AACA,YAAA;AAAiD,UAAA;AAClD,QAAA;AAGD,QAAA;AAAkB,MAAA;AACnB,IAAA;AAEA,MAAA;AACA,MAAA;AACC,QAAA;AAAM,UAAA;AACE,UAAA;AACwD,QAAA;AAChE,MAAA;AAEA,QAAA;AAAM,UAAA;AACE,UAAA;AACuC,QAAA;AAC/C,MAAA;AACD,IAAA;AACD,EAAA;AACD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAAA;AAUC,IAAA;AACA,IAAA;AACC,MAAA;AACA,MAAA;AAAgD,QAAA;AAC/C,QAAA;AACc;AAAA;AAAA,oFAAA;AAAA,MAAA;AAKf,MAAA;AACA,MAAA;AAA2B,IAAA;AAE3B,MAAA;AAAqD,QAAA;AACpD,QAAA;AACA,MAAA;AAED,MAAA;AAAO,IAAA;AACR,EAAA;AACD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAAA;AAyBC,IAAA;AACA,IAAA;AACA,IAAA;AAEA,IAAA;AACA,IAAA;AAEA,IAAA;AACC,MAAA;AAA6D,IAAA;AAE7D,MAAA;AACA,MAAA;AAEA,MAAA;AAEE,QAAA;AACA,QAAA;AACA,QAAA;AACA,QAAA;AACoB,MAAA;AAEtB,MAAA;AACA,MAAA;AAA2C,IAAA;AAI5C,IAAA;AAEA,IAAA;AAEC,MAAA;AAEA,MAAA;AACA,MAAA;AACC,QAAA;AACC,UAAA;AACA,UAAA;AAAgC,QAAA;AAEjC,QAAA;AAAM,MAAA;AAGP,MAAA;AACC,QAAA;AAAyD,UAAA;AACxD,UAAA;AACW,UAAA;AAC6B,QAAA;AAEzC,QAAA;AAAyD,UAAA;AACxD,UAAA;AACoB,UAAA;AACgC,QAAA;AACpD,MAAA;AACF,IAAA;AAEA,MAAA;AACA,MAAA;AACA,MAAA;AAAM,QAAA;AACE,QAAA;AACS,MAAA;AACjB,IAAA;AACD,EAAA;AAEF;AD5FA;AACA;AACA;AACA","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-XPD3ANDG.cjs","sourcesContent":[null,"import { randomUUID } from \"node:crypto\";\nimport type {\n\tA2AModule,\n\tMCPModule,\n\tMemoryModule,\n\tModelModule,\n} from \"@/modules/index.js\";\nimport type { AinAgentPrompts } from \"@/types/agent.js\";\nimport {\n\tMessageRole,\n\ttype ThreadMetadata,\n\ttype ThreadObject,\n\ttype ThreadType,\n} from \"@/types/memory.js\";\nimport type { StreamEvent } from \"@/types/stream\";\nimport {\n\ttype IA2ATool,\n\ttype IAgentTool,\n\ttype IMCPTool,\n\tTOOL_PROTOCOL_TYPE,\n} from \"@/types/tool.js\";\nimport { loggers } from \"@/utils/logger.js\";\n\n/**\n * Service for processing user queries through the agent's AI pipeline.\n *\n * Orchestrates the query processing workflow including intent detection,\n * model inference, tool execution, and response generation. Manages\n * conversation context and coordinates between different modules.\n */\nexport class QueryStreamService {\n\tprivate modelModule: ModelModule;\n\tprivate a2aModule?: A2AModule;\n\tprivate mcpModule?: MCPModule;\n\tprivate memoryModule?: MemoryModule;\n\tprivate prompts?: AinAgentPrompts;\n\n\tconstructor(\n\t\tmodelModule: ModelModule,\n\t\ta2aModule?: A2AModule,\n\t\tmcpModule?: MCPModule,\n\t\tmemoryModule?: MemoryModule,\n\t\tprompts?: AinAgentPrompts,\n\t) {\n\t\tthis.modelModule = modelModule;\n\t\tthis.a2aModule = a2aModule;\n\t\tthis.mcpModule = mcpModule;\n\t\tthis.memoryModule = memoryModule;\n\t\tthis.prompts = prompts;\n\t}\n\n\t/**\n\t * Detects the intent from 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\tpublic async *intentFulfilling(\n\t\tquery: string,\n\t\tthreadId: string,\n\t\tthread?: ThreadObject,\n\t): AsyncGenerator<StreamEvent> {\n\t\ttry {\n\t\t\tconst systemPrompt = `\nToday is ${new Date().toLocaleDateString()}.\n\n${this.prompts?.agent || \"\"}\n\n${this.prompts?.system || \"\"}\n `;\n\n\t\t\tconst modelInstance = this.modelModule.getModel();\n\t\t\tconst messages = modelInstance.generateMessages({\n\t\t\t\tquery,\n\t\t\t\tthread,\n\t\t\t\tsystemPrompt: systemPrompt.trim(),\n\t\t\t});\n\n\t\t\tconst tools: IAgentTool[] = [];\n\t\t\tif (this.mcpModule) {\n\t\t\t\ttools.push(...this.mcpModule.getTools());\n\t\t\t}\n\t\t\tif (this.a2aModule) {\n\t\t\t\ttools.push(...(await this.a2aModule.getTools()));\n\t\t\t}\n\t\t\tconst functions = modelInstance.convertToolsToFunctions(tools);\n\n\t\t\tconst processList: string[] = [];\n\t\t\tlet didCallTool = false;\n\n\t\t\twhile (true) {\n\t\t\t\tconst responseStream =\n\t\t\t\t\tawait modelInstance.fetchStreamWithContextMessage(\n\t\t\t\t\t\tmessages,\n\t\t\t\t\t\tfunctions,\n\t\t\t\t\t);\n\t\t\t\tdidCallTool = false;\n\n\t\t\t\tconst assembledToolCalls: {\n\t\t\t\t\tid: string;\n\t\t\t\t\ttype: \"function\";\n\t\t\t\t\tfunction: { name: string; arguments: string };\n\t\t\t\t}[] = [];\n\n\t\t\t\tloggers.intentStream.debug(\"messages\", { messages });\n\n\t\t\t\tfor await (const chunk of responseStream) {\n\t\t\t\t\tconst delta = chunk.delta;\n\t\t\t\t\tif (delta?.tool_calls) {\n\t\t\t\t\t\tdidCallTool = true;\n\t\t\t\t\t\tfor (const toolCallDelta of delta.tool_calls) {\n\t\t\t\t\t\t\tconst index = toolCallDelta.index;\n\n\t\t\t\t\t\t\tif (!assembledToolCalls[index]) {\n\t\t\t\t\t\t\t\tassembledToolCalls[index] = {\n\t\t\t\t\t\t\t\t\tid: \"\",\n\t\t\t\t\t\t\t\t\ttype: \"function\",\n\t\t\t\t\t\t\t\t\tfunction: { name: \"\", arguments: \"\" },\n\t\t\t\t\t\t\t\t};\n\t\t\t\t\t\t\t}\n\n\t\t\t\t\t\t\tif (toolCallDelta.id) {\n\t\t\t\t\t\t\t\tassembledToolCalls[index].id = toolCallDelta.id;\n\t\t\t\t\t\t\t}\n\n\t\t\t\t\t\t\tif (toolCallDelta.function?.name) {\n\t\t\t\t\t\t\t\tassembledToolCalls[index].function.name =\n\t\t\t\t\t\t\t\t\ttoolCallDelta.function.name;\n\t\t\t\t\t\t\t}\n\n\t\t\t\t\t\t\tif (toolCallDelta.function?.arguments) {\n\t\t\t\t\t\t\t\tassembledToolCalls[index].function.arguments +=\n\t\t\t\t\t\t\t\t\ttoolCallDelta.function.arguments;\n\t\t\t\t\t\t\t}\n\t\t\t\t\t\t}\n\t\t\t\t\t} else if (chunk.delta?.content) {\n\t\t\t\t\t\tyield {\n\t\t\t\t\t\t\tevent: \"text_chunk\",\n\t\t\t\t\t\t\tdata: { delta: chunk.delta.content },\n\t\t\t\t\t\t};\n\t\t\t\t\t}\n\t\t\t\t}\n\n\t\t\t\tloggers.intentStream.debug(\"assembledToolCalls\", {\n\t\t\t\t\tassembledToolCalls,\n\t\t\t\t});\n\n\t\t\t\tif (didCallTool && assembledToolCalls.length > 0) {\n\t\t\t\t\tconst messagePayload = this.a2aModule?.getMessagePayload(\n\t\t\t\t\t\tquery,\n\t\t\t\t\t\tthreadId,\n\t\t\t\t\t);\n\t\t\t\t\tfor (const toolCall of assembledToolCalls) {\n\t\t\t\t\t\tconst toolName = toolCall.function.name;\n\t\t\t\t\t\tconst toolArgs = JSON.parse(toolCall.function.arguments);\n\t\t\t\t\t\tconst selectedTool = tools.filter(\n\t\t\t\t\t\t\t(tool) => tool.id === toolName,\n\t\t\t\t\t\t)[0];\n\n\t\t\t\t\t\tyield { event: \"tool_start\", data: { toolName, toolArgs } };\n\n\t\t\t\t\t\tlet toolResult = \"\";\n\n\t\t\t\t\t\tif (\n\t\t\t\t\t\t\tthis.mcpModule &&\n\t\t\t\t\t\t\tselectedTool.protocol === TOOL_PROTOCOL_TYPE.MCP\n\t\t\t\t\t\t) {\n\t\t\t\t\t\t\tconst toolArgs = JSON.parse(toolCall.function.arguments) as\n\t\t\t\t\t\t\t\t| { [x: string]: unknown }\n\t\t\t\t\t\t\t\t| undefined;\n\t\t\t\t\t\t\tloggers.intent.debug(\"MCP tool call\", { toolName, toolArgs });\n\t\t\t\t\t\t\ttoolResult = await this.mcpModule.useTool(\n\t\t\t\t\t\t\t\tselectedTool as IMCPTool,\n\t\t\t\t\t\t\t\ttoolArgs,\n\t\t\t\t\t\t\t);\n\t\t\t\t\t\t} else if (\n\t\t\t\t\t\t\tthis.a2aModule &&\n\t\t\t\t\t\t\tselectedTool.protocol === TOOL_PROTOCOL_TYPE.A2A\n\t\t\t\t\t\t) {\n\t\t\t\t\t\t\ttoolResult = await this.a2aModule.useTool(\n\t\t\t\t\t\t\t\tselectedTool as IA2ATool,\n\t\t\t\t\t\t\t\tmessagePayload!,\n\t\t\t\t\t\t\t\tthreadId,\n\t\t\t\t\t\t\t);\n\t\t\t\t\t\t} else {\n\t\t\t\t\t\t\t// Unrecognized tool type. It cannot be happened...\n\t\t\t\t\t\t\tloggers.intent.warn(\n\t\t\t\t\t\t\t\t`Unrecognized tool type: ${selectedTool.protocol}`,\n\t\t\t\t\t\t\t);\n\t\t\t\t\t\t\tcontinue;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tyield {\n\t\t\t\t\t\t\tevent: \"tool_output\",\n\t\t\t\t\t\t\tdata: { toolName, result: toolResult },\n\t\t\t\t\t\t};\n\t\t\t\t\t\tloggers.intent.debug(\"toolResult\", { toolResult });\n\n\t\t\t\t\t\tprocessList.push(toolResult);\n\t\t\t\t\t\tmodelInstance.appendMessages(messages, toolResult);\n\t\t\t\t\t}\n\t\t\t\t}\n\n\t\t\t\tif (!didCallTool) break;\n\t\t\t}\n\t\t} catch (error) {\n\t\t\tloggers.intent.error(\"Error in intentFulfilling generator\", { error });\n\t\t\tif (error instanceof Error) {\n\t\t\t\tyield {\n\t\t\t\t\tevent: \"error\",\n\t\t\t\t\tdata: { message: error.message || \"An unknown error occurred.\" },\n\t\t\t\t};\n\t\t\t} else {\n\t\t\t\tyield {\n\t\t\t\t\tevent: \"error\",\n\t\t\t\t\tdata: { message: \"An unknown error occurred.\" },\n\t\t\t\t};\n\t\t\t}\n\t\t}\n\t}\n\n\t/**\n\t * Generates a title for the conversation based on the query.\n\t *\n\t * @param query - The user's input query\n\t * @returns Promise resolving to a generated title\n\t */\n\n\tprivate async generateTitle(query: string): Promise<string> {\n\t\tconst DEFAULT_TITLE = \"New Chat\";\n\t\ttry {\n\t\t\tconst modelInstance = this.modelModule.getModel();\n\t\t\tconst messages = modelInstance.generateMessages({\n\t\t\t\tquery,\n\t\t\t\tsystemPrompt: `You are a helpful assistant that generates titles for conversations.\n Please analyze the user's query and create a concise title that accurately reflects the conversation's core topic.\n The title must be no more than 5 words long.\n Respond with only the title. Do not include any punctuation or extra explanations.`,\n\t\t\t});\n\t\t\tconst response = await modelInstance.fetch(messages);\n\t\t\treturn response.content || DEFAULT_TITLE;\n\t\t} catch (error) {\n\t\t\tloggers.intentStream.error(\"Error generating title\", {\n\t\t\t\terror,\n\t\t\t\tquery,\n\t\t\t});\n\t\t\treturn DEFAULT_TITLE;\n\t\t}\n\t}\n\n\t/**\n\t * Main entry point for processing user queries.\n\t *\n\t * Handles the complete query lifecycle:\n\t * 1. Loads thread 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 type - The type of thread (e.g., chat, workflow)\n\t * @param userId - The user's unique identifier\n\t * @param threadId - Unique thread identifier\n\t * @param query - The user's input query\n\t * @returns Object containing the AI-generated response\n\t */\n\tpublic async *handleQueryStream(\n\t\tthreadMetadata: {\n\t\t\ttype: ThreadType;\n\t\t\tuserId: string;\n\t\t\tthreadId?: string;\n\t\t},\n\t\tquery: string,\n\t): AsyncGenerator<StreamEvent> {\n\t\tconst { type, userId } = threadMetadata;\n\t\tconst queryStartAt = Date.now();\n\t\tconst threadMemory = this.memoryModule?.getThreadMemory();\n\n\t\tlet threadId = threadMetadata.threadId;\n\t\tlet thread: ThreadObject | undefined;\n\n\t\tif (threadId) {\n\t\t\tthread = await threadMemory?.getThread(type, userId, threadId);\n\t\t} else {\n\t\t\tthreadId = randomUUID();\n\t\t\tconst title = await this.generateTitle(query);\n\n\t\t\tconst metadata =\n\t\t\t\t(await threadMemory?.createThread(type, userId, threadId, title)) ||\n\t\t\t\t({\n\t\t\t\t\ttype,\n\t\t\t\t\tthreadId,\n\t\t\t\t\ttitle,\n\t\t\t\t\tupdatedAt: Date.now(),\n\t\t\t\t} as ThreadMetadata);\n\t\t\tloggers.intentStream.info(\"Create new thread\", { metadata });\n\t\t\tyield { event: \"thread_id\", data: metadata };\n\t\t}\n\n\t\t// 2. intent triggering\n\t\tconst _intent = await this.intentTriggering(query);\n\n\t\ttry {\n\t\t\t// 3. intent fulfillment\n\t\t\tconst stream = this.intentFulfilling(query, threadId, thread);\n\n\t\t\tlet finalResponseText = \"\";\n\t\t\tfor await (const event of stream) {\n\t\t\t\tif (event.event === \"text_chunk\" && event.data.delta) {\n\t\t\t\t\tloggers.intentStream.debug(\"text_chunk\", { event });\n\t\t\t\t\tfinalResponseText += event.data.delta;\n\t\t\t\t}\n\t\t\t\tyield event;\n\t\t\t}\n\n\t\t\tif (userId) {\n\t\t\t\tawait threadMemory?.addMessageToThread(userId, threadId, {\n\t\t\t\t\trole: MessageRole.USER,\n\t\t\t\t\ttimestamp: queryStartAt,\n\t\t\t\t\tcontent: { type: \"text\", parts: [query] },\n\t\t\t\t});\n\t\t\t\tawait threadMemory?.addMessageToThread(userId, threadId, {\n\t\t\t\t\trole: MessageRole.MODEL,\n\t\t\t\t\ttimestamp: Date.now(),\n\t\t\t\t\tcontent: { type: \"text\", parts: [finalResponseText] },\n\t\t\t\t});\n\t\t\t}\n\t\t} catch (error) {\n\t\t\tconst message = error instanceof Error ? error.message : \"Stream failed\";\n\t\t\tloggers.intentStream.error(message, { error });\n\t\t\tyield {\n\t\t\t\tevent: \"error\",\n\t\t\t\tdata: { message },\n\t\t\t};\n\t\t}\n\t}\n}\n"]}
|