Add langchain Google import

simonff · Dec 29, 2023 · 2fa0ec9 · 2fa0ec9
1 parent c5006f0
commit 2fa0ec9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py b/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py
@@ -1,4 +1,3 @@
-foo
 import asyncio
 import base64
 import copy
@@ -27,6 +26,7 @@
     QianfanChatEndpoint,
 )
 from langchain.chat_models.base import BaseChatModel
+from langchain_google_genai import ChatGoogleGenerativeAI
 from langchain.llms import (
     AI21,
     Anthropic,