fix: entrypoint runs create_all + column patches before alembic stamp, fixing fresh-install crash

Author: wisdomqin

backend/entrypoint.sh

@@ -1,9 +1,84 @@
 #!/bin/bash
-# Docker entrypoint: run DB migrations, then start the app.
-# Use bash (not sh/dash) to avoid CRLF/set-option compatibility issues.
+# Docker entrypoint: initialize DB tables, then start the app.
+# Order matters:
+#   1. create_all  - creates all tables using SQLAlchemy models (idempotent)
+#   2. alembic stamp head - tells alembic we are at the latest revision (skips migrations)
+#      For existing installs that may have missing columns, safe ALTER TABLE patches run first.
+#   3. uvicorn - starts the FastAPI app
+
 set -e
 
-echo "[entrypoint] Running database migrations..."
-alembic upgrade head
-echo "[entrypoint] Migrations done - starting uvicorn..."
+echo "[entrypoint] Step 1: Creating/verifying database tables..."
+
+python << 'PYEOF'
+import asyncio, sys
+
+async def main():
+    # Import all models to populate Base.metadata before create_all
+    from app.database import Base, engine
+    import app.models.user           # noqa
+    import app.models.agent          # noqa
+    import app.models.task           # noqa
+    import app.models.llm            # noqa
+    import app.models.tool           # noqa
+    import app.models.audit          # noqa
+    import app.models.skill          # noqa
+    import app.models.channel_config # noqa
+    import app.models.schedule       # noqa
+    import app.models.plaza          # noqa
+    import app.models.activity_log   # noqa
+    import app.models.org            # noqa
+    import app.models.system_settings # noqa
+    import app.models.invitation_code # noqa
+    import app.models.tenant         # noqa
+    import app.models.message        # noqa
+    import app.models.chat_session   # noqa
+
+    # Create all tables that don't exist yet (safe to run on every startup)
+    async with engine.begin() as conn:
+        await conn.run_sync(Base.metadata.create_all)
+    print("[entrypoint] Tables created/verified")
+
+    # Apply safe column patches for existing installs that may be missing columns.
+    # All statements use IF NOT EXISTS so they are fully idempotent.
+    patches = [
+        # Quota fields added in v0.2
+        "ALTER TABLE users ADD COLUMN IF NOT EXISTS quota_message_limit INTEGER DEFAULT 50",
+        "ALTER TABLE users ADD COLUMN IF NOT EXISTS quota_message_period VARCHAR(20) DEFAULT 'permanent'",
+        "ALTER TABLE users ADD COLUMN IF NOT EXISTS quota_messages_used INTEGER DEFAULT 0",
+        "ALTER TABLE users ADD COLUMN IF NOT EXISTS quota_period_start TIMESTAMPTZ",
+        "ALTER TABLE users ADD COLUMN IF NOT EXISTS quota_max_agents INTEGER DEFAULT 2",
+        "ALTER TABLE users ADD COLUMN IF NOT EXISTS quota_agent_ttl_hours INTEGER DEFAULT 48",
+        "ALTER TABLE agents ADD COLUMN IF NOT EXISTS expires_at TIMESTAMPTZ",
+        "ALTER TABLE agents ADD COLUMN IF NOT EXISTS is_expired BOOLEAN DEFAULT FALSE",
+        "ALTER TABLE agents ADD COLUMN IF NOT EXISTS llm_calls_today INTEGER DEFAULT 0",
+        "ALTER TABLE agents ADD COLUMN IF NOT EXISTS max_llm_calls_per_day INTEGER DEFAULT 100",
+        "ALTER TABLE agents ADD COLUMN IF NOT EXISTS llm_calls_reset_at TIMESTAMPTZ",
+        # agent_tools source tracking added later
+        "ALTER TABLE agent_tools ADD COLUMN IF NOT EXISTS source VARCHAR(20) NOT NULL DEFAULT 'system'",
+        "ALTER TABLE agent_tools ADD COLUMN IF NOT EXISTS installed_by_agent_id UUID",
+        # chat_sessions channel tracking
+        "ALTER TABLE chat_sessions ADD COLUMN IF NOT EXISTS source_channel VARCHAR(20) NOT NULL DEFAULT 'web'",
+    ]
+
+    from sqlalchemy import text
+    async with engine.begin() as conn:
+        for sql in patches:
+            try:
+                await conn.execute(text(sql))
+            except Exception as e:
+                print(f"[entrypoint] Patch skipped ({e})")
+
+    await engine.dispose()
+    print("[entrypoint] Column patches applied")
+
+asyncio.run(main())
+PYEOF
+
+echo "[entrypoint] Step 2: Stamping alembic to latest revision..."
+# Mark all migrations as applied so alembic does not re-run them on top of create_all.
+# This is safe: create_all already created tables with all current columns.
+alembic stamp head
+
+echo "[entrypoint] Step 3: Starting uvicorn..."
 exec uvicorn app.main:app --host 0.0.0.0 --port 8000