@ridit/lens 0.3.7 → 0.3.8

package/src/components/chat/hooks/useChat.ts

@@ -31,7 +31,16 @@ import {
   type ChatResult,
 } from "../../../utils/chat";
 
-export function useChat(repoPath: string) {
+function hasUnclosedToolTag(text: string): boolean {
+  for (const tag of registry.names()) {
+    if (text.includes(`<${tag}>`) && !text.includes(`</${tag}>`)) return true;
+  }
+  const fences = text.match(/```/g);
+  if (fences && fences.length % 2 !== 0) return true;
+  return false;
+}
+
+export function useChat(repoPath: string, autoForce = false) {
   const [stage, setStage] = useState<ChatStage>({ type: "picking-provider" });
   const [committed, setCommitted] = useState<Message[]>([]);
   const [provider, setProvider] = useState<Provider | null>(null);
@@ -41,8 +50,8 @@ export function useChat(repoPath: string) {
   const [clonedUrls, setClonedUrls] = useState<Set<string>>(new Set());
   const [showTimeline, setShowTimeline] = useState(false);
   const [showReview, setShowReview] = useState(false);
-  const [autoApprove, setAutoApprove] = useState(false);
-  const [forceApprove, setForceApprove] = useState(false);
+  const [autoApprove, setAutoApprove] = useState(autoForce);
+  const [forceApprove, setForceApprove] = useState(autoForce);
   const [showForceWarning, setShowForceWarning] = useState(false);
   const [chatName, setChatName] = useState<string | null>(null);
   const [recentChats, setRecentChats] = useState<string[]>([]);
@@ -53,6 +62,7 @@ export function useChat(repoPath: string) {
   const abortControllerRef = useRef<AbortController | null>(null);
   const toolResultCache = useRef<Map<string, string>>(new Map());
   const batchApprovedRef = useRef(false);
+  const forceApproveRef = useRef(autoForce);
 
   const updateChatName = (name: string) => {
     chatNameRef.current = name;
@@ -65,6 +75,9 @@ export function useChat(repoPath: string) {
   React.useEffect(() => {
     systemPromptRef.current = systemPrompt;
   }, [systemPrompt]);
+  React.useEffect(() => {
+    forceApproveRef.current = forceApprove;
+  }, [forceApprove]);
 
   React.useEffect(() => {
     const chats = listChats(repoPath);
@@ -77,6 +90,13 @@ export function useChat(repoPath: string) {
     }
   }, [allMessages]);
 
+  const pushMsg = (msg: Message, currentAll: Message[]): Message[] => {
+    const next = [...currentAll, msg];
+    setAllMessages(next);
+    setCommitted((prev) => [...prev, msg]);
+    return next;
+  };
+
   const handleError = (currentAll: Message[]) => (err: unknown) => {
     batchApprovedRef.current = false;
     if (err instanceof Error && err.name === "AbortError") {
@@ -88,41 +108,132 @@ export function useChat(repoPath: string) {
       content: `Error: ${err instanceof Error ? err.message : "Something went wrong"}`,
       type: "text",
     };
-    setAllMessages([...currentAll, errMsg]);
-    setCommitted((prev) => [...prev, errMsg]);
+    pushMsg(errMsg, currentAll);
     setStage({ type: "idle" });
   };
 
-  const MAX_AUTO_CONTINUES = 3;
+  const callNext = async (
+    messages: Message[],
+    signal: AbortSignal,
+    maxRetries = 3,
+  ): Promise<ChatResult> => {
+    const currentProvider = providerRef.current;
+    const currentSystemPrompt = systemPromptRef.current;
+
+    if (!currentProvider || signal.aborted)
+      return { text: "", truncated: false };
 
-  function isLikelyTruncated(text: string): boolean {
-    // Check unclosed XML tool tags (dynamic — includes addon tools)
-    for (const tag of registry.names()) {
-      if (text.includes(`<${tag}>`) && !text.includes(`</${tag}>`))
-        return true;
+    let currentMessages = messages;
+
+    for (let attempt = 0; attempt <= maxRetries; attempt++) {
+      if (signal.aborted) return { text: "", truncated: false };
+
+      const result = await callChat(
+        currentProvider,
+        currentSystemPrompt,
+        currentMessages,
+        signal,
+      );
+
+      if (result.text.trim()) return result;
+
+      if (attempt < maxRetries) {
+        const nudge: Message = {
+          role: "assistant",
+          content: `(model stalled — retrying ${attempt + 1}/${maxRetries})`,
+          type: "text",
+        };
+        setCommitted((prev) => [...prev, nudge]);
+        currentMessages = [
+          ...currentMessages,
+          {
+            role: "user",
+            content: "Please continue your response.",
+            type: "text",
+          },
+        ];
+      }
     }
-    // Check unclosed fenced code blocks (```tool\n... without closing ```)
-    const fences = text.match(/```/g);
-    if (fences && fences.length % 2 !== 0) return true;
-    return false;
-  }
 
-  const processResponse = (
+    return { text: "", truncated: false };
+  };
+
+  const MAX_CONTINUATIONS = 3;
+
+  const handleTruncation = async (
+    raw: string,
+    currentAll: Message[],
+    signal: AbortSignal,
+    depth: number,
+  ): Promise<{ text: string; messages: Message[] } | null> => {
+    if (depth >= MAX_CONTINUATIONS) return null;
+
+    const truncNotice: Message = {
+      role: "assistant",
+      content: `(response cut off — continuing ${depth + 1}/${MAX_CONTINUATIONS}…)`,
+      type: "text",
+    };
+    setCommitted((prev) => [...prev, truncNotice]);
+
+    const partialMsg: Message = {
+      role: "assistant",
+      content: raw,
+      type: "text",
+    };
+    const nudgeMsg: Message = {
+      role: "user",
+      content:
+        "Your response was cut off. Please continue exactly from where you left off.",
+      type: "text",
+    };
+    const withContext = [...currentAll, partialMsg, nudgeMsg];
+
+    const result = await callNext(withContext, signal);
+    return { text: result.text ?? "", messages: withContext };
+  };
+
+  const processMemoryTags = (raw: string): string => {
+    const addMatches = [
+      ...raw.matchAll(/<memory-add>([\s\S]*?)<\/memory-add>/g),
+    ];
+    const delMatches = [
+      ...raw.matchAll(/<memory-delete>([\s\S]*?)<\/memory-delete>/g),
+    ];
+    for (const m of addMatches) {
+      const content = m[1]!.trim();
+      if (content) addMemory(content, repoPath);
+    }
+    for (const m of delMatches) {
+      const id = m[1]!.trim();
+      if (id) deleteMemory(id, repoPath);
+    }
+    return raw
+      .replace(/<memory-add>[\s\S]*?<\/memory-add>/g, "")
+      .replace(/<memory-delete>[\s\S]*?<\/memory-delete>/g, "")
+      .trim();
+  };
+
+  const processResponse = async (
     raw: string,
     currentAll: Message[],
     signal: AbortSignal,
     truncated = false,
-    continueCount = 0,
-  ) => {
+    continuationDepth = 0,
+  ): Promise<void> => {
     if (signal.aborted) {
       batchApprovedRef.current = false;
       setStage({ type: "idle" });
       return;
     }
 
-    if (truncated || isLikelyTruncated(raw)) {
-      if (continueCount >= MAX_AUTO_CONTINUES) {
-        // Give up after max attempts — show whatever we have
+    if (truncated || hasUnclosedToolTag(raw)) {
+      const cont = await handleTruncation(
+        raw,
+        currentAll,
+        signal,
+        continuationDepth,
+      );
+      if (!cont) {
         batchApprovedRef.current = false;
         const msg: Message = {
           role: "assistant",
@@ -131,110 +242,116 @@ export function useChat(repoPath: string) {
             "(response was empty after multiple continuation attempts)",
           type: "text",
         };
-        setAllMessages([...currentAll, msg]);
-        setCommitted((prev) => [...prev, msg]);
+        pushMsg(msg, currentAll);
         setStage({ type: "idle" });
         return;
       }
+      return processResponse(
+        cont.text,
+        cont.messages,
+        signal,
+        false,
+        continuationDepth + 1,
+      );
+    }
 
-      // Include the partial response so the model knows where it left off
-      const partialMsg: Message = {
-        role: "assistant",
-        content: raw,
-        type: "text",
-      };
-      const nudgeMsg: Message = {
-        role: "user",
-        content:
-          "Your response was cut off. Please continue exactly from where you left off.",
-        type: "text",
-      };
-      const withContext = [...currentAll, partialMsg, nudgeMsg];
+    const cleanRaw = processMemoryTags(raw);
 
-      const truncMsg: Message = {
-        role: "assistant",
-        content: `(response cut off — auto-continuing ${continueCount + 1}/${MAX_AUTO_CONTINUES}…)`,
-        type: "text",
-      };
-      setAllMessages([...currentAll, truncMsg]);
-      setCommitted((prev) => [...prev, truncMsg]);
+    const parsed = parseResponse(cleanRaw);
 
-      const currentProvider = providerRef.current;
-      const currentSystemPrompt = systemPromptRef.current;
+    if (parsed.kind === "text") {
+      batchApprovedRef.current = false;
 
-      if (!currentProvider) {
+      if (!parsed.content.trim()) {
+        const stallMsg: Message = {
+          role: "assistant",
+          content:
+            '(no response — try sending "continue" or start a new message)',
+          type: "text",
+        };
+        pushMsg(stallMsg, currentAll);
         setStage({ type: "idle" });
         return;
       }
 
-      const nextAbort = new AbortController();
-      abortControllerRef.current = nextAbort;
-      setStage({ type: "thinking" });
-      callChat(
-        currentProvider,
-        currentSystemPrompt,
-        withContext,
-        nextAbort.signal,
-      )
-        .then((result: ChatResult) => {
-          if (nextAbort.signal.aborted) return;
-          processResponse(
-            result.text ?? "",
-            withContext,
-            nextAbort.signal,
-            result.truncated,
-            continueCount + 1,
-          );
-        })
-        .catch(handleError(withContext));
-      return;
-    }
+      const msg: Message = {
+        role: "assistant",
+        content: parsed.content,
+        type: "text",
+      };
+      const withMsg = pushMsg(msg, currentAll);
 
-    const memAddMatches = [
-      ...raw.matchAll(/<memory-add>([\s\S]*?)<\/memory-add>/g),
-    ];
-    const memDelMatches = [
-      ...raw.matchAll(/<memory-delete>([\s\S]*?)<\/memory-delete>/g),
-    ];
-    for (const match of memAddMatches) {
-      const content = match[1]!.trim();
-      if (content) addMemory(content, repoPath);
-    }
-    for (const match of memDelMatches) {
-      const id = match[1]!.trim();
-      if (id) deleteMemory(id, repoPath);
+      const lastUserMsg = [...currentAll]
+        .reverse()
+        .find((m) => m.role === "user");
+      const githubUrl = lastUserMsg
+        ? extractGithubUrl(lastUserMsg.content)
+        : null;
+      if (githubUrl && !clonedUrls.has(githubUrl)) {
+        setTimeout(
+          () => setStage({ type: "clone-offer", repoUrl: githubUrl }),
+          80,
+        );
+      } else {
+        setStage({ type: "idle" });
+      }
+      return;
     }
-    const cleanRaw = raw
-      .replace(/<memory-add>[\s\S]*?<\/memory-add>/g, "")
-      .replace(/<memory-delete>[\s\S]*?<\/memory-delete>/g, "")
-      .trim();
-
-    const parsed = parseResponse(cleanRaw);
 
     if (parsed.kind === "changes") {
       batchApprovedRef.current = false;
+
       if (parsed.patches.length === 0) {
         const msg: Message = {
           role: "assistant",
           content: parsed.content,
           type: "text",
         };
-        setAllMessages([...currentAll, msg]);
-        setCommitted((prev) => [...prev, msg]);
+        pushMsg(msg, currentAll);
         setStage({ type: "idle" });
         return;
       }
+
+      const diffLines = buildDiffs(repoPath, parsed.patches);
+
+      if (forceApproveRef.current) {
+        const assistantMsg: Message = {
+          role: "assistant",
+          content: parsed.content,
+          type: "plan",
+          patches: parsed.patches,
+          diffLines,
+          applied: true,
+        };
+        const withAssistant = [...currentAll, assistantMsg];
+        setAllMessages(withAssistant);
+        setCommitted((prev) => [...prev, assistantMsg]);
+        try {
+          applyPatches(repoPath, parsed.patches);
+          logToolCall(
+            "changes",
+            parsed.patches.map((p) => p.path).join(", "),
+            `Applied changes to ${parsed.patches.length} file(s)`,
+            repoPath,
+          );
+        } catch {}
+        continueAfterChanges(withAssistant, parsed.content || "code changes");
+        return;
+      }
+
       const assistantMsg: Message = {
         role: "assistant",
         content: parsed.content,
         type: "plan",
         patches: parsed.patches,
+        diffLines,
         applied: false,
       };
       const withAssistant = [...currentAll, assistantMsg];
       setAllMessages(withAssistant);
+      setCommitted((prev) => [...prev, assistantMsg]);
       setPendingMsgIndex(withAssistant.length - 1);
-      const diffLines = buildDiffs(repoPath, parsed.patches);
+
       setStage({
         type: "preview",
         patches: parsed.patches,
@@ -253,8 +370,7 @@ export function useChat(repoPath: string) {
           content: parsed.content,
           type: "text",
         };
-        setAllMessages([...currentAll, preambleMsg]);
-        setCommitted((prev) => [...prev, preambleMsg]);
+        pushMsg(preambleMsg, currentAll);
       }
       setStage({
         type: "clone-offer",
@@ -264,215 +380,194 @@ export function useChat(repoPath: string) {
       return;
     }
 
-    if (parsed.kind === "text") {
-      batchApprovedRef.current = false;
-
-      if (!parsed.content.trim()) {
-        const stallMsg: Message = {
-          role: "assistant",
-          content:
-            '(no response — the model may have stalled. Try sending a short follow-up like "continue" or start a new message.)',
-          type: "text",
-        };
-        setAllMessages([...currentAll, stallMsg]);
-        setCommitted((prev) => [...prev, stallMsg]);
+    if (parsed.kind === "tool") {
+      const tool = registry.get(parsed.toolName);
+      if (!tool) {
+        batchApprovedRef.current = false;
         setStage({ type: "idle" });
         return;
       }
 
-      const msg: Message = {
-        role: "assistant",
-        content: parsed.content,
-        type: "text",
-      };
-      const withMsg = [...currentAll, msg];
-      setAllMessages(withMsg);
-      setCommitted((prev) => [...prev, msg]);
-      const lastUserMsg = [...currentAll]
-        .reverse()
-        .find((m) => m.role === "user");
-      const githubUrl = lastUserMsg
-        ? extractGithubUrl(lastUserMsg.content)
-        : null;
-      if (githubUrl && !clonedUrls.has(githubUrl)) {
-        setTimeout(
-          () => setStage({ type: "clone-offer", repoUrl: githubUrl }),
-          80,
-        );
-      } else {
-        setStage({ type: "idle" });
+      if (parsed.content) {
+        const preambleMsg: Message = {
+          role: "assistant",
+          content: parsed.content,
+          type: "text",
+        };
+        pushMsg(preambleMsg, currentAll);
       }
-      return;
-    }
-
-    const tool = registry.get(parsed.toolName);
-    if (!tool) {
-      batchApprovedRef.current = false;
-      setStage({ type: "idle" });
-      return;
-    }
 
-    if (parsed.content) {
-      const preambleMsg: Message = {
-        role: "assistant",
-        content: parsed.content,
-        type: "text",
-      };
-      setAllMessages([...currentAll, preambleMsg]);
-      setCommitted((prev) => [...prev, preambleMsg]);
-    }
+      const isSafe = tool.safe ?? false;
+      const remainder = parsed.remainder;
 
-    const remainder = parsed.remainder;
-    const isSafe = tool.safe ?? false;
+      const executeAndContinue = async (approved: boolean): Promise<void> => {
+        if (approved && remainder) batchApprovedRef.current = true;
 
-    const executeAndContinue = async (approved: boolean) => {
-      if (approved && remainder) {
-        batchApprovedRef.current = true;
-      }
+        const currentProvider = providerRef.current;
+        const currentSystemPrompt = systemPromptRef.current;
 
-      const currentProvider = providerRef.current;
-      const currentSystemPrompt = systemPromptRef.current;
-
-      if (!currentProvider) {
-        batchApprovedRef.current = false;
-        setStage({ type: "idle" });
-        return;
-      }
+        if (!currentProvider) {
+          batchApprovedRef.current = false;
+          setStage({ type: "idle" });
+          return;
+        }
 
-      let result = "(denied by user)";
-
-      if (approved) {
-        const cacheKey = isSafe
-          ? `${parsed.toolName}:${parsed.rawInput}`
-          : null;
-        if (cacheKey && toolResultCache.current.has(cacheKey)) {
-          result =
-            toolResultCache.current.get(cacheKey)! +
-            "\n\n[NOTE: This result was already retrieved earlier. Do not request it again.]";
-        } else {
-          try {
-            setStage({ type: "thinking" });
-            const toolResult = await tool.execute(parsed.input, {
-              repoPath,
-              messages: currentAll,
-            });
-            result = toolResult.value;
-            if (cacheKey && toolResult.kind === "text") {
-              toolResultCache.current.set(cacheKey, result);
+        let result = "(denied by user)";
+
+        if (approved) {
+          const cacheKey = isSafe
+            ? `${parsed.toolName}:${parsed.rawInput}`
+            : null;
+          if (cacheKey && toolResultCache.current.has(cacheKey)) {
+            result =
+              toolResultCache.current.get(cacheKey)! +
+              "\n\n[NOTE: This result was already retrieved earlier. Do not request it again.]";
+          } else {
+            try {
+              setStage({ type: "thinking" });
+              const toolResult = await tool.execute(parsed.input, {
+                repoPath,
+                messages: currentAll,
+              });
+              result = toolResult.value;
+              if (cacheKey && toolResult.kind === "text") {
+                toolResultCache.current.set(cacheKey, result);
+              }
+            } catch (err: unknown) {
+              result = `Error: ${err instanceof Error ? err.message : "failed"}`;
             }
-          } catch (err: unknown) {
-            result = `Error: ${err instanceof Error ? err.message : "failed"}`;
           }
         }
-      }
 
-      if (approved && !result.startsWith("Error:")) {
-        logToolCall(
-          parsed.toolName,
-          tool.summariseInput
-            ? String(tool.summariseInput(parsed.input))
-            : parsed.rawInput,
+        if (approved && !result.startsWith("Error:")) {
+          logToolCall(
+            parsed.toolName,
+            tool.summariseInput
+              ? String(tool.summariseInput(parsed.input))
+              : parsed.rawInput,
+            result,
+            repoPath,
+          );
+        }
+
+        const displayContent = tool.summariseInput
+          ? String(tool.summariseInput(parsed.input))
+          : parsed.rawInput;
+
+        const toolMsg: Message = {
+          role: "assistant",
+          type: "tool",
+          toolName: parsed.toolName as any,
+          content: displayContent,
           result,
-          repoPath,
-        );
-      }
+          approved,
+        };
 
-      const displayContent = tool.summariseInput
-        ? String(tool.summariseInput(parsed.input))
-        : parsed.rawInput;
+        const withTool = pushMsg(toolMsg, currentAll);
 
-      const toolMsg: Message = {
-        role: "assistant",
-        type: "tool",
-        toolName: parsed.toolName as any,
-        content: displayContent,
-        result,
-        approved,
-      };
+        if (approved && remainder && remainder.length > 0) {
+          return processResponse(
+            remainder,
+            withTool,
+            signal,
+            false,
+            continuationDepth,
+          );
+        }
 
-      const withTool = [...currentAll, toolMsg];
-      setAllMessages(withTool);
-      setCommitted((prev) => [...prev, toolMsg]);
+        batchApprovedRef.current = false;
 
-      if (approved && remainder && remainder.length > 0) {
-        processResponse(remainder, withTool, signal, truncated, continueCount);
-        return;
-      }
+        const nextAbort = new AbortController();
+        abortControllerRef.current = nextAbort;
+        setStage({ type: "thinking" });
 
-      batchApprovedRef.current = false;
+        try {
+          const nextResult = await callNext(withTool, nextAbort.signal);
 
-      const nextAbort = new AbortController();
-      abortControllerRef.current = nextAbort;
-      setStage({ type: "thinking" });
-
-      const callWithAutoContinue = async (
-        messages: Message[],
-        maxRetries = 3,
-      ): Promise<ChatResult> => {
-        let currentMessages = messages;
-        for (let i = 0; i < maxRetries; i++) {
-          if (nextAbort.signal.aborted)
-            return { text: "", truncated: false };
-          const result = await callChat(
-            currentProvider,
-            currentSystemPrompt,
-            currentMessages,
-            nextAbort.signal,
-          );
-          if (result.text.trim()) return result;
-          const nudgeMsg: Message = {
-            role: "assistant",
-            content: `(model stalled — auto-continuing, attempt ${i + 1}/${maxRetries})`,
-            type: "text",
-          };
-          setCommitted((prev) => [...prev, nudgeMsg]);
-          setAllMessages((prev) => [...prev, nudgeMsg]);
-          currentMessages = [
-            ...currentMessages,
-            {
-              role: "user",
-              content:
-                "Please continue. Provide your response to the previous tool output.",
+          if (nextAbort.signal.aborted) return;
+
+          if (!nextResult.text.trim()) {
+            const stallMsg: Message = {
+              role: "assistant",
+              content: '(model stopped responding — try sending "continue")',
               type: "text",
-            },
-          ];
-        }
-        return { text: "", truncated: false };
-      };
+            };
+            pushMsg(stallMsg, withTool);
+            setStage({ type: "idle" });
+            return;
+          }
 
-      callWithAutoContinue(withTool)
-        .then((result: ChatResult) => {
-          if (nextAbort.signal.aborted) return;
-          processResponse(
-            result.text ?? "",
+          return processResponse(
+            nextResult.text,
             withTool,
             nextAbort.signal,
-            result.truncated,
+            nextResult.truncated,
           );
-        })
-        .catch(handleError(withTool));
-    };
+        } catch (err) {
+          handleError(withTool)(err);
+        }
+      };
+
+      if (forceApprove || isSafe || batchApprovedRef.current) {
+        return executeAndContinue(true);
+      }
+
+      const permLabel = tool.permissionLabel ?? tool.name;
+      const permValue = tool.summariseInput
+        ? String(tool.summariseInput(parsed.input))
+        : parsed.rawInput;
 
-    if (forceApprove || isSafe || batchApprovedRef.current) {
-      executeAndContinue(true);
+      setStage({
+        type: "permission",
+        tool: {
+          type: parsed.toolName as any,
+          _display: permValue,
+          _label: permLabel,
+        } as any,
+        pendingMessages: currentAll,
+        resolve: executeAndContinue,
+      });
+    }
+  };
+
+  const continueAfterChanges = (currentAll: Message[], summary: string) => {
+    if (!providerRef.current) {
+      setStage({ type: "idle" });
       return;
     }
 
-    const permLabel = tool.permissionLabel ?? tool.name;
-    const permValue = tool.summariseInput
-      ? String(tool.summariseInput(parsed.input))
-      : parsed.rawInput;
-
-    setStage({
-      type: "permission",
-      tool: {
-        type: parsed.toolName as any,
-        _display: permValue,
-        _label: permLabel,
-      } as any,
-      pendingMessages: currentAll,
-      resolve: executeAndContinue,
-    });
+    const resultMsg: Message = {
+      role: "assistant",
+      type: "tool",
+      toolName: "changes",
+      content: summary,
+      result: "Changes applied successfully.",
+      approved: true,
+    };
+
+    const withResult = [...currentAll, resultMsg];
+    setAllMessages(withResult);
+    setCommitted((prev) => [...prev, resultMsg]);
+
+    const abort = new AbortController();
+    abortControllerRef.current = abort;
+    setStage({ type: "thinking" });
+
+    callNext(withResult, abort.signal)
+      .then((result) => {
+        if (abort.signal.aborted) return;
+        if (!result.text.trim()) {
+          setStage({ type: "idle" });
+          return;
+        }
+        return processResponse(
+          result.text,
+          withResult,
+          abort.signal,
+          result.truncated,
+        );
+      })
+      .catch(handleError(withResult));
   };
 
   const sendMessage = (
@@ -509,7 +604,6 @@ export function useChat(repoPath: string) {
       /## TOOLS[\s\S]*?(?=\n## (?!TOOLS))/,
       scopedToolsSection + "\n\n",
     );
-
     if (sessionSummary) {
       scopedSystemPrompt = scopedSystemPrompt.replace(
         /## CODEBASE/,
@@ -518,6 +612,7 @@ export function useChat(repoPath: string) {
     }
 
     setStage({ type: "thinking" });
+
     callChat(currentProvider, scopedSystemPrompt, nextAll, abort.signal)
       .then((result: ChatResult) =>
         processResponse(result.text, nextAll, abort.signal, result.truncated),
@@ -529,6 +624,7 @@ export function useChat(repoPath: string) {
     setProvider(p);
     providerRef.current = p;
     setStage({ type: "loading" });
+
     fetchFileTree(repoPath)
       .catch(() => walkDir(repoPath))
       .then((fileTree) => {
@@ -542,13 +638,16 @@ export function useChat(repoPath: string) {
         const prompt =
           buildSystemPrompt(importantFiles, historySummary, toolsSection) +
           lensContext;
+
         setSystemPrompt(prompt);
         systemPromptRef.current = prompt;
+
         const greeting: Message = {
           role: "assistant",
           content: `Welcome to Lens\nCodebase loaded — ${importantFiles.length} files indexed.${historySummary ? "\n\nI have memory of previous actions in this repo." : ""}${lensFile ? "\n\nFound LENS.md — I have context from a previous analysis of this repo." : ""}\nAsk me anything, tell me what to build, share a URL, or ask me to read/write files.\n\nTip: type /timeline to browse commit history.\nTip: ⭐ Star Lens on GitHub — github.com/ridit-jangra/Lens`,
           type: "text",
         };
+
         setCommitted([greeting]);
         setAllMessages([greeting]);
         setStage({ type: "idle" });
@@ -624,6 +723,7 @@ export function useChat(repoPath: string) {
     handleProviderDone,
     abortThinking,
     applyPatchesAndContinue,
+    continueAfterChanges,
     skipPatches,
     processResponse,
     handleError,