Add rag module: Spring AI 2.0 RAG with pgvector, chunking, reranking and a faithfulness check

Co-Authored-By: Claude Sonnet 5 <[email protected]> Claude-Session: https://claude.ai/code/session_01B38FGKKam5SCGgwgduVAh3
2026-09-21 19:09:05 +00:00
commit 1d4625a1c2
62 changed files with 3715 additions and 0 deletions
@@ -0,0 +1,39 @@
+#!/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"