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

Loading…
Cancel
Save