Update app.py

app.py

@@ -118,11 +118,17 @@ def call_search(purpose, task, history, directory, action_input):
     except Exception as e:
         history += "observation: {}'\n".format(e)
     return "MAIN", None, history, task
+NAME_TO_FUNC = {
+    "MAIN": call_main,
+    "UPDATE-TASK": call_set_task,
+    "SEARCH": call_search,
+    "COMPLETE": end_fn,
 
+}
 def call_main(purpose, task, history, directory, action_input):
     resp = run_gpt(
         ACTION_PROMPT,
-        stop_tokens=["observation:", "task:"],
+        stop_tokens=["observation:", "task:", "thought:","search:"],
         max_tokens=2096,
         purpose=purpose,
         task=task,
@@ -170,16 +176,11 @@ def call_set_task(purpose, task, history, directory, action_input):
 def end_fn(purpose, task, history, directory, action_input):
     task = "END"
     return "COMPLETE", "COMPLETE", history, task
-NAME_TO_FUNC = {
-    "MAIN": call_main,
-    "UPDATE-TASK": call_set_task,
-    "SEARCH": call_search,
-    "COMPLETE": end_fn,
 
-}
 
 
 def run_action(purpose, task, history, directory, action_name, action_input):
+    print(f'action_name::{action_name}')
     try:
         if "RESPONSE" in action_name or "COMPLETE" in action_name:
             action_name="COMPLETE"