Tool Use & MCP

1
import data_designer.config as dd
2
from data_designer.interface import DataDesigner
3
4
# 1. Configure provider
5
6
## Local Stdio provider
7
mcp_provider = dd.LocalStdioMCPProvider(
8
    name="demo-mcp",
9
    command="python",
10
    args=["-m", "my_mcp_server"],
11
)
12
13
## Remote provider
14
# mcp_provider = dd.MCPProvider(
15
#     name="remote-mcp",
16
#     endpoint="https://mcp.example.invalid/sse",
17
#     api_key="REMOTE_MCP_API_KEY",
18
# )
19
20
data_designer = DataDesigner(mcp_providers=[mcp_provider])
21
22
# 2. Create tool config
23
tool_config = dd.ToolConfig(
24
    tool_alias="my-tools",
25
    providers=["demo-mcp"],
26
)
27
28
builder = dd.DataDesignerConfigBuilder(tool_configs=[tool_config])
29
30
# 3. Use tools in column
31
builder.add_column(
32
    dd.LLMTextColumnConfig(
33
        name="answer",
34
        prompt="Use tools to answer: {{ question }}",
35
        model_alias="nvidia-text",
36
        tool_alias="my-tools",
37
    )
38
)