Upload core/models/__init__.py with huggingface_hub

core/models/__init__.py

@@ -8,8 +8,7 @@ from .agent import AgentApproval, AgentDraft
 from .base import (
     Base,
     CasePriority,
-    # Enums
-    CaseStatus,
+    CaseStatus,  # Enums; Database setup
     CaseType,
     ReconciliationType,
     SessionLocal,
@@ -17,7 +16,6 @@ from .base import (
     create_engine_and_session,
     create_tables,
     engine,
-    # Database setup
     get_database_url,
     get_db,
     secure_query_execution,
@@ -54,8 +52,11 @@ from .entity import (
     Relationship,
 )
 from .job import Job
+from .performance import PerformanceMetric
+from .security import APIKey, BlockedIP
 from .user import (
     AccessReview,
+    LoginAttempt,
     Project,
     RookieChecklist,
     Team,
@@ -63,6 +64,7 @@ from .user import (
     User,
     UserDevice,
     UserOnboardingState,
+    UserSession,
 )
 
 # Export all models for backward compatibility
@@ -93,6 +95,8 @@ __all__ = [
     "RookieChecklist",
     "TrainingRecord",
     "AccessReview",
+    "LoginAttempt",
+    "UserSession",
     # Case models
     "Case",
     "Transaction",
@@ -123,4 +127,7 @@ __all__ = [
     "AgentDraft",
     "AgentApproval",
     "Job",
+    "PerformanceMetric",
+    "BlockedIP",
+    "APIKey",
 ]