fix: entrypoint.sh to not fail on first docker up, improved handling …

…of migrations, signals and errors. (#546)

<!-- .github/pull_request_template.md -->

## Description
<!-- Provide a clear description of the changes in this PR -->
In it's current form, the entrypoint.sh script will run but fail with
exit code 3 on the first docker compose up. Technically, running docker
compose up a second time will not throw the same error and the
application works fine. The new changes will improve the first time user
experience and improve on some other aspects.

Summary of Changes:
1- entrypoint.sh to not fail with exit code 3 on first docker up.
2- Improved error and signal handling with set -e.
3- Improved database migration, verification and error handling. Avoids
schema version mismatch and ensures db schema is always in sync with
application code.
4- Added exec before Gunicorn commands to ensure proper signal handling.

## DCO Affirmation
I affirm that all code in every commit of this pull request conforms to
the terms of the Topoteretes Developer Certificate of Origin


<!-- This is an auto-generated comment: release notes by coderabbit.ai
-->
## Summary by CodeRabbit

- **Chores**
	- Improved error handling for smoother database migrations and startup.
	- Updated process management to ensure reliable application launch.
- Optimized worker configuration and introduced a startup delay to
guarantee database readiness.
<!-- end of auto-generated comment: release notes by coderabbit.ai -->

---------

Signed-off-by: soekja <[email protected]>
Co-authored-by: soekja <[email protected]>
Co-authored-by: Vasilije <[email protected]>

entrypoint.sh

@@ -1,33 +1,42 @@
 #!/bin/bash
 
+set -e  # Exit on error
 echo "Debug mode: $DEBUG"
 echo "Environment: $ENVIRONMENT"
 
-
-# # Run Alembic migrations
-# echo "Running database migrations..."
-# poetry run alembic upgrade head
-
-# # Check if the migrations were successful
-# if [ $? -eq 0 ]; then
-#     echo "Migrations completed successfully."
-# else
-#     echo "Migration failed, exiting."
-#     exit 1
-# fi
-
+# Run Alembic migrations with proper error handling
+
+# Note on UserAlreadyExists error handling:
+# During database migrations, we attempt to create a default user. If this user
+# already exists (e.g., from a previous deployment or migration), it's not a
+# critical error and shouldn't prevent the application from starting. This is
+# different from other migration errors which could indicate database schema
+# inconsistencies and should cause the startup to fail. This check allows for
+# smooth redeployments and container restarts while maintaining data integrity.
+echo "Running database migrations..."
+MIGRATION_OUTPUT=$(poetry run alembic upgrade head 2>&1) || {
+    if [[ $MIGRATION_OUTPUT == *"UserAlreadyExists"* ]] || [[ $MIGRATION_OUTPUT == *"User [email protected] already exists"* ]]; then
+        echo "Warning: Default user already exists, continuing startup..."
+    else
+        echo "Migration failed with unexpected error:"
+        echo "$MIGRATION_OUTPUT"
+        exit 1
+    fi
+}
 
 echo "Starting Gunicorn"
 
-if [ "$ENVIRONMENT" = "dev" ]; then
-  if [ "$DEBUG" = true ]; then
-    echo "Waiting for the debugger to attach..."
+# Add startup delay to ensure DB is ready
+sleep 2
 
-    python -m debugpy --wait-for-client --listen 0.0.0.0:5678 -m gunicorn -w 3 -k uvicorn.workers.UvicornWorker -t 30000 --bind=0.0.0.0:8000 --log-level debug --reload cognee.api.client:app
-  else
-    gunicorn -w 3 -k uvicorn.workers.UvicornWorker -t 30000 --bind=0.0.0.0:8000 --log-level debug --reload cognee.api.client:app
-  fi
+# Modified Gunicorn startup with error handling
+if [ "$ENVIRONMENT" = "dev" ]; then
+    if [ "$DEBUG" = true ]; then
+        echo "Waiting for the debugger to attach..."
+        exec python -m debugpy --wait-for-client --listen 0.0.0.0:5678 -m gunicorn -w 3 -k uvicorn.workers.UvicornWorker -t 30000 --bind=0.0.0.0:8000 --log-level debug --reload cognee.api.client:app
+    else
+        exec gunicorn -w 3 -k uvicorn.workers.UvicornWorker -t 30000 --bind=0.0.0.0:8000 --log-level debug --reload cognee.api.client:app
+    fi
 else
-  gunicorn -w 3 -k uvicorn.workers.UvicornWorker -t 30000 --bind=0.0.0.0:8000 --log-level error cognee.api.client:app
-  # python ./cognee/api/client.py
-fi
+    exec gunicorn -w 3 -k uvicorn.workers.UvicornWorker -t 30000 --bind=0.0.0.0:8000 --log-level error cognee.api.client:app 
+fi