phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 18:54:30 +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.llms.voyage.common_utils import VoyageError
from litellm.types.rerank import ( from litellm.types.rerank import (
RerankBilledUnits, RerankBilledUnits,
RerankResponse,
RerankResponseDocument, RerankResponseDocument,
RerankResponseMeta, RerankResponseMeta,
RerankResponseResult, RerankResponseResult,
@ -53,7 +52,6 @@ class VoyageRerankConfig(BaseRerankConfig):
"documents", "documents",
"top_k", "top_k",
"return_documents", "return_documents",
"truncation"
] ]
def map_cohere_rerank_params( def map_cohere_rerank_params(
@ -100,7 +98,6 @@ class VoyageRerankConfig(BaseRerankConfig):
documents=optional_rerank_params["documents"], documents=optional_rerank_params["documents"],
# Voyage API uses top_k instead of top_n # Voyage API uses top_k instead of top_n
top_k=optional_rerank_params.get("top_k", None), 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_documents=optional_rerank_params.get("return_documents", None),
) )
return rerank_request.model_dump(exclude_none=True) 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] max_tokens_per_doc: Optional[int]
top_k: Optional[int] top_k: Optional[int]
class RerankBilledUnits(TypedDict, total=False): class RerankBilledUnits(TypedDict, total=False):
search_units: Optional[int] search_units: Optional[int]
total_tokens: Optional[int] total_tokens: Optional[int]