Significantly upgrade the interactive configuration experience

This commit is contained in:
Ashwin Bharambe 2024-10-05 11:12:46 -07:00 committed by Ashwin Bharambe
parent 8d157a8197
commit 5a7b01d292
7 changed files with 217 additions and 156 deletions

View file

@ -22,8 +22,6 @@ class MetaReferenceShieldType(Enum):
class LlamaGuardShieldConfig(BaseModel):
model: str = "Llama-Guard-3-1B"
excluded_categories: List[str] = []
disable_input_check: bool = False
disable_output_check: bool = False
@field_validator("model")
@classmethod