From 5d4b6e4d4e45b78d752943b7211b0da414fc48cc Mon Sep 17 00:00:00 2001 From: Harrison Chase Date: Mon, 30 Jan 2023 23:59:55 -0800 Subject: [PATCH] conversational agent fix (#818) --- langchain/agents/conversational/base.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/langchain/agents/conversational/base.py b/langchain/agents/conversational/base.py index 3cbccf86a30..d97fc417868 100644 --- a/langchain/agents/conversational/base.py +++ b/langchain/agents/conversational/base.py @@ -75,8 +75,8 @@ class ConversationalAgent(Agent): return self.ai_prefix def _extract_tool_and_input(self, llm_output: str) -> Optional[Tuple[str, str]]: - if f"{self.ai_prefix}: " in llm_output: - return self.ai_prefix, llm_output.split(f"{self.ai_prefix}: ")[-1] + if f"{self.ai_prefix}:" in llm_output: + return self.ai_prefix, llm_output.split(f"{self.ai_prefix}:")[-1].strip() regex = r"Action: (.*?)\nAction Input: (.*)" match = re.search(regex, llm_output) if not match: