| 1 | # SPDX-License-Identifier: Apache-2.0 |
| 2 | # SPDX-FileCopyrightText: Copyright contributors to the vLLM project |
| 3 | |
| 4 | from vllm.config.watermarking import derive_watermark_key |
| 5 | from vllm.v1.watermarking.detector import WatermarkDetection, WatermarkDetector |
| 6 | from vllm.v1.watermarking.factory import create_watermarker |
| 7 | from vllm.v1.watermarking.gumbel import ( |
| 8 | DualKeyGumbelWatermarkDetector, |
| 9 | DualKeyGumbelWatermarker, |
| 10 | GumbelWatermarkDetector, |
| 11 | GumbelWatermarker, |
| 12 | ) |
| 13 | from vllm.v1.watermarking.prfs import ( |
| 14 | PhiloxPRF, |
| 15 | WatermarkPRF, |
| 16 | create_prf, |
| 17 | ) |
| 18 | from vllm.v1.watermarking.watermarker import ( |
| 19 | SupportsSpeculativeDecoding, |
| 20 | Watermarker, |
| 21 | WatermarkSample, |
| 22 | ) |
| 23 | |
| 24 | __all__ = [ |
| 25 | "DualKeyGumbelWatermarkDetector", |
| 26 | "DualKeyGumbelWatermarker", |
| 27 | "GumbelWatermarkDetector", |
| 28 | "GumbelWatermarker", |
| 29 | "PhiloxPRF", |
| 30 | "SupportsSpeculativeDecoding", |
| 31 | "WatermarkDetection", |
| 32 | "WatermarkDetector", |
| 33 | "WatermarkPRF", |
| 34 | "Watermarker", |
| 35 | "WatermarkSample", |
| 36 | "create_watermarker", |
| 37 | "create_prf", |
| 38 | "derive_watermark_key", |
| 39 | ] |