avm/hassio-addons
1
0
Fork 0
mirror of https://github.com/alexbelgium/hassio-addons.git synced 2026-02-06 11:54:53 +01:00
Code Issues Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
alexbelgium
2025-05-25 16:11:02 +02:00
parent 0665a8d7b6
commit ea8f9d83d4
5 changed files with 60 additions and 56 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
immich/rootfs/etc/cont-init.d/99-run.sh
View File

@@ -150,16 +150,16 @@ EOF
bashio::log.info "Database setup completed successfully." bashio::log.info "Database setup completed successfully."
} }
# Function to check if vectors extension is enabled # Function to check if vectorchord extension is enabled
check_vector_extension() { check_vectorchord_extension() {
echo "Checking if 'vectors' extension is enabled..." echo "Checking if 'vectorchord' extension is enabled..."
RESULT=$(psql "postgres://$DB_USERNAME:$DB_PASSWORD@$DB_HOSTNAME:$DB_PORT" -tAc "SELECT extname FROM pg_extension WHERE extname = 'vectors';") RESULT=$(psql "postgres://$DB_USERNAME:$DB_PASSWORD@$DB_HOSTNAME:$DB_PORT" -tAc "SELECT extname FROM pg_extension WHERE extname = 'vectorchord';")
if [[ "$RESULT" == "vectors" ]]; then if [[ "$RESULT" == "vectorchord" ]]; then
echo "✅ 'vectors' extension is enabled." echo "✅ 'vectorchord' extension is enabled."
exit 0 exit 0
else else
bashio::log.warning "❌ 'vectors' extension is NOT enabled." bashio::log.warning "❌ 'vectorchord' extension is NOT enabled."
return 1 return 1
fi fi
} }
@@ -185,4 +185,4 @@ export_db_env
setup_root_user setup_root_user
setup_database setup_database
check_vector_extension check_vectorchord_extension

3
postgres/CHANGELOG.md
View File

@@ -1,3 +1,6 @@
- BREAKING CHANGE : please backup your database before updating
- Remove vector.rs and switch to VectorChord to support immich https://github.com/immich-app/immich/releases/tag/v1.133.0
## 15.7-29 (15-02-2025) ## 15.7-29 (15-02-2025)
- Minor bugs fixed - Minor bugs fixed
## 15.7-28 (15-02-2025) ## 15.7-28 (15-02-2025)

4
postgres/build.json
View File

@@ -1,7 +1,7 @@
{ {
"build_from": { "build_from": {
"aarch64": "tensorchord/pgvecto-rs:pg15-v0.3.0", "aarch64": "ghcr.io/immich-app/postgres:15-vectorchord0.3.0-pgvectors0.3.0",
"amd64": "tensorchord/pgvecto-rs:pg15-v0.3.0", "amd64": "ghcr.io/immich-app/postgres:15-vectorchord0.3.0-pgvectors0.3.0",
"armv7": "postgres:15-alpine" "armv7": "postgres:15-alpine"
}, },
"codenotary": { "codenotary": {

2
postgres/config.json
View File

@@ -6,7 +6,7 @@
], ],
"backup": "cold", "backup": "cold",
"codenotary": "alexandrep.github@gmail.com", "codenotary": "alexandrep.github@gmail.com",
"description": "Postgres 15 with pgvecto.rs support", "description": "Postgres 15 with VectorChord support",
"environment": { "environment": {
"CONFIG_LOCATION": "/config/postgresql.conf", "CONFIG_LOCATION": "/config/postgresql.conf",
"PGDATA": "/config/database" "PGDATA": "/config/database"

91
postgres/rootfs/etc/cont-init.d/99-run.sh
View File

@@ -9,7 +9,7 @@ set -e
CONFIG_HOME="/config" CONFIG_HOME="/config"
PGDATA="${PGDATA:-/config/database}" PGDATA="${PGDATA:-/config/database}"
PG_VERSION_FILE="$PGDATA/pg_major_version" PG_VERSION_FILE="$PGDATA/pg_major_version"
VECTOR_VERSION_FILE="$PGDATA/pgvector_version" VECTORCHORD_VERSION_FILE="$PGDATA/vectorchord_version"
# Define current PostgreSQL major version # Define current PostgreSQL major version
PG_MAJOR_VERSION="${PG_MAJOR:-15}" PG_MAJOR_VERSION="${PG_MAJOR:-15}"
@@ -46,11 +46,11 @@ cd /config || true
bashio::log.info "Starting PostgreSQL..." bashio::log.info "Starting PostgreSQL..."
if [ "$(bashio::info.arch)" = "armv7" ]; then if [ "$(bashio::info.arch)" = "armv7" ]; then
bashio::log.warning "ARMv7 detected: Starting without vectors.so" bashio::log.warning "ARMv7 detected: Starting without vectorchord.so"
docker-entrypoint.sh postgres & true docker-entrypoint.sh postgres & true
exit 0 exit 0
else else
docker-entrypoint.sh postgres -c shared_preload_libraries=vectors.so -c search_path="public, vectors" & true docker-entrypoint.sh postgres -c shared_preload_libraries=vectorchord.so -c search_path="public, vectorchord" & true
fi fi
############################### ###############################
@@ -94,7 +94,7 @@ update_postgres() {
export BACKUP_DIR="/config/backups" export BACKUP_DIR="/config/backups"
export PSQL_VERSION="$PG_MAJOR_VERSION" export PSQL_VERSION="$PG_MAJOR_VERSION"
# Install binaries # Install binaries
apt-get update &>/dev/null apt-get update &>/dev/null
install -y procps rsync postgresql-$PG_MAJOR_VERSION postgresql-$OLD_PG_VERSION &>/dev/null install -y procps rsync postgresql-$PG_MAJOR_VERSION postgresql-$OLD_PG_VERSION &>/dev/null
@@ -110,65 +110,66 @@ update_postgres() {
} }
##################################### #####################################
# Enable & Upgrade pgvector.rs # # Enable & Upgrade vectorchord #
##################################### #####################################
# Function: Check if 'vectors' extension is enabled # Function: Check if 'vectorchord' extension is enabled
check_vector_extension() { check_vectorchord_extension() {
bashio::log.info "Checking if 'vectors' extension is enabled..." bashio::log.info "Checking if 'vectorchord' extension is enabled..."
local result local result
result=$(psql "postgres://$DB_USERNAME:$DB_PASSWORD@$DB_HOSTNAME:$DB_PORT" \ result=$(psql "postgres://$DB_USERNAME:$DB_PASSWORD@$DB_HOSTNAME:$DB_PORT" \
-tAc "SELECT extname FROM pg_extension WHERE extname = 'vectors';") -tAc "SELECT extname FROM pg_extension WHERE extname = 'vectorchord';")
if [[ "$result" == "vectors" ]]; then if [[ "$result" == "vectorchord" ]]; then
bashio::log.info "'vectors' extension is enabled." bashio::log.info "'vectorchord' extension is enabled."
return 0 return 0
else else
bashio::log.error "'vectors' extension is NOT enabled." bashio::log.error "'vectorchord' extension is NOT enabled."
return 1 return 1
fi fi
} }
# Function: Enable (or re-create) 'vectors' extension # Function: Enable (or re-create) 'vectorchord' extension
enable_vector_extension() { enable_vectorchord_extension() {
bashio::log.info "Enabling 'vectors' extension..." bashio::log.info "Enabling 'vectorchord' extension..."
psql "postgres://$DB_USERNAME:$DB_PASSWORD@$DB_HOSTNAME:$DB_PORT" -c "DROP EXTENSION IF EXISTS vectors; CREATE EXTENSION vectors;" >/dev/null 2>&1 psql "postgres://$DB_USERNAME:$DB_PASSWORD@$DB_HOSTNAME:$DB_PORT" -c "DROP EXTENSION IF EXISTS vectorchord;" >/dev/null 2>&1
psql "postgres://$DB_USERNAME:$DB_PASSWORD@$DB_HOSTNAME:$DB_PORT" -c "ALTER EXTENSION vectors UPDATE; SELECT pgvectors_upgrade();" >/dev/null 2>&1 psql "postgres://$DB_USERNAME:$DB_PASSWORD@$DB_HOSTNAME:$DB_PORT" -c "CREATE EXTENSION vectorchord;" >/dev/null 2>&1
psql "postgres://$DB_USERNAME:$DB_PASSWORD@$DB_HOSTNAME:$DB_PORT" -c "ALTER EXTENSION vectorchord UPDATE;" >/dev/null 2>&1
} }
# Function: Store the current pgvector.rs version in a file # Function: Store the current vectorchord version in a file
store_vector_version() { store_vectorchord_version() {
local version local version
version=$(psql "postgres://$DB_USERNAME:$DB_PASSWORD@$DB_HOSTNAME:$DB_PORT" \ version=$(psql "postgres://$DB_USERNAME:$DB_PASSWORD@$DB_HOSTNAME:$DB_PORT" \
-tAc "SELECT extversion FROM pg_extension WHERE extname = 'vectors';") -tAc "SELECT extversion FROM pg_extension WHERE extname = 'vectorchord';")
echo "$version" > "$VECTOR_VERSION_FILE" echo "$version" > "$VECTORCHORD_VERSION_FILE"
} }
# Function: Detect previous and new pgvector.rs versions, and upgrade if needed # Function: Detect previous and new vectorchord versions, and upgrade if needed
upgrade_vector_extension() { upgrade_vectorchord_extension() {
local current_version desired_version local current_version desired_version
current_version=$(cat "$VECTOR_VERSION_FILE" 2>/dev/null || echo "unknown") current_version=$(cat "$VECTORCHORD_VERSION_FILE" 2>/dev/null || echo "unknown")
desired_version=$(psql "postgres://$DB_USERNAME:$DB_PASSWORD@$DB_HOSTNAME:$DB_PORT" \ desired_version=$(psql "postgres://$DB_USERNAME:$DB_PASSWORD@$DB_HOSTNAME:$DB_PORT" \
-tAc "SELECT extversion FROM pg_extension WHERE extname = 'vectors';") -tAc "SELECT extversion FROM pg_extension WHERE extname = 'vectorchord';")
if [[ "$current_version" != "$desired_version" ]]; then if [[ "$current_version" != "$desired_version" ]]; then
bashio::log.warning "Upgrading 'vectors' extension from version $current_version → $desired_version..." bashio::log.warning "Upgrading 'vectorchord' extension from version $current_version → $desired_version..."
psql "postgres://$DB_USERNAME:$DB_PASSWORD@$DB_HOSTNAME:$DB_PORT" -c "ALTER EXTENSION vectors UPDATE;" >/dev/null 2>&1 psql "postgres://$DB_USERNAME:$DB_PASSWORD@$DB_HOSTNAME@$DB_PORT" -c "ALTER EXTENSION vectorchord UPDATE;" >/dev/null 2>&1
# Cleanup outdated indexes # Cleanup outdated indexes if needed (customize this line as needed for your DB schema)
bashio::log.info "Cleaning up outdated vector indexes..." bashio::log.info "Cleaning up outdated vector indexes (if any)..."
psql "postgres://$DB_USERNAME:$DB_PASSWORD@$DB_HOSTNAME:$DB_PORT" \ psql "postgres://$DB_USERNAME:$DB_PASSWORD@$DB_HOSTNAME:$DB_PORT" \
-c "DROP INDEX IF EXISTS clip_index;" >/dev/null 2>&1 -c "DROP INDEX IF EXISTS clip_index;" >/dev/null 2>&1
# Store new pgvector version # Store new vectorchord version
echo "$desired_version" > "$VECTOR_VERSION_FILE" echo "$desired_version" > "$VECTORCHORD_VERSION_FILE"
else else
bashio::log.info "'vectors' extension is already at the latest version ($desired_version)." bashio::log.info "'vectorchord' extension is already at the latest version ($desired_version)."
fi fi
} }
# Function: Troubleshoot vector extension # Function: Troubleshoot vectorchord extension
troubleshoot_vector_extension() { troubleshoot_vectorchord_extension() {
bashio::log.error "Troubleshooting pgvector.rs installation..." bashio::log.error "Troubleshooting vectorchord installation..."
if ! pg_isready -h "$DB_HOSTNAME" -p "$DB_PORT" -U "$DB_USERNAME" >/dev/null 2>&1; then if ! pg_isready -h "$DB_HOSTNAME" -p "$DB_PORT" -U "$DB_USERNAME" >/dev/null 2>&1; then
bashio::log.error "PostgreSQL is not running or unreachable." bashio::log.error "PostgreSQL is not running or unreachable."
@@ -177,9 +178,9 @@ troubleshoot_vector_extension() {
local ext_check local ext_check
ext_check=$(psql "postgres://$DB_USERNAME:$DB_PASSWORD@$DB_HOSTNAME:$DB_PORT" \ ext_check=$(psql "postgres://$DB_USERNAME:$DB_PASSWORD@$DB_HOSTNAME:$DB_PORT" \
-tAc "SELECT count(*) FROM pg_available_extensions WHERE name = 'vectors';") -tAc "SELECT count(*) FROM pg_available_extensions WHERE name = 'vectorchord';")
if [[ "$ext_check" -eq 0 ]]; then if [[ "$ext_check" -eq 0 ]]; then
bashio::log.error "'vectors' extension is missing. Ensure pgvector.rs is installed." bashio::log.error "'vectorchord' extension is missing. Ensure vectorchord is installed."
exit 1 exit 1
fi fi
} }
@@ -188,20 +189,20 @@ troubleshoot_vector_extension() {
# Main Extension Handling # # Main Extension Handling #
################################### ###################################
# Store previous vector version # Store previous vectorchord version
update_postgres update_postgres
if ! check_vector_extension; then if ! check_vectorchord_extension; then
enable_vector_extension enable_vectorchord_extension
fi fi
# Store previous vector version # Store previous vectorchord version
store_vector_version store_vectorchord_version
# Upgrade vector extension if needed # Upgrade vectorchord extension if needed
upgrade_vector_extension upgrade_vectorchord_extension
# Final verification # Final verification
check_vector_extension || troubleshoot_vector_extension check_vectorchord_extension || troubleshoot_vectorchord_extension
bashio::log.info "All initialization steps completed successfully!" ) & true bashio::log.info "All initialization steps completed successfully!" ) & true