blob42/langchain
1
0
Fork 0
forked from Archives/langchain
Code Issues Pull Requests Packages Projects Releases Wiki Activity

small change to allow powerbi tools to all have single inputs (#3864)

Browse Source 
Small change in the tool input so that the single_input_tool function
works against all powerbi tools
This commit is contained in:
Eduard van Valkenburg 2023-05-02 05:22:16 +02:00 committed by GitHub
parent c38cafd6c2
commit 43a0cb4b92
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
langchain/tools/powerbi/tool.py
View File

@ -149,18 +149,16 @@ class ListPowerBITool(BaseTool):
def _run( def _run(
self, self,
*args: Any, tool_input: Optional[str] = None,
run_manager: Optional[CallbackManagerForToolRun] = None, run_manager: Optional[CallbackManagerForToolRun] = None,
**kwargs: Any,
) -> str: ) -> str:
"""Get the names of the tables.""" """Get the names of the tables."""
return ", ".join(self.powerbi.get_table_names()) return ", ".join(self.powerbi.get_table_names())
async def _arun( async def _arun(
self, self,
*args: Any, tool_input: Optional[str] = None,
run_manager: Optional[AsyncCallbackManagerForToolRun] = None, run_manager: Optional[AsyncCallbackManagerForToolRun] = None,
**kwargs: Any,
) -> str: ) -> str:
"""Get the names of the tables.""" """Get the names of the tables."""
return ", ".join(self.powerbi.get_table_names()) return ", ".join(self.powerbi.get_table_names())