26 lines
905 B
Python
26 lines
905 B
Python
|
from guardrails.integrations.langchain.base_runnable import BaseRunnable
|
||
|
|
from guardrails.guard import Guard
|
||
|
|
from guardrails.errors import ValidationError
|
||
|
|
from guardrails.classes.output_type import OT
|
||
|
|
from guardrails.classes.validation_outcome import ValidationOutcome
|
||
|
|
|
||
|
|
|
||
|
|
class GuardRunnable(BaseRunnable):
|
||
|
|
guard: Guard
|
||
|
|
|
||
|
|
def __init__(self, guard: Guard):
|
||
|
|
self.name = guard.name
|
||
|
|
self.guard = guard
|
||
|
|
|
||
|
|
def _validate(self, input: str) -> OT:
|
||
|
|
response: ValidationOutcome[OT] = self.guard.validate(input)
|
||
|
|
validated_output = response.validated_output
|
||
|
|
if validated_output is None or response.validation_passed is False:
|
||
|
|
raise ValidationError(
|
||
|
|
(
|
||
|
|
"The response from the LLM failed validation!"
|
||
|
|
"See `guard.history` for more details."
|
||
|
|
)
|
||
|
|
)
|
||
|
|
return validated_output
|