@falai/agent 0.9.0-alpha-2 → 0.9.2

package/dist/cjs/src/core/Agent.js

@@ -4,17 +4,13 @@
  */
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.Agent = void 0;
-const history_1 = require("../types/history");
 const utils_1 = require("../utils");
 const Route_1 = require("./Route");
-const Step_1 = require("./Step");
 const PersistenceManager_1 = require("./PersistenceManager");
 const SessionManager_1 = require("./SessionManager");
 const RoutingEngine_1 = require("./RoutingEngine");
-const ResponseEngine_1 = require("./ResponseEngine");
-const ToolExecutor_1 = require("./ToolExecutor");
-const ResponsePipeline_1 = require("./ResponsePipeline");
-const constants_1 = require("../constants");
+const ResponseModal_1 = require("./ResponseModal");
+const ToolManager_1 = require("./ToolManager");
 /**
  * Error thrown when data validation fails
  */
@@ -78,14 +74,14 @@ class Agent {
         }
         // Initialize current session if provided
         this.currentSession = options.session;
-        // Initialize routing and response engines
+        // Initialize routing engine
         this.routingEngine = new RoutingEngine_1.RoutingEngine({
             maxCandidates: 5,
             allowRouteSwitch: true,
             switchThreshold: 70,
         });
-        this.responseEngine = new ResponseEngine_1.ResponseEngine();
-        this.responsePipeline = new ResponsePipeline_1.ResponsePipeline(options, this.routes, this.tools, this.routingEngine, this.updateContext.bind(this), this.updateData.bind(this), this.updateCollectedData.bind(this));
+        // Initialize ResponseModal for handling all response generation
+        this.responseModal = new ResponseModal_1.ResponseModal(this);
         // Initialize persistence if configured
         if (options.persistence) {
             try {
@@ -138,12 +134,21 @@ class Agent {
         if (options.knowledgeBase) {
             this.knowledgeBase = { ...options.knowledgeBase };
         }
-        // Initialize session manager
-        this.session = new SessionManager_1.SessionManager(this.persistenceManager);
+        // Initialize session manager with reference to this agent for bidirectional sync
+        this.session = new SessionManager_1.SessionManager(this.persistenceManager, this);
+        // Initialize tool manager with proper type inference
+        this.tool = new ToolManager_1.ToolManager(this);
         // Store sessionId for later use in getOrCreate calls
         if (options.sessionId) {
+            this.session.setDefaultSessionId(options.sessionId);
             // The session will be loaded on first getOrCreate call
-            this.session.getOrCreate(options.sessionId).catch((err) => {
+            this.session.getOrCreate(options.sessionId).then((session) => {
+                // Sync session data to agent collected data
+                if (session.data && Object.keys(session.data).length > 0) {
+                    this.collectedData = { ...session.data };
+                    utils_1.logger.debug("[Agent] Synced session data to collected data:", this.collectedData);
+                }
+            }).catch((err) => {
                 utils_1.logger.error("Failed to start session", err);
             });
         }
@@ -225,6 +230,8 @@ class Agent {
      * Get the current collected data
      */
     getCollectedData() {
+        // Ensure agent collected data is synced with session
+        this.syncSessionDataToCollectedData();
         return { ...this.collectedData };
     }
     /**
@@ -256,6 +263,12 @@ class Agent {
         if (this.currentSession) {
             this.currentSession = (0, utils_1.mergeCollected)(this.currentSession, this.collectedData);
         }
+        // Also update the session manager's session data (avoid circular call)
+        const sessionManagerSession = this.session.current;
+        if (sessionManagerSession) {
+            sessionManagerSession.data = { ...this.collectedData };
+            sessionManagerSession.metadata.lastUpdatedAt = new Date();
+        }
         utils_1.logger.debug("[Agent] Collected data updated:", updates);
     }
     /**
@@ -302,7 +315,7 @@ class Agent {
                     `Must be valid keys from agent schema. Available fields: ${Object.keys(this.schema.properties).join(', ')}.`);
             }
         }
-        const route = new Route_1.Route(options);
+        const route = new Route_1.Route(options, this);
         this.routes.push(route);
         return route;
     }
@@ -326,17 +339,47 @@ class Agent {
         return this;
     }
     /**
-     * Register a tool at the agent level
+     * Add a tool to the agent using the unified Tool interface
+     * Creates and adds the tool to agent scope in one operation (BREAKING CHANGE: replaces createTool)
+     */
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    addTool(tool) {
+        // Validate tool before adding
+        if (!tool || !tool.id || !tool.handler) {
+            throw new Error('Invalid tool: must have id and handler properties');
+        }
+        // Add directly to agent's tools array, preserving the TResult type
+        this.tools.push(tool);
+        utils_1.logger.debug(`[Agent] Added tool to agent scope: ${tool.id}`);
+        return this;
+    }
+    /**
+     * Register a tool at the agent level (legacy method for backward compatibility)
+     * @deprecated Use addTool() with Tool interface instead
      */
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
     createTool(tool) {
+        // Validate tool before adding
+        if (!tool || !tool.id || !tool.handler) {
+            throw new Error('Invalid tool: must have id and handler properties');
+        }
         this.tools.push(tool);
+        utils_1.logger.debug(`[Agent] Created tool (legacy): ${tool.id}`);
         return this;
     }
     /**
      * Register multiple tools at the agent level
      */
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
     registerTools(tools) {
-        tools.forEach((tool) => this.createTool(tool));
+        tools.forEach((tool) => {
+            // Validate each tool before adding
+            if (!tool || !tool.id || !tool.handler) {
+                throw new Error(`Invalid tool in batch: must have id and handler properties (tool: ${tool?.id || 'unknown'})`);
+            }
+            this.tools.push(tool);
+        });
+        utils_1.logger.debug(`[Agent] Registered ${tools.length} tools`);
         return this;
     }
     /**
@@ -412,916 +455,15 @@ class Agent {
      * Generate a response based on history and context as a stream
      */
     async *respondStream(params) {
-        const { history: simpleHistory, signal } = params;
-        const history = (0, utils_1.normalizeHistory)(simpleHistory);
-        // Prepare context and session using the response pipeline
-        this.responsePipeline.setContext(this.context);
-        this.responsePipeline.setCurrentSession(this.currentSession);
-        let session;
-        const responseContext = await this.responsePipeline.prepareResponseContext({
-            contextOverride: params.contextOverride,
-            session: params.session ? (0, utils_1.cloneDeep)(params.session) : undefined,
-        });
-        const { effectiveContext } = responseContext;
-        session = responseContext.session;
-        // Merge agent's collected data into session (agent data takes precedence)
-        if (Object.keys(this.collectedData).length > 0) {
-            session = (0, utils_1.mergeCollected)(session, this.collectedData);
-            utils_1.logger.debug("[Agent] Merged agent collected data into session:", this.collectedData);
-        }
-        // Update our stored context if it was modified by beforeRespond hook
-        this.context = this.responsePipeline.getStoredContext();
-        // PHASE 1: PREPARE - Execute prepare function if current step has one
-        if (session.currentRoute && session.currentStep) {
-            const currentRoute = this.routes.find((r) => r.id === session.currentRoute?.id);
-            if (currentRoute) {
-                const currentStep = currentRoute.getStep(session.currentStep.id);
-                if (currentStep?.prepare) {
-                    utils_1.logger.debug(`[Agent] Executing prepare for step: ${currentStep.id}`);
-                    await this.executePrepareFinalize(currentStep.prepare, effectiveContext, session.data, currentRoute, currentStep);
-                }
-            }
-        }
-        // PHASE 2: ROUTING + STEP SELECTION - Use response pipeline
-        const routingResult = await this.responsePipeline.handleRoutingAndStepSelection({
-            session,
-            history,
-            context: effectiveContext,
-            signal,
-        });
-        const selectedRoute = routingResult.selectedRoute;
-        const selectedStep = routingResult.selectedStep;
-        const responseDirectives = routingResult.responseDirectives;
-        const isRouteComplete = routingResult.isRouteComplete;
-        session = routingResult.session;
-        // PHASE 3: DETERMINE NEXT STEP - Use pipeline method
-        const stepResult = this.responsePipeline.determineNextStep({
-            selectedRoute,
-            selectedStep,
-            session,
-            isRouteComplete,
-        });
-        const nextStep = stepResult.nextStep;
-        session = stepResult.session;
-        if (selectedRoute && !isRouteComplete) {
-            // PHASE 4: RESPONSE GENERATION - Stream message using selected route and step
-            // Get last user message
-            const lastUserMessage = (0, utils_1.getLastMessageFromHistory)(history);
-            // Build response schema for this route (with collect fields from step)
-            const responseSchema = this.responseEngine.responseSchemaForRoute(selectedRoute, nextStep, this.schema);
-            // Check if selected route and next step are defined
-            if (!selectedRoute || !nextStep) {
-                utils_1.logger.error("[Agent] Selected route or next step is not defined", {
-                    selectedRoute,
-                    nextStep,
-                });
-                throw new Error("Selected route or next step is not defined");
-            }
-            // Build response prompt
-            const responsePrompt = await this.responseEngine.buildResponsePrompt({
-                route: selectedRoute,
-                currentStep: nextStep,
-                rules: selectedRoute.getRules(),
-                prohibitions: selectedRoute.getProhibitions(),
-                directives: responseDirectives,
-                history,
-                lastMessage: lastUserMessage,
-                agentOptions: this.options,
-                // Combine agent and route properties according to the specified logic
-                combinedGuidelines: [
-                    ...this.getGuidelines(),
-                    ...selectedRoute.getGuidelines(),
-                ],
-                combinedTerms: this.mergeTerms(this.getTerms(), selectedRoute.getTerms()),
-                context: effectiveContext,
-                session,
-                agentSchema: this.schema,
-            });
-            // Collect available tools for AI
-            const availableTools = this.collectAvailableTools(selectedRoute, nextStep);
-            // Generate message stream using AI provider
-            const stream = this.options.provider.generateMessageStream({
-                prompt: responsePrompt,
-                history,
-                context: effectiveContext,
-                tools: availableTools,
-                signal,
-                parameters: {
-                    jsonSchema: responseSchema,
-                    schemaName: "response_stream_output",
-                },
-            });
-            // Stream chunks to caller
-            for await (const chunk of stream) {
-                let toolCalls = undefined;
-                // Extract tool calls from AI response on final chunk
-                if (chunk.done && chunk.structured?.toolCalls) {
-                    toolCalls = chunk.structured.toolCalls;
-                    // Execute dynamic tool calls
-                    if (toolCalls.length > 0) {
-                        utils_1.logger.debug(`[Agent] Executing ${toolCalls.length} dynamic tool calls`);
-                        for (const toolCall of toolCalls) {
-                            const tool = this.findAvailableTool(toolCall.toolName, selectedRoute);
-                            if (!tool) {
-                                utils_1.logger.warn(`[Agent] Tool not found: ${toolCall.toolName}`);
-                                continue;
-                            }
-                            const toolExecutor = new ToolExecutor_1.ToolExecutor();
-                            const result = await toolExecutor.executeTool({
-                                tool: tool,
-                                context: effectiveContext,
-                                updateContext: this.updateContext.bind(this),
-                                updateData: this.updateCollectedData.bind(this),
-                                history,
-                                data: session.data,
-                                toolArguments: toolCall.arguments,
-                            });
-                            // Update context with tool results
-                            if (result.contextUpdate) {
-                                await this.updateContext(result.contextUpdate);
-                            }
-                            // Update collected data with tool results
-                            if (result.dataUpdate) {
-                                session = await this.updateData(session, result.dataUpdate);
-                                utils_1.logger.debug(`[Agent] Tool updated collected data:`, result.dataUpdate);
-                            }
-                            utils_1.logger.debug(`[Agent] Executed dynamic tool: ${result.toolName} (success: ${result.success})`);
-                        }
-                    }
-                }
-                // TOOL LOOP: Allow AI to make follow-up tool calls after initial tool execution (streaming)
-                const MAX_TOOL_LOOPS = 5;
-                let toolLoopCount = 0;
-                let hasToolCalls = toolCalls && toolCalls.length > 0;
-                while (hasToolCalls && toolLoopCount < MAX_TOOL_LOOPS) {
-                    toolLoopCount++;
-                    utils_1.logger.debug(`[Agent] Starting streaming tool loop ${toolLoopCount}/${MAX_TOOL_LOOPS}`);
-                    // Add tool execution results to history so AI knows what happened
-                    const toolResultsEvents = [];
-                    for (const toolCall of toolCalls || []) {
-                        const tool = this.findAvailableTool(toolCall.toolName, selectedRoute);
-                        if (tool) {
-                            toolResultsEvents.push({
-                                kind: history_1.EventKind.TOOL,
-                                source: history_1.MessageRole.AGENT,
-                                timestamp: new Date().toISOString(),
-                                data: {
-                                    tool_calls: [
-                                        {
-                                            tool_id: toolCall.toolName,
-                                            arguments: toolCall.arguments,
-                                            result: {
-                                                data: "Tool executed successfully",
-                                            },
-                                        },
-                                    ],
-                                },
-                            });
-                        }
-                    }
-                    // Create updated history with tool results
-                    const updatedHistory = [...history, ...toolResultsEvents];
-                    // Make follow-up streaming AI call to see if more tools are needed
-                    const followUpStream = this.options.provider.generateMessageStream({
-                        prompt: responsePrompt,
-                        history: updatedHistory,
-                        context: effectiveContext,
-                        tools: availableTools,
-                        parameters: {
-                            jsonSchema: responseSchema,
-                            schemaName: "tool_followup",
-                        },
-                        signal,
-                    });
-                    let followUpToolCalls;
-                    for await (const followUpChunk of followUpStream) {
-                        // Extract tool calls from follow-up stream
-                        if (followUpChunk.done && followUpChunk.structured?.toolCalls) {
-                            followUpToolCalls = followUpChunk.structured.toolCalls;
-                        }
-                    }
-                    hasToolCalls = followUpToolCalls && followUpToolCalls.length > 0;
-                    if (hasToolCalls) {
-                        utils_1.logger.debug(`[Agent] Follow-up streaming call produced ${followUpToolCalls.length} additional tool calls`);
-                        // Execute the follow-up tool calls
-                        for (const toolCall of followUpToolCalls) {
-                            const tool = this.findAvailableTool(toolCall.toolName, selectedRoute);
-                            if (!tool) {
-                                utils_1.logger.warn(`[Agent] Tool not found in streaming follow-up: ${toolCall.toolName}`);
-                                continue;
-                            }
-                            const toolExecutor = new ToolExecutor_1.ToolExecutor();
-                            const result = await toolExecutor.executeTool({
-                                tool: tool,
-                                context: effectiveContext,
-                                updateContext: this.updateContext.bind(this),
-                                updateData: this.updateCollectedData.bind(this),
-                                history: updatedHistory,
-                                data: session.data,
-                                toolArguments: toolCall.arguments,
-                            });
-                            // Update context with follow-up tool results
-                            if (result.contextUpdate) {
-                                await this.updateContext(result.contextUpdate);
-                            }
-                            if (result.dataUpdate) {
-                                session = await this.updateData(session, result.dataUpdate);
-                                utils_1.logger.debug(`[Agent] Streaming follow-up tool updated collected data:`, result.dataUpdate);
-                            }
-                            utils_1.logger.debug(`[Agent] Executed streaming follow-up tool: ${result.toolName} (success: ${result.success})`);
-                        }
-                        // Update toolCalls for next iteration
-                        toolCalls = followUpToolCalls;
-                    }
-                    else {
-                        utils_1.logger.debug(`[Agent] Streaming tool loop completed after ${toolLoopCount} iterations`);
-                        // Update toolCalls for final response
-                        toolCalls = followUpToolCalls || [];
-                        break;
-                    }
-                }
-                if (toolLoopCount >= MAX_TOOL_LOOPS) {
-                    utils_1.logger.warn(`[Agent] Streaming tool loop limit reached (${MAX_TOOL_LOOPS}), stopping`);
-                }
-                // Extract collected data on final chunk
-                if (chunk.done && chunk.structured && nextStep.collect) {
-                    const collectedData = {};
-                    // The structured response includes both base fields and collected extraction fields
-                    const structuredData = chunk.structured;
-                    for (const field of nextStep.collect) {
-                        const fieldKey = String(field);
-                        if (fieldKey in structuredData) {
-                            collectedData[fieldKey] = structuredData[fieldKey];
-                        }
-                    }
-                    // Merge collected data into session using agent-level data validation
-                    if (Object.keys(collectedData).length > 0) {
-                        // Update agent-level collected data with validation
-                        await this.updateCollectedData(collectedData);
-                        // Update session with validated data
-                        session = await this.updateData(session, collectedData);
-                        utils_1.logger.debug(`[Agent] Collected data:`, collectedData);
-                    }
-                }
-                // Extract any additional data from structured response on final chunk
-                if (chunk.done &&
-                    chunk.structured &&
-                    typeof chunk.structured === "object" &&
-                    "contextUpdate" in chunk.structured) {
-                    await this.updateContext(chunk.structured
-                        .contextUpdate);
-                }
-                // Auto-save session step on final chunk
-                if (chunk.done &&
-                    this.persistenceManager &&
-                    session.id &&
-                    this.options.persistence?.autoSave !== false) {
-                    await this.persistenceManager.saveSessionState(session.id, session);
-                    utils_1.logger.debug(`[Agent] Auto-saved session step to persistence: ${session.id}`);
-                }
-                // Execute finalize function on final chunk
-                if (chunk.done && session.currentRoute && session.currentStep) {
-                    const currentRoute = this.routes.find((r) => r.id === session.currentRoute?.id);
-                    if (currentRoute) {
-                        const currentStep = currentRoute.getStep(session.currentStep.id);
-                        if (currentStep?.finalize) {
-                            utils_1.logger.debug(`[Agent] Executing finalize for step: ${currentStep.id}`);
-                            await this.executePrepareFinalize(currentStep.finalize, effectiveContext, session.data, currentRoute, currentStep);
-                        }
-                    }
-                }
-                // Update current session if we have one
-                if (chunk.done && this.currentSession) {
-                    this.currentSession = session;
-                }
-                yield {
-                    delta: chunk.delta,
-                    accumulated: chunk.accumulated,
-                    done: chunk.done,
-                    session, // Return updated session
-                    toolCalls,
-                    isRouteComplete,
-                    metadata: chunk.metadata,
-                    structured: chunk.structured,
-                };
-            }
-        }
-        else if (isRouteComplete && selectedRoute) {
-            // Route is complete - generate completion message then check for onComplete transition
-            const lastUserMessage = (0, utils_1.getLastMessageFromHistory)(history);
-            // Get endStep spec from route
-            const endStepSpec = selectedRoute.endStepSpec;
-            // Create a temporary step for completion message generation using endStep configuration
-            const completionStep = new Step_1.Step(selectedRoute.id, {
-                description: endStepSpec.description,
-                id: endStepSpec.id || constants_1.END_ROUTE_ID,
-                collect: endStepSpec.collect,
-                requires: endStepSpec.requires,
-                prompt: endStepSpec.prompt ||
-                    "Summarize what was accomplished and confirm completion based on the conversation history and collected data",
-            });
-            // Build response schema for completion
-            const responseSchema = this.responseEngine.responseSchemaForRoute(selectedRoute, completionStep, this.schema);
-            const templateContext = {
-                context: effectiveContext,
-                session,
-                history,
-            };
-            // Build completion response prompt
-            const completionPrompt = await this.responseEngine.buildResponsePrompt({
-                route: selectedRoute,
-                currentStep: completionStep,
-                rules: selectedRoute.getRules(),
-                prohibitions: selectedRoute.getProhibitions(),
-                directives: undefined, // No directives for completion
-                history,
-                lastMessage: lastUserMessage,
-                agentOptions: this.options,
-                // Combine agent and route properties according to the specified logic
-                combinedGuidelines: [
-                    ...this.getGuidelines(),
-                    ...selectedRoute.getGuidelines(),
-                ],
-                combinedTerms: this.mergeTerms(this.getTerms(), selectedRoute.getTerms()),
-                context: effectiveContext,
-                session,
-                agentSchema: this.schema,
-            });
-            // Stream completion message using AI provider
-            const stream = this.options.provider.generateMessageStream({
-                prompt: completionPrompt,
-                history,
-                context: effectiveContext,
-                signal,
-                parameters: {
-                    jsonSchema: responseSchema,
-                    schemaName: "completion_message_stream",
-                },
-            });
-            utils_1.logger.debug(`[Agent] Streaming completion message for route: ${selectedRoute.title}`);
-            // Check for onComplete transition
-            const transitionConfig = await selectedRoute.evaluateOnComplete({ data: session.data }, effectiveContext);
-            if (transitionConfig) {
-                // Find target route by ID or title
-                const targetRoute = this.routes.find((r) => r.id === transitionConfig.nextStep ||
-                    r.title === transitionConfig.nextStep);
-                if (targetRoute) {
-                    const renderedCondition = await (0, utils_1.render)(transitionConfig.condition, templateContext);
-                    // Set pending transition in session
-                    session = {
-                        ...session,
-                        pendingTransition: {
-                            targetRouteId: targetRoute.id,
-                            condition: renderedCondition,
-                            reason: "route_complete",
-                        },
-                    };
-                    utils_1.logger.debug(`[Agent] Route ${selectedRoute.title} completed with pending transition to: ${targetRoute.title}`);
-                }
-                else {
-                    utils_1.logger.warn(`[Agent] Route ${selectedRoute.title} completed but target route not found: ${transitionConfig.nextStep}`);
-                }
-            }
-            // Set step to END_ROUTE marker
-            session = (0, utils_1.enterStep)(session, constants_1.END_ROUTE_ID, "Route completed");
-            utils_1.logger.debug(`[Agent] Route ${selectedRoute.title} completed. Entered END_ROUTE step.`);
-            // Stream completion chunks
-            for await (const chunk of stream) {
-                // Update current session if we have one
-                if (chunk.done && this.currentSession) {
-                    this.currentSession = session;
-                }
-                yield {
-                    delta: chunk.delta,
-                    accumulated: chunk.accumulated,
-                    done: chunk.done,
-                    session,
-                    toolCalls: undefined,
-                    isRouteComplete: true,
-                    metadata: chunk.metadata,
-                    structured: chunk.structured,
-                };
-            }
-        }
-        else {
-            // Fallback: No routes defined, stream a simple response
-            const fallbackPrompt = await this.responseEngine.buildFallbackPrompt({
-                history,
-                agentOptions: this.options,
-                terms: this.terms,
-                guidelines: this.guidelines,
-                context: effectiveContext,
-                session,
-            });
-            const stream = this.options.provider.generateMessageStream({
-                prompt: fallbackPrompt,
-                history,
-                context: effectiveContext,
-                signal,
-                parameters: {
-                    jsonSchema: {
-                        type: "object",
-                        properties: {
-                            message: { type: "string" },
-                        },
-                        required: ["message"],
-                        additionalProperties: false,
-                    },
-                    schemaName: "fallback_stream_response",
-                },
-            });
-            for await (const chunk of stream) {
-                // Update current session if we have one
-                if (chunk.done && this.currentSession) {
-                    this.currentSession = session;
-                }
-                yield {
-                    delta: chunk.delta,
-                    accumulated: chunk.accumulated,
-                    done: chunk.done,
-                    session, // Return updated session
-                    toolCalls: undefined,
-                    isRouteComplete: false,
-                    metadata: chunk.metadata,
-                    structured: chunk.structured,
-                };
-            }
-        }
+        // Delegate to ResponseModal
+        yield* this.responseModal.respondStream(params);
     }
     /**
      * Generate a response based on history and context
      */
     async respond(params) {
-        const { history: simpleHistory, contextOverride, signal } = params;
-        const history = (0, utils_1.normalizeHistory)(simpleHistory);
-        // Get current context (may fetch from provider)
-        let currentContext = await this.getContext();
-        // Call beforeRespond hook if configured
-        if (this.options.hooks?.beforeRespond && currentContext !== undefined) {
-            currentContext = await this.options.hooks.beforeRespond(currentContext);
-            // Update stored context with the result from beforeRespond
-            this.context = currentContext;
-        }
-        // Merge context with override
-        const effectiveContext = {
-            ...currentContext,
-            ...contextOverride,
-        };
-        // Initialize or get session (use current session if available)
-        let session = (0, utils_1.cloneDeep)(params.session) ||
-            (0, utils_1.cloneDeep)(this.currentSession) ||
-            (await this.session.getOrCreate());
-        // Merge agent's collected data into session (agent data takes precedence)
-        if (Object.keys(this.collectedData).length > 0) {
-            session = (0, utils_1.mergeCollected)(session, this.collectedData);
-            utils_1.logger.debug("[Agent] Merged agent collected data into session:", this.collectedData);
-        }
-        // PHASE 1: PREPARE - Execute prepare function if current step has one
-        if (session.currentRoute && session.currentStep) {
-            const currentRoute = this.routes.find((r) => r.id === session.currentRoute?.id);
-            if (currentRoute) {
-                const currentStep = currentRoute.getStep(session.currentStep.id);
-                if (currentStep?.prepare) {
-                    utils_1.logger.debug(`[Agent] Executing prepare for step: ${currentStep.id}`);
-                    await this.executePrepareFinalize(currentStep.prepare, effectiveContext, session.data, currentRoute, currentStep);
-                }
-            }
-        }
-        // PHASE 2: ROUTING + STEP SELECTION - Determine which route and step to use (combined)
-        let selectedRoute;
-        let responseDirectives;
-        let selectedStep;
-        let isRouteComplete = false;
-        // Check for pending transition from previous route completion
-        if (session.pendingTransition) {
-            const targetRoute = this.routes.find((r) => r.id === session.pendingTransition?.targetRouteId);
-            if (targetRoute) {
-                utils_1.logger.debug(`[Agent] Auto-transitioning from pending transition to route: ${targetRoute.title}`);
-                // Clear pending transition and enter new route
-                session = {
-                    ...session,
-                    pendingTransition: undefined,
-                };
-                session = (0, utils_1.enterRoute)(session, targetRoute.id, targetRoute.title);
-                // Merge initial data if available
-                if (targetRoute.initialData) {
-                    session = (0, utils_1.mergeCollected)(session, targetRoute.initialData);
-                }
-                selectedRoute = targetRoute;
-            }
-            else {
-                utils_1.logger.warn(`[Agent] Pending transition target route not found: ${session.pendingTransition.targetRouteId}`);
-                // Clear invalid transition
-                session = {
-                    ...session,
-                    pendingTransition: undefined,
-                };
-            }
-        }
-        // If no pending transition or transition handled, do normal routing
-        if (this.routes.length > 0 && !selectedRoute) {
-            const orchestration = await this.routingEngine.decideRouteAndStep({
-                routes: this.routes,
-                session,
-                history,
-                agentOptions: this.options,
-                provider: this.options.provider,
-                context: effectiveContext,
-                signal,
-            });
-            selectedRoute = orchestration.selectedRoute;
-            selectedStep = orchestration.selectedStep;
-            responseDirectives = orchestration.responseDirectives;
-            session = orchestration.session;
-            isRouteComplete = orchestration.isRouteComplete || false;
-            // Log if route is complete
-            if (isRouteComplete) {
-                utils_1.logger.debug(`[Agent] Route complete: all required data collected, END_ROUTE reached`);
-            }
-        }
-        // PHASE 3: DETERMINE NEXT STEP - Use step from combined decision or get initial step
-        let message;
-        let toolCalls = undefined;
-        let responsePrompt;
-        let availableTools = [];
-        let responseSchema;
-        let nextStep;
-        // Get last user message (needed for both route and completion handling)
-        const lastUserMessage = (0, utils_1.getLastMessageFromHistory)(history);
-        if (selectedRoute && !isRouteComplete) {
-            // If we have a selected step from the combined routing decision, use it
-            if (selectedStep) {
-                nextStep = selectedStep;
-            }
-            else {
-                // New route or no step selected - get initial step or first valid step
-                const candidates = this.routingEngine.getCandidateSteps(selectedRoute, undefined, session.data || {});
-                if (candidates.length > 0) {
-                    nextStep = candidates[0].step;
-                    utils_1.logger.debug(`[Agent] Using first valid step: ${nextStep.id} for new route`);
-                }
-                else {
-                    // Fallback to initial step even if it should be skipped
-                    nextStep = selectedRoute.initialStep;
-                    utils_1.logger.warn(`[Agent] No valid steps found, using initial step: ${nextStep.id}`);
-                }
-            }
-            // Update session with next step
-            session = (0, utils_1.enterStep)(session, nextStep.id, nextStep.description);
-            utils_1.logger.debug(`[Agent] Entered step: ${nextStep.id}`);
-            // PHASE 4: RESPONSE GENERATION - Generate message using selected route and step
-            // Get last user message
-            const lastUserMessage = (0, utils_1.getLastMessageFromHistory)(history);
-            // Build response schema for this route (with collect fields from step)
-            responseSchema = this.responseEngine.responseSchemaForRoute(selectedRoute, nextStep, this.schema);
-            // Build response prompt
-            responsePrompt = await this.responseEngine.buildResponsePrompt({
-                route: selectedRoute,
-                currentStep: nextStep,
-                rules: selectedRoute.getRules(),
-                prohibitions: selectedRoute.getProhibitions(),
-                directives: responseDirectives,
-                history,
-                lastMessage: lastUserMessage,
-                agentOptions: this.options,
-                // Combine agent and route properties according to the specified logic
-                combinedGuidelines: [
-                    ...this.getGuidelines(),
-                    ...selectedRoute.getGuidelines(),
-                ],
-                combinedTerms: this.mergeTerms(this.getTerms(), selectedRoute.getTerms()),
-                context: effectiveContext,
-                session,
-                agentSchema: this.schema,
-            });
-            // Collect available tools for AI
-            availableTools = this.collectAvailableTools(selectedRoute, nextStep);
-        }
-        else {
-            // No route selected - generate basic response without route context
-            utils_1.logger.debug(`[Agent] No route selected, generating basic response`);
-            // Build basic response prompt without route context
-            responsePrompt = await this.responseEngine.buildFallbackPrompt({
-                history,
-                agentOptions: this.options,
-                terms: this.getTerms(),
-                guidelines: this.getGuidelines(),
-                context: effectiveContext,
-                session,
-            });
-            // Use agent-level tools only
-            availableTools = this.collectAvailableTools();
-            responseSchema = undefined;
-        }
-        // Generate message using AI provider (common for both route and no-route cases)
-        const result = await this.options.provider.generateMessage({
-            prompt: responsePrompt,
-            history,
-            context: effectiveContext,
-            tools: availableTools,
-            signal,
-            parameters: responseSchema
-                ? {
-                    jsonSchema: responseSchema,
-                    schemaName: "response_output",
-                }
-                : undefined,
-        });
-        message = result.structured?.message || result.message;
-        // Process dynamic tool calls from AI response (common for both route and no-route cases)
-        if (result.structured?.toolCalls) {
-            toolCalls = result.structured.toolCalls;
-            // Execute dynamic tool calls
-            if (toolCalls.length > 0) {
-                utils_1.logger.debug(`[Agent] Executing ${toolCalls.length} dynamic tool calls`);
-                for (const toolCall of toolCalls) {
-                    const tool = this.findAvailableTool(toolCall.toolName, selectedRoute);
-                    if (!tool) {
-                        utils_1.logger.warn(`[Agent] Tool not found: ${toolCall.toolName}`);
-                        continue;
-                    }
-                    const toolExecutor = new ToolExecutor_1.ToolExecutor();
-                    const toolResult = await toolExecutor.executeTool({
-                        tool: tool,
-                        context: effectiveContext,
-                        updateContext: this.updateContext.bind(this),
-                        updateData: this.updateCollectedData.bind(this),
-                        history,
-                        data: session.data,
-                        toolArguments: toolCall.arguments,
-                    });
-                    // Update context with tool results
-                    if (toolResult.contextUpdate) {
-                        await this.updateContext(toolResult.contextUpdate);
-                    }
-                    // Update collected data with tool results
-                    if (toolResult.dataUpdate) {
-                        session = await this.updateData(session, toolResult.dataUpdate);
-                        utils_1.logger.debug(`[Agent] Tool updated collected data:`, toolResult.dataUpdate);
-                    }
-                    utils_1.logger.debug(`[Agent] Executed dynamic tool: ${toolResult.toolName} (success: ${toolResult.success})`);
-                }
-            }
-        }
-        // TOOL LOOP: Allow AI to make follow-up tool calls after initial tool execution
-        const MAX_TOOL_LOOPS = 5;
-        let toolLoopCount = 0;
-        let hasToolCalls = toolCalls && toolCalls.length > 0;
-        while (hasToolCalls && toolLoopCount < MAX_TOOL_LOOPS) {
-            toolLoopCount++;
-            utils_1.logger.debug(`[Agent] Starting tool loop ${toolLoopCount}/${MAX_TOOL_LOOPS}`);
-            // Add tool execution results to history so AI knows what happened
-            const toolResultsEvents = [];
-            for (const toolCall of toolCalls || []) {
-                const tool = this.findAvailableTool(toolCall.toolName, selectedRoute);
-                if (tool) {
-                    toolResultsEvents.push({
-                        kind: history_1.EventKind.TOOL,
-                        source: history_1.MessageRole.AGENT,
-                        timestamp: new Date().toISOString(),
-                        data: {
-                            tool_calls: [
-                                {
-                                    tool_id: toolCall.toolName,
-                                    arguments: toolCall.arguments,
-                                    result: {
-                                        data: "Tool executed successfully",
-                                    },
-                                },
-                            ],
-                        },
-                    });
-                }
-            }
-            // Create updated history with tool results
-            const updatedHistory = [...history, ...toolResultsEvents];
-            // Make follow-up AI call to see if more tools are needed
-            const followUpResult = await this.options.provider.generateMessage({
-                prompt: responsePrompt,
-                history: updatedHistory,
-                context: effectiveContext,
-                tools: availableTools,
-                parameters: {
-                    jsonSchema: responseSchema,
-                    schemaName: "tool_followup",
-                },
-                signal,
-            });
-            // Check if follow-up call has more tool calls
-            const followUpToolCalls = followUpResult.structured?.toolCalls;
-            hasToolCalls = followUpToolCalls && followUpToolCalls.length > 0;
-            if (hasToolCalls) {
-                utils_1.logger.debug(`[Agent] Follow-up call produced ${followUpToolCalls.length} additional tool calls`);
-                // Execute the follow-up tool calls
-                for (const toolCall of followUpToolCalls) {
-                    const tool = this.findAvailableTool(toolCall.toolName, selectedRoute);
-                    if (!tool) {
-                        utils_1.logger.warn(`[Agent] Tool not found in follow-up: ${toolCall.toolName}`);
-                        continue;
-                    }
-                    const toolExecutor = new ToolExecutor_1.ToolExecutor();
-                    const toolResult = await toolExecutor.executeTool({
-                        tool: tool,
-                        context: effectiveContext,
-                        updateContext: this.updateContext.bind(this),
-                        updateData: this.updateCollectedData.bind(this),
-                        history: updatedHistory,
-                        data: session.data,
-                        toolArguments: toolCall.arguments,
-                    });
-                    // Update context with follow-up tool results
-                    if (toolResult.contextUpdate) {
-                        await this.updateContext(toolResult.contextUpdate);
-                    }
-                    if (toolResult.dataUpdate) {
-                        session = await this.updateData(session, toolResult.dataUpdate);
-                        utils_1.logger.debug(`[Agent] Follow-up tool updated collected data:`, toolResult.dataUpdate);
-                    }
-                    utils_1.logger.debug(`[Agent] Executed follow-up tool: ${toolResult.toolName} (success: ${toolResult.success})`);
-                }
-                // Update toolCalls for next iteration or final response
-                toolCalls = followUpToolCalls;
-            }
-            else {
-                utils_1.logger.debug(`[Agent] Tool loop completed after ${toolLoopCount} iterations`);
-                // Update final message and toolCalls from follow-up result if no more tools
-                message = followUpResult.structured?.message || followUpResult.message;
-                toolCalls = followUpToolCalls || [];
-                break;
-            }
-        }
-        if (toolLoopCount >= MAX_TOOL_LOOPS) {
-            utils_1.logger.warn(`[Agent] Tool loop limit reached (${MAX_TOOL_LOOPS}), stopping`);
-        }
-        // Extract collected data from final response (only for route-based interactions)
-        if (selectedRoute && result.structured && nextStep?.collect) {
-            const collectedData = {};
-            // The structured response includes both base fields and collected extraction fields
-            const structuredData = result.structured;
-            for (const field of nextStep.collect) {
-                const fieldKey = String(field);
-                if (fieldKey in structuredData) {
-                    collectedData[fieldKey] = structuredData[fieldKey];
-                }
-            }
-            // Merge collected data into session using agent-level data validation
-            if (Object.keys(collectedData).length > 0) {
-                // Update agent-level collected data with validation
-                await this.updateCollectedData(collectedData);
-                // Update session with validated data
-                session = await this.updateData(session, collectedData);
-                utils_1.logger.debug(`[Agent] Collected data:`, collectedData);
-            }
-        }
-        // Extract any additional data from structured response
-        if (result.structured &&
-            typeof result.structured === "object" &&
-            "contextUpdate" in result.structured) {
-            await this.updateContext(result.structured
-                .contextUpdate);
-        }
-        // Handle route completion if route is complete
-        if (isRouteComplete) {
-            // Route is complete - generate completion message then check for onComplete transition
-            // Get endStep spec from route
-            const endStepSpec = selectedRoute.endStepSpec;
-            // Create a temporary step for completion message generation using endStep configuration
-            const completionStep = new Step_1.Step(selectedRoute.id, {
-                description: endStepSpec.description,
-                id: endStepSpec.id || constants_1.END_ROUTE_ID,
-                collect: endStepSpec.collect,
-                requires: endStepSpec.requires,
-                prompt: endStepSpec.prompt ||
-                    "Summarize what was accomplished and confirm completion based on the conversation history and collected data",
-            });
-            if (!selectedRoute) {
-                throw new Error("Selected route is not defined");
-            }
-            // Build response schema for completion
-            const responseSchema = this.responseEngine.responseSchemaForRoute(selectedRoute, completionStep, this.schema);
-            const templateContext = {
-                context: effectiveContext,
-                session,
-                history,
-            };
-            // Build completion response prompt
-            const completionPrompt = await this.responseEngine.buildResponsePrompt({
-                route: selectedRoute,
-                currentStep: completionStep,
-                rules: selectedRoute.getRules(),
-                prohibitions: selectedRoute.getProhibitions(),
-                directives: undefined, // No directives for completion
-                history,
-                lastMessage: lastUserMessage,
-                agentOptions: this.options,
-                // Combine agent and route properties according to the specified logic
-                combinedGuidelines: [
-                    ...this.getGuidelines(),
-                    ...selectedRoute.getGuidelines(),
-                ],
-                combinedTerms: this.mergeTerms(this.getTerms(), selectedRoute.getTerms()),
-                context: effectiveContext,
-                session,
-                agentSchema: this.schema,
-            });
-            // Generate completion message using AI provider
-            const completionResult = await this.options.provider.generateMessage({
-                prompt: completionPrompt,
-                history,
-                context: effectiveContext,
-                signal,
-                parameters: {
-                    jsonSchema: responseSchema,
-                    schemaName: "completion_message",
-                },
-            });
-            message =
-                completionResult.structured?.message || completionResult.message;
-            utils_1.logger.debug(`[Agent] Generated completion message for route: ${selectedRoute.title}`);
-            // Check for onComplete transition
-            const transitionConfig = await selectedRoute.evaluateOnComplete({ data: session.data }, effectiveContext);
-            if (transitionConfig) {
-                // Find target route by ID or title
-                const targetRoute = this.routes.find((r) => r.id === transitionConfig.nextStep ||
-                    r.title === transitionConfig.nextStep);
-                if (targetRoute) {
-                    const renderedCondition = await (0, utils_1.render)(transitionConfig.condition, templateContext);
-                    // Set pending transition in session
-                    session = {
-                        ...session,
-                        pendingTransition: {
-                            targetRouteId: targetRoute.id,
-                            condition: renderedCondition,
-                            reason: "route_complete",
-                        },
-                    };
-                    utils_1.logger.debug(`[Agent] Route ${selectedRoute.title} completed with pending transition to: ${targetRoute.title}`);
-                }
-                else {
-                    utils_1.logger.warn(`[Agent] Route ${selectedRoute.title} completed but target route not found: ${transitionConfig.nextStep}`);
-                }
-            }
-            // Set step to END_ROUTE marker
-            session = (0, utils_1.enterStep)(session, constants_1.END_ROUTE_ID, "Route completed");
-            utils_1.logger.debug(`[Agent] Route ${selectedRoute.title} completed. Entered END_ROUTE step.`);
-        }
-        else {
-            // Fallback: No routes defined, generate a simple response
-            const fallbackPrompt = await this.responseEngine.buildFallbackPrompt({
-                history,
-                agentOptions: this.options,
-                terms: this.terms,
-                guidelines: this.guidelines,
-                context: effectiveContext,
-                session,
-            });
-            const result = await this.options.provider.generateMessage({
-                prompt: fallbackPrompt,
-                history,
-                context: effectiveContext,
-                signal,
-                parameters: {
-                    jsonSchema: {
-                        type: "object",
-                        properties: {
-                            message: { type: "string" },
-                        },
-                        required: ["message"],
-                        additionalProperties: false,
-                    },
-                    schemaName: "fallback_response",
-                },
-            });
-            message = result.structured?.message || result.message;
-        }
-        // Auto-save session step to persistence if configured
-        if (this.persistenceManager &&
-            session.id &&
-            this.options.persistence?.autoSave !== false) {
-            await this.persistenceManager.saveSessionState(session.id, session);
-            utils_1.logger.debug(`[Agent] Auto-saved session step to persistence: ${session.id}`);
-        }
-        // Execute finalize function
-        if (session.currentRoute && session.currentStep) {
-            const currentRoute = this.routes.find((r) => r.id === session.currentRoute?.id);
-            if (currentRoute) {
-                const currentStep = currentRoute.getStep(session.currentStep.id);
-                if (currentStep?.finalize) {
-                    utils_1.logger.debug(`[Agent] Executing finalize for step: ${currentStep.id}`);
-                    await this.executePrepareFinalize(currentStep.finalize, effectiveContext, session.data, currentRoute, currentStep);
-                }
-            }
-        }
-        // Update current session if we have one
-        if (this.currentSession) {
-            this.currentSession = session;
-        }
-        return {
-            message,
-            session, // Return updated session with route/step info
-            toolCalls,
-            isRouteComplete, // Indicates if the route has reached END_ROUTE with all data collected
-        };
+        // Delegate to ResponseModal
+        return this.responseModal.respond(params);
     }
     /**
      * Get all routes
@@ -1329,6 +471,27 @@ class Agent {
     getRoutes() {
         return [...this.routes];
     }
+    /**
+     * Get agent options
+     * @internal Used by ResponseModal
+     */
+    getAgentOptions() {
+        return this.options;
+    }
+    /**
+     * Get routing engine
+     * @internal Used by ResponseModal
+     */
+    getRoutingEngine() {
+        return this.routingEngine;
+    }
+    /**
+     * Get the updateData method bound to this agent
+     * @internal Used by ResponseModal
+     */
+    getUpdateDataMethod() {
+        return this.updateData.bind(this);
+    }
     /**
      * Get all terms
      */
@@ -1342,85 +505,45 @@ class Agent {
         return [...this.tools];
     }
     /**
-     * Find an available tool by name for the given route
-     * Route-level tools take precedence over agent-level tools
-     * @private
+     * Get all guidelines
      */
-    findAvailableTool(toolName, route) {
-        // Check route-level tools first (if route provided)
-        if (route) {
-            const routeTool = route
-                .getTools()
-                .find((tool) => tool.id === toolName || tool.name === toolName);
-            if (routeTool)
-                return routeTool;
-        }
-        // Fall back to agent-level tools
-        return this.tools.find((tool) => tool.id === toolName || tool.name === toolName);
+    getGuidelines() {
+        return [...this.guidelines];
     }
     /**
-     * Collect all available tools for the given route and step context
-     * @private
+     * Get the agent's knowledge base
      */
-    collectAvailableTools(route, step) {
-        const availableTools = new Map();
-        // Add agent-level tools
-        this.tools.forEach((tool) => {
-            availableTools.set(tool.id, tool);
-        });
-        // Add route-level tools (these take precedence)
-        if (route) {
-            route.getTools().forEach((tool) => {
-                availableTools.set(tool.id, tool);
-            });
-        }
-        // Filter by step-level allowed tools if specified
-        if (step?.tools) {
-            const allowedToolIds = new Set();
-            const stepTools = [];
-            for (const toolRef of step.tools) {
-                if (typeof toolRef === "string") {
-                    // Reference to registered tool
-                    allowedToolIds.add(toolRef);
-                }
-                else {
-                    // Inline tool definition
-                    if (toolRef.id) {
-                        allowedToolIds.add(toolRef.id);
-                        stepTools.push(toolRef);
-                    }
-                }
-            }
-            // If step specifies tools, only include those
-            if (allowedToolIds.size > 0) {
-                const filteredTools = new Map();
-                for (const toolId of allowedToolIds) {
-                    const tool = availableTools.get(toolId);
-                    if (tool) {
-                        filteredTools.set(toolId, tool);
-                    }
-                }
-                // Add inline tools
-                stepTools.forEach((tool) => {
-                    if (tool.id) {
-                        filteredTools.set(tool.id, tool);
-                    }
-                });
-                availableTools.clear();
-                filteredTools.forEach((tool, id) => availableTools.set(id, tool));
-            }
-        }
-        // Convert to the format expected by AI providers
-        return Array.from(availableTools.values()).map((tool) => ({
-            id: tool.id,
-            name: tool.name || tool.id,
-            description: tool.description,
-            parameters: tool.parameters,
-        }));
+    getKnowledgeBase() {
+        return { ...this.knowledgeBase };
+    }
+    /**
+     * Get the persistence manager (if configured)
+     */
+    getPersistenceManager() {
+        return this.persistenceManager;
+    }
+    /**
+     * Check if persistence is enabled
+     */
+    hasPersistence() {
+        return this.persistenceManager !== undefined;
+    }
+    /**
+     * Set the current session for convenience methods
+     * @param session - Session step to use for subsequent calls
+     */
+    setCurrentSession(session) {
+        this.currentSession = session;
+    }
+    /**
+     * Get the current session (if set)
+     */
+    getCurrentSession() {
+        return this.currentSession;
     }
     /**
      * Execute a prepare or finalize function/tool
-     * @private
+     * @internal Used by ResponseModal
      */
     async executePrepareFinalize(prepareOrFinalize, context, data, route, step) {
         if (!prepareOrFinalize)
@@ -1433,38 +556,22 @@ class Agent {
             // It's a tool reference - find and execute the tool
             let tool;
             if (typeof prepareOrFinalize === "string") {
-                // Tool ID - find it in available tools
-                const availableTools = new Map();
-                // Add agent-level tools
-                this.tools.forEach((t) => {
-                    availableTools.set(t.id, t);
-                });
-                // Add route-level tools
-                if (route) {
-                    route.getTools().forEach((t) => {
-                        availableTools.set(t.id, t);
-                    });
-                }
-                // Add step-level tools
-                if (step?.tools) {
-                    for (const toolRef of step.tools) {
-                        if (typeof toolRef === "string") {
-                            // Keep as is
-                        }
-                        else if (toolRef.id) {
-                            availableTools.set(toolRef.id, toolRef);
-                        }
-                    }
-                }
-                tool = availableTools.get(prepareOrFinalize);
+                // Tool ID - use ToolManager to find it across all scopes
+                tool = this.tool.find(prepareOrFinalize, undefined, step, route);
             }
             else {
-                // Tool object - use directly
-                tool = prepareOrFinalize;
+                // Tool object - validate it has required properties
+                if (prepareOrFinalize.id && typeof prepareOrFinalize.handler === 'function') {
+                    tool = prepareOrFinalize;
+                }
+                else {
+                    utils_1.logger.error(`[Agent] Invalid tool object for prepare/finalize: missing id or invalid handler`);
+                    return;
+                }
             }
             if (tool) {
-                const toolExecutor = new ToolExecutor_1.ToolExecutor();
-                const result = await toolExecutor.executeTool({
+                // Use ToolManager for execution
+                const result = await this.tool.executeTool({
                     tool,
                     context,
                     updateContext: this.updateContext.bind(this),
@@ -1484,73 +591,31 @@ class Agent {
             }
         }
     }
-    /**
-     * Get all guidelines
-     */
-    getGuidelines() {
-        return [...this.guidelines];
-    }
-    /**
-     * Get the agent's knowledge base
-     */
-    getKnowledgeBase() {
-        return { ...this.knowledgeBase };
-    }
-    /**
-     * Merge terms with route-specific taking precedence on conflicts
-     * @private
-     */
-    mergeTerms(agentTerms, routeTerms) {
-        const merged = new Map();
-        // Add agent terms first
-        agentTerms.forEach((term) => {
-            const name = typeof term.name === "string" ? term.name : term.name.toString();
-            merged.set(name, term);
-        });
-        // Add route terms (these take precedence)
-        routeTerms.forEach((term) => {
-            const name = typeof term.name === "string" ? term.name : term.name.toString();
-            merged.set(name, term);
-        });
-        return Array.from(merged.values());
-    }
-    /**
-     * Get the persistence manager (if configured)
-     */
-    getPersistenceManager() {
-        return this.persistenceManager;
-    }
-    /**
-     * Check if persistence is enabled
-     */
-    hasPersistence() {
-        return this.persistenceManager !== undefined;
-    }
-    /**
-     * Set the current session for convenience methods
-     * @param session - Session step to use for subsequent calls
-     */
-    setCurrentSession(session) {
-        this.currentSession = session;
-    }
-    /**
-     * Get the current session (if set)
-     */
-    getCurrentSession() {
-        return this.currentSession;
-    }
     /**
      * Clear the current session
      */
     clearCurrentSession() {
         this.currentSession = undefined;
     }
+    /**
+     * Sync session data to agent collected data
+     * @internal Used to keep agent and session data in sync
+     */
+    syncSessionDataToCollectedData() {
+        const sessionData = this.session.getData();
+        if (sessionData && Object.keys(sessionData).length > 0) {
+            this.collectedData = { ...sessionData };
+            utils_1.logger.debug("[Agent] Synced session data to collected data:", this.collectedData);
+        }
+    }
     /**
      * Get collected data from current session or agent-level collected data
      * @param routeId - Optional route ID to get data for (uses current route if not provided)
      * @returns The collected data from the current session or agent-level data
      */
     getData() {
+        // Ensure agent collected data is synced with session
+        this.syncSessionDataToCollectedData();
         // If we have a current session, use session data
         if (this.currentSession) {
             // With agent-level data, all routes share the same data structure
@@ -1616,44 +681,20 @@ class Agent {
      * Automatically manages conversation history through the session
      */
     async chat(message, options) {
-        // Determine which history to use
-        let history;
-        if (options?.history) {
-            // Use provided history for this response only
-            history = options.history;
-        }
-        else {
-            // Add user message to session history if provided
-            if (message) {
-                await this.session.addMessage("user", message);
-            }
-            history = this.session.getHistory();
-        }
-        // Get or create session
-        let session = await this.session.getOrCreate();
-        // Merge agent's collected data into session (agent data takes precedence)
-        if (Object.keys(this.collectedData).length > 0) {
-            session = (0, utils_1.mergeCollected)(session, this.collectedData);
-            // Update the session manager with the merged data
-            await this.session.setData(this.collectedData);
-            utils_1.logger.debug("[Agent] Merged agent collected data into chat session:", this.collectedData);
-        }
-        // Use existing respond method with session-managed history
-        const result = await this.respond({
-            history,
-            session,
+        // Delegate to ResponseModal.generate()
+        return this.responseModal.generate(message, options);
+    }
+    /**
+     * Modern streaming API - simple interface like chat() but returns a stream
+     * Automatically manages conversation history through the session
+     */
+    async *stream(message, options) {
+        // Delegate to ResponseModal with the same options structure as chat()
+        yield* this.responseModal.stream(message, {
+            history: options?.history,
             contextOverride: options?.contextOverride,
             signal: options?.signal,
         });
-        // Add agent response to session history (only if not using override history)
-        if (!options?.history) {
-            await this.session.addMessage("assistant", result.message);
-        }
-        // Ensure the result includes the current session
-        return {
-            ...result,
-            session: result.session || this.session.current,
-        };
     }
 }
 exports.Agent = Agent;