strip whitespace (#680)

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 = (