wehub-resource-sync
67 lines
1.8 KiB
Python
67 lines
1.8 KiB
Python
#!/usr/bin/env python
|
|
import sys
|
|
import warnings
|
|
|
|
from datetime import datetime
|
|
|
|
from latest_ai_development.crew import LatestAiDevelopment
|
|
|
|
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:
|
|
LatestAiDevelopment().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:
|
|
LatestAiDevelopment().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:
|
|
LatestAiDevelopment().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:
|
|
LatestAiDevelopment().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}")
|