27 lines
756 B
Python
27 lines
756 B
Python
|
from guardrails.formatters.base_formatter import BaseFormatter, PassthroughFormatter
|
||
|
|
|
||
|
|
try:
|
||
|
|
from guardrails.formatters.json_formatter import JsonFormatter
|
||
|
|
except ImportError:
|
||
|
|
JsonFormatter = None
|
||
|
|
|
||
|
|
|
||
|
|
def get_formatter(name: str, *args, **kwargs) -> BaseFormatter:
|
||
|
|
"""Returns a class."""
|
||
|
|
name = name.lower()
|
||
|
|
if name == "jsonformer":
|
||
|
|
if JsonFormatter is None:
|
||
|
|
raise ValueError("jsonformatter requires transformers to be installed.")
|
||
|
|
return JsonFormatter(*args, **kwargs)
|
||
|
|
elif name == "none":
|
||
|
|
return PassthroughFormatter(*args, **kwargs)
|
||
|
|
raise ValueError(f"Unrecognized formatter '{name}'")
|
||
|
|
|
||
|
|
|
||
|
|
__all__ = [
|
||
|
|
"get_formatter",
|
||
|
|
"BaseFormatter",
|
||
|
|
"PassthroughFormatter",
|
||
|
|
"JsonFormatter",
|
||
|
|
]
|