diff --git a/litellm/llms/anthropic/chat/handler.py b/litellm/llms/anthropic/chat/handler.py index 064f73c89d6..59751ae158a 100644 --- a/litellm/llms/anthropic/chat/handler.py +++ b/litellm/llms/anthropic/chat/handler.py @@ -782,7 +782,14 @@ class ModelResponseIterator: usage: Usage | None = None provider_specific_fields: Dict[str, Any] = {} reasoning_content: str | None = None - thinking_blocks: List[Union[ChatCompletionThinkingBlock, ChatCompletionRedactedThinkingBlock]] | None = None + thinking_blocks: ( + List[ + Union[ + ChatCompletionThinkingBlock, ChatCompletionRedactedThinkingBlock + ] + ] + | None + ) = None # Always use index=0 for OpenAI choice format (fixes multi-choice errors) index = 0 diff --git a/litellm/proxy/spend_tracking/cold_storage_handler.py b/litellm/proxy/spend_tracking/cold_storage_handler.py index 9737a4fdc44..aff4c29dc06 100644 --- a/litellm/proxy/spend_tracking/cold_storage_handler.py +++ b/litellm/proxy/spend_tracking/cold_storage_handler.py @@ -60,6 +60,8 @@ class ColdStorageHandler: def _select_custom_logger_for_cold_storage( self, ) -> _custom_logger_compatible_callbacks_literal | None: - cold_storage_custom_logger: _custom_logger_compatible_callbacks_literal | None = litellm.cold_storage_custom_logger + cold_storage_custom_logger: ( + _custom_logger_compatible_callbacks_literal | None + ) = litellm.cold_storage_custom_logger return cold_storage_custom_logger diff --git a/litellm/proxy/spend_tracking/spend_management_endpoints.py b/litellm/proxy/spend_tracking/spend_management_endpoints.py index 2f151d4103a..792d5b7d3c6 100644 --- a/litellm/proxy/spend_tracking/spend_management_endpoints.py +++ b/litellm/proxy/spend_tracking/spend_management_endpoints.py @@ -1789,9 +1789,7 @@ async def ui_view_spend_logs( status_filter: str | None = fastapi.Query( default=None, description="Filter logs by status (e.g., success, failure)" ), - model: str | None = fastapi.Query( - default=None, description="Filter logs by model" - ), + model: str | None = fastapi.Query(default=None, description="Filter logs by model"), model_id: str | None = fastapi.Query( default=None, description="Filter logs by model ID (litellm model deployment id)", diff --git a/litellm/rerank_api/main.py b/litellm/rerank_api/main.py index 23d19319e4d..d19f7c14615 100644 --- a/litellm/rerank_api/main.py +++ b/litellm/rerank_api/main.py @@ -30,7 +30,12 @@ async def arerank( model: str, query: str, documents: List[Union[str, Dict[str, Any]]], - custom_llm_provider: Literal["cohere", "together_ai", "deepinfra", "fireworks_ai", "voyage", "watsonx"] | None = None, + custom_llm_provider: ( + Literal[ + "cohere", "together_ai", "deepinfra", "fireworks_ai", "voyage", "watsonx" + ] + | None + ) = None, top_n: int | None = None, rank_fields: List[str] | None = None, return_documents: bool | None = None, @@ -75,7 +80,21 @@ def rerank( model: str, query: str, documents: List[Union[str, Dict[str, Any]]], - custom_llm_provider: Literal["cohere", "together_ai", "azure_ai", "infinity", "litellm_proxy", "hosted_vllm", "deepinfra", "fireworks_ai", "voyage", "watsonx"] | None = None, + custom_llm_provider: ( + Literal[ + "cohere", + "together_ai", + "azure_ai", + "infinity", + "litellm_proxy", + "hosted_vllm", + "deepinfra", + "fireworks_ai", + "voyage", + "watsonx", + ] + | None + ) = None, top_n: int | None = None, rank_fields: List[str] | None = None, return_documents: bool | None = True,