Add new tests

This commit is contained in:
Brandon Hancock
2024-07-01 15:29:08 -04:00
parent 60c8f86345
commit 5a5276eb5d
5 changed files with 947 additions and 48 deletions

View File

@@ -187,7 +187,6 @@ class BaseAgent(ABC, BaseModel):
"""Get the converter class for the agent to create json/pydantic outputs."""
pass
# TODO: HAVE LORENZE REVIEW THIS WITH OTHER AGENTS
@abstractmethod
def copy(self):
"""Create a copy of the agent."""

View File

@@ -423,7 +423,7 @@ class Crew(BaseModel):
self._finish_execution(task_output)
token_usage = self._calculate_usage_metrics()
token_usage = self.calculate_usage_metrics()
# type: ignore # Incompatible return value type (got "tuple[str, Any]", expected "str")
return self._format_output(task_output, token_usage)