diff --git a/agentlightning/tracer/base.py b/agentlightning/tracer/base.py index 0f4a6afa0..1aae774d1 100644 --- a/agentlightning/tracer/base.py +++ b/agentlightning/tracer/base.py @@ -11,7 +11,7 @@ from agentlightning.types import ParallelWorkerBase if TYPE_CHECKING: - from langchain.callbacks.base import BaseCallbackHandler # type: ignore + from langchain_core.callbacks.base import BaseCallbackHandler # type: ignore logger = logging.getLogger(__name__) diff --git a/pyproject.toml b/pyproject.toml index bfac5e1ad..b32541df2 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -55,10 +55,10 @@ agent = [ "autogen-ext[openai]", "mcp", "openai-agents", - "langgraph", - "langchain[openai]", + "langgraph<1.0", + "langchain[openai]<1.0", "langchain-community", - "langchain-text-splitters", + "langchain-text-splitters<1.0", "sqlparse", "nltk", "uv",