Fix llm not recognizing pydantic tool definition for dict types (#782)

* Update src/tools.py pydantic imports to use v1 from v2 Signed-off-by: JoshuaL3000 <[email protected]>
opea-project · Oct 21, 2024 · f36ca80 · f36ca80
1 parent 7e085a4
commit f36ca80
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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 import BaseModel, Field, create_model
+from pydantic.v1 import BaseModel, Field, create_model
 
 
 def generate_request_function(url):