Update app.py

app.py

@@ -230,7 +230,7 @@ def call_main(purpose, task, history, action_input):
     resp = run_gpt(
         FINDER,
         stop_tokens=["observation:", "task:","action:"],
-        max_tokens=512,
+        max_tokens=2096,
         seed=random.randint(1,1000000000),
         purpose=purpose,
         task=task,
@@ -242,10 +242,7 @@ def call_main(purpose, task, history, action_input):
             continue
         if line.startswith("thought: "):
             history += "{}\n".format(line)
-        if line.startswith("action: COMPLETE"):
-            print("COMPLETE called")
-            return "COMPLETE", None, history, task
-        if line.startswith("action:"):
+        if line.startswith("action: "):
             action_name, action_input = parse_action(line)
             print(f'ACTION::{action_name} -- INPUT :: {action_input}')
             history += "{}\n".format(line)
@@ -422,7 +419,7 @@ def run(purpose,history):
         history = ""
     else:
         history=str(history)
-    action_name = "SEARCH_ENGINE" if task is None else "MAIN"
+    action_name = "MAIN"
     action_input = None
     while True:
         print("")