Skip to content

Jasper

any_guardrail.guardrails.jasper.jasper

Jasper

Bases: StandardGuardrail

Prompt injection detection encoder based models.

For more information, please see the model card:

Parameters:

Name Type Description Default
model_id str | None

HuggingFace path to model.

 None

Raises:

Type Description
ValueError

Can only use model paths for Jasper models from HuggingFace.
Source code in src/any_guardrail/guardrails/jasper/jasper.py 
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
class Jasper(StandardGuardrail):
    """Prompt injection detection encoder based models.

    For more information, please see the model card:

    - [Jasper Deberta](https://huggingface.co/JasperLS/deberta-v3-base-injection)
    - [Jasper Gelectra](https://huggingface.co/JasperLS/gelectra-base-injection).

    Args:
        model_id: HuggingFace path to model.

    Raises:
        ValueError: Can only use model paths for Jasper models from HuggingFace.

    """

    SUPPORTED_MODELS: ClassVar = ["JasperLS/gelectra-base-injection", "JasperLS/deberta-v3-base-injection"]

    def __init__(self, model_id: str | None = None, provider: StandardProvider | None = None) -> None:
        """Initialize the Jasper guardrail."""
        self.model_id = default(model_id, self.SUPPORTED_MODELS)
        self.provider = provider or HuggingFaceProvider()
        self.provider.load_model(self.model_id)

    def _pre_processing(self, input_text: str) -> StandardPreprocessOutput:
        return self.provider.pre_process(input_text)

    def _inference(self, model_inputs: StandardPreprocessOutput) -> StandardInferenceOutput:
        return self.provider.infer(model_inputs)

    def _post_processing(self, model_outputs: StandardInferenceOutput) -> BinaryScoreOutput:
        return match_injection_label(model_outputs, JASPER_INJECTION_LABEL, self.provider.model.config.id2label)  # type: ignore[attr-defined]
__init__(model_id=None, provider=None)

Initialize the Jasper guardrail.

Source code in src/any_guardrail/guardrails/jasper/jasper.py 
30
31
32
33
34
def __init__(self, model_id: str | None = None, provider: StandardProvider | None = None) -> None:
    """Initialize the Jasper guardrail."""
    self.model_id = default(model_id, self.SUPPORTED_MODELS)
    self.provider = provider or HuggingFaceProvider()
    self.provider.load_model(self.model_id)