tensorzero/postgres

TensorZero快速启动Postgres镜像

镜像概述

本镜像为TensorZero项目快速启动设计，基于PostgreSQL数据库，预安装并配置了pgvector、pg_trgm和pg_cron扩展，旨在简化开发环境搭建，满足向量存储、文本相似性搜索及定时任务等需求。

核心功能与特性

• pgvector扩展：支持向量数据类型和高效的向量相似性搜索，适用于AI应用中的嵌入向量存储与检索
• pg_trgm扩展：提供基于trigram（三元语法）的文本相似性搜索功能，支持模糊匹配和文本相似度计算
• pg_cron扩展：实现数据库级别的定时任务调度，支持按时间周期执行SQL命令
• 与标准PostgreSQL镜像完全兼容，保留原生功能与配置方式

使用场景与适用范围

• TensorZero项目的快速启动与开发环境部署
• 需要向量存储和相似性搜索的AI应用（如推荐系统、语义搜索）
• 需文本模糊匹配功能的业务系统（如日志分析、内容检索）
• 需数据库定时任务的自动化场景（如数据清理、统计报表生成）

使用方法与配置说明

基本启动命令

bash
docker run -d \
  --name tensorzero-postgres \
  -e POSTGRES_PASSWORD=your_password \
  -e POSTGRES_DB=tensorzero \
  -p 5432:5432 \
  docker.xuanyuan.run/tensorzero/postgres-quickstart

Docker Compose配置示例

yaml
version: '3.8'
services:
  postgres:
    image: docker.xuanyuan.run/tensorzero/postgres-quickstart
    container_name: tensorzero-postgres
    environment:
      POSTGRES_PASSWORD: your_password
      POSTGRES_USER: tensorzero
      POSTGRES_DB: tensorzero_db
    ports:
      - "5432:5432"
    volumes:
      - postgres_data:/var/lib/postgresql/data
    restart: unless-stopped

volumes:
  postgres_data:

环境变量说明

验证扩展安装

容器启动后，可通过以下方式验证扩展是否正确安装：

1. 连接数据库：

bash
psql -h localhost -p 5432 -U postgres -d tensorzero

2. 执行查询检查扩展：

sql
SELECT extname, extversion FROM pg_extension WHERE extname IN ('vector', 'pg_trgm', 'pg_cron');

预期输出应包含三个扩展的名称及版本信息。

扩展使用示例

pgvector使用示例

sql
-- 创建含向量字段的表
CREATE TABLE documents (
  id SERIAL PRIMARY KEY,
  content TEXT,
  embedding vector(1536)  -- 1536维向量
);

-- 插入向量数据
INSERT INTO documents (content, embedding) 
VALUES ('示例文本', '[0.1, 0.2, ..., 0.9]');

-- 向量相似性搜索
SELECT content, embedding <-> '[0.1, 0.2, ..., 0.9]' AS distance 
FROM docker.xuanyuan.run/documents 
ORDER BY distance 
LIMIT 5;

pg_cron使用示例

sql
-- 创建定时任务（每天凌晨2点执行数据清理）
SELECT cron.schedule(
  'daily-cleanup',
  '0 2 * * *',
  'DELETE FROM logs WHERE created_at < NOW() - INTERVAL ''7 days'''
);

-- 查看所有定时任务
SELECT * FROM cron.job;