Fix linting issues

app/api/routes/__init__.py

@@ -1,3 +1,5 @@
 from app.api.routes.api import router as api_router
 from app.api.routes.health import router as health_router
-from app.api.routes.messages import router as messages_router
+from app.api.routes.messages import router as messages_router
+
+__all__ = ["api_router", "health_router", "messages_router"]

app/db/session.py

@@ -1,6 +1,5 @@
 from pathlib import Path
 from sqlalchemy import create_engine
-from sqlalchemy.ext.declarative import declarative_base
 from sqlalchemy.orm import sessionmaker
 
 # Create DB directory if it doesn't exist

app/models/__init__.py

@@ -1,2 +1,4 @@
 from app.models.base import Base
-from app.models.message import Message
+from app.models.message import Message
+
+__all__ = ["Base", "Message"]

app/schemas/__init__.py

@@ -1 +1,3 @@
-from app.schemas.message import MessageBase, MessageCreate, MessageResponse
+from app.schemas.message import MessageBase, MessageCreate, MessageResponse
+
+__all__ = ["MessageBase", "MessageCreate", "MessageResponse"]

app/schemas/message.py

@@ -1,5 +1,4 @@
 from datetime import datetime
-from typing import Optional
 from pydantic import BaseModel
 
 

migrations/env.py

@@ -5,6 +5,10 @@ from sqlalchemy import pool
 
 from alembic import context
 
+# add your model's MetaData object here
+# for 'autogenerate' support
+from app.models.base import Base
+
 # this is the Alembic Config object, which provides
 # access to the values within the .ini file in use.
 config = context.config
@@ -13,9 +17,6 @@ config = context.config
 # This line sets up loggers basically.
 fileConfig(config.config_file_name)
 
-# add your model's MetaData object here
-# for 'autogenerate' support
-from app.models.base import Base
 target_metadata = Base.metadata
 
 # other values from the config, defined by the needs of env.py,