strip whitespace (#680)

Author: hwchase17

langchain/agents/mrkl/base.py

@@ -41,7 +41,7 @@ def get_action_and_input(llm_output: str) -> Tuple[str, str]:
     match = re.search(regex, llm_output)
     if not match:
         raise ValueError(f"Could not parse LLM output: `{llm_output}`")
-    action = match.group(1)
+    action = match.group(1).strip()
     action_input = match.group(2)
     return action, action_input.strip(" ").strip('"')
 

tests/unit_tests/agents/test_mrkl.py

@@ -19,6 +19,14 @@ def test_get_action_and_input() -> None:
     assert action_input == "NBA"
 
 
+def test_get_action_and_input_whitespace() -> None:
+    """Test getting an action from text."""
+    llm_output = "Thought: I need to search for NBA\nAction: Search \nAction Input: NBA"
+    action, action_input = get_action_and_input(llm_output)
+    assert action == "Search"
+    assert action_input == "NBA"
+
+
 def test_get_final_answer() -> None:
     """Test getting final answer."""
     llm_output = (