feat: use memory with pipeline tools (#113)

Author: mathislucka

src/deepset_mcp/api/pipeline/models.py

@@ -188,3 +188,26 @@ class DeepsetStreamEvent(BaseModel):
     delta: StreamDelta | None = None
     result: DeepsetSearchResponse | None = None
     error: str | None = None
+
+
+class PipelineList(BaseModel):
+    """Response model for listing pipelines."""
+
+    data: list[DeepsetPipeline]
+    has_more: bool
+    total: int
+
+
+class PipelineValidationResultWithYaml(BaseModel):
+    """Model for pipeline validation result that includes the original YAML."""
+
+    validation_result: PipelineValidationResult
+    yaml_config: str
+
+
+class PipelineOperationWithErrors(BaseModel):
+    """Model for pipeline operations that complete with validation errors."""
+
+    message: str
+    validation_result: PipelineValidationResult
+    pipeline: DeepsetPipeline

src/deepset_mcp/api/pipeline/resource.py

@@ -9,6 +9,7 @@
     DeepsetPipeline,
     DeepsetSearchResponse,
     DeepsetStreamEvent,
+    PipelineList,
     PipelineLogList,
     PipelineValidationResult,
     ValidationError,
@@ -73,12 +74,12 @@ async def list(
         self,
         page_number: int = 1,
         limit: int = 10,
-    ) -> list[DeepsetPipeline]:
+    ) -> PipelineList:
         """Retrieve pipeline in the configured workspace with optional pagination.
 
         :param page_number: Page number for paging.
         :param limit: Max number of items to return.
-        :returns: List of DeepsetPipeline instances.
+        :returns: PipelineList with pipelines and metadata.
         """
         params: dict[str, Any] = {
             "page_number": page_number,
@@ -97,10 +98,13 @@ async def list(
 
         if response is not None:
             pipelines = [DeepsetPipeline.model_validate(item) for item in response.get("data", [])]
+            return PipelineList(
+                data=pipelines,
+                has_more=response.get("has_more", False),
+                total=response.get("total", len(pipelines)),
+            )
         else:
-            pipelines = []
-
-        return pipelines
+            return PipelineList(data=[], has_more=False, total=0)
 
     async def get(self, pipeline_name: str, include_yaml: bool = True) -> DeepsetPipeline:
         """Fetch a single pipeline by its name.

src/deepset_mcp/api/protocols.py

@@ -10,6 +10,7 @@
     DeepsetPipeline,
     DeepsetSearchResponse,
     DeepsetStreamEvent,
+    PipelineList,
     PipelineLogList,
     PipelineValidationResult,
 )
@@ -257,7 +258,7 @@ async def list(
         self,
         page_number: int = 1,
         limit: int = 10,
-    ) -> list[DeepsetPipeline]:
+    ) -> PipelineList:
         """List pipelines in the configured workspace with optional pagination."""
         ...
 

src/deepset_mcp/tool_factory.py

@@ -128,14 +128,38 @@ def get_workspace_from_env() -> str:
 # Tool registry with configurations
 TOOL_REGISTRY = {
     # Workspace tools
-    "list_pipelines": (list_pipelines_tool, ToolConfig(needs_client=True, needs_workspace=True)),
-    "create_pipeline": (create_pipeline_tool, ToolConfig(needs_client=True, needs_workspace=True)),
-    "update_pipeline": (update_pipeline_tool, ToolConfig(needs_client=True, needs_workspace=True)),
-    "get_pipeline": (get_pipeline_tool, ToolConfig(needs_client=True, needs_workspace=True)),
-    "deploy_pipeline": (deploy_pipeline_tool, ToolConfig(needs_client=True, needs_workspace=True)),
-    "validate_pipeline": (validate_pipeline_tool, ToolConfig(needs_client=True, needs_workspace=True)),
-    "get_pipeline_logs": (get_pipeline_logs_tool, ToolConfig(needs_client=True, needs_workspace=True)),
-    "search_pipeline": (search_pipeline_tool, ToolConfig(needs_client=True, needs_workspace=True)),
+    "list_pipelines": (
+        list_pipelines_tool,
+        ToolConfig(needs_client=True, needs_workspace=True, memory_type=MemoryType.EXPLORABLE),
+    ),
+    "create_pipeline": (
+        create_pipeline_tool,
+        ToolConfig(needs_client=True, needs_workspace=True, memory_type=MemoryType.BOTH),
+    ),
+    "update_pipeline": (
+        update_pipeline_tool,
+        ToolConfig(needs_client=True, needs_workspace=True, memory_type=MemoryType.BOTH),
+    ),
+    "get_pipeline": (
+        get_pipeline_tool,
+        ToolConfig(needs_client=True, needs_workspace=True, memory_type=MemoryType.EXPLORABLE),
+    ),
+    "deploy_pipeline": (
+        deploy_pipeline_tool,
+        ToolConfig(needs_client=True, needs_workspace=True, memory_type=MemoryType.EXPLORABLE),
+    ),
+    "validate_pipeline": (
+        validate_pipeline_tool,
+        ToolConfig(needs_client=True, needs_workspace=True, memory_type=MemoryType.BOTH),
+    ),
+    "get_pipeline_logs": (
+        get_pipeline_logs_tool,
+        ToolConfig(needs_client=True, needs_workspace=True, memory_type=MemoryType.EXPLORABLE),
+    ),
+    "search_pipeline": (
+        search_pipeline_tool,
+        ToolConfig(needs_client=True, needs_workspace=True, memory_type=MemoryType.EXPLORABLE),
+    ),
     "list_indexes": (
         list_indexes_tool,
         ToolConfig(needs_client=True, needs_workspace=True, memory_type=MemoryType.EXPLORABLE),