Skip to content

Commit 84b0b57

Browse files
Update vllm/entrypoints/openai/protocol.py
Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
1 parent f5ba6ba commit 84b0b57

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
@@ -1229,15 +1229,15 @@ class RerankRequest(OpenAIBaseModel):
12291229
"default: 0). Any priority other than 0 will raise an error "
12301230
"if the served model does not use priority scheduling."),
12311231
)
1232-
rerank_template: Optional[dict[str, str]] = Field(
1232+
score_template: Optional[dict[str, str]] = Field(
12331233
default=None,
12341234
description=("A dictionary containing query_template and "
12351235
"document_template to format the reranker input.")
12361236
)
1237-
rerank_template_kwargs: Optional[dict[str, Any]] = Field(
1237+
score_template_kwargs: Optional[dict[str, Any]] = Field(
12381238
default=None,
1239-
description=("A dictionary of key-value pairs to be formatted into "
1240-
"the rerank model's template.")
1239+
description=("Additional keyword args to pass to the template renderer. "
1240+
"Will be accessible by the rerank template.")
12411241
)
12421242
# --8<-- [end:rerank-extra-params]
12431243

0 commit comments

Comments
 (0)