jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

refactor: update import statements in tools (#2796)

Browse source 
* 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
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-07-18 14:24:06 -03:00 • committed by GitHub
commit d897752f0a
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
4 changed files with 6 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

3
src/backend/base/langflow/base/tools/flow_tool.py
View file

@ -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

2
src/backend/base/langflow/components/tools/PythonCodeStructuredTool.py
View file

@ -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

2
src/backend/base/langflow/components/tools/RetrieverTool.py
View file

@ -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

5
src/backend/base/langflow/components/tools/WikipediaAPI.py
View file

@ -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