langwatch 0.6.0__tar.gz → 0.7.0__tar.gz

{langwatch-0.6.0 → langwatch-0.7.0}/CHANGELOG.md

@@ -1,5 +1,24 @@
 # Changelog
 
+## [0.7.0](https://github.com/langwatch/langwatch/compare/python-sdk@v0.6.1...python-sdk@v0.7.0) (2025-11-05)
+
+
+### Features
+
+* support langchain/langgraph 1.0.0 in python sdk ([#780](https://github.com/langwatch/langwatch/issues/780)) ([70d4814](https://github.com/langwatch/langwatch/commit/70d4814528465d8e58d1ab4f82849ea13d6f564d))
+
+
+### Bug Fixes
+
+* parsing of Long values comming from Strands, add support for gen_ai.server.time_to_first_token ([c060766](https://github.com/langwatch/langwatch/commit/c060766fa093a6c6a881244267c1ed9929c9246b))
+
+## [0.6.1](https://github.com/langwatch/langwatch/compare/python-sdk@v0.6.0...python-sdk@v0.6.1) (2025-10-31)
+
+
+### Bug Fixes
+
+* dataset slug name sync ([#759](https://github.com/langwatch/langwatch/issues/759)) ([d9f87e4](https://github.com/langwatch/langwatch/commit/d9f87e4dc4df610e876f931094d3e86f2c5254d1))
+
 ## [0.6.0](https://github.com/langwatch/langwatch/compare/python-sdk@v0.5.1...python-sdk@v0.6.0) (2025-10-31)
 
 

{langwatch-0.6.0 → langwatch-0.7.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langwatch
-Version: 0.6.0
+Version: 0.7.0
 Summary: LangWatch Python SDK, for monitoring your LLMs
 Author-email: Langwatch Engineers <engineering@langwatch.ai>
 License: MIT
@@ -19,6 +19,7 @@ Requires-Dist: attrs>=24
 Requires-Dist: coolname>=2.2.0
 Requires-Dist: deprecated>=1.2.18
 Requires-Dist: httpx>=0.27.0
+Requires-Dist: langchain-core<2.0.0,>=1.0.0
 Requires-Dist: nanoid<3.0.0,>=2.0.0
 Requires-Dist: openinference-instrumentation-haystack>=0.1.20
 Requires-Dist: openinference-instrumentation-langchain>=0.1.24

{langwatch-0.6.0 → langwatch-0.7.0}/examples/dataset_kitchen_sink.py

@@ -27,7 +27,7 @@ def main(messages=None):
     langwatch.setup()
 
     # Example dataset slug or ID (replace with your actual dataset)
-    dataset_slug_or_id = os.getenv("LANGWATCH_DATASET_ID") or "test-dataset-35"
+    dataset_slug_or_id = os.getenv("LANGWATCH_DATASET_ID") or "test-dataset-1234"
     print(f"Dataset slug or ID: {dataset_slug_or_id}")
     print(
         f"LangWatch endpoint: {os.getenv('LANGWATCH_ENDPOINT') or langwatch._endpoint}"

{langwatch-0.6.0 → langwatch-0.7.0}/examples/langchain_bot.py

@@ -4,10 +4,9 @@ from langchain_openai import ChatOpenAI
 load_dotenv()
 
 import chainlit as cl
-from langchain.prompts import ChatPromptTemplate
-from langchain.schema import StrOutputParser
-from langchain.schema.runnable import Runnable
-from langchain.schema.runnable.config import RunnableConfig
+from langchain_core.prompts import ChatPromptTemplate
+from langchain_core.output_parsers import StrOutputParser
+from langchain_core.runnables import Runnable, RunnableConfig
 
 import langwatch
 
@@ -43,7 +42,6 @@ async def main(message: cl.Message):
         {"question": message.content},
         config=RunnableConfig(
             callbacks=[
-                cl.LangchainCallbackHandler(),
                 langwatch.get_current_trace().get_langchain_callback(),
             ]
         ),

{langwatch-0.6.0 → langwatch-0.7.0}/examples/langchain_bot_with_memory.py

@@ -4,29 +4,23 @@ from langchain_openai import ChatOpenAI
 load_dotenv()
 
 import chainlit as cl
-from langchain.prompts import ChatPromptTemplate, MessagesPlaceholder
-from langchain.schema.runnable import Runnable, RunnableMap
-from langchain.schema.runnable.config import RunnableConfig
-from langchain.memory import ConversationBufferMemory
+from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder
+from langchain_core.runnables import Runnable, RunnableConfig
+from langchain_core.messages import HumanMessage, AIMessage
 
 import langwatch
 
 
-session_memories: dict = {}
+# Store message history per session
+session_message_histories: dict = {}
 
 
 @cl.on_chat_start
 async def on_chat_start():
-    memory = session_memories.setdefault(
-        cl.user_session.get("session_id"),
-        ConversationBufferMemory(return_messages=True, memory_key="chat_history"),
-    )
-    ingress = RunnableMap(
-        {
-            "input": lambda x: x["input"],
-            "chat_history": lambda x: memory.load_memory_variables(x)["chat_history"],
-        }
-    )
+    # Initialize message history for this session
+    session_id = cl.user_session.get("session_id", "default")
+    message_history = session_message_histories.setdefault(session_id, [])
+    
     prompt = ChatPromptTemplate.from_messages(
         [
             ("system", "You are a helpful assistant."),
@@ -35,17 +29,20 @@ async def on_chat_start():
         ]
     )
     llm = ChatOpenAI(temperature=1, model="gpt-5", max_tokens=4096)
-    runnable = ingress | prompt | llm
+    runnable = prompt | llm
 
     cl.user_session.set("runnable", runnable)
-    cl.user_session.set("memory", memory)
+    cl.user_session.set("session_id", session_id)
 
 
 @cl.on_message
 @langwatch.trace()
 async def main(message: cl.Message):
     runnable: Runnable = cl.user_session.get("runnable")  # type: ignore
-    memory = cl.user_session.get("memory")  # type: ignore
+    session_id = cl.user_session.get("session_id", "default")
+    
+    # Get message history for this session
+    message_history = session_message_histories.get(session_id, [])
 
     msg = cl.Message(content="")
 
@@ -54,16 +51,17 @@ async def main(message: cl.Message):
     )
 
     async for chunk in runnable.astream(
-        {"input": message.content},
+        {"input": message.content, "chat_history": message_history},
         config=RunnableConfig(
             callbacks=[
-                cl.LangchainCallbackHandler(),
                 langwatch.get_current_trace().get_langchain_callback(),
             ]
         ),
     ):
         await msg.stream_token(chunk.content)
 
-    memory.save_context({"input": message.content}, {"output": msg.content})  # type: ignore
+    # Save the conversation to message history
+    message_history.append(HumanMessage(content=message.content))
+    message_history.append(AIMessage(content=msg.content))
 
     await msg.send()

{langwatch-0.6.0 → langwatch-0.7.0}/examples/langchain_rag_bot.py

@@ -5,9 +5,9 @@ from langwatch.types import RAGChunk
 load_dotenv()
 
 import chainlit as cl
-from langchain.prompts import ChatPromptTemplate
-from langchain.schema import HumanMessage
-from langchain.schema.runnable.config import RunnableConfig
+from langchain_core.messages import HumanMessage
+from langchain_core.runnables import RunnableConfig
+from langchain.agents import create_agent
 
 import langwatch
 
@@ -15,9 +15,7 @@ from langchain_community.document_loaders import WebBaseLoader
 from langchain_community.vectorstores.faiss import FAISS
 from langchain_openai import ChatOpenAI, OpenAIEmbeddings
 from langchain_text_splitters import RecursiveCharacterTextSplitter
-from langchain.tools.retriever import create_retriever_tool
-from langchain.agents import AgentExecutor, create_tool_calling_agent
-from langchain.tools import BaseTool, StructuredTool, tool
+from langchain_core.tools import create_retriever_tool
 
 loader = WebBaseLoader("https://docs.langwatch.ai")
 docs = loader.load()
@@ -48,24 +46,19 @@ async def on_chat_start():
 
     tools = [retriever_tool]
     model = ChatOpenAI(model="gpt-5", streaming=True)
-    prompt = ChatPromptTemplate.from_messages(
-        [
-            (
-                "system",
-                "You are a helpful assistant that only reply in short tweet-like responses, using lots of emojis and use tools only once.\n\n{agent_scratchpad}",
-            ),
-            ("human", "{question}"),
-        ]
+
+    agent = create_agent(
+        model=model,
+        tools=tools,
+        system_prompt="You are a helpful assistant that only reply in short tweet-like responses, using lots of emojis and use tools only once."
     )
-    agent = create_tool_calling_agent(model, tools, prompt)
-    executor = AgentExecutor(agent=agent, tools=tools, verbose=True)  # type: ignore
-    cl.user_session.set("agent", executor)
+    cl.user_session.set("agent", agent)
 
 
 @cl.on_message
 @langwatch.trace()
 async def main(message: cl.Message):
-    agent: AgentExecutor = cl.user_session.get("agent")  # type: ignore
+    agent = cl.user_session.get("agent")  # type: ignore
 
     msg = cl.Message(content="")
 
@@ -74,24 +67,18 @@ async def main(message: cl.Message):
     )
 
     async for chunk in agent.astream(
-        {
-            "question": message.content,
-            "messages": [HumanMessage(content="Hoi, dit is een test")],
-        },
+        {"messages": [HumanMessage(content=message.content)]},
         config=RunnableConfig(
             callbacks=[
-                cl.LangchainCallbackHandler(),
                 langwatch.get_current_trace().get_langchain_callback(),
             ]
         ),
     ):
-        if "output" in chunk:
-            await msg.stream_token(chunk["output"])
-        elif "actions" in chunk:
-            await msg.stream_token(chunk["actions"][0].log)
-        elif "steps" in chunk:
-            await msg.stream_token(chunk["steps"][0].observation + "\n\n")
-        else:
-            await msg.stream_token("<unammaped chunk>")
+        # In v1, create_agent streams message chunks differently
+        if "model" in chunk:
+            # This is the model response chunk
+            model_chunk = chunk["model"]
+            if hasattr(model_chunk, "content") and model_chunk.content:
+                await msg.stream_token(model_chunk.content)
 
     await msg.send()

{langwatch-0.6.0 → langwatch-0.7.0}/examples/langchain_rag_bot_vertex_ai.py

@@ -8,9 +8,9 @@ from langwatch.types import RAGChunk
 load_dotenv()
 
 import chainlit as cl
-from langchain.prompts import ChatPromptTemplate
-from langchain.schema import HumanMessage
-from langchain.schema.runnable.config import RunnableConfig
+from langchain_core.messages import HumanMessage
+from langchain_core.runnables import RunnableConfig
+from langchain.agents import create_agent
 
 import langwatch
 
@@ -18,9 +18,7 @@ from langchain_community.document_loaders import WebBaseLoader
 from langchain_community.vectorstores.faiss import FAISS
 from langchain_openai import ChatOpenAI, OpenAIEmbeddings
 from langchain_text_splitters import RecursiveCharacterTextSplitter
-from langchain.tools.retriever import create_retriever_tool
-from langchain.agents import AgentExecutor, create_tool_calling_agent
-from langchain.tools import BaseTool, StructuredTool, tool
+from langchain_core.tools import create_retriever_tool
 from langchain_google_vertexai import ChatVertexAI, VertexAI
 
 loader = WebBaseLoader("https://docs.langwatch.ai")
@@ -67,24 +65,19 @@ async def on_chat_start():
         location=os.environ["VERTEXAI_LOCATION"],
         streaming=True,
     )
-    prompt = ChatPromptTemplate.from_messages(
-        [
-            (
-                "system",
-                "You are a helpful assistant that only reply in short tweet-like responses, using lots of emojis and use tools only once.\n\n{agent_scratchpad}",
-            ),
-            ("human", "{question}"),
-        ]
+
+    agent = create_agent(
+        model=model,
+        tools=tools,
+        system_prompt="You are a helpful assistant that only reply in short tweet-like responses, using lots of emojis and use tools only once."
     )
-    agent = create_tool_calling_agent(model, tools, prompt)
-    executor = AgentExecutor(agent=agent, tools=tools, verbose=True)  # type: ignore
-    cl.user_session.set("agent", executor)
+    cl.user_session.set("agent", agent)
 
 
 @cl.on_message
 @langwatch.trace()
 async def main(message: cl.Message):
-    agent: AgentExecutor = cl.user_session.get("agent")  # type: ignore
+    agent = cl.user_session.get("agent")  # type: ignore
 
     msg = cl.Message(content="")
 
@@ -93,24 +86,18 @@ async def main(message: cl.Message):
     )
 
     async for chunk in agent.astream(
-        {
-            "question": message.content,
-            "messages": [HumanMessage(content="Hoi, dit is een test")],
-        },
+        {"messages": [HumanMessage(content=message.content)]},
         config=RunnableConfig(
             callbacks=[
-                cl.LangchainCallbackHandler(),
                 langwatch.get_current_trace().get_langchain_callback(),
             ]
         ),
     ):
-        if "output" in chunk:
-            await msg.stream_token(chunk["output"])
-        elif "actions" in chunk:
-            await msg.stream_token(chunk["actions"][0].log)
-        elif "steps" in chunk:
-            await msg.stream_token(chunk["steps"][0].observation + "\n\n")
-        else:
-            await msg.stream_token("<unammaped chunk>")
+        # In v1, create_agent streams message chunks differently
+        if "model" in chunk:
+            # This is the model response chunk
+            model_chunk = chunk["model"]
+            if hasattr(model_chunk, "content") and model_chunk.content:
+                await msg.stream_token(model_chunk.content)
 
     await msg.send()

{langwatch-0.6.0 → langwatch-0.7.0}/examples/langchain_rag_bot_with_threads.py

@@ -15,9 +15,9 @@ from langwatch.types import RAGChunk
 load_dotenv()
 
 import chainlit as cl
-from langchain.prompts import ChatPromptTemplate
-from langchain.schema import HumanMessage
-from langchain.schema.runnable.config import RunnableConfig
+from langchain_core.messages import HumanMessage
+from langchain_core.runnables import RunnableConfig
+from langchain.agents import create_agent
 
 import langwatch
 
@@ -25,9 +25,7 @@ from langchain_community.document_loaders import WebBaseLoader
 from langchain_community.vectorstores.faiss import FAISS
 from langchain_openai import ChatOpenAI, OpenAIEmbeddings
 from langchain_text_splitters import RecursiveCharacterTextSplitter
-from langchain.tools.retriever import create_retriever_tool
-from langchain.agents import AgentExecutor, create_tool_calling_agent
-from langchain.tools import BaseTool, StructuredTool, tool
+from langchain_core.tools import create_retriever_tool
 import chainlit as cl
 import chainlit.data as cl_data
 from chainlit.data.utils import queue_until_user_message
@@ -254,24 +252,19 @@ async def on_chat_start():
 
     tools = [retriever_tool]
     model = ChatOpenAI(streaming=True)
-    prompt = ChatPromptTemplate.from_messages(
-        [
-            (
-                "system",
-                "You are a helpful assistant that only reply in short tweet-like responses, using lots of emojis and use tools only once.\n\n{agent_scratchpad}",
-            ),
-            ("human", "{question}"),
-        ]
+    
+    agent = create_agent(
+        model=model,
+        tools=tools,
+        system_prompt="You are a helpful assistant that only reply in short tweet-like responses, using lots of emojis and use tools only once."
     )
-    agent = create_tool_calling_agent(model, tools, prompt)
-    executor = AgentExecutor(agent=agent, tools=tools, verbose=True)  # type: ignore
-    cl.user_session.set("agent", executor)
+    cl.user_session.set("agent", agent)
 
 
 @cl.on_message
 @langwatch.trace()
 async def main(message: cl.Message):
-    agent: AgentExecutor = cl.user_session.get("agent")  # type: ignore
+    agent = cl.user_session.get("agent")  # type: ignore
 
     msg = cl.Message(content="")
 
@@ -284,25 +277,19 @@ async def main(message: cl.Message):
     )
 
     async for chunk in agent.astream(
-        {
-            "question": message.content,
-            "messages": [HumanMessage(content="Hoi, dit is een test")],
-        },
+        {"messages": [HumanMessage(content=message.content)]},
         config=RunnableConfig(
             callbacks=[
-                cl.LangchainCallbackHandler(),
                 langwatch.get_current_trace().get_langchain_callback(),
             ]
         ),
     ):
-        if "output" in chunk:
-            await msg.stream_token(chunk["output"])
-        elif "actions" in chunk:
-            await msg.stream_token(chunk["actions"][0].log)
-        elif "steps" in chunk:
-            await msg.stream_token(chunk["steps"][0].observation + "\n\n")
-        else:
-            await msg.stream_token("<unammaped chunk>")
+        # In v1, create_agent streams message chunks differently
+        if "model" in chunk:
+            # This is the model response chunk
+            model_chunk = chunk["model"]
+            if hasattr(model_chunk, "content") and model_chunk.content:
+                await msg.stream_token(model_chunk.content)
 
     await msg.send()
 

{langwatch-0.6.0 → langwatch-0.7.0}/examples/langgraph_rag_bot_with_threads.py

@@ -15,9 +15,8 @@ from langwatch.types import RAGChunk
 load_dotenv()
 
 import chainlit as cl
-from langchain.prompts import ChatPromptTemplate
-from langchain.schema import HumanMessage
-from langchain.schema.runnable.config import RunnableConfig
+from langchain_core.messages import HumanMessage
+from langchain_core.runnables import RunnableConfig
 
 import langwatch
 
@@ -25,9 +24,7 @@ from langchain_community.document_loaders import WebBaseLoader
 from langchain_community.vectorstores.faiss import FAISS
 from langchain_openai import ChatOpenAI, OpenAIEmbeddings
 from langchain_text_splitters import RecursiveCharacterTextSplitter
-from langchain.tools.retriever import create_retriever_tool
-from langchain.agents import AgentExecutor, create_tool_calling_agent
-from langchain.tools import BaseTool, StructuredTool, tool
+from langchain_core.tools import create_retriever_tool
 import chainlit as cl
 import chainlit.data as cl_data
 from chainlit.data.utils import queue_until_user_message
@@ -300,7 +297,6 @@ async def main(message: cl.Message):
         {"messages": [HumanMessage(content=message.content)]},
         config=RunnableConfig(
             callbacks=[
-                cl.LangchainCallbackHandler(),
                 langwatch.get_current_trace().get_langchain_callback(),
             ]
         ),

{langwatch-0.6.0 → langwatch-0.7.0}/examples/langwatch_docs_rag_evaluation.ipynb

@@ -14,7 +14,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 1,
+   "execution_count": null,
    "id": "a61c2b75",
    "metadata": {},
    "outputs": [
@@ -70,7 +70,7 @@
     "from langchain_text_splitters import RecursiveCharacterTextSplitter\n",
     "from langchain.tools.retriever import create_retriever_tool\n",
     "from langchain.agents import AgentExecutor, create_tool_calling_agent\n",
-    "from langchain.tools import BaseTool, StructuredTool, tool\n",
+    "from langchain.tools import BaseTool\n",
     "from langchain_core.documents import Document\n",
     "\n",
     "\n",

{langwatch-0.6.0 → langwatch-0.7.0}/examples/opentelemetry/openinference_langchain_bot.py

@@ -6,10 +6,9 @@ import langwatch
 load_dotenv()
 
 import chainlit as cl
-from langchain.prompts import ChatPromptTemplate
-from langchain.schema import StrOutputParser
-from langchain.schema.runnable import Runnable
-from langchain.schema.runnable.config import RunnableConfig
+from langchain_core.prompts import ChatPromptTemplate
+from langchain_core.output_parsers import StrOutputParser
+from langchain_core.runnables import Runnable, RunnableConfig
 
 from openinference.instrumentation.langchain import LangChainInstrumentor
 
@@ -42,11 +41,7 @@ async def main(message: cl.Message):
 
     async for chunk in runnable.astream(
         {"question": message.content},
-        config=RunnableConfig(
-            callbacks=[
-                cl.LangchainCallbackHandler(),
-            ]
-        ),
+        config=RunnableConfig(),
     ):
         await msg.stream_token(chunk)
 

{langwatch-0.6.0 → langwatch-0.7.0}/examples/opentelemetry/openllmetry_langchain_bot.py

@@ -6,10 +6,9 @@ import langwatch
 load_dotenv()
 
 import chainlit as cl
-from langchain.prompts import ChatPromptTemplate
-from langchain.schema import StrOutputParser
-from langchain.schema.runnable import Runnable
-from langchain.schema.runnable.config import RunnableConfig
+from langchain_core.prompts import ChatPromptTemplate
+from langchain_core.output_parsers import StrOutputParser
+from langchain_core.runnables import Runnable, RunnableConfig
 
 from opentelemetry.instrumentation.langchain import LangchainInstrumentor
 
@@ -43,9 +42,6 @@ async def main(message: cl.Message):
     async for chunk in runnable.astream(
         {"question": message.content},
         config=RunnableConfig(
-            callbacks=[
-                cl.LangchainCallbackHandler(),
-            ],
             metadata={
                 "user_id": "123",
                 "thread_id": "789",

{langwatch-0.6.0 → langwatch-0.7.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "langwatch"
-version = "0.6.0" # remember to also update it in src/langwatch/__version__.py
+version = "0.7.0" # remember to also update it in src/langwatch/__version__.py
 description = "LangWatch Python SDK, for monitoring your LLMs"
 authors = [{ name = "Langwatch Engineers", email = "engineering@langwatch.ai" }]
 requires-python = ">=3.10,<3.14"
@@ -40,6 +40,7 @@ dependencies = [
     "attrs>=24",
     "pksuid>=1.1.2",
     "pyyaml>=6.0.2",
+    "langchain-core>=1.0.0,<2.0.0",
 ]
 
 [project.optional-dependencies]
@@ -75,15 +76,15 @@ examples = [
     "fastapi>=0.115.12",
     "haystack-ai>=2.13.2,<3.0.0",
     "jupyter>=1.0.0,<2.0.0",
-    "langchain>=0.2.11",
-    "langchain_community>=0.2.11",
-    "langchain-openai>=0.1.19",
-    "langchain-text-splitters>=0.3.0,<0.4.0",
-    "langchain-google-vertexai>=1.0.7",
+    "langchain>=1.0.0,<2.0.0",
+    "langchain-community>=0.3.0,<1.0.0",
+    "langchain-openai>=0.3.0,<1.0.0",
+    "langchain-text-splitters>=0.3.0,<1.0.0",
+    "langchain-google-vertexai>=2.0.0,<3.0.0",
     "litellm>=1.52.1",
     "openinference-instrumentation-dspy>=0.1.23,<0.2.0",
     "opentelemetry-instrumentation-fastapi>=0.53b1",
-    "langgraph>=0.2.35,<0.3.0",
+    "langgraph>=1.0.0,<2.0.0",
     "openai>=1.68.2",
     "pydantic>=1,<2.10.2", # v2.10.2 is broken for chainlit
     "pytest>=7.4.2,<8.0.0",
@@ -95,7 +96,7 @@ examples = [
     "opentelemetry-instrumentation-langchain>=0.40.4",
     "openinference-instrumentation-litellm>=0.1.19",
     "pandas>=2.2.3",
-    "strands-agents[otel]>=1.9.0,<2.0.0",
+    "strands-agents[otel]>=1.14.0,<2.0.0",
     "unstructured[pdf]>=0.18.3",
 ]
 tests = ["pytest-asyncio>=0.21.2"]

langwatch-0.7.0/src/langwatch/__version__.py

@@ -0,0 +1,3 @@
+"""Version information for LangWatch."""
+
+__version__ = "0.7.0" # x-release-please-version

{langwatch-0.6.0 → langwatch-0.7.0}/src/langwatch/langchain.py

@@ -11,20 +11,18 @@ from typing import (
     Union,
 )
 from warnings import warn
-from langchain.schema import (
-    LLMResult,
-    AgentAction,
-    AgentFinish,
+from langchain_core.messages import (
     BaseMessage,
     HumanMessage,
     AIMessage,
     SystemMessage,
     FunctionMessage,
-    ChatGeneration,
 )
+from langchain_core.outputs import LLMResult, ChatGeneration
+from langchain_core.agents import AgentAction, AgentFinish
 from langwatch.utils.initialization import ensure_setup
 from opentelemetry.trace import get_current_span, SpanContext
-from langchain.callbacks.base import BaseCallbackHandler
+from langchain_core.callbacks.base import BaseCallbackHandler
 from langwatch.telemetry.span import LangWatchSpan
 from langwatch.telemetry.tracing import LangWatchTrace
 from langwatch.utils.transformation import (
@@ -49,7 +47,7 @@ from langwatch.domain import (
 )
 
 from uuid import UUID
-from langchain.tools import BaseTool
+from langchain_core.tools import BaseTool
 import logging