mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-09 03:18:44 +00:00
fix: run black formatting on UP045-fixed files
Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
parent
5a8b0487a8
commit
ec91088775
4 changed files with 33 additions and 7 deletions
|
|
@ -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
|
||||
|
|
|
|||
|
|
@ -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
|
||||
|
|
|
|||
|
|
@ -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)",
|
||||
|
|
|
|||
|
|
@ -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,
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue