openai_compatible_server.md 17.6 KB
Newer Older
1
(openai-compatible-server)=
2

3
# OpenAI-Compatible Server
4

5
6
7
vLLM provides an HTTP server that implements OpenAI's [Completions API](https://platform.openai.com/docs/api-reference/completions), [Chat API](https://platform.openai.com/docs/api-reference/chat), and more!

You can start the server via the [`vllm serve`](#vllm-serve) command, or through [Docker](#deployment-docker):
8

9
```bash
10
vllm serve NousResearch/Meta-Llama-3-8B-Instruct --dtype auto --api-key token-abc123
11
12
```

13
To call the server, you can use the [official OpenAI Python client](https://github.com/openai/openai-python), or any other HTTP client.
14

15
16
17
18
19
20
21
22
```python
from openai import OpenAI
client = OpenAI(
    base_url="http://localhost:8000/v1",
    api_key="token-abc123",
)

completion = client.chat.completions.create(
23
  model="NousResearch/Meta-Llama-3-8B-Instruct",
24
25
26
27
28
29
30
31
  messages=[
    {"role": "user", "content": "Hello!"}
  ]
)

print(completion.choices[0].message)
```

32
## Supported APIs
33

34
35
We currently support the following OpenAI APIs:

36
- [Completions API](#completions-api) (`/v1/completions`)
37
  - Only applicable to [text generation models](../models/generative_models.md) (`--task generate`).
38
  - *Note: `suffix` parameter is not supported.*
39
- [Chat Completions API](#chat-api) (`/v1/chat/completions`)
40
  - Only applicable to [text generation models](../models/generative_models.md) (`--task generate`) with a [chat template](#chat-template).
41
  - *Note: `parallel_tool_calls` and `user` parameters are ignored.*
42
- [Embeddings API](#embeddings-api) (`/v1/embeddings`)
43
  - Only applicable to [embedding models](../models/pooling_models.md) (`--task embed`).
44
45
- [Transcriptions API](#transcriptions-api) (`/v1/audio/transcriptions`)
  - Only applicable to Automatic Speech Recognition (ASR) models (OpenAI Whisper) (`--task generate`).
46

47
In addition, we have the following custom APIs:
48

49
50
- [Tokenizer API](#tokenizer-api) (`/tokenize`, `/detokenize`)
  - Applicable to any model with a tokenizer.
51
52
- [Pooling API](#pooling-api) (`/pooling`)
  - Applicable to all [pooling models](../models/pooling_models.md).
53
- [Score API](#score-api) (`/score`)
54
  - Applicable to embedding models and [cross-encoder models](../models/pooling_models.md) (`--task score`).
55
56
57
58
59
- [Re-rank API](#rerank-api) (`/rerank`, `/v1/rerank`, `/v2/rerank`)
  - Implements [Jina AI's v1 re-rank API](https://jina.ai/reranker/)
  - Also compatible with [Cohere's v1 & v2 re-rank APIs](https://docs.cohere.com/v2/reference/rerank)
  - Jina and Cohere's APIs are very similar; Jina's includes extra information in the rerank endpoint's response.
  - Only applicable to [cross-encoder models](../models/pooling_models.md) (`--task score`).
60

61
(chat-template)=
62

63
## Chat Template
64

65
66
67
In order for the language model to support chat protocol, vLLM requires the model to include
a chat template in its tokenizer configuration. The chat template is a Jinja2 template that
specifies how are roles, messages, and other chat-specific tokens are encoded in the input.
68

69
An example chat template for `NousResearch/Meta-Llama-3-8B-Instruct` can be found [here](https://github.com/meta-llama/llama3?tab=readme-ov-file#instruction-tuned-models)
70

71
72
73
74
Some models do not provide a chat template even though they are instruction/chat fine-tuned. For those model,
you can manually specify their chat template in the `--chat-template` parameter with the file path to the chat
template, or the template in string form. Without a chat template, the server will not be able to process chat
and all chat requests will error.
75
76

```bash
77
vllm serve <model> --chat-template ./path-to-chat-template.jinja
78
79
```

80
vLLM community provides a set of chat templates for popular models. You can find them under the <gh-dir:examples> directory.
81

82
83
With the inclusion of multi-modal chat APIs, the OpenAI spec now accepts chat messages in a new format which specifies
both a `type` and a `text` field. An example is provided below:
84

85
86
87
88
89
```python
completion = client.chat.completions.create(
  model="NousResearch/Meta-Llama-3-8B-Instruct",
  messages=[
    {"role": "user", "content": [{"type": "text", "text": "Classify this sentiment: vLLM is wonderful!"}]}
90
  ]
91
)
92
93
```

94
Most chat templates for LLMs expect the `content` field to be a string, but there are some newer models like
95
96
97
98
`meta-llama/Llama-Guard-3-1B` that expect the content to be formatted according to the OpenAI schema in the
request. vLLM provides best-effort support to detect this automatically, which is logged as a string like
*"Detected the chat template content format to be..."*, and internally converts incoming requests to match
the detected format, which can be one of:
99

100
101
102
103
- `"string"`: A string.
  - Example: `"Hello world"`
- `"openai"`: A list of dictionaries, similar to OpenAI schema.
  - Example: `[{"type": "text", "text": "Hello world!"}]`
104

105
106
If the result is not what you expect, you can set the `--chat-template-content-format` CLI argument
to override which format to use.
107

108
## Extra Parameters
109

110
111
112
113
114
115
vLLM supports a set of parameters that are not part of the OpenAI API.
In order to use them, you can pass them as extra parameters in the OpenAI client.
Or directly merge them into the JSON payload if you are using HTTP call directly.

```python
completion = client.chat.completions.create(
116
  model="NousResearch/Meta-Llama-3-8B-Instruct",
117
118
119
120
121
122
123
124
125
  messages=[
    {"role": "user", "content": "Classify this sentiment: vLLM is wonderful!"}
  ],
  extra_body={
    "guided_choice": ["positive", "negative"]
  }
)
```

126
## Extra HTTP Headers
127

128
Only `X-Request-Id` HTTP request header is supported for now. It can be enabled
129
with `--enable-request-id-headers`.
130
131
132
133

> Note that enablement of the headers can impact performance significantly at high QPS
> rates. We recommend implementing HTTP headers at the router level (e.g. via Istio),
> rather than within the vLLM layer for this reason.
134
> See [this PR](https://github.com/vllm-project/vllm/pull/11529) for more details.
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157

```python
completion = client.chat.completions.create(
  model="NousResearch/Meta-Llama-3-8B-Instruct",
  messages=[
    {"role": "user", "content": "Classify this sentiment: vLLM is wonderful!"}
  ],
  extra_headers={
    "x-request-id": "sentiment-classification-00001",
  }
)
print(completion._request_id)

completion = client.completions.create(
  model="NousResearch/Meta-Llama-3-8B-Instruct",
  prompt="A robot may not injure a human being",
  extra_headers={
    "x-request-id": "completion-test",
  }
)
print(completion._request_id)
```

158
159
160
## CLI Reference

(vllm-serve)=
161

162
163
164
165
### `vllm serve`

The `vllm serve` command is used to launch the OpenAI-compatible server.

166
:::{argparse}
167
168
169
:module: vllm.entrypoints.openai.cli_args
:func: create_parser_for_docs
:prog: vllm serve
170
:::
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189

#### Configuration file

You can load CLI arguments via a [YAML](https://yaml.org/) config file.
The argument names must be the long form of those outlined [above](#vllm-serve).

For example:

```yaml
# config.yaml

host: "127.0.0.1"
port: 6379
uvicorn-log-level: "info"
```

To use the above config file:

```bash
190
vllm serve SOME_MODEL --config config.yaml
191
192
```

193
:::{note}
194
195
In case an argument is supplied simultaneously using command line and the config file, the value from the command line will take precedence.
The order of priorities is `command line > config file values > defaults`.
196
:::
197
198
199
200

## API Reference

(completions-api)=
201

202
203
### Completions API

204
205
206
Our Completions API is compatible with [OpenAI's Completions API](https://platform.openai.com/docs/api-reference/completions);
you can use the [official OpenAI Python client](https://github.com/openai/openai-python) to interact with it.

207
Code example: <gh-file:examples/online_serving/openai_completion_client.py>
208
209

#### Extra parameters
210

211
The following [sampling parameters](#sampling-params) are supported.
212

213
:::{literalinclude} ../../../vllm/entrypoints/openai/protocol.py
214
215
216
:language: python
:start-after: begin-completion-sampling-params
:end-before: end-completion-sampling-params
217
:::
218
219
220

The following extra parameters are supported:

221
:::{literalinclude} ../../../vllm/entrypoints/openai/protocol.py
222
223
224
:language: python
:start-after: begin-completion-extra-params
:end-before: end-completion-extra-params
225
:::
226

227
(chat-api)=
228

229
### Chat API
230

231
232
Our Chat API is compatible with [OpenAI's Chat Completions API](https://platform.openai.com/docs/api-reference/chat);
you can use the [official OpenAI Python client](https://github.com/openai/openai-python) to interact with it.
233

234
235
We support both [Vision](https://platform.openai.com/docs/guides/vision)- and
[Audio](https://platform.openai.com/docs/guides/audio?audio-generation-quickstart-example=audio-in)-related parameters;
236
see our [Multimodal Inputs](#multimodal-inputs) guide for more information.
237
238
- *Note: `image_url.detail` parameter is not supported.*

239
Code example: <gh-file:examples/online_serving/openai_chat_completion_client.py>
240

241
#### Extra parameters
242

243
The following [sampling parameters](#sampling-params) are supported.
244

245
:::{literalinclude} ../../../vllm/entrypoints/openai/protocol.py
246
247
248
:language: python
:start-after: begin-chat-completion-sampling-params
:end-before: end-chat-completion-sampling-params
249
:::
250
251
252

The following extra parameters are supported:

253
:::{literalinclude} ../../../vllm/entrypoints/openai/protocol.py
254
255
256
:language: python
:start-after: begin-chat-completion-extra-params
:end-before: end-chat-completion-extra-params
257
:::
258

259
(embeddings-api)=
260

261
262
### Embeddings API

263
264
Our Embeddings API is compatible with [OpenAI's Embeddings API](https://platform.openai.com/docs/api-reference/embeddings);
you can use the [official OpenAI Python client](https://github.com/openai/openai-python) to interact with it.
265

266
If the model has a [chat template](#chat-template), you can replace `inputs` with a list of `messages` (same schema as [Chat API](#chat-api))
267
268
which will be treated as a single prompt to the model.

269
:::{tip}
270
This enables multi-modal inputs to be passed to embedding models, see [this page](#multimodal-inputs) for details.
271
:::
272

273
Code example: <gh-file:examples/online_serving/openai_embedding_client.py>
274

275
#### Extra parameters
276

277
The following [pooling parameters](#pooling-params) are supported.
278

279
:::{literalinclude} ../../../vllm/entrypoints/openai/protocol.py
280
:language: python
281
282
:start-after: begin-embedding-pooling-params
:end-before: end-embedding-pooling-params
283
:::
284

285
The following extra parameters are supported by default:
286

287
:::{literalinclude} ../../../vllm/entrypoints/openai/protocol.py
288
:language: python
289
290
:start-after: begin-embedding-extra-params
:end-before: end-embedding-extra-params
291
:::
292

293
For chat-like input (i.e. if `messages` is passed), these extra parameters are supported instead:
294

295
:::{literalinclude} ../../../vllm/entrypoints/openai/protocol.py
296
297
298
:language: python
:start-after: begin-chat-embedding-extra-params
:end-before: end-chat-embedding-extra-params
299
:::
300

301
302
303
304
305
306
307
308
309
310
311
(transcriptions-api)=

### Transcriptions API

Our Transcriptions API is compatible with [OpenAI's Transcriptions API](https://platform.openai.com/docs/api-reference/audio/createTranscription);
you can use the [official OpenAI Python client](https://github.com/openai/openai-python) to interact with it.

<!-- TODO: api enforced limits + uploading audios -->

Code example: <gh-file:examples/online_serving/openai_transcription_client.py>

312
(tokenizer-api)=
313

314
### Tokenizer API
315

316
Our Tokenizer API is a simple wrapper over [HuggingFace-style tokenizers](https://huggingface.co/docs/transformers/en/main_classes/tokenizer).
317
318
319
320
321
It consists of two endpoints:

- `/tokenize` corresponds to calling `tokenizer.encode()`.
- `/detokenize` corresponds to calling `tokenizer.decode()`.

322
(pooling-api)=
323

324
325
326
327
328
329
### Pooling API

Our Pooling API encodes input prompts using a [pooling model](../models/pooling_models.md) and returns the corresponding hidden states.

The input format is the same as [Embeddings API](#embeddings-api), but the output data can contain an arbitrary nested list, not just a 1-D list of floats.

330
Code example: <gh-file:examples/online_serving/openai_pooling_client.py>
331

332
(score-api)=
333

334
335
### Score API

336
Our Score API can apply a cross-encoder model or an embedding model to predict scores for sentence pairs. When using an embedding model the score corresponds to the cosine similarity between each embedding pair.
337
338
Usually, the score for a sentence pair refers to the similarity between two sentences, on a scale of 0 to 1.

339
You can find the documentation for cross encoder models at [sbert.net](https://www.sbert.net/docs/package_reference/cross_encoder/cross_encoder.html).
340

341
Code example: <gh-file:examples/online_serving/openai_cross_encoder_score.py>
342

343
344
345
346
347
#### Single inference

You can pass a string to both `text_1` and `text_2`, forming a single sentence pair.

Request:
348
349

```bash
350
351
352
353
354
355
356
357
358
359
curl -X 'POST' \
  'http://127.0.0.1:8000/score' \
  -H 'accept: application/json' \
  -H 'Content-Type: application/json' \
  -d '{
  "model": "BAAI/bge-reranker-v2-m3",
  "encoding_format": "float",
  "text_1": "What is the capital of France?",
  "text_2": "The capital of France is Paris."
}'
360
361
```

362
Response:
363

364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
```bash
{
  "id": "score-request-id",
  "object": "list",
  "created": 693447,
  "model": "BAAI/bge-reranker-v2-m3",
  "data": [
    {
      "index": 0,
      "object": "score",
      "score": 1
    }
  ],
  "usage": {}
}
379
380
```

381
#### Batch inference
382

383
384
385
You can pass a string to `text_1` and a list to `text_2`, forming multiple sentence pairs
where each pair is built from `text_1` and a string in `text_2`.
The total number of pairs is `len(text_2)`.
386

387
Request:
388

389
390
391
392
393
394
395
396
397
398
399
400
401
402
```bash
curl -X 'POST' \
  'http://127.0.0.1:8000/score' \
  -H 'accept: application/json' \
  -H 'Content-Type: application/json' \
  -d '{
  "model": "BAAI/bge-reranker-v2-m3",
  "text_1": "What is the capital of France?",
  "text_2": [
    "The capital of Brazil is Brasilia.",
    "The capital of France is Paris."
  ]
}'
```
403

404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
Response:

```bash
{
  "id": "score-request-id",
  "object": "list",
  "created": 693570,
  "model": "BAAI/bge-reranker-v2-m3",
  "data": [
    {
      "index": 0,
      "object": "score",
      "score": 0.001094818115234375
    },
    {
      "index": 1,
      "object": "score",
      "score": 1
    }
  ],
  "usage": {}
}
426
```
427

428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
You can pass a list to both `text_1` and `text_2`, forming multiple sentence pairs
where each pair is built from a string in `text_1` and the corresponding string in `text_2` (similar to `zip()`).
The total number of pairs is `len(text_2)`.

Request:

```bash
curl -X 'POST' \
  'http://127.0.0.1:8000/score' \
  -H 'accept: application/json' \
  -H 'Content-Type: application/json' \
  -d '{
  "model": "BAAI/bge-reranker-v2-m3",
  "encoding_format": "float",
  "text_1": [
    "What is the capital of Brazil?",
    "What is the capital of France?"
  ],
  "text_2": [
    "The capital of Brazil is Brasilia.",
    "The capital of France is Paris."
  ]
}'
```
452

453
Response:
454

455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
```bash
{
  "id": "score-request-id",
  "object": "list",
  "created": 693447,
  "model": "BAAI/bge-reranker-v2-m3",
  "data": [
    {
      "index": 0,
      "object": "score",
      "score": 1
    },
    {
      "index": 1,
      "object": "score",
      "score": 1
    }
  ],
  "usage": {}
}
```
476

477
#### Extra parameters
478

479
The following [pooling parameters](#pooling-params) are supported.
480

481
:::{literalinclude} ../../../vllm/entrypoints/openai/protocol.py
482
483
484
:language: python
:start-after: begin-score-pooling-params
:end-before: end-score-pooling-params
485
:::
486

487
488
The following extra parameters are supported:

489
:::{literalinclude} ../../../vllm/entrypoints/openai/protocol.py
490
491
492
:language: python
:start-after: begin-score-extra-params
:end-before: end-score-extra-params
493
:::
494
495
496
497
498

(rerank-api)=

### Re-rank API

499
Our Re-rank API can apply an embedding model or a cross-encoder model to predict relevant scores between a single query, and
500
501
502
each of a list of documents. Usually, the score for a sentence pair refers to the similarity between two sentences, on
a scale of 0 to 1.

503
You can find the documentation for cross encoder models at [sbert.net](https://www.sbert.net/docs/package_reference/cross_encoder/cross_encoder.html).
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567

The rerank endpoints support popular re-rank models such as `BAAI/bge-reranker-base` and other models supporting the
`score` task. Additionally, `/rerank`, `/v1/rerank`, and `/v2/rerank`
endpoints are compatible with both [Jina AI's re-rank API interface](https://jina.ai/reranker/) and
[Cohere's re-rank API interface](https://docs.cohere.com/v2/reference/rerank) to ensure compatibility with
popular open-source tools.

Code example: <gh-file:examples/online_serving/jinaai_rerank_client.py>

#### Example Request

Note that the `top_n` request parameter is optional and will default to the length of the `documents` field.
Result documents will be sorted by relevance, and the `index` property can be used to determine original order.

Request:

```bash
curl -X 'POST' \
  'http://127.0.0.1:8000/v1/rerank' \
  -H 'accept: application/json' \
  -H 'Content-Type: application/json' \
  -d '{
  "model": "BAAI/bge-reranker-base",
  "query": "What is the capital of France?",
  "documents": [
    "The capital of Brazil is Brasilia.",
    "The capital of France is Paris.",
    "Horses and cows are both animals"
  ]
}'
```

Response:

```bash
{
  "id": "rerank-fae51b2b664d4ed38f5969b612edff77",
  "model": "BAAI/bge-reranker-base",
  "usage": {
    "total_tokens": 56
  },
  "results": [
    {
      "index": 1,
      "document": {
        "text": "The capital of France is Paris."
      },
      "relevance_score": 0.99853515625
    },
    {
      "index": 0,
      "document": {
        "text": "The capital of Brazil is Brasilia."
      },
      "relevance_score": 0.0005860328674316406
    }
  ]
}
```

#### Extra parameters

The following [pooling parameters](#pooling-params) are supported.

568
:::{literalinclude} ../../../vllm/entrypoints/openai/protocol.py
569
570
571
:language: python
:start-after: begin-rerank-pooling-params
:end-before: end-rerank-pooling-params
572
:::
573
574
575

The following extra parameters are supported:

576
:::{literalinclude} ../../../vllm/entrypoints/openai/protocol.py
577
578
579
:language: python
:start-after: begin-rerank-extra-params
:end-before: end-rerank-extra-params
580
:::