From 49792e8cc49923d5dabde57d2574320737350587 Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Mon, 10 Mar 2025 21:51:55 -0700
Subject: [PATCH] fix: fix linting error

---
 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 0a65e216dfe..4037c32365e 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"