fix: fix linting error

This commit is contained in:
Krrish Dholakia 2025-03-10 21:51:55 -07:00
parent 92881ee79e
commit 49792e8cc4

View file

@ -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"