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
fix_agent_callbacks
Eduard van Valkenburg 1 year ago committed by GitHub
parent c38cafd6c2
commit 43a0cb4b92
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

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

Loading…
Cancel
Save