fix arg parser that's broken for some reason

This commit is contained in:
Xingyao Wang
2024-04-30 15:09:22 +08:00
parent 12c29a42c3
commit 6cc8961133

View File

@@ -9,8 +9,8 @@ def test_help_message(capsys):
parser.parse_args(['--help'])
captured = capsys.readouterr()
expected_help_message = """
usage: pytest [-h] [-d DIRECTORY] [-t TASK] [-f FILE] [-c AGENT_CLS]
[-m MODEL_NAME] [-i MAX_ITERATIONS] [-n MAX_CHARS]
usage: pytest [-h] [-d DIRECTORY] [-t TASK] [-f FILE] [-c AGENT_CLS] [-m MODEL_NAME] [-i MAX_ITERATIONS]
[-n MAX_CHARS]
Run an agent with a specific task
@@ -19,8 +19,7 @@ options:
-d DIRECTORY, --directory DIRECTORY
The working directory for the agent
-t TASK, --task TASK The task for the agent to perform
-f FILE, --file FILE Path to a file containing the task. Overrides -t if
both are provided.
-f FILE, --file FILE Path to a file containing the task. Overrides -t if both are provided.
-c AGENT_CLS, --agent-cls AGENT_CLS
The agent class to use
-m MODEL_NAME, --model-name MODEL_NAME
@@ -28,8 +27,7 @@ options:
-i MAX_ITERATIONS, --max-iterations MAX_ITERATIONS
The maximum number of iterations to run the agent
-n MAX_CHARS, --max-chars MAX_CHARS
The maximum number of characters to send to and
receive from LLM per task
The maximum number of characters to send to and receive from LLM per task
"""
actual_lines = captured.out.strip().split('\n')
expected_lines = expected_help_message.strip().split('\n')