We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 352a105 commit b63a04eCopy full SHA for b63a04e
litellm/llms/infinity/rerank/transformation.py
@@ -20,13 +20,15 @@
20
21
22
class InfinityRerankConfig(CohereRerankConfig):
23
- def get_complete_url(self, api_base: str, model: str) -> str:
+ 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")
26
# Remove trailing slashes and ensure clean base URL
27
api_base = api_base.rstrip("/")
28
if not api_base.endswith("/rerank"):
29
api_base = f"{api_base}/rerank"
30
return api_base
-
31
+
32
def validate_environment(
33
self,
34
headers: dict,
0 commit comments