mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-06 02:48:13 +00:00
refactor: clean up fresh tech debt from 2026-10-02 (#44362)
Co-authored-by: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
parent
e92bd50de1
commit
d260765652
5 changed files with 1 additions and 8 deletions
|
|
@ -54,7 +54,6 @@ from litellm.proxy.litellm_pre_call_utils import LiteLLMProxyRequestSetup
|
|||
from litellm.proxy.openai_files_endpoints.common_utils import (
|
||||
BATCH_CREATE_HIDDEN_PARAM,
|
||||
FILE_LIST_CONTINUATION_CHUNK_SIZE,
|
||||
ManagedFileIdResolver,
|
||||
_is_base64_encoded_unified_file_id,
|
||||
apply_unified_file_ids,
|
||||
decode_model_from_file_id,
|
||||
|
|
|
|||
|
|
@ -2490,7 +2490,7 @@ class Logging(LiteLLMLoggingBaseClass):
|
|||
response: ModelResponse | TextCompletionResponse | ResponsesAPIResponse | InteractionsAPIResponse,
|
||||
) -> ModelResponse | TextCompletionResponse | ResponsesAPIResponse | InteractionsAPIResponse:
|
||||
model: Final = self.client_facing_stream_model
|
||||
if model is None or getattr(response, "model", None) in (None, model):
|
||||
if model is None or response.model in (None, model):
|
||||
return response
|
||||
return response.model_copy(update={"model": model})
|
||||
|
||||
|
|
|
|||
|
|
@ -46,6 +46,5 @@ class ScalewayRerankConfig(JinaAIRerankConfig):
|
|||
"content-type": "application/json",
|
||||
"authorization": f"Bearer {key}",
|
||||
}
|
||||
# Header names are case-insensitive, so match on the lowercase name.
|
||||
caller_headers: Final = {name: value for name, value in headers.items() if name.lower() not in provider_headers}
|
||||
return {**caller_headers, **provider_headers}
|
||||
|
|
|
|||
|
|
@ -84,10 +84,6 @@ class _SettlementTableActions(Protocol):
|
|||
def update_many(self, *, data: _Claim | _Outcome, where: _RowKey | _UnclaimedRowKey) -> Awaitable[int]: ...
|
||||
|
||||
|
||||
def _settlement_table(prisma_client: "PrismaClient") -> _SettlementTableActions:
|
||||
return BackgroundInteractionSettlementRepository(prisma_client).table
|
||||
|
||||
|
||||
_CLEARED_CREATE_CONTEXT: Final[Mapping[str, object]] = MappingProxyType({})
|
||||
_T = TypeVar("_T")
|
||||
|
||||
|
|
|
|||
|
|
@ -8862,7 +8862,6 @@ class ProviderConfigManager:
|
|||
elif litellm.LlmProviders.INFINITY == provider:
|
||||
return litellm.InfinityRerankConfig()
|
||||
elif provider in (litellm.LlmProviders.JINA_AI, litellm.LlmProviders.SCALEWAY):
|
||||
# Scaleway's rerank API matches Jina's, so its config extends Jina's.
|
||||
return (
|
||||
litellm.ScalewayRerankConfig()
|
||||
if provider == litellm.LlmProviders.SCALEWAY
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue