feat: Add emojis filter for LLM response #21
@@ -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(
|
||||||
|
|||||||
Reference in New Issue
Block a user