mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-07 15:18:29 +00:00
69 lines
1.8 KiB
Python
69 lines
1.8 KiB
Python
#!/usr/bin/env python
|
|
import sys
|
|
import warnings
|
|
|
|
from datetime import datetime
|
|
|
|
from {{folder_name}}.crew import {{crew_name}}
|
|
|
|
warnings.filterwarnings("ignore", category=SyntaxWarning, module="pysbd")
|
|
|
|
# This main file is intended to be a way for you to run your
|
|
# crew locally, so refrain from adding unnecessary logic into this file.
|
|
# Replace with inputs you want to test with, it will automatically
|
|
# interpolate any tasks and agents information
|
|
|
|
def run():
|
|
"""
|
|
Run the crew.
|
|
"""
|
|
inputs = {
|
|
'topic': 'AI LLMs',
|
|
'current_year': str(datetime.now().year)
|
|
}
|
|
|
|
try:
|
|
{{crew_name}}().crew().kickoff(inputs=inputs)
|
|
except Exception as e:
|
|
raise Exception(f"An error occurred while running the crew: {e}")
|
|
|
|
|
|
def train():
|
|
"""
|
|
Train the crew for a given number of iterations.
|
|
"""
|
|
inputs = {
|
|
"topic": "AI LLMs",
|
|
'current_year': str(datetime.now().year)
|
|
}
|
|
try:
|
|
{{crew_name}}().crew().train(n_iterations=int(sys.argv[1]), filename=sys.argv[2], inputs=inputs)
|
|
|
|
except Exception as e:
|
|
raise Exception(f"An error occurred while training the crew: {e}")
|
|
|
|
def replay():
|
|
"""
|
|
Replay the crew execution from a specific task.
|
|
"""
|
|
try:
|
|
{{crew_name}}().crew().replay(task_id=sys.argv[1])
|
|
|
|
except Exception as e:
|
|
raise Exception(f"An error occurred while replaying the crew: {e}")
|
|
|
|
def test():
|
|
"""
|
|
Test the crew execution and returns the results.
|
|
"""
|
|
inputs = {
|
|
"topic": "AI LLMs",
|
|
"current_year": str(datetime.now().year)
|
|
}
|
|
|
|
try:
|
|
{{crew_name}}().crew().test(n_iterations=int(sys.argv[1]), eval_llm=sys.argv[2], inputs=inputs)
|
|
|
|
except Exception as e:
|
|
raise Exception(f"An error occurred while testing the crew: {e}")
|