Skip to content

Commit b63a04e

Browse files
fix(infinity/transformation.py): fix linting error
1 parent 352a105 commit b63a04e

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

Diff for: litellm/llms/infinity/rerank/transformation.py

+4-2
Original file line numberDiff line numberDiff line change
@@ -20,13 +20,15 @@
2020

2121

2222
class InfinityRerankConfig(CohereRerankConfig):
23-
def get_complete_url(self, api_base: str, model: str) -> str:
23+
def get_complete_url(self, api_base: Optional[str], model: str) -> str:
24+
if api_base is None:
25+
raise ValueError("api_base is required for Infinity rerank")
2426
# Remove trailing slashes and ensure clean base URL
2527
api_base = api_base.rstrip("/")
2628
if not api_base.endswith("/rerank"):
2729
api_base = f"{api_base}/rerank"
2830
return api_base
29-
31+
3032
def validate_environment(
3133
self,
3234
headers: dict,

0 commit comments

Comments
 (0)