0.6.7 pre release

.github/workflows/ci.yml

@@ -37,7 +37,7 @@ jobs:
           # Setup Virtual Environment
           python3 -m venv ./.venv
           source .venv/bin/activate
-          poetry install
+          poetry install --extras dev
 
       - name: Check license
         run: |

.pre-commit-config.yaml

@@ -1,6 +1,6 @@
 repos:
   - repo: https://github.com/astral-sh/ruff-pre-commit
-    rev: v0.4.1
+    rev: v0.13.0
     hooks:
       # Performs ruff check with safe fixes
       - id: ruff

guardrails/hub/install.py

@@ -3,7 +3,7 @@
 from string import Template
 from typing import Callable, cast, List
 
-import pkg_resources
+import importlib.metadata
 
 from guardrails.hub.validator_package_service import (
     ValidatorPackageService,
@@ -149,7 +149,7 @@ def install(
     installed_version_message = ""
     with contextlib.suppress(Exception):
         package_name = ValidatorPackageService.get_normalized_package_name(validator_id)
-        installed_version = pkg_resources.get_distribution(package_name).version
+        installed_version = importlib.metadata.version(package_name)
         if installed_version:
             installed_version_message = f" version {installed_version}"
 

guardrails/llm_providers.py

@@ -278,8 +278,7 @@ def _invoke_llm(
             import torch
         except ImportError:
             raise PromptCallableException(
-                "The `torch` package is not installed. "
-                "Install with `pip install torch`"
+                "The `torch` package is not installed. Install with `pip install torch`"
             )
         prompt = messages_to_prompt_string(messages)
         tokenizer = kwargs.pop("tokenizer")
@@ -384,8 +383,7 @@ def _invoke_llm(
             import torch  # noqa: F401 # type: ignore
         except ImportError:
             raise PromptCallableException(
-                "The `torch` package is not installed. "
-                "Install with `pip install torch`"
+                "The `torch` package is not installed. Install with `pip install torch`"
             )
 
         content_key = kwargs.pop("content_key", "generated_text")

guardrails/utils/openai_utils/v1.py

@@ -15,7 +15,7 @@
 
 def get_static_openai_create_func():
     warnings.warn(
-        "This function is deprecated. " " and will be removed in 0.6.0",
+        "This function is deprecated.  and will be removed in 0.6.0",
         DeprecationWarning,
     )
     return openai.completions.create

guardrails/utils/prompt_utils.py

@@ -26,7 +26,7 @@ def prompt_content_for_string_schema(
     description = output_schema.get("description")
     if description:
         prompt_content += (
-            "Here's a description of what I want you to generate: " f"{description}"
+            f"Here's a description of what I want you to generate: {description}"
         )
     validators = validator_map.get(json_path, [])
     if len(validators):

guardrails/utils/structured_data_utils.py

@@ -29,10 +29,10 @@ def set_additional_properties_false_iteratively(schema):
                     current["properties"].keys()
                 )  # this has to be set
             if "maximum" in current:
-                logger.warn("Property maximum is not supported." " Dropping")
+                logger.warn("Property maximum is not supported. Dropping")
                 current.pop("maximum")  # the api does not like these set
             if "minimum" in current:
-                logger.warn("Property maximum is not supported." " Dropping")
+                logger.warn("Property maximum is not supported. Dropping")
                 current.pop("minimum")  # the api does not like these set
             if "default" in current:
                 logger.warn("Property default is not supported. Marking field Required")

guardrails/utils/validator_utils.py

@@ -37,8 +37,7 @@ def parse_rail_arguments(arg_tokens: List[str]) -> List[Any]:
                 t = literal_eval(t)
             except (ValueError, SyntaxError, NameError) as e:
                 raise ValueError(
-                    f"Python expression `{t}` is not valid, "
-                    f"and raised an error: {e}."
+                    f"Python expression `{t}` is not valid, and raised an error: {e}."
                 )
         validator_args.append(t)
     return validator_args

guardrails/validator_base.py

@@ -160,9 +160,9 @@ def __init__(
         # Store the kwargs for the validator.
         self._kwargs = kwargs
 
-        assert (
-            self.rail_alias in validators_registry
-        ), f"Validator {self.__class__.__name__} is not registered. "
+        assert self.rail_alias in validators_registry, (
+            f"Validator {self.__class__.__name__} is not registered. "
+        )
 
     @property
     @deprecated(

guardrails/validator_service/__init__.py

@@ -41,8 +41,7 @@ def should_run_sync():
     bool_values = ["true", "false"]
     if run_sync.lower() not in bool_values:
         warnings.warn(
-            f"GUARDRAILS_RUN_SYNC must be one of {bool_values}!"
-            f" Defaulting to 'false'."
+            f"GUARDRAILS_RUN_SYNC must be one of {bool_values}! Defaulting to 'false'."
         )
     return process_count == 1 or run_sync.lower() == "true"