va/agent
0
0
Fork 0
Code Issues 4 Pull Requests 2 Actions Packages Projects Releases Wiki Activity

feat: Integrate GovernancePlugin for emoji filtering in agent responses

Browse Source
This commit is contained in:
Jorge Angel Juarez Vazquez
2026-03-03 18:47:10 +00:00
parent aabbbbe4c4
commit a1bd2b000f
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/va_agent/agent.py
View File

@@ -10,17 +10,20 @@ from google.cloud.firestore_v1.async_client import AsyncClient
from va_agent.auth import auth_headers_provider from va_agent.auth import auth_headers_provider
from va_agent.config import settings from va_agent.config import settings
from va_agent.session import FirestoreSessionService from va_agent.session import FirestoreSessionService
from va_agent.governance import GovernancePlugin
toolset = McpToolset( toolset = McpToolset(
connection_params=StreamableHTTPConnectionParams(url=settings.mcp_remote_url), connection_params=StreamableHTTPConnectionParams(url=settings.mcp_remote_url),
header_provider=auth_headers_provider, header_provider=auth_headers_provider,
) )
governance = GovernancePlugin()
agent = Agent( agent = Agent(
model=settings.agent_model, model=settings.agent_model,
name=settings.agent_name, name=settings.agent_name,
instruction=settings.agent_instructions, instruction=settings.agent_instructions,
tools=[toolset], tools=[toolset],
after_model_callback=governance.after_model_callback,
) )
session_service = FirestoreSessionService( session_service = FirestoreSessionService(