From a9502872069409039c69b41d4857b2c7791c3752 Mon Sep 17 00:00:00 2001 From: Ryan Dao Date: Thu, 9 Mar 2023 16:36:15 -0800 Subject: [PATCH] Strip trailing whitespaces in agent's stop sequences (#1566) Fixes #1489 --- langchain/agents/agent.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/langchain/agents/agent.py b/langchain/agents/agent.py index 095e7365..01bc233a 100644 --- a/langchain/agents/agent.py +++ b/langchain/agents/agent.py @@ -47,7 +47,10 @@ class Agent(BaseModel): @property def _stop(self) -> List[str]: - return [f"\n{self.observation_prefix}", f"\n\t{self.observation_prefix}"] + return [ + f"\n{self.observation_prefix.rstrip()}", + f"\n\t{self.observation_prefix.rstrip()}", + ] def _construct_scratchpad( self, intermediate_steps: List[Tuple[AgentAction, str]]