diff --git a/litellm/main.py b/litellm/main.py index 8863397bf0..49416fdecb 100644 --- a/litellm/main.py +++ b/litellm/main.py @@ -83,11 +83,11 @@ class LiteLLM: def __init__(self, *, api_key=None, - organization: str | None = None, - base_url: str | None = None, - timeout: Union[float, None] = 600, - max_retries: int | None = litellm.num_retries, - default_headers: Mapping[str, str] | None = None,): + organization: Optional[str] = None, + base_url: Optional[str]= None, + timeout: Optional[float] = 600, + max_retries: Optional[int] = litellm.num_retries, + default_headers: Optional[Mapping[str, str]] = None,): self.params = locals() self.chat = Chat(self.params)