diff --git a/src/paperless_ai/base_model.py b/src/paperless_ai/base_model.py
index 44fe75f5b..61f93b467 100644
--- a/src/paperless_ai/base_model.py
+++ b/src/paperless_ai/base_model.py
@@ -2,6 +2,8 @@ from pydantic import BaseModel
 
 
 class DocumentClassifierSchema(BaseModel):
+    """Schema for document classification suggestions."""
+
     title: str
     tags: list[str]
     correspondents: list[str]