xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2026-05-27 04:27:53 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Bugfix] fix define of RerankDocument (#20877)

Browse Source 
Signed-off-by: liuchenlong <liuchenlong@xiaohongshu.com>
Co-authored-by: liuchenlong <liuchenlong@xiaohongshu.com>
This commit is contained in:
Liuchenlong 2025-07-13 22:32:40 +08:00 committed by GitHub
parent 247102f07f
commit 211b6a6113
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
vllm/entrypoints/openai/protocol.py
View File

@ -30,7 +30,8 @@ from typing_extensions import TypeAlias
from vllm import envs from vllm import envs
from vllm.entrypoints.chat_utils import (ChatCompletionMessageParam, from vllm.entrypoints.chat_utils import (ChatCompletionMessageParam,
random_tool_call_id) random_tool_call_id)
from vllm.entrypoints.score_utils import ScoreMultiModalParam from vllm.entrypoints.score_utils import (ScoreContentPartParam,
ScoreMultiModalParam)
from vllm.logger import init_logger from vllm.logger import init_logger
from vllm.pooling_params import PoolingParams from vllm.pooling_params import PoolingParams
from vllm.sampling_params import (BeamSearchParams, GuidedDecodingParams, from vllm.sampling_params import (BeamSearchParams, GuidedDecodingParams,
@ -1354,7 +1355,7 @@ class RerankRequest(OpenAIBaseModel):
class RerankDocument(BaseModel): class RerankDocument(BaseModel):
text: Optional[str] = None text: Optional[str] = None
multi_modal: Optional[ScoreMultiModalParam] = None multi_modal: Optional[ScoreContentPartParam] = None
class RerankResult(BaseModel): class RerankResult(BaseModel):