Update models/__init__.py

models/__init__.py

@@ -1,6 +1,13 @@
-# This file can be empty or used for re-exporting models
+# models/__init__.py
 from .db import create_db_and_tables, get_session, get_session_context
 from .user import User
 from .chat import ChatMessage, ChatSession
 
-__all__ = ["create_db_and_tables", "get_session", "User", "ChatMessage", "ChatSession"]
+__all__ = [
+    "create_db_and_tables",
+    "get_session",
+    "get_session_context",
+    "User",
+    "ChatMessage",
+    "ChatSession"
+]