update treatement

app.py

@@ -92,12 +92,22 @@ class GPTRemote(LLM):
         
         output = asyncio.run(start_playwright(prompt))
             # output = "test custom llm"
+        
+        output = self.treat_output(output)
+        
         return output
     
     @property
     def _identifying_params(self) -> Mapping[str, Any]:
         return [("n", self.n)]
     
+    def treat_output(output: str):
+        keywords = ['Action:', 'Action Input:', 'Observation:', 'Thought:', 'Final Answer:']
+        for item in keywords:
+            if item in output:
+                output.replace(item, '\n'+item)
+        print("treat output: ", output)
+        return output
     # def _generate(
     #     self,
     #     prompts: List[str],
@@ -625,14 +635,14 @@ Final Answer: the final answer to the original input question"""
 
 FORMAT_INSTRUCTIONS_2 = """Use the following format:
 
--Question: the input question you must answer
--Thought: you should always think about what to do
--Action: the action to take, should be one of [Vector Database Search, Duckduckgo Internet Search, Python REPL, Calculator]
--Action Input: the input to the action
--Observation: the result of the action
+Question: the input question you must answer
+Thought: you should always think about what to do
+Action: the action to take, should be one of [Vector Database Search, Duckduckgo Internet Search, Python REPL, Calculator]
+Action Input: the input to the action
+Observation: the result of the action
 ... (this Thought/Action/Action Input/Observation can repeat N times)
--Thought: I now know the final answer
--Final Answer: the final answer to the original input question"""
+Thought: I now know the final answer
+Final Answer: the final answer to the original input question"""
 
 SUFFIX = """Begin!