From 045c8123c972471de7aed3efd7a4ed198a0045d4 Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@langflow.org>
Date: Wed, 19 Jun 2024 20:02:03 -0300
Subject: [PATCH] refactor: Update GoogleGenerativeAIComponent to handle
 missing 'langchain_google_genai' package

---
 .../langflow/components/models/GoogleGenerativeAIModel.py   | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/backend/base/langflow/components/models/GoogleGenerativeAIModel.py b/src/backend/base/langflow/components/models/GoogleGenerativeAIModel.py
index e810dafc9..734dd48a0 100644
--- a/src/backend/base/langflow/components/models/GoogleGenerativeAIModel.py
+++ b/src/backend/base/langflow/components/models/GoogleGenerativeAIModel.py
@@ -1,4 +1,3 @@
-from langchain_google_genai import ChatGoogleGenerativeAI
 from pydantic.v1 import SecretStr
 
 from langflow.base.constants import STREAM_INFO_TEXT
@@ -84,6 +83,11 @@ class GoogleGenerativeAIComponent(LCModelComponent):
         return result
 
     def build_model(self) -> BaseLanguageModel:
+        try:
+            from langchain_google_genai import ChatGoogleGenerativeAI
+        except ImportError:
+            raise ImportError("The 'langchain_google_genai' package is required to use the Google Generative AI model.")
+
         google_api_key = self.google_api_key
         model = self.model
         max_output_tokens = self.max_output_tokens