Skip to content

Commit b395dfb

Browse files
committed
fix(safety): 修复 __init__.py 末尾换行和 test_filter_is_registered CI 失败
1 parent 86825ee commit b395dfb

2 files changed

Lines changed: 3 additions & 1 deletion

File tree

tests/tools/safety/test_filter.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -35,6 +35,8 @@ class TestSafetyFilter:
3535

3636
def test_filter_is_registered(self):
3737
"""SafetyFilter should be registered with the TOOL filter type."""
38+
# 确保模块导入触发 @register_tool_filter 装饰器
39+
from trpc_agent_sdk.tools.safety import _safety_filter # noqa: F401
3840
from trpc_agent_sdk.filter import get_tool_filter
3941
filter_instance = get_tool_filter("safety_filter")
4042
assert filter_instance is not None

trpc_agent_sdk/tools/safety/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -53,4 +53,4 @@
5353
"SafetyWrapper",
5454
"ScanInput",
5555
"ScriptType",
56-
]
56+
]

0 commit comments

Comments
 (0)