Strip trailing whitespaces in agent's stop sequences (#1566)

Fixes #1489
1 year ago · a950287206
parent 30383abb12
commit a950287206
1 changed files with 4 additions and 1 deletions
--- 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]]