From bcf910097539af09243b7decadaa063ba0244061 Mon Sep 17 00:00:00 2001 From: Sylvain Gugger <35901082+sgugger@users.noreply.github.com> Date: Wed, 10 May 2023 11:13:42 -0400 Subject: [PATCH] Fix new line bug in chat mode for agents (#23267) --- src/transformers/tools/agents.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/src/transformers/tools/agents.py b/src/transformers/tools/agents.py index fd45b17c83772f..a7b07a4932cfb3 100644 --- a/src/transformers/tools/agents.py +++ b/src/transformers/tools/agents.py @@ -264,9 +264,7 @@ def chat(self, task, *, return_code=False, remote=False, **kwargs): """ prompt = self.format_prompt(task, chat_mode=True) result = self.generate_one(prompt, stop=["Human:", "====="]) - self.chat_history = prompt + result - if not self.chat_history.endswith("\n"): - self.chat_history += "\n" + self.chat_history = prompt + result.strip() + "\n" explanation, code = clean_code_for_chat(result) print(f"==Explanation from the agent==\n{explanation}")