Update app.py

app.py

@@ -75,17 +75,7 @@ def run_gpt(
     resp = ""
     for response in stream:
         resp += response.token.text
-    '''
-    resp = openai.ChatCompletion.create(
-        model=MODEL,
-        messages=[
-            {"role": "system", "content": content},
-        ],
-        temperature=0.0,
-        max_tokens=max_tokens,
-        stop=stop_tokens if stop_tokens else None,
-    )["choices"][0]["message"]["content"]
-    '''
+
     if VERBOSE:
         print(LOG_RESPONSE.format(resp))
     return resp
@@ -144,7 +134,7 @@ def call_main(purpose, task, history, directory, action_input):
             print (f'ACTION_INPUT :: {action_input}')
             
             history += "{}\n".format(line)
-            if action_name=="COMPLETE" or action_input=="COMPLETE":
+            if "COMPLETE" in action_name or "COMPLETE" in action_input:
                 task = "END"
                 return action_name, action_input, history, task
             else:
@@ -311,7 +301,7 @@ NAME_TO_FUNC = {
 
 
 def run_action(purpose, task, history, directory, action_name, action_input):
-    if action_name == "COMPLETE":
+    if "COMPLETE" in action_name:
         task="END"
         return action_name, action_input, history, task