scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-06 02:48:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 128

style(cloudflare): apply ruff union and formatting fixes

Browse source
This commit is contained in:
prdai 2026-09-13 15:11:51 +05:30
parent 880e4a43ac
commit 000ec81170
2 changed files with 4 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
litellm/llms/cloudflare/embedding/transformation.py
View file

@ -1,5 +1,4 @@
from collections.abc import Mapping
from typing import Union
import httpx
@ -26,6 +25,6 @@ class CloudflareEmbeddingConfig(VercelAIGatewayEmbeddingConfig):
self,
error_message: str,
status_code: int,
headers: Union[Mapping[object, object], httpx.Headers],
headers: Mapping[object, object] | httpx.Headers,
) -> CloudflareError:
return CloudflareError(status_code=status_code, message=error_message)

12
litellm/llms/cloudflare/rerank/transformation.py
View file

@ -1,6 +1,5 @@
import json
from collections.abc import Mapping, Sequence
from typing import Union
import httpx
from typing_extensions import NotRequired, ReadOnly, TypedDict
@ -28,7 +27,7 @@ class CloudflareRerankRequest(TypedDict):
class CohereRerankParams(TypedDict):
query: ReadOnly[str]
documents: ReadOnly[Sequence[Union[str, Mapping[str, object]]]]
documents: ReadOnly[Sequence[str | Mapping[str, object]]]
top_n: NotRequired[ReadOnly[int]]
return_documents: NotRequired[ReadOnly[bool]]
@ -164,12 +163,7 @@ class CloudflareRerankConfig(BaseRerankConfig):
model: str,
drop_params: bool,
query: str,
documents: Sequence[
Union[
str,
Mapping[str, object],
]
],
documents: Sequence[str | Mapping[str, object]],
custom_llm_provider: str | None = None,
top_n: int | None = None,
rank_fields: Sequence[str] | None = None,
@ -280,6 +274,6 @@ class CloudflareRerankConfig(BaseRerankConfig):
self,
error_message: str,
status_code: int,
headers: Union[Mapping[str, object], httpx.Headers],
headers: Mapping[str, object] | httpx.Headers,
) -> BaseLLMException:
return CloudflareError(status_code=status_code, message=error_message)