0f3522eea6
<!-- .github/pull_request_template.md --> ## Description <!-- Provide a clear description of the changes in this PR --> ## 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.
42 lines
1.8 KiB
Bash
Executable file
42 lines
1.8 KiB
Bash
Executable file
#!/bin/bash
|
|
|
|
set -e # Exit on error
|
|
echo "Debug mode: $DEBUG"
|
|
echo "Environment: $ENVIRONMENT"
|
|
|
|
# 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=$(alembic upgrade head 2>&1) || {
|
|
if [[ $MIGRATION_OUTPUT == *"UserAlreadyExists"* ]] || [[ $MIGRATION_OUTPUT == *"User default_user@example.com 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 "Database migrations done."
|
|
|
|
echo "Starting server..."
|
|
|
|
# Add startup delay to ensure DB is ready
|
|
sleep 2
|
|
|
|
# Modified Gunicorn startup with error handling
|
|
if [ "$ENVIRONMENT" = "dev" ] || [ "$ENVIRONMENT" = "local" ]; then
|
|
if [ "$DEBUG" = "true" ]; then
|
|
echo "Waiting for the debugger to attach..."
|
|
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
|
|
else
|
|
gunicorn -w 3 -k uvicorn.workers.UvicornWorker -t 30000 --bind=0.0.0.0:8000 --log-level error cognee.api.client:app
|
|
fi
|