From 2928a1a3c9f974efaed060d5da309eb0bfd741fc Mon Sep 17 00:00:00 2001 From: Leonid Kuligin Date: Thu, 3 Aug 2023 22:28:42 +0200 Subject: [PATCH] added minimum expected version of SDK to the error description (#8712) #7932 Co-authored-by: Leonid Kuligin --- libs/langchain/langchain/chat_models/vertexai.py | 2 +- libs/langchain/langchain/utilities/vertexai.py | 8 +++++--- 2 files changed, 6 insertions(+), 4 deletions(-) diff --git a/libs/langchain/langchain/chat_models/vertexai.py b/libs/langchain/langchain/chat_models/vertexai.py index b50eac264..d9a08d7f0 100644 --- a/libs/langchain/langchain/chat_models/vertexai.py +++ b/libs/langchain/langchain/chat_models/vertexai.py @@ -111,7 +111,7 @@ class ChatVertexAI(_VertexAICommon, BaseChatModel): values["client"] = ChatModel.from_pretrained(values["model_name"]) except ImportError: - raise_vertex_import_error() + raise_vertex_import_error(minimum_expected_version="1.28.0") return values def _generate( diff --git a/libs/langchain/langchain/utilities/vertexai.py b/libs/langchain/langchain/utilities/vertexai.py index a934987cd..95c493585 100644 --- a/libs/langchain/langchain/utilities/vertexai.py +++ b/libs/langchain/langchain/utilities/vertexai.py @@ -5,15 +5,17 @@ if TYPE_CHECKING: from google.auth.credentials import Credentials -def raise_vertex_import_error() -> None: +def raise_vertex_import_error(minimum_expected_version: str = "1.26.1") -> None: """Raise ImportError related to Vertex SDK being not available. + Args: + minimum_expected_version: The lowest expected version of the SDK. Raises: ImportError: an ImportError that mentions a required version of the SDK. """ - sdk = "'google-cloud-aiplatform>=1.26.1'" raise ImportError( - "Could not import VertexAI. Please, install it with " f"pip install {sdk}" + "Could not import VertexAI. Please, install it with " + f"pip install google-cloud-aiplatform>={minimum_expected_version}" )