Merge remote-tracking branch 'origin/main' into feature/litellm_cleanup

Author: dtam

.github/workflows/install_from_hub.yml

@@ -9,6 +9,8 @@ on:
 jobs:
   install_from_hub:
     runs-on: ubuntu-latest
+    env:
+      GUARDRAILS_API_KEY: ${{ secrets.GUARDRAILS_API_KEY }}
     steps:
     - name: Set up Python
       uses: actions/setup-python@v5
@@ -17,6 +19,8 @@ jobs:
     - name: pip install from main
       run: pip install git+https://github.com/guardrails-ai/guardrails.git@main
     - name: Install PII validator
-      run: guardrails hub install hub://guardrails/detect_pii
+      run: |
+        guardrails configure --token $GUARDRAILS_API_KEY --disable-metrics --enable-remote-inferencing;
+        guardrails hub install hub://guardrails/detect_pii;
     - name: Verify PII validator is addressable
       run: echo 'from guardrails.hub import DetectPII' | python 

guardrails/cli/hub/install.py

@@ -10,9 +10,14 @@
 from guardrails.cli.version import version_warnings_if_applicable
 
 
-@hub_command.command()
+# Quick note: This is the command for `guardrails hub install`.  We change the name of
+# the function def to prevent confusion, lest people import it directly and calling it
+# with a string for package_uris instead of a list, which behaves oddly. If you need to
+# call install from a script, please consider importing install from guardrails,
+# not guardrails.cli.hub.install.
+@hub_command.command(name="install")
 @trace(name="guardrails-cli/hub/install")
-def install(
+def install_cli(
     package_uris: List[str] = typer.Argument(
         ...,
         help="URIs to the packages to install. Example: hub://guardrails/regex_match hub://guardrails/toxic_language",
@@ -33,6 +38,17 @@ def install(
     ),
 ):
     try:
+        if isinstance(package_uris, str):
+            logger.error(
+                f"`install` in {__file__} was called with a string instead of "
+                "a list! This can happen if it is invoked directly instead of "
+                "being run via the CLI. Did you mean to import `from guardrails import "
+                "install` instead?  Recovering..."
+            )
+            package_uris = [
+                package_uris,
+            ]
+
         from guardrails.hub.install import install_multiple
 
         def confirm():

guardrails/hub/install.py

@@ -53,7 +53,7 @@ def install(
     Examples:
         >>> RegexMatch = install("hub://guardrails/regex_match").RegexMatch
 
-        >>> install("hub://guardrails/regex_match);
+        >>> install("hub://guardrails/regex_match")
         >>> import guardrails.hub.regex_match as regex_match
     """
 

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "guardrails-ai"
-version = "0.5.10"
+version = "0.5.11"
 description = "Adding guardrails to large language models."
 authors = ["Guardrails AI <[email protected]>"]
 license = "Apache License 2.0"