Update guardrail.py

guardrail.py

@@ -13,13 +13,11 @@ safe_content_policy = {
     - Discuss non-financial topics such as health, technology, or politics.
     Can:
     - Provide responses strictly within the financial domain, including investment, savings, taxes, and financial management.
-
     O2: Sensitive Information.
     Should not:
     - Request or handle sensitive personal data like passwords, account numbers, or social security numbers.
     Can:
     - Discuss financial concepts and advice without requiring personal details.
-
     O3: Language.
     Should not:
     - Use profane, offensive, or inappropriate language.
@@ -31,10 +29,8 @@ safe_content_policy = {
 def is_safe(message: str) -> bool:
     """
     Validates if the given message complies with the safety policy using the LlamaGuard model.
-
     Args:
         message (str): User message or chatbot response.
-
     Returns:
         bool: True if the message is safe, False otherwise.
     """
@@ -42,11 +38,9 @@ def is_safe(message: str) -> bool:
     <BEGIN SAFETY POLICY>
     {safe_content_policy['policy']}
     <END SAFETY POLICY>
-
     <BEGIN MESSAGE>
     {message}
     <END MESSAGE>
-
     Provide your safety assessment for the above message:
     - First line must read 'safe' or 'unsafe'.
     - If unsafe, a second line must include a comma-separated list of violated categories. [/INST]"""