@ainetwork/adk 0.1.6 → 0.1.7
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-3JKRJYJ6.cjs +111 -0
- package/dist/cjs/chunk-3JKRJYJ6.cjs.map +1 -0
- package/dist/cjs/chunk-466ED5WN.cjs +41 -0
- package/dist/cjs/chunk-466ED5WN.cjs.map +1 -0
- package/dist/cjs/chunk-4ESGHH2M.cjs +20 -0
- package/dist/cjs/chunk-4ESGHH2M.cjs.map +1 -0
- package/dist/cjs/chunk-4HY26I2C.cjs +168 -0
- package/dist/cjs/chunk-4HY26I2C.cjs.map +1 -0
- package/dist/cjs/chunk-5OWM5IH4.cjs +20 -0
- package/dist/cjs/chunk-5OWM5IH4.cjs.map +1 -0
- package/dist/cjs/chunk-6S6GEY3D.cjs +37 -0
- package/dist/cjs/chunk-6S6GEY3D.cjs.map +1 -0
- package/dist/cjs/chunk-ICFTUZP5.cjs +111 -0
- package/dist/cjs/chunk-ICFTUZP5.cjs.map +1 -0
- package/dist/cjs/chunk-MMWEU2MO.cjs +100 -0
- package/dist/cjs/chunk-MMWEU2MO.cjs.map +1 -0
- package/dist/cjs/chunk-NYYEZ4ER.cjs +22 -0
- package/dist/cjs/chunk-NYYEZ4ER.cjs.map +1 -0
- package/dist/cjs/chunk-OBUCVZNU.cjs +37 -0
- package/dist/cjs/chunk-OBUCVZNU.cjs.map +1 -0
- package/dist/cjs/chunk-RMRFCEP4.cjs +8 -0
- package/dist/cjs/chunk-RMRFCEP4.cjs.map +1 -0
- package/dist/cjs/chunk-RPMERDHU.cjs +44 -0
- package/dist/cjs/chunk-RPMERDHU.cjs.map +1 -0
- package/dist/cjs/chunk-RWQDV2XY.cjs +166 -0
- package/dist/cjs/chunk-RWQDV2XY.cjs.map +1 -0
- package/dist/cjs/chunk-UQ72NNHV.cjs +225 -0
- package/dist/cjs/chunk-UQ72NNHV.cjs.map +1 -0
- package/dist/cjs/chunk-XZAKXJYO.cjs +20 -0
- package/dist/cjs/chunk-XZAKXJYO.cjs.map +1 -0
- package/dist/cjs/chunk-ZAZ3KYP5.cjs +43 -0
- package/dist/cjs/chunk-ZAZ3KYP5.cjs.map +1 -0
- package/dist/cjs/controllers/a2a.controller.cjs +3 -3
- package/dist/cjs/controllers/index.cjs +6 -6
- package/dist/cjs/controllers/index.cjs.map +1 -1
- package/dist/cjs/controllers/query.controller.cjs +2 -2
- package/dist/cjs/index.cjs +25 -24
- package/dist/cjs/index.cjs.map +1 -1
- package/dist/cjs/middlewares/error.middleware.cjs +3 -3
- package/dist/cjs/modules/a2a/a2a.module.cjs +3 -3
- package/dist/cjs/modules/index.cjs +9 -9
- package/dist/cjs/modules/index.cjs.map +1 -1
- package/dist/cjs/modules/mcp/mcp.module.cjs +3 -3
- package/dist/cjs/modules/models/base.model.cjs +2 -2
- package/dist/cjs/routes/a2a.routes.cjs +7 -7
- package/dist/cjs/routes/index.cjs +13 -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 +3 -3
- package/dist/cjs/services/index.cjs +9 -5
- package/dist/cjs/services/index.cjs.map +1 -1
- package/dist/cjs/services/query-stream.service.cjs +9 -0
- package/dist/cjs/services/query-stream.service.cjs.map +1 -0
- package/dist/cjs/services/query.service.cjs +4 -4
- package/dist/cjs/types/memory.cjs +2 -2
- package/dist/cjs/types/stream.cjs +1 -0
- package/dist/cjs/types/stream.cjs.map +1 -0
- package/dist/cjs/utils/logger.cjs +2 -2
- package/dist/esm/{chunk-GZFRRFOE.js → chunk-2BITOA2R.js} +1 -1
- package/dist/esm/chunk-2BITOA2R.js.map +1 -0
- package/dist/esm/{chunk-BTZVK3EA.js → chunk-624FPEEV.js} +2 -2
- package/dist/esm/{chunk-U2Y64SAC.js → chunk-A3W52LKL.js} +4 -4
- package/dist/esm/{chunk-SQWZXP4P.js → chunk-DJRRMFKH.js} +2 -2
- package/dist/esm/chunk-EFMGNCWG.js +225 -0
- package/dist/esm/chunk-EFMGNCWG.js.map +1 -0
- package/dist/esm/{chunk-EODMRV4H.js → chunk-HAE333YO.js} +2 -2
- package/dist/esm/{chunk-ILFWNK3W.js → chunk-MAU2MAK2.js} +2 -2
- package/dist/esm/{chunk-N6T2JOGC.js → chunk-MC7CPQ6W.js} +2 -2
- package/dist/esm/{chunk-64GEXWSE.js → chunk-N7VYM2R5.js} +2 -1
- package/dist/esm/chunk-N7VYM2R5.js.map +1 -0
- package/dist/esm/{chunk-F727XG4N.js → chunk-VAZWF5IK.js} +1 -1
- package/dist/esm/{chunk-F727XG4N.js.map → chunk-VAZWF5IK.js.map} +1 -1
- package/dist/esm/chunk-WGHAZ4HG.js +44 -0
- package/dist/esm/chunk-WGHAZ4HG.js.map +1 -0
- package/dist/esm/{chunk-AQ5YZERE.js → chunk-Y6E73I7I.js} +2 -2
- package/dist/esm/{chunk-7QSZHJU5.js → chunk-YT6PD6UF.js} +4 -4
- package/dist/esm/chunk-ZFSIP5KN.js +43 -0
- package/dist/esm/chunk-ZFSIP5KN.js.map +1 -0
- package/dist/esm/controllers/a2a.controller.d.ts +1 -0
- package/dist/esm/controllers/a2a.controller.js +2 -2
- package/dist/esm/controllers/api/model.api.controller.d.ts +1 -0
- package/dist/esm/controllers/index.d.ts +2 -0
- package/dist/esm/controllers/index.js +3 -3
- package/dist/esm/controllers/query.controller.d.ts +5 -1
- package/dist/esm/controllers/query.controller.js +1 -1
- package/dist/esm/index.d.ts +2 -1
- package/dist/esm/index.js +16 -15
- package/dist/esm/index.js.map +1 -1
- package/dist/esm/middlewares/error.middleware.js +2 -2
- package/dist/esm/modules/a2a/a2a.module.js +2 -2
- package/dist/esm/modules/index.d.ts +1 -0
- package/dist/esm/modules/index.js +10 -10
- package/dist/esm/modules/mcp/mcp.module.js +2 -2
- package/dist/esm/modules/memory/base.memory.d.ts +3 -3
- package/dist/esm/modules/models/base.model.d.ts +13 -0
- package/dist/esm/modules/models/base.model.js +1 -1
- package/dist/esm/modules/models/model.module.d.ts +1 -0
- package/dist/esm/routes/a2a.routes.d.ts +1 -0
- package/dist/esm/routes/a2a.routes.js +6 -6
- package/dist/esm/routes/api/models.routes.d.ts +1 -0
- package/dist/esm/routes/api.routes.d.ts +1 -0
- package/dist/esm/routes/api.routes.js +2 -2
- package/dist/esm/routes/index.d.ts +1 -0
- package/dist/esm/routes/index.js +13 -12
- package/dist/esm/routes/query.routes.d.ts +2 -1
- package/dist/esm/routes/query.routes.js +6 -5
- package/dist/esm/services/a2a.service.d.ts +1 -0
- package/dist/esm/services/a2a.service.js +2 -2
- package/dist/esm/services/index.d.ts +3 -0
- package/dist/esm/services/index.js +9 -5
- package/dist/esm/services/query-stream.service.d.ts +74 -0
- package/dist/esm/services/query-stream.service.js +9 -0
- package/dist/esm/services/query-stream.service.js.map +1 -0
- package/dist/esm/services/query.service.d.ts +1 -0
- package/dist/esm/services/query.service.js +3 -3
- package/dist/esm/types/memory.d.ts +9 -1
- package/dist/esm/types/memory.js +1 -1
- package/dist/esm/types/stream.d.ts +47 -0
- package/dist/esm/types/stream.js +1 -0
- package/dist/esm/types/stream.js.map +1 -0
- package/dist/esm/utils/logger.d.ts +1 -0
- package/dist/esm/utils/logger.js +1 -1
- package/package.json +1 -1
- package/dist/esm/chunk-64GEXWSE.js.map +0 -1
- package/dist/esm/chunk-GZFRRFOE.js.map +0 -1
- package/dist/esm/chunk-NH7BYWQV.js +0 -27
- package/dist/esm/chunk-NH7BYWQV.js.map +0 -1
- package/dist/esm/chunk-WIJDRP2A.js +0 -26
- package/dist/esm/chunk-WIJDRP2A.js.map +0 -1
- /package/dist/esm/{chunk-BTZVK3EA.js.map → chunk-624FPEEV.js.map} +0 -0
- /package/dist/esm/{chunk-U2Y64SAC.js.map → chunk-A3W52LKL.js.map} +0 -0
- /package/dist/esm/{chunk-SQWZXP4P.js.map → chunk-DJRRMFKH.js.map} +0 -0
- /package/dist/esm/{chunk-EODMRV4H.js.map → chunk-HAE333YO.js.map} +0 -0
- /package/dist/esm/{chunk-ILFWNK3W.js.map → chunk-MAU2MAK2.js.map} +0 -0
- /package/dist/esm/{chunk-N6T2JOGC.js.map → chunk-MC7CPQ6W.js.map} +0 -0
- /package/dist/esm/{chunk-AQ5YZERE.js.map → chunk-Y6E73I7I.js.map} +0 -0
- /package/dist/esm/{chunk-7QSZHJU5.js.map → chunk-YT6PD6UF.js.map} +0 -0
|
@@ -0,0 +1,111 @@
|
|
|
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; } var _class;
|
|
2
|
+
|
|
3
|
+
var _chunk466ED5WNcjs = require('./chunk-466ED5WN.cjs');
|
|
4
|
+
|
|
5
|
+
// src/services/a2a.service.ts
|
|
6
|
+
var _crypto = require('crypto');
|
|
7
|
+
var A2AService = (_class = class {
|
|
8
|
+
|
|
9
|
+
__init() {this.canceledTasks = /* @__PURE__ */ new Set()}
|
|
10
|
+
constructor(queryService) {;_class.prototype.__init.call(this);_class.prototype.__init2.call(this);_class.prototype.__init3.call(this);
|
|
11
|
+
this.queryService = queryService;
|
|
12
|
+
}
|
|
13
|
+
__init2() {this.cancelTask = async (taskId, _eventBus) => {
|
|
14
|
+
this.canceledTasks.add(taskId);
|
|
15
|
+
}}
|
|
16
|
+
__init3() {this.createTaskStatusUpdateEvent = (taskId, contextId, state, message) => {
|
|
17
|
+
return {
|
|
18
|
+
kind: "status-update",
|
|
19
|
+
taskId,
|
|
20
|
+
contextId,
|
|
21
|
+
status: {
|
|
22
|
+
state,
|
|
23
|
+
message: message ? {
|
|
24
|
+
kind: "message",
|
|
25
|
+
role: "agent",
|
|
26
|
+
messageId: _crypto.randomUUID.call(void 0, ),
|
|
27
|
+
parts: [{ kind: "text", text: message }],
|
|
28
|
+
taskId,
|
|
29
|
+
contextId
|
|
30
|
+
} : void 0,
|
|
31
|
+
timestamp: (/* @__PURE__ */ new Date()).toISOString()
|
|
32
|
+
},
|
|
33
|
+
final: state !== "working"
|
|
34
|
+
};
|
|
35
|
+
}}
|
|
36
|
+
async execute(requestContext, eventBus) {
|
|
37
|
+
const userMessage = requestContext.userMessage;
|
|
38
|
+
const { sessionId } = userMessage.metadata;
|
|
39
|
+
const existingTask = requestContext.task;
|
|
40
|
+
const taskId = _optionalChain([existingTask, 'optionalAccess', _ => _.id]) || _crypto.randomUUID.call(void 0, );
|
|
41
|
+
const contextId = userMessage.contextId || _optionalChain([existingTask, 'optionalAccess', _2 => _2.contextId]) || _crypto.randomUUID.call(void 0, );
|
|
42
|
+
if (!existingTask) {
|
|
43
|
+
const initialTask = {
|
|
44
|
+
kind: "task",
|
|
45
|
+
id: taskId,
|
|
46
|
+
contextId,
|
|
47
|
+
status: {
|
|
48
|
+
state: "submitted",
|
|
49
|
+
timestamp: (/* @__PURE__ */ new Date()).toISOString()
|
|
50
|
+
},
|
|
51
|
+
history: [userMessage],
|
|
52
|
+
metadata: userMessage.metadata,
|
|
53
|
+
artifacts: []
|
|
54
|
+
};
|
|
55
|
+
eventBus.publish(initialTask);
|
|
56
|
+
}
|
|
57
|
+
const workingStatusUpdate = this.createTaskStatusUpdateEvent(
|
|
58
|
+
taskId,
|
|
59
|
+
contextId,
|
|
60
|
+
"working"
|
|
61
|
+
);
|
|
62
|
+
eventBus.publish(workingStatusUpdate);
|
|
63
|
+
const message = userMessage.parts.filter((part) => part.kind === "text").map((part) => part.text).join("\n");
|
|
64
|
+
if (message.length === 0) {
|
|
65
|
+
_chunk466ED5WNcjs.loggers.server.warn(`Empty message received for task ${taskId}.`);
|
|
66
|
+
const failureUpdate = this.createTaskStatusUpdateEvent(
|
|
67
|
+
taskId,
|
|
68
|
+
contextId,
|
|
69
|
+
"failed",
|
|
70
|
+
"No message found to process."
|
|
71
|
+
);
|
|
72
|
+
eventBus.publish(failureUpdate);
|
|
73
|
+
return;
|
|
74
|
+
}
|
|
75
|
+
try {
|
|
76
|
+
const response = await this.queryService.handleQuery(message, sessionId);
|
|
77
|
+
if (this.canceledTasks.has(taskId)) {
|
|
78
|
+
_chunk466ED5WNcjs.loggers.server.info(`Task ${taskId} was canceled.`);
|
|
79
|
+
const canceledUpdate = this.createTaskStatusUpdateEvent(
|
|
80
|
+
taskId,
|
|
81
|
+
contextId,
|
|
82
|
+
"canceled"
|
|
83
|
+
);
|
|
84
|
+
eventBus.publish(canceledUpdate);
|
|
85
|
+
return;
|
|
86
|
+
}
|
|
87
|
+
const finalUpdate = this.createTaskStatusUpdateEvent(
|
|
88
|
+
taskId,
|
|
89
|
+
contextId,
|
|
90
|
+
"completed",
|
|
91
|
+
response.content
|
|
92
|
+
);
|
|
93
|
+
eventBus.publish(finalUpdate);
|
|
94
|
+
_chunk466ED5WNcjs.loggers.server.info(`Task ${taskId} completed successfully.`);
|
|
95
|
+
} catch (error) {
|
|
96
|
+
_chunk466ED5WNcjs.loggers.server.error(`Error processing task ${taskId}:`, error);
|
|
97
|
+
const errorUpdate = this.createTaskStatusUpdateEvent(
|
|
98
|
+
taskId,
|
|
99
|
+
contextId,
|
|
100
|
+
"failed",
|
|
101
|
+
`Agent error: ${error.message}`
|
|
102
|
+
);
|
|
103
|
+
eventBus.publish(errorUpdate);
|
|
104
|
+
}
|
|
105
|
+
}
|
|
106
|
+
}, _class);
|
|
107
|
+
|
|
108
|
+
|
|
109
|
+
|
|
110
|
+
exports.A2AService = A2AService;
|
|
111
|
+
//# sourceMappingURL=chunk-3JKRJYJ6.cjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-3JKRJYJ6.cjs","../../src/services/a2a.service.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACA;ACJA,gCAA2B;AAcpB,IAAM,WAAA,YAAN,MAA0C;AAAA,EACxC;AAAA,iBACA,cAAA,kBAA6B,IAAI,GAAA,CAAY,EAAA;AAAA,EAErD,WAAA,CAAY,YAAA,EAA4B;AACvC,IAAA,IAAA,CAAK,aAAA,EAAe,YAAA;AAAA,EACrB;AAAA,kBAEO,WAAA,EAAa,MAAA,CACnB,MAAA,EACA,SAAA,EAAA,GACmB;AACnB,IAAA,IAAA,CAAK,aAAA,CAAc,GAAA,CAAI,MAAM,CAAA;AAAA,EAC9B,EAAA;AAAA,kBAEQ,4BAAA,EAA8B,CACrC,MAAA,EACA,SAAA,EACA,KAAA,EACA,OAAA,EAAA,GAC2B;AAC3B,IAAA,OAAO;AAAA,MACN,IAAA,EAAM,eAAA;AAAA,MACN,MAAA;AAAA,MACA,SAAA;AAAA,MACA,MAAA,EAAQ;AAAA,QACP,KAAA;AAAA,QACA,OAAA,EAAS,QAAA,EACN;AAAA,UACA,IAAA,EAAM,SAAA;AAAA,UACN,IAAA,EAAM,OAAA;AAAA,UACN,SAAA,EAAW,gCAAA,CAAW;AAAA,UACtB,KAAA,EAAO,CAAC,EAAE,IAAA,EAAM,MAAA,EAAQ,IAAA,EAAM,QAAQ,CAAC,CAAA;AAAA,UACvC,MAAA;AAAA,UACA;AAAA,QACD,EAAA,EACC,KAAA,CAAA;AAAA,QACH,SAAA,EAAA,iBAAW,IAAI,IAAA,CAAK,CAAA,CAAA,CAAE,WAAA,CAAY;AAAA,MACnC,CAAA;AAAA,MACA,KAAA,EAAO,MAAA,IAAU;AAAA,IAClB,CAAA;AAAA,EACD,EAAA;AAAA,EAEA,MAAM,OAAA,CACL,cAAA,EACA,QAAA,EACgB;AAChB,IAAA,MAAM,YAAA,EAAc,cAAA,CAAe,WAAA;AACnC,IAAA,MAAM,EAAE,UAAU,EAAA,EAAI,WAAA,CAAY,QAAA;AAClC,IAAA,MAAM,aAAA,EAAe,cAAA,CAAe,IAAA;AAEpC,IAAA,MAAM,OAAA,kBAAS,YAAA,2BAAc,KAAA,GAAM,gCAAA,CAAW;AAC9C,IAAA,MAAM,UAAA,EACL,WAAA,CAAY,UAAA,mBAAa,YAAA,6BAAc,YAAA,GAAa,gCAAA,CAAW;AAEhE,IAAA,GAAA,CAAI,CAAC,YAAA,EAAc;AAClB,MAAA,MAAM,YAAA,EAAoB;AAAA,QACzB,IAAA,EAAM,MAAA;AAAA,QACN,EAAA,EAAI,MAAA;AAAA,QACJ,SAAA;AAAA,QACA,MAAA,EAAQ;AAAA,UACP,KAAA,EAAO,WAAA;AAAA,UACP,SAAA,EAAA,iBAAW,IAAI,IAAA,CAAK,CAAA,CAAA,CAAE,WAAA,CAAY;AAAA,QACnC,CAAA;AAAA,QACA,OAAA,EAAS,CAAC,WAAW,CAAA;AAAA,QACrB,QAAA,EAAU,WAAA,CAAY,QAAA;AAAA,QACtB,SAAA,EAAW,CAAC;AAAA,MACb,CAAA;AACA,MAAA,QAAA,CAAS,OAAA,CAAQ,WAAW,CAAA;AAAA,IAC7B;AAEA,IAAA,MAAM,oBAAA,EAAsB,IAAA,CAAK,2BAAA;AAAA,MAChC,MAAA;AAAA,MACA,SAAA;AAAA,MACA;AAAA,IACD,CAAA;AACA,IAAA,QAAA,CAAS,OAAA,CAAQ,mBAAmB,CAAA;AAEpC,IAAA,MAAM,QAAA,EAAkB,WAAA,CAAY,KAAA,CAClC,MAAA,CAAO,CAAC,IAAA,EAAA,GAAS,IAAA,CAAK,KAAA,IAAS,MAAM,CAAA,CACrC,GAAA,CAAI,CAAC,IAAA,EAAA,GAAS,IAAA,CAAK,IAAI,CAAA,CACvB,IAAA,CAAK,IAAI,CAAA;AACX,IAAA,GAAA,CAAI,OAAA,CAAQ,OAAA,IAAW,CAAA,EAAG;AACzB,MAAA,yBAAA,CAAQ,MAAA,CAAO,IAAA,CAAK,CAAA,gCAAA,EAAmC,MAAM,CAAA,CAAA,CAAG,CAAA;AAChE,MAAA,MAAM,cAAA,EAAgB,IAAA,CAAK,2BAAA;AAAA,QAC1B,MAAA;AAAA,QACA,SAAA;AAAA,QACA,QAAA;AAAA,QACA;AAAA,MACD,CAAA;AACA,MAAA,QAAA,CAAS,OAAA,CAAQ,aAAa,CAAA;AAC9B,MAAA,MAAA;AAAA,IACD;AAEA,IAAA,IAAI;AACH,MAAA,MAAM,SAAA,EAAW,MAAM,IAAA,CAAK,YAAA,CAAa,WAAA,CAAY,OAAA,EAAS,SAAS,CAAA;AAEvE,MAAA,GAAA,CAAI,IAAA,CAAK,aAAA,CAAc,GAAA,CAAI,MAAM,CAAA,EAAG;AACnC,QAAA,yBAAA,CAAQ,MAAA,CAAO,IAAA,CAAK,CAAA,KAAA,EAAQ,MAAM,CAAA,cAAA,CAAgB,CAAA;AAClD,QAAA,MAAM,eAAA,EAAiB,IAAA,CAAK,2BAAA;AAAA,UAC3B,MAAA;AAAA,UACA,SAAA;AAAA,UACA;AAAA,QACD,CAAA;AACA,QAAA,QAAA,CAAS,OAAA,CAAQ,cAAc,CAAA;AAC/B,QAAA,MAAA;AAAA,MACD;AAEA,MAAA,MAAM,YAAA,EAAc,IAAA,CAAK,2BAAA;AAAA,QACxB,MAAA;AAAA,QACA,SAAA;AAAA,QACA,WAAA;AAAA,QACA,QAAA,CAAS;AAAA,MACV,CAAA;AACA,MAAA,QAAA,CAAS,OAAA,CAAQ,WAAW,CAAA;AAC5B,MAAA,yBAAA,CAAQ,MAAA,CAAO,IAAA,CAAK,CAAA,KAAA,EAAQ,MAAM,CAAA,wBAAA,CAA0B,CAAA;AAAA,IAC7D,EAAA,MAAA,CAAS,KAAA,EAAY;AACpB,MAAA,yBAAA,CAAQ,MAAA,CAAO,KAAA,CAAM,CAAA,sBAAA,EAAyB,MAAM,CAAA,CAAA,CAAA,EAAK,KAAK,CAAA;AAC9D,MAAA,MAAM,YAAA,EAAc,IAAA,CAAK,2BAAA;AAAA,QACxB,MAAA;AAAA,QACA,SAAA;AAAA,QACA,QAAA;AAAA,QACA,CAAA,aAAA,EAAgB,KAAA,CAAM,OAAO,CAAA;AAAA,MAAA;AAE9B,MAAA;AAA4B,IAAA;AAC7B,EAAA;AAEF;ADnCA;AACA;AACA;AACA","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-3JKRJYJ6.cjs","sourcesContent":[null,"import { randomUUID } from \"node:crypto\";\nimport type { Task, TaskStatusUpdateEvent } from \"@a2a-js/sdk\";\nimport type {\n\tAgentExecutor,\n\tExecutionEventBus,\n\tRequestContext,\n} from \"@a2a-js/sdk/server\";\nimport { loggers } from \"@/utils/logger.js\";\nimport type { QueryService } from \"./query.service.js\";\n\n/**\n * Implements the AgentExecutor interface from the a2a-js-sdk.\n * This service is responsible for the core business logic of executing an A2A task.\n */\nexport class A2AService implements AgentExecutor {\n\tprivate queryService: QueryService;\n\tprivate canceledTasks: Set<string> = new Set<string>();\n\n\tconstructor(queryService: QueryService) {\n\t\tthis.queryService = queryService;\n\t}\n\n\tpublic cancelTask = async (\n\t\ttaskId: string,\n\t\t_eventBus: ExecutionEventBus,\n\t): Promise<void> => {\n\t\tthis.canceledTasks.add(taskId);\n\t};\n\n\tprivate createTaskStatusUpdateEvent = (\n\t\ttaskId: string,\n\t\tcontextId: string,\n\t\tstate: \"working\" | \"failed\" | \"canceled\" | \"completed\",\n\t\tmessage?: string,\n\t): TaskStatusUpdateEvent => {\n\t\treturn {\n\t\t\tkind: \"status-update\",\n\t\t\ttaskId: taskId,\n\t\t\tcontextId: contextId,\n\t\t\tstatus: {\n\t\t\t\tstate: state,\n\t\t\t\tmessage: message\n\t\t\t\t\t? {\n\t\t\t\t\t\t\tkind: \"message\",\n\t\t\t\t\t\t\trole: \"agent\",\n\t\t\t\t\t\t\tmessageId: randomUUID(),\n\t\t\t\t\t\t\tparts: [{ kind: \"text\", text: message }],\n\t\t\t\t\t\t\ttaskId: taskId,\n\t\t\t\t\t\t\tcontextId: contextId,\n\t\t\t\t\t\t}\n\t\t\t\t\t: undefined,\n\t\t\t\ttimestamp: new Date().toISOString(),\n\t\t\t},\n\t\t\tfinal: state !== \"working\",\n\t\t};\n\t};\n\n\tasync execute(\n\t\trequestContext: RequestContext,\n\t\teventBus: ExecutionEventBus,\n\t): Promise<void> {\n\t\tconst userMessage = requestContext.userMessage;\n\t\tconst { sessionId } = userMessage.metadata as { sessionId: string };\n\t\tconst existingTask = requestContext.task;\n\n\t\tconst taskId = existingTask?.id || randomUUID();\n\t\tconst contextId =\n\t\t\tuserMessage.contextId || existingTask?.contextId || randomUUID();\n\n\t\tif (!existingTask) {\n\t\t\tconst initialTask: Task = {\n\t\t\t\tkind: \"task\",\n\t\t\t\tid: taskId,\n\t\t\t\tcontextId: contextId,\n\t\t\t\tstatus: {\n\t\t\t\t\tstate: \"submitted\",\n\t\t\t\t\ttimestamp: new Date().toISOString(),\n\t\t\t\t},\n\t\t\t\thistory: [userMessage],\n\t\t\t\tmetadata: userMessage.metadata,\n\t\t\t\tartifacts: [],\n\t\t\t};\n\t\t\teventBus.publish(initialTask);\n\t\t}\n\n\t\tconst workingStatusUpdate = this.createTaskStatusUpdateEvent(\n\t\t\ttaskId,\n\t\t\tcontextId,\n\t\t\t\"working\",\n\t\t);\n\t\teventBus.publish(workingStatusUpdate);\n\n\t\tconst message: string = userMessage.parts\n\t\t\t.filter((part) => part.kind === \"text\")\n\t\t\t.map((part) => part.text)\n\t\t\t.join(\"\\n\");\n\t\tif (message.length === 0) {\n\t\t\tloggers.server.warn(`Empty message received for task ${taskId}.`);\n\t\t\tconst failureUpdate = this.createTaskStatusUpdateEvent(\n\t\t\t\ttaskId,\n\t\t\t\tcontextId,\n\t\t\t\t\"failed\",\n\t\t\t\t\"No message found to process.\",\n\t\t\t);\n\t\t\teventBus.publish(failureUpdate);\n\t\t\treturn;\n\t\t}\n\n\t\ttry {\n\t\t\tconst response = await this.queryService.handleQuery(message, sessionId);\n\n\t\t\tif (this.canceledTasks.has(taskId)) {\n\t\t\t\tloggers.server.info(`Task ${taskId} was canceled.`);\n\t\t\t\tconst canceledUpdate = this.createTaskStatusUpdateEvent(\n\t\t\t\t\ttaskId,\n\t\t\t\t\tcontextId,\n\t\t\t\t\t\"canceled\",\n\t\t\t\t);\n\t\t\t\teventBus.publish(canceledUpdate);\n\t\t\t\treturn;\n\t\t\t}\n\n\t\t\tconst finalUpdate = this.createTaskStatusUpdateEvent(\n\t\t\t\ttaskId,\n\t\t\t\tcontextId,\n\t\t\t\t\"completed\",\n\t\t\t\tresponse.content,\n\t\t\t);\n\t\t\teventBus.publish(finalUpdate);\n\t\t\tloggers.server.info(`Task ${taskId} completed successfully.`);\n\t\t} catch (error: any) {\n\t\t\tloggers.server.error(`Error processing task ${taskId}:`, error);\n\t\t\tconst errorUpdate = this.createTaskStatusUpdateEvent(\n\t\t\t\ttaskId,\n\t\t\t\tcontextId,\n\t\t\t\t\"failed\",\n\t\t\t\t`Agent error: ${error.message}`,\n\t\t\t);\n\t\t\teventBus.publish(errorUpdate);\n\t\t}\n\t}\n}\n"]}
|
|
@@ -0,0 +1,41 @@
|
|
|
1
|
+
"use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }// src/utils/logger.ts
|
|
2
|
+
var _winston = require('winston'); var _winston2 = _interopRequireDefault(_winston);
|
|
3
|
+
var { combine, timestamp, printf, colorize, errors } = _winston2.default.format;
|
|
4
|
+
var createLogger = (service) => {
|
|
5
|
+
const logger2 = _winston2.default.createLogger({
|
|
6
|
+
level: (process.env.LOG_LEVEL || "info").toLowerCase(),
|
|
7
|
+
format: combine(
|
|
8
|
+
errors({ stack: true }),
|
|
9
|
+
colorize(),
|
|
10
|
+
timestamp({ format: "HH:mm:ss" }),
|
|
11
|
+
printf(({ level, message, timestamp: timestamp2, service: service2, stack, ...meta }) => {
|
|
12
|
+
const metaStr = Object.keys(meta).length ? ` | ${JSON.stringify(meta)}` : "";
|
|
13
|
+
const errorStack = stack ? `
|
|
14
|
+
${stack}` : "";
|
|
15
|
+
return `${timestamp2} [${service2}] ${level}: ${message}${metaStr}${errorStack}`;
|
|
16
|
+
})
|
|
17
|
+
),
|
|
18
|
+
defaultMeta: { service },
|
|
19
|
+
transports: [new _winston2.default.transports.Console()]
|
|
20
|
+
});
|
|
21
|
+
return logger2;
|
|
22
|
+
};
|
|
23
|
+
var logger = createLogger("ain-adk");
|
|
24
|
+
var getLogger = (service) => createLogger(service);
|
|
25
|
+
var loggers = {
|
|
26
|
+
agent: getLogger("AINAgent"),
|
|
27
|
+
intent: getLogger("Intent"),
|
|
28
|
+
intentStream: getLogger("IntentStream"),
|
|
29
|
+
mcp: getLogger("MCPModule"),
|
|
30
|
+
a2a: getLogger("A2AModule"),
|
|
31
|
+
model: getLogger("Model"),
|
|
32
|
+
server: getLogger("A2AServer"),
|
|
33
|
+
fol: getLogger("FOL")
|
|
34
|
+
};
|
|
35
|
+
|
|
36
|
+
|
|
37
|
+
|
|
38
|
+
|
|
39
|
+
|
|
40
|
+
exports.logger = logger; exports.getLogger = getLogger; exports.loggers = loggers;
|
|
41
|
+
//# sourceMappingURL=chunk-466ED5WN.cjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-466ED5WN.cjs","../../src/utils/logger.ts"],"names":["logger","timestamp","service"],"mappings":"AAAA;ACAA,oFAAoB;AAEpB,IAAM,EAAE,OAAA,EAAS,SAAA,EAAW,MAAA,EAAQ,QAAA,EAAU,OAAO,EAAA,EAAI,iBAAA,CAAQ,MAAA;AAEjE,IAAM,aAAA,EAAe,CAAC,OAAA,EAAA,GAAoB;AACzC,EAAA,MAAMA,QAAAA,EAAS,iBAAA,CAAQ,YAAA,CAAa;AAAA,IACnC,KAAA,EAAA,CAAQ,OAAA,CAAQ,GAAA,CAAI,UAAA,GAAa,MAAA,CAAA,CAAQ,WAAA,CAAY,CAAA;AAAA,IACrD,MAAA,EAAQ,OAAA;AAAA,MACP,MAAA,CAAO,EAAE,KAAA,EAAO,KAAK,CAAC,CAAA;AAAA,MACtB,QAAA,CAAS,CAAA;AAAA,MACT,SAAA,CAAU,EAAE,MAAA,EAAQ,WAAW,CAAC,CAAA;AAAA,MAChC,MAAA,CAAO,CAAC,EAAE,KAAA,EAAO,OAAA,EAAS,SAAA,EAAAC,UAAAA,EAAW,OAAA,EAAAC,QAAAA,EAAS,KAAA,EAAO,GAAG,KAAK,CAAA,EAAA,GAAM;AAClE,QAAA,MAAM,QAAA,EAAU,MAAA,CAAO,IAAA,CAAK,IAAI,CAAA,CAAE,OAAA,EAC/B,CAAA,GAAA,EAAM,IAAA,CAAK,SAAA,CAAU,IAAI,CAAC,CAAA,EAAA;AAEF,QAAA;AAAe;AACwB,QAAA;AAClE,MAAA;AACF,IAAA;AACuB,IAAA;AACsB,IAAA;AAC7C,EAAA;AAEMF,EAAAA;AACR;AAG4C;AAGsB;AAG3C;AACK,EAAA;AACD,EAAA;AACY,EAAA;AACZ,EAAA;AACA,EAAA;AACF,EAAA;AACK,EAAA;AACT,EAAA;AACrB;ADR8E;AACA;AACA;AACA;AACA;AACA","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-466ED5WN.cjs","sourcesContent":[null,"import winston from \"winston\";\n\nconst { combine, timestamp, printf, colorize, errors } = winston.format;\n\nconst createLogger = (service: string) => {\n\tconst logger = winston.createLogger({\n\t\tlevel: (process.env.LOG_LEVEL || \"info\").toLowerCase(),\n\t\tformat: combine(\n\t\t\terrors({ stack: true }),\n\t\t\tcolorize(),\n\t\t\ttimestamp({ format: \"HH:mm:ss\" }),\n\t\t\tprintf(({ level, message, timestamp, service, stack, ...meta }) => {\n\t\t\t\tconst metaStr = Object.keys(meta).length\n\t\t\t\t\t? ` | ${JSON.stringify(meta)}`\n\t\t\t\t\t: \"\";\n\t\t\t\tconst errorStack = stack ? `\\n${stack}` : \"\";\n\t\t\t\treturn `${timestamp} [${service}] ${level}: ${message}${metaStr}${errorStack}`;\n\t\t\t}),\n\t\t),\n\t\tdefaultMeta: { service },\n\t\ttransports: [new winston.transports.Console()],\n\t});\n\n\treturn logger;\n};\n\n// Default logger instance\nexport const logger = createLogger(\"ain-adk\");\n\n// Factory function for creating service-specific loggers\nexport const getLogger = (service: string) => createLogger(service);\n\n// Convenience methods for different components\nexport const loggers = {\n\tagent: getLogger(\"AINAgent\"),\n\tintent: getLogger(\"Intent\"),\n\tintentStream: getLogger(\"IntentStream\"),\n\tmcp: getLogger(\"MCPModule\"),\n\ta2a: getLogger(\"A2AModule\"),\n\tmodel: getLogger(\"Model\"),\n\tserver: getLogger(\"A2AServer\"),\n\tfol: getLogger(\"FOL\"),\n} as const;\n"]}
|
|
@@ -0,0 +1,20 @@
|
|
|
1
|
+
"use strict";Object.defineProperty(exports, "__esModule", {value: true});// src/types/memory.ts
|
|
2
|
+
var MemoryType = /* @__PURE__ */ ((MemoryType2) => {
|
|
3
|
+
MemoryType2["SESSION"] = "SESSION";
|
|
4
|
+
MemoryType2["INTENT"] = "INTENT";
|
|
5
|
+
MemoryType2["AGENT"] = "AGENT";
|
|
6
|
+
MemoryType2["USER"] = "USER";
|
|
7
|
+
return MemoryType2;
|
|
8
|
+
})(MemoryType || {});
|
|
9
|
+
var ChatRole = /* @__PURE__ */ ((ChatRole2) => {
|
|
10
|
+
ChatRole2["USER"] = "USER";
|
|
11
|
+
ChatRole2["SYSTEM"] = "SYSTEM";
|
|
12
|
+
ChatRole2["MODEL"] = "MODEL";
|
|
13
|
+
return ChatRole2;
|
|
14
|
+
})(ChatRole || {});
|
|
15
|
+
|
|
16
|
+
|
|
17
|
+
|
|
18
|
+
|
|
19
|
+
exports.MemoryType = MemoryType; exports.ChatRole = ChatRole;
|
|
20
|
+
//# sourceMappingURL=chunk-4ESGHH2M.cjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-4ESGHH2M.cjs","../../src/types/memory.ts"],"names":["MemoryType","ChatRole"],"mappings":"AAAA;ACAO,IAAK,WAAA,kBAAL,CAAA,CAAKA,WAAAA,EAAAA,GAAL;AACN,EAAAA,WAAAA,CAAA,SAAA,EAAA,EAAU,SAAA;AACV,EAAAA,WAAAA,CAAA,QAAA,EAAA,EAAS,QAAA;AACT,EAAAA,WAAAA,CAAA,OAAA,EAAA,EAAQ,OAAA;AACR,EAAAA,WAAAA,CAAA,MAAA,EAAA,EAAO,MAAA;AAJI,EAAA,OAAAA,WAAAA;AAAA,CAAA,CAAA,CAAA,WAAA,GAAA,CAAA,CAAA,CAAA;AAUL,IAAK,SAAA,kBAAL,CAAA,CAAKC,SAAAA,EAAAA,GAAL;AAEN,EAAAA,SAAAA,CAAA,MAAA,EAAA,EAAO,MAAA;AAEP,EAAAA,SAAAA,CAAA,QAAA,EAAA,EAAS,QAAA;AAET,EAAAA,SAAAA,CAAA,OAAA,EAAA,EAAQ,OAAA;AANG,EAAA,OAAAA,SAAAA;AAAA,CAAA,CAAA,CAAA,SAAA,GAAA,CAAA,CAAA,CAAA;ADIZ;AACA;AACE;AACA;AACF,6DAAC","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-4ESGHH2M.cjs","sourcesContent":[null,"export enum MemoryType {\n\tSESSION = \"SESSION\",\n\tINTENT = \"INTENT\",\n\tAGENT = \"AGENT\",\n\tUSER = \"USER\",\n}\n\n/**\n * Roles for participants in a chat conversation.\n */\nexport enum ChatRole {\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 chat messages.\n *\n * Supports multi-part content with different types (text, images, etc.).\n */\nexport type ChatContentObject = {\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 chat conversation.\n *\n * @example\n * ```typescript\n * const message: ChatObject = {\n * role: ChatRole.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 ChatObject = {\n\t/** Role of the message sender */\n\trole: ChatRole;\n\t/** Message content with type and parts */\n\tcontent: ChatContentObject;\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 type SessionMetadata = {\n\ttitle?: string;\n\tsessionId: string;\n};\n\n/**\n * Represents a conversation session containing multiple chat messages.\n *\n * Messages are stored in a key-value structure where keys are unique chat IDs\n * and values are the corresponding chat objects.\n *\n * @example\n * ```typescript\n * const session: SessionObject = {\n * chats: {\n * \"<UUID_1>\": { role: ChatRole.USER, content: {...}, timestamp: 1234567890 },\n * \"<UUID_2>\": { role: ChatRole.MODEL, content: {...}, timestamp: 1234567891 }\n * }\n * };\n * ```\n */\nexport type SessionObject = {\n\ttitle?: string;\n\t/** Collection of chat messages indexed by unique chat ID */\n\tchats: {\n\t\t[chatId: string]: ChatObject;\n\t};\n};\n\nexport interface Intent {\n\tname: string;\n\tdescription: string;\n\tprompt?: string;\n\tllm?: string;\n}\n"]}
|
|
@@ -0,0 +1,168 @@
|
|
|
1
|
+
"use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _nullishCoalesce(lhs, rhsFn) { if (lhs != null) { return lhs; } else { return rhsFn(); } } function _optionalChain(ops) { let lastAccessLHS = undefined; let value = ops[0]; let i = 1; while (i < ops.length) { const op = ops[i]; const fn = ops[i + 1]; i += 2; if ((op === 'optionalAccess' || op === 'optionalCall') && value == null) { return undefined; } if (op === 'access' || op === 'optionalAccess') { lastAccessLHS = value; value = fn(value); } else if (op === 'call' || op === 'optionalCall') { value = fn((...args) => value.call(lastAccessLHS, ...args)); lastAccessLHS = undefined; } } return value; } var _class;
|
|
2
|
+
|
|
3
|
+
var _chunkF7I4G6JGcjs = require('./chunk-F7I4G6JG.cjs');
|
|
4
|
+
|
|
5
|
+
|
|
6
|
+
var _chunk466ED5WNcjs = require('./chunk-466ED5WN.cjs');
|
|
7
|
+
|
|
8
|
+
// src/modules/a2a/a2a.module.ts
|
|
9
|
+
var _crypto = require('crypto');
|
|
10
|
+
var _client = require('@a2a-js/sdk/client');
|
|
11
|
+
var A2AModule = (_class = class {constructor() { _class.prototype.__init.call(this);_class.prototype.__init2.call(this);_class.prototype.__init3.call(this); }
|
|
12
|
+
/** Map of A2A server URLs to their corresponding tool instances */
|
|
13
|
+
__init() {this.a2aPeerServers = /* @__PURE__ */ new Map()}
|
|
14
|
+
/** Map of session IDs to their A2A session state */
|
|
15
|
+
__init2() {this.a2aSessions = /* @__PURE__ */ new Map()}
|
|
16
|
+
/**
|
|
17
|
+
* Registers a new A2A peer server URL for connection.
|
|
18
|
+
*
|
|
19
|
+
* @param url - The URL of the A2A-compatible agent to connect to
|
|
20
|
+
*/
|
|
21
|
+
async addA2APeerServer(url) {
|
|
22
|
+
this.a2aPeerServers.set(url, null);
|
|
23
|
+
}
|
|
24
|
+
/**
|
|
25
|
+
* Retrieves tools from all registered A2A peer servers.
|
|
26
|
+
*
|
|
27
|
+
* Attempts to connect to each registered server, fetch their agent cards,
|
|
28
|
+
* and create tool instances. Disables tools for unreachable servers.
|
|
29
|
+
*
|
|
30
|
+
* @returns Promise resolving to array of available A2A tools
|
|
31
|
+
*/
|
|
32
|
+
async getTools() {
|
|
33
|
+
const tools = [];
|
|
34
|
+
for (const url of [...this.a2aPeerServers.keys()]) {
|
|
35
|
+
const tool = this.a2aPeerServers.get(url);
|
|
36
|
+
if (!tool || !tool.enabled) {
|
|
37
|
+
try {
|
|
38
|
+
const client = new (0, _client.A2AClient)(url);
|
|
39
|
+
const card = await client.getAgentCard();
|
|
40
|
+
const toolName = card.name.replaceAll(" ", "-");
|
|
41
|
+
const a2aTool = new (0, _chunkF7I4G6JGcjs.A2ATool)(toolName, client, card);
|
|
42
|
+
tools.push(a2aTool);
|
|
43
|
+
} catch (_error) {
|
|
44
|
+
if (tool) {
|
|
45
|
+
tool.disable();
|
|
46
|
+
}
|
|
47
|
+
}
|
|
48
|
+
} else {
|
|
49
|
+
tools.push(tool);
|
|
50
|
+
}
|
|
51
|
+
}
|
|
52
|
+
return tools;
|
|
53
|
+
}
|
|
54
|
+
/**
|
|
55
|
+
* Gets or creates an A2A session for the given session ID.
|
|
56
|
+
*
|
|
57
|
+
* @param sessionId - The session identifier
|
|
58
|
+
* @returns A2ASession object with task and context IDs
|
|
59
|
+
*/
|
|
60
|
+
__init3() {this.getA2ASessionWithId = (sessionId) => {
|
|
61
|
+
const a2aSession = _nullishCoalesce(this.a2aSessions.get(sessionId), () => ( {
|
|
62
|
+
taskId: void 0,
|
|
63
|
+
contextId: void 0
|
|
64
|
+
}));
|
|
65
|
+
if (!this.a2aSessions.has(sessionId)) {
|
|
66
|
+
this.a2aSessions.set(sessionId, a2aSession);
|
|
67
|
+
}
|
|
68
|
+
return a2aSession;
|
|
69
|
+
}}
|
|
70
|
+
/**
|
|
71
|
+
* Constructs a message payload for A2A communication.
|
|
72
|
+
*
|
|
73
|
+
* Includes session context (task ID and context ID) if available
|
|
74
|
+
* for maintaining conversation continuity.
|
|
75
|
+
*
|
|
76
|
+
* @param query - The message content to send
|
|
77
|
+
* @param sessionId - The session identifier
|
|
78
|
+
* @returns Formatted Message object for A2A protocol
|
|
79
|
+
*/
|
|
80
|
+
getMessagePayload(query, sessionId) {
|
|
81
|
+
const messagePayload = {
|
|
82
|
+
messageId: _crypto.randomUUID.call(void 0, ),
|
|
83
|
+
kind: "message",
|
|
84
|
+
role: "user",
|
|
85
|
+
// FIXME: it could be 'agent'
|
|
86
|
+
metadata: {
|
|
87
|
+
sessionId
|
|
88
|
+
},
|
|
89
|
+
parts: [
|
|
90
|
+
{
|
|
91
|
+
kind: "text",
|
|
92
|
+
text: query
|
|
93
|
+
}
|
|
94
|
+
]
|
|
95
|
+
};
|
|
96
|
+
const a2aSession = this.getA2ASessionWithId(sessionId);
|
|
97
|
+
if (a2aSession.taskId) {
|
|
98
|
+
messagePayload.taskId = a2aSession.taskId;
|
|
99
|
+
}
|
|
100
|
+
if (a2aSession.contextId) {
|
|
101
|
+
messagePayload.contextId = a2aSession.contextId;
|
|
102
|
+
}
|
|
103
|
+
return messagePayload;
|
|
104
|
+
}
|
|
105
|
+
/**
|
|
106
|
+
* Executes an A2A tool by sending a message to the remote agent.
|
|
107
|
+
*
|
|
108
|
+
* Handles streaming responses, maintains session state, and extracts
|
|
109
|
+
* text content from various event types in the response stream.
|
|
110
|
+
*
|
|
111
|
+
* @param tool - The A2ATool instance to use
|
|
112
|
+
* @param messagePayload - The message to send to the agent
|
|
113
|
+
* @param sessionId - The session identifier for context tracking
|
|
114
|
+
* @returns Promise resolving to array of text responses from the agent
|
|
115
|
+
*/
|
|
116
|
+
async useTool(tool, messagePayload, sessionId) {
|
|
117
|
+
const finalText = [];
|
|
118
|
+
const client = tool.client;
|
|
119
|
+
const params = {
|
|
120
|
+
message: messagePayload
|
|
121
|
+
};
|
|
122
|
+
const a2aSession = this.getA2ASessionWithId(sessionId);
|
|
123
|
+
try {
|
|
124
|
+
const stream = client.sendMessageStream(params);
|
|
125
|
+
for await (const event of stream) {
|
|
126
|
+
if (event.kind === "status-update") {
|
|
127
|
+
const typedEvent = event;
|
|
128
|
+
if (typedEvent.final && typedEvent.status.state !== "input-required") {
|
|
129
|
+
a2aSession.taskId = void 0;
|
|
130
|
+
}
|
|
131
|
+
const texts = _optionalChain([typedEvent, 'access', _ => _.status, 'access', _2 => _2.message, 'optionalAccess', _3 => _3.parts, 'access', _4 => _4.filter, 'call', _5 => _5((part) => part.kind === "text"), 'access', _6 => _6.map, 'call', _7 => _7((part) => part.text), 'access', _8 => _8.join, 'call', _9 => _9("\n")]);
|
|
132
|
+
if (texts) {
|
|
133
|
+
finalText.push(texts);
|
|
134
|
+
}
|
|
135
|
+
} else if (event.kind === "message") {
|
|
136
|
+
const msg = event;
|
|
137
|
+
if (msg.taskId && msg.taskId !== a2aSession.taskId) {
|
|
138
|
+
a2aSession.taskId = msg.taskId;
|
|
139
|
+
}
|
|
140
|
+
if (msg.contextId && msg.contextId !== a2aSession.contextId) {
|
|
141
|
+
a2aSession.contextId = msg.contextId;
|
|
142
|
+
}
|
|
143
|
+
} else if (event.kind === "task") {
|
|
144
|
+
const task = event;
|
|
145
|
+
if (task.id !== a2aSession.taskId) {
|
|
146
|
+
a2aSession.taskId = task.id;
|
|
147
|
+
}
|
|
148
|
+
if (task.contextId && task.contextId !== a2aSession.contextId) {
|
|
149
|
+
a2aSession.contextId = task.contextId;
|
|
150
|
+
}
|
|
151
|
+
} else {
|
|
152
|
+
_chunk466ED5WNcjs.loggers.a2a.warn("Received unknown event structure from stream:", {
|
|
153
|
+
event
|
|
154
|
+
});
|
|
155
|
+
}
|
|
156
|
+
}
|
|
157
|
+
} catch (error) {
|
|
158
|
+
_chunk466ED5WNcjs.loggers.a2a.error("Error communicating with agent:", { error });
|
|
159
|
+
tool.disable();
|
|
160
|
+
}
|
|
161
|
+
return finalText;
|
|
162
|
+
}
|
|
163
|
+
}, _class);
|
|
164
|
+
|
|
165
|
+
|
|
166
|
+
|
|
167
|
+
exports.A2AModule = A2AModule;
|
|
168
|
+
//# sourceMappingURL=chunk-4HY26I2C.cjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-4HY26I2C.cjs","../../src/modules/a2a/a2a.module.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACA;ACPA,gCAA2B;AAS3B,4CAA0B;AA+BnB,IAAM,UAAA,YAAN,MAAgB;AAAA;AAAA,iBAEd,eAAA,kBAA8C,IAAI,GAAA,CAAI,EAAA;AAAA;AAAA,kBAEtD,YAAA,kBAAuC,IAAI,GAAA,CAAI,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOvD,MAAa,gBAAA,CAAiB,GAAA,EAA4B;AACzD,IAAA,IAAA,CAAK,cAAA,CAAe,GAAA,CAAI,GAAA,EAAK,IAAI,CAAA;AAAA,EAClC;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAUA,MAAa,QAAA,CAAA,EAA+B;AAC3C,IAAA,MAAM,MAAA,EAAmB,CAAC,CAAA;AAC1B,IAAA,IAAA,CAAA,MAAW,IAAA,GAAO,CAAC,GAAG,IAAA,CAAK,cAAA,CAAe,IAAA,CAAK,CAAC,CAAA,EAAG;AAClD,MAAA,MAAM,KAAA,EAAO,IAAA,CAAK,cAAA,CAAe,GAAA,CAAI,GAAG,CAAA;AACxC,MAAA,GAAA,CAAI,CAAC,KAAA,GAAQ,CAAC,IAAA,CAAK,OAAA,EAAS;AAC3B,QAAA,IAAI;AACH,UAAA,MAAM,OAAA,EAAS,IAAI,sBAAA,CAAU,GAAG,CAAA;AAChC,UAAA,MAAM,KAAA,EAAkB,MAAM,MAAA,CAAO,YAAA,CAAa,CAAA;AAClD,UAAA,MAAM,SAAA,EAAW,IAAA,CAAK,IAAA,CAAK,UAAA,CAAW,GAAA,EAAK,GAAG,CAAA;AAC9C,UAAA,MAAM,QAAA,EAAU,IAAI,8BAAA,CAAQ,QAAA,EAAU,MAAA,EAAQ,IAAI,CAAA;AAElD,UAAA,KAAA,CAAM,IAAA,CAAK,OAAO,CAAA;AAAA,QACnB,EAAA,MAAA,CAAS,MAAA,EAAa;AAErB,UAAA,GAAA,CAAI,IAAA,EAAM;AACT,YAAA,IAAA,CAAK,OAAA,CAAQ,CAAA;AAAA,UACd;AAAA,QACD;AAAA,MACD,EAAA,KAAO;AACN,QAAA,KAAA,CAAM,IAAA,CAAK,IAAI,CAAA;AAAA,MAChB;AAAA,IACD;AACA,IAAA,OAAO,KAAA;AAAA,EACR;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,kBAQQ,oBAAA,EAAsB,CAAC,SAAA,EAAA,GAAkC;AAChE,IAAA,MAAM,WAAA,mBAAa,IAAA,CAAK,WAAA,CAAY,GAAA,CAAI,SAAS,CAAA,UAAK;AAAA,MACrD,MAAA,EAAQ,KAAA,CAAA;AAAA,MACR,SAAA,EAAW,KAAA;AAAA,IACZ,GAAA;AACA,IAAA,GAAA,CAAI,CAAC,IAAA,CAAK,WAAA,CAAY,GAAA,CAAI,SAAS,CAAA,EAAG;AACrC,MAAA,IAAA,CAAK,WAAA,CAAY,GAAA,CAAI,SAAA,EAAW,UAAU,CAAA;AAAA,IAC3C;AAEA,IAAA,OAAO,UAAA;AAAA,EACR,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAYO,iBAAA,CAAkB,KAAA,EAAe,SAAA,EAA4B;AACnE,IAAA,MAAM,eAAA,EAA0B;AAAA,MAC/B,SAAA,EAAW,gCAAA,CAAW;AAAA,MACtB,IAAA,EAAM,SAAA;AAAA,MACN,IAAA,EAAM,MAAA;AAAA;AAAA,MACN,QAAA,EAAU;AAAA,QACT;AAAA,MACD,CAAA;AAAA,MACA,KAAA,EAAO;AAAA,QACN;AAAA,UACC,IAAA,EAAM,MAAA;AAAA,UACN,IAAA,EAAM;AAAA,QACP;AAAA,MACD;AAAA,IACD,CAAA;AAEA,IAAA,MAAM,WAAA,EAAa,IAAA,CAAK,mBAAA,CAAoB,SAAS,CAAA;AACrD,IAAA,GAAA,CAAI,UAAA,CAAW,MAAA,EAAQ;AACtB,MAAA,cAAA,CAAe,OAAA,EAAS,UAAA,CAAW,MAAA;AAAA,IACpC;AACA,IAAA,GAAA,CAAI,UAAA,CAAW,SAAA,EAAW;AACzB,MAAA,cAAA,CAAe,UAAA,EAAY,UAAA,CAAW,SAAA;AAAA,IACvC;AAEA,IAAA,OAAO,cAAA;AAAA,EACR;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAaA,MAAa,OAAA,CACZ,IAAA,EACA,cAAA,EACA,SAAA,EACoB;AACpB,IAAA,MAAM,UAAA,EAAsB,CAAC,CAAA;AAC7B,IAAA,MAAM,OAAA,EAAS,IAAA,CAAK,MAAA;AACpB,IAAA,MAAM,OAAA,EAA4B;AAAA,MACjC,OAAA,EAAS;AAAA,IACV,CAAA;AACA,IAAA,MAAM,WAAA,EAAa,IAAA,CAAK,mBAAA,CAAoB,SAAS,CAAA;AAErD,IAAA,IAAI;AACH,MAAA,MAAM,OAAA,EAAS,MAAA,CAAO,iBAAA,CAAkB,MAAM,CAAA;AAC9C,MAAA,IAAA,MAAA,CAAA,MAAiB,MAAA,GAAS,MAAA,EAAQ;AACjC,QAAA,GAAA,CAAI,KAAA,CAAM,KAAA,IAAS,eAAA,EAAiB;AACnC,UAAA,MAAM,WAAA,EAAa,KAAA;AACnB,UAAA,GAAA,CACC,UAAA,CAAW,MAAA,GACX,UAAA,CAAW,MAAA,CAAO,MAAA,IAAU,gBAAA,EAC3B;AACD,YAAA,UAAA,CAAW,OAAA,EAAS,KAAA,CAAA;AAAA,UACrB;AAEA,UAAA,MAAM,MAAA,kBAAQ,UAAA,mBAAW,MAAA,qBAAO,OAAA,6BAAS,KAAA,qBACvC,MAAA,mBAAO,CAAC,IAAA,EAAA,GAAS,IAAA,CAAK,KAAA,IAAS,MAAM,CAAA,qBACrC,GAAA,mBAAI,CAAC,IAAA,EAAA,GAAmB,IAAA,CAAK,IAAI,CAAA,qBACjC,IAAA,mBAAK,IAAI,GAAA;AACX,UAAA,GAAA,CAAI,KAAA,EAAO;AACV,YAAA,SAAA,CAAU,IAAA,CAAK,KAAK,CAAA;AAAA,UACrB;AAAA,QACD,EAAA,KAAA,GAAA,CAAW,KAAA,CAAM,KAAA,IAAS,SAAA,EAAW;AAEpC,UAAA,MAAM,IAAA,EAAM,KAAA;AACZ,UAAA,GAAA,CAAI,GAAA,CAAI,OAAA,GAAU,GAAA,CAAI,OAAA,IAAW,UAAA,CAAW,MAAA,EAAQ;AACnD,YAAA,UAAA,CAAW,OAAA,EAAS,GAAA,CAAI,MAAA;AAAA,UACzB;AACA,UAAA,GAAA,CAAI,GAAA,CAAI,UAAA,GAAa,GAAA,CAAI,UAAA,IAAc,UAAA,CAAW,SAAA,EAAW;AAC5D,YAAA,UAAA,CAAW,UAAA,EAAY,GAAA,CAAI,SAAA;AAAA,UAC5B;AAAA,QACD,EAAA,KAAA,GAAA,CAAW,KAAA,CAAM,KAAA,IAAS,MAAA,EAAQ;AAEjC,UAAA,MAAM,KAAA,EAAO,KAAA;AACb,UAAA,GAAA,CAAI,IAAA,CAAK,GAAA,IAAO,UAAA,CAAW,MAAA,EAAQ;AAClC,YAAA,UAAA,CAAW,OAAA,EAAS,IAAA,CAAK,EAAA;AAAA,UAC1B;AACA,UAAA,GAAA,CAAI,IAAA,CAAK,UAAA,GAAa,IAAA,CAAK,UAAA,IAAc,UAAA,CAAW,SAAA,EAAW;AAC9D,YAAA,UAAA,CAAW,UAAA,EAAY,IAAA,CAAK,SAAA;AAAA,UAC7B;AAAA,QACD,EAAA,KAAO;AACN,UAAA,yBAAA,CAAQ,GAAA,CAAI,IAAA,CAAK,+CAAA,EAAiD;AAAA,YACjE;AAAA,UACD,CAAC,CAAA;AAAA,QACF;AAAA,MACD;AAAA,IACD,EAAA,MAAA,CAAS,KAAA,EAAgB;AACxB,MAAA,yBAAA,CAAQ,GAAA,CAAI,KAAA,CAAM,iCAAA,EAAmC,EAAE,MAAM,CAAC,CAAA;AAC9D,MAAA,IAAA,CAAK,OAAA,CAAQ,CAAA;AAAA,IAEd;AAEA,IAAA,OAAO,SAAA;AAAA,EACR;AACD,UAAA;ADtDA;AACA;AACE;AACF,8BAAC","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-4HY26I2C.cjs","sourcesContent":[null,"import { randomUUID } from \"node:crypto\";\nimport type {\n\tAgentCard,\n\tMessage,\n\tMessageSendParams,\n\tTask,\n\tTaskStatusUpdateEvent,\n\tTextPart,\n} from \"@a2a-js/sdk\";\nimport { A2AClient } from \"@a2a-js/sdk/client\";\nimport { loggers } from \"@/utils/logger.js\";\nimport { A2ATool } from \"./a2a.tool.js\";\n\n/**\n * Represents an active A2A communication session.\n */\ninterface A2ASession {\n\t/** Current task ID for multi-turn conversations */\n\ttaskId: string | undefined;\n\t/** Context ID for maintaining conversation state */\n\tcontextId: string | undefined;\n}\n\n/**\n * Module for managing Agent-to-Agent (A2A) protocol connections.\n *\n * This module handles connections to other A2A-compatible agents, manages\n * conversation sessions, and provides an interface for inter-agent communication.\n * Supports multi-turn conversations with task and context tracking.\n *\n * @example\n * ```typescript\n * const a2aModule = new A2AModule();\n * await a2aModule.addA2APeerServer(\"https://api.example.com/agent\");\n *\n * const tools = await a2aModule.getTools();\n * const message = a2aModule.getMessagePayload(\"Hello\", \"session-123\");\n * const response = await a2aModule.useTool(tools[0], message, \"session-123\");\n * ```\n */\nexport class A2AModule {\n\t/** Map of A2A server URLs to their corresponding tool instances */\n\tprivate a2aPeerServers: Map<string, A2ATool | null> = new Map();\n\t/** Map of session IDs to their A2A session state */\n\tprivate a2aSessions: Map<string, A2ASession> = new Map();\n\n\t/**\n\t * Registers a new A2A peer server URL for connection.\n\t *\n\t * @param url - The URL of the A2A-compatible agent to connect to\n\t */\n\tpublic async addA2APeerServer(url: string): Promise<void> {\n\t\tthis.a2aPeerServers.set(url, null);\n\t}\n\n\t/**\n\t * Retrieves tools from all registered A2A peer servers.\n\t *\n\t * Attempts to connect to each registered server, fetch their agent cards,\n\t * and create tool instances. Disables tools for unreachable servers.\n\t *\n\t * @returns Promise resolving to array of available A2A tools\n\t */\n\tpublic async getTools(): Promise<A2ATool[]> {\n\t\tconst tools: A2ATool[] = [];\n\t\tfor (const url of [...this.a2aPeerServers.keys()]) {\n\t\t\tconst tool = this.a2aPeerServers.get(url);\n\t\t\tif (!tool || !tool.enabled) {\n\t\t\t\ttry {\n\t\t\t\t\tconst client = new A2AClient(url);\n\t\t\t\t\tconst card: AgentCard = await client.getAgentCard();\n\t\t\t\t\tconst toolName = card.name.replaceAll(\" \", \"-\");\n\t\t\t\t\tconst a2aTool = new A2ATool(toolName, client, card);\n\n\t\t\t\t\ttools.push(a2aTool);\n\t\t\t\t} catch (_error: any) {\n\t\t\t\t\t// Agent not responded\n\t\t\t\t\tif (tool) {\n\t\t\t\t\t\ttool.disable();\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\ttools.push(tool);\n\t\t\t}\n\t\t}\n\t\treturn tools;\n\t}\n\n\t/**\n\t * Gets or creates an A2A session for the given session ID.\n\t *\n\t * @param sessionId - The session identifier\n\t * @returns A2ASession object with task and context IDs\n\t */\n\tprivate getA2ASessionWithId = (sessionId: string): A2ASession => {\n\t\tconst a2aSession = this.a2aSessions.get(sessionId) ?? {\n\t\t\ttaskId: undefined,\n\t\t\tcontextId: undefined,\n\t\t};\n\t\tif (!this.a2aSessions.has(sessionId)) {\n\t\t\tthis.a2aSessions.set(sessionId, a2aSession);\n\t\t}\n\n\t\treturn a2aSession;\n\t};\n\n\t/**\n\t * Constructs a message payload for A2A communication.\n\t *\n\t * Includes session context (task ID and context ID) if available\n\t * for maintaining conversation continuity.\n\t *\n\t * @param query - The message content to send\n\t * @param sessionId - The session identifier\n\t * @returns Formatted Message object for A2A protocol\n\t */\n\tpublic getMessagePayload(query: string, sessionId: string): Message {\n\t\tconst messagePayload: Message = {\n\t\t\tmessageId: randomUUID(),\n\t\t\tkind: \"message\",\n\t\t\trole: \"user\", // FIXME: it could be 'agent'\n\t\t\tmetadata: {\n\t\t\t\tsessionId,\n\t\t\t},\n\t\t\tparts: [\n\t\t\t\t{\n\t\t\t\t\tkind: \"text\",\n\t\t\t\t\ttext: query,\n\t\t\t\t},\n\t\t\t],\n\t\t};\n\n\t\tconst a2aSession = this.getA2ASessionWithId(sessionId);\n\t\tif (a2aSession.taskId) {\n\t\t\tmessagePayload.taskId = a2aSession.taskId;\n\t\t}\n\t\tif (a2aSession.contextId) {\n\t\t\tmessagePayload.contextId = a2aSession.contextId;\n\t\t}\n\n\t\treturn messagePayload;\n\t}\n\n\t/**\n\t * Executes an A2A tool by sending a message to the remote agent.\n\t *\n\t * Handles streaming responses, maintains session state, and extracts\n\t * text content from various event types in the response stream.\n\t *\n\t * @param tool - The A2ATool instance to use\n\t * @param messagePayload - The message to send to the agent\n\t * @param sessionId - The session identifier for context tracking\n\t * @returns Promise resolving to array of text responses from the agent\n\t */\n\tpublic async useTool(\n\t\ttool: A2ATool,\n\t\tmessagePayload: Message,\n\t\tsessionId: string,\n\t): Promise<string[]> {\n\t\tconst finalText: string[] = [];\n\t\tconst client = tool.client;\n\t\tconst params: MessageSendParams = {\n\t\t\tmessage: messagePayload,\n\t\t};\n\t\tconst a2aSession = this.getA2ASessionWithId(sessionId);\n\n\t\ttry {\n\t\t\tconst stream = client.sendMessageStream(params);\n\t\t\tfor await (const event of stream) {\n\t\t\t\tif (event.kind === \"status-update\") {\n\t\t\t\t\tconst typedEvent = event as TaskStatusUpdateEvent;\n\t\t\t\t\tif (\n\t\t\t\t\t\ttypedEvent.final &&\n\t\t\t\t\t\ttypedEvent.status.state !== \"input-required\"\n\t\t\t\t\t) {\n\t\t\t\t\t\ta2aSession.taskId = undefined;\n\t\t\t\t\t}\n\t\t\t\t\t// TODO: handle 'file', 'data' parts\n\t\t\t\t\tconst texts = typedEvent.status.message?.parts\n\t\t\t\t\t\t.filter((part) => part.kind === \"text\")\n\t\t\t\t\t\t.map((part: TextPart) => part.text)\n\t\t\t\t\t\t.join(\"\\n\");\n\t\t\t\t\tif (texts) {\n\t\t\t\t\t\tfinalText.push(texts);\n\t\t\t\t\t}\n\t\t\t\t} else if (event.kind === \"message\") {\n\t\t\t\t\t// FIXME: handling text in 'message'?\n\t\t\t\t\tconst msg = event as Message;\n\t\t\t\t\tif (msg.taskId && msg.taskId !== a2aSession.taskId) {\n\t\t\t\t\t\ta2aSession.taskId = msg.taskId;\n\t\t\t\t\t}\n\t\t\t\t\tif (msg.contextId && msg.contextId !== a2aSession.contextId) {\n\t\t\t\t\t\ta2aSession.contextId = msg.contextId;\n\t\t\t\t\t}\n\t\t\t\t} else if (event.kind === \"task\") {\n\t\t\t\t\t// FIXME: handling text in 'task'?\n\t\t\t\t\tconst task = event as Task;\n\t\t\t\t\tif (task.id !== a2aSession.taskId) {\n\t\t\t\t\t\ta2aSession.taskId = task.id;\n\t\t\t\t\t}\n\t\t\t\t\tif (task.contextId && task.contextId !== a2aSession.contextId) {\n\t\t\t\t\t\ta2aSession.contextId = task.contextId;\n\t\t\t\t\t}\n\t\t\t\t} else {\n\t\t\t\t\tloggers.a2a.warn(\"Received unknown event structure from stream:\", {\n\t\t\t\t\t\tevent,\n\t\t\t\t\t});\n\t\t\t\t}\n\t\t\t}\n\t\t} catch (error: unknown) {\n\t\t\tloggers.a2a.error(\"Error communicating with agent:\", { error });\n\t\t\ttool.disable();\n\t\t\t// TODO: add failed & disabled text for next inference?\n\t\t}\n\n\t\treturn finalText;\n\t}\n}\n"]}
|
|
@@ -0,0 +1,20 @@
|
|
|
1
|
+
"use strict";Object.defineProperty(exports, "__esModule", {value: true});// src/types/memory.ts
|
|
2
|
+
var MemoryType = /* @__PURE__ */ ((MemoryType2) => {
|
|
3
|
+
MemoryType2["SESSION"] = "SESSION";
|
|
4
|
+
MemoryType2["INTENT"] = "INTENT";
|
|
5
|
+
MemoryType2["AGENT"] = "AGENT";
|
|
6
|
+
MemoryType2["USER"] = "USER";
|
|
7
|
+
return MemoryType2;
|
|
8
|
+
})(MemoryType || {});
|
|
9
|
+
var ChatRole = /* @__PURE__ */ ((ChatRole2) => {
|
|
10
|
+
ChatRole2["USER"] = "USER";
|
|
11
|
+
ChatRole2["SYSTEM"] = "SYSTEM";
|
|
12
|
+
ChatRole2["MODEL"] = "MODEL";
|
|
13
|
+
return ChatRole2;
|
|
14
|
+
})(ChatRole || {});
|
|
15
|
+
|
|
16
|
+
|
|
17
|
+
|
|
18
|
+
|
|
19
|
+
exports.MemoryType = MemoryType; exports.ChatRole = ChatRole;
|
|
20
|
+
//# sourceMappingURL=chunk-5OWM5IH4.cjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-5OWM5IH4.cjs","../../src/types/memory.ts"],"names":["MemoryType","ChatRole"],"mappings":"AAAA;ACAO,IAAK,WAAA,kBAAL,CAAA,CAAKA,WAAAA,EAAAA,GAAL;AACN,EAAAA,WAAAA,CAAA,SAAA,EAAA,EAAU,SAAA;AACV,EAAAA,WAAAA,CAAA,QAAA,EAAA,EAAS,QAAA;AACT,EAAAA,WAAAA,CAAA,OAAA,EAAA,EAAQ,OAAA;AACR,EAAAA,WAAAA,CAAA,MAAA,EAAA,EAAO,MAAA;AAJI,EAAA,OAAAA,WAAAA;AAAA,CAAA,CAAA,CAAA,WAAA,GAAA,CAAA,CAAA,CAAA;AAUL,IAAK,SAAA,kBAAL,CAAA,CAAKC,SAAAA,EAAAA,GAAL;AAEN,EAAAA,SAAAA,CAAA,MAAA,EAAA,EAAO,MAAA;AAEP,EAAAA,SAAAA,CAAA,QAAA,EAAA,EAAS,QAAA;AAET,EAAAA,SAAAA,CAAA,OAAA,EAAA,EAAQ,OAAA;AANG,EAAA,OAAAA,SAAAA;AAAA,CAAA,CAAA,CAAA,SAAA,GAAA,CAAA,CAAA,CAAA;ADIZ;AACA;AACE;AACA;AACF,6DAAC","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-5OWM5IH4.cjs","sourcesContent":[null,"export enum MemoryType {\n\tSESSION = \"SESSION\",\n\tINTENT = \"INTENT\",\n\tAGENT = \"AGENT\",\n\tUSER = \"USER\",\n}\n\n/**\n * Roles for participants in a chat conversation.\n */\nexport enum ChatRole {\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 chat messages.\n *\n * Supports multi-part content with different types (text, images, etc.).\n */\nexport type ChatContentObject = {\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 chat conversation.\n *\n * @example\n * ```typescript\n * const message: ChatObject = {\n * role: ChatRole.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 ChatObject = {\n\t/** Role of the message sender */\n\trole: ChatRole;\n\t/** Message content with type and parts */\n\tcontent: ChatContentObject;\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\n/**\n * Represents a conversation session containing multiple chat messages.\n *\n * Messages are stored in a key-value structure where keys are unique chat IDs\n * and values are the corresponding chat objects.\n *\n * @example\n * ```typescript\n * const session: SessionObject = {\n * chats: {\n * \"<UUID_1>\": { role: ChatRole.USER, content: {...}, timestamp: 1234567890 },\n * \"<UUID_2>\": { role: ChatRole.MODEL, content: {...}, timestamp: 1234567891 }\n * }\n * };\n * ```\n */\nexport type SessionObject = {\n\t/** Collection of chat messages indexed by unique chat ID */\n\tchats: {\n\t\t[chatId: string]: ChatObject;\n\t};\n};\n\nexport interface Intent {\n\tname: string;\n\tdescription: string;\n\tprompt?: string;\n\tllm?: string;\n}\n"]}
|
|
@@ -0,0 +1,37 @@
|
|
|
1
|
+
"use strict";Object.defineProperty(exports, "__esModule", {value: true});
|
|
2
|
+
|
|
3
|
+
var _chunkICFTUZP5cjs = require('./chunk-ICFTUZP5.cjs');
|
|
4
|
+
|
|
5
|
+
|
|
6
|
+
var _chunkRWQDV2XYcjs = require('./chunk-RWQDV2XY.cjs');
|
|
7
|
+
|
|
8
|
+
|
|
9
|
+
var _chunk3JKRJYJ6cjs = require('./chunk-3JKRJYJ6.cjs');
|
|
10
|
+
|
|
11
|
+
// src/routes/a2a.routes.ts
|
|
12
|
+
var _server = require('@a2a-js/sdk/server');
|
|
13
|
+
var _express = require('express');
|
|
14
|
+
var createA2ARouter = (agent) => {
|
|
15
|
+
const router = _express.Router.call(void 0, );
|
|
16
|
+
const taskStore = new (0, _server.InMemoryTaskStore)();
|
|
17
|
+
const queryService = new (0, _chunkRWQDV2XYcjs.QueryService)(
|
|
18
|
+
agent.modelModule,
|
|
19
|
+
agent.a2aModule,
|
|
20
|
+
agent.mcpModule,
|
|
21
|
+
agent.memoryModule,
|
|
22
|
+
agent.manifest.prompts
|
|
23
|
+
);
|
|
24
|
+
const a2aService = new (0, _chunk3JKRJYJ6cjs.A2AService)(queryService);
|
|
25
|
+
const a2aController = new (0, _chunkICFTUZP5cjs.A2AController)(
|
|
26
|
+
a2aService,
|
|
27
|
+
taskStore,
|
|
28
|
+
agent.generateAgentCard
|
|
29
|
+
);
|
|
30
|
+
router.post("/a2a", a2aController.handleA2ARequest);
|
|
31
|
+
return router;
|
|
32
|
+
};
|
|
33
|
+
|
|
34
|
+
|
|
35
|
+
|
|
36
|
+
exports.createA2ARouter = createA2ARouter;
|
|
37
|
+
//# sourceMappingURL=chunk-6S6GEY3D.cjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-6S6GEY3D.cjs","../../src/routes/a2a.routes.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACA;ACVA,4CAAkC;AAClC,kCAAuB;AAchB,IAAM,gBAAA,EAAkB,CAAC,KAAA,EAAA,GAA4B;AAC3D,EAAA,MAAM,OAAA,EAAS,6BAAA,CAAO;AAGtB,EAAA,MAAM,UAAA,EAAY,IAAI,8BAAA,CAAkB,CAAA;AACxC,EAAA,MAAM,aAAA,EAAe,IAAI,mCAAA;AAAA,IACxB,KAAA,CAAM,WAAA;AAAA,IACN,KAAA,CAAM,SAAA;AAAA,IACN,KAAA,CAAM,SAAA;AAAA,IACN,KAAA,CAAM,YAAA;AAAA,IACN,KAAA,CAAM,QAAA,CAAS;AAAA,EAChB,CAAA;AACA,EAAA,MAAM,WAAA,EAAa,IAAI,iCAAA,CAAW,YAAY,CAAA;AAC9C,EAAA,MAAM,cAAA,EAAgB,IAAI,oCAAA;AAAA,IACzB,UAAA;AAAA,IACA,SAAA;AAAA,IACA,KAAA,CAAM;AAAA,EACP,CAAA;AAGA,EAAA,MAAA,CAAO,IAAA,CAAK,MAAA,EAAQ,aAAA,CAAc,gBAAgB,CAAA;AAElD,EAAA,OAAO,MAAA;AACR,CAAA;ADNA;AACA;AACE;AACF,0CAAC","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-6S6GEY3D.cjs","sourcesContent":[null,"import { InMemoryTaskStore } from \"@a2a-js/sdk/server\";\nimport { Router } from \"express\";\nimport type { AINAgent } from \"@/index.js\";\nimport { QueryService } from \"@/services/query.service.js\";\nimport { A2AController } from \"../controllers/a2a.controller.js\";\nimport { A2AService } from \"../services/a2a.service.js\";\n\n/**\n * Creates and configures the A2A router.\n * This function is a \"composition root\" for the A2A feature,\n * creating and injecting all necessary dependencies.\n * @param intentAnalyzer The core intent analyzer.\n * @param agentCard The agent's card.\n * @returns An Express Router instance.\n */\nexport const createA2ARouter = (agent: AINAgent): Router => {\n\tconst router = Router();\n\n\t// 1. Create dependencies for the A2A feature\n\tconst taskStore = new InMemoryTaskStore();\n\tconst queryService = new QueryService(\n\t\tagent.modelModule,\n\t\tagent.a2aModule,\n\t\tagent.mcpModule,\n\t\tagent.memoryModule,\n\t\tagent.manifest.prompts,\n\t);\n\tconst a2aService = new A2AService(queryService);\n\tconst a2aController = new A2AController(\n\t\ta2aService,\n\t\ttaskStore,\n\t\tagent.generateAgentCard,\n\t);\n\n\t// 2. Define the route\n\trouter.post(\"/a2a\", a2aController.handleA2ARequest);\n\n\treturn router;\n};\n"]}
|
|
@@ -0,0 +1,111 @@
|
|
|
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; } var _class;
|
|
2
|
+
|
|
3
|
+
var _chunk466ED5WNcjs = require('./chunk-466ED5WN.cjs');
|
|
4
|
+
|
|
5
|
+
// src/controllers/a2a.controller.ts
|
|
6
|
+
|
|
7
|
+
|
|
8
|
+
|
|
9
|
+
|
|
10
|
+
|
|
11
|
+
var _server = require('@a2a-js/sdk/server');
|
|
12
|
+
var AINRequestHandler = class extends _server.DefaultRequestHandler {
|
|
13
|
+
|
|
14
|
+
constructor(cardGenerator, taskStore, executor, eventBusManager = new (0, _server.DefaultExecutionEventBusManager)()) {
|
|
15
|
+
const card = cardGenerator();
|
|
16
|
+
super(card, taskStore, executor, eventBusManager);
|
|
17
|
+
this.cardGenerator = cardGenerator;
|
|
18
|
+
}
|
|
19
|
+
async getAgentCard() {
|
|
20
|
+
return this.cardGenerator();
|
|
21
|
+
}
|
|
22
|
+
};
|
|
23
|
+
var A2AController = (_class = class {
|
|
24
|
+
|
|
25
|
+
constructor(a2aService, taskStore, cardGenerator) {;_class.prototype.__init.call(this);
|
|
26
|
+
const requestHandler = new AINRequestHandler(
|
|
27
|
+
cardGenerator,
|
|
28
|
+
taskStore,
|
|
29
|
+
a2aService
|
|
30
|
+
// Injecting the service here as the executor
|
|
31
|
+
);
|
|
32
|
+
this.jsonRpcTransportHandler = new (0, _server.JsonRpcTransportHandler)(requestHandler);
|
|
33
|
+
}
|
|
34
|
+
/**
|
|
35
|
+
* Handles the POST /a2a request.
|
|
36
|
+
*/
|
|
37
|
+
__init() {this.handleA2ARequest = async (req, res) => {
|
|
38
|
+
try {
|
|
39
|
+
const rpcResponseOrStream = await this.jsonRpcTransportHandler.handle(
|
|
40
|
+
req.body
|
|
41
|
+
);
|
|
42
|
+
if (typeof _optionalChain([rpcResponseOrStream, 'optionalAccess', _2 => _2[Symbol.asyncIterator]]) === "function") {
|
|
43
|
+
const stream = rpcResponseOrStream;
|
|
44
|
+
this.streamSse(req, res, stream);
|
|
45
|
+
} else {
|
|
46
|
+
const rpcResponse = rpcResponseOrStream;
|
|
47
|
+
res.status(200).json(rpcResponse);
|
|
48
|
+
}
|
|
49
|
+
} catch (error) {
|
|
50
|
+
this.handleError(req, res, error);
|
|
51
|
+
}
|
|
52
|
+
}}
|
|
53
|
+
async streamSse(req, res, stream) {
|
|
54
|
+
res.setHeader("Content-Type", "text/event-stream");
|
|
55
|
+
res.setHeader("Cache-Control", "no-cache");
|
|
56
|
+
res.setHeader("Connection", "keep-alive");
|
|
57
|
+
res.flushHeaders();
|
|
58
|
+
try {
|
|
59
|
+
for await (const event of stream) {
|
|
60
|
+
res.write(`id: ${Date.now()}
|
|
61
|
+
`);
|
|
62
|
+
res.write(`data: ${JSON.stringify(event)}
|
|
63
|
+
|
|
64
|
+
`);
|
|
65
|
+
}
|
|
66
|
+
} catch (streamError) {
|
|
67
|
+
_chunk466ED5WNcjs.loggers.server.error(
|
|
68
|
+
`Error during SSE streaming (request ${_optionalChain([req, 'access', _3 => _3.body, 'optionalAccess', _4 => _4.id])}):`,
|
|
69
|
+
streamError
|
|
70
|
+
);
|
|
71
|
+
const a2aError = streamError instanceof _server.A2AError ? streamError : _server.A2AError.internalError("Streaming error");
|
|
72
|
+
const errorResponse = {
|
|
73
|
+
jsonrpc: "2.0",
|
|
74
|
+
id: _optionalChain([req, 'access', _5 => _5.body, 'optionalAccess', _6 => _6.id]) || null,
|
|
75
|
+
error: a2aError.toJSONRPCError()
|
|
76
|
+
};
|
|
77
|
+
res.write(`id: ${Date.now()}
|
|
78
|
+
`);
|
|
79
|
+
res.write("event: error\n");
|
|
80
|
+
res.write(`data: ${JSON.stringify(errorResponse)}
|
|
81
|
+
|
|
82
|
+
`);
|
|
83
|
+
} finally {
|
|
84
|
+
if (!res.writableEnded) {
|
|
85
|
+
res.end();
|
|
86
|
+
}
|
|
87
|
+
}
|
|
88
|
+
}
|
|
89
|
+
handleError(req, res, error) {
|
|
90
|
+
_chunk466ED5WNcjs.loggers.server.error(
|
|
91
|
+
"Unhandled error in AINAgent A2A POST handler:",
|
|
92
|
+
error
|
|
93
|
+
);
|
|
94
|
+
const a2aError = error instanceof _server.A2AError ? error : _server.A2AError.internalError("General processing error");
|
|
95
|
+
const errorResponse = {
|
|
96
|
+
jsonrpc: "2.0",
|
|
97
|
+
id: _optionalChain([req, 'access', _7 => _7.body, 'optionalAccess', _8 => _8.id]) || null,
|
|
98
|
+
error: a2aError.toJSONRPCError()
|
|
99
|
+
};
|
|
100
|
+
if (!res.headersSent) {
|
|
101
|
+
res.status(500).json(errorResponse);
|
|
102
|
+
} else if (!res.writableEnded) {
|
|
103
|
+
res.end();
|
|
104
|
+
}
|
|
105
|
+
}
|
|
106
|
+
}, _class);
|
|
107
|
+
|
|
108
|
+
|
|
109
|
+
|
|
110
|
+
exports.A2AController = A2AController;
|
|
111
|
+
//# sourceMappingURL=chunk-ICFTUZP5.cjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-ICFTUZP5.cjs","../../src/controllers/a2a.controller.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACA;ACOA;AACC;AACA;AACA;AACA;AAAA,4CACM;AAUP,IAAM,kBAAA,EAAN,MAAA,QAAgC,8BAAsB;AAAA,EAC7C;AAAA,EAER,WAAA,CACC,aAAA,EACA,SAAA,EACA,QAAA,EACA,gBAAA,EAA4C,IAAI,4CAAA,CAAgC,CAAA,EAC/E;AACD,IAAA,MAAM,KAAA,EAAO,aAAA,CAAc,CAAA;AAC3B,IAAA,KAAA,CAAM,IAAA,EAAM,SAAA,EAAW,QAAA,EAAU,eAAe,CAAA;AAChD,IAAA,IAAA,CAAK,cAAA,EAAgB,aAAA;AAAA,EACtB;AAAA,EAEA,MAAM,YAAA,CAAA,EAAmC;AACxC,IAAA,OAAO,IAAA,CAAK,aAAA,CAAc,CAAA;AAAA,EAC3B;AACD,CAAA;AAEO,IAAM,cAAA,YAAN,MAAoB;AAAA,EAClB;AAAA,EAER,WAAA,CACC,UAAA,EACA,SAAA,EACA,aAAA,EACC;AAED,IAAA,MAAM,eAAA,EAAiB,IAAI,iBAAA;AAAA,MAC1B,aAAA;AAAA,MACA,SAAA;AAAA,MACA;AAAA;AAAA,IACD,CAAA;AACA,IAAA,IAAA,CAAK,wBAAA,EAA0B,IAAI,oCAAA,CAAwB,cAAc,CAAA;AAAA,EAC1E;AAAA;AAAA;AAAA;AAAA,iBAKO,iBAAA,EAAmB,MAAA,CAAO,GAAA,EAAc,GAAA,EAAA,GAAkB;AAChE,IAAA,IAAI;AACH,MAAA,MAAM,oBAAA,EAAsB,MAAM,IAAA,CAAK,uBAAA,CAAwB,MAAA;AAAA,QAC9D,GAAA,CAAI;AAAA,MACL,CAAA;AAGA,MAAA,GAAA,CACC,uBAAQ,mBAAA,4BAAA,CAA8B,MAAA,CAAO,aAAa,IAAA,IAC1D,UAAA,EACC;AACD,QAAA,MAAM,OAAA,EAAS,mBAAA;AAKf,QAAA,IAAA,CAAK,SAAA,CAAU,GAAA,EAAK,GAAA,EAAK,MAAM,CAAA;AAAA,MAChC,EAAA,KAAO;AAEN,QAAA,MAAM,YAAA,EAAc,mBAAA;AACpB,QAAA,GAAA,CAAI,MAAA,CAAO,GAAG,CAAA,CAAE,IAAA,CAAK,WAAW,CAAA;AAAA,MACjC;AAAA,IACD,EAAA,MAAA,CAAS,KAAA,EAAgB;AACxB,MAAA,IAAA,CAAK,WAAA,CAAY,GAAA,EAAK,GAAA,EAAK,KAAK,CAAA;AAAA,IACjC;AAAA,EACD,EAAA;AAAA,EAEA,MAAc,SAAA,CACb,GAAA,EACA,GAAA,EACA,MAAA,EACC;AACD,IAAA,GAAA,CAAI,SAAA,CAAU,cAAA,EAAgB,mBAAmB,CAAA;AACjD,IAAA,GAAA,CAAI,SAAA,CAAU,eAAA,EAAiB,UAAU,CAAA;AACzC,IAAA,GAAA,CAAI,SAAA,CAAU,YAAA,EAAc,YAAY,CAAA;AACxC,IAAA,GAAA,CAAI,YAAA,CAAa,CAAA;AAEjB,IAAA,IAAI;AACH,MAAA,IAAA,MAAA,CAAA,MAAiB,MAAA,GAAS,MAAA,EAAQ;AACjC,QAAA,GAAA,CAAI,KAAA,CAAM,CAAA,IAAA,EAAO,IAAA,CAAK,GAAA,CAAI,CAAC,CAAA;AAAA,CAAI,CAAA;AAC/B,QAAA,GAAA,CAAI,KAAA,CAAM,CAAA,MAAA,EAAS,IAAA,CAAK,SAAA,CAAU,KAAK,CAAC,CAAA;AAAA;AAAA,CAAM,CAAA;AAAA,MAC/C;AAAA,IACD,EAAA,MAAA,CAAS,WAAA,EAAsB;AAC9B,MAAA,yBAAA,CAAQ,MAAA,CAAO,KAAA;AAAA,QACd,CAAA,oCAAA,kBAAuC,GAAA,qBAAI,IAAA,6BAAM,IAAE,CAAA,EAAA,CAAA;AAAA,QACnD;AAAA,MACD,CAAA;AACA,MAAA,MAAM,SAAA,EACL,YAAA,WAAuB,iBAAA,EACpB,YAAA,EACA,gBAAA,CAAS,aAAA,CAAc,iBAAiB,CAAA;AAC5C,MAAA,MAAM,cAAA,EAAsC;AAAA,QAC3C,OAAA,EAAS,KAAA;AAAA,QACT,EAAA,kBAAI,GAAA,qBAAI,IAAA,6BAAM,KAAA,GAAM,IAAA;AAAA,QACpB,KAAA,EAAO,QAAA,CAAS,cAAA,CAAe;AAAA,MAChC,CAAA;AACA,MAAA,GAAA,CAAI,KAAA,CAAM,CAAA,IAAA,EAAO,IAAA,CAAK,GAAA,CAAI,CAAC,CAAA;AAAA,CAAI,CAAA;AAC/B,MAAA,GAAA,CAAI,KAAA,CAAM,gBAAgB,CAAA;AAC1B,MAAA,GAAA,CAAI,KAAA,CAAM,CAAA,MAAA,EAAS,IAAA,CAAK,SAAA,CAAU,aAAa,CAAC,CAAA;AAAA;AAAA,CAAM,CAAA;AAAA,IACvD,EAAA,QAAE;AACD,MAAA,GAAA,CAAI,CAAC,GAAA,CAAI,aAAA,EAAe;AACvB,QAAA,GAAA,CAAI,GAAA,CAAI,CAAA;AAAA,MACT;AAAA,IACD;AAAA,EACD;AAAA,EAEQ,WAAA,CAAY,GAAA,EAAc,GAAA,EAAe,KAAA,EAAgB;AAChE,IAAA,yBAAA,CAAQ,MAAA,CAAO,KAAA;AAAA,MACd,+CAAA;AAAA,MACA;AAAA,IACD,CAAA;AACA,IAAA,MAAM,SAAA,EACL,MAAA,WAAiB,iBAAA,EACd,MAAA,EACA,gBAAA,CAAS,aAAA,CAAc,0BAA0B,CAAA;AACrD,IAAA,MAAM,cAAA,EAAsC;AAAA,MAC3C,OAAA,EAAS,KAAA;AAAA,MACT,EAAA,kBAAI,GAAA,qBAAI,IAAA,6BAAM,KAAA,GAAM,IAAA;AAAA,MACpB,KAAA,EAAO,QAAA,CAAS,cAAA,CAAe;AAAA,IAChC,CAAA;AACA,IAAA,GAAA,CAAI,CAAC,GAAA,CAAI,WAAA,EAAa;AACrB,MAAA,GAAA,CAAI,MAAA,CAAO,GAAG,CAAA,CAAE,IAAA,CAAK,aAAa,CAAA;AAAA,IACnC,EAAA,KAAA,GAAA,CAAW,CAAC,GAAA,CAAI,aAAA,EAAe;AAC9B,MAAA,GAAA,CAAI,GAAA,CAAI,CAAA;AAAA,IACT;AAAA,EACD;AACD,UAAA;AD7CA;AACA;AACE;AACF,sCAAC","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-ICFTUZP5.cjs","sourcesContent":[null,"import type {\n\tA2AResponse,\n\tAgentCard,\n\tJSONRPCErrorResponse,\n\tJSONRPCSuccessResponse,\n} from \"@a2a-js/sdk\";\nimport type {\n\tAgentExecutor,\n\tExecutionEventBusManager,\n\tTaskStore,\n} from \"@a2a-js/sdk/server\";\nimport {\n\tA2AError,\n\tDefaultExecutionEventBusManager,\n\tDefaultRequestHandler,\n\tJsonRpcTransportHandler,\n} from \"@a2a-js/sdk/server\";\nimport type { Request, Response } from \"express\";\nimport type { A2AService } from \"@/services/a2a.service.js\";\nimport { loggers } from \"@/utils/logger.js\";\n\n/**\n * Handles the transport layer for A2A communication.\n * It orchestrates the A2A-JS SDK components and manages the HTTP request/response lifecycle.\n */\n\nclass AINRequestHandler extends DefaultRequestHandler {\n\tprivate cardGenerator: () => AgentCard;\n\n\tconstructor(\n\t\tcardGenerator: () => AgentCard,\n\t\ttaskStore: TaskStore,\n\t\texecutor: AgentExecutor,\n\t\teventBusManager: ExecutionEventBusManager = new DefaultExecutionEventBusManager(),\n\t) {\n\t\tconst card = cardGenerator();\n\t\tsuper(card, taskStore, executor, eventBusManager);\n\t\tthis.cardGenerator = cardGenerator;\n\t}\n\n\tasync getAgentCard(): Promise<AgentCard> {\n\t\treturn this.cardGenerator();\n\t}\n}\n\nexport class A2AController {\n\tprivate jsonRpcTransportHandler: JsonRpcTransportHandler;\n\n\tconstructor(\n\t\ta2aService: A2AService,\n\t\ttaskStore: TaskStore,\n\t\tcardGenerator: () => AgentCard,\n\t) {\n\t\t// The controller is responsible for setting up the SDK components.\n\t\tconst requestHandler = new AINRequestHandler(\n\t\t\tcardGenerator,\n\t\t\ttaskStore,\n\t\t\ta2aService, // Injecting the service here as the executor\n\t\t);\n\t\tthis.jsonRpcTransportHandler = new JsonRpcTransportHandler(requestHandler);\n\t}\n\n\t/**\n\t * Handles the POST /a2a request.\n\t */\n\tpublic handleA2ARequest = async (req: Request, res: Response) => {\n\t\ttry {\n\t\t\tconst rpcResponseOrStream = await this.jsonRpcTransportHandler.handle(\n\t\t\t\treq.body,\n\t\t\t);\n\n\t\t\t// Handle streaming responses (AsyncGenerator)\n\t\t\tif (\n\t\t\t\ttypeof (rpcResponseOrStream as any)?.[Symbol.asyncIterator] ===\n\t\t\t\t\"function\"\n\t\t\t) {\n\t\t\t\tconst stream = rpcResponseOrStream as AsyncGenerator<\n\t\t\t\t\tJSONRPCSuccessResponse,\n\t\t\t\t\tvoid,\n\t\t\t\t\tundefined\n\t\t\t\t>;\n\t\t\t\tthis.streamSse(req, res, stream);\n\t\t\t} else {\n\t\t\t\t// Handle single JSON-RPC response\n\t\t\t\tconst rpcResponse = rpcResponseOrStream as A2AResponse;\n\t\t\t\tres.status(200).json(rpcResponse);\n\t\t\t}\n\t\t} catch (error: unknown) {\n\t\t\tthis.handleError(req, res, error);\n\t\t}\n\t};\n\n\tprivate async streamSse(\n\t\treq: Request,\n\t\tres: Response,\n\t\tstream: AsyncGenerator<JSONRPCSuccessResponse, void, undefined>,\n\t) {\n\t\tres.setHeader(\"Content-Type\", \"text/event-stream\");\n\t\tres.setHeader(\"Cache-Control\", \"no-cache\");\n\t\tres.setHeader(\"Connection\", \"keep-alive\");\n\t\tres.flushHeaders();\n\n\t\ttry {\n\t\t\tfor await (const event of stream) {\n\t\t\t\tres.write(`id: ${Date.now()}\\n`);\n\t\t\t\tres.write(`data: ${JSON.stringify(event)}\\n\\n`);\n\t\t\t}\n\t\t} catch (streamError: unknown) {\n\t\t\tloggers.server.error(\n\t\t\t\t`Error during SSE streaming (request ${req.body?.id}):`,\n\t\t\t\tstreamError,\n\t\t\t);\n\t\t\tconst a2aError =\n\t\t\t\tstreamError instanceof A2AError\n\t\t\t\t\t? streamError\n\t\t\t\t\t: A2AError.internalError(\"Streaming error\");\n\t\t\tconst errorResponse: JSONRPCErrorResponse = {\n\t\t\t\tjsonrpc: \"2.0\",\n\t\t\t\tid: req.body?.id || null,\n\t\t\t\terror: a2aError.toJSONRPCError(),\n\t\t\t};\n\t\t\tres.write(`id: ${Date.now()}\\n`);\n\t\t\tres.write(\"event: error\\n\");\n\t\t\tres.write(`data: ${JSON.stringify(errorResponse)}\\n\\n`);\n\t\t} finally {\n\t\t\tif (!res.writableEnded) {\n\t\t\t\tres.end();\n\t\t\t}\n\t\t}\n\t}\n\n\tprivate handleError(req: Request, res: Response, error: unknown) {\n\t\tloggers.server.error(\n\t\t\t\"Unhandled error in AINAgent A2A POST handler:\",\n\t\t\terror,\n\t\t);\n\t\tconst a2aError =\n\t\t\terror instanceof A2AError\n\t\t\t\t? error\n\t\t\t\t: A2AError.internalError(\"General processing error\");\n\t\tconst errorResponse: JSONRPCErrorResponse = {\n\t\t\tjsonrpc: \"2.0\",\n\t\t\tid: req.body?.id || null,\n\t\t\terror: a2aError.toJSONRPCError(),\n\t\t};\n\t\tif (!res.headersSent) {\n\t\t\tres.status(500).json(errorResponse);\n\t\t} else if (!res.writableEnded) {\n\t\t\tres.end();\n\t\t}\n\t}\n}\n"]}
|