undo stream implementation

Author: AlejandroEsquivel

guardrails/run/async_stream_runner.py

@@ -13,7 +13,6 @@
 from guardrails.classes import ValidationOutcome
 from guardrails.classes.history import Call, Inputs, Iteration, Outputs
 from guardrails.classes.output_type import OutputTypes
-from guardrails.classes.validation.validation_summary import ValidationSummary
 from guardrails.constants import pass_status
 from guardrails.llm_providers import (
     AsyncLiteLLMCallable,
@@ -168,16 +167,11 @@ async def async_step(
                     )
                 validation_response += cast(str, validated_fragment)
                 passed = call_log.status == pass_status
-                validator_logs = iteration.validator_logs
-                validation_summaries = ValidationSummary.from_validator_logs_only_fails(
-                    validator_logs
-                )
                 yield ValidationOutcome(
                     call_id=call_log.id,  # type: ignore
                     raw_llm_output=chunk_text,
                     validated_output=validated_fragment,
                     validation_passed=passed,
-                    validation_summaries=validation_summaries,
                 )
         else:
             async for chunk in stream_output:
@@ -213,16 +207,11 @@ async def async_step(
                     validation_response = cast(list, validated_fragment)
                 else:
                     validation_response = cast(dict, validated_fragment)
-                validator_logs = iteration.validator_logs
-                validation_summaries = ValidationSummary.from_validator_logs_only_fails(
-                    validator_logs
-                )
                 yield ValidationOutcome(
                     call_id=call_log.id,  # type: ignore
                     raw_llm_output=fragment,
                     validated_output=chunk_text,
                     validation_passed=validated_fragment is not None,
-                    validation_summaries=validation_summaries,
                 )
 
         iteration.outputs.raw_output = fragment

guardrails/run/stream_runner.py

@@ -1,10 +1,8 @@
 from typing import Any, Dict, Iterator, List, Optional, Tuple, Union, cast
 
-
 from guardrails import validator_service
 from guardrails.classes.history import Call, Inputs, Iteration, Outputs
 from guardrails.classes.output_type import OT, OutputTypes
-from guardrails.classes.validation.validation_summary import ValidationSummary
 from guardrails.classes.validation_outcome import ValidationOutcome
 from guardrails.llm_providers import (
     LiteLLMCallable,
@@ -181,7 +179,8 @@ def prepare_chunk_generator(stream) -> Iterator[Tuple[Any, bool]]:
                 "$",
                 validate_subschema=True,
             )
-            for chunk_index, res in enumerate(gen):
+
+            for res in gen:
                 chunk = res.chunk
                 original_text = res.original_text
                 if isinstance(chunk, SkeletonReAsk):
@@ -200,20 +199,12 @@ def prepare_chunk_generator(stream) -> Iterator[Tuple[Any, bool]]:
                 # 5. Convert validated fragment to a pretty JSON string
                 validation_response += cast(str, chunk)
                 passed = call_log.status == pass_status
-
-                validator_logs = call_log.iterations.last.validator_logs
-
-                validation_summaries = ValidationSummary.from_validator_logs_only_fails(
-                    validator_logs
-                )
-
                 yield ValidationOutcome(
                     call_id=call_log.id,  # type: ignore
                     #  The chunk or the whole output?
                     raw_llm_output=original_text,
                     validated_output=chunk,
                     validation_passed=passed,
-                    validation_summaries=validation_summaries,
                 )
 
         # handle non string schema
@@ -258,17 +249,11 @@ def prepare_chunk_generator(stream) -> Iterator[Tuple[Any, bool]]:
                 else:
                     validation_response = cast(dict, validated_fragment)
                 # 5. Convert validated fragment to a pretty JSON string
-
-                validator_logs = iteration.validator_logs
-                validation_summaries = ValidationSummary.from_validator_logs_only_fails(
-                    validator_logs
-                )
                 yield ValidationOutcome(
                     call_id=call_log.id,  # type: ignore
                     raw_llm_output=fragment,
                     validated_output=validated_fragment,
                     validation_passed=validated_fragment is not None,
-                    validation_summaries=validation_summaries,
                 )
 
         # # Finally, add to logs