From 682f946f2916a86d6f1c7d6e6f3d2dfdd2151442 Mon Sep 17 00:00:00 2001 From: Shreya Rajpal Date: Sat, 30 Mar 2024 23:36:23 -0700 Subject: [PATCH] update tree for guard logs --- guardrails/classes/history/iteration.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/guardrails/classes/history/iteration.py b/guardrails/classes/history/iteration.py index 7d96b56ea..8530dea16 100644 --- a/guardrails/classes/history/iteration.py +++ b/guardrails/classes/history/iteration.py @@ -215,7 +215,9 @@ def create_msg_history_table( self.raw_output or "", title="Raw LLM Output", style="on #F5F5DC" ), Panel( - pretty_repr(self.validation_response), + self.validation_response + if isinstance(self.validation_response, str) + else pretty_repr(self.validation_response), title="Validated Output", style="on #F0FFF0", ),