Unverified Commit 211b6a61 authored by Liuchenlong's avatar Liuchenlong Committed by GitHub
Browse files

[Bugfix] fix define of RerankDocument (#20877)


Signed-off-by: default avatarliuchenlong <liuchenlong@xiaohongshu.com>
Co-authored-by: default avatarliuchenlong <liuchenlong@xiaohongshu.com>
parent 247102f0
...@@ -30,7 +30,8 @@ from typing_extensions import TypeAlias ...@@ -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): ...@@ -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):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment