30 lines
1.8 KiB
Python
30 lines
1.8 KiB
Python
import argparse
|
|
import asyncio
|
|
from Agent_E.test.tests_processor import run_tests
|
|
|
|
if __name__ == "__main__":
|
|
# Create the parser
|
|
parser = argparse.ArgumentParser(description='Run test suite for specified range of test tasks.')
|
|
|
|
# Add arguments
|
|
parser.add_argument('-s', '--take_screenshots', type=bool, default=False,
|
|
help='Take screenshots after every operation performed (default: False)')
|
|
parser.add_argument('-wait', '--wait_time_non_headless', type=int, default=5,
|
|
help='Time to wait between test tasks when running in non-headless mode (default: 10 seconds)')
|
|
parser.add_argument("-ids", "--task_ids", type=str, nargs='+', help="List of task IDs to execute")
|
|
parser.add_argument('-dir', '--logdir', type=str, default="../AgentOccam-Trajectories",
|
|
help='Logdir.')
|
|
parser.add_argument('-log', '--logname', type=str, default="Agent-E",
|
|
help='Logname.')
|
|
parser.add_argument('-id', '--test_results_id', type=str, default="",
|
|
help='A unique identifier for the test results. If not provided, a timestamp is used.')
|
|
parser.add_argument('-config', '--relative_task_dir', type=str, default="webvoyager",
|
|
help='Path to the test configuration file.')
|
|
|
|
# Parse the command line arguments
|
|
args = parser.parse_args()
|
|
|
|
# Run the main function with the provided or default arguments, not passing browser_manager or AutoGenWrapper will cause the test processor to create new instances of them
|
|
asyncio.run(run_tests(None, None, args.task_ids, logdir=args.logdir, logname=args.logname, relative_task_dir=args.relative_task_dir,
|
|
take_screenshots=args.take_screenshots, wait_time_non_headless=args.wait_time_non_headless))
|