nemoguardrails.rails.llm.options.GenerationRailsOptions

class nemoguardrails.rails.llm.options.GenerationRailsOptions

Bases: BaseModel

Options for what rails should be used during the generation.

input: Union[bool, List[str]]

output: Union[bool, List[str]]

retrieval: Union[bool, List[str]]

dialog: bool

tool_output: Union[bool, List[str]]

tool_input: Union[bool, List[str]]

model_config: ClassVar[ConfigDict] = {}

Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].