@wingman-ai/gateway 0.2.5 → 0.3.1

package/dist/cli/config/schema.js

@@ -16,6 +16,34 @@ const SkillsConfigSchema = object({
     githubToken: string().optional().describe("GitHub personal access token for higher API rate limits"),
     skillsDirectory: string().default("skills").describe("Directory to install skills in")
 });
+const BrowserTransportSchema = external_zod_enum([
+    "auto",
+    "playwright",
+    "relay"
+]).describe("Browser automation transport preference");
+const BrowserConfigSchema = object({
+    profilesDir: string().optional().default(".wingman/browser-profiles").describe("Base directory for named browser_control profiles (relative to workspace unless absolute)"),
+    profiles: record(string(), string()).optional().default({}).describe("Optional mapping of profile ID to explicit profile path (relative to workspace unless absolute)"),
+    extensionsDir: string().optional().default(".wingman/browser-extensions").describe("Base directory for named browser extensions (relative to workspace unless absolute)"),
+    extensions: record(string(), string()).optional().default({}).describe("Optional mapping of extension ID to unpacked extension path (relative to workspace unless absolute)"),
+    defaultExtensions: array(string().min(1)).optional().default([]).describe("Optional default extension IDs loaded for browser profile open and browser_control runs"),
+    defaultProfile: string().min(1).optional().describe("Optional default browser profile ID used when an agent does not specify browserProfile"),
+    transport: BrowserTransportSchema.optional().default("auto").describe('Default browser transport preference for browser_control ("auto", "playwright", or "relay")'),
+    relay: object({
+        enabled: external_zod_boolean().optional().default(false).describe("Enable local browser relay server for extension/CDP bridging"),
+        host: string().optional().default("127.0.0.1").describe("Relay bind host (security: keep this loopback)"),
+        port: number().int().min(1).max(65535).optional().default(18792).describe("Relay bind port"),
+        requireAuth: external_zod_boolean().optional().default(true).describe("Require relay clients to authenticate with a token"),
+        authToken: string().min(16).optional().describe("Relay shared secret for extension/CDP clients"),
+        maxMessageBytes: number().int().min(1024).max(2097152).optional().default(262144).describe("Maximum relay websocket message size")
+    }).optional().default({
+        enabled: false,
+        host: "127.0.0.1",
+        port: 18792,
+        requireAuth: true,
+        maxMessageBytes: 262144
+    })
+});
 const SummarizationConfigSchema = object({
     enabled: external_zod_boolean().optional().default(true).describe("Enable conversation history summarization"),
     maxTokensBeforeSummary: number().min(1000).max(1000000).optional().default(12000).describe("Token threshold before summarizing conversation history"),
@@ -238,6 +266,21 @@ const WingmanConfigSchema = object({
         repositoryName: "skills",
         skillsDirectory: "skills"
     }),
+    browser: BrowserConfigSchema.optional().default({
+        profilesDir: ".wingman/browser-profiles",
+        profiles: {},
+        extensionsDir: ".wingman/browser-extensions",
+        extensions: {},
+        defaultExtensions: [],
+        transport: "auto",
+        relay: {
+            enabled: false,
+            host: "127.0.0.1",
+            port: 18792,
+            requireAuth: true,
+            maxMessageBytes: 262144
+        }
+    }),
     gateway: GatewayConfigSchema.optional().default({
         host: "127.0.0.1",
         port: 18789,
@@ -279,4 +322,4 @@ function validateConfig(data) {
         };
     }
 }
-export { AgentsConfigSchema, GatewayConfigSchema, HumanInTheLoopConfigSchema, ModelRetryConfigSchema, SearchConfigSchema, SkillsConfigSchema, SummarizationConfigSchema, ToolRetryConfigSchema, WingmanConfigSchema, validateConfig };
+export { AgentsConfigSchema, BrowserConfigSchema, BrowserTransportSchema, GatewayConfigSchema, HumanInTheLoopConfigSchema, ModelRetryConfigSchema, SearchConfigSchema, SkillsConfigSchema, SummarizationConfigSchema, ToolRetryConfigSchema, WingmanConfigSchema, validateConfig };

package/dist/cli/core/agentInvoker.cjs

@@ -26,7 +26,9 @@ __webpack_require__.r(__webpack_exports__);
 __webpack_require__.d(__webpack_exports__, {
     resolveToolRetryMiddlewareSettings: ()=>resolveToolRetryMiddlewareSettings,
     resolveSummarizationMiddlewareSettings: ()=>resolveSummarizationMiddlewareSettings,
+    resolveAgentMemorySources: ()=>resolveAgentMemorySources,
     OUTPUT_VIRTUAL_PATH: ()=>OUTPUT_VIRTUAL_PATH,
+    emitCompletionAndContinuePostProcessing: ()=>emitCompletionAndContinuePostProcessing,
     resolveExecutionWorkspace: ()=>resolveExecutionWorkspace,
     buildUserContent: ()=>buildUserContent,
     detectStreamErrorMessage: ()=>detectStreamErrorMessage,
@@ -34,15 +36,16 @@ __webpack_require__.d(__webpack_exports__, {
     chunkHasAssistantText: ()=>chunkHasAssistantText,
     resolveModelRetryMiddlewareSettings: ()=>resolveModelRetryMiddlewareSettings,
     AgentInvoker: ()=>AgentInvoker,
-    WORKDIR_VIRTUAL_PATH: ()=>WORKDIR_VIRTUAL_PATH,
     configureDeepAgentSummarizationMiddleware: ()=>configureDeepAgentSummarizationMiddleware,
+    WORKDIR_VIRTUAL_PATH: ()=>WORKDIR_VIRTUAL_PATH,
     detectToolEventContext: ()=>detectToolEventContext,
     resolveExternalOutputMount: ()=>resolveExternalOutputMount,
-    evaluateStreamingCompletion: ()=>evaluateStreamingCompletion,
     resolveHumanInTheLoopSettings: ()=>resolveHumanInTheLoopSettings,
     trackRootLangGraphRunId: ()=>trackRootLangGraphRunId,
     isRootLangGraphTerminalEvent: ()=>isRootLangGraphTerminalEvent,
-    selectStreamingFallbackText: ()=>selectStreamingFallbackText
+    selectStreamingFallbackText: ()=>selectStreamingFallbackText,
+    AGENTS_MEMORY_VIRTUAL_PATHS: ()=>AGENTS_MEMORY_VIRTUAL_PATHS,
+    resolveAgentExecutionWorkspace: ()=>resolveAgentExecutionWorkspace
 });
 const external_node_fs_namespaceObject = require("node:fs");
 const external_node_path_namespaceObject = require("node:path");
@@ -70,6 +73,9 @@ function _define_property(obj, key, value) {
 }
 const WORKDIR_VIRTUAL_PATH = "/workdir/";
 const OUTPUT_VIRTUAL_PATH = "/output/";
+const AGENTS_MEMORY_VIRTUAL_PATHS = [
+    "/AGENTS.md"
+];
 const DEFAULT_DEEPAGENT_MODEL = "claude-sonnet-4-5-20250929";
 const isPathWithinRoot = (targetPath, rootPath)=>{
     const normalizedTarget = (0, external_node_path_namespaceObject.normalize)(targetPath);
@@ -81,6 +87,11 @@ const resolveExecutionWorkspace = (workspace, workdir)=>{
     if ((0, external_node_path_namespaceObject.isAbsolute)(workdir)) return (0, external_node_path_namespaceObject.normalize)(workdir);
     return (0, external_node_path_namespaceObject.normalize)((0, external_node_path_namespaceObject.join)(workspace, workdir));
 };
+const resolveAgentExecutionWorkspace = (workspace, workdir, defaultOutputDir)=>{
+    const preferredWorkdir = workdir || defaultOutputDir || null;
+    return resolveExecutionWorkspace(workspace, preferredWorkdir);
+};
+const resolveAgentMemorySources = (executionWorkspace)=>AGENTS_MEMORY_VIRTUAL_PATHS.filter((memoryPath)=>(0, external_node_fs_namespaceObject.existsSync)((0, external_node_path_namespaceObject.join)(executionWorkspace, memoryPath.replace(/^\/+/, ""))));
 const toWorkspaceAliasVirtualPath = (absolutePath)=>{
     const normalized = (0, external_node_path_namespaceObject.normalize)(absolutePath);
     if (!(0, external_node_path_namespaceObject.isAbsolute)(normalized)) return null;
@@ -284,24 +295,19 @@ const trackRootLangGraphRunId = (currentRootLangGraphRunId, chunk)=>{
     return extractEventRunId(eventRecord) || currentRootLangGraphRunId;
 };
 const isRootLangGraphTerminalEvent = (chunk, rootLangGraphRunId)=>{
-    if (!rootLangGraphRunId) return false;
     const eventRecord = extractStreamEventRecord(chunk);
     if (!eventRecord || !isRootLangGraphChainEvent(eventRecord, "on_chain_end")) return false;
+    if (!rootLangGraphRunId) return true;
     const chunkRunId = extractEventRunId(eventRecord);
+    if (!chunkRunId) return true;
     return Boolean(chunkRunId && chunkRunId === rootLangGraphRunId);
 };
-const evaluateStreamingCompletion = (input)=>{
-    if (!input.sawAssistantText && !input.fallbackText) {
-        const message = input.streamErrorMessage ? `Model call failed: ${input.streamErrorMessage}` : "Model completed without a response. Check provider logs for request errors.";
-        return {
-            status: "blocked",
-            reason: input.streamErrorMessage ? "stream_error" : "empty_stream_response",
-            message
-        };
-    }
-    return {
-        status: "ok"
-    };
+const emitCompletionAndContinuePostProcessing = (input)=>{
+    input.outputManager.emitAgentComplete(input.result);
+    if (!input.postProcess) return;
+    input.postProcess().catch((error)=>{
+        input.logger?.debug("Failed post-completion processing for streamed agent response", error);
+    });
 };
 class AgentInvoker {
     findAllAgents() {
@@ -315,14 +321,11 @@ class AgentInvoker {
         let cancellationHandled = false;
         let activeToolName = null;
         let lastToolName = null;
-        let sawAssistantText = false;
-        let streamErrorMessage;
         let rootLangGraphRunId;
-        let preInvocationMessages = null;
         const isCancelled = ()=>options?.signal?.aborted === true;
         try {
             const hookSessionId = sessionId || (0, external_uuid_namespaceObject.v4)();
-            const executionWorkspace = resolveExecutionWorkspace(this.workspace, this.workdir);
+            const executionWorkspace = resolveAgentExecutionWorkspace(this.workspace, this.workdir, this.defaultOutputDir);
             const effectiveWorkdir = this.workdir ? executionWorkspace : null;
             const loader = new agentLoader_cjs_namespaceObject.AgentLoader(this.configDir, this.workspace, this.wingmanConfig, executionWorkspace, {
                 terminalOwnerId: `${agentName}:${hookSessionId}`,
@@ -341,7 +344,9 @@ class AgentInvoker {
             if (targetAgent.mcpUseGlobal && this.wingmanConfig.mcp) mcpConfigs.push(this.wingmanConfig.mcp);
             if (mcpConfigs.length > 0) {
                 this.logger.debug("Initializing MCP client for agent invocation");
-                this.mcpManager = new mcpClientManager_cjs_namespaceObject.MCPClientManager(mcpConfigs, this.logger);
+                this.mcpManager = new mcpClientManager_cjs_namespaceObject.MCPClientManager(mcpConfigs, this.logger, {
+                    executionWorkspace
+                });
                 await this.mcpManager.initialize();
                 const mcpTools = await this.mcpManager.getTools();
                 if (mcpTools.length > 0) {
@@ -358,6 +363,7 @@ class AgentInvoker {
             const normalizedSkillsDirectory = skillsDirectory.replace(/^\/+|\/+$/g, "");
             const skillsVirtualPath = `/${normalizedSkillsDirectory}/`;
             const outputMount = resolveExternalOutputMount(executionWorkspace, effectiveWorkdir, this.defaultOutputDir);
+            const memorySources = resolveAgentMemorySources(executionWorkspace);
             const middleware = [
                 (0, media_compat_cjs_namespaceObject.mediaCompatibilityMiddleware)({
                     model: targetAgent.model
@@ -442,6 +448,7 @@ class AgentInvoker {
                 middleware: middleware,
                 interruptOn: hitlSettings?.interruptOn,
                 skills: skillsSources,
+                memory: memorySources,
                 subagents: targetAgent.subagents || [],
                 checkpointer: checkpointer
             });
@@ -450,15 +457,6 @@ class AgentInvoker {
             const userContent = buildUserContent(prompt, attachments, targetAgent.model);
             if (this.sessionManager && sessionId) {
                 this.logger.debug(`Using streaming with session: ${sessionId}`);
-                try {
-                    const messages = await this.sessionManager.listMessages(sessionId);
-                    preInvocationMessages = messages.map((message)=>({
-                            role: message.role,
-                            content: message.content
-                        }));
-                } catch (stateError) {
-                    this.logger.debug("Failed to capture pre-invocation session state", stateError);
-                }
                 const stream = await standaloneAgent.streamEvents({
                     messages: [
                         {
@@ -476,8 +474,6 @@ class AgentInvoker {
                 });
                 for await (const chunk of stream){
                     rootLangGraphRunId = trackRootLangGraphRunId(rootLangGraphRunId, chunk);
-                    if (!sawAssistantText && chunkHasAssistantText(chunk)) sawAssistantText = true;
-                    if (!streamErrorMessage) streamErrorMessage = detectStreamErrorMessage(chunk);
                     const toolEvent = detectToolEventContext(chunk);
                     if (toolEvent) {
                         lastToolName = toolEvent.toolName;
@@ -507,37 +503,15 @@ class AgentInvoker {
                         cancelled: true
                     };
                 }
-                let fallbackText;
-                if (!sawAssistantText && this.sessionManager && sessionId && preInvocationMessages) try {
-                    const sessionMessages = await this.sessionManager.listMessages(sessionId);
-                    fallbackText = selectStreamingFallbackText(preInvocationMessages, sessionMessages.map((message)=>({
-                            role: message.role,
-                            content: message.content
-                        })));
-                } catch (stateError) {
-                    this.logger.debug("Failed to derive streaming fallback text from session state", stateError);
-                }
-                const completionOutcome = evaluateStreamingCompletion({
-                    sawAssistantText,
-                    fallbackText,
-                    streamErrorMessage
-                });
-                if ("blocked" === completionOutcome.status) {
-                    this.logger.warn(completionOutcome.message);
-                    this.outputManager.emitAgentError(completionOutcome.message);
-                    return {
-                        blocked: true,
-                        reason: completionOutcome.reason
-                    };
-                }
-                this.logger.info("Agent streaming completed successfully", {
-                    usedFallbackText: Boolean(fallbackText)
-                });
-                this.outputManager.emitAgentComplete({
-                    streaming: true,
-                    ...fallbackText ? {
-                        fallbackText
-                    } : {}
+                this.logger.info("Agent streaming completed successfully");
+                const completionPayload = {
+                    streaming: true
+                };
+                emitCompletionAndContinuePostProcessing({
+                    outputManager: this.outputManager,
+                    result: completionPayload,
+                    postProcess: ()=>this.materializeSessionImages(sessionId),
+                    logger: this.logger
                 });
                 return {
                     streaming: true
@@ -573,7 +547,12 @@ class AgentInvoker {
                     };
                 }
                 this.logger.info("Agent completed successfully");
-                this.outputManager.emitAgentComplete(result);
+                emitCompletionAndContinuePostProcessing({
+                    outputManager: this.outputManager,
+                    result,
+                    postProcess: ()=>this.materializeSessionImages(sessionId),
+                    logger: this.logger
+                });
                 return result;
             }
         } catch (error) {
@@ -605,6 +584,10 @@ class AgentInvoker {
                 description: a.description
             }));
     }
+    async materializeSessionImages(sessionId) {
+        if (!this.sessionManager || !sessionId) return;
+        await this.sessionManager.listMessages(sessionId);
+    }
     constructor(options){
         _define_property(this, "loader", void 0);
         _define_property(this, "outputManager", void 0);
@@ -815,6 +798,7 @@ function buildAttachmentPreview(attachments) {
     if (hasImage) return "[image]";
     return "";
 }
+exports.AGENTS_MEMORY_VIRTUAL_PATHS = __webpack_exports__.AGENTS_MEMORY_VIRTUAL_PATHS;
 exports.AgentInvoker = __webpack_exports__.AgentInvoker;
 exports.OUTPUT_VIRTUAL_PATH = __webpack_exports__.OUTPUT_VIRTUAL_PATH;
 exports.WORKDIR_VIRTUAL_PATH = __webpack_exports__.WORKDIR_VIRTUAL_PATH;
@@ -823,8 +807,10 @@ exports.chunkHasAssistantText = __webpack_exports__.chunkHasAssistantText;
 exports.configureDeepAgentSummarizationMiddleware = __webpack_exports__.configureDeepAgentSummarizationMiddleware;
 exports.detectStreamErrorMessage = __webpack_exports__.detectStreamErrorMessage;
 exports.detectToolEventContext = __webpack_exports__.detectToolEventContext;
-exports.evaluateStreamingCompletion = __webpack_exports__.evaluateStreamingCompletion;
+exports.emitCompletionAndContinuePostProcessing = __webpack_exports__.emitCompletionAndContinuePostProcessing;
 exports.isRootLangGraphTerminalEvent = __webpack_exports__.isRootLangGraphTerminalEvent;
+exports.resolveAgentExecutionWorkspace = __webpack_exports__.resolveAgentExecutionWorkspace;
+exports.resolveAgentMemorySources = __webpack_exports__.resolveAgentMemorySources;
 exports.resolveExecutionWorkspace = __webpack_exports__.resolveExecutionWorkspace;
 exports.resolveExternalOutputMount = __webpack_exports__.resolveExternalOutputMount;
 exports.resolveHumanInTheLoopSettings = __webpack_exports__.resolveHumanInTheLoopSettings;
@@ -835,6 +821,7 @@ exports.selectStreamingFallbackText = __webpack_exports__.selectStreamingFallbac
 exports.toWorkspaceAliasVirtualPath = __webpack_exports__.toWorkspaceAliasVirtualPath;
 exports.trackRootLangGraphRunId = __webpack_exports__.trackRootLangGraphRunId;
 for(var __rspack_i in __webpack_exports__)if (-1 === [
+    "AGENTS_MEMORY_VIRTUAL_PATHS",
     "AgentInvoker",
     "OUTPUT_VIRTUAL_PATH",
     "WORKDIR_VIRTUAL_PATH",
@@ -843,8 +830,10 @@ for(var __rspack_i in __webpack_exports__)if (-1 === [
     "configureDeepAgentSummarizationMiddleware",
     "detectStreamErrorMessage",
     "detectToolEventContext",
-    "evaluateStreamingCompletion",
+    "emitCompletionAndContinuePostProcessing",
     "isRootLangGraphTerminalEvent",
+    "resolveAgentExecutionWorkspace",
+    "resolveAgentMemorySources",
     "resolveExecutionWorkspace",
     "resolveExternalOutputMount",
     "resolveHumanInTheLoopSettings",

package/dist/cli/core/agentInvoker.d.ts

@@ -78,6 +78,7 @@ type UserContentPart = {
 };
 export declare const WORKDIR_VIRTUAL_PATH = "/workdir/";
 export declare const OUTPUT_VIRTUAL_PATH = "/output/";
+export declare const AGENTS_MEMORY_VIRTUAL_PATHS: readonly ["/AGENTS.md"];
 export type ExternalOutputMount = {
     virtualPath: string | null;
     absolutePath: string | null;
@@ -105,6 +106,8 @@ export type HumanInTheLoopSettings = {
     }>;
 };
 export declare const resolveExecutionWorkspace: (workspace: string, workdir?: string | null) => string;
+export declare const resolveAgentExecutionWorkspace: (workspace: string, workdir?: string | null, defaultOutputDir?: string | null) => string;
+export declare const resolveAgentMemorySources: (executionWorkspace: string) => string[];
 export declare const toWorkspaceAliasVirtualPath: (absolutePath: string) => string | null;
 export declare const resolveExternalOutputMount: (workspace: string, workdir?: string | null, defaultOutputDir?: string | null) => ExternalOutputMount;
 export declare const resolveSummarizationMiddlewareSettings: (config: WingmanConfigType) => SummarizationMiddlewareSettings | null;
@@ -128,19 +131,12 @@ export declare const selectStreamingFallbackText: (previousMessages: Array<{
 export declare const detectStreamErrorMessage: (chunk: unknown) => string | undefined;
 export declare const trackRootLangGraphRunId: (currentRootLangGraphRunId: string | undefined, chunk: unknown) => string | undefined;
 export declare const isRootLangGraphTerminalEvent: (chunk: unknown, rootLangGraphRunId?: string) => boolean;
-type StreamingCompletionOutcome = {
-    status: "blocked";
-    reason: "stream_error" | "empty_stream_response";
-    message: string;
-} | {
-    status: "ok";
-};
-type EvaluateStreamingCompletionInput = {
-    sawAssistantText: boolean;
-    fallbackText?: string;
-    streamErrorMessage?: string;
-};
-export declare const evaluateStreamingCompletion: (input: EvaluateStreamingCompletionInput) => StreamingCompletionOutcome;
+export declare const emitCompletionAndContinuePostProcessing: (input: {
+    outputManager: Pick<OutputManager, "emitAgentComplete">;
+    result: unknown;
+    postProcess?: () => Promise<void>;
+    logger?: Pick<Logger, "debug">;
+}) => void;
 export declare class AgentInvoker {
     private loader;
     private outputManager;
@@ -170,6 +166,7 @@ export declare class AgentInvoker {
         name: string;
         description: string;
     }>;
+    private materializeSessionImages;
 }
 export declare function buildUserContent(prompt: string, attachments?: MediaAttachment[], model?: unknown): string | UserContentPart[];
 export {};

package/dist/cli/core/agentInvoker.js

@@ -24,6 +24,9 @@ function _define_property(obj, key, value) {
 }
 const WORKDIR_VIRTUAL_PATH = "/workdir/";
 const OUTPUT_VIRTUAL_PATH = "/output/";
+const AGENTS_MEMORY_VIRTUAL_PATHS = [
+    "/AGENTS.md"
+];
 const DEFAULT_DEEPAGENT_MODEL = "claude-sonnet-4-5-20250929";
 const isPathWithinRoot = (targetPath, rootPath)=>{
     const normalizedTarget = normalize(targetPath);
@@ -35,6 +38,11 @@ const resolveExecutionWorkspace = (workspace, workdir)=>{
     if (isAbsolute(workdir)) return normalize(workdir);
     return normalize(join(workspace, workdir));
 };
+const resolveAgentExecutionWorkspace = (workspace, workdir, defaultOutputDir)=>{
+    const preferredWorkdir = workdir || defaultOutputDir || null;
+    return resolveExecutionWorkspace(workspace, preferredWorkdir);
+};
+const resolveAgentMemorySources = (executionWorkspace)=>AGENTS_MEMORY_VIRTUAL_PATHS.filter((memoryPath)=>existsSync(join(executionWorkspace, memoryPath.replace(/^\/+/, ""))));
 const toWorkspaceAliasVirtualPath = (absolutePath)=>{
     const normalized = normalize(absolutePath);
     if (!isAbsolute(normalized)) return null;
@@ -238,24 +246,19 @@ const trackRootLangGraphRunId = (currentRootLangGraphRunId, chunk)=>{
     return extractEventRunId(eventRecord) || currentRootLangGraphRunId;
 };
 const isRootLangGraphTerminalEvent = (chunk, rootLangGraphRunId)=>{
-    if (!rootLangGraphRunId) return false;
     const eventRecord = extractStreamEventRecord(chunk);
     if (!eventRecord || !isRootLangGraphChainEvent(eventRecord, "on_chain_end")) return false;
+    if (!rootLangGraphRunId) return true;
     const chunkRunId = extractEventRunId(eventRecord);
+    if (!chunkRunId) return true;
     return Boolean(chunkRunId && chunkRunId === rootLangGraphRunId);
 };
-const evaluateStreamingCompletion = (input)=>{
-    if (!input.sawAssistantText && !input.fallbackText) {
-        const message = input.streamErrorMessage ? `Model call failed: ${input.streamErrorMessage}` : "Model completed without a response. Check provider logs for request errors.";
-        return {
-            status: "blocked",
-            reason: input.streamErrorMessage ? "stream_error" : "empty_stream_response",
-            message
-        };
-    }
-    return {
-        status: "ok"
-    };
+const emitCompletionAndContinuePostProcessing = (input)=>{
+    input.outputManager.emitAgentComplete(input.result);
+    if (!input.postProcess) return;
+    input.postProcess().catch((error)=>{
+        input.logger?.debug("Failed post-completion processing for streamed agent response", error);
+    });
 };
 class AgentInvoker {
     findAllAgents() {
@@ -269,14 +272,11 @@ class AgentInvoker {
         let cancellationHandled = false;
         let activeToolName = null;
         let lastToolName = null;
-        let sawAssistantText = false;
-        let streamErrorMessage;
         let rootLangGraphRunId;
-        let preInvocationMessages = null;
         const isCancelled = ()=>options?.signal?.aborted === true;
         try {
             const hookSessionId = sessionId || v4();
-            const executionWorkspace = resolveExecutionWorkspace(this.workspace, this.workdir);
+            const executionWorkspace = resolveAgentExecutionWorkspace(this.workspace, this.workdir, this.defaultOutputDir);
             const effectiveWorkdir = this.workdir ? executionWorkspace : null;
             const loader = new AgentLoader(this.configDir, this.workspace, this.wingmanConfig, executionWorkspace, {
                 terminalOwnerId: `${agentName}:${hookSessionId}`,
@@ -295,7 +295,9 @@ class AgentInvoker {
             if (targetAgent.mcpUseGlobal && this.wingmanConfig.mcp) mcpConfigs.push(this.wingmanConfig.mcp);
             if (mcpConfigs.length > 0) {
                 this.logger.debug("Initializing MCP client for agent invocation");
-                this.mcpManager = new MCPClientManager(mcpConfigs, this.logger);
+                this.mcpManager = new MCPClientManager(mcpConfigs, this.logger, {
+                    executionWorkspace
+                });
                 await this.mcpManager.initialize();
                 const mcpTools = await this.mcpManager.getTools();
                 if (mcpTools.length > 0) {
@@ -312,6 +314,7 @@ class AgentInvoker {
             const normalizedSkillsDirectory = skillsDirectory.replace(/^\/+|\/+$/g, "");
             const skillsVirtualPath = `/${normalizedSkillsDirectory}/`;
             const outputMount = resolveExternalOutputMount(executionWorkspace, effectiveWorkdir, this.defaultOutputDir);
+            const memorySources = resolveAgentMemorySources(executionWorkspace);
             const middleware = [
                 mediaCompatibilityMiddleware({
                     model: targetAgent.model
@@ -396,6 +399,7 @@ class AgentInvoker {
                 middleware: middleware,
                 interruptOn: hitlSettings?.interruptOn,
                 skills: skillsSources,
+                memory: memorySources,
                 subagents: targetAgent.subagents || [],
                 checkpointer: checkpointer
             });
@@ -404,15 +408,6 @@ class AgentInvoker {
             const userContent = buildUserContent(prompt, attachments, targetAgent.model);
             if (this.sessionManager && sessionId) {
                 this.logger.debug(`Using streaming with session: ${sessionId}`);
-                try {
-                    const messages = await this.sessionManager.listMessages(sessionId);
-                    preInvocationMessages = messages.map((message)=>({
-                            role: message.role,
-                            content: message.content
-                        }));
-                } catch (stateError) {
-                    this.logger.debug("Failed to capture pre-invocation session state", stateError);
-                }
                 const stream = await standaloneAgent.streamEvents({
                     messages: [
                         {
@@ -430,8 +425,6 @@ class AgentInvoker {
                 });
                 for await (const chunk of stream){
                     rootLangGraphRunId = trackRootLangGraphRunId(rootLangGraphRunId, chunk);
-                    if (!sawAssistantText && chunkHasAssistantText(chunk)) sawAssistantText = true;
-                    if (!streamErrorMessage) streamErrorMessage = detectStreamErrorMessage(chunk);
                     const toolEvent = detectToolEventContext(chunk);
                     if (toolEvent) {
                         lastToolName = toolEvent.toolName;
@@ -461,37 +454,15 @@ class AgentInvoker {
                         cancelled: true
                     };
                 }
-                let fallbackText;
-                if (!sawAssistantText && this.sessionManager && sessionId && preInvocationMessages) try {
-                    const sessionMessages = await this.sessionManager.listMessages(sessionId);
-                    fallbackText = selectStreamingFallbackText(preInvocationMessages, sessionMessages.map((message)=>({
-                            role: message.role,
-                            content: message.content
-                        })));
-                } catch (stateError) {
-                    this.logger.debug("Failed to derive streaming fallback text from session state", stateError);
-                }
-                const completionOutcome = evaluateStreamingCompletion({
-                    sawAssistantText,
-                    fallbackText,
-                    streamErrorMessage
-                });
-                if ("blocked" === completionOutcome.status) {
-                    this.logger.warn(completionOutcome.message);
-                    this.outputManager.emitAgentError(completionOutcome.message);
-                    return {
-                        blocked: true,
-                        reason: completionOutcome.reason
-                    };
-                }
-                this.logger.info("Agent streaming completed successfully", {
-                    usedFallbackText: Boolean(fallbackText)
-                });
-                this.outputManager.emitAgentComplete({
-                    streaming: true,
-                    ...fallbackText ? {
-                        fallbackText
-                    } : {}
+                this.logger.info("Agent streaming completed successfully");
+                const completionPayload = {
+                    streaming: true
+                };
+                emitCompletionAndContinuePostProcessing({
+                    outputManager: this.outputManager,
+                    result: completionPayload,
+                    postProcess: ()=>this.materializeSessionImages(sessionId),
+                    logger: this.logger
                 });
                 return {
                     streaming: true
@@ -527,7 +498,12 @@ class AgentInvoker {
                     };
                 }
                 this.logger.info("Agent completed successfully");
-                this.outputManager.emitAgentComplete(result);
+                emitCompletionAndContinuePostProcessing({
+                    outputManager: this.outputManager,
+                    result,
+                    postProcess: ()=>this.materializeSessionImages(sessionId),
+                    logger: this.logger
+                });
                 return result;
             }
         } catch (error) {
@@ -559,6 +535,10 @@ class AgentInvoker {
                 description: a.description
             }));
     }
+    async materializeSessionImages(sessionId) {
+        if (!this.sessionManager || !sessionId) return;
+        await this.sessionManager.listMessages(sessionId);
+    }
     constructor(options){
         _define_property(this, "loader", void 0);
         _define_property(this, "outputManager", void 0);
@@ -769,4 +749,4 @@ function buildAttachmentPreview(attachments) {
     if (hasImage) return "[image]";
     return "";
 }
-export { AgentInvoker, OUTPUT_VIRTUAL_PATH, WORKDIR_VIRTUAL_PATH, buildUserContent, chunkHasAssistantText, configureDeepAgentSummarizationMiddleware, detectStreamErrorMessage, detectToolEventContext, evaluateStreamingCompletion, isRootLangGraphTerminalEvent, resolveExecutionWorkspace, resolveExternalOutputMount, resolveHumanInTheLoopSettings, resolveModelRetryMiddlewareSettings, resolveSummarizationMiddlewareSettings, resolveToolRetryMiddlewareSettings, selectStreamingFallbackText, toWorkspaceAliasVirtualPath, trackRootLangGraphRunId };
+export { AGENTS_MEMORY_VIRTUAL_PATHS, AgentInvoker, OUTPUT_VIRTUAL_PATH, WORKDIR_VIRTUAL_PATH, buildUserContent, chunkHasAssistantText, configureDeepAgentSummarizationMiddleware, detectStreamErrorMessage, detectToolEventContext, emitCompletionAndContinuePostProcessing, isRootLangGraphTerminalEvent, resolveAgentExecutionWorkspace, resolveAgentMemorySources, resolveExecutionWorkspace, resolveExternalOutputMount, resolveHumanInTheLoopSettings, resolveModelRetryMiddlewareSettings, resolveSummarizationMiddlewareSettings, resolveToolRetryMiddlewareSettings, selectStreamingFallbackText, toWorkspaceAliasVirtualPath, trackRootLangGraphRunId };