diff --git a/src/crewai/cli/templates/main.py b/src/crewai/cli/templates/main.py index f300f4912..94ac35b93 100644 --- a/src/crewai/cli/templates/main.py +++ b/src/crewai/cli/templates/main.py @@ -48,7 +48,7 @@ def test(): "topic": "AI LLMs" } try: - {{crew_name}}Crew().crew().test(n_iterations=int(sys.argv[1]), model=sys.argv[2], inputs=inputs) + {{crew_name}}Crew().crew().test(n_iterations=int(sys.argv[1]), openai_model_name=sys.argv[2], inputs=inputs) except Exception as e: raise Exception(f"An error occurred while replaying the crew: {e}")