Fixed "NameError: name 'GoogleGenerativeAIEmbeddings' is not defined"

arjuuuuunnnnn · web-flow · commit bd8afafce580 · 2024-05-09T09:36:06.000+05:30
diff --git a/scrapegraphai/graphs/abstract_graph.py b/scrapegraphai/graphs/abstract_graph.py
@@ -7,6 +7,7 @@
 from langchain_community.embeddings import HuggingFaceHubEmbeddings, OllamaEmbeddings
 from ..helpers import models_tokens
 from ..models import AzureOpenAI, Bedrock, Gemini, Groq, HuggingFace, Ollama, OpenAI, Anthropic
+from langchain_google_genai.embeddings import GoogleGenerativeAIEmbeddings
 
 
 class AbstractGraph(ABC):
