minor fixes

Author: dtam

guardrails/classes/history/iteration.py

@@ -198,7 +198,10 @@ def create_messages_table(
             table.add_column("Content")
 
             for msg in messages:
-                table.add_row(str(msg["role"]), msg["content"])  # type: ignore
+                if hasattr(msg["content"], "source"):
+                    table.add_row(str(msg["role"]), msg["content"].source)  # type: ignore
+                else:
+                    table.add_row(str(msg["role"]), msg["content"])  # type: ignore
 
             return table
 

guardrails/run/utils.py

@@ -20,6 +20,7 @@ def messages_source(messages: MessageHistory) -> MessageHistory:
         content = (
             msg["content"].source
             if isinstance(msg["content"], Prompt)
+            or isinstance(msg["content"], Instructions)
             else msg["content"]
         )
         msg_copy["content"] = content