mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-15 23:31:29 +00:00
fix(transformation.py): fix linting error
This commit is contained in:
parent
c58941d49c
commit
f3be2e1fc9
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