@ainetwork/adk 0.3.0 → 0.3.1

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 (50) hide show
  1. package/dist/cjs/chunk-7QXQYCZP.cjs +249 -0
  2. package/dist/cjs/chunk-7QXQYCZP.cjs.map +1 -0
  3. package/dist/cjs/chunk-G42JXC76.cjs +41 -0
  4. package/dist/cjs/chunk-G42JXC76.cjs.map +1 -0
  5. package/dist/cjs/chunk-K63YCRBA.cjs +105 -0
  6. package/dist/cjs/chunk-K63YCRBA.cjs.map +1 -0
  7. package/dist/cjs/chunk-NYDNXGKB.cjs +108 -0
  8. package/dist/cjs/chunk-NYDNXGKB.cjs.map +1 -0
  9. package/dist/cjs/chunk-TF2SCL3K.cjs +36 -0
  10. package/dist/cjs/chunk-TF2SCL3K.cjs.map +1 -0
  11. package/dist/cjs/index.cjs +15 -15
  12. package/dist/cjs/index.cjs.map +1 -1
  13. package/dist/cjs/modules/a2a/a2a.module.cjs +1 -1
  14. package/dist/cjs/modules/index.cjs +9 -9
  15. package/dist/cjs/modules/index.cjs.map +1 -1
  16. package/dist/cjs/routes/a2a.routes.cjs +7 -7
  17. package/dist/cjs/routes/api.routes.cjs +4 -4
  18. package/dist/cjs/routes/index.cjs +14 -14
  19. package/dist/cjs/routes/index.cjs.map +1 -1
  20. package/dist/cjs/routes/intent.routes.cjs +6 -6
  21. package/dist/cjs/routes/query.routes.cjs +8 -8
  22. package/dist/cjs/services/index.cjs +7 -7
  23. package/dist/cjs/services/index.cjs.map +1 -1
  24. package/dist/cjs/services/intents/fulfill-stream.service.cjs +3 -3
  25. package/dist/cjs/services/intents/fulfill.service.cjs +1 -1
  26. package/dist/cjs/services/query-stream.service.cjs +6 -6
  27. package/dist/cjs/services/query.service.cjs +5 -5
  28. package/dist/esm/{chunk-ZK2KSFGQ.js → chunk-3FE3GPF2.js} +2 -2
  29. package/dist/esm/{chunk-3QCXIHM7.js → chunk-RJZ4QSDX.js} +2 -2
  30. package/dist/esm/{chunk-4NESTSZ3.js → chunk-V24ATQLM.js} +6 -3
  31. package/dist/esm/chunk-V24ATQLM.js.map +1 -0
  32. package/dist/esm/{chunk-YVN6VFRT.js → chunk-WRKUZMNO.js} +2 -2
  33. package/dist/esm/index.js +8 -8
  34. package/dist/esm/index.js.map +1 -1
  35. package/dist/esm/modules/a2a/a2a.module.js +1 -1
  36. package/dist/esm/modules/index.js +1 -1
  37. package/dist/esm/routes/a2a.routes.js +4 -4
  38. package/dist/esm/routes/index.js +6 -6
  39. package/dist/esm/routes/intent.routes.js +5 -5
  40. package/dist/esm/routes/query.routes.js +4 -4
  41. package/dist/esm/services/index.js +6 -6
  42. package/dist/esm/services/intents/fulfill-stream.service.js +2 -2
  43. package/dist/esm/services/intents/fulfill.service.js +1 -1
  44. package/dist/esm/services/query-stream.service.js +3 -3
  45. package/dist/esm/services/query.service.js +1 -1
  46. package/package.json +1 -1
  47. package/dist/esm/chunk-4NESTSZ3.js.map +0 -1
  48. /package/dist/esm/{chunk-ZK2KSFGQ.js.map → chunk-3FE3GPF2.js.map} +0 -0
  49. /package/dist/esm/{chunk-3QCXIHM7.js.map → chunk-RJZ4QSDX.js.map} +0 -0
  50. /package/dist/esm/{chunk-YVN6VFRT.js.map → chunk-WRKUZMNO.js.map} +0 -0
@@ -0,0 +1,249 @@
1
+ "use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _optionalChain(ops) { let lastAccessLHS = undefined; let value = ops[0]; let i = 1; while (i < ops.length) { const op = ops[i]; const fn = ops[i + 1]; i += 2; if ((op === 'optionalAccess' || op === 'optionalCall') && value == null) { return undefined; } if (op === 'access' || op === 'optionalAccess') { lastAccessLHS = value; value = fn(value); } else if (op === 'call' || op === 'optionalCall') { value = fn((...args) => value.call(lastAccessLHS, ...args)); lastAccessLHS = undefined; } } return value; }
2
+
3
+ var _chunk7HBOMBZGcjs = require('./chunk-7HBOMBZG.cjs');
4
+
5
+
6
+ var _chunk466ED5WNcjs = require('./chunk-466ED5WN.cjs');
7
+
8
+ // src/services/intents/fulfill-stream.service.ts
9
+ var _crypto = require('crypto');
10
+ var IntentFulfillStreamService = class {
11
+
12
+
13
+
14
+
15
+ constructor(modelModule, a2aModule, mcpModule, memoryModule) {
16
+ this.modelModule = modelModule;
17
+ this.a2aModule = a2aModule;
18
+ this.mcpModule = mcpModule;
19
+ this.memoryModule = memoryModule;
20
+ }
21
+ async addToThreadMessages(thread, params) {
22
+ try {
23
+ const threadMemory = _optionalChain([this, 'access', _ => _.memoryModule, 'optionalAccess', _2 => _2.getThreadMemory, 'call', _3 => _3()]);
24
+ const { userId, threadId } = thread;
25
+ const newMessage = {
26
+ messageId: _crypto.randomUUID.call(void 0, ),
27
+ role: params.role,
28
+ timestamp: Date.now(),
29
+ content: { type: "text", parts: [params.content] },
30
+ metadata: params.metadata
31
+ };
32
+ thread.messages.push(newMessage);
33
+ await _optionalChain([threadMemory, 'optionalAccess', _4 => _4.addMessagesToThread, 'call', _5 => _5(userId, threadId, [newMessage])]);
34
+ } catch (error) {
35
+ _chunk466ED5WNcjs.loggers.intentStream.error("Error adding message to thread", error);
36
+ }
37
+ }
38
+ /**
39
+ * Fulfills the detected intent by generating a streaming response.
40
+ *
41
+ * Manages the complete inference loop including:
42
+ * - Loading prompts and conversation history
43
+ * - Collecting available tools from modules
44
+ * - Executing model inference with tool support
45
+ * - Processing tool calls iteratively until completion
46
+ * - Streaming results as Server-Sent Events
47
+ *
48
+ * @param query - The user's input query
49
+ * @param threadId - Thread identifier for context
50
+ * @param thread - Previous conversation history
51
+ * @param intent - Optional detected intent with custom prompt
52
+ * @returns AsyncGenerator yielding StreamEvent objects
53
+ */
54
+ async *intentFulfilling(query, thread, intent) {
55
+ const agentMemory = _optionalChain([this, 'access', _6 => _6.memoryModule, 'optionalAccess', _7 => _7.getAgentMemory, 'call', _8 => _8()]);
56
+ const fulfillPrompt = await _chunk7HBOMBZGcjs.createFulfillPrompt.call(void 0, agentMemory, intent);
57
+ const modelInstance = this.modelModule.getModel();
58
+ const modelOptions = this.modelModule.getModelOptions();
59
+ const messages = modelInstance.generateMessages({
60
+ query,
61
+ thread,
62
+ systemPrompt: fulfillPrompt.trim()
63
+ });
64
+ _chunk466ED5WNcjs.loggers.intent.debug("Intent fulfillment start", {
65
+ threadId: thread.threadId,
66
+ messages
67
+ });
68
+ const tools = [];
69
+ this.mcpModule && tools.push(...this.mcpModule.getTools());
70
+ this.a2aModule && tools.push(...await this.a2aModule.getTools());
71
+ const processList = [];
72
+ while (true) {
73
+ const functions = modelInstance.convertToolsToFunctions(tools);
74
+ const responseStream = await modelInstance.fetchStreamWithContextMessage(
75
+ messages,
76
+ functions,
77
+ modelOptions
78
+ );
79
+ const assembledToolCalls = [];
80
+ for await (const chunk of responseStream) {
81
+ const delta = chunk.delta;
82
+ if (_optionalChain([delta, 'optionalAccess', _9 => _9.tool_calls])) {
83
+ for (const { index, id, function: func } of delta.tool_calls) {
84
+ assembledToolCalls[index] ??= {
85
+ id: "",
86
+ type: "function",
87
+ function: { name: "", arguments: "" }
88
+ };
89
+ if (id) assembledToolCalls[index].id = id;
90
+ if (_optionalChain([func, 'optionalAccess', _10 => _10.name])) assembledToolCalls[index].function.name = func.name;
91
+ if (_optionalChain([func, 'optionalAccess', _11 => _11.arguments]))
92
+ assembledToolCalls[index].function.arguments += func.arguments;
93
+ }
94
+ } else if (_optionalChain([chunk, 'access', _12 => _12.delta, 'optionalAccess', _13 => _13.content])) {
95
+ yield {
96
+ event: "text_chunk",
97
+ data: { delta: chunk.delta.content }
98
+ };
99
+ }
100
+ }
101
+ _chunk466ED5WNcjs.loggers.intentStream.debug("assembledToolCalls", {
102
+ threadId: thread.threadId,
103
+ assembledToolCalls
104
+ });
105
+ if (assembledToolCalls.length > 0) {
106
+ for (const toolCall of assembledToolCalls) {
107
+ const toolCallId = _crypto.randomUUID.call(void 0, );
108
+ const toolName = toolCall.function.name;
109
+ let selectedTool;
110
+ for (const [index, toolTmp] of tools.entries()) {
111
+ if (toolTmp.toolName === toolName) {
112
+ if (toolTmp.protocol === "A2A" /* A2A */) {
113
+ selectedTool = tools.splice(index, 1)[0];
114
+ break;
115
+ }
116
+ selectedTool = toolTmp;
117
+ }
118
+ }
119
+ if (!selectedTool) {
120
+ continue;
121
+ }
122
+ let toolResult = "";
123
+ if (this.mcpModule && selectedTool.protocol === "MCP" /* MCP */) {
124
+ const toolArgs = JSON.parse(toolCall.function.arguments);
125
+ yield {
126
+ event: "tool_start",
127
+ data: {
128
+ toolCallId,
129
+ protocol: "MCP" /* MCP */,
130
+ toolName,
131
+ toolArgs
132
+ }
133
+ };
134
+ _chunk466ED5WNcjs.loggers.intent.info("MCP tool call", { toolName, toolArgs });
135
+ toolResult = await this.mcpModule.useTool(selectedTool, toolArgs);
136
+ } else if (this.a2aModule && selectedTool.protocol === "A2A" /* A2A */) {
137
+ yield {
138
+ event: "tool_start",
139
+ data: {
140
+ toolCallId,
141
+ protocol: "A2A" /* A2A */,
142
+ toolName,
143
+ toolArgs: null
144
+ }
145
+ };
146
+ _chunk466ED5WNcjs.loggers.intent.info("A2A tool call", { toolName });
147
+ toolResult = await this.a2aModule.useTool(
148
+ selectedTool,
149
+ query,
150
+ thread.threadId
151
+ );
152
+ } else {
153
+ _chunk466ED5WNcjs.loggers.intent.warn(
154
+ `Unrecognized tool type: ${selectedTool.protocol}`
155
+ );
156
+ continue;
157
+ }
158
+ yield {
159
+ event: "tool_output",
160
+ data: {
161
+ toolCallId,
162
+ protocol: selectedTool.protocol,
163
+ toolName,
164
+ result: toolResult
165
+ }
166
+ };
167
+ _chunk466ED5WNcjs.loggers.intent.debug("Tool Result", { toolResult });
168
+ processList.push(toolResult);
169
+ modelInstance.appendMessages(messages, toolResult);
170
+ }
171
+ } else {
172
+ break;
173
+ }
174
+ }
175
+ _chunk466ED5WNcjs.loggers.intent.debug("Intent fulfillment completed", {
176
+ threadId: thread.threadId,
177
+ toolCallsExecuted: processList.length,
178
+ intentName: _optionalChain([intent, 'optionalAccess', _14 => _14.name])
179
+ });
180
+ }
181
+ /**
182
+ * Detects the intent from context.
183
+ *
184
+ * @param intents - The user's input query
185
+ * @param thread - The thread history
186
+ * @returns The detected intent
187
+ */
188
+ async *intentFulfillStream(intents, thread) {
189
+ const streamStartTime = Date.now();
190
+ _chunk466ED5WNcjs.loggers.intentStream.info("Stream session started", {
191
+ threadId: thread.threadId,
192
+ intentCount: intents.length,
193
+ startTime: new Date(streamStartTime).toISOString()
194
+ });
195
+ let finalResponseText = "";
196
+ for (let i = 0; i < intents.length; i++) {
197
+ const { subquery, intent, actionPlan } = intents[i];
198
+ _chunk466ED5WNcjs.loggers.intent.info(`Process query: ${subquery}, ${_optionalChain([intent, 'optionalAccess', _15 => _15.name])}`);
199
+ _chunk466ED5WNcjs.loggers.intent.info(`Action plan: ${actionPlan}`);
200
+ finalResponseText !== "" && thread.messages.push({
201
+ messageId: _crypto.randomUUID.call(void 0, ),
202
+ role: "MODEL" /* MODEL */,
203
+ timestamp: Date.now(),
204
+ content: { type: "text", parts: [finalResponseText] },
205
+ metadata: { isThinking: true }
206
+ });
207
+ await this.addToThreadMessages(thread, {
208
+ role: "MODEL" /* MODEL */,
209
+ content: subquery,
210
+ metadata: {
211
+ subquery,
212
+ isThinking: true,
213
+ actionPlan
214
+ }
215
+ });
216
+ yield {
217
+ event: "intent_process",
218
+ data: { subquery, actionPlan: actionPlan || "" }
219
+ };
220
+ const stream = this.intentFulfilling(subquery, thread, intent);
221
+ finalResponseText = "";
222
+ for await (const event of stream) {
223
+ if (event.event === "text_chunk" && event.data.delta) {
224
+ finalResponseText += event.data.delta;
225
+ }
226
+ if (event.event === "text_chunk" && i !== intents.length - 1) {
227
+ continue;
228
+ }
229
+ yield event;
230
+ }
231
+ }
232
+ await this.addToThreadMessages(thread, {
233
+ role: "MODEL" /* MODEL */,
234
+ content: finalResponseText
235
+ });
236
+ const streamEndTime = Date.now();
237
+ const streamDuration = streamEndTime - streamStartTime;
238
+ _chunk466ED5WNcjs.loggers.intentStream.info("Stream session completed", {
239
+ threadId: thread.threadId,
240
+ duration: `${streamDuration}ms`,
241
+ endTime: new Date(streamEndTime).toISOString()
242
+ });
243
+ }
244
+ };
245
+
246
+
247
+
248
+ exports.IntentFulfillStreamService = IntentFulfillStreamService;
249
+ //# sourceMappingURL=chunk-7QXQYCZP.cjs.map
@@ -0,0 +1 @@
1
+ {"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-7QXQYCZP.cjs","../../src/services/intents/fulfill-stream.service.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACA;ACPA,gCAA2B;AAmBpB,IAAM,2BAAA,EAAN,MAAiC;AAAA,EAC/B;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EAER,WAAA,CACC,WAAA,EACA,SAAA,EACA,SAAA,EACA,YAAA,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;AAAA,EACrB;AAAA,EAEA,MAAc,mBAAA,CACb,MAAA,EACA,MAAA,EAKC;AACD,IAAA,IAAI;AACH,MAAA,MAAM,aAAA,kBAAe,IAAA,mBAAK,YAAA,6BAAc,eAAA,mBAAgB,GAAA;AACxD,MAAA,MAAM,EAAE,MAAA,EAAQ,SAAS,EAAA,EAAI,MAAA;AAC7B,MAAA,MAAM,WAAA,EAA4B;AAAA,QACjC,SAAA,EAAW,gCAAA,CAAW;AAAA,QACtB,IAAA,EAAM,MAAA,CAAO,IAAA;AAAA,QACb,SAAA,EAAW,IAAA,CAAK,GAAA,CAAI,CAAA;AAAA,QACpB,OAAA,EAAS,EAAE,IAAA,EAAM,MAAA,EAAQ,KAAA,EAAO,CAAC,MAAA,CAAO,OAAO,EAAE,CAAA;AAAA,QACjD,QAAA,EAAU,MAAA,CAAO;AAAA,MAClB,CAAA;AACA,MAAA,MAAA,CAAO,QAAA,CAAS,IAAA,CAAK,UAAU,CAAA;AAC/B,MAAA,sBAAM,YAAA,6BAAc,mBAAA,mBAAoB,MAAA,EAAQ,QAAA,EAAU,CAAC,UAAU,CAAC,GAAA;AAAA,IACvE,EAAA,MAAA,CAAS,KAAA,EAAO;AACf,MAAA,yBAAA,CAAQ,YAAA,CAAa,KAAA,CAAM,gCAAA,EAAkC,KAAK,CAAA;AAAA,IACnE;AAAA,EACD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAkBA,MAAA,CAAe,gBAAA,CACd,KAAA,EACA,MAAA,EACA,MAAA,EAC8B;AAC9B,IAAA,MAAM,YAAA,kBAAc,IAAA,qBAAK,YAAA,6BAAc,cAAA,mBAAe,GAAA;AACtD,IAAA,MAAM,cAAA,EAAgB,MAAM,mDAAA,WAAoB,EAAa,MAAM,CAAA;AAEnE,IAAA,MAAM,cAAA,EAAgB,IAAA,CAAK,WAAA,CAAY,QAAA,CAAS,CAAA;AAChD,IAAA,MAAM,aAAA,EAAe,IAAA,CAAK,WAAA,CAAY,eAAA,CAAgB,CAAA;AACtD,IAAA,MAAM,SAAA,EAAW,aAAA,CAAc,gBAAA,CAAiB;AAAA,MAC/C,KAAA;AAAA,MACA,MAAA;AAAA,MACA,YAAA,EAAc,aAAA,CAAc,IAAA,CAAK;AAAA,IAClC,CAAC,CAAA;AAED,IAAA,yBAAA,CAAQ,MAAA,CAAO,KAAA,CAAM,0BAAA,EAA4B;AAAA,MAChD,QAAA,EAAU,MAAA,CAAO,QAAA;AAAA,MACjB;AAAA,IACD,CAAC,CAAA;AAED,IAAA,MAAM,MAAA,EAAyB,CAAC,CAAA;AAChC,IAAA,IAAA,CAAK,UAAA,GAAa,KAAA,CAAM,IAAA,CAAK,GAAG,IAAA,CAAK,SAAA,CAAU,QAAA,CAAS,CAAC,CAAA;AACzD,IAAA,IAAA,CAAK,UAAA,GAAa,KAAA,CAAM,IAAA,CAAK,GAAI,MAAM,IAAA,CAAK,SAAA,CAAU,QAAA,CAAS,CAAE,CAAA;AAEjE,IAAA,MAAM,YAAA,EAAwB,CAAC,CAAA;AAE/B,IAAA,MAAA,CAAO,IAAA,EAAM;AACZ,MAAA,MAAM,UAAA,EAAY,aAAA,CAAc,uBAAA,CAAwB,KAAK,CAAA;AAC7D,MAAA,MAAM,eAAA,EAAiB,MAAM,aAAA,CAAc,6BAAA;AAAA,QAC1C,QAAA;AAAA,QACA,SAAA;AAAA,QACA;AAAA,MACD,CAAA;AAEA,MAAA,MAAM,mBAAA,EAIA,CAAC,CAAA;AAEP,MAAA,IAAA,MAAA,CAAA,MAAiB,MAAA,GAAS,cAAA,EAAgB;AACzC,QAAA,MAAM,MAAA,EAAQ,KAAA,CAAM,KAAA;AACpB,QAAA,GAAA,iBAAI,KAAA,6BAAO,YAAA,EAAY;AACtB,UAAA,IAAA,CAAA,MAAW,EAAE,KAAA,EAAO,EAAA,EAAI,QAAA,EAAU,KAAK,EAAA,GAAK,KAAA,CAAM,UAAA,EAAY;AAC7D,YAAA,kBAAA,CAAmB,KAAK,EAAA,IAAM;AAAA,cAC7B,EAAA,EAAI,EAAA;AAAA,cACJ,IAAA,EAAM,UAAA;AAAA,cACN,QAAA,EAAU,EAAE,IAAA,EAAM,EAAA,EAAI,SAAA,EAAW,GAAG;AAAA,YACrC,CAAA;AAEA,YAAA,GAAA,CAAI,EAAA,EAAI,kBAAA,CAAmB,KAAK,CAAA,CAAE,GAAA,EAAK,EAAA;AACvC,YAAA,GAAA,iBAAI,IAAA,+BAAM,MAAA,EAAM,kBAAA,CAAmB,KAAK,CAAA,CAAE,QAAA,CAAS,KAAA,EAAO,IAAA,CAAK,IAAA;AAC/D,YAAA,GAAA,iBAAI,IAAA,+BAAM,WAAA;AACT,cAAA,kBAAA,CAAmB,KAAK,CAAA,CAAE,QAAA,CAAS,UAAA,GAAa,IAAA,CAAK,SAAA;AAAA,UACvD;AAAA,QACD,EAAA,KAAA,GAAA,iBAAW,KAAA,uBAAM,KAAA,+BAAO,SAAA,EAAS;AAChC,UAAA,MAAM;AAAA,YACL,KAAA,EAAO,YAAA;AAAA,YACP,IAAA,EAAM,EAAE,KAAA,EAAO,KAAA,CAAM,KAAA,CAAM,QAAQ;AAAA,UACpC,CAAA;AAAA,QACD;AAAA,MACD;AAEA,MAAA,yBAAA,CAAQ,YAAA,CAAa,KAAA,CAAM,oBAAA,EAAsB;AAAA,QAChD,QAAA,EAAU,MAAA,CAAO,QAAA;AAAA,QACjB;AAAA,MACD,CAAC,CAAA;AAED,MAAA,GAAA,CAAI,kBAAA,CAAmB,OAAA,EAAS,CAAA,EAAG;AAClC,QAAA,IAAA,CAAA,MAAW,SAAA,GAAY,kBAAA,EAAoB;AAC1C,UAAA,MAAM,WAAA,EAAa,gCAAA,CAAW;AAC9B,UAAA,MAAM,SAAA,EAAW,QAAA,CAAS,QAAA,CAAS,IAAA;AACnC,UAAA,IAAI,YAAA;AACJ,UAAA,IAAA,CAAA,MAAW,CAAC,KAAA,EAAO,OAAO,EAAA,GAAK,KAAA,CAAM,OAAA,CAAQ,CAAA,EAAG;AAC/C,YAAA,GAAA,CAAI,OAAA,CAAQ,SAAA,IAAa,QAAA,EAAU;AAClC,cAAA,GAAA,CAAI,OAAA,CAAQ,SAAA,IAAA,eAAA,EAA0C;AAErD,gBAAA,aAAA,EAAe,KAAA,CAAM,MAAA,CAAO,KAAA,EAAO,CAAC,CAAA,CAAE,CAAC,CAAA;AACvC,gBAAA,KAAA;AAAA,cACD;AACA,cAAA,aAAA,EAAe,OAAA;AAAA,YAChB;AAAA,UACD;AAEA,UAAA,GAAA,CAAI,CAAC,YAAA,EAAc;AAElB,YAAA,QAAA;AAAA,UACD;AAEA,UAAA,IAAI,WAAA,EAAa,EAAA;AACjB,UAAA,GAAA,CACC,IAAA,CAAK,UAAA,GACL,YAAA,CAAa,SAAA,IAAA,eAAA,EACZ;AACD,YAAA,MAAM,SAAA,EAAW,IAAA,CAAK,KAAA,CAAM,QAAA,CAAS,QAAA,CAAS,SAAS,CAAA;AAGvD,YAAA,MAAM;AAAA,cACL,KAAA,EAAO,YAAA;AAAA,cACP,IAAA,EAAM;AAAA,gBACL,UAAA;AAAA,gBACA,QAAA,EAAA,eAAA;AAAA,gBACA,QAAA;AAAA,gBACA;AAAA,cACD;AAAA,YACD,CAAA;AACA,YAAA,yBAAA,CAAQ,MAAA,CAAO,IAAA,CAAK,eAAA,EAAiB,EAAE,QAAA,EAAU,SAAS,CAAC,CAAA;AAC3D,YAAA,WAAA,EAAa,MAAM,IAAA,CAAK,SAAA,CAAU,OAAA,CAAQ,YAAA,EAAc,QAAQ,CAAA;AAAA,UACjE,EAAA,KAAA,GAAA,CACC,IAAA,CAAK,UAAA,GACL,YAAA,CAAa,SAAA,IAAA,eAAA,EACZ;AACD,YAAA,MAAM;AAAA,cACL,KAAA,EAAO,YAAA;AAAA,cACP,IAAA,EAAM;AAAA,gBACL,UAAA;AAAA,gBACA,QAAA,EAAA,eAAA;AAAA,gBACA,QAAA;AAAA,gBACA,QAAA,EAAU;AAAA,cACX;AAAA,YACD,CAAA;AACA,YAAA,yBAAA,CAAQ,MAAA,CAAO,IAAA,CAAK,eAAA,EAAiB,EAAE,SAAS,CAAC,CAAA;AACjD,YAAA,WAAA,EAAa,MAAM,IAAA,CAAK,SAAA,CAAU,OAAA;AAAA,cACjC,YAAA;AAAA,cACA,KAAA;AAAA,cACA,MAAA,CAAO;AAAA,YACR,CAAA;AAAA,UACD,EAAA,KAAO;AAEN,YAAA,yBAAA,CAAQ,MAAA,CAAO,IAAA;AAAA,cACd,CAAA,wBAAA,EAA2B,YAAA,CAAa,QAAQ,CAAA;AAAA,YAAA;AAEjD,YAAA;AAAA,UAAA;AAED,UAAA;AAAM,YAAA;AACE,YAAA;AACD,cAAA;AACL,cAAA;AACuB,cAAA;AACvB,cAAA;AACQ,YAAA;AACT,UAAA;AAGD,UAAA;AAEA,UAAA;AACA,UAAA;AAAiD,QAAA;AAGlD,MAAA;AAEA,QAAA;AAAA,MAAA;AACD,IAAA;AAGD,IAAA;AAAqD,MAAA;AACnC,MAAA;AACc,MAAA;AACX,IAAA;AACpB,EAAA;AACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAAA;AAaC,IAAA;AACA,IAAA;AAAoD,MAAA;AAClC,MAAA;AACI,MAAA;AAC4B,IAAA;AAGlD,IAAA;AAEA,IAAA;AACC,MAAA;AACA,MAAA;AACA,MAAA;AAGA,MAAA;AACsB,QAAA;AACE,QAAA;AACtB,QAAA;AACoB,QAAA;AACgC,QAAA;AACvB,MAAA;AAE/B,MAAA;AAAuC,QAAA;AACtC,QAAA;AACS,QAAA;AACC,UAAA;AACT,UAAA;AACY,UAAA;AACZ,QAAA;AACD,MAAA;AAGD,MAAA;AAAM,QAAA;AACE,QAAA;AACwC,MAAA;AAGhD,MAAA;AAEA,MAAA;AACA,MAAA;AACC,QAAA;AACC,UAAA;AAAgC,QAAA;AAGjC,QAAA;AACC,UAAA;AAAA,QAAA;AAED,QAAA;AAAM,MAAA;AACP,IAAA;AAGD,IAAA;AAAuC,MAAA;AACtC,MAAA;AACS,IAAA;AAGV,IAAA;AACA,IAAA;AAEA,IAAA;AAAsD,MAAA;AACpC,MAAA;AACU,MAAA;AACkB,IAAA;AAC7C,EAAA;AAEH;AD5EA;AACA;AACA;AACA","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-7QXQYCZP.cjs","sourcesContent":[null,"import { randomUUID } from \"node:crypto\";\nimport type {\n\tA2AModule,\n\tMCPModule,\n\tMemoryModule,\n\tModelModule,\n} from \"@/modules\";\nimport { CONNECTOR_PROTOCOL_TYPE, type ConnectorTool } from \"@/types/connector\";\nimport {\n\ttype Intent,\n\ttype MessageObject,\n\tMessageRole,\n\ttype ThreadObject,\n\ttype TriggeredIntent,\n} from \"@/types/memory\";\nimport type { StreamEvent } from \"@/types/stream\";\nimport { loggers } from \"@/utils/logger\";\nimport { createFulfillPrompt } from \"../utils/fulfill.common\";\n\nexport class IntentFulfillStreamService {\n\tprivate modelModule: ModelModule;\n\tprivate a2aModule?: A2AModule;\n\tprivate mcpModule?: MCPModule;\n\tprivate memoryModule?: MemoryModule;\n\n\tconstructor(\n\t\tmodelModule: ModelModule,\n\t\ta2aModule?: A2AModule,\n\t\tmcpModule?: MCPModule,\n\t\tmemoryModule?: MemoryModule,\n\t) {\n\t\tthis.modelModule = modelModule;\n\t\tthis.a2aModule = a2aModule;\n\t\tthis.mcpModule = mcpModule;\n\t\tthis.memoryModule = memoryModule;\n\t}\n\n\tprivate async addToThreadMessages(\n\t\tthread: ThreadObject,\n\t\tparams: {\n\t\t\trole: MessageRole;\n\t\t\tcontent: string;\n\t\t\tmetadata?: Record<string, unknown>;\n\t\t},\n\t) {\n\t\ttry {\n\t\t\tconst threadMemory = this.memoryModule?.getThreadMemory();\n\t\t\tconst { userId, threadId } = thread;\n\t\t\tconst newMessage: MessageObject = {\n\t\t\t\tmessageId: randomUUID(),\n\t\t\t\trole: params.role,\n\t\t\t\ttimestamp: Date.now(),\n\t\t\t\tcontent: { type: \"text\", parts: [params.content] },\n\t\t\t\tmetadata: params.metadata,\n\t\t\t};\n\t\t\tthread.messages.push(newMessage);\n\t\t\tawait threadMemory?.addMessagesToThread(userId, threadId, [newMessage]);\n\t\t} catch (error) {\n\t\t\tloggers.intentStream.error(\"Error adding message to thread\", error);\n\t\t}\n\t}\n\n\t/**\n\t * Fulfills the detected intent by generating a streaming response.\n\t *\n\t * Manages the complete inference loop including:\n\t * - Loading prompts and conversation history\n\t * - Collecting available tools from modules\n\t * - Executing model inference with tool support\n\t * - Processing tool calls iteratively until completion\n\t * - Streaming results as Server-Sent Events\n\t *\n\t * @param query - The user's input query\n\t * @param threadId - Thread identifier for context\n\t * @param thread - Previous conversation history\n\t * @param intent - Optional detected intent with custom prompt\n\t * @returns AsyncGenerator yielding StreamEvent objects\n\t */\n\tprivate async *intentFulfilling(\n\t\tquery: string,\n\t\tthread: ThreadObject,\n\t\tintent?: Intent,\n\t): AsyncGenerator<StreamEvent> {\n\t\tconst agentMemory = this.memoryModule?.getAgentMemory();\n\t\tconst fulfillPrompt = await createFulfillPrompt(agentMemory, intent);\n\n\t\tconst modelInstance = this.modelModule.getModel();\n\t\tconst modelOptions = this.modelModule.getModelOptions();\n\t\tconst messages = modelInstance.generateMessages({\n\t\t\tquery,\n\t\t\tthread,\n\t\t\tsystemPrompt: fulfillPrompt.trim(),\n\t\t});\n\n\t\tloggers.intent.debug(\"Intent fulfillment start\", {\n\t\t\tthreadId: thread.threadId,\n\t\t\tmessages,\n\t\t});\n\n\t\tconst tools: ConnectorTool[] = [];\n\t\tthis.mcpModule && tools.push(...this.mcpModule.getTools());\n\t\tthis.a2aModule && tools.push(...(await this.a2aModule.getTools()));\n\n\t\tconst processList: string[] = [];\n\n\t\twhile (true) {\n\t\t\tconst functions = modelInstance.convertToolsToFunctions(tools);\n\t\t\tconst responseStream = await modelInstance.fetchStreamWithContextMessage(\n\t\t\t\tmessages,\n\t\t\t\tfunctions,\n\t\t\t\tmodelOptions,\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\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\tthreadId: thread.threadId,\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\tlet selectedTool: ConnectorTool | undefined;\n\t\t\t\t\tfor (const [index, toolTmp] of tools.entries()) {\n\t\t\t\t\t\tif (toolTmp.toolName === toolName) {\n\t\t\t\t\t\t\tif (toolTmp.protocol === CONNECTOR_PROTOCOL_TYPE.A2A) {\n\t\t\t\t\t\t\t\t// remove used tool to prevent infinite loop\n\t\t\t\t\t\t\t\tselectedTool = tools.splice(index, 1)[0];\n\t\t\t\t\t\t\t\tbreak;\n\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\tselectedTool = toolTmp;\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\n\t\t\t\t\tif (!selectedTool) {\n\t\t\t\t\t\t// it cannot be happened...\n\t\t\t\t\t\tcontinue;\n\t\t\t\t\t}\n\n\t\t\t\t\tlet toolResult = \"\";\n\t\t\t\t\tif (\n\t\t\t\t\t\tthis.mcpModule &&\n\t\t\t\t\t\tselectedTool.protocol === CONNECTOR_PROTOCOL_TYPE.MCP\n\t\t\t\t\t) {\n\t\t\t\t\t\tconst toolArgs = 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: CONNECTOR_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.info(\"MCP tool call\", { toolName, toolArgs });\n\t\t\t\t\t\ttoolResult = await this.mcpModule.useTool(selectedTool, toolArgs);\n\t\t\t\t\t} else if (\n\t\t\t\t\t\tthis.a2aModule &&\n\t\t\t\t\t\tselectedTool.protocol === CONNECTOR_PROTOCOL_TYPE.A2A\n\t\t\t\t\t) {\n\t\t\t\t\t\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: CONNECTOR_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.info(\"A2A tool call\", { toolName });\n\t\t\t\t\t\ttoolResult = await this.a2aModule.useTool(\n\t\t\t\t\t\t\tselectedTool,\n\t\t\t\t\t\t\tquery,\n\t\t\t\t\t\t\tthread.threadId,\n\t\t\t\t\t\t);\n\t\t\t\t\t} else {\n\t\t\t\t\t\t// Unrecognized tool type. It cannot be happened...\n\t\t\t\t\t\tloggers.intent.warn(\n\t\t\t\t\t\t\t`Unrecognized tool type: ${selectedTool.protocol}`,\n\t\t\t\t\t\t);\n\t\t\t\t\t\tcontinue;\n\t\t\t\t\t}\n\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\n\t\t\t\t\tloggers.intent.debug(\"Tool Result\", { toolResult });\n\n\t\t\t\t\tprocessList.push(toolResult);\n\t\t\t\t\tmodelInstance.appendMessages(messages, toolResult);\n\n\t\t\t\t\t// remove used tool to prevent infinite loop\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\tbreak;\n\t\t\t}\n\t\t}\n\n\t\tloggers.intent.debug(\"Intent fulfillment completed\", {\n\t\t\tthreadId: thread.threadId,\n\t\t\ttoolCallsExecuted: processList.length,\n\t\t\tintentName: intent?.name,\n\t\t});\n\t}\n\n\t/**\n\t * Detects the intent from context.\n\t *\n\t * @param intents - The user's input query\n\t * @param thread - The thread history\n\t * @returns The detected intent\n\t */\n\tpublic async *intentFulfillStream(\n\t\tintents: Array<TriggeredIntent>,\n\t\tthread: ThreadObject,\n\t): AsyncGenerator<StreamEvent> {\n\t\tconst streamStartTime = Date.now();\n\t\tloggers.intentStream.info(\"Stream session started\", {\n\t\t\tthreadId: thread.threadId,\n\t\t\tintentCount: intents.length,\n\t\t\tstartTime: new Date(streamStartTime).toISOString(),\n\t\t});\n\n\t\tlet finalResponseText = \"\";\n\n\t\tfor (let i = 0; i < intents.length; i++) {\n\t\t\tconst { subquery, intent, actionPlan } = intents[i];\n\t\t\tloggers.intent.info(`Process query: ${subquery}, ${intent?.name}`);\n\t\t\tloggers.intent.info(`Action plan: ${actionPlan}`);\n\n\t\t\t// only use for inference, not stored in memory\n\t\t\tfinalResponseText !== \"\" &&\n\t\t\t\tthread.messages.push({\n\t\t\t\t\tmessageId: randomUUID(),\n\t\t\t\t\trole: MessageRole.MODEL,\n\t\t\t\t\ttimestamp: Date.now(),\n\t\t\t\t\tcontent: { type: \"text\", parts: [finalResponseText] },\n\t\t\t\t\tmetadata: { isThinking: true },\n\t\t\t\t});\n\t\t\tawait this.addToThreadMessages(thread, {\n\t\t\t\trole: MessageRole.MODEL,\n\t\t\t\tcontent: subquery,\n\t\t\t\tmetadata: {\n\t\t\t\t\tsubquery,\n\t\t\t\t\tisThinking: true,\n\t\t\t\t\tactionPlan: actionPlan,\n\t\t\t\t},\n\t\t\t});\n\n\t\t\tyield {\n\t\t\t\tevent: \"intent_process\",\n\t\t\t\tdata: { subquery, actionPlan: actionPlan || \"\" },\n\t\t\t};\n\n\t\t\tconst stream = this.intentFulfilling(subquery, thread, intent);\n\n\t\t\tfinalResponseText = \"\";\n\t\t\tfor await (const event of stream) {\n\t\t\t\tif (event.event === \"text_chunk\" && event.data.delta) {\n\t\t\t\t\tfinalResponseText += event.data.delta;\n\t\t\t\t}\n\n\t\t\t\tif (event.event === \"text_chunk\" && i !== intents.length - 1) {\n\t\t\t\t\tcontinue; // skip intermediate text_chunk events\n\t\t\t\t}\n\t\t\t\tyield event;\n\t\t\t}\n\t\t}\n\n\t\tawait this.addToThreadMessages(thread, {\n\t\t\trole: MessageRole.MODEL,\n\t\t\tcontent: finalResponseText,\n\t\t});\n\n\t\tconst streamEndTime = Date.now();\n\t\tconst streamDuration = streamEndTime - streamStartTime;\n\n\t\tloggers.intentStream.info(\"Stream session completed\", {\n\t\t\tthreadId: thread.threadId,\n\t\t\tduration: `${streamDuration}ms`,\n\t\t\tendTime: new Date(streamEndTime).toISOString(),\n\t\t});\n\t}\n}\n"]}
@@ -0,0 +1,41 @@
1
+ "use strict";Object.defineProperty(exports, "__esModule", {value: true});
2
+
3
+ var _chunkK63YCRBAcjs = require('./chunk-K63YCRBA.cjs');
4
+
5
+
6
+ var _chunkNYDNXGKBcjs = require('./chunk-NYDNXGKB.cjs');
7
+
8
+
9
+ var _chunkAQXHOGJDcjs = require('./chunk-AQXHOGJD.cjs');
10
+
11
+ // src/routes/query.routes.ts
12
+ var _express = require('express');
13
+ var createQueryRouter = (agent, allowStream = false) => {
14
+ const router = _express.Router.call(void 0, );
15
+ const queryService = new (0, _chunkK63YCRBAcjs.QueryService)(
16
+ agent.modelModule,
17
+ agent.a2aModule,
18
+ agent.mcpModule,
19
+ agent.memoryModule
20
+ );
21
+ let queryStreamService;
22
+ if (allowStream) {
23
+ queryStreamService = new (0, _chunkNYDNXGKBcjs.QueryStreamService)(
24
+ agent.modelModule,
25
+ agent.a2aModule,
26
+ agent.mcpModule,
27
+ agent.memoryModule
28
+ );
29
+ }
30
+ const queryController = new (0, _chunkAQXHOGJDcjs.QueryController)(queryService, queryStreamService);
31
+ router.post("/", queryController.handleQueryRequest);
32
+ if (allowStream) {
33
+ router.post("/stream", queryController.handleQueryStreamRequest);
34
+ }
35
+ return router;
36
+ };
37
+
38
+
39
+
40
+ exports.createQueryRouter = createQueryRouter;
41
+ //# sourceMappingURL=chunk-G42JXC76.cjs.map
@@ -0,0 +1 @@
1
+ {"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-G42JXC76.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;AAAA,EACP,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;AAAA,IACP,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-G42JXC76.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);\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);\n\t}\n\n\tconst queryController = new QueryController(queryService, queryStreamService);\n\trouter.post(\"/\", queryController.handleQueryRequest);\n\n\tif (allowStream) {\n\t\trouter.post(\"/stream\", queryController.handleQueryStreamRequest);\n\t}\n\n\treturn router;\n};\n"]}
@@ -0,0 +1,105 @@
1
+ "use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _optionalChain(ops) { let lastAccessLHS = undefined; let value = ops[0]; let i = 1; while (i < ops.length) { const op = ops[i]; const fn = ops[i + 1]; i += 2; if ((op === 'optionalAccess' || op === 'optionalCall') && value == null) { return undefined; } if (op === 'access' || op === 'optionalAccess') { lastAccessLHS = value; value = fn(value); } else if (op === 'call' || op === 'optionalCall') { value = fn((...args) => value.call(lastAccessLHS, ...args)); lastAccessLHS = undefined; } } return value; }
2
+
3
+ var _chunkX5INT5HEcjs = require('./chunk-X5INT5HE.cjs');
4
+
5
+
6
+ var _chunkMO3H4KHEcjs = require('./chunk-MO3H4KHE.cjs');
7
+
8
+
9
+ var _chunkTF7S7SPOcjs = require('./chunk-TF7S7SPO.cjs');
10
+
11
+
12
+ var _chunkZ6TQ7TFOcjs = require('./chunk-Z6TQ7TFO.cjs');
13
+
14
+
15
+ var _chunk466ED5WNcjs = require('./chunk-466ED5WN.cjs');
16
+
17
+ // src/services/query.service.ts
18
+ var _crypto = require('crypto');
19
+ var _httpstatuscodes = require('http-status-codes');
20
+ var QueryService = class {
21
+
22
+
23
+
24
+
25
+ constructor(modelModule, a2aModule, mcpModule, memoryModule) {
26
+ this.modelModule = modelModule;
27
+ this.memoryModule = memoryModule;
28
+ this.intentTriggerService = new (0, _chunkTF7S7SPOcjs.IntentTriggerService)(
29
+ modelModule,
30
+ memoryModule
31
+ );
32
+ this.intentFulfillService = new (0, _chunkX5INT5HEcjs.IntentFulfillService)(
33
+ modelModule,
34
+ a2aModule,
35
+ mcpModule,
36
+ memoryModule
37
+ );
38
+ }
39
+ /**
40
+ * Main entry point for processing user queries.
41
+ *
42
+ * Handles the complete query lifecycle:
43
+ * 1. Loads or creates thread from memory
44
+ * 2. Detects intent from the query
45
+ * 3. Fulfills the intent with AI response
46
+ * 4. Updates conversation history
47
+ *
48
+ * @param threadMetadata - Metadata containing type, userId, and optional threadId
49
+ * @param threadMetadata.type - The type of thread (e.g., chat, workflow)
50
+ * @param threadMetadata.userId - The user's unique identifier
51
+ * @param threadMetadata.threadId - Optional thread identifier
52
+ * @param query - The user's input query
53
+ * @returns Promise resolving to object with content property containing the response
54
+ */
55
+ async handleQuery(threadMetadata, query, isA2A) {
56
+ const { type, userId, options } = threadMetadata;
57
+ const threadMemory = _optionalChain([this, 'access', _ => _.memoryModule, 'optionalAccess', _2 => _2.getThreadMemory, 'call', _3 => _3()]);
58
+ let threadId = threadMetadata.threadId;
59
+ let thread;
60
+ if (threadId) {
61
+ thread = await _optionalChain([threadMemory, 'optionalAccess', _4 => _4.getThread, 'call', _5 => _5(userId, threadId)]);
62
+ if (!thread && !isA2A) {
63
+ throw new (0, _chunkZ6TQ7TFOcjs.AinHttpError)(_httpstatuscodes.StatusCodes.NOT_FOUND, "Thread not found");
64
+ }
65
+ }
66
+ threadId ??= _crypto.randomUUID.call(void 0, );
67
+ if (!thread) {
68
+ const title = await _chunkMO3H4KHEcjs.generateTitle.call(void 0, this.modelModule, query, options);
69
+ const metadata = await _optionalChain([threadMemory, 'optionalAccess', _6 => _6.createThread, 'call', _7 => _7(
70
+ type,
71
+ userId,
72
+ threadId,
73
+ title
74
+ )]) || { type, userId, threadId, title };
75
+ thread = { ...metadata, messages: [] };
76
+ _chunk466ED5WNcjs.loggers.intent.info(`Create new thread: ${threadId}`);
77
+ }
78
+ const triggeredIntent = await this.intentTriggerService.intentTriggering(query, thread);
79
+ _chunk466ED5WNcjs.loggers.intent.debug("Triggered intents", { triggeredIntent });
80
+ await _optionalChain([threadMemory, 'optionalAccess', _8 => _8.addMessagesToThread, 'call', _9 => _9(userId, threadId, [
81
+ {
82
+ messageId: _crypto.randomUUID.call(void 0, ),
83
+ role: "USER" /* USER */,
84
+ timestamp: Date.now(),
85
+ content: { type: "text", parts: [query] },
86
+ metadata: {
87
+ intents: triggeredIntent.filter((intent) => !!intent.intent).map((intent) => ({
88
+ id: _optionalChain([intent, 'access', _10 => _10.intent, 'optionalAccess', _11 => _11.id]),
89
+ subquery: intent.subquery
90
+ }))
91
+ }
92
+ }
93
+ ])]);
94
+ const result = await this.intentFulfillService.intentFulfill(
95
+ triggeredIntent,
96
+ thread
97
+ );
98
+ return { content: result };
99
+ }
100
+ };
101
+
102
+
103
+
104
+ exports.QueryService = QueryService;
105
+ //# sourceMappingURL=chunk-K63YCRBA.cjs.map
@@ -0,0 +1 @@
1
+ {"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-K63YCRBA.cjs","../../src/services/query.service.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACA;AChBA,gCAA2B;AAC3B,oDAA4B;AA4BrB,IAAM,aAAA,EAAN,MAAmB;AAAA,EACjB;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EAER,WAAA,CACC,WAAA,EACA,SAAA,EACA,SAAA,EACA,YAAA,EACC;AACD,IAAA,IAAA,CAAK,YAAA,EAAc,WAAA;AACnB,IAAA,IAAA,CAAK,aAAA,EAAe,YAAA;AACpB,IAAA,IAAA,CAAK,qBAAA,EAAuB,IAAI,2CAAA;AAAA,MAC/B,WAAA;AAAA,MACA;AAAA,IACD,CAAA;AACA,IAAA,IAAA,CAAK,qBAAA,EAAuB,IAAI,2CAAA;AAAA,MAC/B,WAAA;AAAA,MACA,SAAA;AAAA,MACA,SAAA;AAAA,MACA;AAAA,IACD,CAAA;AAAA,EACD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAkBA,MAAa,WAAA,CACZ,cAAA,EAMA,KAAA,EACA,KAAA,EACC;AACD,IAAA,MAAM,EAAE,IAAA,EAAM,MAAA,EAAQ,QAAQ,EAAA,EAAI,cAAA;AAClC,IAAA,MAAM,aAAA,kBAAe,IAAA,mBAAK,YAAA,6BAAc,eAAA,mBAAgB,GAAA;AAGxD,IAAA,IAAI,SAAA,EAAW,cAAA,CAAe,QAAA;AAC9B,IAAA,IAAI,MAAA;AACJ,IAAA,GAAA,CAAI,QAAA,EAAU;AACb,MAAA,OAAA,EAAS,sBAAM,YAAA,6BAAc,SAAA,mBAAU,MAAA,EAAQ,QAAQ,GAAA;AACvD,MAAA,GAAA,CAAI,CAAC,OAAA,GAAU,CAAC,KAAA,EAAO;AACtB,QAAA,MAAM,IAAI,mCAAA,CAAa,4BAAA,CAAY,SAAA,EAAW,kBAAkB,CAAA;AAAA,MACjE;AAAA,IACD;AAEA,IAAA,SAAA,IAAa,gCAAA,CAAW;AACxB,IAAA,GAAA,CAAI,CAAC,MAAA,EAAQ;AACZ,MAAA,MAAM,MAAA,EAAQ,MAAM,6CAAA,IAAc,CAAK,WAAA,EAAa,KAAA,EAAO,OAAO,CAAA;AAClE,MAAA,MAAM,SAAA,EAA4B,sBAAM,YAAA,6BAAc,YAAA;AAAA,QACrD,IAAA;AAAA,QACA,MAAA;AAAA,QACA,QAAA;AAAA,QACA;AAAA,MACD,IAAA,GAAM,EAAE,IAAA,EAAM,MAAA,EAAQ,QAAA,EAAU,MAAM,CAAA;AACtC,MAAA,OAAA,EAAS,EAAE,GAAG,QAAA,EAAU,QAAA,EAAU,CAAC,EAAE,CAAA;AACrC,MAAA,yBAAA,CAAQ,MAAA,CAAO,IAAA,CAAK,CAAA,mBAAA,EAAsB,QAAQ,CAAA,CAAA;AACnD,IAAA;AAIY,IAAA;AACgC,IAAA;AAGI,IAAA;AAC/C,MAAA;AACuB,QAAA;AACtB,QAAA;AACoB,QAAA;AACoB,QAAA;AAC9B,QAAA;AAEc,UAAA;AAEF,YAAA;AACF,YAAA;AAChB,UAAA;AACJ,QAAA;AACD,MAAA;AACA,IAAA;AAG8C,IAAA;AAC9C,MAAA;AACA,MAAA;AACD,IAAA;AAEyB,IAAA;AAC1B,EAAA;AACD;ADtCyD;AACA;AACA;AACA","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-K63YCRBA.cjs","sourcesContent":[null,"import { randomUUID } from \"node:crypto\";\nimport { StatusCodes } from \"http-status-codes\";\nimport type {\n\tA2AModule,\n\tMCPModule,\n\tMemoryModule,\n\tModelFetchOptions,\n\tModelModule,\n} from \"@/modules/index.js\";\nimport { AinHttpError } from \"@/types/agent.js\";\nimport {\n\tMessageRole,\n\ttype ThreadMetadata,\n\ttype ThreadObject,\n\ttype ThreadType,\n\ttype TriggeredIntent,\n} from \"@/types/memory.js\";\nimport { loggers } from \"@/utils/logger.js\";\nimport { IntentFulfillService } from \"./intents/fulfill.service\";\nimport { IntentTriggerService } from \"./intents/trigger.service\";\nimport { generateTitle } from \"./utils/query.common\";\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 QueryService {\n\tprivate modelModule: ModelModule;\n\tprivate memoryModule?: MemoryModule;\n\tprivate intentTriggerService: IntentTriggerService;\n\tprivate intentFulfillService: IntentFulfillService;\n\n\tconstructor(\n\t\tmodelModule: ModelModule,\n\t\ta2aModule?: A2AModule,\n\t\tmcpModule?: MCPModule,\n\t\tmemoryModule?: MemoryModule,\n\t) {\n\t\tthis.modelModule = modelModule;\n\t\tthis.memoryModule = memoryModule;\n\t\tthis.intentTriggerService = new IntentTriggerService(\n\t\t\tmodelModule,\n\t\t\tmemoryModule,\n\t\t);\n\t\tthis.intentFulfillService = new IntentFulfillService(\n\t\t\tmodelModule,\n\t\t\ta2aModule,\n\t\t\tmcpModule,\n\t\t\tmemoryModule,\n\t\t);\n\t}\n\n\t/**\n\t * Main entry point for processing user queries.\n\t *\n\t * Handles the complete query lifecycle:\n\t * 1. Loads or creates thread from memory\n\t * 2. Detects intent from the query\n\t * 3. Fulfills the intent with AI response\n\t * 4. Updates conversation history\n\t *\n\t * @param 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 Promise resolving to object with content property containing the response\n\t */\n\tpublic async handleQuery(\n\t\tthreadMetadata: {\n\t\t\ttype: ThreadType;\n\t\t\tuserId: string;\n\t\t\tthreadId?: string;\n\t\t\toptions?: ModelFetchOptions;\n\t\t},\n\t\tquery: string,\n\t\tisA2A?: boolean,\n\t) {\n\t\tconst { type, userId, options } = threadMetadata;\n\t\tconst threadMemory = this.memoryModule?.getThreadMemory();\n\n\t\t// 1. Load or create thread\n\t\tlet threadId = threadMetadata.threadId;\n\t\tlet thread: ThreadObject | undefined;\n\t\tif (threadId) {\n\t\t\tthread = await threadMemory?.getThread(userId, threadId);\n\t\t\tif (!thread && !isA2A) {\n\t\t\t\tthrow new AinHttpError(StatusCodes.NOT_FOUND, \"Thread not found\");\n\t\t\t}\n\t\t}\n\n\t\tthreadId ??= randomUUID();\n\t\tif (!thread) {\n\t\t\tconst title = await generateTitle(this.modelModule, query, options);\n\t\t\tconst metadata: ThreadMetadata = (await threadMemory?.createThread(\n\t\t\t\ttype,\n\t\t\t\tuserId,\n\t\t\t\tthreadId,\n\t\t\t\ttitle,\n\t\t\t)) || { type, userId, threadId, title };\n\t\t\tthread = { ...metadata, messages: [] };\n\t\t\tloggers.intent.info(`Create new thread: ${threadId}`);\n\t\t}\n\n\t\t// 2. intent triggering\n\t\tconst triggeredIntent: Array<TriggeredIntent> =\n\t\t\tawait this.intentTriggerService.intentTriggering(query, thread);\n\t\tloggers.intent.debug(\"Triggered intents\", { triggeredIntent });\n\n\t\t// only add for storage, not for inference\n\t\tawait threadMemory?.addMessagesToThread(userId, threadId, [\n\t\t\t{\n\t\t\t\tmessageId: randomUUID(),\n\t\t\t\trole: MessageRole.USER,\n\t\t\t\ttimestamp: Date.now(),\n\t\t\t\tcontent: { type: \"text\", parts: [query] },\n\t\t\t\tmetadata: {\n\t\t\t\t\tintents: triggeredIntent\n\t\t\t\t\t\t.filter((intent) => !!intent.intent)\n\t\t\t\t\t\t.map((intent) => ({\n\t\t\t\t\t\t\tid: intent.intent?.id,\n\t\t\t\t\t\t\tsubquery: intent.subquery,\n\t\t\t\t\t\t})),\n\t\t\t\t},\n\t\t\t},\n\t\t]);\n\n\t\t// 3. intent fulfillment\n\t\tconst result = await this.intentFulfillService.intentFulfill(\n\t\t\ttriggeredIntent,\n\t\t\tthread,\n\t\t);\n\n\t\treturn { content: result };\n\t}\n}\n"]}
@@ -0,0 +1,108 @@
1
+ "use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _optionalChain(ops) { let lastAccessLHS = undefined; let value = ops[0]; let i = 1; while (i < ops.length) { const op = ops[i]; const fn = ops[i + 1]; i += 2; if ((op === 'optionalAccess' || op === 'optionalCall') && value == null) { return undefined; } if (op === 'access' || op === 'optionalAccess') { lastAccessLHS = value; value = fn(value); } else if (op === 'call' || op === 'optionalCall') { value = fn((...args) => value.call(lastAccessLHS, ...args)); lastAccessLHS = undefined; } } return value; }
2
+
3
+ var _chunk7QXQYCZPcjs = require('./chunk-7QXQYCZP.cjs');
4
+
5
+
6
+ var _chunkMO3H4KHEcjs = require('./chunk-MO3H4KHE.cjs');
7
+
8
+
9
+ var _chunkTF7S7SPOcjs = require('./chunk-TF7S7SPO.cjs');
10
+
11
+
12
+ var _chunkZ6TQ7TFOcjs = require('./chunk-Z6TQ7TFO.cjs');
13
+
14
+
15
+ var _chunk466ED5WNcjs = require('./chunk-466ED5WN.cjs');
16
+
17
+ // src/services/query-stream.service.ts
18
+ var _crypto = require('crypto');
19
+ var _httpstatuscodes = require('http-status-codes');
20
+ var QueryStreamService = class {
21
+
22
+
23
+
24
+
25
+ constructor(modelModule, a2aModule, mcpModule, memoryModule) {
26
+ this.modelModule = modelModule;
27
+ this.memoryModule = memoryModule;
28
+ this.intentTriggerService = new (0, _chunkTF7S7SPOcjs.IntentTriggerService)(
29
+ modelModule,
30
+ memoryModule
31
+ );
32
+ this.intentFulfillStreamService = new (0, _chunk7QXQYCZPcjs.IntentFulfillStreamService)(
33
+ modelModule,
34
+ a2aModule,
35
+ mcpModule,
36
+ memoryModule
37
+ );
38
+ }
39
+ /**
40
+ * Main entry point for processing streaming user queries.
41
+ *
42
+ * Handles the complete query lifecycle:
43
+ * 1. Loads or creates thread from memory
44
+ * 2. Detects intent from the query
45
+ * 3. Fulfills the intent with streaming AI response
46
+ * 4. Updates conversation history in real-time
47
+ *
48
+ * @param threadMetadata - Metadata containing type, userId, and optional threadId
49
+ * @param threadMetadata.type - The type of thread (e.g., chat, workflow)
50
+ * @param threadMetadata.userId - The user's unique identifier
51
+ * @param threadMetadata.threadId - Optional thread identifier
52
+ * @param query - The user's input query
53
+ * @returns AsyncGenerator yielding StreamEvent objects for SSE
54
+ */
55
+ async *handleQueryStream(threadMetadata, query, isA2A) {
56
+ const { type, userId, options } = threadMetadata;
57
+ const threadMemory = _optionalChain([this, 'access', _ => _.memoryModule, 'optionalAccess', _2 => _2.getThreadMemory, 'call', _3 => _3()]);
58
+ let threadId = threadMetadata.threadId;
59
+ let thread;
60
+ if (threadId) {
61
+ thread = await _optionalChain([threadMemory, 'optionalAccess', _4 => _4.getThread, 'call', _5 => _5(userId, threadId)]);
62
+ if (!thread && !isA2A) {
63
+ throw new (0, _chunkZ6TQ7TFOcjs.AinHttpError)(_httpstatuscodes.StatusCodes.NOT_FOUND, "Thread not found");
64
+ }
65
+ }
66
+ threadId ??= _crypto.randomUUID.call(void 0, );
67
+ if (!thread) {
68
+ const title = await _chunkMO3H4KHEcjs.generateTitle.call(void 0, this.modelModule, query, options);
69
+ const metadata = await _optionalChain([threadMemory, 'optionalAccess', _6 => _6.createThread, 'call', _7 => _7(
70
+ type,
71
+ userId,
72
+ threadId,
73
+ title
74
+ )]) || { type, userId, threadId, title };
75
+ thread = { ...metadata, messages: [] };
76
+ _chunk466ED5WNcjs.loggers.intent.info(`Create new thread: ${threadId}`);
77
+ yield { event: "thread_id", data: { type, userId, threadId, title } };
78
+ }
79
+ const triggeredIntent = await this.intentTriggerService.intentTriggering(query, thread);
80
+ _chunk466ED5WNcjs.loggers.intent.debug("Triggered intents", { triggeredIntent });
81
+ await _optionalChain([threadMemory, 'optionalAccess', _8 => _8.addMessagesToThread, 'call', _9 => _9(userId, threadId, [
82
+ {
83
+ messageId: _crypto.randomUUID.call(void 0, ),
84
+ role: "USER" /* USER */,
85
+ timestamp: Date.now(),
86
+ content: { type: "text", parts: [query] },
87
+ metadata: {
88
+ intents: triggeredIntent.filter((intent) => !!intent.intent).map((intent) => ({
89
+ id: _optionalChain([intent, 'access', _10 => _10.intent, 'optionalAccess', _11 => _11.id]),
90
+ subquery: intent.subquery
91
+ }))
92
+ }
93
+ }
94
+ ])]);
95
+ const stream = this.intentFulfillStreamService.intentFulfillStream(
96
+ triggeredIntent,
97
+ thread
98
+ );
99
+ for await (const event of stream) {
100
+ yield event;
101
+ }
102
+ }
103
+ };
104
+
105
+
106
+
107
+ exports.QueryStreamService = QueryStreamService;
108
+ //# sourceMappingURL=chunk-NYDNXGKB.cjs.map
@@ -0,0 +1 @@
1
+ {"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-NYDNXGKB.cjs","../../src/services/query-stream.service.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACA;AChBA,gCAA2B;AAC3B,oDAA4B;AA6BrB,IAAM,mBAAA,EAAN,MAAyB;AAAA,EACvB;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EAER,WAAA,CACC,WAAA,EACA,SAAA,EACA,SAAA,EACA,YAAA,EACC;AACD,IAAA,IAAA,CAAK,YAAA,EAAc,WAAA;AACnB,IAAA,IAAA,CAAK,aAAA,EAAe,YAAA;AACpB,IAAA,IAAA,CAAK,qBAAA,EAAuB,IAAI,2CAAA;AAAA,MAC/B,WAAA;AAAA,MACA;AAAA,IACD,CAAA;AACA,IAAA,IAAA,CAAK,2BAAA,EAA6B,IAAI,iDAAA;AAAA,MACrC,WAAA;AAAA,MACA,SAAA;AAAA,MACA,SAAA;AAAA,MACA;AAAA,IACD,CAAA;AAAA,EACD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAkBA,MAAA,CAAc,iBAAA,CACb,cAAA,EAMA,KAAA,EACA,KAAA,EAC8B;AAC9B,IAAA,MAAM,EAAE,IAAA,EAAM,MAAA,EAAQ,QAAQ,EAAA,EAAI,cAAA;AAClC,IAAA,MAAM,aAAA,kBAAe,IAAA,mBAAK,YAAA,6BAAc,eAAA,mBAAgB,GAAA;AAGxD,IAAA,IAAI,SAAA,EAAW,cAAA,CAAe,QAAA;AAC9B,IAAA,IAAI,MAAA;AACJ,IAAA,GAAA,CAAI,QAAA,EAAU;AACb,MAAA,OAAA,EAAS,sBAAM,YAAA,6BAAc,SAAA,mBAAU,MAAA,EAAQ,QAAQ,GAAA;AACvD,MAAA,GAAA,CAAI,CAAC,OAAA,GAAU,CAAC,KAAA,EAAO;AACtB,QAAA,MAAM,IAAI,mCAAA,CAAa,4BAAA,CAAY,SAAA,EAAW,kBAAkB,CAAA;AAAA,MACjE;AAAA,IACD;AAEA,IAAA,SAAA,IAAa,gCAAA,CAAW;AACxB,IAAA,GAAA,CAAI,CAAC,MAAA,EAAQ;AACZ,MAAA,MAAM,MAAA,EAAQ,MAAM,6CAAA,IAAc,CAAK,WAAA,EAAa,KAAA,EAAO,OAAO,CAAA;AAClE,MAAA,MAAM,SAAA,EAA4B,sBAAM,YAAA,6BAAc,YAAA;AAAA,QACrD,IAAA;AAAA,QACA,MAAA;AAAA,QACA,QAAA;AAAA,QACA;AAAA,MACD,IAAA,GAAM,EAAE,IAAA,EAAM,MAAA,EAAQ,QAAA,EAAU,MAAM,CAAA;AACtC,MAAA,OAAA,EAAS,EAAE,GAAG,QAAA,EAAU,QAAA,EAAU,CAAC,EAAE,CAAA;AACrC,MAAA,yBAAA,CAAQ,MAAA,CAAO,IAAA,CAAK,CAAA,mBAAA,EAAsB,QAAQ,CAAA,CAAA;AACA,MAAA;AACnD,IAAA;AAIY,IAAA;AACgC,IAAA;AAGI,IAAA;AAC/C,MAAA;AACuB,QAAA;AACtB,QAAA;AACoB,QAAA;AACoB,QAAA;AAC9B,QAAA;AAEc,UAAA;AAEF,YAAA;AACF,YAAA;AAChB,UAAA;AACJ,QAAA;AACD,MAAA;AACA,IAAA;AAG8C,IAAA;AAC9C,MAAA;AACA,MAAA;AACD,IAAA;AAEkC,IAAA;AAC3B,MAAA;AACP,IAAA;AACD,EAAA;AACD;ADvCyD;AACA;AACA;AACA","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-NYDNXGKB.cjs","sourcesContent":[null,"import { randomUUID } from \"node:crypto\";\nimport { StatusCodes } from \"http-status-codes\";\nimport type {\n\tA2AModule,\n\tMCPModule,\n\tMemoryModule,\n\tModelFetchOptions,\n\tModelModule,\n} from \"@/modules/index.js\";\nimport { AinHttpError } from \"@/types/agent.js\";\nimport {\n\tMessageRole,\n\ttype ThreadMetadata,\n\ttype ThreadObject,\n\ttype ThreadType,\n\ttype TriggeredIntent,\n} from \"@/types/memory.js\";\nimport type { StreamEvent } from \"@/types/stream\";\nimport { loggers } from \"@/utils/logger.js\";\nimport { IntentFulfillStreamService } from \"./intents/fulfill-stream.service\";\nimport { IntentTriggerService } from \"./intents/trigger.service\";\nimport { generateTitle } from \"./utils/query.common\";\n\n/**\n * Service for processing user queries through the agent's AI pipeline.\n *\n * Orchestrates the query processing workflow including intent detection,\n * model inference, tool execution, and response generation. Manages\n * conversation context and coordinates between different modules.\n */\nexport class QueryStreamService {\n\tprivate modelModule: ModelModule;\n\tprivate memoryModule?: MemoryModule;\n\tprivate intentTriggerService: IntentTriggerService;\n\tprivate intentFulfillStreamService: IntentFulfillStreamService;\n\n\tconstructor(\n\t\tmodelModule: ModelModule,\n\t\ta2aModule?: A2AModule,\n\t\tmcpModule?: MCPModule,\n\t\tmemoryModule?: MemoryModule,\n\t) {\n\t\tthis.modelModule = modelModule;\n\t\tthis.memoryModule = memoryModule;\n\t\tthis.intentTriggerService = new IntentTriggerService(\n\t\t\tmodelModule,\n\t\t\tmemoryModule,\n\t\t);\n\t\tthis.intentFulfillStreamService = new IntentFulfillStreamService(\n\t\t\tmodelModule,\n\t\t\ta2aModule,\n\t\t\tmcpModule,\n\t\t\tmemoryModule,\n\t\t);\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\toptions?: ModelFetchOptions;\n\t\t},\n\t\tquery: string,\n\t\tisA2A?: boolean,\n\t): AsyncGenerator<StreamEvent> {\n\t\tconst { type, userId, options } = threadMetadata;\n\t\tconst threadMemory = this.memoryModule?.getThreadMemory();\n\n\t\t// 1. Load or create thread\n\t\tlet threadId = threadMetadata.threadId;\n\t\tlet thread: ThreadObject | undefined;\n\t\tif (threadId) {\n\t\t\tthread = await threadMemory?.getThread(userId, threadId);\n\t\t\tif (!thread && !isA2A) {\n\t\t\t\tthrow new AinHttpError(StatusCodes.NOT_FOUND, \"Thread not found\");\n\t\t\t}\n\t\t}\n\n\t\tthreadId ??= randomUUID();\n\t\tif (!thread) {\n\t\t\tconst title = await generateTitle(this.modelModule, query, options);\n\t\t\tconst metadata: ThreadMetadata = (await threadMemory?.createThread(\n\t\t\t\ttype,\n\t\t\t\tuserId,\n\t\t\t\tthreadId,\n\t\t\t\ttitle,\n\t\t\t)) || { type, userId, threadId, title };\n\t\t\tthread = { ...metadata, messages: [] };\n\t\t\tloggers.intent.info(`Create new thread: ${threadId}`);\n\t\t\tyield { event: \"thread_id\", data: { type, userId, threadId, title } };\n\t\t}\n\n\t\t// 2. intent triggering\n\t\tconst triggeredIntent: Array<TriggeredIntent> =\n\t\t\tawait this.intentTriggerService.intentTriggering(query, thread);\n\t\tloggers.intent.debug(\"Triggered intents\", { triggeredIntent });\n\n\t\t// only add for storage, not for inference\n\t\tawait threadMemory?.addMessagesToThread(userId, threadId, [\n\t\t\t{\n\t\t\t\tmessageId: randomUUID(),\n\t\t\t\trole: MessageRole.USER,\n\t\t\t\ttimestamp: Date.now(),\n\t\t\t\tcontent: { type: \"text\", parts: [query] },\n\t\t\t\tmetadata: {\n\t\t\t\t\tintents: triggeredIntent\n\t\t\t\t\t\t.filter((intent) => !!intent.intent)\n\t\t\t\t\t\t.map((intent) => ({\n\t\t\t\t\t\t\tid: intent.intent?.id,\n\t\t\t\t\t\t\tsubquery: intent.subquery,\n\t\t\t\t\t\t})),\n\t\t\t\t},\n\t\t\t},\n\t\t]);\n\n\t\t// 3. intent fulfillment\n\t\tconst stream = this.intentFulfillStreamService.intentFulfillStream(\n\t\t\ttriggeredIntent,\n\t\t\tthread,\n\t\t);\n\n\t\tfor await (const event of stream) {\n\t\t\tyield event;\n\t\t}\n\t}\n}\n"]}
@@ -0,0 +1,36 @@
1
+ "use strict";Object.defineProperty(exports, "__esModule", {value: true});
2
+
3
+ var _chunkBDQDZXZIcjs = require('./chunk-BDQDZXZI.cjs');
4
+
5
+
6
+ var _chunkNYDNXGKBcjs = require('./chunk-NYDNXGKB.cjs');
7
+
8
+
9
+ var _chunkICFTUZP5cjs = require('./chunk-ICFTUZP5.cjs');
10
+
11
+ // src/routes/a2a.routes.ts
12
+ var _server = require('@a2a-js/sdk/server');
13
+ var _express = require('express');
14
+ var createA2ARouter = (agent) => {
15
+ const router = _express.Router.call(void 0, );
16
+ const taskStore = new (0, _server.InMemoryTaskStore)();
17
+ const queryStreamService = new (0, _chunkNYDNXGKBcjs.QueryStreamService)(
18
+ agent.modelModule,
19
+ agent.a2aModule,
20
+ agent.mcpModule,
21
+ agent.memoryModule
22
+ );
23
+ const a2aService = new (0, _chunkBDQDZXZIcjs.A2AService)(queryStreamService);
24
+ const a2aController = new (0, _chunkICFTUZP5cjs.A2AController)(
25
+ a2aService,
26
+ taskStore,
27
+ agent.generateAgentCard
28
+ );
29
+ router.post("/", a2aController.handleA2ARequest);
30
+ return router;
31
+ };
32
+
33
+
34
+
35
+ exports.createA2ARouter = createA2ARouter;
36
+ //# sourceMappingURL=chunk-TF2SCL3K.cjs.map
@@ -0,0 +1 @@
1
+ {"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-TF2SCL3K.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;AAAA,EACP,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-TF2SCL3K.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);\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"]}