diff --git a/app/plugins/builtins/ai_assistant/routes.py b/app/plugins/builtins/ai_assistant/routes.py
index e3e4440..f72e331 100644
--- a/app/plugins/builtins/ai_assistant/routes.py
+++ b/app/plugins/builtins/ai_assistant/routes.py
@@ -17,6 +17,7 @@ from app.plugins.builtins.ai_assistant.models import (
     AIAgent,
     AIChatSession,
     AIModel,
+    AIPreset,
     AIProvider,
 )
 from app.plugins.builtins.ai_assistant.schemas import (