strip whitespace (#680)

This commit is contained in:
Harrison Chase 2023-01-21 16:03:48 -08:00 committed by GitHub
parent 2f57d18b25
commit a2eeaf3d43
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 9 additions and 1 deletions

View File

@ -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('"')

View File

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