Update models/__init__.py

models/__init__.py

@@ -1,13 +1,16 @@
 # models/__init__.py
 from .db import create_db_and_tables, get_session, get_session_context
-from .user import User
-from .chat import ChatMessage, ChatSession
+from .user import User, UserCreate, UserPublic # Export all relevant User models/schemas
+from .chat import ChatMessage, ChatSession, ChatMessageCreate # Export all relevant Chat models/schemas
 
 __all__ = [
     "create_db_and_tables",
     "get_session",
     "get_session_context",
     "User",
+    "UserCreate",
+    "UserPublic",
     "ChatMessage",
-    "ChatSession"
+    "ChatSession",
+    "ChatMessageCreate",
 ]