entrypoint
Register Stained Glass Output Protection with vLLM.
register_vllm_plugin is declared under the vllm.general_plugins entry point group, so vLLM calls it automatically in every vLLM process,
including the process that builds the OpenAI-compatible FastAPI application.
Launching Output Protection therefore needs no special command:
export HUGGING_FACE_HUB_TOKEN=<secret>
export SG_CLIENT_PUBLIC_KEY_HEADER_NAME=x-client-public-key
vllm serve meta-llama/Meta-Llama-3.1-8B-Instruct
The resulting Output Protected vLLM server is available at http://localhost:8000/, and exposes an OpenAI compatible API that accepts prompt
embeds. Any CLI arguments valid for vllm serve may be passed.
Prompt embeddings are enabled by default (see prompt_embeds), so
--enable-prompt-embeds need not be passed even on platforms that compose the container command themselves.
Deprecated
python -m stainedglass_output_protection.vllm.entrypoint still works but it is deprecated: it prepends serve to
its arguments and delegates to vLLM's own CLI. Launch with vllm serve instead.
Functions:
| Name | Description |
|---|---|
launch_vllm_with_output_protection |
Launch a vLLM OpenAI-compatible API server with Stained Glass Output Protection enabled. |
register_vllm_plugin |
Patch vLLM with everything Stained Glass Output Protection needs. Called by vLLM's plugin system in every vLLM process. |
launch_vllm_with_output_protection
¶
Launch a vLLM OpenAI-compatible API server with Stained Glass Output Protection enabled.
Deprecated in favour of vllm serve, which this now delegates to. Delegating rather than reimplementing keeps this path identical to
vllm serve.
Parameters:
| Name | Type | Description | Default |
|---|---|---|---|
|
list[str] | None
|
Arguments to serve with, defaulting to this process's own command-line arguments. |
None
|
register_vllm_plugin
¶
Patch vLLM with everything Stained Glass Output Protection needs. Called by vLLM's plugin system in every vLLM process.