phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-24 18:24:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

removed duplicate imports

Browse source
This commit is contained in:
Prathamesh 2025-04-16 17:02:02 +05:30
parent b803c9dca4
commit 6ad3d6e57c
2 changed files with 0 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

3
litellm/llms/voyage/rerank/transformation.py
View file

@ -10,7 +10,6 @@ from litellm.types.rerank import OptionalRerankParams, RerankRequest, RerankResp
from litellm.llms.voyage.common_utils import VoyageError
from litellm.types.rerank import (
RerankBilledUnits,
RerankResponse,
RerankResponseDocument,
RerankResponseMeta,
RerankResponseResult,
@ -53,7 +52,6 @@ class VoyageRerankConfig(BaseRerankConfig):
"documents",
"top_k",
"return_documents",
"truncation"
]
def map_cohere_rerank_params(
@ -100,7 +98,6 @@ class VoyageRerankConfig(BaseRerankConfig):
documents=optional_rerank_params["documents"],
# Voyage API uses top_k instead of top_n
top_k=optional_rerank_params.get("top_k", None),
truncation=optional_rerank_params.get("truncation", None),
return_documents=optional_rerank_params.get("return_documents", None),
)
return rerank_request.model_dump(exclude_none=True)

1
litellm/types/rerank.py
View file

@ -32,7 +32,6 @@ class OptionalRerankParams(TypedDict, total=False):
max_tokens_per_doc: Optional[int]
top_k: Optional[int]
class RerankBilledUnits(TypedDict, total=False):
search_units: Optional[int]
total_tokens: Optional[int]