diff --git a/analyse_results.py b/analyse_results.py index 552865e..c83a83a 100644 --- a/analyse_results.py +++ b/analyse_results.py @@ -76,7 +76,7 @@ def analyze_json_file(file_path): if 'turns' in data and isinstance(data['turns'], list): for turn in reversed(data['turns']): # Check turns from the end if turn.get('role') == 'system' and isinstance(turn.get('content'), str): - if "Task successful ended with code : 2" in turn['content'] or "Task ended in score: 1" in turn["content"]: + if "Task successful ended with code : 2" in turn['content'] or "Task ended with score : 1" in turn["content"] or "Task ended in score: 1" in turn["content"]: return True return False except FileNotFoundError: