Skip to content

Commit f201b57

Browse files
committed
mark agent callback as TODO
1 parent 6c9625b commit f201b57

File tree

2 files changed

+4
-31
lines changed

2 files changed

+4
-31
lines changed

docs/content/5.tools/5.guardrail-tools.md

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -8,8 +8,6 @@ navigation:
88
## 概述
99

1010
VeADK 基于 Agent 的插件机制,提供了内容安全护栏工具 `content_safety`。该工具通过以下回调函数嵌入 Agent 的执行流程,实现多阶段内容审计:
11-
- Before Agent Callback
12-
- After Agent Callback
1311
- Before Model Callback
1412
- After Model Callback
1513
- Before Tool Callback
@@ -34,12 +32,11 @@ agent = Agent(
3432
name="robot",
3533
description="A robot can help user.",
3634
instruction="Talk with user friendly.",
37-
before_agent_callback=content_safety.before_agent_callback,
38-
before_model_callback=content_safety.before_model_callback,
35+
# before_model_callback=content_safety.before_model_callback, # TODO
3936
after_model_callback=content_safety.after_model_callback,
4037
before_tool_callback=content_safety.before_tool_callback,
4138
after_tool_callback=content_safety.after_tool_callback,
42-
after_agent_callback=content_safety.after_agent_callback
39+
# after_agent_callback=content_safety.after_agent_callback # TODO
4340
)
4441

4542
runner = Runner(agent=agent)

veadk/tools/builtin_tools/llm_shield.py

Lines changed: 2 additions & 26 deletions
Original file line numberDiff line numberDiff line change
@@ -45,12 +45,10 @@ class LLMShieldPlugin(BasePlugin):
4545
```python
4646
from veadk.tools.builtin_tools.llm_shield import content_safety
4747
agent = Agent(
48-
before_agent_callback=content_safety.before_agent_callback,
4948
before_model_callback=content_safety.before_model_callback,
5049
after_model_callback=content_safety.after_model_callback,
5150
before_tool_callback=content_safety.before_tool_callback,
5251
after_tool_callback=content_safety.after_tool_callback,
53-
after_agent_callback=content_safety.after_agent_callback
5452
)
5553
```
5654
"""
@@ -205,33 +203,11 @@ def _request_llm_shield(self, message: str, role: str) -> Optional[str]:
205203
def before_agent_callback(
206204
self, callback_context: CallbackContext, **kwargs
207205
) -> None:
208-
"""
209-
Callback executed before agent processing.
210-
211-
Currently implements no specific logic for pre-agent processing.
212-
213-
Args:
214-
callback_context (CallbackContext): The callback execution context
215-
**kwargs: Additional keyword arguments
216-
217-
Returns:
218-
None
219-
"""
206+
# TODO: Implement agent-level input validation and context analysis
220207
return None
221208

222209
def after_agent_callback(self, callback_context: CallbackContext, **kwargs) -> None:
223-
"""
224-
Callback executed after agent processing.
225-
226-
Currently implements no specific logic for post-agent processing.
227-
228-
Args:
229-
callback_context (CallbackContext): The callback execution context
230-
**kwargs: Additional keyword arguments
231-
232-
Returns:
233-
None
234-
"""
210+
# TODO: Implement post-agent analysis and context analysis
235211
return None
236212

237213
def before_model_callback(

0 commit comments

Comments
 (0)