From 9f2acfe91f3dafab2aef1c465810b7ef8780eee7 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Jo=C3=A3o=20Moura?= Date: Wed, 6 Nov 2024 23:03:58 -0300 Subject: [PATCH] making sure we don't check for agents that were not used in the crew --- src/crewai/crew.py | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) diff --git a/src/crewai/crew.py b/src/crewai/crew.py index e65024ed6..7bcaa82ad 100644 --- a/src/crewai/crew.py +++ b/src/crewai/crew.py @@ -445,13 +445,14 @@ class Crew(BaseModel): training_data = CrewTrainingHandler(TRAINING_DATA_FILE).load() for agent in train_crew.agents: - result = TaskEvaluator(agent).evaluate_training_data( - training_data=training_data, agent_id=str(agent.id) - ) + if training_data.get(str(agent.id)): + result = TaskEvaluator(agent).evaluate_training_data( + training_data=training_data, agent_id=str(agent.id) + ) - CrewTrainingHandler(filename).save_trained_data( - agent_id=str(agent.role), trained_data=result.model_dump() - ) + CrewTrainingHandler(filename).save_trained_data( + agent_id=str(agent.role), trained_data=result.model_dump() + ) def kickoff( self,