Update app.py

app.py

@@ -229,7 +229,7 @@ def compress_history(purpose, task, history):
 def call_main(purpose, task, history, action_input):
     resp = run_gpt(
         FINDER,
-        stop_tokens=["observation:", "task:","action:"],
+        stop_tokens=["observation:", "task:"],
         max_tokens=2096,
         seed=random.randint(1,1000000000),
         purpose=purpose,
@@ -243,10 +243,11 @@ def call_main(purpose, task, history, action_input):
             continue
         if line.startswith("thought: "):
             history += "{}\n".format(line)
+            
         if line.startswith("action: "):
             action_name, action_input = parse_action(line)
             print(f'ACTION::{action_name} -- INPUT :: {action_input}')
-            history += "{}\n".format(line)
+            #history += "{}\n".format(line)
             return action_name, action_input, history, task
         else:
             pass
@@ -330,14 +331,14 @@ def find_all(purpose,task,history, url):
             print (f'out:: {out}')
             history += "observation: the search results are:\n {}\n".format(out)
             task = "compile report"
-            return "UPDATE-TASK", None, history, task
+            return "MAIN", None, history, task
         else: 
-            history += "observation: I need to trigger a search using the following syntax:\naction: SCRAPE_WEBSITE action_input=URL\n"
-            return "UPDATE-TASK", None, history, task
+            history += "observation: An Error occured\nI need to trigger a search using the following syntax:\naction: SCRAPE_WEBSITE action_input=URL\n"
+            return "MAIN", None, history, task
     except Exception as e:
         print (e)
         history += "observation: I need to trigger a search using the following syntax:\naction: SCRAPE_WEBSITE action_input=URL\n"
-        return "UPDATE-TASK", None, history, task
+        return "MAIN", None, history, task
 
         #else:
     #    history = "observation: The search query I used did not return a valid response"