Fixed bug in shell reset

Author: rusiaaman

Diff for: pyproject.toml

@@ -1,7 +1,7 @@
 [project]
 authors = [{ name = "Aman Rusia", email = "[email protected]" }]
 name = "wcgw"
-version = "3.0.3"
+version = "3.0.4"
 description = "Shell and coding agent on claude and chatgpt"
 readme = "README.md"
 requires-python = ">=3.11"

Diff for: src/wcgw/client/bash_state/bash_state.py

@@ -468,16 +468,6 @@ def load_state(
         cwd: str,
     ) -> None:
         """Create a new BashState instance from a serialized state dictionary"""
-        if (
-            self._bash_command_mode == bash_command_mode
-            and ((self._cwd == cwd) or not cwd)
-            and (self._file_edit_mode == file_edit_mode)
-            and (self._write_if_empty_mode == write_if_empty_mode)
-            and (self._mode == mode)
-            and (self._whitelist_for_overwrite == set(whitelist_for_overwrite))
-        ):
-            # No need to reset shell if the state is the same
-            return
         self._bash_command_mode = bash_command_mode
         self._cwd = cwd or self._cwd
         self._file_edit_mode = file_edit_mode

Diff for: src/wcgw/client/tools.py

@@ -324,7 +324,7 @@ def try_open_file(file_path: str) -> None:
                 break
             except:
                 continue
-    
+
     # Try to open the file if a command is available
     if open_cmd:
         try:
@@ -698,7 +698,7 @@ def get_tool_output(
         context.console.print("Calling task memory tool")
         relevant_files = []
         warnings = ""
-        arg.project_root_path = os.path.expanduser(arg.project_root_path) 
+        arg.project_root_path = os.path.expanduser(arg.project_root_path)
         for fglob in arg.relevant_file_globs:
             fglob = expand_user(fglob)
             if not os.path.isabs(fglob) and arg.project_root_path:
@@ -708,7 +708,9 @@ def get_tool_output(
             if not globs:
                 warnings += f"Warning: No files found for the glob: {fglob}\n"
         relevant_files_data = read_files(relevant_files[:10_000], None, context)
-        save_path = save_memory(arg, relevant_files_data, context.bash_state.serialize())
+        save_path = save_memory(
+            arg, relevant_files_data, context.bash_state.serialize()
+        )
         if not relevant_files and arg.relevant_file_globs:
             output_ = f'Error: No files found for the given globs. Context file successfully saved at "{save_path}", but please fix the error.'
         elif warnings: