From a1bd2b000f2a60b78ef30d487016b1173720c515 Mon Sep 17 00:00:00 2001
From: A8080816 <angel.juarez.vazquez@banorte.com>
Date: Tue, 3 Mar 2026 18:47:10 +0000
Subject: [PATCH] feat: Integrate GovernancePlugin for emoji filtering in agent
 responses

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

diff --git a/src/va_agent/agent.py b/src/va_agent/agent.py
index 3ffcc13..cbccff9 100644
--- a/src/va_agent/agent.py
+++ b/src/va_agent/agent.py
@@ -10,17 +10,20 @@ from google.cloud.firestore_v1.async_client import AsyncClient
 from va_agent.auth import auth_headers_provider
 from va_agent.config import settings
 from va_agent.session import FirestoreSessionService
+from va_agent.governance import GovernancePlugin
 
 toolset = McpToolset(
     connection_params=StreamableHTTPConnectionParams(url=settings.mcp_remote_url),
     header_provider=auth_headers_provider,
 )
 
+governance = GovernancePlugin()
 agent = Agent(
     model=settings.agent_model,
     name=settings.agent_name,
     instruction=settings.agent_instructions,
     tools=[toolset],
+    after_model_callback=governance.after_model_callback,
 )
 
 session_service = FirestoreSessionService(