Skip to content

Commit dfe0910

Browse files
added to guard.py
1 parent 5620dc3 commit dfe0910

File tree

1 file changed

+4
-5
lines changed

1 file changed

+4
-5
lines changed

guardrails/guard.py

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -33,6 +33,7 @@
3333
from guardrails.api_client import GuardrailsApiClient
3434
from guardrails.classes.output_type import OT
3535
from guardrails.classes.validation.validation_result import ErrorSpan
36+
from guardrails.classes.validation.validation_summary import ValidationSummary
3637
from guardrails.classes.validation_outcome import ValidationOutcome
3738
from guardrails.classes.credentials import Credentials
3839
from guardrails.classes.execution import GuardExecutionOptions
@@ -1193,10 +1194,8 @@ def _single_server_call(self, *, payload: Dict[str, Any]) -> ValidationOutcome[O
11931194
)
11941195
self.history.extend([Call.from_interface(call) for call in guard_history])
11951196

1196-
# TODO Validation Summary
1197-
# validator_logs = self.history.last.iterations.last.validator_logs
1198-
# validation_summaries = ValidationSummary.
1199-
# from_validator_logs(validator_logs)
1197+
validator_logs = self.history.last.iterations.last.validator_logs
1198+
validation_summaries = ValidationSummary.from_validator_logs(validator_logs)
12001199

12011200
# TODO: See if the below statement is still true
12021201
# Our interfaces are too different for this to work right now.
@@ -1208,12 +1207,12 @@ def _single_server_call(self, *, payload: Dict[str, Any]) -> ValidationOutcome[O
12081207
if validation_output.validated_output
12091208
else None
12101209
)
1211-
# TODO: Validation Summary
12121210
return ValidationOutcome[OT](
12131211
call_id=validation_output.call_id, # type: ignore
12141212
raw_llm_output=validation_output.raw_llm_output,
12151213
validated_output=validated_output,
12161214
validation_passed=(validation_output.validation_passed is True),
1215+
validation_summaries=validation_summaries,
12171216
)
12181217
else:
12191218
raise ValueError("Guard does not have an api client!")

0 commit comments

Comments
 (0)