add forceReload event on any tool result. (#673)

* add forceReload event on any tool result.

* rename forceReload event

* chore: rename forceReload

* fix: compression stop, only stop the chat when it's compressed and more than 40 messages from the last user message.

* fix: combine comppression and chat suggestion use similar logic.

* enable send if the user dimisses the request to start a new chat

* fix: open chat suggestion box when restoring a large chat.

Author: MarcMcIntosh

refact-agent/gui/src/app/middleware.ts

@@ -13,6 +13,7 @@ import {
   setIsWaitingForResponse,
   upsertToolCall,
   sendCurrentChatToLspAfterToolCallUpdate,
+  chatResponse,
 } from "../features/Chat/Thread";
 import { statisticsApi } from "../services/refact/statistics";
 import { integrationsApi } from "../services/refact/integrations";
@@ -38,8 +39,12 @@ import {
   updateConfirmationAfterIdeToolUse,
 } from "../features/ToolConfirmation/confirmationSlice";
 import { setInitialAgentUsage } from "../features/AgentUsage/agentUsageSlice";
-import { ideToolCallResponse } from "../hooks/useEventBusForIDE";
+import {
+  ideToolCallResponse,
+  ideForceReloadProjectTreeFiles,
+} from "../hooks/useEventBusForIDE";
 import { upsertToolCallIntoHistory } from "../features/History/historySlice";
+import { isToolResponse } from "../events";
 
 const AUTH_ERROR_MESSAGE =
   "There is an issue with your API key. Check out your API Key or re-login";
@@ -536,3 +541,16 @@ startListening({
     }
   },
 });
+
+// JB file refresh
+// TBD: this could include diff messages to
+startListening({
+  actionCreator: chatResponse,
+  effect: (action, listenerApi) => {
+    const state = listenerApi.getState();
+    if (state.config.host !== "jetbrains") return;
+    if (!isToolResponse(action.payload)) return;
+    if (!window.postIntellijMessage) return;
+    window.postIntellijMessage(ideForceReloadProjectTreeFiles());
+  },
+});

refact-agent/gui/src/components/Chat/Chat.tsx

@@ -10,7 +10,6 @@ import {
   useGetCapsQuery,
   useCapsForToolUse,
   useAgentUsage,
-  useLastSentCompressionStop,
 } from "../../hooks";
 import { type Config } from "../../features/Config/configSlice";
 import {
@@ -54,7 +53,6 @@ export const Chat: React.FC<ChatProps> = ({
   const { submit, abort, retryFromIndex } = useSendChatRequest();
 
   const chatToolUse = useAppSelector(getSelectedToolUse);
-  const compressionStop = useLastSentCompressionStop();
   const threadNewChatSuggested = useAppSelector(selectThreadNewChatSuggested);
   const messages = useAppSelector(selectMessages);
   const capsForToolUse = useCapsForToolUse();
@@ -106,9 +104,8 @@ export const Chat: React.FC<ChatProps> = ({
         <AgentUsage />
         <SuggestNewChat
           shouldBeVisible={
-            compressionStop.stopped ||
-            (threadNewChatSuggested.wasSuggested &&
-              !threadNewChatSuggested.wasRejectedByUser)
+            threadNewChatSuggested.wasSuggested &&
+            !threadNewChatSuggested.wasRejectedByUser
           }
         />
         {!isStreaming && preventSend && unCalledTools && (

refact-agent/gui/src/components/ChatForm/SuggestNewChat/SuggestNewChat.tsx

@@ -13,6 +13,7 @@ import {
 import { popBackTo, push } from "../../../features/Pages/pagesSlice";
 import { telemetryApi } from "../../../services/refact";
 import {
+  enableSend,
   newChatAction,
   selectChatId,
   setIsNewChatSuggestionRejected,
@@ -64,6 +65,7 @@ export const SuggestNewChat = ({
 
   const handleClose = () => {
     dispatch(setIsNewChatSuggestionRejected({ chatId, value: true }));
+    dispatch(enableSend({ id: chatId }));
 
     void sendTelemetryEvent({
       scope: `dismissedNewChatSuggestionWarning`,

refact-agent/gui/src/components/UsageCounter/useUsageCounter.ts

@@ -3,7 +3,6 @@ import {
   selectIsStreaming,
   selectIsWaiting,
   selectMessages,
-  // selectLastSentCompression,
 } from "../../features/Chat";
 import { useAppSelector, useLastSentCompressionStop } from "../../hooks";
 import {
@@ -33,19 +32,17 @@ export function useUsageCounter() {
   }, [currentThreadUsage]);
 
   const isOverflown = useMemo(() => {
-    if (!compressionStop.stopped) return false;
     if (compressionStop.strength === "low") return true;
     if (compressionStop.strength === "medium") return true;
     if (compressionStop.strength === "high") return true;
     return false;
-  }, [compressionStop.stopped, compressionStop.strength]);
+  }, [compressionStop.strength]);
 
   const isWarning = useMemo(() => {
-    if (!compressionStop.stopped) return false;
     if (compressionStop.strength === "medium") return true;
     if (compressionStop.strength === "high") return true;
     return false;
-  }, [compressionStop.stopped, compressionStop.strength]);
+  }, [compressionStop.strength]);
 
   const shouldShow = useMemo(() => {
     return messages.length > 0 && !isStreaming && !isWaiting;

refact-agent/gui/src/features/Chat/Thread/reducer.ts

@@ -39,7 +39,6 @@ import {
   setIsNewChatSuggestionRejected,
   upsertToolCall,
   setIncreaseMaxTokens,
-  setThreadPaused,
 } from "./actions";
 import { formatChatResponse } from "./utils";
 import {
@@ -79,7 +78,6 @@ const createChatThread = (
     boost_reasoning: false,
     automatic_patch: false,
     increase_max_tokens: false,
-    paused: false,
   };
   return chat;
 };
@@ -209,7 +207,7 @@ export const chatReducer = createReducer(initialState, (builder) => {
       action.payload.compression_strength !== "absent"
     ) {
       state.thread.new_chat_suggested = {
-        ...state.thread.new_chat_suggested,
+        wasRejectedByUser: false,
         wasSuggested: true,
       };
     }
@@ -243,15 +241,14 @@ export const chatReducer = createReducer(initialState, (builder) => {
 
   builder.addCase(setIsNewChatSuggested, (state, action) => {
     if (state.thread.id !== action.payload.chatId) return state;
-    state.thread.paused = true;
     state.thread.new_chat_suggested = {
       wasSuggested: action.payload.value,
     };
   });
 
   builder.addCase(setIsNewChatSuggestionRejected, (state, action) => {
     if (state.thread.id !== action.payload.chatId) return state;
-    state.thread.paused = false;
+    state.prevent_send = false;
     state.thread.new_chat_suggested = {
       ...state.thread.new_chat_suggested,
       wasRejectedByUser: action.payload.value,
@@ -279,7 +276,6 @@ export const chatReducer = createReducer(initialState, (builder) => {
     state.streaming = true;
     state.thread.read = false;
     state.prevent_send = false;
-    state.thread.paused = false;
   });
 
   builder.addCase(removeChatFromCache, (state, action) => {
@@ -337,7 +333,7 @@ export const chatReducer = createReducer(initialState, (builder) => {
       lastUserMessage.compression_strength !== "absent"
     ) {
       state.thread.new_chat_suggested = {
-        ...state.thread.new_chat_suggested,
+        wasRejectedByUser: false,
         wasSuggested: true,
       };
     }
@@ -427,10 +423,6 @@ export const chatReducer = createReducer(initialState, (builder) => {
     state.thread.increase_max_tokens = action.payload;
   });
 
-  builder.addCase(setThreadPaused, (state, action) => {
-    state.thread.paused = action.payload;
-  });
-
   builder.addMatcher(
     capsApi.endpoints.getCaps.matchFulfilled,
     (state, action) => {

refact-agent/gui/src/features/Chat/Thread/selectors.ts

@@ -94,8 +94,3 @@ export const selectLastSentCompression = createSelector(
     return lastCompression;
   },
 );
-
-export const selectThreadPaused = createSelector(
-  selectThread,
-  (thread) => thread.paused ?? false,
-);

refact-agent/gui/src/features/Chat/Thread/types.ts

@@ -29,7 +29,6 @@ export type ChatThread = {
   currentMaximumContextTokens?: number;
   currentMessageContextTokens?: number;
   increase_max_tokens?: boolean;
-  paused?: boolean;
 };
 
 export type SuggestedChat = {

refact-agent/gui/src/hooks/useCompressionStop.ts

@@ -2,10 +2,12 @@ import { useEffect, useCallback, useMemo } from "react";
 import { useAppSelector } from "./useAppSelector";
 import { useAppDispatch } from "./useAppDispatch";
 import {
+  selectChatId,
   selectLastSentCompression,
   selectMessages,
-  selectThreadPaused,
-  setThreadPaused,
+  setIsNewChatSuggested,
+  setIsNewChatSuggestionRejected,
+  setPreventSend,
 } from "../features/Chat";
 import { takeFromEndWhile } from "../utils";
 import { isUserMessage } from "../events";
@@ -14,12 +16,7 @@ export function useLastSentCompressionStop() {
   const dispatch = useAppDispatch();
   const lastSentCompression = useAppSelector(selectLastSentCompression);
   const messages = useAppSelector(selectMessages);
-  const stopped = useAppSelector(selectThreadPaused);
-  useEffect(() => {
-    if (lastSentCompression && lastSentCompression !== "absent" && !stopped) {
-      dispatch(setThreadPaused(true));
-    }
-  }, [dispatch, lastSentCompression, stopped]);
+  const chatId = useAppSelector(selectChatId);
 
   const messagesFromLastUserMessage = useMemo(() => {
     return takeFromEndWhile(messages, (message) => !isUserMessage(message))
@@ -30,16 +27,16 @@ export function useLastSentCompressionStop() {
     if (
       lastSentCompression &&
       lastSentCompression !== "absent" &&
-      messagesFromLastUserMessage >= 40 &&
-      !stopped
+      messagesFromLastUserMessage >= 40
     ) {
-      dispatch(setThreadPaused(true));
+      dispatch(setPreventSend({ id: chatId }));
+      dispatch(setIsNewChatSuggested({ chatId, value: true }));
     }
-  }, [dispatch, lastSentCompression, messagesFromLastUserMessage, stopped]);
+  }, [chatId, dispatch, lastSentCompression, messagesFromLastUserMessage]);
 
   const resume = useCallback(() => {
-    dispatch(setThreadPaused(false));
-  }, [dispatch]);
+    dispatch(setIsNewChatSuggestionRejected({ chatId, value: true }));
+  }, [chatId, dispatch]);
 
-  return { stopped, resume, strength: lastSentCompression };
+  return { resume, strength: lastSentCompression };
 }

refact-agent/gui/src/hooks/useDiffFileReload.ts

@@ -9,6 +9,7 @@ import { isDiffMessage } from "../services/refact";
  * Hook to handle file reloading for diff messages in JetBrains IDE
  * Ensures each file is only reloaded once per message
  */
+// Note this won't work if the chat is in the cache.
 export function useDiffFileReload() {
   const messages = useAppSelector(selectMessages);
   const configIdeHost = useAppSelector(selectConfig).host;

refact-agent/gui/src/hooks/useEventBusForIDE.ts

@@ -63,6 +63,10 @@ export const ideToolCallResponse = createAction<{
   accepted: boolean | "indeterminate";
 }>("ide/toolEditResponse");
 
+export const ideForceReloadProjectTreeFiles = createAction(
+  "ide/forceReloadProjectTreeFiles",
+);
+
 export const useEventsBusForIDE = () => {
   const [sendTelemetryEvent] =
     telemetryApi.useLazySendTelemetryChatEventQuery();

refact-agent/gui/src/hooks/useSendChatRequest.ts

@@ -56,7 +56,6 @@ import {
 
 import { v4 as uuidv4 } from "uuid";
 import { upsertToolCallIntoHistory } from "../features/History/historySlice";
-import { useLastSentCompressionStop } from "./useCompressionStop";
 
 type SubmitHandlerParams =
   | {
@@ -356,7 +355,6 @@ export function useAutoSend() {
   const sendImmediately = useAppSelector(selectSendImmediately);
   const wasInteracted = useAppSelector(getToolsInteractionStatus); // shows if tool confirmation popup was interacted by user
   const areToolsConfirmed = useAppSelector(getToolsConfirmationStatus);
-  const compressionStop = useLastSentCompressionStop();
   const { sendMessages, abort, messagesWithSystemPrompt } =
     useSendChatRequest();
   // TODO: make a selector for this, or show tool formation
@@ -381,7 +379,6 @@ export function useAutoSend() {
       const lastMessage = currentMessages.slice(-1)[0];
 
       if (
-        !compressionStop.stopped &&
         isAssistantMessage(lastMessage) &&
         lastMessage.tool_calls &&
         lastMessage.tool_calls.length > 0
@@ -418,6 +415,5 @@ export function useAutoSend() {
     isIntegration,
     thread.mode,
     thread,
-    compressionStop.stopped,
   ]);
 }