diff --git a/pyproject.toml b/pyproject.toml
index 97e474c..070cdb8 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -4,10 +4,12 @@ version = "0.1.20"
 description = "CodeInterpreterAPI is an (unofficial) open source python interface for the ChatGPT CodeInterpreter."
 authors = [{ name = "Shroominic", email = "contact@shroominic.com" }]
 dependencies = [
-    "langchain>=0.1, <0.2",
+    "langchain>=0.1",
     "langchain_openai",
     "codeboxapi==0.1.19",
     "pyzmq==25.1.2",
+    "langchain-community>=0.0.20",
+    "langchain-core>=0.1",
 ]
 license = { file = "LICENSE" }
 readme = "README.md"
diff --git a/src/codeinterpreterapi/session.py b/src/codeinterpreterapi/session.py
index 3ec1f01..f0a8310 100644
--- a/src/codeinterpreterapi/session.py
+++ b/src/codeinterpreterapi/session.py
@@ -15,8 +15,8 @@
     ConversationalChatAgent,
 )
 from langchain.agents.openai_functions_agent.base import OpenAIFunctionsAgent
-from langchain.callbacks.base import Callbacks
-from langchain.chat_models.base import BaseChatModel
+from langchain_core.callbacks import Callbacks
+from langchain_core.language_models import BaseChatModel
 from langchain.memory.buffer import ConversationBufferMemory
 from langchain_community.chat_message_histories.in_memory import ChatMessageHistory
 from langchain_community.chat_message_histories.postgres import (