azaddjan/pgvector

概述：使用Docker运行PgVector

本文档提供在Docker容器中运行包含PgVector扩展的PostgreSQL的详细步骤，包括验证扩展、创建表、插入数据及执行查询等操作说明。

核心功能与特性

• 预安装PgVector扩展，支持向量数据类型及相关操作
• 基于PostgreSQL 17构建，确保数据库功能完整性
• 简化向量数据存储与查询流程，适用于AI应用场景

使用场景

• AI应用中的向量嵌入存储
• 相似性搜索与向量数据分析
• 机器学习模型输出向量的持久化存储

运行与测试步骤

1. 启动Docker容器

bash
docker run --name pgvector-test -e POSTGRES_PASSWORD=postgres -d azaddjan/pgvector:v0.8.0-postgres17

此命令启动一个预安装PgVector扩展的PostgreSQL容器。

2. 连接PostgreSQL实例

bash
docker exec -it pgvector-test psql -U postgres

通过psql命令行工具连接到PostgreSQL实例。

3. 验证PgVector扩展

执行以下SQL命令检查vector扩展是否可用：

sql
SELECT * FROM pg_available_extensions WHERE name = 'vector';

4. 创建含向量列的表

创建用于存储向量数据的表：

sql
CREATE TABLE items (id serial primary key, embedding vector(3));

5. 插入向量数据

向表中插入向量数据：

sql
INSERT INTO items (embedding) VALUES ('[1, 2, 3]');

6. 查询向量数据

从表中检索数据：

sql
SELECT * FROM items;

7. 退出PostgreSQL CLI

使用以下命令退出psql：

\q

8. 停止并删除容器

停止容器：

bash
docker stop pgvector-test

删除容器：

bash
docker rm pgvector-test

总结

本指南演示了在Docker容器中设置和测试PgVector扩展的完整流程，涵盖扩展验证、表创建、数据操作等核心步骤。使用完毕后可停止并删除容器以释放资源。