diff --git a/src/crewai/utilities/training_handler.py b/src/crewai/utilities/training_handler.py index 5cadde619..b6b3c38b6 100644 --- a/src/crewai/utilities/training_handler.py +++ b/src/crewai/utilities/training_handler.py @@ -1,3 +1,5 @@ +import os + from crewai.utilities.file_handler import PickleHandler @@ -29,3 +31,10 @@ class CrewTrainingHandler(PickleHandler): data[agent_id] = {train_iteration: new_data} self.save(data) + + def clear(self) -> None: + """Clear the training data by removing the file or resetting its contents.""" + if os.path.exists(self.file_path): + with open(self.file_path, "wb") as file: + # Overwrite with an empty dictionary + self.save({})