feat: Add before_model_callback to Agent initialization

2026-03-04 16:59:46 +00:00
parent 1c255c5ccf
commit 7d5309c9d0
1 changed files with 1 additions and 0 deletions
--- a/src/va_agent/agent.py
+++ b/src/va_agent/agent.py
@@ -23,6 +23,7 @@ agent = Agent(
    name=settings.agent_name,
    instruction=settings.agent_instructions,
    tools=[toolset],
+    before_model_callback=governance.before_model_callback,
    after_model_callback=governance.after_model_callback,
 )