diff --git a/app/plugins/builtins/ai_assistant/routes.py b/app/plugins/builtins/ai_assistant/routes.py
index b09ac47..5d3eafb 100644
--- a/app/plugins/builtins/ai_assistant/routes.py
+++ b/app/plugins/builtins/ai_assistant/routes.py
@@ -12,6 +12,7 @@ from sqlalchemy import select
 from sqlalchemy.ext.asyncio import AsyncSession
 
 from app.core.db import get_db, set_tenant_context
+from app.core.visibility import apply_visibility_filter
 from app.deps import get_current_user, require_permission
 from app.plugins.builtins.ai_assistant.models import (
     AIAgent,