@ainetwork/adk 0.2.1 → 0.2.3

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.
Files changed (190) hide show
  1. package/README.md +50 -24
  2. package/dist/cjs/{chunk-EOPW5OPE.cjs → chunk-77SJGFVU.cjs} +7 -7
  3. package/dist/cjs/{chunk-4DXS33VP.cjs.map → chunk-77SJGFVU.cjs.map} +1 -1
  4. package/dist/cjs/{chunk-GB3ON7UV.cjs → chunk-7PZP7YSA.cjs} +5 -5
  5. package/dist/cjs/{chunk-GB3ON7UV.cjs.map → chunk-7PZP7YSA.cjs.map} +1 -1
  6. package/dist/cjs/{chunk-ZZK4ACTV.cjs → chunk-A2ANDQNY.cjs} +75 -27
  7. package/dist/cjs/chunk-A2ANDQNY.cjs.map +1 -0
  8. package/dist/cjs/{chunk-MIWLK57K.cjs → chunk-A2I7D6JW.cjs} +1 -1
  9. package/dist/cjs/chunk-A2I7D6JW.cjs.map +1 -0
  10. package/dist/cjs/{chunk-6YVTJEMQ.cjs → chunk-BBFYVWE2.cjs} +17 -42
  11. package/dist/cjs/chunk-BBFYVWE2.cjs.map +1 -0
  12. package/dist/cjs/{chunk-4DXS33VP.cjs → chunk-CHC3HECE.cjs} +7 -7
  13. package/dist/cjs/{chunk-EOPW5OPE.cjs.map → chunk-CHC3HECE.cjs.map} +1 -1
  14. package/dist/cjs/{chunk-LRBRG2DS.cjs → chunk-E35E4LQY.cjs} +4 -4
  15. package/dist/cjs/chunk-E35E4LQY.cjs.map +1 -0
  16. package/dist/cjs/{chunk-45G7LDK7.cjs → chunk-FRWC3FQ4.cjs} +3 -3
  17. package/dist/cjs/{chunk-45G7LDK7.cjs.map → chunk-FRWC3FQ4.cjs.map} +1 -1
  18. package/dist/cjs/{chunk-2C33PZTV.cjs → chunk-HCN6JEJA.cjs} +6 -6
  19. package/dist/cjs/chunk-HCN6JEJA.cjs.map +1 -0
  20. package/dist/cjs/chunk-J7GCQY3S.cjs +18 -0
  21. package/dist/cjs/chunk-J7GCQY3S.cjs.map +1 -0
  22. package/dist/cjs/{chunk-AIE3SVG7.cjs → chunk-KHD7UAFW.cjs} +22 -24
  23. package/dist/cjs/chunk-KHD7UAFW.cjs.map +1 -0
  24. package/dist/cjs/{chunk-BOKIE5SM.cjs → chunk-LMR6B76V.cjs} +5 -5
  25. package/dist/cjs/chunk-LMR6B76V.cjs.map +1 -0
  26. package/dist/cjs/{chunk-ZZPJ5CPR.cjs → chunk-MCDRCKR6.cjs} +29 -24
  27. package/dist/cjs/chunk-MCDRCKR6.cjs.map +1 -0
  28. package/dist/cjs/{chunk-344VLDMS.cjs → chunk-O4QQH7VT.cjs} +7 -7
  29. package/dist/cjs/{chunk-RX72NXXR.cjs.map → chunk-O4QQH7VT.cjs.map} +1 -1
  30. package/dist/cjs/{chunk-55IME4BG.cjs → chunk-OKRBMMK2.cjs} +46 -24
  31. package/dist/cjs/chunk-OKRBMMK2.cjs.map +1 -0
  32. package/dist/cjs/{chunk-7GWYMGWW.cjs → chunk-PAWGHHSY.cjs} +15 -14
  33. package/dist/cjs/chunk-PAWGHHSY.cjs.map +1 -0
  34. package/dist/cjs/{chunk-DX2JH3FJ.cjs → chunk-QHQLU7BI.cjs} +1 -1
  35. package/dist/cjs/{chunk-DX2JH3FJ.cjs.map → chunk-QHQLU7BI.cjs.map} +1 -1
  36. package/dist/cjs/{chunk-QWASSVCU.cjs → chunk-RIJV7J5Z.cjs} +1 -1
  37. package/dist/cjs/chunk-RIJV7J5Z.cjs.map +1 -0
  38. package/dist/cjs/{chunk-HT3PACCL.cjs → chunk-UPQVXJK6.cjs} +6 -6
  39. package/dist/cjs/chunk-UPQVXJK6.cjs.map +1 -0
  40. package/dist/cjs/{chunk-3J3ORQOG.cjs → chunk-VLMHDHXT.cjs} +3 -3
  41. package/dist/cjs/{chunk-3J3ORQOG.cjs.map → chunk-VLMHDHXT.cjs.map} +1 -1
  42. package/dist/cjs/{chunk-RX72NXXR.cjs → chunk-W5VM75PL.cjs} +7 -7
  43. package/dist/cjs/{chunk-344VLDMS.cjs.map → chunk-W5VM75PL.cjs.map} +1 -1
  44. package/dist/cjs/{chunk-JXKLEQD6.cjs → chunk-WUOME625.cjs} +6 -6
  45. package/dist/cjs/chunk-WUOME625.cjs.map +1 -0
  46. package/dist/cjs/{chunk-CU5MOHJU.cjs → chunk-ZPYMLCQZ.cjs} +13 -14
  47. package/dist/cjs/chunk-ZPYMLCQZ.cjs.map +1 -0
  48. package/dist/cjs/controllers/api/threads.api.controller.cjs +2 -2
  49. package/dist/cjs/controllers/index.cjs +7 -7
  50. package/dist/cjs/controllers/index.cjs.map +1 -1
  51. package/dist/cjs/controllers/query.controller.cjs +3 -3
  52. package/dist/cjs/index.cjs +46 -32
  53. package/dist/cjs/index.cjs.map +1 -1
  54. package/dist/cjs/middlewares/auth.middleware.cjs +3 -3
  55. package/dist/cjs/modules/a2a/a2a.module.cjs +3 -3
  56. package/dist/cjs/modules/index.cjs +9 -9
  57. package/dist/cjs/modules/index.cjs.map +1 -1
  58. package/dist/cjs/modules/models/model.module.cjs +2 -2
  59. package/dist/cjs/routes/a2a.routes.cjs +7 -6
  60. package/dist/cjs/routes/a2a.routes.cjs.map +1 -1
  61. package/dist/cjs/routes/api/threads.routes.cjs +4 -4
  62. package/dist/cjs/routes/api.routes.cjs +5 -5
  63. package/dist/cjs/routes/index.cjs +17 -17
  64. package/dist/cjs/routes/index.cjs.map +1 -1
  65. package/dist/cjs/routes/query.routes.cjs +7 -7
  66. package/dist/cjs/services/a2a.service.cjs +2 -2
  67. package/dist/cjs/services/index.cjs +6 -5
  68. package/dist/cjs/services/index.cjs.map +1 -1
  69. package/dist/cjs/services/query-stream.service.cjs +4 -3
  70. package/dist/cjs/services/query-stream.service.cjs.map +1 -1
  71. package/dist/cjs/services/query.service.cjs +3 -3
  72. package/dist/cjs/types/agent.cjs +2 -2
  73. package/dist/cjs/types/memory.cjs +2 -2
  74. package/dist/esm/{chunk-7D4SKPIS.js → chunk-4C2TLCWI.js} +2 -2
  75. package/dist/esm/{chunk-EIHZ4S5V.js → chunk-4R4TB2ZJ.js} +6 -6
  76. package/dist/esm/{chunk-PN6NMQBM.js → chunk-5JIXQ6PU.js} +2 -2
  77. package/dist/esm/chunk-5JIXQ6PU.js.map +1 -0
  78. package/dist/esm/{chunk-CWFIK2EK.js → chunk-KDVMHOLB.js} +11 -10
  79. package/dist/esm/chunk-KDVMHOLB.js.map +1 -0
  80. package/dist/esm/{chunk-LJVVSUF7.js → chunk-MJEPJ62P.js} +15 -40
  81. package/dist/esm/chunk-MJEPJ62P.js.map +1 -0
  82. package/dist/esm/{chunk-2L2KJNAM.js → chunk-NIT7ITZV.js} +6 -6
  83. package/dist/esm/chunk-NIT7ITZV.js.map +1 -0
  84. package/dist/esm/{chunk-MNAWSBYP.js → chunk-POQFOL6Y.js} +1 -1
  85. package/dist/esm/chunk-POQFOL6Y.js.map +1 -0
  86. package/dist/esm/{chunk-2RW77MJB.js → chunk-PUUZRUE6.js} +47 -25
  87. package/dist/esm/chunk-PUUZRUE6.js.map +1 -0
  88. package/dist/esm/{chunk-SYCFBEUA.js → chunk-QHWDMU43.js} +3 -3
  89. package/dist/esm/{chunk-WLT2B2SZ.js → chunk-RL6SRSSK.js} +1 -1
  90. package/dist/esm/chunk-RL6SRSSK.js.map +1 -0
  91. package/dist/esm/{chunk-VDXWZ4NK.js → chunk-TV2GJAEH.js} +2 -2
  92. package/dist/esm/{chunk-ILKEX4ZV.js → chunk-UK6K6ZPH.js} +26 -17
  93. package/dist/esm/chunk-UK6K6ZPH.js.map +1 -0
  94. package/dist/esm/{chunk-PHKK5NHK.js → chunk-VNQCUGC3.js} +1 -1
  95. package/dist/esm/chunk-VNQCUGC3.js.map +1 -0
  96. package/dist/esm/{chunk-FL3JRQM4.js → chunk-ZBUAT73P.js} +1 -1
  97. package/dist/esm/{chunk-FL3JRQM4.js.map → chunk-ZBUAT73P.js.map} +1 -1
  98. package/dist/esm/controllers/a2a.controller.d.ts +2 -2
  99. package/dist/esm/controllers/api/threads.api.controller.js +1 -1
  100. package/dist/esm/controllers/index.d.ts +3 -3
  101. package/dist/esm/controllers/index.js +3 -3
  102. package/dist/esm/controllers/query.controller.d.ts +2 -2
  103. package/dist/esm/controllers/query.controller.js +2 -2
  104. package/dist/esm/index.d.ts +1 -2
  105. package/dist/esm/index.js +37 -23
  106. package/dist/esm/index.js.map +1 -1
  107. package/dist/esm/middlewares/auth.middleware.js +2 -2
  108. package/dist/esm/modules/a2a/a2a.module.d.ts +2 -9
  109. package/dist/esm/modules/a2a/a2a.module.js +2 -2
  110. package/dist/esm/modules/index.js +10 -10
  111. package/dist/esm/modules/models/model.module.d.ts +17 -0
  112. package/dist/esm/modules/models/model.module.js +1 -1
  113. package/dist/esm/routes/a2a.routes.js +5 -4
  114. package/dist/esm/routes/api/threads.routes.js +3 -3
  115. package/dist/esm/routes/api.routes.js +4 -4
  116. package/dist/esm/routes/index.js +11 -11
  117. package/dist/esm/routes/query.routes.js +6 -6
  118. package/dist/esm/services/a2a.service.d.ts +4 -4
  119. package/dist/esm/services/a2a.service.js +1 -1
  120. package/dist/esm/services/index.js +7 -6
  121. package/dist/esm/services/query-stream.service.d.ts +13 -10
  122. package/dist/esm/services/query-stream.service.js +3 -2
  123. package/dist/esm/services/query.service.d.ts +7 -4
  124. package/dist/esm/services/query.service.js +2 -2
  125. package/dist/esm/types/agent.d.ts +0 -2
  126. package/dist/esm/types/agent.js +1 -1
  127. package/dist/esm/types/memory.d.ts +6 -7
  128. package/dist/esm/types/memory.js +1 -1
  129. package/package.json +9 -4
  130. package/dist/cjs/chunk-2C33PZTV.cjs.map +0 -1
  131. package/dist/cjs/chunk-55IME4BG.cjs.map +0 -1
  132. package/dist/cjs/chunk-6YVTJEMQ.cjs.map +0 -1
  133. package/dist/cjs/chunk-7GWYMGWW.cjs.map +0 -1
  134. package/dist/cjs/chunk-7HUWKIXU.cjs +0 -128
  135. package/dist/cjs/chunk-7HUWKIXU.cjs.map +0 -1
  136. package/dist/cjs/chunk-AIE3SVG7.cjs.map +0 -1
  137. package/dist/cjs/chunk-BBCZU5VZ.cjs +0 -17
  138. package/dist/cjs/chunk-BBCZU5VZ.cjs.map +0 -1
  139. package/dist/cjs/chunk-BOKIE5SM.cjs.map +0 -1
  140. package/dist/cjs/chunk-CU5MOHJU.cjs.map +0 -1
  141. package/dist/cjs/chunk-CYW6ICAW.cjs +0 -106
  142. package/dist/cjs/chunk-CYW6ICAW.cjs.map +0 -1
  143. package/dist/cjs/chunk-EAK7MP6X.cjs +0 -22
  144. package/dist/cjs/chunk-EAK7MP6X.cjs.map +0 -1
  145. package/dist/cjs/chunk-F4CJCUAK.cjs +0 -22
  146. package/dist/cjs/chunk-F4CJCUAK.cjs.map +0 -1
  147. package/dist/cjs/chunk-FCZQBJTF.cjs +0 -36
  148. package/dist/cjs/chunk-FCZQBJTF.cjs.map +0 -1
  149. package/dist/cjs/chunk-G5KEV27W.cjs +0 -45
  150. package/dist/cjs/chunk-G5KEV27W.cjs.map +0 -1
  151. package/dist/cjs/chunk-H35XHCTA.cjs +0 -22
  152. package/dist/cjs/chunk-H35XHCTA.cjs.map +0 -1
  153. package/dist/cjs/chunk-HT3PACCL.cjs.map +0 -1
  154. package/dist/cjs/chunk-JXKLEQD6.cjs.map +0 -1
  155. package/dist/cjs/chunk-KE7SL3H6.cjs +0 -251
  156. package/dist/cjs/chunk-KE7SL3H6.cjs.map +0 -1
  157. package/dist/cjs/chunk-KRLUWTLN.cjs +0 -35
  158. package/dist/cjs/chunk-KRLUWTLN.cjs.map +0 -1
  159. package/dist/cjs/chunk-LRBRG2DS.cjs.map +0 -1
  160. package/dist/cjs/chunk-MIWLK57K.cjs.map +0 -1
  161. package/dist/cjs/chunk-MLUUV7UY.cjs +0 -36
  162. package/dist/cjs/chunk-MLUUV7UY.cjs.map +0 -1
  163. package/dist/cjs/chunk-QWASSVCU.cjs.map +0 -1
  164. package/dist/cjs/chunk-T2AKW5PA.cjs +0 -45
  165. package/dist/cjs/chunk-T2AKW5PA.cjs.map +0 -1
  166. package/dist/cjs/chunk-UTFWH24D.cjs +0 -22
  167. package/dist/cjs/chunk-UTFWH24D.cjs.map +0 -1
  168. package/dist/cjs/chunk-W52YCYRF.cjs +0 -43
  169. package/dist/cjs/chunk-W52YCYRF.cjs.map +0 -1
  170. package/dist/cjs/chunk-YNBMJ5QX.cjs +0 -129
  171. package/dist/cjs/chunk-YNBMJ5QX.cjs.map +0 -1
  172. package/dist/cjs/chunk-ZZK4ACTV.cjs.map +0 -1
  173. package/dist/cjs/chunk-ZZPJ5CPR.cjs.map +0 -1
  174. package/dist/cjs/controllers/api/thread.api.controller.cjs +0 -8
  175. package/dist/cjs/controllers/api/thread.api.controller.cjs.map +0 -1
  176. package/dist/cjs/routes/api/thread.routes.cjs +0 -9
  177. package/dist/cjs/routes/api/thread.routes.cjs.map +0 -1
  178. package/dist/esm/chunk-2L2KJNAM.js.map +0 -1
  179. package/dist/esm/chunk-2RW77MJB.js.map +0 -1
  180. package/dist/esm/chunk-CWFIK2EK.js.map +0 -1
  181. package/dist/esm/chunk-ILKEX4ZV.js.map +0 -1
  182. package/dist/esm/chunk-LJVVSUF7.js.map +0 -1
  183. package/dist/esm/chunk-MNAWSBYP.js.map +0 -1
  184. package/dist/esm/chunk-PHKK5NHK.js.map +0 -1
  185. package/dist/esm/chunk-PN6NMQBM.js.map +0 -1
  186. package/dist/esm/chunk-WLT2B2SZ.js.map +0 -1
  187. /package/dist/esm/{chunk-7D4SKPIS.js.map → chunk-4C2TLCWI.js.map} +0 -0
  188. /package/dist/esm/{chunk-EIHZ4S5V.js.map → chunk-4R4TB2ZJ.js.map} +0 -0
  189. /package/dist/esm/{chunk-SYCFBEUA.js.map → chunk-QHWDMU43.js.map} +0 -0
  190. /package/dist/esm/{chunk-VDXWZ4NK.js.map → chunk-TV2GJAEH.js.map} +0 -0
@@ -1,4 +1,4 @@
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;
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
2
 
3
3
  var _chunkF7I4G6JGcjs = require('./chunk-F7I4G6JG.cjs');
4
4
 
@@ -8,11 +8,11 @@ var _chunk466ED5WNcjs = require('./chunk-466ED5WN.cjs');
8
8
  // src/modules/a2a/a2a.module.ts
9
9
  var _crypto = require('crypto');
10
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); }
11
+ var A2AModule = (_class = class {constructor() { _class.prototype.__init.call(this);_class.prototype.__init2.call(this);_class.prototype.__init3.call(this); }
12
12
  /** Map of A2A server URLs to their corresponding tool instances */
13
13
  __init() {this.a2aPeerServers = /* @__PURE__ */ new Map()}
14
14
  /** Map of session IDs to their A2A session state */
15
- __init2() {this.a2aSessions = /* @__PURE__ */ new Map()}
15
+ __init2() {this.a2aTasks = /* @__PURE__ */ new Map()}
16
16
  __init3() {this.agentId = _crypto.randomUUID.call(void 0, )}
17
17
  /* FIXME */
18
18
  /**
@@ -53,22 +53,6 @@ var A2AModule = (_class = class {constructor() { _class.prototype.__init.call(th
53
53
  }
54
54
  return tools;
55
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
56
  /**
73
57
  * Constructs a message payload for A2A communication.
74
58
  *
@@ -87,22 +71,18 @@ var A2AModule = (_class = class {constructor() { _class.prototype.__init.call(th
87
71
  // FIXME: it could be 'agent'
88
72
  metadata: {
89
73
  agentId: this.agentId,
90
- type: "CHAT" /* CHAT */,
91
- threadId
74
+ type: "CHAT" /* CHAT */
92
75
  },
93
76
  parts: [
94
77
  {
95
78
  kind: "text",
96
79
  text: query
97
80
  }
98
- ]
81
+ ],
82
+ contextId: threadId
99
83
  };
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;
84
+ if (this.a2aTasks.has(threadId)) {
85
+ messagePayload.taskId = this.a2aTasks.get(threadId);
106
86
  }
107
87
  return messagePayload;
108
88
  }
@@ -117,20 +97,20 @@ var A2AModule = (_class = class {constructor() { _class.prototype.__init.call(th
117
97
  * @param threadId - The session identifier for context tracking
118
98
  * @returns Promise resolving to array of text responses from the agent
119
99
  */
120
- async useTool(tool, messagePayload, threadId) {
100
+ async useTool(tool, query, threadId) {
121
101
  const finalText = [];
122
102
  const client = tool.client;
103
+ const messagePayload = this.getMessagePayload(query, threadId);
123
104
  const params = {
124
105
  message: messagePayload
125
106
  };
126
- const a2aSession = this.getA2ASessionWithId(threadId);
127
107
  try {
128
108
  const stream = client.sendMessageStream(params);
129
109
  for await (const event of stream) {
130
110
  if (event.kind === "status-update") {
131
111
  const typedEvent = event;
132
112
  if (typedEvent.final && typedEvent.status.state !== "input-required") {
133
- a2aSession.taskId = void 0;
113
+ this.a2aTasks.delete(threadId);
134
114
  }
135
115
  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
116
  if (texts) {
@@ -138,19 +118,14 @@ var A2AModule = (_class = class {constructor() { _class.prototype.__init.call(th
138
118
  }
139
119
  } else if (event.kind === "message") {
140
120
  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;
121
+ const taskId = this.a2aTasks.get(threadId);
122
+ if (msg.taskId && msg.taskId !== taskId) {
123
+ this.a2aTasks.set(threadId, msg.taskId);
146
124
  }
147
125
  } else if (event.kind === "task") {
148
126
  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;
127
+ if (task.id !== this.a2aTasks.get(threadId)) {
128
+ this.a2aTasks.set(threadId, task.id);
154
129
  }
155
130
  } else {
156
131
  _chunk466ED5WNcjs.loggers.a2a.warn("Received unknown event structure from stream:", {
@@ -173,4 +148,4 @@ ${finalText.join("\n")}`;
173
148
 
174
149
 
175
150
  exports.A2AModule = A2AModule;
176
- //# sourceMappingURL=chunk-6YVTJEMQ.cjs.map
151
+ //# sourceMappingURL=chunk-BBFYVWE2.cjs.map
@@ -0,0 +1 @@
1
+ {"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-BBFYVWE2.cjs","../../src/modules/a2a/a2a.module.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACA;ACPA,gCAA2B;AAS3B,4CAA0B;AAYnB,IAAM,UAAA,YAAN,MAAgB;AAAA;AAAA,iBAEd,eAAA,kBAA8C,IAAI,GAAA,CAAI,EAAA;AAAA;AAAA,kBAEtD,SAAA,kBAAgC,IAAI,GAAA,CAAI,EAAA;AAAA,kBACxC,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;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;AAAA,MACD,CAAA;AAAA,MACA,KAAA,EAAO;AAAA,QACN;AAAA,UACC,IAAA,EAAM,MAAA;AAAA,UACN,IAAA,EAAM;AAAA,QACP;AAAA,MACD,CAAA;AAAA,MACA,SAAA,EAAW;AAAA,IACZ,CAAA;AAEA,IAAA,GAAA,CAAI,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,QAAQ,CAAA,EAAG;AAChC,MAAA,cAAA,CAAe,OAAA,EAAS,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,QAAQ,CAAA;AAAA,IACnD;AAEA,IAAA,OAAO,cAAA;AAAA,EACR;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAaA,MAAa,OAAA,CACZ,IAAA,EACA,KAAA,EACA,QAAA,EACkB;AAClB,IAAA,MAAM,UAAA,EAAsB,CAAC,CAAA;AAC7B,IAAA,MAAM,OAAA,EAAS,IAAA,CAAK,MAAA;AACpB,IAAA,MAAM,eAAA,EAAiB,IAAA,CAAK,iBAAA,CAAkB,KAAA,EAAO,QAAQ,CAAA;AAC7D,IAAA,MAAM,OAAA,EAA4B;AAAA,MACjC,OAAA,EAAS;AAAA,IACV,CAAA;AAEA,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,IAAA,CAAK,QAAA,CAAS,MAAA,CAAO,QAAQ,CAAA;AAAA,UAC9B;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,MAAM,OAAA,EAAS,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,QAAQ,CAAA;AACzC,UAAA,GAAA,CAAI,GAAA,CAAI,OAAA,GAAU,GAAA,CAAI,OAAA,IAAW,MAAA,EAAQ;AACxC,YAAA,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,QAAA,EAAU,GAAA,CAAI,MAAM,CAAA;AAAA,UACvC;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,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,QAAQ,CAAA,EAAG;AAC5C,YAAA,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,QAAA,EAAU,IAAA,CAAK,EAAE,CAAA;AAAA,UACpC;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;AD7BqE;AACA;AACA;AACA","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-BBFYVWE2.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 * 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 a2aTasks: Map<string, string> = 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 * 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},\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\tcontextId: threadId,\n\t\t};\n\n\t\tif (this.a2aTasks.has(threadId)) {\n\t\t\tmessagePayload.taskId = this.a2aTasks.get(threadId);\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\tquery: string,\n\t\tthreadId: string,\n\t): Promise<string> {\n\t\tconst finalText: string[] = [];\n\t\tconst client = tool.client;\n\t\tconst messagePayload = this.getMessagePayload(query, threadId);\n\t\tconst params: MessageSendParams = {\n\t\t\tmessage: messagePayload,\n\t\t};\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\tthis.a2aTasks.delete(threadId);\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\tconst taskId = this.a2aTasks.get(threadId);\n\t\t\t\t\tif (msg.taskId && msg.taskId !== taskId) {\n\t\t\t\t\t\tthis.a2aTasks.set(threadId, msg.taskId);\n\t\t\t\t\t}\n\t\t\t\t} else if (event.kind === \"task\") {\n\t\t\t\t\t// establishing the Task ID\n\t\t\t\t\tconst task = event as Task;\n\t\t\t\t\tif (task.id !== this.a2aTasks.get(threadId)) {\n\t\t\t\t\t\tthis.a2aTasks.set(threadId, task.id);\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"]}
@@ -1,18 +1,18 @@
1
1
  "use strict";Object.defineProperty(exports, "__esModule", {value: true});
2
2
 
3
- var _chunkZZK4ACTVcjs = require('./chunk-ZZK4ACTV.cjs');
3
+ var _chunkPAWGHHSYcjs = require('./chunk-PAWGHHSY.cjs');
4
4
 
5
5
 
6
- var _chunk3J3ORQOGcjs = require('./chunk-3J3ORQOG.cjs');
6
+ var _chunkVLMHDHXTcjs = require('./chunk-VLMHDHXT.cjs');
7
7
 
8
8
 
9
- var _chunk7GWYMGWWcjs = require('./chunk-7GWYMGWW.cjs');
9
+ var _chunkA2ANDQNYcjs = require('./chunk-A2ANDQNY.cjs');
10
10
 
11
11
  // src/routes/query.routes.ts
12
12
  var _express = require('express');
13
13
  var createQueryRouter = (agent, allowStream = false) => {
14
14
  const router = _express.Router.call(void 0, );
15
- const queryService = new (0, _chunk7GWYMGWWcjs.QueryService)(
15
+ const queryService = new (0, _chunkPAWGHHSYcjs.QueryService)(
16
16
  agent.modelModule,
17
17
  agent.a2aModule,
18
18
  agent.mcpModule,
@@ -21,7 +21,7 @@ var createQueryRouter = (agent, allowStream = false) => {
21
21
  );
22
22
  let queryStreamService;
23
23
  if (allowStream) {
24
- queryStreamService = new (0, _chunkZZK4ACTVcjs.QueryStreamService)(
24
+ queryStreamService = new (0, _chunkA2ANDQNYcjs.QueryStreamService)(
25
25
  agent.modelModule,
26
26
  agent.a2aModule,
27
27
  agent.mcpModule,
@@ -29,7 +29,7 @@ var createQueryRouter = (agent, allowStream = false) => {
29
29
  agent.manifest.prompts
30
30
  );
31
31
  }
32
- const queryController = new (0, _chunk3J3ORQOGcjs.QueryController)(queryService, queryStreamService);
32
+ const queryController = new (0, _chunkVLMHDHXTcjs.QueryController)(queryService, queryStreamService);
33
33
  router.post("/", queryController.handleQueryRequest);
34
34
  if (allowStream) {
35
35
  router.post("/stream", queryController.handleQueryStreamRequest);
@@ -40,4 +40,4 @@ var createQueryRouter = (agent, allowStream = false) => {
40
40
 
41
41
 
42
42
  exports.createQueryRouter = createQueryRouter;
43
- //# sourceMappingURL=chunk-4DXS33VP.cjs.map
43
+ //# sourceMappingURL=chunk-CHC3HECE.cjs.map
@@ -1 +1 @@
1
- {"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-EOPW5OPE.cjs","../../src/routes/query.routes.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACA;ACVA,kCAAuB;AAMhB,IAAM,kBAAA,EAAoB,CAChC,KAAA,EACA,YAAA,EAAc,KAAA,EAAA,GACF;AACZ,EAAA,MAAM,OAAA,EAAS,6BAAA,CAAO;AAEtB,EAAA,MAAM,aAAA,EAAe,IAAI,mCAAA;AAAA,IACxB,KAAA,CAAM,WAAA;AAAA,IACN,KAAA,CAAM,SAAA;AAAA,IACN,KAAA,CAAM,SAAA;AAAA,IACN,KAAA,CAAM,YAAA;AAAA,IACN,KAAA,CAAM,QAAA,CAAS;AAAA,EAChB,CAAA;AAEA,EAAA,IAAI,kBAAA;AAEJ,EAAA,GAAA,CAAI,WAAA,EAAa;AAChB,IAAA,mBAAA,EAAqB,IAAI,yCAAA;AAAA,MACxB,KAAA,CAAM,WAAA;AAAA,MACN,KAAA,CAAM,SAAA;AAAA,MACN,KAAA,CAAM,SAAA;AAAA,MACN,KAAA,CAAM,YAAA;AAAA,MACN,KAAA,CAAM,QAAA,CAAS;AAAA,IAChB,CAAA;AAAA,EACD;AAEA,EAAA,MAAM,gBAAA,EAAkB,IAAI,sCAAA,CAAgB,YAAA,EAAc,kBAAkB,CAAA;AAC5E,EAAA,MAAA,CAAO,IAAA,CAAK,GAAA,EAAK,eAAA,CAAgB,kBAAkB,CAAA;AAEnD,EAAA,GAAA,CAAI,WAAA,EAAa;AAChB,IAAA,MAAA,CAAO,IAAA,CAAK,SAAA,EAAW,eAAA,CAAgB,wBAAwB,CAAA;AAAA,EAChE;AAEA,EAAA,OAAO,MAAA;AACR,CAAA;ADFA;AACA;AACE;AACF,8CAAC","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-EOPW5OPE.cjs","sourcesContent":[null,"import { Router } from \"express\";\nimport { QueryController } from \"@/controllers/query.controller.js\";\nimport type { AINAgent } from \"@/index.js\";\nimport { QueryService } from \"@/services/query.service.js\";\nimport { QueryStreamService } from \"@/services/query-stream.service.js\";\n\nexport const createQueryRouter = (\n\tagent: AINAgent,\n\tallowStream = false,\n): Router => {\n\tconst router = Router();\n\n\tconst queryService = new QueryService(\n\t\tagent.modelModule,\n\t\tagent.a2aModule,\n\t\tagent.mcpModule,\n\t\tagent.memoryModule,\n\t\tagent.manifest.prompts,\n\t);\n\n\tlet queryStreamService: QueryStreamService | undefined;\n\n\tif (allowStream) {\n\t\tqueryStreamService = new QueryStreamService(\n\t\t\tagent.modelModule,\n\t\t\tagent.a2aModule,\n\t\t\tagent.mcpModule,\n\t\t\tagent.memoryModule,\n\t\t\tagent.manifest.prompts,\n\t\t);\n\t}\n\n\tconst queryController = new QueryController(queryService, queryStreamService);\n\trouter.post(\"/\", queryController.handleQueryRequest);\n\n\tif (allowStream) {\n\t\trouter.post(\"/stream\", queryController.handleQueryStreamRequest);\n\t}\n\n\treturn router;\n};\n"]}
1
+ {"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-CHC3HECE.cjs","../../src/routes/query.routes.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACA;ACVA,kCAAuB;AAMhB,IAAM,kBAAA,EAAoB,CAChC,KAAA,EACA,YAAA,EAAc,KAAA,EAAA,GACF;AACZ,EAAA,MAAM,OAAA,EAAS,6BAAA,CAAO;AAEtB,EAAA,MAAM,aAAA,EAAe,IAAI,mCAAA;AAAA,IACxB,KAAA,CAAM,WAAA;AAAA,IACN,KAAA,CAAM,SAAA;AAAA,IACN,KAAA,CAAM,SAAA;AAAA,IACN,KAAA,CAAM,YAAA;AAAA,IACN,KAAA,CAAM,QAAA,CAAS;AAAA,EAChB,CAAA;AAEA,EAAA,IAAI,kBAAA;AAEJ,EAAA,GAAA,CAAI,WAAA,EAAa;AAChB,IAAA,mBAAA,EAAqB,IAAI,yCAAA;AAAA,MACxB,KAAA,CAAM,WAAA;AAAA,MACN,KAAA,CAAM,SAAA;AAAA,MACN,KAAA,CAAM,SAAA;AAAA,MACN,KAAA,CAAM,YAAA;AAAA,MACN,KAAA,CAAM,QAAA,CAAS;AAAA,IAChB,CAAA;AAAA,EACD;AAEA,EAAA,MAAM,gBAAA,EAAkB,IAAI,sCAAA,CAAgB,YAAA,EAAc,kBAAkB,CAAA;AAC5E,EAAA,MAAA,CAAO,IAAA,CAAK,GAAA,EAAK,eAAA,CAAgB,kBAAkB,CAAA;AAEnD,EAAA,GAAA,CAAI,WAAA,EAAa;AAChB,IAAA,MAAA,CAAO,IAAA,CAAK,SAAA,EAAW,eAAA,CAAgB,wBAAwB,CAAA;AAAA,EAChE;AAEA,EAAA,OAAO,MAAA;AACR,CAAA;ADFA;AACA;AACE;AACF,8CAAC","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-CHC3HECE.cjs","sourcesContent":[null,"import { Router } from \"express\";\nimport { QueryController } from \"@/controllers/query.controller.js\";\nimport type { AINAgent } from \"@/index.js\";\nimport { QueryService } from \"@/services/query.service.js\";\nimport { QueryStreamService } from \"@/services/query-stream.service.js\";\n\nexport const createQueryRouter = (\n\tagent: AINAgent,\n\tallowStream = false,\n): Router => {\n\tconst router = Router();\n\n\tconst queryService = new QueryService(\n\t\tagent.modelModule,\n\t\tagent.a2aModule,\n\t\tagent.mcpModule,\n\t\tagent.memoryModule,\n\t\tagent.manifest.prompts,\n\t);\n\n\tlet queryStreamService: QueryStreamService | undefined;\n\n\tif (allowStream) {\n\t\tqueryStreamService = new QueryStreamService(\n\t\t\tagent.modelModule,\n\t\t\tagent.a2aModule,\n\t\t\tagent.mcpModule,\n\t\t\tagent.memoryModule,\n\t\t\tagent.manifest.prompts,\n\t\t);\n\t}\n\n\tconst queryController = new QueryController(queryService, queryStreamService);\n\trouter.post(\"/\", queryController.handleQueryRequest);\n\n\tif (allowStream) {\n\t\trouter.post(\"/stream\", queryController.handleQueryStreamRequest);\n\t}\n\n\treturn router;\n};\n"]}
@@ -1,6 +1,6 @@
1
1
  "use strict";Object.defineProperty(exports, "__esModule", {value: true});
2
2
 
3
- var _chunkDX2JH3FJcjs = require('./chunk-DX2JH3FJ.cjs');
3
+ var _chunkQHQLU7BIcjs = require('./chunk-QHQLU7BI.cjs');
4
4
 
5
5
  // src/middlewares/auth.middleware.ts
6
6
  var _httpstatuscodes = require('http-status-codes');
@@ -17,7 +17,7 @@ var AuthMiddleware = class {
17
17
  res.locals.userId = authRes.userId;
18
18
  next();
19
19
  } else {
20
- const error = new (0, _chunkDX2JH3FJcjs.AinHttpError)(
20
+ const error = new (0, _chunkQHQLU7BIcjs.AinHttpError)(
21
21
  _httpstatuscodes.StatusCodes.UNAUTHORIZED,
22
22
  "Unauthorized"
23
23
  );
@@ -25,7 +25,7 @@ var AuthMiddleware = class {
25
25
  }
26
26
  } catch (e) {
27
27
  if (!e.status) {
28
- const error = new (0, _chunkDX2JH3FJcjs.AinHttpError)(
28
+ const error = new (0, _chunkQHQLU7BIcjs.AinHttpError)(
29
29
  _httpstatuscodes.StatusCodes.INTERNAL_SERVER_ERROR,
30
30
  `Authentication error: ${JSON.stringify(e)}`
31
31
  );
@@ -40,4 +40,4 @@ var AuthMiddleware = class {
40
40
 
41
41
 
42
42
  exports.AuthMiddleware = AuthMiddleware;
43
- //# sourceMappingURL=chunk-LRBRG2DS.cjs.map
43
+ //# sourceMappingURL=chunk-E35E4LQY.cjs.map
@@ -0,0 +1 @@
1
+ {"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-E35E4LQY.cjs","../../src/middlewares/auth.middleware.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACA;ACHA,oDAA4B;AAKrB,IAAM,eAAA,EAAN,MAAqB;AAAA,EACnB;AAAA,EACR,WAAA,CAAY,IAAA,EAAgB;AAC3B,IAAA,IAAA,CAAK,KAAA,EAAO,IAAA;AAAA,EACb;AAAA,EAEO,UAAA,CAAA,EAA6B;AACnC,IAAA,OAAO,MAAA,CAAO,GAAA,EAAc,GAAA,EAAe,IAAA,EAAA,GAAuB;AACjE,MAAA,IAAI;AACH,QAAA,MAAM,QAAA,EAAwB,MAAM,IAAA,CAAK,IAAA,CAAK,YAAA,CAAa,GAAA,EAAK,GAAG,CAAA;AACnE,QAAA,GAAA,CAAI,OAAA,CAAQ,eAAA,EAAiB;AAC5B,UAAA,GAAA,CAAI,MAAA,CAAO,OAAA,EAAS,OAAA,CAAQ,MAAA;AAC5B,UAAA,IAAA,CAAK,CAAA;AAAA,QACN,EAAA,KAAO;AACN,UAAA,MAAM,MAAA,EAAsB,IAAI,mCAAA;AAAA,YAC/B,4BAAA,CAAY,YAAA;AAAA,YACZ;AAAA,UACD,CAAA;AACA,UAAA,MAAM,KAAA;AAAA,QACP;AAAA,MACD,EAAA,MAAA,CAAS,CAAA,EAAQ;AAChB,QAAA,GAAA,CAAI,CAAC,CAAA,CAAE,MAAA,EAAQ;AACd,UAAA,MAAM,MAAA,EAAsB,IAAI,mCAAA;AAAA,YAC/B,4BAAA,CAAY,qBAAA;AAAA,YACZ,CAAA,sBAAA,EAAyB,IAAA,CAAK,SAAA,CAAU,CAAC,CAAC,CAAA;AAAA,UAAA;AAE3C,UAAA;AAAM,QAAA;AAEP,QAAA;AAAM,MAAA;AACP,IAAA;AACD,EAAA;AAEF;ADAA;AACA;AACA;AACA","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-E35E4LQY.cjs","sourcesContent":[null,"import type { NextFunction, Request, RequestHandler, Response } from \"express\";\nimport { StatusCodes } from \"http-status-codes\";\nimport type { BaseAuth } from \"@/modules/auth/base.auth\";\nimport { AinHttpError } from \"@/types/agent\";\nimport type { AuthResponse } from \"@/types/auth\";\n\nexport class AuthMiddleware {\n\tprivate auth: BaseAuth;\n\tconstructor(auth: BaseAuth) {\n\t\tthis.auth = auth;\n\t}\n\n\tpublic middleware(): RequestHandler {\n\t\treturn async (req: Request, res: Response, next: NextFunction) => {\n\t\t\ttry {\n\t\t\t\tconst authRes: AuthResponse = await this.auth.authenticate(req, res);\n\t\t\t\tif (authRes.isAuthenticated) {\n\t\t\t\t\tres.locals.userId = authRes.userId;\n\t\t\t\t\tnext();\n\t\t\t\t} else {\n\t\t\t\t\tconst error: AinHttpError = new AinHttpError(\n\t\t\t\t\t\tStatusCodes.UNAUTHORIZED,\n\t\t\t\t\t\t\"Unauthorized\",\n\t\t\t\t\t);\n\t\t\t\t\tthrow error;\n\t\t\t\t}\n\t\t\t} catch (e: any) {\n\t\t\t\tif (!e.status) {\n\t\t\t\t\tconst error: AinHttpError = new AinHttpError(\n\t\t\t\t\t\tStatusCodes.INTERNAL_SERVER_ERROR,\n\t\t\t\t\t\t`Authentication error: ${JSON.stringify(e)}`,\n\t\t\t\t\t);\n\t\t\t\t\tthrow error;\n\t\t\t\t}\n\t\t\t\tthrow e;\n\t\t\t}\n\t\t};\n\t}\n}\n"]}
@@ -3,7 +3,7 @@
3
3
  var _chunkX46GJS7Xcjs = require('./chunk-X46GJS7X.cjs');
4
4
 
5
5
 
6
- var _chunkGB3ON7UVcjs = require('./chunk-GB3ON7UV.cjs');
6
+ var _chunk7PZP7YSAcjs = require('./chunk-7PZP7YSA.cjs');
7
7
 
8
8
  // src/routes/api.routes.ts
9
9
  var _express = require('express');
@@ -11,7 +11,7 @@ var createApiRouter = (agent) => {
11
11
  const router = _express.Router.call(void 0, );
12
12
  router.use("/model", _chunkX46GJS7Xcjs.createModelApiRouter.call(void 0, agent.modelModule));
13
13
  if (agent.memoryModule) {
14
- router.use("/threads", _chunkGB3ON7UVcjs.createThreadApiRouter.call(void 0, agent.memoryModule));
14
+ router.use("/threads", _chunk7PZP7YSAcjs.createThreadApiRouter.call(void 0, agent.memoryModule));
15
15
  }
16
16
  return router;
17
17
  };
@@ -19,4 +19,4 @@ var createApiRouter = (agent) => {
19
19
 
20
20
 
21
21
  exports.createApiRouter = createApiRouter;
22
- //# sourceMappingURL=chunk-45G7LDK7.cjs.map
22
+ //# sourceMappingURL=chunk-FRWC3FQ4.cjs.map
@@ -1 +1 @@
1
- {"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-45G7LDK7.cjs","../../src/routes/api.routes.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACA;ACPA,kCAAuB;AAMhB,IAAM,gBAAA,EAAkB,CAAC,KAAA,EAAA,GAA4B;AAC3D,EAAA,MAAM,OAAA,EAAS,6BAAA,CAAO;AAEtB,EAAA,MAAA,CAAO,GAAA,CAAI,QAAA,EAAU,oDAAA,KAAqB,CAAM,WAAW,CAAC,CAAA;AAC5D,EAAA,GAAA,CAAI,KAAA,CAAM,YAAA,EAAc;AACvB,IAAA,MAAA,CAAO,GAAA,CAAI,UAAA,EAAY,qDAAA,KAAsB,CAAM,YAAY,CAAC,CAAA;AAAA,EACjE;AAEA,EAAA,OAAO,MAAA;AACR,CAAA;ADEA;AACA;AACE;AACF,0CAAC","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-45G7LDK7.cjs","sourcesContent":[null,"import { Router } from \"express\";\nimport type { AINAgent } from \"@/index.js\";\n\nimport { createModelApiRouter } from \"./api/model.routes.js\";\nimport { createThreadApiRouter } from \"./api/threads.routes.js\";\n\nexport const createApiRouter = (agent: AINAgent): Router => {\n\tconst router = Router();\n\n\trouter.use(\"/model\", createModelApiRouter(agent.modelModule));\n\tif (agent.memoryModule) {\n\t\trouter.use(\"/threads\", createThreadApiRouter(agent.memoryModule));\n\t}\n\n\treturn router;\n};\n"]}
1
+ {"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-FRWC3FQ4.cjs","../../src/routes/api.routes.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACA;ACPA,kCAAuB;AAMhB,IAAM,gBAAA,EAAkB,CAAC,KAAA,EAAA,GAA4B;AAC3D,EAAA,MAAM,OAAA,EAAS,6BAAA,CAAO;AAEtB,EAAA,MAAA,CAAO,GAAA,CAAI,QAAA,EAAU,oDAAA,KAAqB,CAAM,WAAW,CAAC,CAAA;AAC5D,EAAA,GAAA,CAAI,KAAA,CAAM,YAAA,EAAc;AACvB,IAAA,MAAA,CAAO,GAAA,CAAI,UAAA,EAAY,qDAAA,KAAsB,CAAM,YAAY,CAAC,CAAA;AAAA,EACjE;AAEA,EAAA,OAAO,MAAA;AACR,CAAA;ADEA;AACA;AACE;AACF,0CAAC","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-FRWC3FQ4.cjs","sourcesContent":[null,"import { Router } from \"express\";\nimport type { AINAgent } from \"@/index.js\";\n\nimport { createModelApiRouter } from \"./api/model.routes.js\";\nimport { createThreadApiRouter } from \"./api/threads.routes.js\";\n\nexport const createApiRouter = (agent: AINAgent): Router => {\n\tconst router = Router();\n\n\trouter.use(\"/model\", createModelApiRouter(agent.modelModule));\n\tif (agent.memoryModule) {\n\t\trouter.use(\"/threads\", createThreadApiRouter(agent.memoryModule));\n\t}\n\n\treturn router;\n};\n"]}
@@ -1,12 +1,12 @@
1
1
  "use strict";Object.defineProperty(exports, "__esModule", {value: true});
2
2
 
3
- var _chunkKE7SL3H6cjs = require('./chunk-KE7SL3H6.cjs');
3
+ var _chunkOKRBMMK2cjs = require('./chunk-OKRBMMK2.cjs');
4
4
 
5
5
 
6
- var _chunkICFTUZP5cjs = require('./chunk-ICFTUZP5.cjs');
6
+ var _chunkA2ANDQNYcjs = require('./chunk-A2ANDQNY.cjs');
7
7
 
8
8
 
9
- var _chunk55IME4BGcjs = require('./chunk-55IME4BG.cjs');
9
+ var _chunkICFTUZP5cjs = require('./chunk-ICFTUZP5.cjs');
10
10
 
11
11
  // src/routes/a2a.routes.ts
12
12
  var _server = require('@a2a-js/sdk/server');
@@ -14,14 +14,14 @@ var _express = require('express');
14
14
  var createA2ARouter = (agent) => {
15
15
  const router = _express.Router.call(void 0, );
16
16
  const taskStore = new (0, _server.InMemoryTaskStore)();
17
- const queryService = new (0, _chunkKE7SL3H6cjs.QueryService)(
17
+ const queryStreamService = new (0, _chunkA2ANDQNYcjs.QueryStreamService)(
18
18
  agent.modelModule,
19
19
  agent.a2aModule,
20
20
  agent.mcpModule,
21
21
  agent.memoryModule,
22
22
  agent.manifest.prompts
23
23
  );
24
- const a2aService = new (0, _chunk55IME4BGcjs.A2AService)(queryService);
24
+ const a2aService = new (0, _chunkOKRBMMK2cjs.A2AService)(queryStreamService);
25
25
  const a2aController = new (0, _chunkICFTUZP5cjs.A2AController)(
26
26
  a2aService,
27
27
  taskStore,
@@ -34,4 +34,4 @@ var createA2ARouter = (agent) => {
34
34
 
35
35
 
36
36
  exports.createA2ARouter = createA2ARouter;
37
- //# sourceMappingURL=chunk-2C33PZTV.cjs.map
37
+ //# sourceMappingURL=chunk-HCN6JEJA.cjs.map
@@ -0,0 +1 @@
1
+ {"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-HCN6JEJA.cjs","../../src/routes/a2a.routes.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACA;ACVA,4CAAkC;AAClC,kCAAuB;AAchB,IAAM,gBAAA,EAAkB,CAAC,KAAA,EAAA,GAA4B;AAC3D,EAAA,MAAM,OAAA,EAAS,6BAAA,CAAO;AAGtB,EAAA,MAAM,UAAA,EAAY,IAAI,8BAAA,CAAkB,CAAA;AACxC,EAAA,MAAM,mBAAA,EAAqB,IAAI,yCAAA;AAAA,IAC9B,KAAA,CAAM,WAAA;AAAA,IACN,KAAA,CAAM,SAAA;AAAA,IACN,KAAA,CAAM,SAAA;AAAA,IACN,KAAA,CAAM,YAAA;AAAA,IACN,KAAA,CAAM,QAAA,CAAS;AAAA,EAChB,CAAA;AACA,EAAA,MAAM,WAAA,EAAa,IAAI,iCAAA,CAAW,kBAAkB,CAAA;AACpD,EAAA,MAAM,cAAA,EAAgB,IAAI,oCAAA;AAAA,IACzB,UAAA;AAAA,IACA,SAAA;AAAA,IACA,KAAA,CAAM;AAAA,EACP,CAAA;AAGA,EAAA,MAAA,CAAO,IAAA,CAAK,GAAA,EAAK,aAAA,CAAc,gBAAgB,CAAA;AAE/C,EAAA,OAAO,MAAA;AACR,CAAA;ADNA;AACA;AACE;AACF,0CAAC","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-HCN6JEJA.cjs","sourcesContent":[null,"import { InMemoryTaskStore } from \"@a2a-js/sdk/server\";\nimport { Router } from \"express\";\nimport type { AINAgent } from \"@/index.js\";\nimport { QueryStreamService } from \"@/services/query-stream.service.js\";\nimport { A2AController } from \"../controllers/a2a.controller.js\";\nimport { A2AService } from \"../services/a2a.service.js\";\n\n/**\n * Creates and configures the A2A router.\n * This function is a \"composition root\" for the A2A feature,\n * creating and injecting all necessary dependencies.\n * @param intentAnalyzer The core intent analyzer.\n * @param agentCard The agent's card.\n * @returns An Express Router instance.\n */\nexport const createA2ARouter = (agent: AINAgent): Router => {\n\tconst router = Router();\n\n\t// 1. Create dependencies for the A2A feature\n\tconst taskStore = new InMemoryTaskStore();\n\tconst queryStreamService = new QueryStreamService(\n\t\tagent.modelModule,\n\t\tagent.a2aModule,\n\t\tagent.mcpModule,\n\t\tagent.memoryModule,\n\t\tagent.manifest.prompts,\n\t);\n\tconst a2aService = new A2AService(queryStreamService);\n\tconst a2aController = new A2AController(\n\t\ta2aService,\n\t\ttaskStore,\n\t\tagent.generateAgentCard,\n\t);\n\n\t// 2. Define the route\n\trouter.post(\"/\", a2aController.handleA2ARequest);\n\n\treturn router;\n};\n"]}
@@ -0,0 +1,18 @@
1
+ "use strict";Object.defineProperty(exports, "__esModule", {value: true});// src/types/memory.ts
2
+ var MessageRole = /* @__PURE__ */ ((MessageRole2) => {
3
+ MessageRole2["USER"] = "USER";
4
+ MessageRole2["SYSTEM"] = "SYSTEM";
5
+ MessageRole2["MODEL"] = "MODEL";
6
+ return MessageRole2;
7
+ })(MessageRole || {});
8
+ var ThreadType = /* @__PURE__ */ ((ThreadType2) => {
9
+ ThreadType2["WORKFLOW"] = "WORKFLOW";
10
+ ThreadType2["CHAT"] = "CHAT";
11
+ return ThreadType2;
12
+ })(ThreadType || {});
13
+
14
+
15
+
16
+
17
+ exports.MessageRole = MessageRole; exports.ThreadType = ThreadType;
18
+ //# sourceMappingURL=chunk-J7GCQY3S.cjs.map
@@ -0,0 +1 @@
1
+ {"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-J7GCQY3S.cjs","../../src/types/memory.ts"],"names":["MessageRole","ThreadType"],"mappings":"AAAA;ACGO,IAAK,YAAA,kBAAL,CAAA,CAAKA,YAAAA,EAAAA,GAAL;AAEN,EAAAA,YAAAA,CAAA,MAAA,EAAA,EAAO,MAAA;AAEP,EAAAA,YAAAA,CAAA,QAAA,EAAA,EAAS,QAAA;AAET,EAAAA,YAAAA,CAAA,OAAA,EAAA,EAAQ,OAAA;AANG,EAAA,OAAAA,YAAAA;AAAA,CAAA,CAAA,CAAA,YAAA,GAAA,CAAA,CAAA,CAAA;AAiDL,IAAK,WAAA,kBAAL,CAAA,CAAKC,WAAAA,EAAAA,GAAL;AACN,EAAAA,WAAAA,CAAA,UAAA,EAAA,EAAW,UAAA;AACX,EAAAA,WAAAA,CAAA,MAAA,EAAA,EAAO,MAAA;AAFI,EAAA,OAAAA,WAAAA;AAAA,CAAA,CAAA,CAAA,WAAA,GAAA,CAAA,CAAA,CAAA;ADxCZ;AACA;AACE;AACA;AACF,mEAAC","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-J7GCQY3S.cjs","sourcesContent":[null,"/**\n * Roles for participants in a message.\n */\nexport enum MessageRole {\n\t/** User/human participant */\n\tUSER = \"USER\",\n\t/** System-generated messages or instructions */\n\tSYSTEM = \"SYSTEM\",\n\t/** AI model responses */\n\tMODEL = \"MODEL\",\n}\n\n/**\n * Content structure for message content.\n *\n * Supports multi-part content with different types (text, images, etc.).\n */\nexport type MessageContentObject = {\n\t/** Content type (e.g., \"text\", \"image\", \"tool_use\") */\n\ttype: string;\n\t/** Array of content parts, structure depends on content type */\n\tparts: any[];\n};\n\n/**\n * Represents a single message in a thread.\n *\n * @example\n * ```typescript\n * const message: MessageObject = {\n * role: MessageRole.USER,\n * content: {\n * type: \"text\",\n * parts: [\"Hello, how can you help me?\"]\n * },\n * timestamp: Date.now(),\n * metadata: { source: \"web-ui\" }\n * };\n * ```\n */\nexport type MessageObject = {\n\tmessageId: string;\n\t/** Role of the message sender */\n\trole: MessageRole;\n\t/** Message content with type and parts */\n\tcontent: MessageContentObject;\n\t/** Unix timestamp when the message was created */\n\ttimestamp: number;\n\t/** Optional metadata for additional context */\n\tmetadata?: { [key: string]: unknown };\n};\n\nexport enum ThreadType {\n\tWORKFLOW = \"WORKFLOW\",\n\tCHAT = \"CHAT\",\n}\n\nexport type ThreadMetadata = {\n\ttype: ThreadType;\n\ttitle: string;\n\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 * { messageId: <UUID_1>, role: MessageRole.USER, content: {...}, timestamp: 1234567890 },\n * { messageId: <UUID_2> ,role: MessageRole.MODEL, content: {...}, timestamp: 1234567891 }\n * ]\n * };\n * ```\n */\nexport type ThreadObject = {\n\ttype: ThreadType;\n\ttitle: string;\n\tmessages: Array<MessageObject>;\n};\n\nexport interface Intent {\n\tname: string;\n\tdescription: string;\n\tprompt?: string;\n\tllm?: string;\n}\n"]}
@@ -72,18 +72,20 @@ Please select and answer the most appropriate intent name from the available int
72
72
  return intent;
73
73
  }
74
74
  /**
75
- * Fulfills the detected intent by generating a response.
75
+ * Fulfills the detected intent by generating a streaming response.
76
76
  *
77
77
  * Manages the complete inference loop including:
78
78
  * - Loading prompts and conversation history
79
79
  * - Collecting available tools from modules
80
80
  * - Executing model inference with tool support
81
81
  * - Processing tool calls iteratively until completion
82
+ * - Streaming results as Server-Sent Events
82
83
  *
83
84
  * @param query - The user's input query
84
85
  * @param threadId - Thread identifier for context
85
86
  * @param thread - Previous conversation history
86
- * @returns Object containing process steps and final response
87
+ * @param intent - Optional detected intent with custom prompt
88
+ * @returns AsyncGenerator yielding StreamEvent objects
87
89
  */
88
90
  async *intentFulfilling(query, threadId, thread, intent) {
89
91
  const systemPrompt = `
@@ -138,10 +140,6 @@ ${_optionalChain([intent, 'optionalAccess', _8 => _8.prompt]) || ""}
138
140
  assembledToolCalls
139
141
  });
140
142
  if (assembledToolCalls.length > 0) {
141
- const messagePayload = _optionalChain([this, 'access', _14 => _14.a2aModule, 'optionalAccess', _15 => _15.getMessagePayload, 'call', _16 => _16(
142
- query,
143
- threadId
144
- )]);
145
143
  for (const toolCall of assembledToolCalls) {
146
144
  const toolCallId = _crypto.randomUUID.call(void 0, );
147
145
  const toolName = toolCall.function.name;
@@ -176,8 +174,7 @@ ${_optionalChain([intent, 'optionalAccess', _8 => _8.prompt]) || ""}
176
174
  _chunk466ED5WNcjs.loggers.intent.debug("A2A tool call", { toolName });
177
175
  toolResult = await this.a2aModule.useTool(
178
176
  selectedTool,
179
- // biome-ignore lint/style/noNonNullAssertion: <a2aModule is guaranteed to be defined>
180
- messagePayload,
177
+ query,
181
178
  threadId
182
179
  );
183
180
  } else {
@@ -232,32 +229,33 @@ ${_optionalChain([intent, 'optionalAccess', _8 => _8.prompt]) || ""}
232
229
  }
233
230
  }
234
231
  /**
235
- * Main entry point for processing user queries.
232
+ * Main entry point for processing streaming user queries.
236
233
  *
237
234
  * Handles the complete query lifecycle:
238
- * 1. Loads thread from memory
235
+ * 1. Loads or creates thread from memory
239
236
  * 2. Detects intent from the query
240
- * 3. Fulfills the intent with AI response
241
- * 4. Updates conversation history
237
+ * 3. Fulfills the intent with streaming AI response
238
+ * 4. Updates conversation history in real-time
242
239
  *
243
- * @param type - The type of thread (e.g., chat, workflow)
244
- * @param userId - The user's unique identifier
245
- * @param threadId - Unique thread identifier
240
+ * @param threadMetadata - Metadata containing type, userId, and optional threadId
241
+ * @param threadMetadata.type - The type of thread (e.g., chat, workflow)
242
+ * @param threadMetadata.userId - The user's unique identifier
243
+ * @param threadMetadata.threadId - Optional thread identifier
246
244
  * @param query - The user's input query
247
- * @returns Object containing the AI-generated response
245
+ * @returns AsyncGenerator yielding StreamEvent objects for SSE
248
246
  */
249
247
  async *handleQueryStream(threadMetadata, query) {
250
248
  const { type, userId } = threadMetadata;
251
249
  const queryStartAt = Date.now();
252
- const threadMemory = _optionalChain([this, 'access', _17 => _17.memoryModule, 'optionalAccess', _18 => _18.getThreadMemory, 'call', _19 => _19()]);
250
+ const threadMemory = _optionalChain([this, 'access', _14 => _14.memoryModule, 'optionalAccess', _15 => _15.getThreadMemory, 'call', _16 => _16()]);
253
251
  let threadId = threadMetadata.threadId;
254
252
  let thread;
255
253
  if (threadId) {
256
- thread = await _optionalChain([threadMemory, 'optionalAccess', _20 => _20.getThread, 'call', _21 => _21(userId, threadId)]);
254
+ thread = await _optionalChain([threadMemory, 'optionalAccess', _17 => _17.getThread, 'call', _18 => _18(userId, threadId)]);
257
255
  } else {
258
256
  threadId = _crypto.randomUUID.call(void 0, );
259
257
  const title = await this.generateTitle(query);
260
- const metadata = await _optionalChain([threadMemory, 'optionalAccess', _22 => _22.createThread, 'call', _23 => _23(type, userId, threadId, title)]) || {
258
+ const metadata = await _optionalChain([threadMemory, 'optionalAccess', _19 => _19.createThread, 'call', _20 => _20(type, userId, threadId, title)]) || {
261
259
  type,
262
260
  threadId,
263
261
  title,
@@ -266,7 +264,7 @@ ${_optionalChain([intent, 'optionalAccess', _8 => _8.prompt]) || ""}
266
264
  _chunk466ED5WNcjs.loggers.intentStream.info("Create new thread", { metadata });
267
265
  yield { event: "thread_id", data: metadata };
268
266
  }
269
- await _optionalChain([threadMemory, 'optionalAccess', _24 => _24.addMessagesToThread, 'call', _25 => _25(userId, threadId, [
267
+ await _optionalChain([threadMemory, 'optionalAccess', _21 => _21.addMessagesToThread, 'call', _22 => _22(userId, threadId, [
270
268
  {
271
269
  role: "USER" /* USER */,
272
270
  timestamp: queryStartAt,
@@ -281,7 +279,7 @@ ${_optionalChain([intent, 'optionalAccess', _8 => _8.prompt]) || ""}
281
279
  _chunk466ED5WNcjs.loggers.intentStream.debug("text_chunk", { event });
282
280
  finalResponseText += event.data.delta;
283
281
  } else if (event.event === "tool_start") {
284
- await _optionalChain([threadMemory, 'optionalAccess', _26 => _26.addMessagesToThread, 'call', _27 => _27(userId, threadId, [
282
+ await _optionalChain([threadMemory, 'optionalAccess', _23 => _23.addMessagesToThread, 'call', _24 => _24(userId, threadId, [
285
283
  {
286
284
  role: "MODEL" /* MODEL */,
287
285
  timestamp: Date.now(),
@@ -297,7 +295,7 @@ ${_optionalChain([intent, 'optionalAccess', _8 => _8.prompt]) || ""}
297
295
  }
298
296
  ])]);
299
297
  } else if (event.event === "tool_output") {
300
- await _optionalChain([threadMemory, 'optionalAccess', _28 => _28.addMessagesToThread, 'call', _29 => _29(userId, threadId, [
298
+ await _optionalChain([threadMemory, 'optionalAccess', _25 => _25.addMessagesToThread, 'call', _26 => _26(userId, threadId, [
301
299
  {
302
300
  role: "MODEL" /* MODEL */,
303
301
  timestamp: Date.now(),
@@ -312,7 +310,7 @@ ${_optionalChain([intent, 'optionalAccess', _8 => _8.prompt]) || ""}
312
310
  }
313
311
  yield event;
314
312
  }
315
- await _optionalChain([threadMemory, 'optionalAccess', _30 => _30.addMessagesToThread, 'call', _31 => _31(userId, threadId, [
313
+ await _optionalChain([threadMemory, 'optionalAccess', _27 => _27.addMessagesToThread, 'call', _28 => _28(userId, threadId, [
316
314
  {
317
315
  role: "MODEL" /* MODEL */,
318
316
  timestamp: Date.now(),
@@ -325,4 +323,4 @@ ${_optionalChain([intent, 'optionalAccess', _8 => _8.prompt]) || ""}
325
323
 
326
324
 
327
325
  exports.QueryStreamService = QueryStreamService;
328
- //# sourceMappingURL=chunk-AIE3SVG7.cjs.map
326
+ //# sourceMappingURL=chunk-KHD7UAFW.cjs.map
@@ -0,0 +1 @@
1
+ {"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-KHD7UAFW.cjs","../../src/services/query-stream.service.ts"],"names":["intent"],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACA;ACJA,gCAA2B;AA+BpB,IAAM,mBAAA,EAAN,MAAyB;AAAA,EACvB;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EAER,WAAA,CACC,WAAA,EACA,SAAA,EACA,SAAA,EACA,YAAA,EACA,OAAA,EACC;AACD,IAAA,IAAA,CAAK,YAAA,EAAc,WAAA;AACnB,IAAA,IAAA,CAAK,UAAA,EAAY,SAAA;AACjB,IAAA,IAAA,CAAK,UAAA,EAAY,SAAA;AACjB,IAAA,IAAA,CAAK,aAAA,EAAe,YAAA;AACpB,IAAA,IAAA,CAAK,QAAA,EAAU,OAAA;AAAA,EAChB;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EASA,MAAc,gBAAA,CACb,KAAA,EACA,MAAA,EAC8B;AAC9B,IAAA,MAAM,cAAA,EAAgB,IAAA,CAAK,WAAA,CAAY,QAAA,CAAS,CAAA;AAChD,IAAA,MAAM,aAAA,kBAAe,IAAA,mBAAK,YAAA,6BAAc,eAAA,mBAAgB,GAAA;AACxD,IAAA,GAAA,CAAI,CAAC,YAAA,EAAc;AAClB,MAAA,OAAO,KAAA,CAAA;AAAA,IACR;AAGA,IAAA,MAAM,QAAA,EAAU,MAAM,YAAA,CAAa,WAAA,CAAY,CAAA;AAE/C,IAAA,GAAA,CAAI,OAAA,CAAQ,OAAA,IAAW,CAAA,EAAG;AACzB,MAAA,yBAAA,CAAQ,YAAA,CAAa,IAAA,CAAK,iBAAiB,CAAA;AAC3C,MAAA,OAAO,KAAA,CAAA;AAAA,IACR;AAEA,IAAA,MAAM,WAAA,EAAa,OAAA,CACjB,GAAA,CAAI,CAACA,OAAAA,EAAAA,GAAW,CAAA,EAAA,EAAKA,OAAAA,CAAO,IAAI,CAAA,EAAA,EAAKA,OAAAA,CAAO,WAAW,CAAA,CAAA;AAOvC,IAAA;AAOT,MAAA;AAGF,MAAA;AAC0B,MAAA;AAEpB,IAAA;AAEQ,IAAA;AAAA;AAAA;AAGX;AAAA;AAAA;AAAA,qFAAA;AAKU,IAAA;AAAA;AAEL;AAAA;AAEW,qBAAA;AAAA;AAAA;AAAA,yFAAA;AAKsB,IAAA;AACxC,MAAA;AACP,MAAA;AACA,IAAA;AAEkD,IAAA;AAC5B,IAAA;AACc,MAAA;AACrC,IAAA;AACyC,IAAA;AACmB,IAAA;AAC/C,IAAA;AACoC,MAAA;AACjD,IAAA;AACO,IAAA;AACR,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAuB+B,EAAA;AACT,IAAA;AACmB,SAAA;AAAA;AAEf;AAAA;AAEC;AAAA;AAEN;AACd,CAAA;AAE0C,IAAA;AACA,IAAA;AAC/C,MAAA;AACA,MAAA;AACgC,MAAA;AAChC,IAAA;AAE4B,IAAA;AAC4B,IAAA;AACQ,IAAA;AAEJ,IAAA;AAE9B,IAAA;AAElB,IAAA;AAC+B,MAAA;AAC1C,QAAA;AACA,QAAA;AACD,MAAA;AAMO,MAAA;AAE4C,MAAA;AAET,MAAA;AACrB,QAAA;AACG,QAAA;AACwC,UAAA;AAC/B,YAAA;AACzB,cAAA;AACE,cAAA;AAC8B,cAAA;AACrC,YAAA;AAEuC,YAAA;AACwB,YAAA;AACrD,YAAA;AAC4C,cAAA;AACvD,UAAA;AACgC,QAAA;AAC1B,UAAA;AACE,YAAA;AAC4B,YAAA;AACpC,UAAA;AACD,QAAA;AACD,MAAA;AAEiD,MAAA;AAChD,QAAA;AACA,MAAA;AAEkC,MAAA;AACS,QAAA;AACZ,UAAA;AACK,UAAA;AACgC,UAAA;AAElD,UAAA;AAIf,UAAA;AACsD,YAAA;AAGjD,YAAA;AACE,cAAA;AACD,cAAA;AACL,gBAAA;AACA,gBAAA;AACA,gBAAA;AACA,gBAAA;AACD,cAAA;AACD,YAAA;AAC4D,YAAA;AAC1B,YAAA;AACjC,cAAA;AACA,cAAA;AACD,YAAA;AAIC,UAAA;AACK,YAAA;AACE,cAAA;AACD,cAAA;AACL,gBAAA;AACA,gBAAA;AACA,gBAAA;AACU,gBAAA;AACX,cAAA;AACD,YAAA;AACkD,YAAA;AAChB,YAAA;AACjC,cAAA;AACA,cAAA;AACA,cAAA;AACD,YAAA;AACM,UAAA;AAES,YAAA;AACkC,cAAA;AACjD,YAAA;AACA,YAAA;AACD,UAAA;AACM,UAAA;AACE,YAAA;AACD,YAAA;AACL,cAAA;AACuB,cAAA;AACvB,cAAA;AACQ,cAAA;AACT,YAAA;AACD,UAAA;AACiD,UAAA;AAEtB,UAAA;AACsB,UAAA;AAClD,QAAA;AACM,MAAA;AACN,QAAA;AACD,MAAA;AACD,IAAA;AACD,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAS4D,EAAA;AACrC,IAAA;AAClB,IAAA;AAC6C,MAAA;AACA,MAAA;AAC/C,QAAA;AACc,QAAA;AAAA;AAAA;AAAA,oFAAA;AAId,MAAA;AACkD,MAAA;AACxB,MAAA;AACZ,IAAA;AACsC,MAAA;AACpD,QAAA;AACA,QAAA;AACA,MAAA;AACM,MAAA;AACR,IAAA;AACD,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAyB+B,EAAA;AACL,IAAA;AACK,IAAA;AAC0B,IAAA;AAG1B,IAAA;AAC1B,IAAA;AACU,IAAA;AAC0C,MAAA;AACjD,IAAA;AACgB,MAAA;AACsB,MAAA;AAI1C,MAAA;AACA,QAAA;AACA,QAAA;AACA,QAAA;AACoB,QAAA;AACrB,MAAA;AAC0D,MAAA;AAChB,MAAA;AAC5C,IAAA;AAE0D,IAAA;AACzD,MAAA;AACC,QAAA;AACW,QAAA;AAC6B,QAAA;AACzC,MAAA;AACA,IAAA;AAGuD,IAAA;AAGY,IAAA;AAE5C,IAAA;AACU,IAAA;AACqB,MAAA;AACH,QAAA;AAClB,QAAA;AACQ,MAAA;AACkB,QAAA;AACzD,UAAA;AACC,YAAA;AACoB,YAAA;AACX,YAAA;AACF,cAAA;AACqC,cAAA;AAC5C,YAAA;AACU,YAAA;AACc,cAAA;AACF,cAAA;AACA,cAAA;AACtB,YAAA;AACD,UAAA;AACA,QAAA;AACwC,MAAA;AACiB,QAAA;AACzD,UAAA;AACC,YAAA;AACoB,YAAA;AACgC,YAAA;AAC1C,YAAA;AACc,cAAA;AACF,cAAA;AACA,cAAA;AACtB,YAAA;AACD,UAAA;AACA,QAAA;AACF,MAAA;AACM,MAAA;AACP,IAAA;AAE0D,IAAA;AACzD,MAAA;AACC,QAAA;AACoB,QAAA;AACgC,QAAA;AACrD,MAAA;AACA,IAAA;AACF,EAAA;AACD;ADjH0F;AACA;AACA;AACA","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-KHD7UAFW.cjs","sourcesContent":[null,"import { randomUUID } from \"node:crypto\";\nimport type {\n\tA2AModule,\n\tMCPModule,\n\tMemoryModule,\n\tModelModule,\n} from \"@/modules/index.js\";\nimport type { AinAgentPrompts } from \"@/types/agent.js\";\nimport {\n\ttype Intent,\n\tMessageRole,\n\ttype ThreadMetadata,\n\ttype ThreadObject,\n\ttype ThreadType,\n} from \"@/types/memory.js\";\nimport type { StreamEvent } from \"@/types/stream\";\nimport {\n\ttype IA2ATool,\n\ttype IAgentTool,\n\ttype IMCPTool,\n\tTOOL_PROTOCOL_TYPE,\n} from \"@/types/tool.js\";\nimport { loggers } from \"@/utils/logger.js\";\n\n/**\n * Service for processing user queries through the agent's AI pipeline.\n *\n * Orchestrates the query processing workflow including intent detection,\n * model inference, tool execution, and response generation. Manages\n * conversation context and coordinates between different modules.\n */\nexport class QueryStreamService {\n\tprivate modelModule: ModelModule;\n\tprivate a2aModule?: A2AModule;\n\tprivate mcpModule?: MCPModule;\n\tprivate memoryModule?: MemoryModule;\n\tprivate prompts?: AinAgentPrompts;\n\n\tconstructor(\n\t\tmodelModule: ModelModule,\n\t\ta2aModule?: A2AModule,\n\t\tmcpModule?: MCPModule,\n\t\tmemoryModule?: MemoryModule,\n\t\tprompts?: AinAgentPrompts,\n\t) {\n\t\tthis.modelModule = modelModule;\n\t\tthis.a2aModule = a2aModule;\n\t\tthis.mcpModule = mcpModule;\n\t\tthis.memoryModule = memoryModule;\n\t\tthis.prompts = prompts;\n\t}\n\n\t/**\n\t * Detects the intent from context.\n\t *\n\t * @param query - The user's input query\n\t * @param thread - The thread history\n\t * @returns The detected intent\n\t */\n\tprivate async intentTriggering(\n\t\tquery: string,\n\t\tthread: ThreadObject | undefined,\n\t): Promise<Intent | undefined> {\n\t\tconst modelInstance = this.modelModule.getModel();\n\t\tconst intentMemory = this.memoryModule?.getIntentMemory();\n\t\tif (!intentMemory) {\n\t\t\treturn undefined;\n\t\t}\n\n\t\t// 인텐트 목록 가져오기\n\t\tconst intents = await intentMemory.listIntents();\n\n\t\tif (intents.length === 0) {\n\t\t\tloggers.intentStream.warn(\"No intent found\");\n\t\t\treturn undefined;\n\t\t}\n\n\t\tconst intentList = intents\n\t\t\t.map((intent) => `- ${intent.name}: ${intent.description}`)\n\t\t\t.join(\"\\n\");\n\n\t\t// Convert session history to a string\n\t\tconst historyMessages = !thread\n\t\t\t? \"\"\n\t\t\t: Object.entries(thread.messages)\n\t\t\t\t\t.sort(([, a], [, b]) => a.timestamp - b.timestamp)\n\t\t\t\t\t.map(([_chatId, chat]) => {\n\t\t\t\t\t\tconst role =\n\t\t\t\t\t\t\tchat.role === \"USER\"\n\t\t\t\t\t\t\t\t? \"User\"\n\t\t\t\t\t\t\t\t: chat.role === \"MODEL\"\n\t\t\t\t\t\t\t\t\t? \"Assistant\"\n\t\t\t\t\t\t\t\t\t: \"System\";\n\t\t\t\t\t\tconst content = Array.isArray(chat.content.parts)\n\t\t\t\t\t\t\t? chat.content.parts.join(\" \")\n\t\t\t\t\t\t\t: String(chat.content.parts);\n\t\t\t\t\t\treturn `${role}: \"\"\"${content}\"\"\"`;\n\t\t\t\t\t})\n\t\t\t\t\t.join(\"\\n\");\n\n\t\tconst systemPrompt = `You are an expert in accurately identifying user intentions.\n\nAvailable intent list:\n${intentList}\n\nPlease select and answer only from the above intent list. \nPlease return only the exact intent name without any additional explanations or text.`;\n\n\t\tconst userMessage = `The following is the conversation history with the user:\n\n${historyMessages}\n\nLast user question: \"${query}\"\n\nBased on the above conversation history, please determine what the intention of the last user question is. \nPlease select and answer the most appropriate intent name from the available intent list.`;\n\n\t\tconst messages = modelInstance.generateMessages({\n\t\t\tquery: userMessage,\n\t\t\tsystemPrompt,\n\t\t});\n\n\t\tconst response = await modelInstance.fetch(messages);\n\t\tif (!response.content) {\n\t\t\tthrow new Error(\"No intent detected\");\n\t\t}\n\t\tconst intentName = response.content.trim();\n\t\tconst intent = await intentMemory.getIntentByName(intentName);\n\t\tif (!intent) {\n\t\t\tthrow new Error(`No intent found: ${intentName}`);\n\t\t}\n\t\treturn intent;\n\t}\n\n\t/**\n\t * Fulfills the detected intent by generating a streaming response.\n\t *\n\t * Manages the complete inference loop including:\n\t * - Loading prompts and conversation history\n\t * - Collecting available tools from modules\n\t * - Executing model inference with tool support\n\t * - Processing tool calls iteratively until completion\n\t * - Streaming results as Server-Sent Events\n\t *\n\t * @param query - The user's input query\n\t * @param threadId - Thread identifier for context\n\t * @param thread - Previous conversation history\n\t * @param intent - Optional detected intent with custom prompt\n\t * @returns AsyncGenerator yielding StreamEvent objects\n\t */\n\tpublic async *intentFulfilling(\n\t\tquery: string,\n\t\tthreadId: string,\n\t\tthread?: ThreadObject,\n\t\tintent?: Intent,\n\t): AsyncGenerator<StreamEvent> {\n\t\tconst systemPrompt = `\nToday is ${new Date().toLocaleDateString()}.\n\n${this.prompts?.agent || \"\"}\n\n${this.prompts?.system || \"\"}\n\n${intent?.prompt || \"\"}\n\t`.trim();\n\n\t\tconst modelInstance = this.modelModule.getModel();\n\t\tconst messages = modelInstance.generateMessages({\n\t\t\tquery,\n\t\t\tthread,\n\t\t\tsystemPrompt: systemPrompt.trim(),\n\t\t});\n\n\t\tconst tools: IAgentTool[] = [];\n\t\tthis.mcpModule && tools.push(...this.mcpModule.getTools());\n\t\tthis.a2aModule && tools.push(...(await this.a2aModule.getTools()));\n\n\t\tconst functions = modelInstance.convertToolsToFunctions(tools);\n\n\t\tconst processList: string[] = [];\n\n\t\twhile (true) {\n\t\t\tconst responseStream = await modelInstance.fetchStreamWithContextMessage(\n\t\t\t\tmessages,\n\t\t\t\tfunctions,\n\t\t\t);\n\n\t\t\tconst assembledToolCalls: {\n\t\t\t\tid: string;\n\t\t\t\ttype: \"function\";\n\t\t\t\tfunction: { name: string; arguments: string };\n\t\t\t}[] = [];\n\n\t\t\tloggers.intentStream.debug(\"messages\", { messages });\n\n\t\t\tfor await (const chunk of responseStream) {\n\t\t\t\tconst delta = chunk.delta;\n\t\t\t\tif (delta?.tool_calls) {\n\t\t\t\t\tfor (const { index, id, function: func } of delta.tool_calls) {\n\t\t\t\t\t\tassembledToolCalls[index] ??= {\n\t\t\t\t\t\t\tid: \"\",\n\t\t\t\t\t\t\ttype: \"function\",\n\t\t\t\t\t\t\tfunction: { name: \"\", arguments: \"\" },\n\t\t\t\t\t\t};\n\n\t\t\t\t\t\tif (id) assembledToolCalls[index].id = id;\n\t\t\t\t\t\tif (func?.name) assembledToolCalls[index].function.name = func.name;\n\t\t\t\t\t\tif (func?.arguments)\n\t\t\t\t\t\t\tassembledToolCalls[index].function.arguments += func.arguments;\n\t\t\t\t\t}\n\t\t\t\t} else if (chunk.delta?.content) {\n\t\t\t\t\tyield {\n\t\t\t\t\t\tevent: \"text_chunk\",\n\t\t\t\t\t\tdata: { delta: chunk.delta.content },\n\t\t\t\t\t};\n\t\t\t\t}\n\t\t\t}\n\n\t\t\tloggers.intentStream.debug(\"assembledToolCalls\", {\n\t\t\t\tassembledToolCalls,\n\t\t\t});\n\n\t\t\tif (assembledToolCalls.length > 0) {\n\t\t\t\tfor (const toolCall of assembledToolCalls) {\n\t\t\t\t\tconst toolCallId = randomUUID();\n\t\t\t\t\tconst toolName = toolCall.function.name;\n\t\t\t\t\tconst selectedTool = tools.filter((tool) => tool.id === toolName)[0];\n\n\t\t\t\t\tlet toolResult = \"\";\n\t\t\t\t\tif (\n\t\t\t\t\t\tthis.mcpModule &&\n\t\t\t\t\t\tselectedTool.protocol === TOOL_PROTOCOL_TYPE.MCP\n\t\t\t\t\t) {\n\t\t\t\t\t\tconst toolArgs = JSON.parse(toolCall.function.arguments) as\n\t\t\t\t\t\t\t| { [x: string]: unknown }\n\t\t\t\t\t\t\t| undefined;\n\t\t\t\t\t\tyield {\n\t\t\t\t\t\t\tevent: \"tool_start\",\n\t\t\t\t\t\t\tdata: {\n\t\t\t\t\t\t\t\ttoolCallId,\n\t\t\t\t\t\t\t\tprotocol: TOOL_PROTOCOL_TYPE.MCP,\n\t\t\t\t\t\t\t\ttoolName,\n\t\t\t\t\t\t\t\ttoolArgs,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t};\n\t\t\t\t\t\tloggers.intent.debug(\"MCP tool call\", { toolName, toolArgs });\n\t\t\t\t\t\ttoolResult = await this.mcpModule.useTool(\n\t\t\t\t\t\t\tselectedTool as IMCPTool,\n\t\t\t\t\t\t\ttoolArgs,\n\t\t\t\t\t\t);\n\t\t\t\t\t} else if (\n\t\t\t\t\t\tthis.a2aModule &&\n\t\t\t\t\t\tselectedTool.protocol === TOOL_PROTOCOL_TYPE.A2A\n\t\t\t\t\t) {\n\t\t\t\t\t\tyield {\n\t\t\t\t\t\t\tevent: \"tool_start\",\n\t\t\t\t\t\t\tdata: {\n\t\t\t\t\t\t\t\ttoolCallId,\n\t\t\t\t\t\t\t\tprotocol: TOOL_PROTOCOL_TYPE.A2A,\n\t\t\t\t\t\t\t\ttoolName,\n\t\t\t\t\t\t\t\ttoolArgs: null,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t};\n\t\t\t\t\t\tloggers.intent.debug(\"A2A tool call\", { toolName });\n\t\t\t\t\t\ttoolResult = await this.a2aModule.useTool(\n\t\t\t\t\t\t\tselectedTool as IA2ATool,\n\t\t\t\t\t\t\tquery,\n\t\t\t\t\t\t\tthreadId,\n\t\t\t\t\t\t);\n\t\t\t\t\t} else {\n\t\t\t\t\t\t// Unrecognized tool type. It cannot be happened...\n\t\t\t\t\t\tloggers.intent.warn(\n\t\t\t\t\t\t\t`Unrecognized tool type: ${selectedTool.protocol}`,\n\t\t\t\t\t\t);\n\t\t\t\t\t\tcontinue;\n\t\t\t\t\t}\n\t\t\t\t\tyield {\n\t\t\t\t\t\tevent: \"tool_output\",\n\t\t\t\t\t\tdata: {\n\t\t\t\t\t\t\ttoolCallId,\n\t\t\t\t\t\t\tprotocol: selectedTool.protocol,\n\t\t\t\t\t\t\ttoolName,\n\t\t\t\t\t\t\tresult: toolResult,\n\t\t\t\t\t\t},\n\t\t\t\t\t};\n\t\t\t\t\tloggers.intent.debug(\"toolResult\", { toolResult });\n\n\t\t\t\t\tprocessList.push(toolResult);\n\t\t\t\t\tmodelInstance.appendMessages(messages, toolResult);\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\tbreak;\n\t\t\t}\n\t\t}\n\t}\n\n\t/**\n\t * Generates a title for the conversation based on the query.\n\t *\n\t * @param query - The user's input query\n\t * @returns Promise resolving to a generated title\n\t */\n\n\tprivate async generateTitle(query: string): Promise<string> {\n\t\tconst DEFAULT_TITLE = \"New Chat\";\n\t\ttry {\n\t\t\tconst modelInstance = this.modelModule.getModel();\n\t\t\tconst messages = modelInstance.generateMessages({\n\t\t\t\tquery,\n\t\t\t\tsystemPrompt: `You are a helpful assistant that generates titles for conversations.\n Please analyze the user's query and create a concise title that accurately reflects the conversation's core topic.\n The title must be no more than 5 words long.\n Respond with only the title. Do not include any punctuation or extra explanations.`,\n\t\t\t});\n\t\t\tconst response = await modelInstance.fetch(messages);\n\t\t\treturn response.content || DEFAULT_TITLE;\n\t\t} catch (error) {\n\t\t\tloggers.intentStream.error(\"Error generating title\", {\n\t\t\t\terror,\n\t\t\t\tquery,\n\t\t\t});\n\t\t\treturn DEFAULT_TITLE;\n\t\t}\n\t}\n\n\t/**\n\t * Main entry point for processing streaming user queries.\n\t *\n\t * Handles the complete query lifecycle:\n\t * 1. Loads or creates thread from memory\n\t * 2. Detects intent from the query\n\t * 3. Fulfills the intent with streaming AI response\n\t * 4. Updates conversation history in real-time\n\t *\n\t * @param threadMetadata - Metadata containing type, userId, and optional threadId\n\t * @param threadMetadata.type - The type of thread (e.g., chat, workflow)\n\t * @param threadMetadata.userId - The user's unique identifier\n\t * @param threadMetadata.threadId - Optional thread identifier\n\t * @param query - The user's input query\n\t * @returns AsyncGenerator yielding StreamEvent objects for SSE\n\t */\n\tpublic async *handleQueryStream(\n\t\tthreadMetadata: {\n\t\t\ttype: ThreadType;\n\t\t\tuserId: string;\n\t\t\tthreadId?: string;\n\t\t},\n\t\tquery: string,\n\t): AsyncGenerator<StreamEvent> {\n\t\tconst { type, userId } = threadMetadata;\n\t\tconst queryStartAt = Date.now();\n\t\tconst threadMemory = this.memoryModule?.getThreadMemory();\n\n\t\t// 1. Load or create thread\n\t\tlet threadId = threadMetadata.threadId;\n\t\tlet thread: ThreadObject | undefined;\n\t\tif (threadId) {\n\t\t\tthread = await threadMemory?.getThread(userId, threadId);\n\t\t} else {\n\t\t\tthreadId = randomUUID();\n\t\t\tconst title = await this.generateTitle(query);\n\n\t\t\tconst metadata =\n\t\t\t\t(await threadMemory?.createThread(type, userId, threadId, title)) ||\n\t\t\t\t({\n\t\t\t\t\ttype,\n\t\t\t\t\tthreadId,\n\t\t\t\t\ttitle,\n\t\t\t\t\tupdatedAt: Date.now(),\n\t\t\t\t} as ThreadMetadata);\n\t\t\tloggers.intentStream.info(\"Create new thread\", { metadata });\n\t\t\tyield { event: \"thread_id\", data: metadata };\n\t\t}\n\n\t\tawait threadMemory?.addMessagesToThread(userId, threadId, [\n\t\t\t{\n\t\t\t\trole: MessageRole.USER,\n\t\t\t\ttimestamp: queryStartAt,\n\t\t\t\tcontent: { type: \"text\", parts: [query] },\n\t\t\t},\n\t\t]);\n\n\t\t// 2. intent triggering\n\t\tconst intent = await this.intentTriggering(query, thread);\n\n\t\t// 3. intent fulfillment\n\t\tconst stream = this.intentFulfilling(query, threadId, thread, intent);\n\n\t\tlet finalResponseText = \"\";\n\t\tfor await (const event of stream) {\n\t\t\tif (event.event === \"text_chunk\" && event.data.delta) {\n\t\t\t\tloggers.intentStream.debug(\"text_chunk\", { event });\n\t\t\t\tfinalResponseText += event.data.delta;\n\t\t\t} else if (event.event === \"tool_start\") {\n\t\t\t\tawait threadMemory?.addMessagesToThread(userId, threadId, [\n\t\t\t\t\t{\n\t\t\t\t\t\trole: MessageRole.MODEL,\n\t\t\t\t\t\ttimestamp: Date.now(),\n\t\t\t\t\t\tcontent: {\n\t\t\t\t\t\t\ttype: \"text\",\n\t\t\t\t\t\t\tparts: [JSON.stringify(event.data.toolArgs)],\n\t\t\t\t\t\t},\n\t\t\t\t\t\tmetadata: {\n\t\t\t\t\t\t\ttoolCallId: event.data.toolCallId,\n\t\t\t\t\t\t\ttoolName: event.data.toolName,\n\t\t\t\t\t\t\tprotocol: event.data.protocol,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t]);\n\t\t\t} else if (event.event === \"tool_output\") {\n\t\t\t\tawait threadMemory?.addMessagesToThread(userId, threadId, [\n\t\t\t\t\t{\n\t\t\t\t\t\trole: MessageRole.MODEL,\n\t\t\t\t\t\ttimestamp: Date.now(),\n\t\t\t\t\t\tcontent: { type: \"text\", parts: [event.data.result] },\n\t\t\t\t\t\tmetadata: {\n\t\t\t\t\t\t\ttoolCallId: event.data.toolCallId,\n\t\t\t\t\t\t\ttoolName: event.data.toolName,\n\t\t\t\t\t\t\tprotocol: event.data.protocol,\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t]);\n\t\t\t}\n\t\t\tyield event;\n\t\t}\n\n\t\tawait threadMemory?.addMessagesToThread(userId, threadId, [\n\t\t\t{\n\t\t\t\trole: MessageRole.MODEL,\n\t\t\t\ttimestamp: Date.now(),\n\t\t\t\tcontent: { type: \"text\", parts: [finalResponseText] },\n\t\t\t},\n\t\t]);\n\t}\n}\n"]}
@@ -1,9 +1,9 @@
1
1
  "use strict";Object.defineProperty(exports, "__esModule", {value: true});
2
2
 
3
- var _chunk55IME4BGcjs = require('./chunk-55IME4BG.cjs');
3
+ var _chunkOKRBMMK2cjs = require('./chunk-OKRBMMK2.cjs');
4
4
 
5
5
 
6
- var _chunk7GWYMGWWcjs = require('./chunk-7GWYMGWW.cjs');
6
+ var _chunkKHD7UAFWcjs = require('./chunk-KHD7UAFW.cjs');
7
7
 
8
8
 
9
9
  var _chunkICFTUZP5cjs = require('./chunk-ICFTUZP5.cjs');
@@ -14,14 +14,14 @@ var _express = require('express');
14
14
  var createA2ARouter = (agent) => {
15
15
  const router = _express.Router.call(void 0, );
16
16
  const taskStore = new (0, _server.InMemoryTaskStore)();
17
- const queryService = new (0, _chunk7GWYMGWWcjs.QueryService)(
17
+ const queryStreamService = new (0, _chunkKHD7UAFWcjs.QueryStreamService)(
18
18
  agent.modelModule,
19
19
  agent.a2aModule,
20
20
  agent.mcpModule,
21
21
  agent.memoryModule,
22
22
  agent.manifest.prompts
23
23
  );
24
- const a2aService = new (0, _chunk55IME4BGcjs.A2AService)(queryService);
24
+ const a2aService = new (0, _chunkOKRBMMK2cjs.A2AService)(queryStreamService);
25
25
  const a2aController = new (0, _chunkICFTUZP5cjs.A2AController)(
26
26
  a2aService,
27
27
  taskStore,
@@ -34,4 +34,4 @@ var createA2ARouter = (agent) => {
34
34
 
35
35
 
36
36
  exports.createA2ARouter = createA2ARouter;
37
- //# sourceMappingURL=chunk-BOKIE5SM.cjs.map
37
+ //# sourceMappingURL=chunk-LMR6B76V.cjs.map
@@ -0,0 +1 @@
1
+ {"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-LMR6B76V.cjs","../../src/routes/a2a.routes.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACA;ACVA,4CAAkC;AAClC,kCAAuB;AAchB,IAAM,gBAAA,EAAkB,CAAC,KAAA,EAAA,GAA4B;AAC3D,EAAA,MAAM,OAAA,EAAS,6BAAA,CAAO;AAGtB,EAAA,MAAM,UAAA,EAAY,IAAI,8BAAA,CAAkB,CAAA;AACxC,EAAA,MAAM,mBAAA,EAAqB,IAAI,yCAAA;AAAA,IAC9B,KAAA,CAAM,WAAA;AAAA,IACN,KAAA,CAAM,SAAA;AAAA,IACN,KAAA,CAAM,SAAA;AAAA,IACN,KAAA,CAAM,YAAA;AAAA,IACN,KAAA,CAAM,QAAA,CAAS;AAAA,EAChB,CAAA;AACA,EAAA,MAAM,WAAA,EAAa,IAAI,iCAAA,CAAW,kBAAkB,CAAA;AACpD,EAAA,MAAM,cAAA,EAAgB,IAAI,oCAAA;AAAA,IACzB,UAAA;AAAA,IACA,SAAA;AAAA,IACA,KAAA,CAAM;AAAA,EACP,CAAA;AAGA,EAAA,MAAA,CAAO,IAAA,CAAK,GAAA,EAAK,aAAA,CAAc,gBAAgB,CAAA;AAE/C,EAAA,OAAO,MAAA;AACR,CAAA;ADNA;AACA;AACE;AACF,0CAAC","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-LMR6B76V.cjs","sourcesContent":[null,"import { InMemoryTaskStore } from \"@a2a-js/sdk/server\";\nimport { Router } from \"express\";\nimport type { AINAgent } from \"@/index.js\";\nimport { QueryStreamService } from \"@/services/query-stream.service.js\";\nimport { A2AController } from \"../controllers/a2a.controller.js\";\nimport { A2AService } from \"../services/a2a.service.js\";\n\n/**\n * Creates and configures the A2A router.\n * This function is a \"composition root\" for the A2A feature,\n * creating and injecting all necessary dependencies.\n * @param intentAnalyzer The core intent analyzer.\n * @param agentCard The agent's card.\n * @returns An Express Router instance.\n */\nexport const createA2ARouter = (agent: AINAgent): Router => {\n\tconst router = Router();\n\n\t// 1. Create dependencies for the A2A feature\n\tconst taskStore = new InMemoryTaskStore();\n\tconst queryStreamService = new QueryStreamService(\n\t\tagent.modelModule,\n\t\tagent.a2aModule,\n\t\tagent.mcpModule,\n\t\tagent.memoryModule,\n\t\tagent.manifest.prompts,\n\t);\n\tconst a2aService = new A2AService(queryStreamService);\n\tconst a2aController = new A2AController(\n\t\ta2aService,\n\t\ttaskStore,\n\t\tagent.generateAgentCard,\n\t);\n\n\t// 2. Define the route\n\trouter.post(\"/\", a2aController.handleA2ARequest);\n\n\treturn router;\n};\n"]}