diff --git a/pyrit/executor/attack/printer/console_printer.py b/pyrit/executor/attack/printer/console_printer.py index 7f954fd4e..272d275bb 100644 --- a/pyrit/executor/attack/printer/console_printer.py +++ b/pyrit/executor/attack/printer/console_printer.py @@ -152,13 +152,18 @@ async def print_conversation_async( # Display images if present await display_image_response(piece) - # Print scores with better formatting (only if auxiliary scores are requested) + # Always print objective scores + scores = self._memory.get_prompt_scores(prompt_ids=[str(piece.id)]) + if scores: + print() + self._print_colored(f"{self._indent}📊 Scores:", Style.DIM, Fore.MAGENTA) + objective_score = [score for score in scores if score.score_category == "objective"][0] + self.print_score(objective_score) + + # Print auxiliary scores only if requested if include_auxiliary_scores: - scores = self._memory.get_prompt_scores(prompt_ids=[str(piece.id)]) - if scores: - print() - self._print_colored(f"{self._indent}📊 Scores:", Style.DIM, Fore.MAGENTA) - for score in scores: + for score in scores: + if score.score_category == "auxiliary": self._print_score(score) print()