scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-06 08:16:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 79

style: apply black formatting

Browse source 
Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
This commit is contained in:
Oguz Vuruskaner 2026-03-31 12:29:34 +03:00
parent 18f75241e1
commit 891718d064
2 changed files with 4 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
litellm/llms/deepinfra/rerank/transformation.py
View file

@ -108,7 +108,9 @@ class DeepinfraRerankConfig(BaseRerankConfig):
# Start with the basic parameters
optional_rerank_params = {}
if query:
optional_rerank_params["queries"] = [query] # Single-element array; Deepinfra broadcasts it across all documents
optional_rerank_params["queries"] = [
query
] # Single-element array; Deepinfra broadcasts it across all documents
if non_default_params is not None:
for k, v in non_default_params.items():

5
tests/test_litellm/llms/deepinfra/test_deepinfra_rerank_transformation.py
View file

@ -42,7 +42,6 @@ class TestDeepinfraRerankTransform:
with pytest.raises(ValueError, match="Deepinfra API Base is required"):
self.config.get_complete_url(None, model)
def test_map_cohere_rerank_params_basic(self):
"""Test basic parameter mapping for DeepInfra rerank."""
params = self.config.map_cohere_rerank_params(
@ -238,9 +237,7 @@ class TestDeepinfraRerankTransform:
query="query1",
documents=documents,
)
assert params["queries"] == [
"query1"
], f"Failed for {num_docs} documents"
assert params["queries"] == ["query1"], f"Failed for {num_docs} documents"
def test_get_error_class_basic(self):
"""Test error class generation for basic error."""