Method Upsert

pkg/vector/pgvector/store.go:85–113

Upsert 将文档插入或更新到 pgvector 表。

(ctx context.Context, docs []vector.Document)

// Upsert 将文档插入或更新到 pgvector 表。
func (s *Store) Upsert(ctx context.Context, docs []vector.Document) error {
	if len(docs) == 0 {
		return nil
	}

	const tmpl = `
INSERT INTO %s (id, namespace, embedding, metadata)
VALUES ($1, $2, $3, COALESCE($4, '{}'::jsonb))
ON CONFLICT (id) DO UPDATE
SET namespace = EXCLUDED.namespace,
    embedding = EXCLUDED.embedding,
    metadata  = EXCLUDED.metadata;
`
	query := fmt.Sprintf(tmpl, s.table)

	for _, d := range docs {
		if d.ID == "" || len(d.Embedding) == 0 {
			continue
		}
		if len(d.Embedding) != s.dim {
			return fmt.Errorf("embedding dimension mismatch for id=%s: got %d, want %d", d.ID, len(d.Embedding), s.dim)
		}
		_, err := s.pool.Exec(ctx, query, d.ID, d.Namespace, float32SliceToPgVector(d.Embedding), d.Metadata)
		if err != nil {
			return fmt.Errorf("upsert vector %s: %w", d.ID, err)
		}
	}
	return nil
}

// Delete 从 pgvector 表中删除指定文档。
func (s *Store) Delete(ctx context.Context, ids []string) error {