From 7d5309c9d0641d816ed627fcbc5a4d708a319f3e Mon Sep 17 00:00:00 2001
From: A8080816 <angel.juarez.vazquez@banorte.com>
Date: Wed, 4 Mar 2026 16:59:46 +0000
Subject: [PATCH] feat: Add before_model_callback to Agent initialization

---
 src/va_agent/agent.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/va_agent/agent.py b/src/va_agent/agent.py
index cbccff9..7bd1224 100644
--- 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,
 )