fix: run black formatting on UP045-fixed files

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
Sameer Kankute 2026-06-24 17:26:57 +05:30
parent 5a8b0487a8
commit ec91088775
No known key found for this signature in database
4 changed files with 33 additions and 7 deletions

View file

@ -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

View file

@ -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

View file

@ -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)",

View file

@ -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,