lint

Author: dtam

guardrails/actions/reask.py

@@ -297,8 +297,9 @@ def get_reask_setup_for_string(
     if messages is None:
         messages = Messages(
             [
-                {"role": "system", "content": instructions}, 
-                {"role": "user", "content": prompt}]
+                {"role": "system", "content": instructions},
+                {"role": "user", "content": prompt},
+            ]
         )
 
     messages = messages.format(

guardrails/classes/history/call.py

@@ -79,8 +79,8 @@ def prompt_params(self) -> Optional[Dict]:
 
     @property
     def messages(self) -> Optional[Union[Messages, list[dict[str, str]]]]:
-        """The messages as provided by the user when initializing or calling the
-        Guard."""
+        """The messages as provided by the user when initializing or calling
+        the Guard."""
         return self.inputs.messages
 
     @property

guardrails/run/async_stream_runner.py

@@ -200,7 +200,7 @@ async def async_step(
     def get_chunk_text(self, chunk: Any, api: Union[PromptCallableBase, None]) -> str:
         """Get the text from a chunk."""
         chunk_text = ""
-    
+
         try:
             finished = chunk.choices[0].finish_reason
             content = chunk.choices[0].delta.content

guardrails/run/stream_runner.py

@@ -5,7 +5,6 @@
 from guardrails.classes.output_type import OT, OutputTypes
 from guardrails.classes.validation_outcome import ValidationOutcome
 from guardrails.llm_providers import (
-    LiteLLMCallable,
     PromptCallableBase,
 )
 from guardrails.run.runner import Runner

tests/integration_tests/test_assets/custom_llm.py

@@ -1,6 +1,3 @@
-from typing import Dict, List, Optional
-
-
 def mock_llm(
     messages,
     *args,

tests/integration_tests/test_assets/lists_object.py

@@ -16,7 +16,12 @@ class Item(BaseModel):
 PYDANTIC_RAIL_WITH_LIST = List[Item]
 
 
-RAIL_SPEC_WITH_LIST = """
+message = (
+    '<message role="user">'
+    "Create a list of items that may be found in a grocery store."
+    "</message>"
+)
+RAIL_SPEC_WITH_LIST = f"""
 <rail version="0.1">
   <output type="list">
     <object>
@@ -25,7 +30,7 @@ class Item(BaseModel):
     </object>
   </output>
   <messages>
-  <message role="user">Create a list of items that may be found in a grocery store.</message>
+  {message}
   </messages>
 </rail>
 """

tests/integration_tests/test_formatters.py

@@ -25,7 +25,11 @@ class Foo(BaseModel):
         bez: List[str]
 
     g = Guard.from_pydantic(Foo, output_formatter="jsonformer")
-    response = g(model.generate, tokenizer=tokenizer, messages=[{"content": "test","role": "user"}])
+    response = g(
+        model.generate,
+        tokenizer=tokenizer,
+        messages=[{"content": "test", "role": "user"}],
+    )
     validated_output = response.validated_output
     assert isinstance(validated_output, dict)
     assert "bar" in validated_output
@@ -45,7 +49,7 @@ class Foo(BaseModel):
         bez: List[str]
 
     g = Guard.from_pydantic(Foo, output_formatter="jsonformer")
-    response = g(model, messages=[{"content": "Sample:","role": "user"}])
+    response = g(model, messages=[{"content": "Sample:", "role": "user"}])
     validated_output = response.validated_output
     assert isinstance(validated_output, dict)
     assert "bar" in validated_output

tests/unit_tests/actions/test_reask.py

@@ -550,10 +550,7 @@ def test_get_reask_prompt(
         ]
     )
 
-    (
-        reask_schema,
-        reask_messages
-    ) = get_reask_setup(
+    (reask_schema, reask_messages) = get_reask_setup(
         output_type,
         output_schema,
         validation_map={},

tests/unit_tests/test_validator_base.py

@@ -485,7 +485,9 @@ def mock_llm_api(messages, *args, **kwargs):
                 }
             ],
         )
-    assert str(excinfo.value) == "Validation failed for field with errors: must be exactly two words"
+    assert str(excinfo.value) == (
+        "Validation failed for field with errors:" " must be exactly two words"
+    )
     assert isinstance(guard.history.first.exception, ValidationError)
     assert guard.history.first.exception == excinfo.value
 
@@ -565,8 +567,8 @@ async def mock_llm_api(messages, *args, **kwargs) -> str:
             }
         ],
     )
-
-    assert guard.history.first.iterations.first.outputs.validation_response == "What kind"
+    first_iter = guard.history.first.iterations.first
+    assert first_iter.outputs.validation_response == "What kind"
 
     # rail prompt validation
     guard = AsyncGuard.from_rail_string(
@@ -767,4 +769,3 @@ async def custom_llm(messages, *args, **kwargs) -> str:
     assert str(excinfo.value) == unstructured_messages_error
     assert isinstance(guard.history.last.exception, ValidationError)
     assert guard.history.last.exception == excinfo.value
-