update filtering

app.py

@@ -93,7 +93,8 @@ class GPTRemote(LLM):
         output = asyncio.run(start_playwright(prompt))
             # output = "test custom llm"
         # print(type(output))
-        print(output)
+        print("-" * 20)
+        print('Raw: \n', output)
         keywords = ['Action:', 'Action Input:', 'Observation:', 'Thought:', 'Final Answer:']
         # print("Judge 1: ", 'Action:' in output)
         # print("Judge 2: ", 'Action Input:' in output)
@@ -104,7 +105,15 @@ class GPTRemote(LLM):
         for item in keywords:
             if item in output:
                 output = output.replace(item, '\n'+item)
-        print("treat output: ", output)
+        
+        if 'Thought:' not in output:
+            output = 'Thought:'+ output
+
+        if 'Action Input:' in output and 'Observation:' in output:
+            output = output.split('Observation:')[0]
+
+        print("-" * 20)
+        print("Treated output: \n", output)
         return output
     
     @property