Skip to content

Commit c8ab3e2

Browse files
authored
feat: only expose tool params that are needed (#151)
1 parent 362dbd3 commit c8ab3e2

1 file changed

Lines changed: 18 additions & 3 deletions

File tree

src/deepset_mcp/tool_factory.py

Lines changed: 18 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -169,11 +169,21 @@ def get_workspace_from_env() -> str:
169169
),
170170
"create_pipeline": (
171171
create_pipeline_tool,
172-
ToolConfig(needs_client=True, needs_workspace=True, memory_type=MemoryType.BOTH),
172+
ToolConfig(
173+
needs_client=True,
174+
needs_workspace=True,
175+
memory_type=MemoryType.BOTH,
176+
custom_args={"skip_validation_errors": True},
177+
),
173178
),
174179
"update_pipeline": (
175180
update_pipeline_tool,
176-
ToolConfig(needs_client=True, needs_workspace=True, memory_type=MemoryType.BOTH),
181+
ToolConfig(
182+
needs_client=True,
183+
needs_workspace=True,
184+
memory_type=MemoryType.BOTH,
185+
custom_args={"skip_validation_errors": True},
186+
),
177187
),
178188
"get_pipeline": (
179189
get_pipeline_tool,
@@ -222,7 +232,12 @@ def get_workspace_from_env() -> str:
222232
),
223233
"list_pipeline_templates": (
224234
list_pipeline_templates_tool,
225-
ToolConfig(needs_client=True, needs_workspace=True, memory_type=MemoryType.EXPLORABLE),
235+
ToolConfig(
236+
needs_client=True,
237+
needs_workspace=True,
238+
memory_type=MemoryType.EXPLORABLE,
239+
custom_args={"filter": None, "field": "created_at", "order": "DESC", "limit": 100},
240+
),
226241
),
227242
"get_pipeline_template": (
228243
get_pipeline_template_tool,

0 commit comments

Comments
 (0)