@falai/agent 0.9.0-alpha-2 → 0.9.0

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 (75) hide show
  1. package/dist/cjs/src/core/Agent.d.ts +31 -41
  2. package/dist/cjs/src/core/Agent.d.ts.map +1 -1
  3. package/dist/cjs/src/core/Agent.js +72 -1075
  4. package/dist/cjs/src/core/Agent.js.map +1 -1
  5. package/dist/cjs/src/core/ResponseModal.d.ts +205 -0
  6. package/dist/cjs/src/core/ResponseModal.d.ts.map +1 -0
  7. package/dist/cjs/src/core/ResponseModal.js +1328 -0
  8. package/dist/cjs/src/core/ResponseModal.js.map +1 -0
  9. package/dist/cjs/src/core/ResponsePipeline.js +1 -1
  10. package/dist/cjs/src/core/ResponsePipeline.js.map +1 -1
  11. package/dist/cjs/src/index.d.ts +3 -1
  12. package/dist/cjs/src/index.d.ts.map +1 -1
  13. package/dist/cjs/src/index.js +7 -1
  14. package/dist/cjs/src/index.js.map +1 -1
  15. package/dist/cjs/src/types/agent.d.ts +1 -0
  16. package/dist/cjs/src/types/agent.d.ts.map +1 -1
  17. package/dist/cjs/src/types/ai.d.ts +1 -1
  18. package/dist/cjs/src/types/ai.d.ts.map +1 -1
  19. package/dist/cjs/src/utils/clone.d.ts.map +1 -1
  20. package/dist/cjs/src/utils/clone.js +0 -4
  21. package/dist/cjs/src/utils/clone.js.map +1 -1
  22. package/dist/cjs/src/utils/history.d.ts +30 -1
  23. package/dist/cjs/src/utils/history.d.ts.map +1 -1
  24. package/dist/cjs/src/utils/history.js +169 -23
  25. package/dist/cjs/src/utils/history.js.map +1 -1
  26. package/dist/cjs/src/utils/index.d.ts +1 -1
  27. package/dist/cjs/src/utils/index.d.ts.map +1 -1
  28. package/dist/cjs/src/utils/index.js +5 -1
  29. package/dist/cjs/src/utils/index.js.map +1 -1
  30. package/dist/src/core/Agent.d.ts +31 -41
  31. package/dist/src/core/Agent.d.ts.map +1 -1
  32. package/dist/src/core/Agent.js +73 -1076
  33. package/dist/src/core/Agent.js.map +1 -1
  34. package/dist/src/core/ResponseModal.d.ts +205 -0
  35. package/dist/src/core/ResponseModal.d.ts.map +1 -0
  36. package/dist/src/core/ResponseModal.js +1323 -0
  37. package/dist/src/core/ResponseModal.js.map +1 -0
  38. package/dist/src/core/ResponsePipeline.js +1 -1
  39. package/dist/src/core/ResponsePipeline.js.map +1 -1
  40. package/dist/src/index.d.ts +3 -1
  41. package/dist/src/index.d.ts.map +1 -1
  42. package/dist/src/index.js +2 -1
  43. package/dist/src/index.js.map +1 -1
  44. package/dist/src/types/agent.d.ts +1 -0
  45. package/dist/src/types/agent.d.ts.map +1 -1
  46. package/dist/src/types/ai.d.ts +1 -1
  47. package/dist/src/types/ai.d.ts.map +1 -1
  48. package/dist/src/utils/clone.d.ts.map +1 -1
  49. package/dist/src/utils/clone.js +0 -4
  50. package/dist/src/utils/clone.js.map +1 -1
  51. package/dist/src/utils/history.d.ts +30 -1
  52. package/dist/src/utils/history.d.ts.map +1 -1
  53. package/dist/src/utils/history.js +165 -23
  54. package/dist/src/utils/history.js.map +1 -1
  55. package/dist/src/utils/index.d.ts +1 -1
  56. package/dist/src/utils/index.d.ts.map +1 -1
  57. package/dist/src/utils/index.js +1 -1
  58. package/dist/src/utils/index.js.map +1 -1
  59. package/docs/README.md +2 -1
  60. package/docs/api/README.md +160 -0
  61. package/docs/api/overview.md +66 -1
  62. package/docs/guides/migration/README.md +72 -0
  63. package/docs/guides/migration/response-modal-refactor.md +518 -0
  64. package/examples/advanced-patterns/streaming-responses.ts +169 -96
  65. package/examples/core-concepts/modern-streaming-api.ts +309 -0
  66. package/package.json +1 -1
  67. package/src/core/Agent.ts +95 -1488
  68. package/src/core/ResponseModal.ts +1722 -0
  69. package/src/core/ResponsePipeline.ts +1 -1
  70. package/src/index.ts +11 -0
  71. package/src/types/agent.ts +1 -0
  72. package/src/types/ai.ts +1 -1
  73. package/src/utils/clone.ts +6 -8
  74. package/src/utils/history.ts +190 -27
  75. package/src/utils/index.ts +4 -0
@@ -0,0 +1,1328 @@
1
+ "use strict";
2
+ /**
3
+ * ResponseModal handles all response generation logic for the Agent
4
+ * Provides both streaming and non-streaming response generation with unified logic
5
+ */
6
+ Object.defineProperty(exports, "__esModule", { value: true });
7
+ exports.ResponseModal = exports.ResponseGenerationError = void 0;
8
+ const types_1 = require("../types");
9
+ const Step_1 = require("./Step");
10
+ const ResponseEngine_1 = require("./ResponseEngine");
11
+ const ResponsePipeline_1 = require("./ResponsePipeline");
12
+ const utils_1 = require("../utils");
13
+ const ToolExecutor_1 = require("./ToolExecutor");
14
+ const constants_1 = require("../constants");
15
+ /**
16
+ * Error class for response generation failures
17
+ */
18
+ class ResponseGenerationError extends Error {
19
+ constructor(message, details) {
20
+ super(message);
21
+ this.details = details;
22
+ this.name = 'ResponseGenerationError';
23
+ // Preserve stack trace from original error if available
24
+ if (details?.originalError instanceof Error && details.originalError.stack) {
25
+ this.stack = `${this.stack}\nCaused by: ${details.originalError.stack}`;
26
+ }
27
+ }
28
+ /**
29
+ * Create a ResponseGenerationError from an unknown error
30
+ */
31
+ static fromError(error, phase, params, context) {
32
+ const message = error instanceof Error ? error.message : String(error);
33
+ return new ResponseGenerationError(`Response generation failed in ${phase}: ${message}`, { originalError: error, params, phase, context });
34
+ }
35
+ /**
36
+ * Check if an error is a ResponseGenerationError
37
+ */
38
+ static isResponseGenerationError(error) {
39
+ return error instanceof ResponseGenerationError;
40
+ }
41
+ }
42
+ exports.ResponseGenerationError = ResponseGenerationError;
43
+ /**
44
+ * ResponseModal class that encapsulates all response generation logic
45
+ * Uses unified approach for both streaming and non-streaming responses
46
+ */
47
+ class ResponseModal {
48
+ constructor(agent, options) {
49
+ this.agent = agent;
50
+ this.options = options;
51
+ // Initialize response engine
52
+ this.responseEngine = new ResponseEngine_1.ResponseEngine();
53
+ // Initialize response pipeline with agent dependencies
54
+ this.responsePipeline = new ResponsePipeline_1.ResponsePipeline(this.agent.getAgentOptions(), this.agent.getRoutes(), this.agent.getTools(), this.agent.getRoutingEngine(), this.agent.updateContext.bind(this.agent), this.agent.getUpdateDataMethod(), this.agent.updateCollectedData.bind(this.agent));
55
+ }
56
+ /**
57
+ * Generate a non-streaming response using unified logic
58
+ */
59
+ async respond(params) {
60
+ try {
61
+ // Use unified response preparation and routing
62
+ const responseContext = await this.prepareUnifiedResponseContext(params);
63
+ // Generate response using unified logic
64
+ const result = await this.generateUnifiedResponse(responseContext);
65
+ // Finalize session
66
+ await this.finalizeSession(result.session, responseContext.effectiveContext);
67
+ return result;
68
+ }
69
+ catch (error) {
70
+ throw new ResponseGenerationError(`Failed to generate response: ${error instanceof Error ? error.message : String(error)}`, { originalError: error, params, phase: 'response_generation' });
71
+ }
72
+ }
73
+ /**
74
+ * Generate a streaming response using unified logic
75
+ */
76
+ async *respondStream(params) {
77
+ try {
78
+ // Use unified response preparation and routing
79
+ const responseContext = await this.prepareUnifiedResponseContext(params);
80
+ // Generate streaming response using unified logic
81
+ yield* this.generateUnifiedStreamingResponse(responseContext);
82
+ }
83
+ catch (error) {
84
+ // Stream error to caller
85
+ yield {
86
+ delta: "",
87
+ accumulated: "",
88
+ done: true,
89
+ session: params.session || await this.agent.session.getOrCreate(),
90
+ error: new ResponseGenerationError(`Streaming response failed: ${error instanceof Error ? error.message : String(error)}`, { originalError: error, params, phase: 'streaming' }),
91
+ };
92
+ }
93
+ }
94
+ /**
95
+ * Modern streaming API - simple interface like chat()
96
+ */
97
+ async *stream(message, options) {
98
+ // Determine which history to use
99
+ let history;
100
+ if (options?.history) {
101
+ // Use provided history for this response only
102
+ history = options.history;
103
+ }
104
+ else {
105
+ // Add user message to session history if provided
106
+ if (message) {
107
+ await this.agent.session.addMessage("user", message);
108
+ }
109
+ history = this.agent.session.getHistory();
110
+ }
111
+ // Get or create session
112
+ let session = await this.agent.session.getOrCreate();
113
+ // Merge agent's collected data into session (agent data takes precedence)
114
+ const collectedData = this.agent.getCollectedData();
115
+ if (Object.keys(collectedData).length > 0) {
116
+ session = (0, utils_1.mergeCollected)(session, collectedData);
117
+ // Update the session manager with the merged data
118
+ await this.agent.session.setData(collectedData);
119
+ utils_1.logger.debug("[ResponseModal] Merged agent collected data into stream session:", collectedData);
120
+ }
121
+ // Stream response using existing respondStream method
122
+ let finalMessage = "";
123
+ for await (const chunk of this.respondStream({
124
+ history,
125
+ session,
126
+ contextOverride: options?.contextOverride,
127
+ signal: options?.signal,
128
+ })) {
129
+ // Accumulate the final message for session history
130
+ if (chunk.done) {
131
+ finalMessage = chunk.accumulated;
132
+ }
133
+ yield chunk;
134
+ }
135
+ // Add agent response to session history (only if not using override history)
136
+ if (!options?.history && finalMessage) {
137
+ await this.agent.session.addMessage("assistant", finalMessage);
138
+ }
139
+ }
140
+ /**
141
+ * Modern non-streaming API - equivalent to chat() but more explicit
142
+ */
143
+ async generate(message, options) {
144
+ // Determine which history to use
145
+ let history;
146
+ if (options?.history) {
147
+ // Use provided history for this response only
148
+ history = options.history;
149
+ }
150
+ else {
151
+ // Add user message to session history if provided
152
+ if (message) {
153
+ await this.agent.session.addMessage("user", message);
154
+ }
155
+ history = this.agent.session.getHistory();
156
+ }
157
+ // Get or create session
158
+ let session = await this.agent.session.getOrCreate();
159
+ // Merge agent's collected data into session (agent data takes precedence)
160
+ const collectedData = this.agent.getCollectedData();
161
+ if (Object.keys(collectedData).length > 0) {
162
+ session = (0, utils_1.mergeCollected)(session, collectedData);
163
+ // Update the session manager with the merged data
164
+ await this.agent.session.setData(collectedData);
165
+ utils_1.logger.debug("[ResponseModal] Merged agent collected data into generate session:", collectedData);
166
+ }
167
+ // Generate response using existing respond method
168
+ const result = await this.respond({
169
+ history,
170
+ session,
171
+ contextOverride: options?.contextOverride,
172
+ signal: options?.signal,
173
+ });
174
+ // Add agent response to session history (only if not using override history)
175
+ if (!options?.history) {
176
+ await this.agent.session.addMessage("assistant", result.message);
177
+ }
178
+ // Ensure the result includes the current session
179
+ return {
180
+ ...result,
181
+ session: result.session || this.agent.session.current,
182
+ };
183
+ }
184
+ /**
185
+ * Get the response engine instance
186
+ * @internal
187
+ */
188
+ getResponseEngine() {
189
+ return this.responseEngine;
190
+ }
191
+ /**
192
+ * Get the response pipeline instance
193
+ * @internal
194
+ */
195
+ getResponsePipeline() {
196
+ return this.responsePipeline;
197
+ }
198
+ // UNIFIED RESPONSE LOGIC - Consolidates common logic between streaming and non-streaming
199
+ // ============================================================================
200
+ /**
201
+ * Unified response preparation - handles context setup, session management, and routing
202
+ * This consolidates common logic between streaming and non-streaming responses
203
+ * @private
204
+ */
205
+ async prepareUnifiedResponseContext(params) {
206
+ try {
207
+ const { history: simpleHistory, contextOverride, signal } = params;
208
+ // Validate input parameters
209
+ if (!simpleHistory) {
210
+ throw new ResponseGenerationError('History is required for response generation', { params, phase: 'validation' });
211
+ }
212
+ // Convert HistoryItem[] to Event[] for internal processing
213
+ const historyEvents = (0, utils_1.historyToEvents)(simpleHistory);
214
+ // Keep original HistoryItem[] format for external APIs
215
+ const history = simpleHistory;
216
+ // Use ResponsePipeline for optimized context and session preparation
217
+ // This leverages existing optimizations and avoids code duplication
218
+ let responseContext;
219
+ try {
220
+ // Set current context and session in pipeline for consistency
221
+ this.responsePipeline.setContext(await this.agent.getContext());
222
+ this.responsePipeline.setCurrentSession(this.agent.getCurrentSession());
223
+ responseContext = await this.responsePipeline.prepareResponseContext({
224
+ contextOverride,
225
+ session: params.session ? (0, utils_1.cloneDeep)(params.session) : undefined,
226
+ });
227
+ }
228
+ catch (error) {
229
+ throw ResponseGenerationError.fromError(error, 'pipeline_context_preparation', params);
230
+ }
231
+ const { effectiveContext } = responseContext;
232
+ let session = responseContext.session;
233
+ // Update our stored context if it was modified by beforeRespond hook
234
+ const storedContext = this.responsePipeline.getStoredContext();
235
+ if (storedContext !== undefined) {
236
+ try {
237
+ await this.agent.updateContext(storedContext);
238
+ }
239
+ catch (error) {
240
+ throw ResponseGenerationError.fromError(error, 'context_update_from_pipeline', params, { storedContext });
241
+ }
242
+ }
243
+ // Merge agent's collected data into session (agent data takes precedence)
244
+ const collectedData = this.agent.getCollectedData();
245
+ if (Object.keys(collectedData).length > 0) {
246
+ try {
247
+ session = (0, utils_1.mergeCollected)(session, collectedData);
248
+ utils_1.logger.debug("[ResponseModal] Merged agent collected data into session:", collectedData);
249
+ }
250
+ catch (error) {
251
+ throw ResponseGenerationError.fromError(error, 'data_merging', params, { collectedData });
252
+ }
253
+ }
254
+ // PHASE 1: PREPARE - Execute prepare function if current step has one
255
+ try {
256
+ await this.executeStepPrepare(session, effectiveContext);
257
+ }
258
+ catch (error) {
259
+ throw ResponseGenerationError.fromError(error, 'step_preparation', params, { session, effectiveContext });
260
+ }
261
+ // PHASE 2: ROUTING + STEP SELECTION - Determine which route and step to use
262
+ let routingResult;
263
+ try {
264
+ routingResult = await this.handleUnifiedRoutingAndStepSelection({
265
+ session,
266
+ history: historyEvents,
267
+ context: effectiveContext,
268
+ signal,
269
+ });
270
+ }
271
+ catch (error) {
272
+ throw ResponseGenerationError.fromError(error, 'routing_and_step_selection', params, { session, effectiveContext });
273
+ }
274
+ return {
275
+ effectiveContext,
276
+ session: routingResult.session,
277
+ history,
278
+ selectedRoute: routingResult.selectedRoute,
279
+ selectedStep: routingResult.selectedStep,
280
+ responseDirectives: routingResult.responseDirectives,
281
+ isRouteComplete: routingResult.isRouteComplete,
282
+ };
283
+ }
284
+ catch (error) {
285
+ // Re-throw ResponseGenerationError as-is, wrap others
286
+ if (ResponseGenerationError.isResponseGenerationError(error)) {
287
+ throw error;
288
+ }
289
+ throw ResponseGenerationError.fromError(error, 'preparation', params);
290
+ }
291
+ }
292
+ /**
293
+ * Unified routing and step selection logic using ResponsePipeline for optimization
294
+ * @private
295
+ */
296
+ async handleUnifiedRoutingAndStepSelection(params) {
297
+ try {
298
+ // Use the ResponsePipeline for optimized routing and step selection
299
+ // This avoids duplicate logic and leverages existing optimizations
300
+ // ResponsePipeline expects Event[] for history
301
+ const routingResult = await this.responsePipeline.handleRoutingAndStepSelection({
302
+ session: params.session,
303
+ history: params.history, // Already Event[]
304
+ context: params.context,
305
+ signal: params.signal,
306
+ });
307
+ // Determine next step using pipeline method for consistency
308
+ const stepResult = this.responsePipeline.determineNextStep({
309
+ selectedRoute: routingResult.selectedRoute,
310
+ selectedStep: routingResult.selectedStep,
311
+ session: routingResult.session,
312
+ isRouteComplete: routingResult.isRouteComplete,
313
+ });
314
+ return {
315
+ selectedRoute: routingResult.selectedRoute,
316
+ selectedStep: stepResult.nextStep, // Use the determined next step
317
+ responseDirectives: routingResult.responseDirectives,
318
+ session: stepResult.session,
319
+ isRouteComplete: routingResult.isRouteComplete,
320
+ };
321
+ }
322
+ catch (error) {
323
+ throw ResponseGenerationError.fromError(error, 'routing_optimization', params);
324
+ }
325
+ }
326
+ /**
327
+ * Unified response generation for non-streaming responses
328
+ * @private
329
+ */
330
+ async generateUnifiedResponse(responseContext) {
331
+ const { effectiveContext, session: initialSession, history, selectedRoute, selectedStep, responseDirectives, isRouteComplete } = responseContext;
332
+ let session = initialSession;
333
+ // Get last user message (needed for both route and completion handling)
334
+ // Convert HistoryItem[] to Event[] for internal processing
335
+ const historyEvents = (0, utils_1.historyToEvents)(history);
336
+ const lastMessageText = (0, utils_1.getLastMessageFromHistory)(historyEvents);
337
+ let message;
338
+ let toolCalls = undefined;
339
+ if (selectedRoute && !isRouteComplete) {
340
+ // Handle normal route processing
341
+ const result = await this.processRouteResponse({
342
+ selectedRoute,
343
+ selectedStep,
344
+ responseDirectives,
345
+ session,
346
+ history,
347
+ context: effectiveContext,
348
+ lastMessageText,
349
+ historyEvents,
350
+ signal: responseContext.history ? undefined : undefined, // TODO: Fix signal passing
351
+ });
352
+ message = result.message;
353
+ toolCalls = result.toolCalls;
354
+ session = result.session;
355
+ }
356
+ else if (isRouteComplete && selectedRoute) {
357
+ // Handle route completion
358
+ message = await this.handleRouteCompletion({
359
+ selectedRoute,
360
+ session,
361
+ context: effectiveContext,
362
+ lastMessageText,
363
+ historyEvents,
364
+ });
365
+ // Set step to END_ROUTE marker
366
+ session = (0, utils_1.enterStep)(session, constants_1.END_ROUTE_ID, "Route completed");
367
+ utils_1.logger.debug(`[ResponseModal] Route ${selectedRoute.title} completed. Entered END_ROUTE step.`);
368
+ }
369
+ else {
370
+ // Fallback: No routes defined, generate a simple response
371
+ message = await this.generateFallbackResponse({
372
+ history: historyEvents, // Use Event[] for fallback response
373
+ context: effectiveContext,
374
+ session,
375
+ });
376
+ }
377
+ return {
378
+ message,
379
+ session,
380
+ toolCalls,
381
+ isRouteComplete,
382
+ };
383
+ }
384
+ /**
385
+ * Unified streaming response generation
386
+ * @private
387
+ */
388
+ async *generateUnifiedStreamingResponse(responseContext) {
389
+ const { effectiveContext, session: initialSession, history, selectedRoute, selectedStep, responseDirectives, isRouteComplete } = responseContext;
390
+ const session = initialSession;
391
+ // Get last user message (needed for both route and completion handling)
392
+ // Convert HistoryItem[] to Event[] for internal processing
393
+ const historyEvents = (0, utils_1.historyToEvents)(history);
394
+ const lastMessageText = (0, utils_1.getLastMessageFromHistory)(historyEvents);
395
+ if (selectedRoute && !isRouteComplete) {
396
+ // Handle normal route processing with streaming
397
+ yield* this.processRouteStreamingResponse({
398
+ selectedRoute,
399
+ selectedStep,
400
+ responseDirectives,
401
+ session,
402
+ history,
403
+ context: effectiveContext,
404
+ lastMessageText,
405
+ historyEvents,
406
+ });
407
+ }
408
+ else if (isRouteComplete && selectedRoute) {
409
+ // Handle route completion streaming
410
+ yield* this.streamRouteCompletion({
411
+ selectedRoute,
412
+ session,
413
+ context: effectiveContext,
414
+ lastMessageText,
415
+ historyEvents,
416
+ });
417
+ }
418
+ else {
419
+ // Fallback: No routes defined, stream a simple response
420
+ yield* this.streamFallbackResponse({
421
+ history: historyEvents, // Use Event[] for fallback response
422
+ context: effectiveContext,
423
+ session,
424
+ });
425
+ }
426
+ }
427
+ /**
428
+ * Execute prepare function for current step if available
429
+ * @private
430
+ */
431
+ async executeStepPrepare(session, context) {
432
+ if (session.currentRoute && session.currentStep) {
433
+ const currentRoute = this.agent.getRoutes().find((r) => r.id === session.currentRoute?.id);
434
+ if (currentRoute) {
435
+ const currentStep = currentRoute.getStep(session.currentStep.id);
436
+ if (currentStep?.prepare) {
437
+ utils_1.logger.debug(`[ResponseModal] Executing prepare for step: ${currentStep.id}`);
438
+ await this.executePrepareFinalize(currentStep.prepare, context, session.data, currentRoute, currentStep);
439
+ }
440
+ }
441
+ }
442
+ }
443
+ /**
444
+ * Execute finalize function for current step if available
445
+ * @private
446
+ */
447
+ async executeStepFinalize(session, context) {
448
+ if (session.currentRoute && session.currentStep) {
449
+ const currentRoute = this.agent.getRoutes().find((r) => r.id === session.currentRoute?.id);
450
+ if (currentRoute) {
451
+ const currentStep = currentRoute.getStep(session.currentStep.id);
452
+ if (currentStep?.finalize) {
453
+ utils_1.logger.debug(`[ResponseModal] Executing finalize for step: ${currentStep.id}`);
454
+ await this.executePrepareFinalize(currentStep.finalize, context, session.data, currentRoute, currentStep);
455
+ }
456
+ }
457
+ }
458
+ }
459
+ /**
460
+ * Process route response with unified tool execution and data collection
461
+ * @private
462
+ */
463
+ async processRouteResponse(params) {
464
+ const { selectedRoute, selectedStep, responseDirectives, history, context, lastMessageText, historyEvents, signal } = params;
465
+ let session = params.session;
466
+ // Determine next step
467
+ let nextStep;
468
+ if (selectedStep) {
469
+ nextStep = selectedStep;
470
+ }
471
+ else {
472
+ // New route or no step selected - get initial step or first valid step
473
+ const routingEngine = this.agent.getRoutingEngine();
474
+ const candidates = routingEngine.getCandidateSteps(selectedRoute, undefined, session.data || {});
475
+ if (candidates.length > 0) {
476
+ nextStep = candidates[0].step;
477
+ utils_1.logger.debug(`[ResponseModal] Using first valid step: ${nextStep.id} for new route`);
478
+ }
479
+ else {
480
+ // Fallback to initial step even if it should be skipped
481
+ nextStep = selectedRoute.initialStep;
482
+ utils_1.logger.warn(`[ResponseModal] No valid steps found, using initial step: ${nextStep.id}`);
483
+ }
484
+ }
485
+ // Update session with next step
486
+ session = (0, utils_1.enterStep)(session, nextStep.id, nextStep.description);
487
+ utils_1.logger.debug(`[ResponseModal] Entered step: ${nextStep.id}`);
488
+ // Build response schema for this route (with collect fields from step)
489
+ const responseSchema = this.responseEngine.responseSchemaForRoute(selectedRoute, nextStep, this.agent.getSchema());
490
+ // Build response prompt
491
+ const responsePrompt = await this.responseEngine.buildResponsePrompt({
492
+ route: selectedRoute,
493
+ currentStep: nextStep,
494
+ rules: selectedRoute.getRules(),
495
+ prohibitions: selectedRoute.getProhibitions(),
496
+ directives: responseDirectives,
497
+ history: historyEvents, // Use Event[] for buildResponsePrompt
498
+ lastMessage: lastMessageText, // Use string for buildResponsePrompt
499
+ agentOptions: this.agent.getAgentOptions(),
500
+ combinedGuidelines: [...this.agent.getGuidelines(), ...selectedRoute.getGuidelines()],
501
+ combinedTerms: this.mergeTerms(this.agent.getTerms(), selectedRoute.getTerms()),
502
+ context,
503
+ session,
504
+ agentSchema: this.agent.getSchema(),
505
+ });
506
+ // Collect available tools for AI
507
+ const availableTools = this.collectAvailableTools(selectedRoute, nextStep);
508
+ // Generate message using AI provider
509
+ const agentOptions = this.agent.getAgentOptions();
510
+ const result = await agentOptions.provider.generateMessage({
511
+ prompt: responsePrompt,
512
+ history: historyEvents, // Use Event[] for AI provider
513
+ context,
514
+ tools: availableTools,
515
+ signal,
516
+ parameters: responseSchema ? { jsonSchema: responseSchema, schemaName: "response_output" } : undefined,
517
+ });
518
+ let message = result.structured?.message || result.message;
519
+ let toolCalls = result.structured?.toolCalls;
520
+ // Execute tools with unified loop handling
521
+ const toolResult = await this.executeUnifiedToolLoop({
522
+ toolCalls,
523
+ context,
524
+ session,
525
+ history,
526
+ selectedRoute,
527
+ responsePrompt,
528
+ availableTools,
529
+ responseSchema,
530
+ signal,
531
+ });
532
+ session = toolResult.session;
533
+ toolCalls = toolResult.finalToolCalls;
534
+ if (toolResult.finalMessage) {
535
+ message = toolResult.finalMessage;
536
+ }
537
+ // Collect data from response
538
+ session = await this.collectDataFromResponse({ result, selectedRoute, nextStep, session });
539
+ return { message, toolCalls, session };
540
+ }
541
+ /**
542
+ * Process route streaming response with unified tool execution and data collection
543
+ * @private
544
+ */
545
+ async *processRouteStreamingResponse(params) {
546
+ const { selectedRoute, selectedStep, responseDirectives, history, context, lastMessageText, historyEvents, signal } = params;
547
+ let session = params.session;
548
+ // Determine next step (same logic as non-streaming)
549
+ let nextStep;
550
+ if (selectedStep) {
551
+ nextStep = selectedStep;
552
+ }
553
+ else {
554
+ const routingEngine = this.agent.getRoutingEngine();
555
+ const candidates = routingEngine.getCandidateSteps(selectedRoute, undefined, session.data || {});
556
+ if (candidates.length > 0) {
557
+ nextStep = candidates[0].step;
558
+ utils_1.logger.debug(`[ResponseModal] Using first valid step: ${nextStep.id} for new route`);
559
+ }
560
+ else {
561
+ nextStep = selectedRoute.initialStep;
562
+ utils_1.logger.warn(`[ResponseModal] No valid steps found, using initial step: ${nextStep.id}`);
563
+ }
564
+ }
565
+ // Update session with next step
566
+ session = (0, utils_1.enterStep)(session, nextStep.id, nextStep.description);
567
+ utils_1.logger.debug(`[ResponseModal] Entered step: ${nextStep.id}`);
568
+ // Build response schema and prompt (same as non-streaming)
569
+ const responseSchema = this.responseEngine.responseSchemaForRoute(selectedRoute, nextStep, this.agent.getSchema());
570
+ const responsePrompt = await this.responseEngine.buildResponsePrompt({
571
+ route: selectedRoute,
572
+ currentStep: nextStep,
573
+ rules: selectedRoute.getRules(),
574
+ prohibitions: selectedRoute.getProhibitions(),
575
+ directives: responseDirectives,
576
+ history: historyEvents, // Use Event[] for buildResponsePrompt
577
+ lastMessage: lastMessageText, // Use string for buildResponsePrompt
578
+ agentOptions: this.agent.getAgentOptions(),
579
+ combinedGuidelines: [...this.agent.getGuidelines(), ...selectedRoute.getGuidelines()],
580
+ combinedTerms: this.mergeTerms(this.agent.getTerms(), selectedRoute.getTerms()),
581
+ context,
582
+ session,
583
+ agentSchema: this.agent.getSchema(),
584
+ });
585
+ // Collect available tools for AI
586
+ const availableTools = this.collectAvailableTools(selectedRoute, nextStep);
587
+ // Generate message stream using AI provider
588
+ const agentOptions = this.agent.getAgentOptions();
589
+ const stream = agentOptions.provider.generateMessageStream({
590
+ prompt: responsePrompt,
591
+ history: historyEvents, // Use Event[] for AI provider
592
+ context,
593
+ tools: availableTools,
594
+ signal,
595
+ parameters: { jsonSchema: responseSchema, schemaName: "response_stream_output" },
596
+ });
597
+ // Stream chunks with unified tool handling
598
+ for await (const chunk of stream) {
599
+ let toolCalls = undefined;
600
+ // Extract tool calls from AI response on final chunk
601
+ if (chunk.done && chunk.structured?.toolCalls) {
602
+ toolCalls = chunk.structured.toolCalls;
603
+ // Execute tools with unified loop handling
604
+ const toolResult = await this.executeUnifiedToolLoop({
605
+ toolCalls,
606
+ context,
607
+ session,
608
+ history,
609
+ selectedRoute,
610
+ responsePrompt,
611
+ availableTools,
612
+ responseSchema,
613
+ signal,
614
+ });
615
+ session = toolResult.session;
616
+ toolCalls = toolResult.finalToolCalls;
617
+ }
618
+ // Extract collected data on final chunk
619
+ if (chunk.done && chunk.structured && nextStep.collect) {
620
+ session = await this.collectDataFromResponse({
621
+ result: { structured: chunk.structured },
622
+ selectedRoute,
623
+ nextStep,
624
+ session,
625
+ });
626
+ }
627
+ // Handle session finalization on final chunk
628
+ if (chunk.done) {
629
+ await this.finalizeSession(session, context);
630
+ }
631
+ yield {
632
+ delta: chunk.delta,
633
+ accumulated: chunk.accumulated,
634
+ done: chunk.done,
635
+ session,
636
+ toolCalls,
637
+ isRouteComplete: false,
638
+ metadata: chunk.metadata,
639
+ structured: chunk.structured,
640
+ };
641
+ }
642
+ }
643
+ /**
644
+ * Unified tool execution logic with loop handling
645
+ * Consolidates the complex tool execution logic from both streaming and non-streaming responses
646
+ * @private
647
+ */
648
+ async executeUnifiedToolLoop(params) {
649
+ try {
650
+ const { context, history, selectedRoute, responsePrompt, availableTools, responseSchema, signal } = params;
651
+ let { toolCalls, session } = params;
652
+ // Convert HistoryItem[] to Event[] for internal processing
653
+ const historyEvents = (0, utils_1.historyToEvents)(history);
654
+ // Execute initial dynamic tool calls
655
+ if (toolCalls && toolCalls.length > 0) {
656
+ utils_1.logger.debug(`[ResponseModal] Executing ${toolCalls.length} dynamic tool calls`);
657
+ for (const toolCall of toolCalls) {
658
+ const tool = this.findAvailableTool(toolCall.toolName, selectedRoute);
659
+ if (!tool) {
660
+ utils_1.logger.warn(`[ResponseModal] Tool not found: ${toolCall.toolName}`);
661
+ continue;
662
+ }
663
+ try {
664
+ const toolExecutor = new ToolExecutor_1.ToolExecutor();
665
+ const toolResult = await toolExecutor.executeTool({
666
+ tool: tool,
667
+ context,
668
+ updateContext: this.agent.updateContext.bind(this.agent),
669
+ updateData: this.agent.updateCollectedData.bind(this.agent),
670
+ history: historyEvents, // Use Event[] for tool execution
671
+ data: session.data,
672
+ toolArguments: toolCall.arguments,
673
+ });
674
+ // Check if tool execution was successful
675
+ if (!toolResult.success) {
676
+ utils_1.logger.error(`[ResponseModal] Tool execution failed: ${toolCall.toolName} - ${toolResult.error}`);
677
+ // Continue with other tools rather than failing completely
678
+ continue;
679
+ }
680
+ // Update context with tool results
681
+ if (toolResult.contextUpdate) {
682
+ try {
683
+ await this.agent.updateContext(toolResult.contextUpdate);
684
+ }
685
+ catch (error) {
686
+ utils_1.logger.error(`[ResponseModal] Failed to update context from tool ${toolCall.toolName}:`, error);
687
+ // Continue execution but log the error
688
+ }
689
+ }
690
+ // Update collected data with tool results
691
+ if (toolResult.dataUpdate) {
692
+ try {
693
+ const updateDataMethod = this.agent.getUpdateDataMethod();
694
+ session = await updateDataMethod(session, toolResult.dataUpdate);
695
+ utils_1.logger.debug(`[ResponseModal] Tool updated collected data:`, toolResult.dataUpdate);
696
+ }
697
+ catch (error) {
698
+ utils_1.logger.error(`[ResponseModal] Failed to update data from tool ${toolCall.toolName}:`, error);
699
+ // Continue execution but log the error
700
+ }
701
+ }
702
+ utils_1.logger.debug(`[ResponseModal] Executed dynamic tool: ${toolResult.toolName} (success: ${toolResult.success})`);
703
+ }
704
+ catch (error) {
705
+ utils_1.logger.error(`[ResponseModal] Tool execution error for ${toolCall.toolName}:`, error);
706
+ // Continue with other tools rather than failing the entire response
707
+ continue;
708
+ }
709
+ }
710
+ }
711
+ // TOOL LOOP: Allow AI to make follow-up tool calls after initial tool execution
712
+ const MAX_TOOL_LOOPS = this.options?.maxToolLoops || 5;
713
+ let toolLoopCount = 0;
714
+ let hasToolCalls = toolCalls && toolCalls.length > 0;
715
+ let finalMessage;
716
+ while (hasToolCalls && toolLoopCount < MAX_TOOL_LOOPS) {
717
+ toolLoopCount++;
718
+ utils_1.logger.debug(`[ResponseModal] Starting tool loop ${toolLoopCount}/${MAX_TOOL_LOOPS}`);
719
+ // Create tool result events with proper Event format structure
720
+ const toolResultEvents = [];
721
+ for (const toolCall of toolCalls || []) {
722
+ const tool = this.findAvailableTool(toolCall.toolName, selectedRoute);
723
+ if (tool) {
724
+ // Create proper Event format for tool results
725
+ const toolResultEvent = {
726
+ kind: types_1.EventKind.TOOL,
727
+ source: types_1.MessageRole.AGENT,
728
+ timestamp: new Date().toISOString(),
729
+ data: {
730
+ tool_calls: [
731
+ {
732
+ tool_id: toolCall.toolName,
733
+ arguments: toolCall.arguments,
734
+ result: {
735
+ data: "Tool executed successfully",
736
+ },
737
+ },
738
+ ],
739
+ },
740
+ };
741
+ toolResultEvents.push(toolResultEvent);
742
+ }
743
+ }
744
+ // Create updated history with tool results (combine Event arrays)
745
+ const updatedHistoryEvents = [...historyEvents, ...toolResultEvents];
746
+ // Make follow-up AI call to see if more tools are needed
747
+ const agentOptions = this.agent.getAgentOptions();
748
+ const followUpResult = await agentOptions.provider.generateMessage({
749
+ prompt: responsePrompt,
750
+ history: updatedHistoryEvents, // Use Event[] for AI provider
751
+ context,
752
+ tools: availableTools,
753
+ parameters: responseSchema ? {
754
+ jsonSchema: responseSchema,
755
+ schemaName: "tool_followup",
756
+ } : undefined,
757
+ signal,
758
+ });
759
+ // Check if follow-up call has more tool calls
760
+ const followUpToolCalls = followUpResult.structured?.toolCalls;
761
+ hasToolCalls = followUpToolCalls && followUpToolCalls.length > 0;
762
+ if (hasToolCalls) {
763
+ utils_1.logger.debug(`[ResponseModal] Follow-up call produced ${followUpToolCalls.length} additional tool calls`);
764
+ // Execute the follow-up tool calls
765
+ for (const toolCall of followUpToolCalls) {
766
+ const tool = this.findAvailableTool(toolCall.toolName, selectedRoute);
767
+ if (!tool) {
768
+ utils_1.logger.warn(`[ResponseModal] Tool not found in follow-up: ${toolCall.toolName}`);
769
+ continue;
770
+ }
771
+ try {
772
+ const toolExecutor = new ToolExecutor_1.ToolExecutor();
773
+ const toolResult = await toolExecutor.executeTool({
774
+ tool: tool,
775
+ context,
776
+ updateContext: this.agent.updateContext.bind(this.agent),
777
+ updateData: this.agent.updateCollectedData.bind(this.agent),
778
+ history: updatedHistoryEvents, // Use Event[] for tool execution
779
+ data: session.data,
780
+ toolArguments: toolCall.arguments,
781
+ });
782
+ // Check if tool execution was successful
783
+ if (!toolResult.success) {
784
+ utils_1.logger.error(`[ResponseModal] Follow-up tool execution failed: ${toolCall.toolName} - ${toolResult.error}`);
785
+ continue;
786
+ }
787
+ // Update context with follow-up tool results
788
+ if (toolResult.contextUpdate) {
789
+ try {
790
+ await this.agent.updateContext(toolResult.contextUpdate);
791
+ }
792
+ catch (error) {
793
+ utils_1.logger.error(`[ResponseModal] Failed to update context from follow-up tool ${toolCall.toolName}:`, error);
794
+ }
795
+ }
796
+ if (toolResult.dataUpdate) {
797
+ try {
798
+ const updateDataMethod = this.agent.getUpdateDataMethod();
799
+ session = await updateDataMethod(session, toolResult.dataUpdate);
800
+ utils_1.logger.debug(`[ResponseModal] Follow-up tool updated collected data:`, toolResult.dataUpdate);
801
+ }
802
+ catch (error) {
803
+ utils_1.logger.error(`[ResponseModal] Failed to update data from follow-up tool ${toolCall.toolName}:`, error);
804
+ }
805
+ }
806
+ utils_1.logger.debug(`[ResponseModal] Executed follow-up tool: ${toolResult.toolName} (success: ${toolResult.success})`);
807
+ }
808
+ catch (error) {
809
+ utils_1.logger.error(`[ResponseModal] Follow-up tool execution error for ${toolCall.toolName}:`, error);
810
+ continue;
811
+ }
812
+ }
813
+ // Update toolCalls for next iteration or final response
814
+ toolCalls = followUpToolCalls;
815
+ }
816
+ else {
817
+ utils_1.logger.debug(`[ResponseModal] Tool loop completed after ${toolLoopCount} iterations`);
818
+ // Update final message and toolCalls from follow-up result if no more tools
819
+ finalMessage = followUpResult.structured?.message || followUpResult.message;
820
+ toolCalls = followUpToolCalls || [];
821
+ break;
822
+ }
823
+ }
824
+ if (toolLoopCount >= MAX_TOOL_LOOPS) {
825
+ utils_1.logger.warn(`[ResponseModal] Tool loop limit reached (${MAX_TOOL_LOOPS}), stopping`);
826
+ }
827
+ return {
828
+ session,
829
+ finalToolCalls: toolCalls,
830
+ finalMessage,
831
+ };
832
+ }
833
+ catch (error) {
834
+ throw ResponseGenerationError.fromError(error, 'tool_execution', params, {
835
+ toolCallsCount: params.toolCalls?.length || 0,
836
+ availableToolsCount: params.availableTools.length
837
+ });
838
+ }
839
+ } /**
840
+ * Unified data collection from AI response
841
+ * @private
842
+ */
843
+ async collectDataFromResponse(params) {
844
+ try {
845
+ const { result, selectedRoute, nextStep, session } = params;
846
+ let updatedSession = session;
847
+ // Extract collected data from final response (only for route-based interactions)
848
+ if (selectedRoute && result.structured && nextStep?.collect) {
849
+ try {
850
+ const collectedData = {};
851
+ // AgentStructuredResponse extends Record<string, unknown>, so we can safely access properties
852
+ const structuredData = result.structured;
853
+ for (const field of nextStep.collect) {
854
+ const fieldKey = String(field);
855
+ if (fieldKey in structuredData) {
856
+ collectedData[fieldKey] = structuredData[fieldKey];
857
+ }
858
+ }
859
+ // Merge collected data into session using agent-level data validation
860
+ if (Object.keys(collectedData).length > 0) {
861
+ try {
862
+ // Update agent-level collected data with validation
863
+ await this.agent.updateCollectedData(collectedData);
864
+ // Update session with validated data
865
+ const updateDataMethod = this.agent.getUpdateDataMethod();
866
+ updatedSession = await updateDataMethod(updatedSession, collectedData);
867
+ utils_1.logger.debug(`[ResponseModal] Collected data:`, collectedData);
868
+ }
869
+ catch (error) {
870
+ utils_1.logger.error(`[ResponseModal] Failed to update collected data:`, error);
871
+ // Continue without updating data rather than failing completely
872
+ }
873
+ }
874
+ }
875
+ catch (error) {
876
+ utils_1.logger.error(`[ResponseModal] Error during data collection:`, error);
877
+ // Continue without collecting data rather than failing completely
878
+ }
879
+ }
880
+ // Extract any additional data from structured response
881
+ // Since AgentStructuredResponse extends Record<string, unknown>, we can safely check for additional properties
882
+ if (result.structured && "contextUpdate" in result.structured) {
883
+ try {
884
+ const contextUpdate = result.structured.contextUpdate;
885
+ if (contextUpdate) {
886
+ await this.agent.updateContext(contextUpdate);
887
+ }
888
+ }
889
+ catch (error) {
890
+ utils_1.logger.error(`[ResponseModal] Failed to update context from structured response:`, error);
891
+ // Continue without updating context rather than failing completely
892
+ }
893
+ }
894
+ return updatedSession;
895
+ }
896
+ catch (error) {
897
+ utils_1.logger.error(`[ResponseModal] Error in collectDataFromResponse:`, error);
898
+ // Return original session if data collection fails completely
899
+ return params.session;
900
+ }
901
+ }
902
+ /**
903
+ * Handle route completion logic
904
+ * @private
905
+ */
906
+ async handleRouteCompletion(params) {
907
+ const { selectedRoute, session, context, lastMessageText, historyEvents, signal } = params;
908
+ // Get endStep spec from route
909
+ const endStepSpec = selectedRoute.endStepSpec;
910
+ // Create a temporary step for completion message generation using endStep configuration
911
+ const completionStep = new Step_1.Step(selectedRoute.id, {
912
+ description: endStepSpec.description,
913
+ id: endStepSpec.id || constants_1.END_ROUTE_ID,
914
+ collect: endStepSpec.collect,
915
+ requires: endStepSpec.requires,
916
+ prompt: endStepSpec.prompt || "Summarize what was accomplished and confirm completion based on the conversation history and collected data",
917
+ });
918
+ // Build response schema for completion
919
+ const responseSchema = this.responseEngine.responseSchemaForRoute(selectedRoute, completionStep, this.agent.getSchema());
920
+ const templateContext = { context, session, history: historyEvents }; // Use Event[] for template context
921
+ // Build completion response prompt
922
+ const completionPrompt = await this.responseEngine.buildResponsePrompt({
923
+ route: selectedRoute,
924
+ currentStep: completionStep,
925
+ rules: selectedRoute.getRules(),
926
+ prohibitions: selectedRoute.getProhibitions(),
927
+ directives: undefined, // No directives for completion
928
+ history: historyEvents, // Use Event[] for buildResponsePrompt
929
+ lastMessage: lastMessageText, // Use string for buildResponsePrompt
930
+ agentOptions: this.agent.getAgentOptions(),
931
+ combinedGuidelines: [...this.agent.getGuidelines(), ...selectedRoute.getGuidelines()],
932
+ combinedTerms: this.mergeTerms(this.agent.getTerms(), selectedRoute.getTerms()),
933
+ context,
934
+ session,
935
+ agentSchema: this.agent.getSchema(),
936
+ });
937
+ // Generate completion message using AI provider
938
+ const agentOptions = this.agent.getAgentOptions();
939
+ const completionResult = await agentOptions.provider.generateMessage({
940
+ prompt: completionPrompt,
941
+ history: historyEvents, // Use Event[] for AI provider
942
+ context,
943
+ signal,
944
+ parameters: { jsonSchema: responseSchema, schemaName: "completion_message" },
945
+ });
946
+ const message = completionResult.structured?.message || completionResult.message;
947
+ utils_1.logger.debug(`[ResponseModal] Generated completion message for route: ${selectedRoute.title}`);
948
+ // Check for onComplete transition
949
+ const transitionConfig = await selectedRoute.evaluateOnComplete({ data: session.data }, context);
950
+ if (transitionConfig) {
951
+ // Find target route by ID or title
952
+ const targetRoute = this.agent.getRoutes().find((r) => r.id === transitionConfig.nextStep || r.title === transitionConfig.nextStep);
953
+ if (targetRoute) {
954
+ const renderedCondition = await (0, utils_1.render)(transitionConfig.condition, templateContext);
955
+ // Set pending transition in session
956
+ session.pendingTransition = {
957
+ targetRouteId: targetRoute.id,
958
+ condition: renderedCondition,
959
+ reason: "route_complete",
960
+ };
961
+ utils_1.logger.debug(`[ResponseModal] Route ${selectedRoute.title} completed with pending transition to: ${targetRoute.title}`);
962
+ }
963
+ else {
964
+ utils_1.logger.warn(`[ResponseModal] Route ${selectedRoute.title} completed but target route not found: ${transitionConfig.nextStep}`);
965
+ }
966
+ }
967
+ return message;
968
+ }
969
+ /**
970
+ * Stream route completion response
971
+ * @private
972
+ */
973
+ async *streamRouteCompletion(params) {
974
+ const { selectedRoute, context, lastMessageText, historyEvents, signal } = params;
975
+ let session = params.session;
976
+ // Get endStep spec from route
977
+ const endStepSpec = selectedRoute.endStepSpec;
978
+ // Create a temporary step for completion message generation using endStep configuration
979
+ const completionStep = new Step_1.Step(selectedRoute.id, {
980
+ description: endStepSpec.description,
981
+ id: endStepSpec.id || constants_1.END_ROUTE_ID,
982
+ collect: endStepSpec.collect,
983
+ requires: endStepSpec.requires,
984
+ prompt: endStepSpec.prompt || "Summarize what was accomplished and confirm completion based on the conversation history and collected data",
985
+ });
986
+ // Build response schema for completion
987
+ const responseSchema = this.responseEngine.responseSchemaForRoute(selectedRoute, completionStep, this.agent.getSchema());
988
+ const templateContext = { context, session, history: historyEvents }; // Use Event[] for template context
989
+ // Build completion response prompt
990
+ const completionPrompt = await this.responseEngine.buildResponsePrompt({
991
+ route: selectedRoute,
992
+ currentStep: completionStep,
993
+ rules: selectedRoute.getRules(),
994
+ prohibitions: selectedRoute.getProhibitions(),
995
+ directives: undefined, // No directives for completion
996
+ history: historyEvents, // Use Event[] for buildResponsePrompt
997
+ lastMessage: lastMessageText, // Use string for buildResponsePrompt
998
+ agentOptions: this.agent.getAgentOptions(),
999
+ combinedGuidelines: [...this.agent.getGuidelines(), ...selectedRoute.getGuidelines()],
1000
+ combinedTerms: this.mergeTerms(this.agent.getTerms(), selectedRoute.getTerms()),
1001
+ context,
1002
+ session,
1003
+ agentSchema: this.agent.getSchema(),
1004
+ });
1005
+ // Stream completion message using AI provider
1006
+ const agentOptions = this.agent.getAgentOptions();
1007
+ const stream = agentOptions.provider.generateMessageStream({
1008
+ prompt: completionPrompt,
1009
+ history: historyEvents, // Use Event[] for AI provider
1010
+ context,
1011
+ signal,
1012
+ parameters: { jsonSchema: responseSchema, schemaName: "completion_message_stream" },
1013
+ });
1014
+ utils_1.logger.debug(`[ResponseModal] Streaming completion message for route: ${selectedRoute.title}`);
1015
+ // Check for onComplete transition
1016
+ const transitionConfig = await selectedRoute.evaluateOnComplete({ data: session.data }, context);
1017
+ if (transitionConfig) {
1018
+ // Find target route by ID or title
1019
+ const targetRoute = this.agent.getRoutes().find((r) => r.id === transitionConfig.nextStep || r.title === transitionConfig.nextStep);
1020
+ if (targetRoute) {
1021
+ const renderedCondition = await (0, utils_1.render)(transitionConfig.condition, templateContext);
1022
+ // Set pending transition in session
1023
+ session = {
1024
+ ...session,
1025
+ pendingTransition: {
1026
+ targetRouteId: targetRoute.id,
1027
+ condition: renderedCondition,
1028
+ reason: "route_complete",
1029
+ },
1030
+ };
1031
+ utils_1.logger.debug(`[ResponseModal] Route ${selectedRoute.title} completed with pending transition to: ${targetRoute.title}`);
1032
+ }
1033
+ else {
1034
+ utils_1.logger.warn(`[ResponseModal] Route ${selectedRoute.title} completed but target route not found: ${transitionConfig.nextStep}`);
1035
+ }
1036
+ }
1037
+ // Set step to END_ROUTE marker
1038
+ session = (0, utils_1.enterStep)(session, constants_1.END_ROUTE_ID, "Route completed");
1039
+ utils_1.logger.debug(`[ResponseModal] Route ${selectedRoute.title} completed. Entered END_ROUTE step.`);
1040
+ // Stream completion chunks
1041
+ for await (const chunk of stream) {
1042
+ // Update current session if we have one
1043
+ if (chunk.done) {
1044
+ await this.finalizeSession(session, context);
1045
+ }
1046
+ yield {
1047
+ delta: chunk.delta,
1048
+ accumulated: chunk.accumulated,
1049
+ done: chunk.done,
1050
+ session,
1051
+ toolCalls: undefined,
1052
+ isRouteComplete: true,
1053
+ metadata: chunk.metadata,
1054
+ structured: chunk.structured,
1055
+ };
1056
+ }
1057
+ }
1058
+ /**
1059
+ * Generate fallback response when no routes are available
1060
+ * @private
1061
+ */
1062
+ async generateFallbackResponse(params) {
1063
+ const { history, context, session, signal } = params;
1064
+ utils_1.logger.debug(`[ResponseModal] No route selected, generating basic response`);
1065
+ // Build basic response prompt without route context
1066
+ const fallbackPrompt = await this.responseEngine.buildFallbackPrompt({
1067
+ history,
1068
+ agentOptions: this.agent.getAgentOptions(),
1069
+ terms: this.agent.getTerms(),
1070
+ guidelines: this.agent.getGuidelines(),
1071
+ context,
1072
+ session,
1073
+ });
1074
+ const agentOptions = this.agent.getAgentOptions();
1075
+ const result = await agentOptions.provider.generateMessage({
1076
+ prompt: fallbackPrompt,
1077
+ history,
1078
+ context,
1079
+ signal,
1080
+ parameters: {
1081
+ jsonSchema: {
1082
+ type: "object",
1083
+ properties: { message: { type: "string" } },
1084
+ required: ["message"],
1085
+ additionalProperties: false,
1086
+ },
1087
+ schemaName: "fallback_response",
1088
+ },
1089
+ });
1090
+ return result.structured?.message || result.message;
1091
+ }
1092
+ /**
1093
+ * Stream fallback response when no routes are available
1094
+ * @private
1095
+ */
1096
+ async *streamFallbackResponse(params) {
1097
+ const { history, context, session, signal } = params;
1098
+ const fallbackPrompt = await this.responseEngine.buildFallbackPrompt({
1099
+ history,
1100
+ agentOptions: this.agent.getAgentOptions(),
1101
+ terms: this.agent.getTerms(),
1102
+ guidelines: this.agent.getGuidelines(),
1103
+ context,
1104
+ session,
1105
+ });
1106
+ const agentOptions = this.agent.getAgentOptions();
1107
+ const stream = agentOptions.provider.generateMessageStream({
1108
+ prompt: fallbackPrompt,
1109
+ history,
1110
+ context,
1111
+ signal,
1112
+ parameters: {
1113
+ jsonSchema: {
1114
+ type: "object",
1115
+ properties: { message: { type: "string" } },
1116
+ required: ["message"],
1117
+ additionalProperties: false,
1118
+ },
1119
+ schemaName: "fallback_stream_response",
1120
+ },
1121
+ });
1122
+ for await (const chunk of stream) {
1123
+ // Update current session if we have one
1124
+ if (chunk.done) {
1125
+ await this.finalizeSession(session, context);
1126
+ }
1127
+ yield {
1128
+ delta: chunk.delta,
1129
+ accumulated: chunk.accumulated,
1130
+ done: chunk.done,
1131
+ session,
1132
+ toolCalls: undefined,
1133
+ isRouteComplete: false,
1134
+ metadata: chunk.metadata,
1135
+ structured: chunk.structured,
1136
+ };
1137
+ }
1138
+ }
1139
+ /**
1140
+ * Handle session persistence and finalization
1141
+ * @private
1142
+ */
1143
+ async finalizeSession(session, context) {
1144
+ // Auto-save session step to persistence if configured
1145
+ const persistenceManager = this.agent.getPersistenceManager();
1146
+ const agentOptions = this.agent.getAgentOptions();
1147
+ if (persistenceManager &&
1148
+ session.id &&
1149
+ (this.options?.enableAutoSave !== false && agentOptions.persistence?.autoSave !== false)) {
1150
+ await persistenceManager.saveSessionState(session.id, session);
1151
+ utils_1.logger.debug(`[ResponseModal] Auto-saved session step to persistence: ${session.id}`);
1152
+ }
1153
+ // Execute finalize function
1154
+ await this.executeStepFinalize(session, context);
1155
+ // Update current session if we have one
1156
+ const currentSession = this.agent.getCurrentSession();
1157
+ if (currentSession) {
1158
+ this.agent.setCurrentSession(session);
1159
+ }
1160
+ }
1161
+ // ============================================================================
1162
+ // UTILITY METHODS - Helper methods for tool management and other utilities
1163
+ // ============================================================================
1164
+ /**
1165
+ * Find an available tool by name for the given route
1166
+ * Route-level tools take precedence over agent-level tools
1167
+ * @private
1168
+ */
1169
+ findAvailableTool(toolName, route) {
1170
+ // Check route-level tools first (if route provided)
1171
+ if (route) {
1172
+ const routeTool = route
1173
+ .getTools()
1174
+ .find((tool) => tool.id === toolName || tool.name === toolName);
1175
+ if (routeTool)
1176
+ return routeTool;
1177
+ }
1178
+ // Fall back to agent-level tools
1179
+ const agentTools = this.agent.getTools();
1180
+ return agentTools.find((tool) => tool.id === toolName || tool.name === toolName);
1181
+ }
1182
+ /**
1183
+ * Collect all available tools for the given route and step context
1184
+ * @private
1185
+ */
1186
+ collectAvailableTools(route, step) {
1187
+ const availableTools = new Map();
1188
+ // Add agent-level tools
1189
+ this.agent.getTools().forEach((tool) => {
1190
+ availableTools.set(tool.id, tool);
1191
+ });
1192
+ // Add route-level tools (these take precedence)
1193
+ if (route) {
1194
+ route.getTools().forEach((tool) => {
1195
+ availableTools.set(tool.id, tool);
1196
+ });
1197
+ }
1198
+ // Filter by step-level allowed tools if specified
1199
+ if (step?.tools) {
1200
+ const allowedToolIds = new Set();
1201
+ const stepTools = [];
1202
+ for (const toolRef of step.tools) {
1203
+ if (typeof toolRef === "string") {
1204
+ // Reference to registered tool
1205
+ allowedToolIds.add(toolRef);
1206
+ }
1207
+ else {
1208
+ // Inline tool definition
1209
+ if (toolRef.id) {
1210
+ allowedToolIds.add(toolRef.id);
1211
+ stepTools.push(toolRef);
1212
+ }
1213
+ }
1214
+ }
1215
+ // If step specifies tools, only include those
1216
+ if (allowedToolIds.size > 0) {
1217
+ const filteredTools = new Map();
1218
+ for (const toolId of Array.from(allowedToolIds)) {
1219
+ const tool = availableTools.get(toolId);
1220
+ if (tool) {
1221
+ filteredTools.set(toolId, tool);
1222
+ }
1223
+ }
1224
+ // Add inline tools
1225
+ stepTools.forEach((tool) => {
1226
+ if (tool.id) {
1227
+ filteredTools.set(tool.id, tool);
1228
+ }
1229
+ });
1230
+ availableTools.clear();
1231
+ filteredTools.forEach((tool, id) => availableTools.set(id, tool));
1232
+ }
1233
+ }
1234
+ // Convert to the format expected by AI providers
1235
+ return Array.from(availableTools.values()).map((tool) => ({
1236
+ id: tool.id,
1237
+ name: tool.name || tool.id,
1238
+ description: tool.description,
1239
+ parameters: tool.parameters,
1240
+ }));
1241
+ }
1242
+ /**
1243
+ * Execute a prepare or finalize function/tool
1244
+ * @private
1245
+ */
1246
+ async executePrepareFinalize(prepareOrFinalize, context, data, route, step) {
1247
+ if (!prepareOrFinalize)
1248
+ return;
1249
+ if (typeof prepareOrFinalize === "function") {
1250
+ // It's a function - call it directly
1251
+ await prepareOrFinalize(context, data);
1252
+ }
1253
+ else {
1254
+ // It's a tool reference - find and execute the tool
1255
+ let tool;
1256
+ if (typeof prepareOrFinalize === "string") {
1257
+ // Tool ID - find it in available tools
1258
+ const availableTools = new Map();
1259
+ // Add agent-level tools
1260
+ this.agent.getTools().forEach((t) => {
1261
+ availableTools.set(t.id, t);
1262
+ });
1263
+ // Add route-level tools
1264
+ if (route) {
1265
+ route.getTools().forEach((t) => {
1266
+ availableTools.set(t.id, t);
1267
+ });
1268
+ }
1269
+ // Add step-level tools
1270
+ if (step?.tools) {
1271
+ for (const toolRef of step.tools) {
1272
+ if (typeof toolRef === "string") {
1273
+ // Keep as is
1274
+ }
1275
+ else if (typeof toolRef === 'object' && 'id' in toolRef && toolRef.id) {
1276
+ availableTools.set(toolRef.id, toolRef);
1277
+ }
1278
+ }
1279
+ }
1280
+ tool = availableTools.get(prepareOrFinalize);
1281
+ }
1282
+ else {
1283
+ // Tool object - use directly
1284
+ tool = prepareOrFinalize;
1285
+ }
1286
+ if (tool) {
1287
+ const toolExecutor = new ToolExecutor_1.ToolExecutor();
1288
+ const result = await toolExecutor.executeTool({
1289
+ tool,
1290
+ context,
1291
+ updateContext: this.agent.updateContext.bind(this.agent),
1292
+ updateData: this.agent.updateCollectedData.bind(this.agent),
1293
+ history: [], // Empty history for prepare/finalize
1294
+ data,
1295
+ });
1296
+ if (!result.success) {
1297
+ utils_1.logger.error(`[ResponseModal] Tool execution failed in prepare/finalize: ${result.error}`);
1298
+ throw new Error(`Tool execution failed: ${result.error}`);
1299
+ }
1300
+ }
1301
+ else {
1302
+ utils_1.logger.warn(`[ResponseModal] Tool not found for prepare/finalize: ${typeof prepareOrFinalize === "string"
1303
+ ? prepareOrFinalize
1304
+ : "inline tool"}`);
1305
+ }
1306
+ }
1307
+ }
1308
+ /**
1309
+ * Merge terms with route-specific taking precedence on conflicts
1310
+ * @private
1311
+ */
1312
+ mergeTerms(agentTerms, routeTerms) {
1313
+ const merged = new Map();
1314
+ // Add agent terms first
1315
+ agentTerms.forEach((term) => {
1316
+ const name = typeof term.name === "string" ? term.name : term.name.toString();
1317
+ merged.set(name, term);
1318
+ });
1319
+ // Add route terms (these take precedence)
1320
+ routeTerms.forEach((term) => {
1321
+ const name = typeof term.name === "string" ? term.name : term.name.toString();
1322
+ merged.set(name, term);
1323
+ });
1324
+ return Array.from(merged.values());
1325
+ }
1326
+ }
1327
+ exports.ResponseModal = ResponseModal;
1328
+ //# sourceMappingURL=ResponseModal.js.map