39 lines
1.2 KiB
Bash
39 lines
1.2 KiB
Bash
|
#!/bin/bash
|
||
|
|
set -e
|
||
|
|
|
||
|
|
# PostgreSQL initialization script for AI stack
|
||
|
|
# This script runs on first database initialization
|
||
|
|
# Creates all databases required by AI services
|
||
|
|
|
||
|
|
echo "Starting AI stack database initialization..."
|
||
|
|
|
||
|
|
psql -v ON_ERROR_STOP=1 --username "$POSTGRES_USER" --dbname "$POSTGRES_DB" <<-EOSQL
|
||
|
|
-- Create databases for AI services
|
||
|
|
-- Open WebUI database
|
||
|
|
SELECT 'CREATE DATABASE openwebui'
|
||
|
|
WHERE NOT EXISTS (SELECT FROM pg_database WHERE datname = 'openwebui')\gexec
|
||
|
|
|
||
|
|
-- LiteLLM proxy database
|
||
|
|
SELECT 'CREATE DATABASE litellm'
|
||
|
|
WHERE NOT EXISTS (SELECT FROM pg_database WHERE datname = 'litellm')\gexec
|
||
|
|
|
||
|
|
-- Grant privileges to all databases
|
||
|
|
GRANT ALL PRIVILEGES ON DATABASE openwebui TO $POSTGRES_USER;
|
||
|
|
GRANT ALL PRIVILEGES ON DATABASE litellm TO $POSTGRES_USER;
|
||
|
|
|
||
|
|
-- Log success
|
||
|
|
SELECT 'AI stack databases initialized:' AS status;
|
||
|
|
SELECT datname FROM pg_database
|
||
|
|
WHERE datname IN ('openwebui', 'litellm')
|
||
|
|
ORDER BY datname;
|
||
|
|
EOSQL
|
||
|
|
|
||
|
|
echo ""
|
||
|
|
echo "✓ PostgreSQL initialization completed"
|
||
|
|
echo "✓ All AI stack databases created successfully"
|
||
|
|
echo ""
|
||
|
|
echo "Databases available:"
|
||
|
|
echo " • openwebui - Open WebUI application database"
|
||
|
|
echo " • litellm - LiteLLM proxy database"
|
||
|
|
echo ""
|