forked from Archives/langchain
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
This commit is contained in:
parent
c38cafd6c2
commit
43a0cb4b92
@ -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…
Reference in New Issue
Block a user