From 7c0633e5de74d77b8a22eda5a8137b91036eed69 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Jo=C3=A3o=20Moura?= Date: Sun, 22 Sep 2024 17:04:40 -0300 Subject: [PATCH] linter --- src/crewai/crew.py | 19 +++++++++++++------ 1 file changed, 13 insertions(+), 6 deletions(-) diff --git a/src/crewai/crew.py b/src/crewai/crew.py index 6c6992e07..91382c8ac 100644 --- a/src/crewai/crew.py +++ b/src/crewai/crew.py @@ -224,11 +224,19 @@ class Crew(BaseModel): def create_crew_memory(self) -> "Crew": """Set private attributes.""" if self.memory: - self._long_term_memory = self.long_term_memory if self.long_term_memory else LongTermMemory() - self._short_term_memory = self.short_term_memory if self.short_term_memory else ShortTermMemory( - crew=self, embedder_config=self.embedder + self._long_term_memory = ( + self.long_term_memory if self.long_term_memory else LongTermMemory() + ) + self._short_term_memory = ( + self.short_term_memory + if self.short_term_memory + else ShortTermMemory(crew=self, embedder_config=self.embedder) + ) + self._entity_memory = ( + self.entity_memory + if self.entity_memory + else EntityMemory(crew=self, embedder_config=self.embedder) ) - self._entity_memory = self.entity_memory if self.entity_memory else EntityMemory(crew=self, embedder_config=self.embedder) return self @model_validator(mode="after") @@ -948,10 +956,9 @@ class Crew(BaseModel): inputs: Optional[Dict[str, Any]] = None, ) -> None: """Test and evaluate the Crew with the given inputs for n iterations concurrently using concurrent.futures.""" - # type: ignore[arg-type] self._test_execution_span = self._telemetry.test_execution_span( self, n_iterations, inputs, openai_model_name - ) + ) # type: ignore[arg-type] evaluator = CrewEvaluator(self, openai_model_name) # type: ignore[arg-type] for i in range(1, n_iterations + 1):