From 24b8dcff1fe6c795890fa8bc21b74ea920ed2469 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Mon, 10 Mar 2025 20:45:58 -0700 Subject: [PATCH] get_complete_url --- litellm/llms/triton/completion/transformation.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/litellm/llms/triton/completion/transformation.py b/litellm/llms/triton/completion/transformation.py index 0a65e216df..4037c32365 100644 --- a/litellm/llms/triton/completion/transformation.py +++ b/litellm/llms/triton/completion/transformation.py @@ -69,11 +69,13 @@ class TritonConfig(BaseConfig): def get_complete_url( self, - api_base: str, + api_base: Optional[str], model: str, optional_params: dict, stream: Optional[bool] = None, ) -> str: + if api_base is None: + raise ValueError("api_base is required") llm_type = self._get_triton_llm_type(api_base) if llm_type == "generate" and stream: return api_base + "_stream"