|
|
@ -19,6 +19,14 @@ def test_get_action_and_input() -> None:
|
|
|
|
assert action_input == "NBA"
|
|
|
|
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:
|
|
|
|
def test_get_final_answer() -> None:
|
|
|
|
"""Test getting final answer."""
|
|
|
|
"""Test getting final answer."""
|
|
|
|
llm_output = (
|
|
|
|
llm_output = (
|
|
|
|