style(tools): fix flake8 warnings — unused imports and line length

Author: DNKYr

tests/tools/safety/test_filter.py

@@ -7,7 +7,6 @@
 
 from unittest.mock import AsyncMock
 from unittest.mock import MagicMock
-from unittest.mock import patch
 
 import pytest
 

tests/tools/safety/test_policy.py

@@ -11,8 +11,6 @@
 from trpc_agent_sdk.tools.safety._policy import (
     SafetyPolicy,
     PolicyRuleConfig,
-    WhitelistConfig,
-    BlocklistConfig,
 )
 from trpc_agent_sdk.tools.safety._types import RiskType, Decision, RiskLevel
 

tests/tools/safety/test_scanner.py

@@ -5,8 +5,6 @@
 # tRPC-Agent-Python is licensed under Apache-2.0.
 """Core integration tests for the ToolSafetyScanner."""
 
-import pytest
-
 from trpc_agent_sdk.tools.safety._types import Decision
 from trpc_agent_sdk.tools.safety._types import RiskLevel
 from trpc_agent_sdk.tools.safety._types import RiskType

trpc_agent_sdk/tools/safety/_audit.py

@@ -15,7 +15,6 @@
 from datetime import datetime
 from datetime import timezone
 from pathlib import Path
-from typing import Optional
 
 from ._types import AuditEvent
 from ._types import ScanReport

trpc_agent_sdk/tools/safety/_policy.py

@@ -9,7 +9,6 @@
 """
 
 from pathlib import Path
-from typing import Optional
 
 import yaml
 from pydantic import BaseModel

trpc_agent_sdk/tools/safety/_rules.py

@@ -191,7 +191,7 @@ class AstRule:
         message="Potential sensitive information exposure detected",
         recommendation="Do not log or output API keys, tokens, or passwords",
         patterns=[
-            r"(print|write|log|logger)\.?[^(]*\([^)]*(api_key|API_KEY|password|PASSWORD|token|TOKEN|secret|SECRET|private_key|PRIVATE_KEY)",
+            r"(print|write|log|logger)\.?[^(]*\([^)]*(api_key|API_KEY|password|PASSWORD|token|TOKEN|secret|SECRET|private_key|PRIVATE_KEY)",  # noqa: E501
             r"(api_key|API_KEY|password|PASSWORD|token|TOKEN|secret|SECRET)\s*=\s*[^#\n]{3,}",
         ],
     ),