adding agent to task output

crewAIInc · May 16, 2024 · a336381 · a336381
1 parent 208c3a7
commit a336381
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/src/crewai/task.py b/src/crewai/task.py
@@ -201,6 +201,7 @@ def _execute(self, agent, task, context, tools):
  description=self.description,
  exported_output=exported_output,
  raw_output=result,
+ agent=agent.role,
  )
 
  if self.callback:

diff --git a/src/crewai/tasks/task_output.py b/src/crewai/tasks/task_output.py
@@ -11,6 +11,7 @@ class TaskOutput(BaseModel):
  exported_output: Union[str, BaseModel] = Field(
  description="Output of the task", default=None
  )
+ agent: str = Field(description="Agent that executed the task")
  raw_output: str = Field(description="Result of the task")
 
  @model_validator(mode="after")