@@ -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