From d897752f0a01629cc5469a3e2c7f37551ba29c2f Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Thu, 18 Jul 2024 14:24:06 -0300 Subject: [PATCH] refactor: update import statements in tools (#2796) * fix: update import statement for WikipediaAPI.py * fix: update import statement for flow_tool.py * fix: update import statement in StructuredTool * fix: update import in retriever tool --- src/backend/base/langflow/base/tools/flow_tool.py | 3 +-- .../langflow/components/tools/PythonCodeStructuredTool.py | 2 +- src/backend/base/langflow/components/tools/RetrieverTool.py | 2 +- src/backend/base/langflow/components/tools/WikipediaAPI.py | 5 +++-- 4 files changed, 6 insertions(+), 6 deletions(-) diff --git a/src/backend/base/langflow/base/tools/flow_tool.py b/src/backend/base/langflow/base/tools/flow_tool.py index 4f767e4da..03936ac58 100644 --- a/src/backend/base/langflow/base/tools/flow_tool.py +++ b/src/backend/base/langflow/base/tools/flow_tool.py @@ -1,9 +1,8 @@ from typing import Any, List, Optional, Type from asyncer import syncify -from langchain.tools import BaseTool from langchain_core.runnables import RunnableConfig -from langchain_core.tools import ToolException +from langchain_core.tools import BaseTool, ToolException from pydantic.v1 import BaseModel from langflow.base.flow_processing.utils import build_data_from_result_data, format_flow_output_data diff --git a/src/backend/base/langflow/components/tools/PythonCodeStructuredTool.py b/src/backend/base/langflow/components/tools/PythonCodeStructuredTool.py index ef3c3b8a8..3d7078545 100644 --- a/src/backend/base/langflow/components/tools/PythonCodeStructuredTool.py +++ b/src/backend/base/langflow/components/tools/PythonCodeStructuredTool.py @@ -2,7 +2,7 @@ import ast from typing import Any, Dict, List, Optional from langchain.agents import Tool -from langchain.tools import StructuredTool +from langchain_core.tools import StructuredTool from langflow.custom import CustomComponent from langflow.schema.dotdict import dotdict diff --git a/src/backend/base/langflow/components/tools/RetrieverTool.py b/src/backend/base/langflow/components/tools/RetrieverTool.py index 6a87bcbf9..43dfbcba6 100644 --- a/src/backend/base/langflow/components/tools/RetrieverTool.py +++ b/src/backend/base/langflow/components/tools/RetrieverTool.py @@ -1,4 +1,4 @@ -from langchain.tools.retriever import create_retriever_tool +from langchain_core.tools import create_retriever_tool from langflow.custom import CustomComponent from langflow.field_typing import BaseRetriever, Tool diff --git a/src/backend/base/langflow/components/tools/WikipediaAPI.py b/src/backend/base/langflow/components/tools/WikipediaAPI.py index 33cdaedb1..21c166050 100644 --- a/src/backend/base/langflow/components/tools/WikipediaAPI.py +++ b/src/backend/base/langflow/components/tools/WikipediaAPI.py @@ -1,4 +1,5 @@ -from langchain.tools import WikipediaQueryRun +from typing import cast +from langchain_community.tools import WikipediaQueryRun from langchain_community.utilities.wikipedia import WikipediaAPIWrapper from langflow.base.langchain_utilities.model import LCToolComponent @@ -34,7 +35,7 @@ class WikipediaAPIComponent(LCToolComponent): def build_tool(self) -> Tool: wrapper = self._build_wrapper() - return WikipediaQueryRun(api_wrapper=wrapper) + return cast(Tool, WikipediaQueryRun(api_wrapper=wrapper)) def _build_wrapper(self) -> WikipediaAPIWrapper: return WikipediaAPIWrapper( # type: ignore