mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-10 00:28:31 +00:00
1f0001b644
New CrewAIBaseModel: Base for Agent, Crew, Task. Includes generated, frozen UUID. Adds hashing capability Migrate to ConfigDict: Replaces class Config with model_config, see this deprecation note . Benefits: Adds auditing capability with frozen UUIDs.
41 lines
1.3 KiB
Python
41 lines
1.3 KiB
Python
from typing import Any, List, Optional
|
|
|
|
from pydantic import Field, model_validator
|
|
|
|
from crewai.agent import Agent
|
|
from crewai.base.model import CrewAIBaseModel
|
|
|
|
|
|
class Task(CrewAIBaseModel):
|
|
"""Class that represent a task to be executed."""
|
|
|
|
description: str = Field(description="Description of the actual task.")
|
|
agent: Optional[Agent] = Field(
|
|
description="Agent responsible for the task.", default=None
|
|
)
|
|
tools: List[Any] = Field(
|
|
default_factory=list,
|
|
description="Tools the agent are limited to use for this task.",
|
|
)
|
|
|
|
@model_validator(mode="after")
|
|
def check_tools(self):
|
|
if not self.tools and (self.agent and self.agent.tools):
|
|
self.tools.extend(self.agent.tools)
|
|
return self
|
|
|
|
def execute(self, context: str = None) -> str:
|
|
"""Execute the task.
|
|
|
|
Returns:
|
|
Output of the task.
|
|
"""
|
|
if self.agent:
|
|
return self.agent.execute_task(
|
|
task=self.description, context=context, tools=self.tools
|
|
)
|
|
else:
|
|
raise Exception(
|
|
f"The task '{self.description}' has no agent assigned, therefore it can't be executed directly and should be executed in a Crew using a specific process that support that, either consensual or hierarchical."
|
|
)
|