Skip to content

Commit 211b6a6

Browse files
Liuchenlongliuchenlong
andauthored
[Bugfix] fix define of RerankDocument (#20877)
Signed-off-by: liuchenlong <liuchenlong@xiaohongshu.com> Co-authored-by: liuchenlong <liuchenlong@xiaohongshu.com>
1 parent 247102f commit 211b6a6

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

vllm/entrypoints/openai/protocol.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,8 @@
3030
from vllm import envs
3131
from vllm.entrypoints.chat_utils import (ChatCompletionMessageParam,
3232
random_tool_call_id)
33-
from vllm.entrypoints.score_utils import ScoreMultiModalParam
33+
from vllm.entrypoints.score_utils import (ScoreContentPartParam,
34+
ScoreMultiModalParam)
3435
from vllm.logger import init_logger
3536
from vllm.pooling_params import PoolingParams
3637
from vllm.sampling_params import (BeamSearchParams, GuidedDecodingParams,
@@ -1354,7 +1355,7 @@ def to_pooling_params(self, *, use_cross_encoder: bool = False):
13541355

13551356
class RerankDocument(BaseModel):
13561357
text: Optional[str] = None
1357-
multi_modal: Optional[ScoreMultiModalParam] = None
1358+
multi_modal: Optional[ScoreContentPartParam] = None
13581359

13591360

13601361
class RerankResult(BaseModel):

0 commit comments

Comments
 (0)