diff --git a/backend/chainlit/langchain/callbacks.py b/backend/chainlit/langchain/callbacks.py
index f841c0c56d..a666cd8dc8 100644
--- a/backend/chainlit/langchain/callbacks.py
+++ b/backend/chainlit/langchain/callbacks.py
@@ -3,9 +3,9 @@
 from uuid import UUID
 
 import pydantic
-from langchain.callbacks.tracers.schemas import Run
-from langchain.load.dump import dumps
-from langchain.schema import BaseMessage
+from langsmith.schemas import Run
+from langchain_core.load.dump import dumps 
+from langchain_core.messages import BaseMessage
 from langchain_core.outputs import ChatGenerationChunk, GenerationChunk
 from langchain_core.tracers.base import AsyncBaseTracer
 from literalai import ChatGeneration, CompletionGeneration, GenerationMessage