Spaces:
Running
Running
File size: 2,088 Bytes
6231986 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 |
#!/usr/bin/env python
import sys
import warnings
from datetime import datetime
from coder.crew import Coder
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(assignment):
"""
Run the crew.
"""
# assignment = "Write a python program to calculate the first 10,000 terms \
# of this series, multiplying the total by 4: 1 - 1/3 + 1/5 - 1/7 + ..."
inputs = {
'assignment': assignment,
}
return Coder().crew().kickoff(inputs=inputs)
# try:
# return Coder().crew().kickoff(inputs=inputs)
# # print(result.raw)
# # return result.raw
# 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:
# Coder().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:
# Coder().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:
# Coder().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}")
|