defaults for empty registry; always return registry even if empty

Author: CalebCourier

guardrails/cli/hub/list.py

@@ -10,10 +10,6 @@ def list():
     """List all installed validators."""
     registry = get_registry()
 
-    if not registry:
-        console.print("No validators installed.")
-        return
-
     validators = registry.validators
     if not validators:
         console.print("No validators installed.")

guardrails/hub/__init__.py

@@ -19,8 +19,6 @@ def _build_export_map() -> dict:
     module import paths (e.g. "guardrails_grhub_detect_pii").
     """
     registry = get_registry()
-    if not registry:
-        return {}
     export_map = {}
     for entry in registry.validators.values():
         import_path = entry.import_path

guardrails/hub/registry.py

@@ -11,7 +11,7 @@ def get_registry_path() -> Path:
     return Path(os.getcwd()) / ".guardrails" / "hub_registry.json"
 
 
-def get_registry() -> ValidatorRegistry | None:
+def get_registry() -> ValidatorRegistry:
     registry_file = get_registry_path()
     try:
         registry_str = registry_file.read_text()
@@ -20,4 +20,4 @@ def get_registry() -> ValidatorRegistry | None:
         return registry
     except (json.JSONDecodeError, OSError):
         logger.warning("Failed to read hub registry at %s", registry_file)
-        return None
+        return ValidatorRegistry()

guardrails/types/validator_registry.py

@@ -11,5 +11,5 @@ class ValidatorRegistryEntry(BaseModel):
 
 
 class ValidatorRegistry(BaseModel):
-    version: int
-    validators: dict[str, ValidatorRegistryEntry]
+    version: int = Field(default=1)
+    validators: dict[str, ValidatorRegistryEntry] = Field(default_factory=dict)

guardrails/validator_base.py

@@ -572,11 +572,9 @@ def decorator(
 def try_to_import_from_hub(validator_key: str):
     try:
         hub_registry = get_registry()
-        if not hub_registry:
-            return
-        import_path = hub_registry.validators[validator_key].import_path
-        if import_path:
-            importlib.import_module(import_path)
+        validator_entry = hub_registry.validators.get(validator_key)
+        if validator_entry and validator_entry.import_path:
+            importlib.import_module(validator_entry.import_path)
     except (ImportError, KeyError):
         logger.error("Could not import from hub. Validators may not work properly.")