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

making sure we don't check for agents that were not used in the crew

Browse Source
This commit is contained in:
João Moura
2024-11-06 23:03:58 -03:00
parent bd3f4042e2
commit 1267219c3d
1 changed files with 7 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
src/crewai/crew.py
View File

@@ -445,13 +445,14 @@ class Crew(BaseModel):
training_data = CrewTrainingHandler(TRAINING_DATA_FILE).load() training_data = CrewTrainingHandler(TRAINING_DATA_FILE).load()
for agent in train_crew.agents: for agent in train_crew.agents:
result = TaskEvaluator(agent).evaluate_training_data( if training_data.get(str(agent.id)):
training_data=training_data, agent_id=str(agent.id) result = TaskEvaluator(agent).evaluate_training_data(
) training_data=training_data, agent_id=str(agent.id)
)
CrewTrainingHandler(filename).save_trained_data( CrewTrainingHandler(filename).save_trained_data(
agent_id=str(agent.role), trained_data=result.model_dump() agent_id=str(agent.role), trained_data=result.model_dump()
) )
def kickoff( def kickoff(
self, self,