diff --git a/litellm/llms/voyage/rerank/transformation.py b/litellm/llms/voyage/rerank/transformation.py index c95b8c9f15..6a22902e20 100644 --- a/litellm/llms/voyage/rerank/transformation.py +++ b/litellm/llms/voyage/rerank/transformation.py @@ -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) diff --git a/litellm/types/rerank.py b/litellm/types/rerank.py index 92fbbbcad4..4b6563101e 100644 --- a/litellm/types/rerank.py +++ b/litellm/types/rerank.py @@ -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]