Compare commits

...

1 Commits

Author SHA1 Message Date
João Moura
a336381849 adding agent to task output 2024-05-16 05:12:32 -03:00
2 changed files with 2 additions and 0 deletions

View File

@@ -201,6 +201,7 @@ class Task(BaseModel):
description=self.description,
exported_output=exported_output,
raw_output=result,
agent=agent.role,
)
if self.callback:

View File

@@ -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")