From 7b08a31c6a90490930ff2560da3d1ae241ad5c5d Mon Sep 17 00:00:00 2001 From: "Roy B.a" Date: Sat, 29 Nov 2025 00:43:57 +0200 Subject: [PATCH] fix: Update LangChain imports for v1.x compatibility Fixes #2607 - Replace deprecated `langchain.*` imports with `langchain_core.*` - Updates imports for Run, dumps, and BaseMessage - Resolves ModuleNotFoundError when using LangchainCallbackHandler with LangChain v1.x --- backend/chainlit/langchain/callbacks.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/backend/chainlit/langchain/callbacks.py b/backend/chainlit/langchain/callbacks.py index 0f811cc7ea..d1be77841b 100644 --- a/backend/chainlit/langchain/callbacks.py +++ b/backend/chainlit/langchain/callbacks.py @@ -3,11 +3,11 @@ 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 langchain_core.load import dumps +from langchain_core.messages import BaseMessage from langchain_core.outputs import ChatGenerationChunk, GenerationChunk from langchain_core.tracers.base import AsyncBaseTracer +from langchain_core.tracers.schemas import Run from literalai import ChatGeneration, CompletionGeneration, GenerationMessage from literalai.observability.step import TrueStepType