Skip to content

Deepset

any_guardrail.guardrails.deepset.deepset

Deepset

Bases: HuggingFace[dict[str, Any], dict[str, Any], bool, None, float]

Wrapper for prompt injection detection model from Deepset.

For more information, please see the model card:

Source code in src/any_guardrail/guardrails/deepset/deepset.py 
10
11
12
13
14
15
16
17
18
19
20
21
22
23
class Deepset(HuggingFace[dict[str, Any], dict[str, Any], bool, None, float]):
    """Wrapper for prompt injection detection model from Deepset.

    For more information, please see the model card:

    - [Deepset](https://huggingface.co/deepset/deberta-v3-base-injection).
    """

    SUPPORTED_MODELS: ClassVar = ["deepset/deberta-v3-base-injection"]

    def _post_processing(
        self, model_outputs: GuardrailInferenceOutput[dict[str, Any]]
    ) -> GuardrailOutput[bool, None, float]:
        return _match_injection_label(model_outputs, DEEPSET_INJECTION_LABEL, self.model.config.id2label)