Import ChatOpenAI and AzureChatOpenAI from the updated langchain loca…

…tion:
jupyterlab · Feb 25, 2024 · 475193f · 475193f
1 parent 642ac53
commit 475193f
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py b/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py
@@ -22,12 +22,14 @@
 from langchain.schema import LLMResult
 from langchain.utils import get_from_dict_or_env
 from langchain_community.chat_models import (
-    AzureChatOpenAI,
     BedrockChat,
     ChatAnthropic,
-    ChatOpenAI,
     QianfanChatEndpoint,
 )
+from langchain_openai.chat_models import (
+    AzureChatOpenAI,
+    ChatOpenAI,
+)
 from langchain_community.llms import (
     AI21,
     Anthropic,