fix some more tests

Author: dtam

guardrails/run/async_runner.py

@@ -1,6 +1,6 @@
 import copy
 from functools import partial
-from typing import Any, Dict, List, Optional, Union, cast
+from typing import Any, Dict, List, Optional, cast
 
 
 from guardrails import validator_service
@@ -11,7 +11,7 @@
 from guardrails.llm_providers import AsyncPromptCallableBase
 from guardrails.logger import set_scope
 from guardrails.run.runner import Runner
-from guardrails.run.utils import messages_source, messages_string
+from guardrails.run.utils import messages_source
 from guardrails.schema.validator import schema_validation
 from guardrails.hub_telemetry.hub_tracing import async_trace
 from guardrails.types.inputs import MessageHistory
@@ -25,6 +25,7 @@
 from guardrails.constants import fail_status
 from guardrails.prompt import Prompt
 
+
 class AsyncRunner(Runner):
     def __init__(
         self,
@@ -331,9 +332,7 @@ async def prepare_messages(
             formatted_messages.append(msg_copy)
 
         if "messages" in self.validation_map:
-            await self.validate_messages(
-                call_log, formatted_messages, attempt_number
-            )
+            await self.validate_messages(call_log, formatted_messages, attempt_number)
 
         return formatted_messages
 
@@ -348,9 +347,11 @@ async def validate_messages(
                 else msg["content"]
             )
             inputs = Inputs(
-                        llm_output=content,
-                    )
-            iteration = Iteration(call_id=call_log.id, index=attempt_number, inputs=inputs)
+                llm_output=content,
+            )
+            iteration = Iteration(
+                call_id=call_log.id, index=attempt_number, inputs=inputs
+            )
             call_log.iterations.insert(0, iteration)
             value, _metadata = await validator_service.async_validate(
                 value=content,
@@ -364,7 +365,7 @@ async def validate_messages(
             validated_msg = validator_service.post_process_validation(
                 value, attempt_number, iteration, OutputTypes.STRING
             )
-            
+
             iteration.outputs.validation_response = validated_msg
 
             if isinstance(validated_msg, ReAsk):
@@ -374,4 +375,4 @@ async def validate_messages(
 
             msg["content"] = cast(str, validated_msg)
 
-        return messages  # type: ignore
+        return messages  # type: ignore

guardrails/run/async_stream_runner.py

@@ -205,13 +205,13 @@ def get_chunk_text(self, chunk: Any, api: Union[PromptCallableBase, None]) -> st
             content = chunk.choices[0].text
             if not finished and content:
                 chunk_text = content
-        except Exception as e:
+        except Exception:
             try:
                 finished = chunk.choices[0].finish_reason
                 content = chunk.choices[0].delta.content
                 if not finished and content:
                     chunk_text = content
-            except Exception as e:
+            except Exception:
                 try:
                     chunk_text = chunk
                 except Exception as e:

guardrails/run/stream_runner.py

@@ -253,13 +253,13 @@ def get_chunk_text(self, chunk: Any, api: Union[PromptCallableBase, None]) -> st
             content = chunk.choices[0].text
             if not finished and content:
                 chunk_text = content
-        except Exception as e:
+        except Exception:
             try:
                 finished = chunk.choices[0].finish_reason
                 content = chunk.choices[0].delta.content
                 if not finished and content:
                     chunk_text = content
-            except Exception as e:
+            except Exception:
                 try:
                     chunk_text = chunk
                 except Exception as e: