diff --git a/comps/agent/langchain/src/tools.py b/comps/agent/langchain/src/tools.py index 5ac0906db..ff3918cc9 100644 --- a/comps/agent/langchain/src/tools.py +++ b/comps/agent/langchain/src/tools.py @@ -8,7 +8,7 @@ import yaml from langchain.tools import BaseTool, StructuredTool from langchain_community.agent_toolkits.load_tools import load_tools -from pydantic.v1 import BaseModel, Field, create_model +from pydantic import BaseModel, Field, create_model def generate_request_function(url):