diff --git a/litellm/llms/vertex_httpx.py b/litellm/llms/vertex_httpx.py index 4fb554fe499..9fc9080b086 100644 --- a/litellm/llms/vertex_httpx.py +++ b/litellm/llms/vertex_httpx.py @@ -3,7 +3,7 @@ import json from enum import Enum import requests # type: ignore import time -from typing import Callable, Optional, Union, List, Any +from typing import Callable, Optional, Union, List, Any, Tuple from litellm.utils import ModelResponse, Usage, CustomStreamWrapper, map_finish_reason import litellm, uuid import httpx, inspect # type: ignore @@ -34,7 +34,7 @@ class VertexLLM(BaseLLM): self._credentials: Optional[Credentials] = None self.project_id: Optional[str] = None - def load_auth(self) -> tuple[Any, str]: + def load_auth(self) -> Tuple[Any, str]: from google.auth.transport.requests import Request # type: ignore[import-untyped] from google.auth.credentials import Credentials # type: ignore[import-untyped] import google.auth as google_auth