From 43a0cb4b92245139ebc5c6783262210b5c9f67d8 Mon Sep 17 00:00:00 2001 From: Eduard van Valkenburg Date: Tue, 2 May 2023 05:22:16 +0200 Subject: [PATCH] small change to allow powerbi tools to all have single inputs (#3864) Small change in the tool input so that the single_input_tool function works against all powerbi tools --- langchain/tools/powerbi/tool.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/langchain/tools/powerbi/tool.py b/langchain/tools/powerbi/tool.py index 633f99d3..58e27b2e 100644 --- a/langchain/tools/powerbi/tool.py +++ b/langchain/tools/powerbi/tool.py @@ -149,18 +149,16 @@ class ListPowerBITool(BaseTool): def _run( self, - *args: Any, + tool_input: Optional[str] = None, run_manager: Optional[CallbackManagerForToolRun] = None, - **kwargs: Any, ) -> str: """Get the names of the tables.""" return ", ".join(self.powerbi.get_table_names()) async def _arun( self, - *args: Any, + tool_input: Optional[str] = None, run_manager: Optional[AsyncCallbackManagerForToolRun] = None, - **kwargs: Any, ) -> str: """Get the names of the tables.""" return ", ".join(self.powerbi.get_table_names())