initial crew class

This commit is contained in:
Joao Moura
2023-11-05 14:26:59 -03:00
parent ffd0444f11
commit 15d4871b52

View File

@@ -1,7 +1,24 @@
"""Crew of agents."""
from typing import List
from pydantic import BaseModel, Field
from .process import Process
from .agent import Agent
from .task import Task
class Crew(BaseModel):
description: str = Field(description="Description and of the crew being created.")
goal: str = Field(description="Objective of the crew being created.")
"""
Class that represents a group of agents, how they should work together and
their tasks.
"""
goal: str = Field(description="Objective of the crew being created.")
process: Process = Field(description="Process that the crew will follow.")
tasks: List[Task] = Field(description="List of tasks")
agents: List[Agent] = Field(description="List of agents in this crew.")
def kickoff(self) -> str:
"""
Kickoff the crew to work on it's tasks.
Returns:
output (List[str]): Output of the crew for each task.
"""
return "Crew is executing task"