mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-02 02:11:58 +00:00
style: black reformat after SGLang rerank fix
Signed-off-by: Tai An <antai12232931@outlook.com>
This commit is contained in:
parent
ffbcbf62e1
commit
8d2845ef4e
1 changed files with 1 additions and 3 deletions
|
|
@ -168,9 +168,7 @@ class HostedVLLMRerankConfig(BaseRerankConfig):
|
|||
message=error_message, status_code=status_code, headers=headers
|
||||
)
|
||||
|
||||
def _transform_response(
|
||||
self, response: Union[dict, list]
|
||||
) -> RerankResponse:
|
||||
def _transform_response(self, response: Union[dict, list]) -> RerankResponse:
|
||||
# SGLang's /v1/rerank endpoint returns a bare list of result objects
|
||||
# instead of the OpenAI/Cohere-style {"results": [...], "usage": {...}}
|
||||
# envelope. Normalize both shapes here so hosted_vllm covers both.
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue