update output

app.py

@@ -94,15 +94,19 @@ class GPTRemote(LLM):
             # output = "test custom llm"
         print(type(output))
         print(output)
-        output2 = self.treat_output(output)
-        
-        return output2
+        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
     
     @property
     def _identifying_params(self) -> Mapping[str, Any]:
         return [("n", self.n)]
     
-    def treat_output(text: str):
+    def treat_output(text):
+        
         keywords = ['Action:', 'Action Input:', 'Observation:', 'Thought:', 'Final Answer:']
         for item in keywords:
             if item in text: