Update app.py

app.py

@@ -169,7 +169,7 @@ def call_set_task(purpose, task, history, directory, action_input):
 
 def end_fn(purpose, task, history, directory, action_input):
     task = "END"
-    return "COMPLETE", None, history, task
+    return "COMPLETE", "COMPLETE", history, task
 NAME_TO_FUNC = {
     "MAIN": call_main,
     "UPDATE-TASK": call_set_task,
@@ -184,7 +184,7 @@ def run_action(purpose, task, history, directory, action_name, action_input):
         if "RESPONSE" in action_name or "COMPLETE" in action_name:
             action_name="COMPLETE"
             task="END"
-            return action_name, action_input, history, task
+            return action_name, "COMPLETE", history, task
     
         # compress the history when it is long
         if len(history.split("\n")) > MAX_HISTORY: