fix: tests

Author: sestinj

gui/src/pages/gui/Chat.tsx

@@ -49,6 +49,8 @@ import { cancelStream } from "../../redux/thunks/cancelStream";
 import { EmptyChatBody } from "./EmptyChatBody";
 import { ExploreDialogWatcher } from "./ExploreDialogWatcher";
 import { useAutoScroll } from "./useAutoScroll";
+import { useStore } from "react-redux";
+import { RootState } from "../../redux/store";
 
 // Helper function to find the index of the latest conversation summary
 function findLatestSummaryIndex(history: ChatHistoryItem[]): number {
@@ -98,13 +100,11 @@ function fallbackRender({ error, resetErrorBoundary }: any) {
 export function Chat() {
   const dispatch = useAppDispatch();
   const ideMessenger = useContext(IdeMessengerContext);
+  const reduxStore = useStore<RootState>();
   const onboardingCard = useOnboardingCard();
   const showSessionTabs = useAppSelector(
     (store) => store.config.config.ui?.showSessionTabs,
   );
-  const selectedModels = useAppSelector(
-    (store) => store.config?.config.selectedModelByRole,
-  );
   const isStreaming = useAppSelector((state) => state.session.isStreaming);
   const [stepsOpen] = useState<(boolean | undefined)[]>([]);
   const mainTextInputRef = useRef<HTMLInputElement>(null);
@@ -115,7 +115,6 @@ export function Chat() {
     (state) => state.config.config.ui?.showChatScrollbar,
   );
   const codeToEdit = useAppSelector((state) => state.editModeState.codeToEdit);
-  const mode = useAppSelector((store) => store.session.mode);
   const isInEdit = useAppSelector((store) => store.session.isInEdit);
 
   const lastSessionId = useAppSelector((state) => state.session.lastSessionId);
@@ -152,18 +151,23 @@ export function Chat() {
     isStreaming,
   );
 
-  const pendingToolCalls = useAppSelector(selectPendingToolCalls);
-  const pendingApplyStates = useAppSelector(selectDoneApplyStates);
-
   const sendInput = useCallback(
     (
       editorState: JSONContent,
       modifiers: InputModifiers,
       index?: number,
       editorToClearOnSend?: Editor,
     ) => {
+      const stateSnapshot = reduxStore.getState();
+      const latestPendingToolCalls = selectPendingToolCalls(stateSnapshot);
+      const latestPendingApplyStates = selectDoneApplyStates(stateSnapshot);
+      const isCurrentlyInEdit = stateSnapshot.session.isInEdit;
+      const codeToEditSnapshot = stateSnapshot.editModeState.codeToEdit;
+      const selectedModelByRole =
+        stateSnapshot.config.config.selectedModelByRole;
+
       // Cancel all pending tool calls
-      pendingToolCalls.forEach((toolCallState) => {
+      latestPendingToolCalls.forEach((toolCallState) => {
         dispatch(
           cancelToolCall({
             toolCallId: toolCallState.toolCallId,
@@ -172,19 +176,20 @@ export function Chat() {
       });
 
       // Reject all pending apply states
-      pendingApplyStates.forEach((applyState) => {
+      latestPendingApplyStates.forEach((applyState) => {
         if (applyState.status !== "closed") {
           ideMessenger.post("rejectDiff", applyState);
         }
       });
-      const model = isInEdit
-        ? (selectedModels?.edit ?? selectedModels?.chat)
-        : selectedModels?.chat;
+      const model = isCurrentlyInEdit
+        ? selectedModelByRole.edit ?? selectedModelByRole.chat
+        : selectedModelByRole.chat;
+
       if (!model) {
         return;
       }
 
-      if (isInEdit && codeToEdit.length === 0) {
+      if (isCurrentlyInEdit && codeToEditSnapshot.length === 0) {
         return;
       }
 
@@ -212,11 +217,11 @@ export function Chat() {
       //   }
       // }
 
-      if (isInEdit) {
+      if (isCurrentlyInEdit) {
         void dispatch(
           streamEditThunk({
             editorState,
-            codeToEdit,
+            codeToEdit: codeToEditSnapshot,
           }),
         );
       } else {
@@ -227,15 +232,7 @@ export function Chat() {
         }
       }
     },
-    [
-      history,
-      selectedModels,
-      mode,
-      isInEdit,
-      codeToEdit,
-      pendingToolCalls,
-      pendingApplyStates,
-    ],
+    [dispatch, ideMessenger, reduxStore],
   );
 
   useWebviewListener(