updated handler exceptions

databases/sql_models/rules/rules.py

@@ -1,7 +1,9 @@
-from sqlalchemy import String, Boolean
-from databases.sql_models.core_mixin import CrudCollection
+from sqlalchemy import String
 from sqlalchemy.orm import mapped_column, Mapped
 
+from databases.language_models.rules.rules import EndpointRestrictionLanguageModel
+from databases.sql_models.core_mixin import CrudCollection
+
 
 class EndpointRestriction(CrudCollection):
     """
@@ -10,6 +12,7 @@ class EndpointRestriction(CrudCollection):
 
     __tablename__ = "endpoint_restriction"
     __exclude__fields__ = []
+    __language_model__ = EndpointRestrictionLanguageModel
 
     endpoint_function: Mapped[str] = mapped_column(
         String, server_default="", comment="Function name of the API endpoint"