Skip to content

Commit

Permalink
change spell
Browse files Browse the repository at this point in the history
  • Loading branch information
tentru committed Dec 13, 2024
1 parent 0d6c1eb commit aaf8e6e
Show file tree
Hide file tree
Showing 2 changed files with 8 additions and 8 deletions.
6 changes: 3 additions & 3 deletions modules/crew/crew.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,8 +6,8 @@
# https://docs.crewai.com/concepts/crews#example-crew-class-with-decorators

@CrewBase
class Cfdw():
"""Cfdw crew"""
class CrewAgent():
"""CrewAgent crew"""

# Learn more about YAML configuration files here:
# Agents: https://docs.crewai.com/concepts/agents#yaml-configuration-recommended
Expand Down Expand Up @@ -49,7 +49,7 @@ def reporting_task(self) -> Task:

@crew
def crew(self) -> Crew:
"""Creates the Cfdw crew"""
"""Creates the CrewAgent crew"""
# To learn how to add knowledge sources to your crew, check out the documentation:
# https://docs.crewai.com/concepts/knowledge#what-is-knowledge

Expand Down
10 changes: 5 additions & 5 deletions modules/crew/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@
import sys
import warnings

from cfdw.crew import Cfdw
from crew import CrewAgent

warnings.filterwarnings("ignore", category=SyntaxWarning, module="pysbd")

Expand All @@ -18,7 +18,7 @@ def run():
inputs = {
'topic': 'AI LLMs'
}
Cfdw().crew().kickoff(inputs=inputs)
CrewAgent().crew().kickoff(inputs=inputs)


def train():
Expand All @@ -29,7 +29,7 @@ def train():
"topic": "AI LLMs"
}
try:
Cfdw().crew().train(n_iterations=int(sys.argv[1]), filename=sys.argv[2], inputs=inputs)
CrewAgent().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}")
Expand All @@ -39,7 +39,7 @@ def replay():
Replay the crew execution from a specific task.
"""
try:
Cfdw().crew().replay(task_id=sys.argv[1])
CrewAgent().crew().replay(task_id=sys.argv[1])

except Exception as e:
raise Exception(f"An error occurred while replaying the crew: {e}")
Expand All @@ -52,7 +52,7 @@ def test():
"topic": "AI LLMs"
}
try:
Cfdw().crew().test(n_iterations=int(sys.argv[1]), openai_model_name=sys.argv[2], inputs=inputs)
CrewAgent().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}")

0 comments on commit aaf8e6e

Please sign in to comment.