@zds-ai/cli 0.1.4 → 0.1.6

package/dist/agent/grok-agent.js

@@ -5,6 +5,7 @@ import { ChatHistoryManager } from "../utils/chat-history-manager.js";
 import { logApiError } from "../utils/error-logger.js";
 import { parseImagesFromMessage, hasImageReferences } from "../utils/image-encoder.js";
 import { getTextContent } from "../utils/content-utils.js";
+import { Variable } from "./prompt-variables.js";
 import fs from "fs";
 import { TextEditorTool, MorphEditorTool, ZshTool, ConfirmationTool, SearchTool, EnvTool, IntrospectTool, ClearCacheTool, CharacterTool, TaskTool, InternetTool, ImageTool, FileConversionTool, RestartTool } from "../tools/index.js";
 import { EventEmitter } from "events";
@@ -109,6 +110,7 @@ export class GrokAgent extends EventEmitter {
     apiKeyEnvVar = "GROK_API_KEY";
     pendingContextEditSession = null;
     rephraseState = null;
+    hookPrefillText = null;
     constructor(apiKey, baseURL, model, maxToolRounds, debugLogFile, startupHookOutput, temperature, maxTokens) {
         super();
         const manager = getSettingsManager();
@@ -160,6 +162,7 @@ export class GrokAgent extends EventEmitter {
     }
     startupHookOutput;
     systemPrompt = "Initializing..."; // THE system prompt (always at messages[0])
+    hasRunInstanceHook = false;
     /**
      * Initialize the agent with dynamic system prompt
      * Must be called after construction
@@ -167,35 +170,19 @@ export class GrokAgent extends EventEmitter {
     async initialize() {
         // Build system message
         await this.buildSystemMessage();
-        // Execute instance hook on every startup (fresh or not)
-        const settings = getSettingsManager();
-        const instanceHookPath = settings.getInstanceHook();
-        if (instanceHookPath) {
-            const hookResult = await executeOperationHook(instanceHookPath, "instance", {}, 30000, false, // Instance hook is not mandatory
-            this.getCurrentTokenCount(), this.getMaxContextSize());
-            if (hookResult.approved && hookResult.commands && hookResult.commands.length > 0) {
-                // Apply hook commands (ENV, TOOL_RESULT, MODEL, SYSTEM)
-                await this.processHookResult(hookResult);
-            }
-        }
     }
     /**
      * Build/rebuild the system message with current tool availability
      * Updates this.systemPrompt which is always used for messages[0]
      */
     async buildSystemMessage() {
-        // Add startup hook output if provided
-        const startupHookSection = this.startupHookOutput
-            ? `${this.startupHookOutput}\n`
-            : "";
         // Generate dynamic tool list using introspect tool
         const toolsResult = await this.introspect.introspect("tools");
         const toolsSection = toolsResult.success ? toolsResult.output : "Tools: Unknown";
+        // Set APP:TOOLS variable
+        Variable.set("APP:TOOLS", toolsSection);
         // Build THE system prompt
-        this.systemPrompt = `${startupHookSection}
-${toolsSection}
-
-Current working directory: ${process.cwd()}`;
+        this.systemPrompt = Variable.renderFull('SYSTEM');
         // Update messages[0] with the system prompt
         this.messages[0] = {
             role: "system",
@@ -355,16 +342,27 @@ Current working directory: ${process.cwd()}`;
         let isSystemRephrase = false;
         let messageToSend = message;
         let messageType = "user";
+        let prefillText;
         if (message.startsWith("/system rephrase")) {
             isRephraseCommand = true;
             isSystemRephrase = true;
             messageToSend = message.substring(8).trim(); // Strip "/system " (8 chars including space)
             messageType = "system";
+            // Extract prefill text after "/system rephrase "
+            const prefillMatch = message.match(/^\/system rephrase\s+(.+)$/);
+            if (prefillMatch) {
+                prefillText = prefillMatch[1];
+            }
         }
         else if (message.startsWith("/rephrase")) {
             isRephraseCommand = true;
             messageToSend = message; // Keep full text including "/rephrase"
             messageType = "user";
+            // Extract prefill text after "/rephrase "
+            const prefillMatch = message.match(/^\/rephrase\s+(.+)$/);
+            if (prefillMatch) {
+                prefillText = prefillMatch[1];
+            }
         }
         // If this is a rephrase command, find the last assistant message
         if (isRephraseCommand) {
@@ -381,7 +379,7 @@ Current working directory: ${process.cwd()}`;
             }
             // Store rephrase state (will be updated with newResponseIndex after response)
             // For now, just mark that we're in rephrase mode
-            this.setRephraseState(lastAssistantIndex, this.chatHistory.length, -1, messageType);
+            this.setRephraseState(lastAssistantIndex, this.chatHistory.length, -1, messageType, prefillText);
         }
         // Before adding the new user message, check if there are incomplete tool calls
         // from a previous interrupted turn. This prevents malformed message sequences
@@ -413,29 +411,83 @@ Current working directory: ${process.cwd()}`;
                 }
             }
         }
+        // Clear one-shot variables
+        Variable.clearOneShot();
+        // Execute instance hook once per session (after first clearOneShot)
+        if (!this.hasRunInstanceHook) {
+            this.hasRunInstanceHook = true;
+            const settings = getSettingsManager();
+            const instanceHookPath = settings.getInstanceHook();
+            if (instanceHookPath) {
+                const hookResult = await executeOperationHook(instanceHookPath, "instance", {}, 30000, false, // Instance hook is not mandatory
+                this.getCurrentTokenCount(), this.getMaxContextSize());
+                if (hookResult.approved && hookResult.commands && hookResult.commands.length > 0) {
+                    // Apply hook commands (ENV, TOOL_RESULT, MODEL, SYSTEM, SET*)
+                    const results = applyHookCommands(hookResult.commands);
+                    // Apply prompt variables from SET* commands
+                    for (const [varName, value] of results.promptVars.entries()) {
+                        Variable.set(varName, value);
+                    }
+                    // Process other hook commands (MODEL, BACKEND, ENV)
+                    await this.processHookCommands(results);
+                    // Add SYSTEM message to messages array if present
+                    if (results.system) {
+                        this.messages.push({
+                            role: 'system',
+                            content: results.system
+                        });
+                    }
+                    // Store prefill text from hook if present
+                    if (results.prefill) {
+                        this.hookPrefillText = results.prefill;
+                    }
+                }
+            }
+        }
+        // Parse images once if present (for both text extraction and later assembly)
+        const parsed = hasImageReferences(messageToSend)
+            ? parseImagesFromMessage(messageToSend)
+            : { text: messageToSend, images: [] };
+        // Set USER:PROMPT variable (text only, images stripped)
+        Variable.set("USER:PROMPT", parsed.text);
+        // Execute prePrompt hook if configured
+        const hookPath = getSettingsManager().getPrePromptHook();
+        if (hookPath) {
+            const hookResult = await executeOperationHook(hookPath, "prePrompt", { USER_MESSAGE: parsed.text }, 30000, false, // prePrompt hook is never mandatory
+            this.getCurrentTokenCount(), this.getMaxContextSize());
+            if (hookResult.approved && hookResult.commands) {
+                const results = applyHookCommands(hookResult.commands);
+                // Set prompt variables from hook output (SET, SET_FILE, SET_TEMP_FILE)
+                for (const [varName, value] of results.promptVars.entries()) {
+                    Variable.set(varName, value);
+                }
+                // Process other hook commands (MODEL, BACKEND, SYSTEM, etc.)
+                await this.processHookCommands(results);
+                // Store prefill text from hook if present
+                if (results.prefill) {
+                    this.hookPrefillText = results.prefill;
+                }
+            }
+        }
+        // Assemble final message from variables
+        const assembledMessage = Variable.renderFull("USER");
         // Add user/system message to conversation
-        // Check for image references and parse if present
         // Note: System messages can only have string content, so images are only supported for user messages
         const supportsVision = this.grokClient.getSupportsVision();
-        let messageContent = messageToSend;
-        if (messageType === "user" && hasImageReferences(messageToSend) && supportsVision) {
-            // Parse images from message (only for user messages)
-            const parsed = parseImagesFromMessage(messageToSend);
-            if (parsed.images.length > 0) {
-                // Construct content array with text and images
-                messageContent = [
-                    { type: "text", text: parsed.text },
-                    ...parsed.images
-                ];
-            }
-            else {
-                // No valid images found (errors will be in parsed.text)
-                messageContent = parsed.text;
-            }
+        let messageContent = assembledMessage;
+        if (messageType === "user" && parsed.images.length > 0 && supportsVision) {
+            // Construct content array with assembled text and images
+            messageContent = [
+                { type: "text", text: assembledMessage },
+                ...parsed.images
+            ];
         }
         const userEntry = {
             type: messageType,
             content: messageContent,
+            originalContent: messageType === "user" ? (parsed.images.length > 0 && supportsVision
+                ? [{ type: "text", text: parsed.text }, ...parsed.images]
+                : parsed.text) : undefined,
             timestamp: new Date(),
         };
         this.chatHistory.push(userEntry);
@@ -453,6 +505,20 @@ Current working directory: ${process.cwd()}`;
         let toolRounds = 0;
         let consecutiveNonToolResponses = 0;
         try {
+            // If this is a rephrase with prefill text, add the assistant message now
+            if (this.rephraseState?.prefillText) {
+                this.messages.push({
+                    role: "assistant",
+                    content: this.rephraseState.prefillText
+                });
+            }
+            // If a hook returned prefill text, add the assistant message now
+            if (this.hookPrefillText) {
+                this.messages.push({
+                    role: "assistant",
+                    content: this.hookPrefillText
+                });
+            }
             // Always fetch tools fresh - getAllGrokTools() handles lazy refresh internally
             const supportsTools = this.grokClient.getSupportsTools();
             let currentResponse = await this.grokClient.chat(this.messages, supportsTools ? await getAllGrokTools() : [], undefined, this.isGrokModel() && this.shouldUseSearchFor(message)
@@ -626,7 +692,16 @@ Current working directory: ${process.cwd()}`;
                 }
                 else {
                     // No tool calls in this response - only add it if there's actual content
-                    const trimmedContent = assistantMessage.content?.trim();
+                    let trimmedContent = assistantMessage.content?.trim();
+                    // If this was a rephrase with prefill, prepend the prefill text to the response
+                    if (trimmedContent && this.rephraseState?.prefillText) {
+                        trimmedContent = this.rephraseState.prefillText + trimmedContent;
+                    }
+                    // If a hook provided prefill, prepend it to the response
+                    if (trimmedContent && this.hookPrefillText) {
+                        trimmedContent = this.hookPrefillText + trimmedContent;
+                        this.hookPrefillText = null; // Clear after use
+                    }
                     if (trimmedContent) {
                         const responseEntry = {
                             type: "assistant",
@@ -642,7 +717,7 @@ Current working directory: ${process.cwd()}`;
                         // Update rephrase state with the new response index
                         if (this.rephraseState && this.rephraseState.newResponseIndex === -1) {
                             const newResponseIndex = this.chatHistory.length - 1;
-                            this.setRephraseState(this.rephraseState.originalAssistantMessageIndex, this.rephraseState.rephraseRequestIndex, newResponseIndex, this.rephraseState.messageType);
+                            this.setRephraseState(this.rephraseState.originalAssistantMessageIndex, this.rephraseState.rephraseRequestIndex, newResponseIndex, this.rephraseState.messageType, this.rephraseState.prefillText);
                         }
                     }
                     // TODO: HACK - This is a temporary fix to prevent duplicate responses.
@@ -691,6 +766,22 @@ Current working directory: ${process.cwd()}`;
             return newEntries;
         }
         catch (error) {
+            // Check if context is too large (413 error when vision already disabled)
+            if (error.message && error.message.startsWith('CONTEXT_TOO_LARGE:')) {
+                const beforeCount = this.chatHistory.length;
+                this.compactContext(20);
+                const afterCount = this.chatHistory.length;
+                const removedCount = beforeCount - afterCount;
+                const compactEntry = {
+                    type: "system",
+                    content: `Context was too large for backend. Automatically compacted: removed ${removedCount} older messages, keeping last 20 messages. Please retry your request.`,
+                    timestamp: new Date(),
+                };
+                this.chatHistory.push(compactEntry);
+                // Mark first message as processed
+                this.firstMessageProcessed = true;
+                return [userEntry, compactEntry];
+            }
             const errorEntry = {
                 type: "assistant",
                 content: `Sorry, I encountered an error: ${error.message}`,
@@ -804,16 +895,27 @@ Current working directory: ${process.cwd()}`;
         let isSystemRephrase = false;
         let messageToSend = message;
         let messageType = "user";
+        let prefillText;
         if (message.startsWith("/system rephrase")) {
             isRephraseCommand = true;
             isSystemRephrase = true;
             messageToSend = message.substring(8).trim(); // Strip "/system " (8 chars including space)
             messageType = "system";
+            // Extract prefill text after "/system rephrase "
+            const prefillMatch = message.match(/^\/system rephrase\s+(.+)$/);
+            if (prefillMatch) {
+                prefillText = prefillMatch[1];
+            }
         }
         else if (message.startsWith("/rephrase")) {
             isRephraseCommand = true;
             messageToSend = message; // Keep full text including "/rephrase"
             messageType = "user";
+            // Extract prefill text after "/rephrase "
+            const prefillMatch = message.match(/^\/rephrase\s+(.+)$/);
+            if (prefillMatch) {
+                prefillText = prefillMatch[1];
+            }
         }
         // If this is a rephrase command, find the last assistant message
         if (isRephraseCommand) {
@@ -830,7 +932,7 @@ Current working directory: ${process.cwd()}`;
             }
             // Store rephrase state (will be updated with newResponseIndex after response)
             // For now, just mark that we're in rephrase mode
-            this.setRephraseState(lastAssistantIndex, this.chatHistory.length, -1, messageType);
+            this.setRephraseState(lastAssistantIndex, this.chatHistory.length, -1, messageType, prefillText);
         }
         // Before adding the new user message, check if there are incomplete tool calls
         // from a previous interrupted turn. This prevents malformed message sequences
@@ -862,29 +964,83 @@ Current working directory: ${process.cwd()}`;
                 }
             }
         }
+        // Clear one-shot variables
+        Variable.clearOneShot();
+        // Execute instance hook once per session (after first clearOneShot)
+        if (!this.hasRunInstanceHook) {
+            this.hasRunInstanceHook = true;
+            const settings = getSettingsManager();
+            const instanceHookPath = settings.getInstanceHook();
+            if (instanceHookPath) {
+                const hookResult = await executeOperationHook(instanceHookPath, "instance", {}, 30000, false, // Instance hook is not mandatory
+                this.getCurrentTokenCount(), this.getMaxContextSize());
+                if (hookResult.approved && hookResult.commands && hookResult.commands.length > 0) {
+                    // Apply hook commands (ENV, TOOL_RESULT, MODEL, SYSTEM, SET*)
+                    const results = applyHookCommands(hookResult.commands);
+                    // Apply prompt variables from SET* commands
+                    for (const [varName, value] of results.promptVars.entries()) {
+                        Variable.set(varName, value);
+                    }
+                    // Process other hook commands (MODEL, BACKEND, ENV)
+                    await this.processHookCommands(results);
+                    // Add SYSTEM message to messages array if present
+                    if (results.system) {
+                        this.messages.push({
+                            role: 'system',
+                            content: results.system
+                        });
+                    }
+                    // Store prefill text from hook if present
+                    if (results.prefill) {
+                        this.hookPrefillText = results.prefill;
+                    }
+                }
+            }
+        }
+        // Parse images once if present (for both text extraction and later assembly)
+        const parsed = hasImageReferences(messageToSend)
+            ? parseImagesFromMessage(messageToSend)
+            : { text: messageToSend, images: [] };
+        // Set USER:PROMPT variable (text only, images stripped)
+        Variable.set("USER:PROMPT", parsed.text);
+        // Execute prePrompt hook if configured
+        const hookPath = getSettingsManager().getPrePromptHook();
+        if (hookPath) {
+            const hookResult = await executeOperationHook(hookPath, "prePrompt", { USER_MESSAGE: parsed.text }, 30000, false, // prePrompt hook is never mandatory
+            this.getCurrentTokenCount(), this.getMaxContextSize());
+            if (hookResult.approved && hookResult.commands) {
+                const results = applyHookCommands(hookResult.commands);
+                // Set prompt variables from hook output (SET, SET_FILE, SET_TEMP_FILE)
+                for (const [varName, value] of results.promptVars.entries()) {
+                    Variable.set(varName, value);
+                }
+                // Process other hook commands (MODEL, BACKEND, SYSTEM, etc.)
+                await this.processHookCommands(results);
+                // Store prefill text from hook if present
+                if (results.prefill) {
+                    this.hookPrefillText = results.prefill;
+                }
+            }
+        }
+        // Assemble final message from variables
+        const assembledMessage = Variable.renderFull("USER");
         // Add user/system message to both API conversation and chat history
-        // Check for image references and parse if present
         // Note: System messages can only have string content, so images are only supported for user messages
         const supportsVision = this.grokClient.getSupportsVision();
-        let messageContent = messageToSend;
-        if (messageType === "user" && hasImageReferences(messageToSend) && supportsVision) {
-            // Parse images from message (only for user messages)
-            const parsed = parseImagesFromMessage(messageToSend);
-            if (parsed.images.length > 0) {
-                // Construct content array with text and images
-                messageContent = [
-                    { type: "text", text: parsed.text },
-                    ...parsed.images
-                ];
-            }
-            else {
-                // No valid images found (errors will be in parsed.text)
-                messageContent = parsed.text;
-            }
+        let messageContent = assembledMessage;
+        if (messageType === "user" && parsed.images.length > 0 && supportsVision) {
+            // Construct content array with assembled text and images
+            messageContent = [
+                { type: "text", text: assembledMessage },
+                ...parsed.images
+            ];
         }
         const userEntry = {
             type: messageType,
             content: messageContent,
+            originalContent: messageType === "user" ? (parsed.images.length > 0 && supportsVision
+                ? [{ type: "text", text: parsed.text }, ...parsed.images]
+                : parsed.text) : undefined,
             timestamp: new Date(),
         };
         this.chatHistory.push(userEntry);
@@ -894,7 +1050,7 @@ Current working directory: ${process.cwd()}`;
         }
         else {
             // System messages must have string content only
-            this.messages.push({ role: "system", content: typeof messageContent === "string" ? messageContent : messageToSend });
+            this.messages.push({ role: "system", content: typeof messageContent === "string" ? messageContent : assembledMessage });
         }
         await this.emitContextChange();
         // Yield user message so UI can display it immediately
@@ -902,6 +1058,20 @@ Current working directory: ${process.cwd()}`;
             type: "user_message",
             userEntry: userEntry,
         };
+        // If this is a rephrase with prefill text, add the assistant message now
+        if (this.rephraseState?.prefillText) {
+            this.messages.push({
+                role: "assistant",
+                content: this.rephraseState.prefillText
+            });
+        }
+        // If a hook returned prefill text, add the assistant message now
+        if (this.hookPrefillText) {
+            this.messages.push({
+                role: "assistant",
+                content: this.hookPrefillText
+            });
+        }
         // Calculate input tokens
         let inputTokens = this.tokenCounter.countMessageTokens(this.messages);
         yield {
@@ -940,6 +1110,23 @@ Current working directory: ${process.cwd()}`;
                 let tool_calls_yielded = false;
                 let streamFinished = false;
                 let insideThinkTag = false;
+                // If this is a rephrase with prefill, yield the prefill text first and add to accumulated content
+                if (this.rephraseState?.prefillText) {
+                    yield {
+                        type: "content",
+                        content: this.rephraseState.prefillText,
+                    };
+                    accumulatedContent = this.rephraseState.prefillText;
+                }
+                // If a hook provided prefill, yield it first and add to accumulated content
+                if (this.hookPrefillText) {
+                    yield {
+                        type: "content",
+                        content: this.hookPrefillText,
+                    };
+                    accumulatedContent = this.hookPrefillText;
+                    this.hookPrefillText = null; // Clear after use
+                }
                 try {
                     for await (const chunk of stream) {
                         // Check for cancellation in the streaming loop
@@ -1237,6 +1424,25 @@ Current working directory: ${process.cwd()}`;
                 yield { type: "done" };
                 return;
             }
+            // Check if context is too large (413 error when vision already disabled)
+            if (error.message && error.message.startsWith('CONTEXT_TOO_LARGE:')) {
+                const beforeCount = this.chatHistory.length;
+                this.compactContext(20);
+                const afterCount = this.chatHistory.length;
+                const removedCount = beforeCount - afterCount;
+                const compactEntry = {
+                    type: "system",
+                    content: `Context was too large for backend. Automatically compacted: removed ${removedCount} older messages, keeping last 20 messages. Please retry your request.`,
+                    timestamp: new Date(),
+                };
+                this.chatHistory.push(compactEntry);
+                yield {
+                    type: "content",
+                    content: getTextContent(compactEntry.content),
+                };
+                yield { type: "done" };
+                return;
+            }
             const errorEntry = {
                 type: "assistant",
                 content: `Sorry, I encountered an error: ${error.message}`,
@@ -1496,7 +1702,7 @@ Current working directory: ${process.cwd()}`;
                 case "generateImage":
                     return await this.imageTool.generateImage(args.prompt, args.negativePrompt, args.width, args.height, args.model, args.sampler, args.configScale, args.numSteps, args.nsfw, args.name, args.move, args.seed);
                 case "captionImage":
-                    return await this.imageTool.captionImage(args.filename, args.prompt);
+                    return await this.imageTool.captionImage(args.filename, args.backend);
                 case "pngInfo":
                     return await this.imageTool.pngInfo(args.filename);
                 case "listImageModels":
@@ -1683,8 +1889,8 @@ Current working directory: ${process.cwd()}`;
     clearPendingContextEditSession() {
         this.pendingContextEditSession = null;
     }
-    setRephraseState(originalAssistantMessageIndex, rephraseRequestIndex, newResponseIndex, messageType) {
-        this.rephraseState = { originalAssistantMessageIndex, rephraseRequestIndex, newResponseIndex, messageType };
+    setRephraseState(originalAssistantMessageIndex, rephraseRequestIndex, newResponseIndex, messageType, prefillText) {
+        this.rephraseState = { originalAssistantMessageIndex, rephraseRequestIndex, newResponseIndex, messageType, prefillText };
     }
     getRephraseState() {
         return this.rephraseState;
@@ -2038,6 +2244,8 @@ Current working directory: ${process.cwd()}`;
     }
     setModel(model) {
         this.grokClient.setModel(model);
+        // Reset supportsVision flag for new model
+        this.grokClient.setSupportsVision(true);
         // Update token counter for new model
         this.tokenCounter.dispose();
         this.tokenCounter = createTokenCounter(model);
@@ -2412,6 +2620,7 @@ Current working directory: ${process.cwd()}`;
             baseUrl: this.grokClient.getBaseURL(),
             apiKeyEnvVar: this.apiKeyEnvVar,
             model: this.getCurrentModel(),
+            supportsVision: this.grokClient.getSupportsVision(),
         };
     }
     /**
@@ -2425,10 +2634,15 @@ Current working directory: ${process.cwd()}`;
         // Restore cwd early (hooks may need correct working directory)
         if (state.cwd) {
             try {
-                process.chdir(state.cwd);
+                const fs = await import('fs');
+                // Only attempt to change directory if it exists
+                if (fs.existsSync(state.cwd)) {
+                    process.chdir(state.cwd);
+                }
+                // Silently skip if directory doesn't exist (common in containerized environments)
             }
             catch (error) {
-                console.warn(`Failed to restore working directory to ${state.cwd}:`, error);
+                // Silently skip on any error - working directory restoration is non-critical
             }
         }
         // Restore backend/baseUrl/apiKeyEnvVar/model if present (creates initial client)
@@ -2441,6 +2655,10 @@ Current working directory: ${process.cwd()}`;
                     const model = state.model || this.getCurrentModel();
                     this.grokClient = new GrokClient(apiKey, model, state.baseUrl, state.backend);
                     this.apiKeyEnvVar = state.apiKeyEnvVar;
+                    // Restore supportsVision flag if present
+                    if (state.supportsVision !== undefined) {
+                        this.grokClient.setSupportsVision(state.supportsVision);
+                    }
                     // Reinitialize MCP servers when restoring session
                     try {
                         const config = loadMCPConfig();
@@ -2469,30 +2687,122 @@ Current working directory: ${process.cwd()}`;
         // Restore persona (hook may change backend/model and sets env vars)
         if (state.persona) {
             try {
-                await this.setPersona(state.persona, state.personaColor);
+                const result = await this.setPersona(state.persona, state.personaColor);
+                if (!result.success) {
+                    // If persona hook failed (e.g., backend test failed), still set the persona values
+                    // but don't change backend/model. This prevents losing persona state on transitory errors.
+                    console.warn(`Persona hook failed, setting persona without backend change: ${result.error}`);
+                    this.persona = state.persona;
+                    this.personaColor = state.personaColor;
+                    process.env.ZDS_AI_AGENT_PERSONA = state.persona;
+                }
             }
             catch (error) {
                 console.warn(`Failed to restore persona "${state.persona}":`, error);
+                // Still set persona values even if hook crashed
+                this.persona = state.persona;
+                this.personaColor = state.personaColor;
+                process.env.ZDS_AI_AGENT_PERSONA = state.persona;
             }
         }
         // Restore mood (hook sets env vars)
         if (state.mood) {
             try {
-                await this.setMood(state.mood, state.moodColor);
+                const result = await this.setMood(state.mood, state.moodColor);
+                if (!result.success) {
+                    // If mood hook failed (e.g., backend test failed), still set the mood values
+                    // but don't change backend/model. This prevents losing mood state on transitory errors.
+                    console.warn(`Mood hook failed, setting mood without backend change: ${result.error}`);
+                    this.mood = state.mood;
+                    this.moodColor = state.moodColor;
+                    process.env.ZDS_AI_AGENT_MOOD = state.mood;
+                }
             }
             catch (error) {
                 console.warn(`Failed to restore mood "${state.mood}":`, error);
+                // Still set mood values even if hook crashed
+                this.mood = state.mood;
+                this.moodColor = state.moodColor;
+                process.env.ZDS_AI_AGENT_MOOD = state.mood;
             }
         }
         // Restore active task (hook sets env vars)
         if (state.activeTask) {
             try {
-                await this.startActiveTask(state.activeTask, state.activeTaskAction, state.activeTaskColor);
+                const result = await this.startActiveTask(state.activeTask, state.activeTaskAction, state.activeTaskColor);
+                if (!result.success) {
+                    // If task hook failed (e.g., backend test failed), still set the task values
+                    // but don't change backend/model. This prevents losing task state on transitory errors.
+                    console.warn(`Task hook failed, setting active task without backend change: ${result.error}`);
+                    this.activeTask = state.activeTask;
+                    this.activeTaskAction = state.activeTaskAction;
+                    this.activeTaskColor = state.activeTaskColor;
+                    process.env.ZDS_AI_AGENT_ACTIVE_TASK = state.activeTask;
+                    process.env.ZDS_AI_AGENT_ACTIVE_TASK_ACTION = state.activeTaskAction;
+                }
             }
             catch (error) {
                 console.warn(`Failed to restore active task "${state.activeTask}":`, error);
+                // Still set task values even if hook crashed
+                this.activeTask = state.activeTask;
+                this.activeTaskAction = state.activeTaskAction;
+                this.activeTaskColor = state.activeTaskColor;
+                process.env.ZDS_AI_AGENT_ACTIVE_TASK = state.activeTask;
+                process.env.ZDS_AI_AGENT_ACTIVE_TASK_ACTION = state.activeTaskAction;
+            }
+        }
+    }
+    /**
+     * Compact conversation context by keeping system prompt and last N messages
+     * Reduces context size when it grows too large for backend to handle
+     * @returns Number of messages removed
+     */
+    compactContext(keepLastMessages = 20) {
+        if (this.chatHistory.length <= keepLastMessages) {
+            // Nothing to compact
+            return 0;
+        }
+        const removedCount = this.chatHistory.length - keepLastMessages;
+        const keptMessages = this.chatHistory.slice(-keepLastMessages);
+        // Clear both arrays
+        this.chatHistory = keptMessages;
+        this.messages = [];
+        // Add system message noting the compaction
+        const compactionNote = {
+            type: 'system',
+            content: `Context compacted: removed ${removedCount} older messages, keeping last ${keepLastMessages} messages.`,
+            timestamp: new Date()
+        };
+        this.chatHistory.push(compactionNote);
+        // Rebuild this.messages from compacted chatHistory
+        for (const entry of this.chatHistory) {
+            if (entry.type === 'system') {
+                this.messages.push({
+                    role: 'system',
+                    content: entry.content
+                });
+            }
+            else if (entry.type === 'user') {
+                this.messages.push({
+                    role: 'user',
+                    content: entry.content
+                });
+            }
+            else if (entry.type === 'assistant') {
+                this.messages.push({
+                    role: 'assistant',
+                    content: entry.content
+                });
+            }
+            else if (entry.type === 'tool_result') {
+                this.messages.push({
+                    role: 'tool',
+                    tool_call_id: entry.toolResult.output || '',
+                    content: JSON.stringify(entry.toolResult)
+                });
             }
         }
+        return removedCount;
     }
     /**
      * Get all tool instances and their class names for display purposes