From 631fce5b7c49a8ecc62dcc3fcb72e8f950a8c1bd Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Jo=C3=A3o=20Moura?= Date: Tue, 5 Mar 2024 10:12:49 -0300 Subject: [PATCH] Overridding classes __repr__ --- src/crewai/agent.py | 3 +++ src/crewai/crew.py | 3 +++ src/crewai/task.py | 3 +++ 3 files changed, 9 insertions(+) diff --git a/src/crewai/agent.py b/src/crewai/agent.py index ecf18a42d..a26085d48 100644 --- a/src/crewai/agent.py +++ b/src/crewai/agent.py @@ -318,3 +318,6 @@ class Agent(BaseModel): @staticmethod def __tools_names(tools) -> str: return ", ".join([t.name for t in tools]) + + def __repr__(self): + return f"Agent(role={self.role}, goal={self.goal}, backstory={self.backstory})" diff --git a/src/crewai/crew.py b/src/crewai/crew.py index ec3cfcb13..20f89d77f 100644 --- a/src/crewai/crew.py +++ b/src/crewai/crew.py @@ -287,3 +287,6 @@ class Crew(BaseModel): if self.max_rpm: self._rpm_controller.stop_rpm_counter() self._telemetry.end_crew(self, output) + + def __repr__(self): + return f"Crew(id={self.id}, process={self.process}, number_of_agents={len(self.agents)}, number_of_tasks={len(self.tasks)})" diff --git a/src/crewai/task.py b/src/crewai/task.py index 3b1fd1a13..ff7af1b89 100644 --- a/src/crewai/task.py +++ b/src/crewai/task.py @@ -244,3 +244,6 @@ class Task(BaseModel): with open(self.output_file, "w") as file: file.write(result) return None + + def __repr__(self): + return f"Task(description={self.description}, expected_output={self.expected_output})"