@ainetwork/adk 0.1.12 → 0.1.13
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +9 -4
- package/dist/cjs/chunk-36ECGO5C.cjs +269 -0
- package/dist/cjs/chunk-36ECGO5C.cjs.map +1 -0
- package/dist/cjs/chunk-3J3ORQOG.cjs +65 -0
- package/dist/cjs/chunk-3J3ORQOG.cjs.map +1 -0
- package/dist/cjs/chunk-3QNRWPMM.cjs +37 -0
- package/dist/cjs/chunk-3QNRWPMM.cjs.map +1 -0
- package/dist/cjs/chunk-4MCEFDVX.cjs +43 -0
- package/dist/cjs/chunk-4MCEFDVX.cjs.map +1 -0
- package/dist/cjs/chunk-55IME4BG.cjs +114 -0
- package/dist/cjs/chunk-55IME4BG.cjs.map +1 -0
- package/dist/cjs/chunk-5T53A5X4.cjs +12 -0
- package/dist/cjs/chunk-5T53A5X4.cjs.map +1 -0
- package/dist/cjs/chunk-65M7DC3L.cjs +172 -0
- package/dist/cjs/chunk-65M7DC3L.cjs.map +1 -0
- package/dist/cjs/chunk-6BKOR3ZH.cjs +111 -0
- package/dist/cjs/chunk-6BKOR3ZH.cjs.map +1 -0
- package/dist/cjs/chunk-6YVTJEMQ.cjs +176 -0
- package/dist/cjs/chunk-6YVTJEMQ.cjs.map +1 -0
- package/dist/cjs/chunk-7EGJAURF.cjs +205 -0
- package/dist/cjs/chunk-7EGJAURF.cjs.map +1 -0
- package/dist/cjs/chunk-7EOUPEHU.cjs +243 -0
- package/dist/cjs/chunk-7EOUPEHU.cjs.map +1 -0
- package/dist/cjs/chunk-7K2R2UKB.cjs +8 -0
- package/dist/cjs/chunk-7K2R2UKB.cjs.map +1 -0
- package/dist/cjs/chunk-7TEDNNSL.cjs +18 -0
- package/dist/cjs/chunk-7TEDNNSL.cjs.map +1 -0
- package/dist/cjs/chunk-7XDYJNTA.cjs +35 -0
- package/dist/cjs/chunk-7XDYJNTA.cjs.map +1 -0
- package/dist/cjs/chunk-BBCZU5VZ.cjs +17 -0
- package/dist/cjs/chunk-BBCZU5VZ.cjs.map +1 -0
- package/dist/cjs/chunk-BNY3RLYC.cjs +43 -0
- package/dist/cjs/chunk-BNY3RLYC.cjs.map +1 -0
- package/dist/cjs/chunk-BYD5RS6L.cjs +12 -0
- package/dist/cjs/chunk-BYD5RS6L.cjs.map +1 -0
- package/dist/cjs/chunk-EAK7MP6X.cjs +22 -0
- package/dist/cjs/chunk-EAK7MP6X.cjs.map +1 -0
- package/dist/cjs/chunk-H65RVLWK.cjs +267 -0
- package/dist/cjs/chunk-H65RVLWK.cjs.map +1 -0
- package/dist/cjs/chunk-IS2IVXXA.cjs +176 -0
- package/dist/cjs/chunk-IS2IVXXA.cjs.map +1 -0
- package/dist/{esm/chunk-3NEV3O2X.js → cjs/chunk-J4Z7ZJUJ.cjs} +57 -55
- package/dist/cjs/chunk-J4Z7ZJUJ.cjs.map +1 -0
- package/dist/cjs/chunk-J6PDRTNM.cjs +37 -0
- package/dist/cjs/chunk-J6PDRTNM.cjs.map +1 -0
- package/dist/cjs/chunk-JAPFZA2K.cjs +205 -0
- package/dist/cjs/chunk-JAPFZA2K.cjs.map +1 -0
- package/dist/cjs/chunk-K4P2MYZF.cjs +43 -0
- package/dist/cjs/chunk-K4P2MYZF.cjs.map +1 -0
- package/dist/cjs/chunk-KDNNRPBV.cjs +43 -0
- package/dist/cjs/chunk-KDNNRPBV.cjs.map +1 -0
- package/dist/cjs/chunk-KRLUWTLN.cjs +35 -0
- package/dist/cjs/chunk-KRLUWTLN.cjs.map +1 -0
- package/dist/cjs/chunk-M4MYRCSF.cjs +18 -0
- package/dist/cjs/chunk-M4MYRCSF.cjs.map +1 -0
- package/dist/cjs/chunk-MVN5XJGZ.cjs +66 -0
- package/dist/cjs/chunk-MVN5XJGZ.cjs.map +1 -0
- package/dist/cjs/chunk-O6XQRAWR.cjs +45 -0
- package/dist/cjs/chunk-O6XQRAWR.cjs.map +1 -0
- package/dist/cjs/chunk-PDF6B6A3.cjs +37 -0
- package/dist/cjs/chunk-PDF6B6A3.cjs.map +1 -0
- package/dist/cjs/chunk-PT2CVI52.cjs +162 -0
- package/dist/cjs/chunk-PT2CVI52.cjs.map +1 -0
- package/dist/cjs/chunk-QUSBMECP.cjs +37 -0
- package/dist/cjs/chunk-QUSBMECP.cjs.map +1 -0
- package/dist/cjs/chunk-SAXFX7U2.cjs +37 -0
- package/dist/cjs/chunk-SAXFX7U2.cjs.map +1 -0
- package/dist/cjs/chunk-SCOMNXDA.cjs +51 -0
- package/dist/cjs/chunk-SCOMNXDA.cjs.map +1 -0
- package/dist/cjs/chunk-SGW3CSPD.cjs +43 -0
- package/dist/cjs/chunk-SGW3CSPD.cjs.map +1 -0
- package/dist/cjs/chunk-SJHSHDJL.cjs +65 -0
- package/dist/cjs/chunk-SJHSHDJL.cjs.map +1 -0
- package/dist/cjs/chunk-UAWGHNZD.cjs +22 -0
- package/dist/cjs/chunk-UAWGHNZD.cjs.map +1 -0
- package/dist/cjs/chunk-UBEB55WC.cjs +43 -0
- package/dist/cjs/chunk-UBEB55WC.cjs.map +1 -0
- package/dist/cjs/chunk-UPNSAV5A.cjs +18 -0
- package/dist/cjs/chunk-UPNSAV5A.cjs.map +1 -0
- package/dist/cjs/chunk-UW43BI5Z.cjs +17 -0
- package/dist/cjs/chunk-UW43BI5Z.cjs.map +1 -0
- package/dist/cjs/chunk-WNDA2G4Q.cjs +43 -0
- package/dist/cjs/chunk-WNDA2G4Q.cjs.map +1 -0
- package/dist/cjs/chunk-X46GJS7X.cjs +17 -0
- package/dist/cjs/chunk-X46GJS7X.cjs.map +1 -0
- package/dist/cjs/chunk-XPD3ANDG.cjs +267 -0
- package/dist/cjs/chunk-XPD3ANDG.cjs.map +1 -0
- package/dist/cjs/chunk-XPTRG6IC.cjs +43 -0
- package/dist/cjs/chunk-XPTRG6IC.cjs.map +1 -0
- package/dist/cjs/chunk-XVMXTCJI.cjs +64 -0
- package/dist/cjs/chunk-XVMXTCJI.cjs.map +1 -0
- package/dist/cjs/chunk-YQ5P2K3X.cjs +37 -0
- package/dist/cjs/chunk-YQ5P2K3X.cjs.map +1 -0
- package/dist/cjs/controllers/api/thread.api.controller.cjs +8 -0
- package/dist/cjs/controllers/api/thread.api.controller.cjs.map +1 -0
- package/dist/cjs/controllers/index.cjs +3 -3
- package/dist/cjs/controllers/index.cjs.map +1 -1
- package/dist/cjs/controllers/query.controller.cjs +3 -2
- package/dist/cjs/controllers/query.controller.cjs.map +1 -1
- package/dist/cjs/index.cjs +17 -16
- package/dist/cjs/index.cjs.map +1 -1
- package/dist/cjs/modules/a2a/a2a.module.cjs +3 -2
- package/dist/cjs/modules/a2a/a2a.module.cjs.map +1 -1
- package/dist/cjs/modules/index.cjs +5 -4
- package/dist/cjs/modules/index.cjs.map +1 -1
- package/dist/cjs/modules/memory/memory.module.cjs +2 -2
- package/dist/cjs/modules/models/base.model.cjs +2 -2
- package/dist/cjs/routes/a2a.routes.cjs +5 -5
- package/dist/cjs/routes/api/model.routes.cjs +8 -0
- package/dist/cjs/routes/api/model.routes.cjs.map +1 -0
- package/dist/cjs/routes/api/thread.routes.cjs +9 -0
- package/dist/cjs/routes/api/thread.routes.cjs.map +1 -0
- package/dist/cjs/routes/api.routes.cjs +5 -5
- package/dist/cjs/routes/index.cjs +12 -12
- package/dist/cjs/routes/index.cjs.map +1 -1
- package/dist/cjs/routes/query.routes.cjs +7 -6
- package/dist/cjs/routes/query.routes.cjs.map +1 -1
- package/dist/cjs/services/a2a.service.cjs +2 -2
- package/dist/cjs/services/index.cjs +5 -5
- package/dist/cjs/services/query-stream.service.cjs +3 -3
- package/dist/cjs/services/query.service.cjs +3 -3
- package/dist/cjs/types/memory.cjs +2 -2
- package/dist/cjs/types/memory.cjs.map +1 -1
- package/dist/esm/chunk-2BYUKYVB.js +243 -0
- package/dist/esm/chunk-2BYUKYVB.js.map +1 -0
- package/dist/esm/{chunk-Y6E73I7I.js → chunk-2RW77MJB.js} +6 -3
- package/dist/esm/chunk-2RW77MJB.js.map +1 -0
- package/dist/esm/chunk-7D4SKPIS.js +65 -0
- package/dist/esm/chunk-7D4SKPIS.js.map +1 -0
- package/dist/esm/{chunk-5EPK37PE.js → chunk-ADIAIRNH.js} +10 -10
- package/dist/esm/chunk-ADIAIRNH.js.map +1 -0
- package/dist/esm/{chunk-2Z7GJVJJ.js → chunk-AMRKK46V.js} +2 -2
- package/dist/esm/chunk-AMRKK46V.js.map +1 -0
- package/dist/esm/{chunk-2BITOA2R.js → chunk-CDFLKADH.js} +1 -1
- package/dist/esm/chunk-CDFLKADH.js.map +1 -0
- package/dist/esm/{chunk-DRUCJYPJ.js → chunk-EGR6OTTN.js} +65 -22
- package/dist/esm/chunk-EGR6OTTN.js.map +1 -0
- package/dist/esm/{chunk-5N4X25IK.js → chunk-G6VR5RSV.js} +5 -5
- package/dist/esm/chunk-HOFAK5J2.js +22 -0
- package/dist/esm/chunk-HOFAK5J2.js.map +1 -0
- package/dist/esm/{chunk-GNWF6QG4.js → chunk-JCWH7LIC.js} +6 -6
- package/dist/esm/{chunk-C6SGV7JV.js → chunk-LJVVSUF7.js} +17 -13
- package/dist/esm/chunk-LJVVSUF7.js.map +1 -0
- package/dist/esm/chunk-MNAWSBYP.js +18 -0
- package/dist/esm/chunk-MNAWSBYP.js.map +1 -0
- package/dist/esm/chunk-VA3XOZBV.js +17 -0
- package/dist/esm/chunk-VA3XOZBV.js.map +1 -0
- package/dist/esm/{chunk-CKTURH34.js → chunk-XBMWWTSX.js} +9 -9
- package/dist/esm/chunk-XBMWWTSX.js.map +1 -0
- package/dist/esm/controllers/a2a.controller.d.ts +1 -1
- package/dist/esm/controllers/api/{session.api.controller.d.ts → thread.api.controller.d.ts} +3 -3
- package/dist/esm/controllers/api/thread.api.controller.js +8 -0
- package/dist/esm/controllers/index.d.ts +2 -2
- package/dist/esm/controllers/index.js +7 -7
- package/dist/esm/controllers/query.controller.d.ts +1 -1
- package/dist/esm/controllers/query.controller.js +2 -1
- package/dist/esm/index.d.ts +1 -0
- package/dist/esm/index.js +17 -16
- package/dist/esm/index.js.map +1 -1
- package/dist/esm/modules/a2a/a2a.module.d.ts +6 -15
- package/dist/esm/modules/a2a/a2a.module.js +2 -1
- package/dist/esm/modules/index.d.ts +1 -1
- package/dist/esm/modules/index.js +8 -7
- package/dist/esm/modules/memory/base.memory.d.ts +9 -9
- package/dist/esm/modules/memory/memory.module.d.ts +4 -4
- package/dist/esm/modules/memory/memory.module.js +1 -1
- package/dist/esm/modules/models/base.model.d.ts +4 -4
- package/dist/esm/modules/models/base.model.js +1 -1
- package/dist/esm/routes/a2a.routes.js +4 -4
- package/dist/esm/routes/api/{models.routes.js → model.routes.js} +2 -2
- package/dist/esm/routes/api/{sessions.routes.d.ts → thread.routes.d.ts} +2 -2
- package/dist/esm/routes/api/thread.routes.js +9 -0
- package/dist/esm/routes/api.routes.js +4 -4
- package/dist/esm/routes/index.d.ts +2 -2
- package/dist/esm/routes/index.js +16 -16
- package/dist/esm/routes/query.routes.js +6 -5
- package/dist/esm/services/a2a.service.d.ts +1 -1
- package/dist/esm/services/a2a.service.js +1 -1
- package/dist/esm/services/index.d.ts +1 -2
- package/dist/esm/services/index.js +6 -6
- package/dist/esm/services/query-stream.service.d.ts +13 -8
- package/dist/esm/services/query-stream.service.js +2 -2
- package/dist/esm/services/query.service.d.ts +19 -8
- package/dist/esm/services/query.service.js +2 -2
- package/dist/esm/types/memory.d.ts +32 -32
- package/dist/esm/types/memory.js +5 -5
- package/dist/esm/types/stream.d.ts +32 -2
- package/package.json +1 -1
- package/dist/esm/chunk-2BITOA2R.js.map +0 -1
- package/dist/esm/chunk-2Z7GJVJJ.js.map +0 -1
- package/dist/esm/chunk-3NEV3O2X.js.map +0 -1
- package/dist/esm/chunk-5EPK37PE.js.map +0 -1
- package/dist/esm/chunk-C6SGV7JV.js.map +0 -1
- package/dist/esm/chunk-CKTURH34.js.map +0 -1
- package/dist/esm/chunk-DRUCJYPJ.js.map +0 -1
- package/dist/esm/chunk-Q6HHVAO6.js +0 -45
- package/dist/esm/chunk-Q6HHVAO6.js.map +0 -1
- package/dist/esm/chunk-QROUPWWO.js +0 -17
- package/dist/esm/chunk-QROUPWWO.js.map +0 -1
- package/dist/esm/chunk-VAZWF5IK.js +0 -20
- package/dist/esm/chunk-VAZWF5IK.js.map +0 -1
- package/dist/esm/chunk-Y6E73I7I.js.map +0 -1
- package/dist/esm/chunk-YT6PD6UF.js +0 -22
- package/dist/esm/chunk-YT6PD6UF.js.map +0 -1
- package/dist/esm/controllers/api/session.api.controller.js +0 -8
- package/dist/esm/routes/api/sessions.routes.js +0 -9
- /package/dist/esm/{chunk-5N4X25IK.js.map → chunk-G6VR5RSV.js.map} +0 -0
- /package/dist/esm/{chunk-GNWF6QG4.js.map → chunk-JCWH7LIC.js.map} +0 -0
- /package/dist/esm/controllers/api/{session.api.controller.js.map → thread.api.controller.js.map} +0 -0
- /package/dist/esm/routes/api/{models.routes.d.ts → model.routes.d.ts} +0 -0
- /package/dist/esm/routes/api/{models.routes.js.map → model.routes.js.map} +0 -0
- /package/dist/esm/routes/api/{sessions.routes.js.map → thread.routes.js.map} +0 -0
|
@@ -0,0 +1,12 @@
|
|
|
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
|
+
|
|
9
|
+
|
|
10
|
+
|
|
11
|
+
exports.MessageRole = MessageRole;
|
|
12
|
+
//# sourceMappingURL=chunk-5T53A5X4.cjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-5T53A5X4.cjs","../../src/types/memory.ts"],"names":["MessageRole"],"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;ADIZ;AACA;AACE;AACF,kCAAC","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-5T53A5X4.cjs","sourcesContent":[null,"/**\n * Roles for participants in a message.\n */\nexport enum MessageRole {\n\t/** User/human participant */\n\tUSER = \"USER\",\n\t/** System-generated messages or instructions */\n\tSYSTEM = \"SYSTEM\",\n\t/** AI model responses */\n\tMODEL = \"MODEL\",\n}\n\n/**\n * Content structure for message content.\n *\n * Supports multi-part content with different types (text, images, etc.).\n */\nexport type MessageContentObject = {\n\t/** Content type (e.g., \"text\", \"image\", \"tool_use\") */\n\ttype: string;\n\t/** Array of content parts, structure depends on content type */\n\tparts: any[];\n};\n\n/**\n * Represents a single message in a thread.\n *\n * @example\n * ```typescript\n * const message: MessageObject = {\n * role: MessageRole.USER,\n * content: {\n * type: \"text\",\n * parts: [\"Hello, how can you help me?\"]\n * },\n * timestamp: Date.now(),\n * metadata: { source: \"web-ui\" }\n * };\n * ```\n */\nexport type MessageObject = {\n\t/** Role of the message sender */\n\trole: MessageRole;\n\t/** Message content with type and parts */\n\tcontent: MessageContentObject;\n\t/** Unix timestamp when the message was created */\n\ttimestamp: number;\n\t/** Optional metadata for additional context */\n\tmetadata?: { [key: string]: unknown };\n};\n\nexport type ThreadMetadata = {\n\ttype: \"workflow\" | \"chat\";\n\ttitle: string;\n\tthreadId: string;\n\tupdatedAt: number;\n};\n\n/**\n * Represents a conversation thread containing multiple messages.\n *\n * Messages are stored in a key-value structure where keys are unique message IDs\n * and values are the corresponding message objects.\n *\n * @example\n * ```typescript\n * const thread: ThreadObject = {\n * \t title: \"New conversation\",\n * messages: {\n * \"<UUID_1>\": { role: MessageRole.USER, content: {...}, timestamp: 1234567890 },\n * \"<UUID_2>\": { role: MessageRole.MODEL, content: {...}, timestamp: 1234567891 }\n * }\n * };\n * ```\n */\nexport type ThreadObject = {\n\ttype: \"workflow\" | \"chat\";\n\ttitle: string;\n\t/* Collection of messages indexed by unique message ID */\n\tmessages: {\n\t\t[messageId: string]: MessageObject;\n\t};\n};\n\nexport interface Intent {\n\tname: string;\n\tdescription: string;\n\tprompt?: string;\n\tllm?: string;\n}\n"]}
|
|
@@ -0,0 +1,172 @@
|
|
|
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 threadId - The session identifier
|
|
58
|
+
* @returns A2ASession object with task and context IDs
|
|
59
|
+
*/
|
|
60
|
+
__init3() {this.getA2ASessionWithId = (threadId) => {
|
|
61
|
+
const a2aSession = _nullishCoalesce(this.a2aSessions.get(threadId), () => ( {
|
|
62
|
+
taskId: void 0,
|
|
63
|
+
contextId: void 0
|
|
64
|
+
}));
|
|
65
|
+
if (!this.a2aSessions.has(threadId)) {
|
|
66
|
+
this.a2aSessions.set(threadId, 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 threadId - The session identifier
|
|
78
|
+
* @returns Formatted Message object for A2A protocol
|
|
79
|
+
*/
|
|
80
|
+
getMessagePayload(query, threadId) {
|
|
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
|
+
threadId
|
|
88
|
+
},
|
|
89
|
+
parts: [
|
|
90
|
+
{
|
|
91
|
+
kind: "text",
|
|
92
|
+
text: query
|
|
93
|
+
}
|
|
94
|
+
]
|
|
95
|
+
};
|
|
96
|
+
const a2aSession = this.getA2ASessionWithId(threadId);
|
|
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 threadId - The session identifier for context tracking
|
|
114
|
+
* @returns Promise resolving to array of text responses from the agent
|
|
115
|
+
*/
|
|
116
|
+
async useTool(tool, messagePayload, threadId) {
|
|
117
|
+
const finalText = [];
|
|
118
|
+
const client = tool.client;
|
|
119
|
+
const params = {
|
|
120
|
+
message: messagePayload
|
|
121
|
+
};
|
|
122
|
+
const a2aSession = this.getA2ASessionWithId(threadId);
|
|
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
|
+
const toolResult = `[Bot Called A2A Tool ${tool.card.name}]
|
|
161
|
+
${typeof error === "string" ? error : JSON.stringify(error, null, 2)}`;
|
|
162
|
+
return toolResult;
|
|
163
|
+
}
|
|
164
|
+
return `[Bot Called A2A Tool ${tool.card.name}]
|
|
165
|
+
${finalText.join("\n")}`;
|
|
166
|
+
}
|
|
167
|
+
}, _class);
|
|
168
|
+
|
|
169
|
+
|
|
170
|
+
|
|
171
|
+
exports.A2AModule = A2AModule;
|
|
172
|
+
//# sourceMappingURL=chunk-65M7DC3L.cjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-65M7DC3L.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,QAAA,EAAA,GAAiC;AAC/D,IAAA,MAAM,WAAA,mBAAa,IAAA,CAAK,WAAA,CAAY,GAAA,CAAI,QAAQ,CAAA,UAAK;AAAA,MACpD,MAAA,EAAQ,KAAA,CAAA;AAAA,MACR,SAAA,EAAW,KAAA;AAAA,IACZ,GAAA;AACA,IAAA,GAAA,CAAI,CAAC,IAAA,CAAK,WAAA,CAAY,GAAA,CAAI,QAAQ,CAAA,EAAG;AACpC,MAAA,IAAA,CAAK,WAAA,CAAY,GAAA,CAAI,QAAA,EAAU,UAAU,CAAA;AAAA,IAC1C;AAEA,IAAA,OAAO,UAAA;AAAA,EACR,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAYO,iBAAA,CAAkB,KAAA,EAAe,QAAA,EAA2B;AAClE,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,QAAQ,CAAA;AACpD,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,QAAA,EACkB;AAClB,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,QAAQ,CAAA;AAEpD,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,EAAO;AACf,MAAA,yBAAA,CAAQ,GAAA,CAAI,KAAA,CAAM,iCAAA,EAAmC,EAAE,MAAM,CAAC,CAAA;AAC9D,MAAA,IAAA,CAAK,OAAA,CAAQ,CAAA;AACb,MAAA,MAAM,WAAA,EAAa,CAAA,qBAAA,EAAwB,IAAA,CAAK,IAAA,CAAK,IAAI,CAAA;AAAA,EAAM,OAAO,MAAA,IAAU,SAAA,EAAW,MAAA,EAAQ,IAAA,CAAK,SAAA,CAAU,KAAA,EAAO,IAAA,EAAM,CAAC,CAAC,CAAA,CAAA;AAC1H,MAAA;AACR,IAAA;AAE6C,IAAA;AAA0B;AACxE,EAAA;AACD;ADnDqE;AACA;AACA;AACA","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-65M7DC3L.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 threadId - The session identifier\n\t * @returns A2ASession object with task and context IDs\n\t */\n\tprivate getA2ASessionWithId = (threadId: string): A2ASession => {\n\t\tconst a2aSession = this.a2aSessions.get(threadId) ?? {\n\t\t\ttaskId: undefined,\n\t\t\tcontextId: undefined,\n\t\t};\n\t\tif (!this.a2aSessions.has(threadId)) {\n\t\t\tthis.a2aSessions.set(threadId, 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 threadId - The session identifier\n\t * @returns Formatted Message object for A2A protocol\n\t */\n\tpublic getMessagePayload(query: string, threadId: 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\tthreadId,\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(threadId);\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 threadId - 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\tthreadId: 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(threadId);\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) {\n\t\t\tloggers.a2a.error(\"Error communicating with agent:\", { error });\n\t\t\ttool.disable();\n\t\t\tconst toolResult = `[Bot Called A2A Tool ${tool.card.name}]\\n${typeof error === \"string\" ? error : JSON.stringify(error, null, 2)}`;\n\t\t\treturn toolResult;\n\t\t}\n\n\t\treturn `[Bot Called A2A Tool ${tool.card.name}]\\n${finalText.join(\"\\n\")}`;\n\t}\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/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 { threadId } = 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, threadId);
|
|
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-6BKOR3ZH.cjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-6BKOR3ZH.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,SAAS,EAAA,EAAI,WAAA,CAAY,QAAA;AACjC,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,QAAQ,CAAA;AAEtE,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-6BKOR3ZH.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 { threadId } = userMessage.metadata as { threadId: 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, threadId);\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,176 @@
|
|
|
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);_class.prototype.__init4.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
|
+
__init3() {this.agentId = _crypto.randomUUID.call(void 0, )}
|
|
17
|
+
/* FIXME */
|
|
18
|
+
/**
|
|
19
|
+
* Registers a new A2A peer server URL for connection.
|
|
20
|
+
*
|
|
21
|
+
* @param url - The URL of the A2A-compatible agent to connect to
|
|
22
|
+
*/
|
|
23
|
+
async addA2APeerServer(url) {
|
|
24
|
+
this.a2aPeerServers.set(url, null);
|
|
25
|
+
}
|
|
26
|
+
/**
|
|
27
|
+
* Retrieves tools from all registered A2A peer servers.
|
|
28
|
+
*
|
|
29
|
+
* Attempts to connect to each registered server, fetch their agent cards,
|
|
30
|
+
* and create tool instances. Disables tools for unreachable servers.
|
|
31
|
+
*
|
|
32
|
+
* @returns Promise resolving to array of available A2A tools
|
|
33
|
+
*/
|
|
34
|
+
async getTools() {
|
|
35
|
+
const tools = [];
|
|
36
|
+
for (const url of [...this.a2aPeerServers.keys()]) {
|
|
37
|
+
const tool = this.a2aPeerServers.get(url);
|
|
38
|
+
if (!tool || !tool.enabled) {
|
|
39
|
+
try {
|
|
40
|
+
const client = new (0, _client.A2AClient)(url);
|
|
41
|
+
const card = await client.getAgentCard();
|
|
42
|
+
const toolName = card.name.replaceAll(" ", "-");
|
|
43
|
+
const a2aTool = new (0, _chunkF7I4G6JGcjs.A2ATool)(toolName, client, card);
|
|
44
|
+
tools.push(a2aTool);
|
|
45
|
+
} catch (_error) {
|
|
46
|
+
if (tool) {
|
|
47
|
+
tool.disable();
|
|
48
|
+
}
|
|
49
|
+
}
|
|
50
|
+
} else {
|
|
51
|
+
tools.push(tool);
|
|
52
|
+
}
|
|
53
|
+
}
|
|
54
|
+
return tools;
|
|
55
|
+
}
|
|
56
|
+
/**
|
|
57
|
+
* Gets or creates an A2A session for the given session ID.
|
|
58
|
+
*
|
|
59
|
+
* @param threadId - The session identifier
|
|
60
|
+
* @returns A2ASession object with task and context IDs
|
|
61
|
+
*/
|
|
62
|
+
__init4() {this.getA2ASessionWithId = (threadId) => {
|
|
63
|
+
const a2aSession = _nullishCoalesce(this.a2aSessions.get(threadId), () => ( {
|
|
64
|
+
taskId: void 0,
|
|
65
|
+
contextId: void 0
|
|
66
|
+
}));
|
|
67
|
+
if (!this.a2aSessions.has(threadId)) {
|
|
68
|
+
this.a2aSessions.set(threadId, a2aSession);
|
|
69
|
+
}
|
|
70
|
+
return a2aSession;
|
|
71
|
+
}}
|
|
72
|
+
/**
|
|
73
|
+
* Constructs a message payload for A2A communication.
|
|
74
|
+
*
|
|
75
|
+
* Includes session context (task ID and context ID) if available
|
|
76
|
+
* for maintaining conversation continuity.
|
|
77
|
+
*
|
|
78
|
+
* @param query - The message content to send
|
|
79
|
+
* @param threadId - The session identifier
|
|
80
|
+
* @returns Formatted Message object for A2A protocol
|
|
81
|
+
*/
|
|
82
|
+
getMessagePayload(query, threadId) {
|
|
83
|
+
const messagePayload = {
|
|
84
|
+
messageId: _crypto.randomUUID.call(void 0, ),
|
|
85
|
+
kind: "message",
|
|
86
|
+
role: "user",
|
|
87
|
+
// FIXME: it could be 'agent'
|
|
88
|
+
metadata: {
|
|
89
|
+
agentId: this.agentId,
|
|
90
|
+
type: "CHAT" /* CHAT */,
|
|
91
|
+
threadId
|
|
92
|
+
},
|
|
93
|
+
parts: [
|
|
94
|
+
{
|
|
95
|
+
kind: "text",
|
|
96
|
+
text: query
|
|
97
|
+
}
|
|
98
|
+
]
|
|
99
|
+
};
|
|
100
|
+
const a2aSession = this.getA2ASessionWithId(threadId);
|
|
101
|
+
if (a2aSession.taskId) {
|
|
102
|
+
messagePayload.taskId = a2aSession.taskId;
|
|
103
|
+
}
|
|
104
|
+
if (a2aSession.contextId) {
|
|
105
|
+
messagePayload.contextId = a2aSession.contextId;
|
|
106
|
+
}
|
|
107
|
+
return messagePayload;
|
|
108
|
+
}
|
|
109
|
+
/**
|
|
110
|
+
* Executes an A2A tool by sending a message to the remote agent.
|
|
111
|
+
*
|
|
112
|
+
* Handles streaming responses, maintains session state, and extracts
|
|
113
|
+
* text content from various event types in the response stream.
|
|
114
|
+
*
|
|
115
|
+
* @param tool - The A2ATool instance to use
|
|
116
|
+
* @param messagePayload - The message to send to the agent
|
|
117
|
+
* @param threadId - The session identifier for context tracking
|
|
118
|
+
* @returns Promise resolving to array of text responses from the agent
|
|
119
|
+
*/
|
|
120
|
+
async useTool(tool, messagePayload, threadId) {
|
|
121
|
+
const finalText = [];
|
|
122
|
+
const client = tool.client;
|
|
123
|
+
const params = {
|
|
124
|
+
message: messagePayload
|
|
125
|
+
};
|
|
126
|
+
const a2aSession = this.getA2ASessionWithId(threadId);
|
|
127
|
+
try {
|
|
128
|
+
const stream = client.sendMessageStream(params);
|
|
129
|
+
for await (const event of stream) {
|
|
130
|
+
if (event.kind === "status-update") {
|
|
131
|
+
const typedEvent = event;
|
|
132
|
+
if (typedEvent.final && typedEvent.status.state !== "input-required") {
|
|
133
|
+
a2aSession.taskId = void 0;
|
|
134
|
+
}
|
|
135
|
+
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")]);
|
|
136
|
+
if (texts) {
|
|
137
|
+
finalText.push(texts);
|
|
138
|
+
}
|
|
139
|
+
} else if (event.kind === "message") {
|
|
140
|
+
const msg = event;
|
|
141
|
+
if (msg.taskId && msg.taskId !== a2aSession.taskId) {
|
|
142
|
+
a2aSession.taskId = msg.taskId;
|
|
143
|
+
}
|
|
144
|
+
if (msg.contextId && msg.contextId !== a2aSession.contextId) {
|
|
145
|
+
a2aSession.contextId = msg.contextId;
|
|
146
|
+
}
|
|
147
|
+
} else if (event.kind === "task") {
|
|
148
|
+
const task = event;
|
|
149
|
+
if (task.id !== a2aSession.taskId) {
|
|
150
|
+
a2aSession.taskId = task.id;
|
|
151
|
+
}
|
|
152
|
+
if (task.contextId && task.contextId !== a2aSession.contextId) {
|
|
153
|
+
a2aSession.contextId = task.contextId;
|
|
154
|
+
}
|
|
155
|
+
} else {
|
|
156
|
+
_chunk466ED5WNcjs.loggers.a2a.warn("Received unknown event structure from stream:", {
|
|
157
|
+
event
|
|
158
|
+
});
|
|
159
|
+
}
|
|
160
|
+
}
|
|
161
|
+
} catch (error) {
|
|
162
|
+
_chunk466ED5WNcjs.loggers.a2a.error("Error communicating with agent:", { error });
|
|
163
|
+
tool.disable();
|
|
164
|
+
const toolResult = `[Bot Called A2A Tool ${tool.card.name}]
|
|
165
|
+
${typeof error === "string" ? error : JSON.stringify(error, null, 2)}`;
|
|
166
|
+
return toolResult;
|
|
167
|
+
}
|
|
168
|
+
return `[Bot Called A2A Tool ${tool.card.name}]
|
|
169
|
+
${finalText.join("\n")}`;
|
|
170
|
+
}
|
|
171
|
+
}, _class);
|
|
172
|
+
|
|
173
|
+
|
|
174
|
+
|
|
175
|
+
exports.A2AModule = A2AModule;
|
|
176
|
+
//# sourceMappingURL=chunk-6YVTJEMQ.cjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-6YVTJEMQ.cjs","../../src/modules/a2a/a2a.module.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACA;ACPA,gCAA2B;AAS3B,4CAA0B;AAsBnB,IAAM,UAAA,YAAN,MAAgB;AAAA;AAAA,iBAEd,eAAA,kBAA8C,IAAI,GAAA,CAAI,EAAA;AAAA;AAAA,kBAEtD,YAAA,kBAAuC,IAAI,GAAA,CAAI,EAAA;AAAA,kBAC/C,QAAA,EAAkB,gCAAA,EAAW;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOrC,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,QAAA,EAAA,GAAiC;AAC/D,IAAA,MAAM,WAAA,mBAAa,IAAA,CAAK,WAAA,CAAY,GAAA,CAAI,QAAQ,CAAA,UAAK;AAAA,MACpD,MAAA,EAAQ,KAAA,CAAA;AAAA,MACR,SAAA,EAAW,KAAA;AAAA,IACZ,GAAA;AACA,IAAA,GAAA,CAAI,CAAC,IAAA,CAAK,WAAA,CAAY,GAAA,CAAI,QAAQ,CAAA,EAAG;AACpC,MAAA,IAAA,CAAK,WAAA,CAAY,GAAA,CAAI,QAAA,EAAU,UAAU,CAAA;AAAA,IAC1C;AAEA,IAAA,OAAO,UAAA;AAAA,EACR,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAYO,iBAAA,CAAkB,KAAA,EAAe,QAAA,EAA2B;AAClE,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,OAAA,EAAS,IAAA,CAAK,OAAA;AAAA,QACd,IAAA,EAAA,iBAAA;AAAA,QACA;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,QAAQ,CAAA;AACpD,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,QAAA,EACkB;AAClB,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,QAAQ,CAAA;AAEpD,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,EAAO;AACf,MAAA,yBAAA,CAAQ,GAAA,CAAI,KAAA,CAAM,iCAAA,EAAmC,EAAE,MAAM,CAAC,CAAA;AAC9D,MAAA,IAAA,CAAK,OAAA,CAAQ,CAAA;AACb,MAAA,MAAM,WAAA,EAAa,CAAA,qBAAA,EAAwB,IAAA,CAAK,IAAA,CAAK,IAAI,CAAA;AAAA,EAAM,OAAO,MAAA,IAAU,SAAA,EAAW,MAAA,EAAQ,IAAA,CAAK,SAAA,CAAU,KAAA,EAAO,IAAA,EAAM,CAAC,CAAC,CAAA,CAAA;AAC1H,MAAA;AACR,IAAA;AAE6C,IAAA;AAA0B;AACxE,EAAA;AACD;ADzCqE;AACA;AACA;AACA","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-6YVTJEMQ.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 { ThreadType } from \"@/types/memory.js\";\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 */\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\tprivate agentId: string = randomUUID(); /* FIXME */\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 threadId - The session identifier\n\t * @returns A2ASession object with task and context IDs\n\t */\n\tprivate getA2ASessionWithId = (threadId: string): A2ASession => {\n\t\tconst a2aSession = this.a2aSessions.get(threadId) ?? {\n\t\t\ttaskId: undefined,\n\t\t\tcontextId: undefined,\n\t\t};\n\t\tif (!this.a2aSessions.has(threadId)) {\n\t\t\tthis.a2aSessions.set(threadId, 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 threadId - The session identifier\n\t * @returns Formatted Message object for A2A protocol\n\t */\n\tpublic getMessagePayload(query: string, threadId: 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\tagentId: this.agentId,\n\t\t\t\ttype: ThreadType.CHAT,\n\t\t\t\tthreadId,\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(threadId);\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 threadId - 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\tthreadId: 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(threadId);\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) {\n\t\t\tloggers.a2a.error(\"Error communicating with agent:\", { error });\n\t\t\ttool.disable();\n\t\t\tconst toolResult = `[Bot Called A2A Tool ${tool.card.name}]\\n${typeof error === \"string\" ? error : JSON.stringify(error, null, 2)}`;\n\t\t\treturn toolResult;\n\t\t}\n\n\t\treturn `[Bot Called A2A Tool ${tool.card.name}]\\n${finalText.join(\"\\n\")}`;\n\t}\n}\n"]}
|