disable comment stripping and simplify state machine

replayio · Dec 9, 2024 · 50b3864 · 50b3864
1 parent 30b7eab
commit 50b3864
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 13 deletions.
diff --git a/openhands/controller/agent_controller.py b/openhands/controller/agent_controller.py
@@ -333,17 +333,7 @@ async def _handle_message_action(self, action: MessageAction) -> None:
                 # Tell the agent to stop analyzing and start editing:
                 self.state.replay_phase = ReplayDebuggingPhase.Edit
                 self.event_stream.add_event(
-                    MessageAction(
-                        content='Implement the changes, then wait for user confirmation.'
-                    ),
-                    EventSource.USER,
-                )
-            elif self.state.replay_phase == ReplayDebuggingPhase.Edit:
-                self.state.replay_phase = ReplayDebuggingPhase.Normal
-                self.event_stream.add_event(
-                    MessageAction(
-                        content='Double check your changes and publish a PR without waiting for user confirmation.'
-                    ),
+                    MessageAction(content='Implement the changes.'),
                     EventSource.USER,
                 )
             else:

diff --git a/openhands/resolver/resolve_issue.py b/openhands/resolver/resolve_issue.py
@@ -198,8 +198,8 @@ async def complete_runtime(
     if not isinstance(obs, CmdOutputObservation) or obs.exit_code != 0:
         raise RuntimeError(f'Failed to git add. Observation: {obs}')
 
-    base_git_patch = await get_git_patch(runtime, base_commit)
-    strip_replay_comments(base_git_patch)
+    # base_git_patch = await get_git_patch(runtime, base_commit)
+    # strip_replay_comments(base_git_patch)
 
     git_patch = await get_git_patch(runtime, base_commit)