andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-11 00:58:30 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

linter

Browse Source
This commit is contained in:
João Moura
2024-09-22 17:04:40 -03:00
parent 6682fe3d89
commit 7c0633e5de
1 changed files with 13 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
src/crewai/crew.py
View File

@@ -224,11 +224,19 @@ class Crew(BaseModel):
def create_crew_memory(self) -> "Crew": def create_crew_memory(self) -> "Crew":
"""Set private attributes.""" """Set private attributes."""
if self.memory: if self.memory:
self._long_term_memory = self.long_term_memory if self.long_term_memory else LongTermMemory() self._long_term_memory = (
self._short_term_memory = self.short_term_memory if self.short_term_memory else ShortTermMemory( self.long_term_memory if self.long_term_memory else LongTermMemory()
crew=self, embedder_config=self.embedder )
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 return self
@model_validator(mode="after") @model_validator(mode="after")
@@ -948,10 +956,9 @@ class Crew(BaseModel):
inputs: Optional[Dict[str, Any]] = None, inputs: Optional[Dict[str, Any]] = None,
) -> None: ) -> None:
"""Test and evaluate the Crew with the given inputs for n iterations concurrently using concurrent.futures.""" """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._test_execution_span = self._telemetry.test_execution_span(
self, n_iterations, inputs, openai_model_name self, n_iterations, inputs, openai_model_name
) ) # type: ignore[arg-type]
evaluator = CrewEvaluator(self, 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): for i in range(1, n_iterations + 1):