fix: enable the messages tag in RAIL

guardrails-ai · Oct 10, 2024 · d889ca0 · d889ca0
1 parent 30eeb4c
commit d889ca0
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/guardrails/guard.py b/guardrails/guard.py
@@ -979,7 +979,12 @@ def __call__(
         """
         instructions = instructions or self._exec_opts.instructions
         prompt = prompt or self._exec_opts.prompt
-        msg_history = msg_history or kwargs.get("messages", None) or []
+        msg_history = (
+            msg_history
+            or kwargs.get("messages", None)
+            or self._exec_opts.messages
+            or []
+        )
         if prompt is None:
             if msg_history is not None and not len(msg_history):
                 raise RuntimeError(

diff --git a/guardrails/schema/rail_schema.py b/guardrails/schema/rail_schema.py
@@ -455,7 +455,7 @@ def rail_string_to_schema(rail_string: str) -> ProcessedSchema:
                 role = message.attrib.get("role")
                 content = message.text
                 extracted_reask_messages.append({"role": role, "content": content})
-        processed_schema.exec_opts.messages = extracted_reask_messages
+        processed_schema.exec_opts.reask_messages = extracted_reask_messages
 
     return processed_schema