From 85f673c7eab133b686d4825342def038dd9957c2 Mon Sep 17 00:00:00 2001 From: Eugene Yurtsev Date: Thu, 12 Sep 2024 16:51:28 -0400 Subject: [PATCH] langchain-cli[patch]: Update to work with langchain 0.3 (#26403) Update to work with pydantic 2 / langchain 0.3 --- .../integration_template/integration_template/chat_models.py | 4 ++-- .../integration_template/integration_template/tools.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/libs/cli/langchain_cli/integration_template/integration_template/chat_models.py b/libs/cli/langchain_cli/integration_template/integration_template/chat_models.py index ed5134e763e..d299fb2ff26 100644 --- a/libs/cli/langchain_cli/integration_template/integration_template/chat_models.py +++ b/libs/cli/langchain_cli/integration_template/integration_template/chat_models.py @@ -116,7 +116,7 @@ class Chat__ModuleName__(BaseChatModel): Tool calling: .. code-block:: python - from langchain_core.pydantic_v1 import BaseModel, Field + from pydantic import BaseModel, Field class GetWeather(BaseModel): '''Get the current weather in a given location''' @@ -144,7 +144,7 @@ class Chat__ModuleName__(BaseChatModel): from typing import Optional - from langchain_core.pydantic_v1 import BaseModel, Field + from pydantic import BaseModel, Field class Joke(BaseModel): '''Joke to tell user.''' diff --git a/libs/cli/langchain_cli/integration_template/integration_template/tools.py b/libs/cli/langchain_cli/integration_template/integration_template/tools.py index 0ada7bcef56..57deb006f06 100644 --- a/libs/cli/langchain_cli/integration_template/integration_template/tools.py +++ b/libs/cli/langchain_cli/integration_template/integration_template/tools.py @@ -5,8 +5,8 @@ from typing import Optional, Type from langchain_core.callbacks import ( CallbackManagerForToolRun, ) -from langchain_core.pydantic_v1 import BaseModel from langchain_core.tools import BaseTool +from pydantic import BaseModel class __ModuleName__Input(BaseModel): @@ -62,7 +62,7 @@ class __ModuleName__Tool(BaseTool): .. code-block:: python # TODO: output of invocation - """ # noqa: E501 + """ # noqa: E501 # TODO: Set tool name and description name: str = "TODO: Tool name"