From 475193f30ba084b5b27477a6c85ff59644a4d313 Mon Sep 17 00:00:00 2001
From: Steven Tartakovsky <tartakovsky.developer@gmail.com>
Date: Sat, 24 Feb 2024 18:24:28 -0800
Subject: [PATCH] Import ChatOpenAI and AzureChatOpenAI from the updated
 langchain location:

---
 packages/jupyter-ai-magics/jupyter_ai_magics/providers.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py b/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py
index 2704e8f32..44b946bbb 100644
--- 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,