Skip to content

Commit a8bb136

Browse files
committed
fix: normalize input tokens and preserve all server config args/env
1 parent 14b7e1c commit a8bb136

File tree

1 file changed

+3
-4
lines changed
  • src/datapilot/core/mcp_utils

1 file changed

+3
-4
lines changed

src/datapilot/core/mcp_utils/mcp.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -122,13 +122,12 @@ def create_mcp_proxy():
122122
"config": input_configs,
123123
"command": server_config["command"],
124124
"args": [
125-
f"${{{input_id}}}" if any(input_val in str(arg) for input_val in inputs.values()) else arg
126-
for arg, input_id in zip(server_config.get("args", []), inputs.keys())
125+
arg.replace("${input:", "${") if isinstance(arg, str) else arg
126+
for arg in server_config.get("args", [])
127127
],
128128
"env": {
129-
k: f"${{{input_id}}}" if any(input_val in str(v) for input_val in inputs.values()) else v
129+
k: v.replace("${input:", "${") if isinstance(v, str) else v
130130
for k, v in server_config.get("env", {}).items()
131-
for input_id in inputs
132131
},
133132
**output
134133
}

0 commit comments

Comments
 (0)