base_prompt: Remove warning on old schema

Author: irgolic

guardrails/prompt/base_prompt.py

@@ -1,6 +1,5 @@
 """Class for representing a prompt entry."""
 import re
-import warnings
 from string import Template
 from typing import Optional
 
@@ -48,13 +47,6 @@ def substitute_constants(self, text):
         """Substitute constants in the prompt."""
         # Substitute constants by reading the constants file.
         # Regex to extract all occurrences of ${gr.<constant_name>}
-        if self.uses_old_constant_schema(text):
-            warnings.warn(
-                "It appears that you are using an old schema for gaurdrails variables, "
-                "follow the new namespaced convention "
-                "documented here: https://docs.guardrailsai.com/0-2-migration/"
-            )
-
         matches = re.findall(r"\${gr\.(\w+)}", text)
 
         # Substitute all occurrences of ${gr.<constant_name>}
@@ -66,13 +58,6 @@ def substitute_constants(self, text):
 
         return text
 
-    def uses_old_constant_schema(self, text) -> bool:
-        matches = re.findall(r"@(\w+)", text)
-        if len(matches) == 0:
-            return False
-        else:
-            return True
-
     def get_prompt_variables(self):
         return self.variable_names
 

tests/unit_tests/test_prompt.py

@@ -1,7 +1,6 @@
 """Unit tests for prompt and instructions parsing."""
 
 from string import Template
-from unittest import mock
 
 import pytest
 from pydantic import BaseModel, Field
@@ -232,31 +231,6 @@ def test_substitute_constants(prompt_str, final_prompt):
     assert prompt.source == final_prompt
 
 
-# TODO: Deprecate when we can confirm migration off the old, non-namespaced standard
-@pytest.mark.parametrize(
-    "text, is_old_schema",
-    [
-        (RAIL_WITH_OLD_CONSTANT_SCHEMA, True),  # Test with a single match
-        (
-            RAIL_WITH_FORMAT_INSTRUCTIONS,
-            False,
-        ),  # Test with no matches/correct namespacing
-    ],
-)
-def test_uses_old_constant_schema(text, is_old_schema):
-    with mock.patch("warnings.warn") as warn_mock:
-        guard = gd.Guard.from_rail_string(text)
-        assert guard.prompt.uses_old_constant_schema(text) == is_old_schema
-        if is_old_schema:
-            # we only check for the warning when we have an older schema
-            warn_mock.assert_called_once_with(
-                """It appears that you are using an old schema for gaurdrails\
- variables, follow the new namespaced convention documented here:\
- https://docs.guardrailsai.com/0-2-migration/\
-"""
-            )
-
-
 class TestResponse(BaseModel):
     grade: int = Field(description="The grade of the response")