From f5cbfe1ebba3f9ed9d9ee8f47d228e2d0117347a Mon Sep 17 00:00:00 2001
From: Xingyao Wang <xingyao6@illinois.edu>
Date: Thu, 2 May 2024 20:06:52 +0800
Subject: [PATCH] attemp to fix "Current task state not recognized"

---
 opendevin/server/agent/agent.py | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/opendevin/server/agent/agent.py b/opendevin/server/agent/agent.py
index 193af40eb3..f8b0c1e167 100644
--- a/opendevin/server/agent/agent.py
+++ b/opendevin/server/agent/agent.py
@@ -16,9 +16,13 @@ from opendevin.server.session import session_manager
 
 # new task state to valid old task states
 VALID_TASK_STATE_MAP: Dict[TaskStateAction, List[TaskState]] = {
-    TaskStateAction.PAUSE: [TaskState.RUNNING],
+    TaskStateAction.PAUSE: [TaskState.RUNNING, TaskState.AWAITING_USER_INPUT],
     TaskStateAction.RESUME: [TaskState.PAUSED],
-    TaskStateAction.STOP: [TaskState.RUNNING, TaskState.PAUSED],
+    TaskStateAction.STOP: [
+        TaskState.RUNNING,
+        TaskState.PAUSED,
+        TaskState.AWAITING_USER_INPUT,
+    ],
 }
 IGNORED_TASK_STATE_MAP: Dict[TaskStateAction, List[TaskState]] = {
     TaskStateAction.PAUSE: [