@ainetwork/adk 0.2.7 → 0.2.8
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/dist/cjs/chunk-3SYBDGMH.cjs +37 -0
- package/dist/cjs/chunk-3SYBDGMH.cjs.map +1 -0
- package/dist/cjs/chunk-67EDI57F.cjs +37 -0
- package/dist/cjs/chunk-67EDI57F.cjs.map +1 -0
- package/dist/cjs/chunk-AY3PG2UZ.cjs +126 -0
- package/dist/cjs/chunk-AY3PG2UZ.cjs.map +1 -0
- package/dist/cjs/chunk-B2XCYG4C.cjs +43 -0
- package/dist/cjs/chunk-B2XCYG4C.cjs.map +1 -0
- package/dist/cjs/chunk-CD3RWOCR.cjs +18 -0
- package/dist/cjs/chunk-CD3RWOCR.cjs.map +1 -0
- package/dist/cjs/chunk-IKPQ52KY.cjs +156 -0
- package/dist/cjs/chunk-IKPQ52KY.cjs.map +1 -0
- package/dist/cjs/chunk-J5WX6GPC.cjs +43 -0
- package/dist/cjs/chunk-J5WX6GPC.cjs.map +1 -0
- package/dist/cjs/chunk-JUPAGIP4.cjs +43 -0
- package/dist/cjs/chunk-JUPAGIP4.cjs.map +1 -0
- package/dist/cjs/chunk-KAF2EDPZ.cjs +123 -0
- package/dist/cjs/chunk-KAF2EDPZ.cjs.map +1 -0
- package/dist/cjs/chunk-MXDXXOXE.cjs +126 -0
- package/dist/cjs/chunk-MXDXXOXE.cjs.map +1 -0
- package/dist/cjs/chunk-OAPMGKE2.cjs +123 -0
- package/dist/cjs/chunk-OAPMGKE2.cjs.map +1 -0
- package/dist/cjs/chunk-PNN5WGWA.cjs +43 -0
- package/dist/cjs/chunk-PNN5WGWA.cjs.map +1 -0
- package/dist/cjs/chunk-QCEMS4A3.cjs +168 -0
- package/dist/cjs/chunk-QCEMS4A3.cjs.map +1 -0
- package/dist/cjs/chunk-TYLHXACM.cjs +103 -0
- package/dist/cjs/chunk-TYLHXACM.cjs.map +1 -0
- package/dist/cjs/chunk-W4PRBE6H.cjs +222 -0
- package/dist/cjs/chunk-W4PRBE6H.cjs.map +1 -0
- package/dist/cjs/chunk-WDPCMSZM.cjs +37 -0
- package/dist/cjs/chunk-WDPCMSZM.cjs.map +1 -0
- package/dist/cjs/chunk-WGFYVOP5.cjs +30 -0
- package/dist/cjs/chunk-WGFYVOP5.cjs.map +1 -0
- package/dist/cjs/chunk-YIRGAWMS.cjs +30 -0
- package/dist/cjs/chunk-YIRGAWMS.cjs.map +1 -0
- package/dist/cjs/chunk-ZS3QMRHK.cjs +165 -0
- package/dist/cjs/chunk-ZS3QMRHK.cjs.map +1 -0
- package/dist/cjs/controllers/index.cjs +6 -6
- package/dist/cjs/controllers/index.cjs.map +1 -1
- package/dist/cjs/index.cjs +24 -21
- package/dist/cjs/index.cjs.map +1 -1
- package/dist/cjs/modules/a2a/a2a.module.cjs +3 -3
- package/dist/cjs/modules/index.cjs +10 -10
- package/dist/cjs/modules/index.cjs.map +1 -1
- package/dist/cjs/routes/a2a.routes.cjs +8 -6
- package/dist/cjs/routes/a2a.routes.cjs.map +1 -1
- package/dist/cjs/routes/index.cjs +18 -15
- package/dist/cjs/routes/index.cjs.map +1 -1
- package/dist/cjs/routes/query.routes.cjs +9 -6
- package/dist/cjs/routes/query.routes.cjs.map +1 -1
- package/dist/cjs/services/index.cjs +8 -5
- package/dist/cjs/services/index.cjs.map +1 -1
- package/dist/cjs/services/intents/fulfill-stream.service.cjs +10 -0
- package/dist/cjs/services/intents/fulfill-stream.service.cjs.map +1 -0
- package/dist/cjs/services/intents/fulfill.service.cjs +10 -0
- package/dist/cjs/services/intents/fulfill.service.cjs.map +1 -0
- package/dist/cjs/services/intents/trigger.service.cjs +8 -0
- package/dist/cjs/services/intents/trigger.service.cjs.map +1 -0
- package/dist/cjs/services/query-stream.service.cjs +6 -4
- package/dist/cjs/services/query-stream.service.cjs.map +1 -1
- package/dist/cjs/services/query.service.cjs +6 -4
- package/dist/cjs/services/query.service.cjs.map +1 -1
- package/dist/cjs/types/memory.cjs +2 -2
- package/dist/esm/{chunk-5C53F6HQ.js → chunk-5T7E533R.js} +2 -2
- package/dist/esm/{chunk-NTTSP6VU.js → chunk-6OHUQ6DV.js} +3 -3
- package/dist/esm/{chunk-P4ADSRP5.js → chunk-E2RV322Z.js} +12 -172
- package/dist/esm/chunk-E2RV322Z.js.map +1 -0
- package/dist/esm/chunk-EYZ5Z7FG.js +156 -0
- package/dist/esm/chunk-EYZ5Z7FG.js.map +1 -0
- package/dist/esm/chunk-IRGR2OHS.js +123 -0
- package/dist/esm/chunk-IRGR2OHS.js.map +1 -0
- package/dist/esm/chunk-OCONOO5Z.js +103 -0
- package/dist/esm/chunk-OCONOO5Z.js.map +1 -0
- package/dist/esm/{chunk-NWL3BMUH.js → chunk-SHQWRBWS.js} +7 -3
- package/dist/esm/chunk-SHQWRBWS.js.map +1 -0
- package/dist/esm/{chunk-UQK33CJX.js → chunk-X6JNM3XR.js} +1 -1
- package/dist/esm/{chunk-UQK33CJX.js.map → chunk-X6JNM3XR.js.map} +1 -1
- package/dist/esm/chunk-XTPDMHUI.js +126 -0
- package/dist/esm/chunk-XTPDMHUI.js.map +1 -0
- package/dist/esm/index.js +10 -7
- package/dist/esm/index.js.map +1 -1
- package/dist/esm/modules/a2a/a2a.module.js +2 -2
- package/dist/esm/modules/index.js +5 -5
- package/dist/esm/routes/a2a.routes.js +6 -4
- package/dist/esm/routes/index.js +9 -6
- package/dist/esm/routes/query.routes.js +8 -5
- package/dist/esm/services/index.js +7 -4
- package/dist/esm/services/intents/fulfill-stream.service.d.ts +53 -0
- package/dist/esm/services/intents/fulfill-stream.service.js +10 -0
- package/dist/esm/services/intents/fulfill-stream.service.js.map +1 -0
- package/dist/esm/services/intents/fulfill.service.d.ts +53 -0
- package/dist/esm/services/intents/fulfill.service.js +10 -0
- package/dist/esm/services/intents/fulfill.service.js.map +1 -0
- package/dist/esm/services/intents/trigger.service.d.ts +23 -0
- package/dist/esm/services/intents/trigger.service.js +8 -0
- package/dist/esm/services/intents/trigger.service.js.map +1 -0
- package/dist/esm/services/query-stream.service.d.ts +3 -30
- package/dist/esm/services/query-stream.service.js +5 -3
- package/dist/esm/services/query.service.d.ts +2 -27
- package/dist/esm/services/query.service.js +5 -3
- package/dist/esm/types/memory.d.ts +6 -1
- package/dist/esm/types/memory.js +1 -1
- package/package.json +1 -1
- package/dist/esm/chunk-NWL3BMUH.js.map +0 -1
- package/dist/esm/chunk-P4ADSRP5.js.map +0 -1
- package/dist/esm/chunk-ZBA55AN4.js +0 -257
- package/dist/esm/chunk-ZBA55AN4.js.map +0 -1
- /package/dist/esm/{chunk-5C53F6HQ.js.map → chunk-5T7E533R.js.map} +0 -0
- /package/dist/esm/{chunk-NTTSP6VU.js.map → chunk-6OHUQ6DV.js.map} +0 -0
|
@@ -0,0 +1,37 @@
|
|
|
1
|
+
"use strict";Object.defineProperty(exports, "__esModule", {value: true});
|
|
2
|
+
|
|
3
|
+
var _chunkBDQDZXZIcjs = require('./chunk-BDQDZXZI.cjs');
|
|
4
|
+
|
|
5
|
+
|
|
6
|
+
var _chunkMXDXXOXEcjs = require('./chunk-MXDXXOXE.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 queryStreamService = new (0, _chunkMXDXXOXEcjs.QueryStreamService)(
|
|
18
|
+
agent.modelModule,
|
|
19
|
+
agent.a2aModule,
|
|
20
|
+
agent.mcpModule,
|
|
21
|
+
agent.memoryModule,
|
|
22
|
+
agent.manifest.prompts
|
|
23
|
+
);
|
|
24
|
+
const a2aService = new (0, _chunkBDQDZXZIcjs.A2AService)(queryStreamService);
|
|
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-3SYBDGMH.cjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-3SYBDGMH.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,mBAAA,EAAqB,IAAI,yCAAA;AAAA,IAC9B,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,kBAAkB,CAAA;AACpD,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-3SYBDGMH.cjs","sourcesContent":[null,"import { InMemoryTaskStore } from \"@a2a-js/sdk/server\";\nimport { Router } from \"express\";\nimport type { AINAgent } from \"@/index.js\";\nimport { QueryStreamService } from \"@/services/query-stream.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 queryStreamService = new QueryStreamService(\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(queryStreamService);\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 _chunkBDQDZXZIcjs = require('./chunk-BDQDZXZI.cjs');
|
|
4
|
+
|
|
5
|
+
|
|
6
|
+
var _chunkAY3PG2UZcjs = require('./chunk-AY3PG2UZ.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 queryStreamService = new (0, _chunkAY3PG2UZcjs.QueryStreamService)(
|
|
18
|
+
agent.modelModule,
|
|
19
|
+
agent.a2aModule,
|
|
20
|
+
agent.mcpModule,
|
|
21
|
+
agent.memoryModule,
|
|
22
|
+
agent.manifest.prompts
|
|
23
|
+
);
|
|
24
|
+
const a2aService = new (0, _chunkBDQDZXZIcjs.A2AService)(queryStreamService);
|
|
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-67EDI57F.cjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-67EDI57F.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,mBAAA,EAAqB,IAAI,yCAAA;AAAA,IAC9B,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,kBAAkB,CAAA;AACpD,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-67EDI57F.cjs","sourcesContent":[null,"import { InMemoryTaskStore } from \"@a2a-js/sdk/server\";\nimport { Router } from \"express\";\nimport type { AINAgent } from \"@/index.js\";\nimport { QueryStreamService } from \"@/services/query-stream.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 queryStreamService = new QueryStreamService(\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(queryStreamService);\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,126 @@
|
|
|
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 _chunkW4PRBE6Hcjs = require('./chunk-W4PRBE6H.cjs');
|
|
4
|
+
|
|
5
|
+
|
|
6
|
+
var _chunkTYLHXACMcjs = require('./chunk-TYLHXACM.cjs');
|
|
7
|
+
|
|
8
|
+
|
|
9
|
+
var _chunk466ED5WNcjs = require('./chunk-466ED5WN.cjs');
|
|
10
|
+
|
|
11
|
+
|
|
12
|
+
var _chunkQHQLU7BIcjs = require('./chunk-QHQLU7BI.cjs');
|
|
13
|
+
|
|
14
|
+
// src/services/query-stream.service.ts
|
|
15
|
+
var _crypto = require('crypto');
|
|
16
|
+
var _httpstatuscodes = require('http-status-codes');
|
|
17
|
+
var QueryStreamService = class {
|
|
18
|
+
|
|
19
|
+
|
|
20
|
+
|
|
21
|
+
|
|
22
|
+
constructor(modelModule, a2aModule, mcpModule, memoryModule, prompts) {
|
|
23
|
+
this.modelModule = modelModule;
|
|
24
|
+
this.memoryModule = memoryModule;
|
|
25
|
+
this.intentTriggerService = new (0, _chunkTYLHXACMcjs.IntentTriggerService)(
|
|
26
|
+
modelModule,
|
|
27
|
+
memoryModule
|
|
28
|
+
);
|
|
29
|
+
this.intentFulfillStreamService = new (0, _chunkW4PRBE6Hcjs.IntentFulfillStreamService)(
|
|
30
|
+
modelModule,
|
|
31
|
+
a2aModule,
|
|
32
|
+
mcpModule,
|
|
33
|
+
memoryModule,
|
|
34
|
+
prompts
|
|
35
|
+
);
|
|
36
|
+
}
|
|
37
|
+
/**
|
|
38
|
+
* Generates a title for the conversation based on the query.
|
|
39
|
+
*
|
|
40
|
+
* @param query - The user's input query
|
|
41
|
+
* @returns Promise resolving to a generated title
|
|
42
|
+
*/
|
|
43
|
+
async generateTitle(query) {
|
|
44
|
+
const DEFAULT_TITLE = "New Chat";
|
|
45
|
+
try {
|
|
46
|
+
const modelInstance = this.modelModule.getModel();
|
|
47
|
+
const messages = modelInstance.generateMessages({
|
|
48
|
+
query,
|
|
49
|
+
systemPrompt: `You are a helpful assistant that generates titles for conversations.
|
|
50
|
+
Please analyze the user's query and create a concise title that accurately reflects the conversation's core topic.
|
|
51
|
+
The title must be no more than 5 words long.
|
|
52
|
+
Respond with only the title. Do not include any punctuation or extra explanations.`
|
|
53
|
+
});
|
|
54
|
+
const response = await modelInstance.fetch(messages);
|
|
55
|
+
return response.content || DEFAULT_TITLE;
|
|
56
|
+
} catch (error) {
|
|
57
|
+
_chunk466ED5WNcjs.loggers.intentStream.error("Error generating title", {
|
|
58
|
+
error,
|
|
59
|
+
query
|
|
60
|
+
});
|
|
61
|
+
return DEFAULT_TITLE;
|
|
62
|
+
}
|
|
63
|
+
}
|
|
64
|
+
/**
|
|
65
|
+
* Main entry point for processing streaming user queries.
|
|
66
|
+
*
|
|
67
|
+
* Handles the complete query lifecycle:
|
|
68
|
+
* 1. Loads or creates thread from memory
|
|
69
|
+
* 2. Detects intent from the query
|
|
70
|
+
* 3. Fulfills the intent with streaming AI response
|
|
71
|
+
* 4. Updates conversation history in real-time
|
|
72
|
+
*
|
|
73
|
+
* @param threadMetadata - Metadata containing type, userId, and optional threadId
|
|
74
|
+
* @param threadMetadata.type - The type of thread (e.g., chat, workflow)
|
|
75
|
+
* @param threadMetadata.userId - The user's unique identifier
|
|
76
|
+
* @param threadMetadata.threadId - Optional thread identifier
|
|
77
|
+
* @param query - The user's input query
|
|
78
|
+
* @returns AsyncGenerator yielding StreamEvent objects for SSE
|
|
79
|
+
*/
|
|
80
|
+
async *handleQueryStream(threadMetadata, query, isA2A) {
|
|
81
|
+
const { type, userId } = threadMetadata;
|
|
82
|
+
const threadMemory = _optionalChain([this, 'access', _ => _.memoryModule, 'optionalAccess', _2 => _2.getThreadMemory, 'call', _3 => _3()]);
|
|
83
|
+
let threadId = threadMetadata.threadId;
|
|
84
|
+
let thread;
|
|
85
|
+
if (threadId) {
|
|
86
|
+
thread = await _optionalChain([threadMemory, 'optionalAccess', _4 => _4.getThread, 'call', _5 => _5(userId, threadId)]);
|
|
87
|
+
if (!thread && !isA2A) {
|
|
88
|
+
throw new (0, _chunkQHQLU7BIcjs.AinHttpError)(_httpstatuscodes.StatusCodes.NOT_FOUND, "Thread not found");
|
|
89
|
+
}
|
|
90
|
+
}
|
|
91
|
+
threadId ??= _crypto.randomUUID.call(void 0, );
|
|
92
|
+
if (!thread) {
|
|
93
|
+
const title = await this.generateTitle(query);
|
|
94
|
+
const metadata = await _optionalChain([threadMemory, 'optionalAccess', _6 => _6.createThread, 'call', _7 => _7(
|
|
95
|
+
type,
|
|
96
|
+
userId,
|
|
97
|
+
threadId,
|
|
98
|
+
title
|
|
99
|
+
)]) || { type, userId, threadId, title };
|
|
100
|
+
thread = { ...metadata, messages: [] };
|
|
101
|
+
_chunk466ED5WNcjs.loggers.intent.info(`Create new thread: ${threadId}`);
|
|
102
|
+
yield { event: "thread_id", data: { type, userId, threadId, title } };
|
|
103
|
+
}
|
|
104
|
+
await _optionalChain([threadMemory, 'optionalAccess', _8 => _8.addMessagesToThread, 'call', _9 => _9(userId, threadId, [
|
|
105
|
+
{
|
|
106
|
+
messageId: _crypto.randomUUID.call(void 0, ),
|
|
107
|
+
role: "USER" /* USER */,
|
|
108
|
+
timestamp: Date.now(),
|
|
109
|
+
content: { type: "text", parts: [query] }
|
|
110
|
+
}
|
|
111
|
+
])]);
|
|
112
|
+
const triggeredIntent = await this.intentTriggerService.intentTriggering(query, thread);
|
|
113
|
+
const stream = this.intentFulfillStreamService.intentFulfillStream(
|
|
114
|
+
triggeredIntent,
|
|
115
|
+
thread
|
|
116
|
+
);
|
|
117
|
+
for await (const event of stream) {
|
|
118
|
+
yield event;
|
|
119
|
+
}
|
|
120
|
+
}
|
|
121
|
+
};
|
|
122
|
+
|
|
123
|
+
|
|
124
|
+
|
|
125
|
+
exports.QueryStreamService = QueryStreamService;
|
|
126
|
+
//# sourceMappingURL=chunk-AY3PG2UZ.cjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-AY3PG2UZ.cjs","../../src/services/query-stream.service.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACA;ACbA,gCAA2B;AAC3B,oDAA4B;AA2BrB,IAAM,mBAAA,EAAN,MAAyB;AAAA,EACvB;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,aAAA,EAAe,YAAA;AACpB,IAAA,IAAA,CAAK,qBAAA,EAAuB,IAAI,2CAAA;AAAA,MAC/B,WAAA;AAAA,MACA;AAAA,IACD,CAAA;AACA,IAAA,IAAA,CAAK,2BAAA,EAA6B,IAAI,iDAAA;AAAA,MACrC,WAAA;AAAA,MACA,SAAA;AAAA,MACA,SAAA;AAAA,MACA,YAAA;AAAA,MACA;AAAA,IACD,CAAA;AAAA,EACD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EASA,MAAc,aAAA,CAAc,KAAA,EAAgC;AAC3D,IAAA,MAAM,cAAA,EAAgB,UAAA;AACtB,IAAA,IAAI;AACH,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,YAAA,EAAc,CAAA;AAAA;AAAA;AAAA,oFAAA;AAAA,MAIf,CAAC,CAAA;AACD,MAAA,MAAM,SAAA,EAAW,MAAM,aAAA,CAAc,KAAA,CAAM,QAAQ,CAAA;AACnD,MAAA,OAAO,QAAA,CAAS,QAAA,GAAW,aAAA;AAAA,IAC5B,EAAA,MAAA,CAAS,KAAA,EAAO;AACf,MAAA,yBAAA,CAAQ,YAAA,CAAa,KAAA,CAAM,wBAAA,EAA0B;AAAA,QACpD,KAAA;AAAA,QACA;AAAA,MACD,CAAC,CAAA;AACD,MAAA,OAAO,aAAA;AAAA,IACR;AAAA,EACD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAkBA,MAAA,CAAc,iBAAA,CACb,cAAA,EAKA,KAAA,EACA,KAAA,EAC8B;AAC9B,IAAA,MAAM,EAAE,IAAA,EAAM,OAAO,EAAA,EAAI,cAAA;AACzB,IAAA,MAAM,aAAA,kBAAe,IAAA,mBAAK,YAAA,6BAAc,eAAA,mBAAgB,GAAA;AAGxD,IAAA,IAAI,SAAA,EAAW,cAAA,CAAe,QAAA;AAC9B,IAAA,IAAI,MAAA;AACJ,IAAA,GAAA,CAAI,QAAA,EAAU;AACb,MAAA,OAAA,EAAS,sBAAM,YAAA,6BAAc,SAAA,mBAAU,MAAA,EAAQ,QAAQ,GAAA;AACvD,MAAA,GAAA,CAAI,CAAC,OAAA,GAAU,CAAC,KAAA,EAAO;AACtB,QAAA,MAAM,IAAI,mCAAA,CAAa,4BAAA,CAAY,SAAA,EAAW,kBAAkB,CAAA;AAAA,MACjE;AAAA,IACD;AAEA,IAAA,SAAA,IAAa,gCAAA,CAAW;AACxB,IAAA,GAAA,CAAI,CAAC,MAAA,EAAQ;AACZ,MAAA,MAAM,MAAA,EAAQ,MAAM,IAAA,CAAK,aAAA,CAAc,KAAK,CAAA;AAC5C,MAAA,MAAM,SAAA,EAA4B,sBAAM,YAAA,6BAAc,YAAA;AAAA,QACrD,IAAA;AAAA,QACA,MAAA;AAAA,QACA,QAAA;AAAA,QACA;AAAA,MACD,IAAA,GAAM,EAAE,IAAA,EAAM,MAAA,EAAQ,QAAA,EAAU,MAAM,CAAA;AACtC,MAAA,OAAA,EAAS,EAAE,GAAG,QAAA,EAAU,QAAA,EAAU,CAAC,EAAE,CAAA;AACrC,MAAA,yBAAA,CAAQ,MAAA,CAAO,IAAA,CAAK,CAAA,mBAAA,EAAsB,QAAQ,CAAA,CAAA;AACA,MAAA;AACnD,IAAA;AAGgD,IAAA;AAC/C,MAAA;AACuB,QAAA;AACtB,QAAA;AACoB,QAAA;AACoB,QAAA;AACzC,MAAA;AACA,IAAA;AAIW,IAAA;AAGmC,IAAA;AAC9C,MAAA;AACA,MAAA;AACD,IAAA;AAEkC,IAAA;AAC3B,MAAA;AACP,IAAA;AACD,EAAA;AACD;ADxCyD;AACA;AACA;AACA","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-AY3PG2UZ.cjs","sourcesContent":[null,"import { randomUUID } from \"node:crypto\";\nimport { StatusCodes } from \"http-status-codes\";\nimport type {\n\tA2AModule,\n\tMCPModule,\n\tMemoryModule,\n\tModelModule,\n} from \"@/modules/index.js\";\nimport { type AinAgentPrompts, AinHttpError } from \"@/types/agent.js\";\nimport {\n\tMessageRole,\n\ttype ThreadMetadata,\n\ttype ThreadObject,\n\ttype ThreadType,\n\ttype TriggeredIntent,\n} from \"@/types/memory.js\";\nimport type { StreamEvent } from \"@/types/stream\";\nimport { loggers } from \"@/utils/logger.js\";\nimport { IntentFulfillStreamService } from \"./intents/fulfill-stream.service\";\nimport { IntentTriggerService } from \"./intents/trigger.service\";\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 memoryModule?: MemoryModule;\n\tprivate intentTriggerService: IntentTriggerService;\n\tprivate intentFulfillStreamService: IntentFulfillStreamService;\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.memoryModule = memoryModule;\n\t\tthis.intentTriggerService = new IntentTriggerService(\n\t\t\tmodelModule,\n\t\t\tmemoryModule,\n\t\t);\n\t\tthis.intentFulfillStreamService = new IntentFulfillStreamService(\n\t\t\tmodelModule,\n\t\t\ta2aModule,\n\t\t\tmcpModule,\n\t\t\tmemoryModule,\n\t\t\tprompts,\n\t\t);\n\t}\n\n\t/**\n\t * Generates a title for the conversation based on the query.\n\t *\n\t * @param query - The user's input query\n\t * @returns Promise resolving to a generated title\n\t */\n\n\tprivate async generateTitle(query: string): Promise<string> {\n\t\tconst DEFAULT_TITLE = \"New Chat\";\n\t\ttry {\n\t\t\tconst modelInstance = this.modelModule.getModel();\n\t\t\tconst messages = modelInstance.generateMessages({\n\t\t\t\tquery,\n\t\t\t\tsystemPrompt: `You are a helpful assistant that generates titles for conversations.\n Please analyze the user's query and create a concise title that accurately reflects the conversation's core topic.\n The title must be no more than 5 words long.\n Respond with only the title. Do not include any punctuation or extra explanations.`,\n\t\t\t});\n\t\t\tconst response = await modelInstance.fetch(messages);\n\t\t\treturn response.content || DEFAULT_TITLE;\n\t\t} catch (error) {\n\t\t\tloggers.intentStream.error(\"Error generating title\", {\n\t\t\t\terror,\n\t\t\t\tquery,\n\t\t\t});\n\t\t\treturn DEFAULT_TITLE;\n\t\t}\n\t}\n\n\t/**\n\t * Main entry point for processing streaming user queries.\n\t *\n\t * Handles the complete query lifecycle:\n\t * 1. Loads or creates thread from memory\n\t * 2. Detects intent from the query\n\t * 3. Fulfills the intent with streaming AI response\n\t * 4. Updates conversation history in real-time\n\t *\n\t * @param threadMetadata - Metadata containing type, userId, and optional threadId\n\t * @param threadMetadata.type - The type of thread (e.g., chat, workflow)\n\t * @param threadMetadata.userId - The user's unique identifier\n\t * @param threadMetadata.threadId - Optional thread identifier\n\t * @param query - The user's input query\n\t * @returns AsyncGenerator yielding StreamEvent objects for SSE\n\t */\n\tpublic async *handleQueryStream(\n\t\tthreadMetadata: {\n\t\t\ttype: ThreadType;\n\t\t\tuserId: string;\n\t\t\tthreadId?: string;\n\t\t},\n\t\tquery: string,\n\t\tisA2A?: boolean,\n\t): AsyncGenerator<StreamEvent> {\n\t\tconst { type, userId } = threadMetadata;\n\t\tconst threadMemory = this.memoryModule?.getThreadMemory();\n\n\t\t// 1. Load or create thread\n\t\tlet threadId = threadMetadata.threadId;\n\t\tlet thread: ThreadObject | undefined;\n\t\tif (threadId) {\n\t\t\tthread = await threadMemory?.getThread(userId, threadId);\n\t\t\tif (!thread && !isA2A) {\n\t\t\t\tthrow new AinHttpError(StatusCodes.NOT_FOUND, \"Thread not found\");\n\t\t\t}\n\t\t}\n\n\t\tthreadId ??= randomUUID();\n\t\tif (!thread) {\n\t\t\tconst title = await this.generateTitle(query);\n\t\t\tconst metadata: ThreadMetadata = (await threadMemory?.createThread(\n\t\t\t\ttype,\n\t\t\t\tuserId,\n\t\t\t\tthreadId,\n\t\t\t\ttitle,\n\t\t\t)) || { type, userId, threadId, title };\n\t\t\tthread = { ...metadata, messages: [] };\n\t\t\tloggers.intent.info(`Create new thread: ${threadId}`);\n\t\t\tyield { event: \"thread_id\", data: { type, userId, threadId, title } };\n\t\t}\n\n\t\t// only add for storage, not for inference\n\t\tawait threadMemory?.addMessagesToThread(userId, threadId, [\n\t\t\t{\n\t\t\t\tmessageId: randomUUID(),\n\t\t\t\trole: MessageRole.USER,\n\t\t\t\ttimestamp: Date.now(),\n\t\t\t\tcontent: { type: \"text\", parts: [query] },\n\t\t\t},\n\t\t]);\n\n\t\t// 2. intent triggering\n\t\tconst triggeredIntent: Array<TriggeredIntent> =\n\t\t\tawait this.intentTriggerService.intentTriggering(query, thread);\n\n\t\t// 3. intent fulfillment\n\t\tconst stream = this.intentFulfillStreamService.intentFulfillStream(\n\t\t\ttriggeredIntent,\n\t\t\tthread,\n\t\t);\n\n\t\tfor await (const event of stream) {\n\t\t\tyield event;\n\t\t}\n\t}\n}\n"]}
|
|
@@ -0,0 +1,43 @@
|
|
|
1
|
+
"use strict";Object.defineProperty(exports, "__esModule", {value: true});
|
|
2
|
+
|
|
3
|
+
var _chunkKAF2EDPZcjs = require('./chunk-KAF2EDPZ.cjs');
|
|
4
|
+
|
|
5
|
+
|
|
6
|
+
var _chunkMXDXXOXEcjs = require('./chunk-MXDXXOXE.cjs');
|
|
7
|
+
|
|
8
|
+
|
|
9
|
+
var _chunkVLMHDHXTcjs = require('./chunk-VLMHDHXT.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, _chunkKAF2EDPZcjs.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, _chunkMXDXXOXEcjs.QueryStreamService)(
|
|
25
|
+
agent.modelModule,
|
|
26
|
+
agent.a2aModule,
|
|
27
|
+
agent.mcpModule,
|
|
28
|
+
agent.memoryModule,
|
|
29
|
+
agent.manifest.prompts
|
|
30
|
+
);
|
|
31
|
+
}
|
|
32
|
+
const queryController = new (0, _chunkVLMHDHXTcjs.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-B2XCYG4C.cjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-B2XCYG4C.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-B2XCYG4C.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-CD3RWOCR.cjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-CD3RWOCR.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;AAiDL,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;ADxCZ;AACA;AACE;AACA;AACF,mEAAC","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-CD3RWOCR.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\tmessageId: string;\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\tuserId: string;\n\tthreadId: string;\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 * { messageId: <UUID_1>, role: MessageRole.USER, content: {...}, timestamp: 1234567890 },\n * { messageId: <UUID_2> ,role: MessageRole.MODEL, content: {...}, timestamp: 1234567891 }\n * ]\n * };\n * ```\n */\nexport type ThreadObject = {\n\tuserId: string;\n\tthreadId: string;\n\ttype: ThreadType;\n\ttitle: string;\n\tmessages: Array<MessageObject>;\n};\n\nexport interface Intent {\n\tid: string;\n\tname: string;\n\tdescription: string;\n\tprompt?: string;\n\ttriggeringSentences?: Array<string>;\n\ttags?: Array<string>;\n}\n\nexport type TriggeredIntent = {\n\tsubquery: string;\n\tintent?: Intent;\n\tactionPlan?: string;\n};\n"]}
|
|
@@ -0,0 +1,156 @@
|
|
|
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/intents/fulfill.service.ts
|
|
6
|
+
var _crypto = require('crypto');
|
|
7
|
+
var IntentFulfillService = 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
|
+
async addToThreadMessages(thread, params) {
|
|
21
|
+
try {
|
|
22
|
+
const threadMemory = _optionalChain([this, 'access', _ => _.memoryModule, 'optionalAccess', _2 => _2.getThreadMemory, 'call', _3 => _3()]);
|
|
23
|
+
const { userId, threadId } = thread;
|
|
24
|
+
const newMessage = {
|
|
25
|
+
messageId: _crypto.randomUUID.call(void 0, ),
|
|
26
|
+
role: params.role,
|
|
27
|
+
timestamp: Date.now(),
|
|
28
|
+
content: { type: "text", parts: [params.content] },
|
|
29
|
+
metadata: params.metadata
|
|
30
|
+
};
|
|
31
|
+
thread.messages.push(newMessage);
|
|
32
|
+
await _optionalChain([threadMemory, 'optionalAccess', _4 => _4.addMessagesToThread, 'call', _5 => _5(userId, threadId, [newMessage])]);
|
|
33
|
+
} catch (error) {
|
|
34
|
+
_chunk466ED5WNcjs.loggers.intentStream.error("Error adding message to thread", error);
|
|
35
|
+
}
|
|
36
|
+
}
|
|
37
|
+
/**
|
|
38
|
+
* Fulfills the detected intent by generating a streaming response.
|
|
39
|
+
*
|
|
40
|
+
* Manages the complete inference loop including:
|
|
41
|
+
* - Loading prompts and conversation history
|
|
42
|
+
* - Collecting available tools from modules
|
|
43
|
+
* - Executing model inference with tool support
|
|
44
|
+
* - Processing tool calls iteratively until completion
|
|
45
|
+
* - Streaming results as Server-Sent Events
|
|
46
|
+
*
|
|
47
|
+
* @param query - The user's input query
|
|
48
|
+
* @param threadId - Thread identifier for context
|
|
49
|
+
* @param thread - Previous conversation history
|
|
50
|
+
* @param intent - Optional detected intent with custom prompt
|
|
51
|
+
* @returns AsyncGenerator yielding StreamEvent objects
|
|
52
|
+
*/
|
|
53
|
+
async intentFulfilling(query, thread, intent) {
|
|
54
|
+
const systemPrompt = `
|
|
55
|
+
Today is ${(/* @__PURE__ */ new Date()).toLocaleDateString()}.
|
|
56
|
+
|
|
57
|
+
${_optionalChain([this, 'access', _6 => _6.prompts, 'optionalAccess', _7 => _7.agent]) || ""}
|
|
58
|
+
|
|
59
|
+
${_optionalChain([this, 'access', _8 => _8.prompts, 'optionalAccess', _9 => _9.system]) || ""}
|
|
60
|
+
|
|
61
|
+
${_optionalChain([intent, 'optionalAccess', _10 => _10.prompt]) || ""}
|
|
62
|
+
`.trim();
|
|
63
|
+
const modelInstance = this.modelModule.getModel();
|
|
64
|
+
const messages = modelInstance.generateMessages({
|
|
65
|
+
query,
|
|
66
|
+
thread,
|
|
67
|
+
systemPrompt: systemPrompt.trim()
|
|
68
|
+
});
|
|
69
|
+
const tools = [];
|
|
70
|
+
this.mcpModule && tools.push(...this.mcpModule.getTools());
|
|
71
|
+
this.a2aModule && tools.push(...await this.a2aModule.getTools());
|
|
72
|
+
const functions = modelInstance.convertToolsToFunctions(tools);
|
|
73
|
+
let finalMessage = "";
|
|
74
|
+
while (true) {
|
|
75
|
+
const response = await modelInstance.fetchWithContextMessage(
|
|
76
|
+
messages,
|
|
77
|
+
functions
|
|
78
|
+
);
|
|
79
|
+
_chunk466ED5WNcjs.loggers.intentStream.info("messages", { messages });
|
|
80
|
+
const { content, toolCalls } = response;
|
|
81
|
+
if (toolCalls) {
|
|
82
|
+
for (const toolCall of toolCalls) {
|
|
83
|
+
const toolName = toolCall.name;
|
|
84
|
+
const selectedTool = tools.filter(
|
|
85
|
+
(tool) => tool.toolName === toolName
|
|
86
|
+
)[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(selectedTool, toolArgs);
|
|
92
|
+
} else if (this.a2aModule && selectedTool.protocol === "A2A" /* A2A */) {
|
|
93
|
+
toolResult = await this.a2aModule.useTool(
|
|
94
|
+
selectedTool,
|
|
95
|
+
query,
|
|
96
|
+
thread.threadId
|
|
97
|
+
);
|
|
98
|
+
} else {
|
|
99
|
+
_chunk466ED5WNcjs.loggers.intent.warn(
|
|
100
|
+
`Unrecognized tool type: ${selectedTool.protocol}`
|
|
101
|
+
);
|
|
102
|
+
continue;
|
|
103
|
+
}
|
|
104
|
+
_chunk466ED5WNcjs.loggers.intent.debug("toolResult", { toolResult });
|
|
105
|
+
modelInstance.appendMessages(messages, toolResult);
|
|
106
|
+
}
|
|
107
|
+
} else if (content) {
|
|
108
|
+
finalMessage = content;
|
|
109
|
+
break;
|
|
110
|
+
}
|
|
111
|
+
}
|
|
112
|
+
return finalMessage;
|
|
113
|
+
}
|
|
114
|
+
/**
|
|
115
|
+
* Detects the intent from context.
|
|
116
|
+
*
|
|
117
|
+
* @param intents - The user's input query
|
|
118
|
+
* @param thread - The thread history
|
|
119
|
+
* @returns The detected intent
|
|
120
|
+
*/
|
|
121
|
+
async intentFulfill(intents, thread) {
|
|
122
|
+
let finalResponseText = "";
|
|
123
|
+
for (let i = 0; i < intents.length; i++) {
|
|
124
|
+
const { subquery, intent, actionPlan } = intents[i];
|
|
125
|
+
_chunk466ED5WNcjs.loggers.intent.info(`Process query: ${subquery}, ${_optionalChain([intent, 'optionalAccess', _11 => _11.name])}`);
|
|
126
|
+
_chunk466ED5WNcjs.loggers.intent.info(`Action plan: ${actionPlan}`);
|
|
127
|
+
finalResponseText !== "" && thread.messages.push({
|
|
128
|
+
messageId: _crypto.randomUUID.call(void 0, ),
|
|
129
|
+
role: "MODEL" /* MODEL */,
|
|
130
|
+
timestamp: Date.now(),
|
|
131
|
+
content: { type: "text", parts: [finalResponseText] },
|
|
132
|
+
metadata: { isThinking: true }
|
|
133
|
+
});
|
|
134
|
+
await this.addToThreadMessages(thread, {
|
|
135
|
+
role: "MODEL" /* MODEL */,
|
|
136
|
+
content: subquery,
|
|
137
|
+
metadata: {
|
|
138
|
+
subquery,
|
|
139
|
+
isThinking: true,
|
|
140
|
+
actionPlan
|
|
141
|
+
}
|
|
142
|
+
});
|
|
143
|
+
finalResponseText = await this.intentFulfilling(subquery, thread, intent);
|
|
144
|
+
}
|
|
145
|
+
await this.addToThreadMessages(thread, {
|
|
146
|
+
role: "MODEL" /* MODEL */,
|
|
147
|
+
content: finalResponseText
|
|
148
|
+
});
|
|
149
|
+
return finalResponseText;
|
|
150
|
+
}
|
|
151
|
+
};
|
|
152
|
+
|
|
153
|
+
|
|
154
|
+
|
|
155
|
+
exports.IntentFulfillService = IntentFulfillService;
|
|
156
|
+
//# sourceMappingURL=chunk-IKPQ52KY.cjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-IKPQ52KY.cjs","../../src/services/intents/fulfill.service.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACA;ACJA,gCAA2B;AAmBpB,IAAM,qBAAA,EAAN,MAA2B;AAAA,EACzB;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,EAEA,MAAc,mBAAA,CACb,MAAA,EACA,MAAA,EACC;AACD,IAAA,IAAI;AACH,MAAA,MAAM,aAAA,kBAAe,IAAA,mBAAK,YAAA,6BAAc,eAAA,mBAAgB,GAAA;AACxD,MAAA,MAAM,EAAE,MAAA,EAAQ,SAAS,EAAA,EAAI,MAAA;AAC7B,MAAA,MAAM,WAAA,EAA4B;AAAA,QACjC,SAAA,EAAW,gCAAA,CAAW;AAAA,QACtB,IAAA,EAAM,MAAA,CAAO,IAAA;AAAA,QACb,SAAA,EAAW,IAAA,CAAK,GAAA,CAAI,CAAA;AAAA,QACpB,OAAA,EAAS,EAAE,IAAA,EAAM,MAAA,EAAQ,KAAA,EAAO,CAAC,MAAA,CAAO,OAAO,EAAE,CAAA;AAAA,QACjD,QAAA,EAAU,MAAA,CAAO;AAAA,MAClB,CAAA;AACA,MAAA,MAAA,CAAO,QAAA,CAAS,IAAA,CAAK,UAAU,CAAA;AAC/B,MAAA,sBAAM,YAAA,6BAAc,mBAAA,mBAAoB,MAAA,EAAQ,QAAA,EAAU,CAAC,UAAU,CAAC,GAAA;AAAA,IACvE,EAAA,MAAA,CAAS,KAAA,EAAO;AACf,MAAA,yBAAA,CAAQ,YAAA,CAAa,KAAA,CAAM,gCAAA,EAAkC,KAAK,CAAA;AAAA,IACnE;AAAA,EACD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAkBA,MAAa,gBAAA,CACZ,KAAA,EACA,MAAA,EACA,MAAA,EACkB;AAClB,IAAA,MAAM,aAAA,EAAe,CAAA;AAAA,SAAA,EAAA,iBACZ,IAAI,IAAA,CAAK,CAAA,CAAA,CAAE,kBAAA,CAAmB,CAAC,CAAA;AAAA;AAAA,kBAExC,IAAA,qBAAK,OAAA,6BAAS,QAAA,GAAS,EAAE,CAAA;AAAA;AAAA,kBAEzB,IAAA,qBAAK,OAAA,6BAAS,SAAA,GAAU,EAAE,CAAA;AAAA;AAAA,kBAE1B,MAAA,+BAAQ,SAAA,GAAU,EAAE,CAAA;AAAA,CAAA,CAAA,CACnB,IAAA,CAAK,CAAA;AAEN,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,EAAyB,CAAC,CAAA;AAChC,IAAA,IAAA,CAAK,UAAA,GAAa,KAAA,CAAM,IAAA,CAAK,GAAG,IAAA,CAAK,SAAA,CAAU,QAAA,CAAS,CAAC,CAAA;AACzD,IAAA,IAAA,CAAK,UAAA,GAAa,KAAA,CAAM,IAAA,CAAK,GAAI,MAAM,IAAA,CAAK,SAAA,CAAU,QAAA,CAAS,CAAE,CAAA;AAEjE,IAAA,MAAM,UAAA,EAAY,aAAA,CAAc,uBAAA,CAAwB,KAAK,CAAA;AAC7D,IAAA,IAAI,aAAA,EAAe,EAAA;AAEnB,IAAA,MAAA,CAAO,IAAA,EAAM;AACZ,MAAA,MAAM,SAAA,EAAW,MAAM,aAAA,CAAc,uBAAA;AAAA,QACpC,QAAA;AAAA,QACA;AAAA,MACD,CAAA;AAEA,MAAA,yBAAA,CAAQ,YAAA,CAAa,IAAA,CAAK,UAAA,EAAY,EAAE,SAAS,CAAC,CAAA;AAElD,MAAA,MAAM,EAAE,OAAA,EAAS,UAAU,EAAA,EAAI,QAAA;AAE/B,MAAA,GAAA,CAAI,SAAA,EAAW;AACd,QAAA,IAAA,CAAA,MAAW,SAAA,GAAY,SAAA,EAAW;AACjC,UAAA,MAAM,SAAA,EAAW,QAAA,CAAS,IAAA;AAC1B,UAAA,MAAM,aAAA,EAAe,KAAA,CAAM,MAAA;AAAA,YAC1B,CAAC,IAAA,EAAA,GAAS,IAAA,CAAK,SAAA,IAAa;AAAA,UAC7B,CAAA,CAAE,CAAC,CAAA;AAEH,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,CAAQ,YAAA,EAAc,QAAQ,CAAA;AAAA,UACjE,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,KAAA;AAAA,cACA,MAAA,CAAO;AAAA,YACR,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;AAAiD,QAAA;AAClD,MAAA;AAEA,QAAA;AACA,QAAA;AAAA,MAAA;AACD,IAAA;AAGD,IAAA;AAAO,EAAA;AACR;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAAA;AAaC,IAAA;AACA,IAAA;AACC,MAAA;AACA,MAAA;AACA,MAAA;AAGA,MAAA;AACsB,QAAA;AACE,QAAA;AACtB,QAAA;AACoB,QAAA;AACgC,QAAA;AACvB,MAAA;AAE/B,MAAA;AAAuC,QAAA;AACtC,QAAA;AACS,QAAA;AACC,UAAA;AACT,UAAA;AACY,UAAA;AACZ,QAAA;AACD,MAAA;AAGD,MAAA;AAAwE,IAAA;AAGzE,IAAA;AAAuC,MAAA;AACtC,MAAA;AACS,IAAA;AAGV,IAAA;AAAO,EAAA;AAET;AD1DA;AACA;AACA;AACA","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-IKPQ52KY.cjs","sourcesContent":[null,"import { randomUUID } from \"node:crypto\";\nimport type {\n\tA2AModule,\n\tMCPModule,\n\tMemoryModule,\n\tModelModule,\n} from \"@/modules\";\nimport type { AinAgentPrompts } from \"@/types/agent\";\nimport { CONNECTOR_PROTOCOL_TYPE, type ConnectorTool } from \"@/types/connector\";\nimport {\n\ttype Intent,\n\ttype MessageObject,\n\tMessageRole,\n\ttype ThreadObject,\n\ttype TriggeredIntent,\n} from \"@/types/memory\";\nimport { StreamEvent } from \"@/types/stream\";\nimport { loggers } from \"@/utils/logger\";\n\nexport class IntentFulfillService {\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\tprivate async addToThreadMessages(\n\t\tthread: ThreadObject,\n\t\tparams: { role: MessageRole; content: string; metadata?: any },\n\t) {\n\t\ttry {\n\t\t\tconst threadMemory = this.memoryModule?.getThreadMemory();\n\t\t\tconst { userId, threadId } = thread;\n\t\t\tconst newMessage: MessageObject = {\n\t\t\t\tmessageId: randomUUID(),\n\t\t\t\trole: params.role,\n\t\t\t\ttimestamp: Date.now(),\n\t\t\t\tcontent: { type: \"text\", parts: [params.content] },\n\t\t\t\tmetadata: params.metadata,\n\t\t\t};\n\t\t\tthread.messages.push(newMessage);\n\t\t\tawait threadMemory?.addMessagesToThread(userId, threadId, [newMessage]);\n\t\t} catch (error) {\n\t\t\tloggers.intentStream.error(\"Error adding message to thread\", error);\n\t\t}\n\t}\n\n\t/**\n\t * Fulfills the detected intent by generating a streaming response.\n\t *\n\t * Manages the complete inference loop including:\n\t * - Loading prompts and conversation history\n\t * - Collecting available tools from modules\n\t * - Executing model inference with tool support\n\t * - Processing tool calls iteratively until completion\n\t * - Streaming results as Server-Sent Events\n\t *\n\t * @param query - The user's input query\n\t * @param threadId - Thread identifier for context\n\t * @param thread - Previous conversation history\n\t * @param intent - Optional detected intent with custom prompt\n\t * @returns AsyncGenerator yielding StreamEvent objects\n\t */\n\tpublic async intentFulfilling(\n\t\tquery: string,\n\t\tthread: ThreadObject,\n\t\tintent?: Intent,\n\t): Promise<string> {\n\t\tconst systemPrompt = `\nToday is ${new Date().toLocaleDateString()}.\n\n${this.prompts?.agent || \"\"}\n\n${this.prompts?.system || \"\"}\n\n${intent?.prompt || \"\"}\n\t`.trim();\n\n\t\tconst modelInstance = this.modelModule.getModel();\n\t\tconst messages = modelInstance.generateMessages({\n\t\t\tquery,\n\t\t\tthread,\n\t\t\tsystemPrompt: systemPrompt.trim(),\n\t\t});\n\n\t\tconst tools: ConnectorTool[] = [];\n\t\tthis.mcpModule && tools.push(...this.mcpModule.getTools());\n\t\tthis.a2aModule && tools.push(...(await this.a2aModule.getTools()));\n\n\t\tconst functions = modelInstance.convertToolsToFunctions(tools);\n\t\tlet finalMessage = \"\";\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\n\t\t\tloggers.intentStream.info(\"messages\", { messages });\n\n\t\t\tconst { content, toolCalls } = response;\n\n\t\t\tif (toolCalls) {\n\t\t\t\tfor (const toolCall of toolCalls) {\n\t\t\t\t\tconst toolName = toolCall.name;\n\t\t\t\t\tconst selectedTool = tools.filter(\n\t\t\t\t\t\t(tool) => tool.toolName === toolName,\n\t\t\t\t\t)[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 === CONNECTOR_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(selectedTool, toolArgs);\n\t\t\t\t\t} else if (\n\t\t\t\t\t\tthis.a2aModule &&\n\t\t\t\t\t\tselectedTool.protocol === CONNECTOR_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,\n\t\t\t\t\t\t\tquery,\n\t\t\t\t\t\t\tthread.threadId,\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\tmodelInstance.appendMessages(messages, toolResult);\n\t\t\t\t}\n\t\t\t} else if (content) {\n\t\t\t\tfinalMessage = content;\n\t\t\t\tbreak;\n\t\t\t}\n\t\t}\n\n\t\treturn finalMessage;\n\t}\n\n\t/**\n\t * Detects the intent from context.\n\t *\n\t * @param intents - The user's input query\n\t * @param thread - The thread history\n\t * @returns The detected intent\n\t */\n\tpublic async intentFulfill(\n\t\tintents: Array<TriggeredIntent>,\n\t\tthread: ThreadObject,\n\t): Promise<string> {\n\t\tlet finalResponseText = \"\";\n\t\tfor (let i = 0; i < intents.length; i++) {\n\t\t\tconst { subquery, intent, actionPlan } = intents[i];\n\t\t\tloggers.intent.info(`Process query: ${subquery}, ${intent?.name}`);\n\t\t\tloggers.intent.info(`Action plan: ${actionPlan}`);\n\n\t\t\t// only use for inference, not stored in memory\n\t\t\tfinalResponseText !== \"\" &&\n\t\t\t\tthread.messages.push({\n\t\t\t\t\tmessageId: randomUUID(),\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\tmetadata: { isThinking: true },\n\t\t\t\t});\n\t\t\tawait this.addToThreadMessages(thread, {\n\t\t\t\trole: MessageRole.MODEL,\n\t\t\t\tcontent: subquery,\n\t\t\t\tmetadata: {\n\t\t\t\t\tsubquery,\n\t\t\t\t\tisThinking: true,\n\t\t\t\t\tactionPlan: actionPlan,\n\t\t\t\t},\n\t\t\t});\n\n\t\t\tfinalResponseText = await this.intentFulfilling(subquery, thread, intent);\n\t\t}\n\n\t\tawait this.addToThreadMessages(thread, {\n\t\t\trole: MessageRole.MODEL,\n\t\t\tcontent: finalResponseText,\n\t\t});\n\n\t\treturn finalResponseText;\n\t}\n}\n"]}
|
|
@@ -0,0 +1,43 @@
|
|
|
1
|
+
"use strict";Object.defineProperty(exports, "__esModule", {value: true});
|
|
2
|
+
|
|
3
|
+
var _chunkMXDXXOXEcjs = require('./chunk-MXDXXOXE.cjs');
|
|
4
|
+
|
|
5
|
+
|
|
6
|
+
var _chunkKAF2EDPZcjs = require('./chunk-KAF2EDPZ.cjs');
|
|
7
|
+
|
|
8
|
+
|
|
9
|
+
var _chunkVLMHDHXTcjs = require('./chunk-VLMHDHXT.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, _chunkKAF2EDPZcjs.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, _chunkMXDXXOXEcjs.QueryStreamService)(
|
|
25
|
+
agent.modelModule,
|
|
26
|
+
agent.a2aModule,
|
|
27
|
+
agent.mcpModule,
|
|
28
|
+
agent.memoryModule,
|
|
29
|
+
agent.manifest.prompts
|
|
30
|
+
);
|
|
31
|
+
}
|
|
32
|
+
const queryController = new (0, _chunkVLMHDHXTcjs.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-J5WX6GPC.cjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-J5WX6GPC.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-J5WX6GPC.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,43 @@
|
|
|
1
|
+
"use strict";Object.defineProperty(exports, "__esModule", {value: true});
|
|
2
|
+
|
|
3
|
+
var _chunkOAPMGKE2cjs = require('./chunk-OAPMGKE2.cjs');
|
|
4
|
+
|
|
5
|
+
|
|
6
|
+
var _chunkAY3PG2UZcjs = require('./chunk-AY3PG2UZ.cjs');
|
|
7
|
+
|
|
8
|
+
|
|
9
|
+
var _chunkVLMHDHXTcjs = require('./chunk-VLMHDHXT.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, _chunkOAPMGKE2cjs.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, _chunkAY3PG2UZcjs.QueryStreamService)(
|
|
25
|
+
agent.modelModule,
|
|
26
|
+
agent.a2aModule,
|
|
27
|
+
agent.mcpModule,
|
|
28
|
+
agent.memoryModule,
|
|
29
|
+
agent.manifest.prompts
|
|
30
|
+
);
|
|
31
|
+
}
|
|
32
|
+
const queryController = new (0, _chunkVLMHDHXTcjs.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-JUPAGIP4.cjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-JUPAGIP4.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-JUPAGIP4.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"]}
|