phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix(handler.py): fix linting error

Browse source
This commit is contained in:
Krrish Dholakia 2024-11-23 02:08:50 +05:30
parent 541326731f
commit 31943bf2ad
1 changed files with 2 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/llms/azure_ai/rerank/handler.py
View file

@ -4,6 +4,7 @@ import httpx
from litellm.litellm_core_utils.litellm_logging import Logging as LiteLLMLoggingObj from litellm.litellm_core_utils.litellm_logging import Logging as LiteLLMLoggingObj
from litellm.llms.cohere.rerank import CohereRerank from litellm.llms.cohere.rerank import CohereRerank
from litellm.llms.custom_httpx.http_handler import AsyncHTTPHandler, HTTPHandler
from litellm.types.rerank import RerankResponse from litellm.types.rerank import RerankResponse
@ -73,6 +74,7 @@ class AzureAIRerank(CohereRerank):
return_documents: Optional[bool] = True, return_documents: Optional[bool] = True,
max_chunks_per_doc: Optional[int] = None, max_chunks_per_doc: Optional[int] = None,
_is_async: Optional[bool] = False, _is_async: Optional[bool] = False,
client: Optional[Union[HTTPHandler, AsyncHTTPHandler]] = None,
) -> RerankResponse: ) -> RerankResponse:
if headers is None: if headers is None: