Claude
1d4625a1c2
Co-Authored-By: Claude Sonnet 5 <[email protected]> Claude-Session: https://claude.ai/code/session_01B38FGKKam5SCGgwgduVAh3
40 lines
2.1 KiB
Bash
Executable File
40 lines
2.1 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
# Starts a throwaway PostgreSQL with the pgvector extension, with no Docker, and prepares the
|
|
# ragdb database. Use it when docker compose is not available. It is what produced the
|
|
# transcripts in docs/output/ in this repository's authoring environment.
|
|
#
|
|
# Requires PostgreSQL 14+ and the pgvector package (Debian/Ubuntu: apt install postgresql-16 postgresql-16-pgvector).
|
|
# Environment: PGDATA (default /tmp/pgrag/data), PGPORT (default 5439).
|
|
# Afterwards: export RAG_PG_URL=jdbc:postgresql://127.0.0.1:${PGPORT:-5439}/ragdb
|
|
set -euo pipefail
|
|
|
|
PGDATA="${PGDATA:-/tmp/pgrag/data}"
|
|
PGPORT="${PGPORT:-5439}"
|
|
SOCKDIR="$(dirname "$PGDATA")"
|
|
BIN="$(ls -d /usr/lib/postgresql/*/bin 2>/dev/null | sort -V | tail -1)"
|
|
[ -x "$BIN/initdb" ] || { echo "PostgreSQL server binaries not found under /usr/lib/postgresql" >&2; exit 1; }
|
|
HERE="$(cd "$(dirname "$0")/.." && pwd)"
|
|
|
|
# PostgreSQL refuses to run as root, so hand the work to the postgres user in that case.
|
|
as_pg() { if [ "$(id -u)" = 0 ]; then su postgres -c "$*"; else bash -c "$*"; fi; }
|
|
|
|
mkdir -p "$SOCKDIR"
|
|
[ "$(id -u)" = 0 ] && chown postgres "$SOCKDIR"
|
|
if [ ! -d "$PGDATA/base" ]; then
|
|
as_pg "'$BIN/initdb' -D '$PGDATA' -A trust >'$SOCKDIR/initdb.log' 2>&1"
|
|
fi
|
|
if ! as_pg "'$BIN/pg_ctl' -D '$PGDATA' status" >/dev/null 2>&1; then
|
|
as_pg "'$BIN/pg_ctl' -D '$PGDATA' -o \"-p $PGPORT -c listen_addresses=127.0.0.1 -c unix_socket_directories=$SOCKDIR\" -l '$SOCKDIR/pg.log' -w start"
|
|
fi
|
|
|
|
PSQL="$BIN/psql -h $SOCKDIR -p $PGPORT -U postgres -v ON_ERROR_STOP=1 -q"
|
|
as_pg "$PSQL -d postgres -tc \"select 1 from pg_roles where rolname='raguser'\"" | grep -q 1 \
|
|
|| as_pg "$PSQL -d postgres -c \"create role raguser login superuser password 'ragpass'\""
|
|
as_pg "$PSQL -d postgres -tc \"select 1 from pg_database where datname='ragdb'\"" | grep -q 1 \
|
|
|| as_pg "$PSQL -d postgres -c 'create database ragdb owner raguser'"
|
|
as_pg "$PSQL -d ragdb -c 'create extension if not exists vector'"
|
|
as_pg "$PSQL -d ragdb -f '$HERE/init.sql'"
|
|
|
|
echo "PostgreSQL is up on 127.0.0.1:$PGPORT (database ragdb, user raguser, password ragpass)"
|
|
echo "export RAG_PG_URL=jdbc:postgresql://127.0.0.1:$PGPORT/ragdb"
|