make check formatting

Author: steven10a

examples/basic/tool_guardrails.py

@@ -97,7 +97,7 @@ def reject_phone_numbers(data: ToolOutputGuardrailData) -> ToolGuardrailFunction
             message="User data not retrieved as it contains a phone number which is restricted.",
             output_info={"redacted": "phone_number"},
         )
-    return ToolGuardrailFunctionOutput()
+    return ToolGuardrailFunctionOutput(output_info="Phone number check passed")
 
 
 # Apply guardrails to tools

src/agents/tool_guardrails.py

@@ -2,7 +2,7 @@
 
 import inspect
 from collections.abc import Awaitable
-from dataclasses import dataclass
+from dataclasses import dataclass, field
 from typing import TYPE_CHECKING, Any, Callable, Generic, Literal, overload
 
 from typing_extensions import TypedDict, TypeVar
@@ -66,19 +66,16 @@ class ToolGuardrailFunctionOutput:
     information about the checks it performed and granular results.
     """
 
-    behavior: RejectContentBehavior | RaiseExceptionBehavior | AllowBehavior | None = None
+    behavior: RejectContentBehavior | RaiseExceptionBehavior | AllowBehavior = field(
+        default_factory=lambda: AllowBehavior(type="allow")
+    )
     """
     Defines how the system should respond when this guardrail result is processed.
-    - None/allow: Allow normal tool execution to continue without interference (default)
+    - allow: Allow normal tool execution to continue without interference (default)
     - reject_content: Reject the tool call/output but continue execution with a message to the model
     - raise_exception: Halt execution by raising a ToolGuardrailTripwireTriggered exception
     """
 
-    def __post_init__(self) -> None:
-        """Set default behavior if none specified."""
-        if self.behavior is None:
-            self.behavior = AllowBehavior(type="allow")
-
     @classmethod
     def allow(cls, output_info: Any = None) -> ToolGuardrailFunctionOutput:
         """Create a guardrail output that allows the tool execution to continue normally.

tests/extensions/memory/test_advanced_sqlite_session.py

@@ -70,6 +70,8 @@ def create_mock_run_result(
         final_output="test output",
         input_guardrail_results=[],
         output_guardrail_results=[],
+        tool_input_guardrail_results=[],
+        tool_output_guardrail_results=[],
         context_wrapper=context_wrapper,
         _last_agent=agent,
     )

tests/test_result_cast.py

@@ -14,6 +14,8 @@ def create_run_result(final_output: Any) -> RunResult:
         final_output=final_output,
         input_guardrail_results=[],
         output_guardrail_results=[],
+        tool_input_guardrail_results=[],
+        tool_output_guardrail_results=[],
         _last_agent=Agent(name="test"),
         context_wrapper=RunContextWrapper(context=None),
     )