Skip to content

Commit 744db15

Browse files
authored
Change the default value of truncate_prompt_tokens in the embedding model to -1,By default, the model is truncated according to its maximum length.
1 parent 51d5e9b commit 744db15

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

vllm/entrypoints/openai/protocol.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1298,7 +1298,7 @@ class EmbeddingCompletionRequest(OpenAIBaseModel):
12981298
encoding_format: Literal["float", "base64"] = "float"
12991299
dimensions: Optional[int] = None
13001300
user: Optional[str] = None
1301-
truncate_prompt_tokens: Optional[Annotated[int, Field(ge=-1)]] = None
1301+
truncate_prompt_tokens: Optional[Annotated[int, Field(ge=-1)]] = -1
13021302

13031303
# --8<-- [start:embedding-extra-params]
13041304
add_special_tokens: bool = Field(
@@ -1339,7 +1339,7 @@ class EmbeddingChatRequest(OpenAIBaseModel):
13391339
encoding_format: Literal["float", "base64"] = "float"
13401340
dimensions: Optional[int] = None
13411341
user: Optional[str] = None
1342-
truncate_prompt_tokens: Optional[Annotated[int, Field(ge=-1)]] = None
1342+
truncate_prompt_tokens: Optional[Annotated[int, Field(ge=-1)]] = -1
13431343

13441344
# --8<-- [start:chat-embedding-extra-params]
13451345
add_generation_prompt: bool = Field(
@@ -1461,7 +1461,7 @@ class ScoreRequest(OpenAIBaseModel):
14611461
model: Optional[str] = None
14621462
text_1: Union[list[str], str, ScoreMultiModalParam]
14631463
text_2: Union[list[str], str, ScoreMultiModalParam]
1464-
truncate_prompt_tokens: Optional[Annotated[int, Field(ge=-1)]] = None
1464+
truncate_prompt_tokens: Optional[Annotated[int, Field(ge=-1)]] = -1
14651465

14661466
# --8<-- [start:score-extra-params]
14671467

@@ -1493,7 +1493,7 @@ class RerankRequest(OpenAIBaseModel):
14931493
query: Union[str, ScoreMultiModalParam]
14941494
documents: Union[list[str], ScoreMultiModalParam]
14951495
top_n: int = Field(default_factory=lambda: 0)
1496-
truncate_prompt_tokens: Optional[Annotated[int, Field(ge=-1)]] = None
1496+
truncate_prompt_tokens: Optional[Annotated[int, Field(ge=-1)]] = -1
14971497

14981498
# --8<-- [start:rerank-extra-params]
14991499

0 commit comments

Comments
 (0)