diff --git a/knowledgehub/embeddings/langchain_based.py b/knowledgehub/embeddings/langchain_based.py
index 0aef4f1..9bd0e7b 100644
--- a/knowledgehub/embeddings/langchain_based.py
+++ b/knowledgehub/embeddings/langchain_based.py
@@ -122,7 +122,7 @@ class OpenAIEmbeddings(LCEmbeddingMixin, BaseEmbeddings):
 
     def _get_lc_class(self):
         try:
-            from langchain_community.embeddings import OpenAIEmbeddings
+            from langchain_openai import OpenAIEmbeddings
         except ImportError:
             from langchain.embeddings import OpenAIEmbeddings
 
@@ -152,7 +152,7 @@ class AzureOpenAIEmbeddings(LCEmbeddingMixin, BaseEmbeddings):
 
     def _get_lc_class(self):
         try:
-            from langchain_community.embeddings import AzureOpenAIEmbeddings
+            from langchain_openai import AzureOpenAIEmbeddings
         except ImportError:
             from langchain.embeddings import AzureOpenAIEmbeddings
 
diff --git a/knowledgehub/llms/chats/langchain_based.py b/knowledgehub/llms/chats/langchain_based.py
index 9a815ad..c5c2469 100644
--- a/knowledgehub/llms/chats/langchain_based.py
+++ b/knowledgehub/llms/chats/langchain_based.py
@@ -163,7 +163,7 @@ class AzureChatOpenAI(LCChatMixin, ChatLLM):
 
     def _get_lc_class(self):
         try:
-            from langchain_community.chat_models import AzureChatOpenAI
+            from langchain_openai import AzureChatOpenAI
         except ImportError:
             from langchain.chat_models import AzureChatOpenAI
 
diff --git a/knowledgehub/llms/completions/langchain_based.py b/knowledgehub/llms/completions/langchain_based.py
index 65122dd..8ad5a59 100644
--- a/knowledgehub/llms/completions/langchain_based.py
+++ b/knowledgehub/llms/completions/langchain_based.py
@@ -187,7 +187,7 @@ class AzureOpenAI(LCCompletionMixin, LLM):
 
     def _get_lc_class(self):
         try:
-            from langchain_community.llms import AzureOpenAI
+            from langchain_openai import AzureOpenAI
         except ImportError:
             from langchain.llms import AzureOpenAI
 
diff --git a/pyproject.toml b/pyproject.toml
index fdc9e33..1a05069 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -50,6 +50,7 @@ dev = [
     "sphinx",
     "coverage",
     "openai",
+    "langchain-openai",
     "chromadb",
     "wikipedia",
     "duckduckgo-search",