scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-05 02:41:56 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 49

style: black reformat after SGLang rerank fix

Browse source 
Signed-off-by: Tai An <antai12232931@outlook.com>
This commit is contained in:
Tai An 2026-05-28 06:23:36 -07:00
parent ffbcbf62e1
commit 8d2845ef4e
1 changed files with 1 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
litellm/llms/hosted_vllm/rerank/transformation.py
View file

@ -168,9 +168,7 @@ class HostedVLLMRerankConfig(BaseRerankConfig):
message=error_message, status_code=status_code, headers=headers message=error_message, status_code=status_code, headers=headers
) )
def _transform_response( def _transform_response(self, response: Union[dict, list]) -> RerankResponse:
self, response: Union[dict, list]
) -> RerankResponse:
# SGLang's /v1/rerank endpoint returns a bare list of result objects # SGLang's /v1/rerank endpoint returns a bare list of result objects
# instead of the OpenAI/Cohere-style {"results": [...], "usage": {...}} # instead of the OpenAI/Cohere-style {"results": [...], "usage": {...}}
# envelope. Normalize both shapes here so hosted_vllm covers both. # envelope. Normalize both shapes here so hosted_vllm covers both.