Skip to content

Commit 14b7e1c

Browse files
committed
fix: add server_config param to list_tools function
1 parent 51ae30f commit 14b7e1c

File tree

1 file changed

+2
-2
lines changed
  • src/datapilot/core/mcp_utils

1 file changed

+2
-2
lines changed

src/datapilot/core/mcp_utils/mcp.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -115,7 +115,7 @@ def create_mcp_proxy():
115115
}
116116

117117
# Execute with processed parameters
118-
output = asyncio.run(list_tools(command=server_config["command"], args=processed_args, env=processed_env))
118+
output = asyncio.run(list_tools(server_config=server_config, command=server_config["command"], args=processed_args, env=processed_env))
119119
# Add processed parameters to output
120120
output_with_name = {
121121
"name": server_name,
@@ -141,7 +141,7 @@ def create_mcp_proxy():
141141
click.secho(f"\nFailed to copy to clipboard: {str(e)}", fg="yellow")
142142

143143

144-
async def list_tools(command: str, args: list[str], env: dict[str, str]):
144+
async def list_tools(server_config: dict, command: str, args: list[str], env: dict[str, str]):
145145
command_path = shutil.which(command)
146146
if not command_path:
147147
raise click.UsageError(f"Command not found: {command}")

0 commit comments

Comments
 (0)