Настройка PostgreSQL с pgvector в контейнере Docker

Этот пост объясняет, как настроить локальный экземпляр pgvector, базы данных векторов, в контейнере Docker. Pgvector является популярным вариантом для поиска похожих векторов в Postgres и часто используется в конвейерах RAG (Усилении генерации с помощью поиска). Для начала извлеките последний образ pgvector из репозитория Docker, указав желаемую версию сервера Postgres. Запустите образ, установив пароль root-пользователя и открыв стандартный порт Postgres. Создайте базу данных внутри контейнера с помощью команды docker exec. Подключитесь к базе данных из вашего приложения, установив необходимые зависимости и установив переменную среды DATABASE_URL. Создайте соединение с базой данных с помощью класса pg.Pool и инициализируйте расширение pgvector. Создайте новую таблицу с помощью метода PGVectorStore.initialize. С настройкой vectorStore вы можете добавлять контент и запрашивать контекст с помощью методов vectorStore.addDocuments и vectorStore.similaritySearch. Этот setup предоставляет основу для дальнейшего исследования возможностей pgvector.