Overridding classes __repr__

This commit is contained in:
João Moura
2024-03-05 10:12:49 -03:00
parent ef9f85f5d2
commit 631fce5b7c
3 changed files with 9 additions and 0 deletions

View File

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

View File

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

View File

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