jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

refactor: Update GoogleGenerativeAIComponent to handle missing 'langchain_google_genai' package

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-06-19 20:02:03 -03:00
commit 045c8123c9
1 changed files with 5 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

6
src/backend/base/langflow/components/models/GoogleGenerativeAIModel.py
View file

@ -1,4 +1,3 @@
from langchain_google_genai import ChatGoogleGenerativeAI
from pydantic.v1 import SecretStr from pydantic.v1 import SecretStr
from langflow.base.constants import STREAM_INFO_TEXT from langflow.base.constants import STREAM_INFO_TEXT
@ -84,6 +83,11 @@ class GoogleGenerativeAIComponent(LCModelComponent):
return result return result
def build_model(self) -> BaseLanguageModel: 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 google_api_key = self.google_api_key
model = self.model model = self.model
max_output_tokens = self.max_output_tokens max_output_tokens = self.max_output_tokens