mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-11 22:51:28 +00:00
fix: fix linting error
This commit is contained in:
parent
92881ee79e
commit
49792e8cc4
1 changed files with 3 additions and 1 deletions
|
|
@ -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"
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue