chore: template tool models should reside in tool module (#188)

Author: mathislucka

src/deepset_mcp/api/pipeline_template/models.py

@@ -74,23 +74,3 @@ def populate_yaml_config(cls, values: Any) -> Any:
             values["yaml_config"] = yaml_config
 
         return values
-
-
-class PipelineTemplateSearchResult(BaseModel):
-    """Model representing a search result for pipeline templates."""
-
-    template: PipelineTemplate
-    "Pipeline template that matched the search criteria"
-    similarity_score: float
-    "Relevance score indicating how well the template matches the search"
-
-
-class PipelineTemplateSearchResults(BaseModel):
-    """Response model for pipeline template search results."""
-
-    results: list[PipelineTemplateSearchResult]
-    "List of pipeline templates matching the search criteria"
-    query: str
-    "Original search query string"
-    total_found: int
-    "Total number of templates found matching the search criteria"

src/deepset_mcp/tools/pipeline_template.py

@@ -3,12 +3,11 @@
 # SPDX-License-Identifier: Apache-2.0
 
 import numpy as np
+from pydantic import BaseModel
 
 from deepset_mcp.api.exceptions import ResourceNotFoundError, UnexpectedAPIError
 from deepset_mcp.api.pipeline_template.models import (
     PipelineTemplate,
-    PipelineTemplateSearchResult,
-    PipelineTemplateSearchResults,
     PipelineType,
 )
 from deepset_mcp.api.protocols import AsyncClientProtocol
@@ -61,6 +60,26 @@ async def get_template(*, client: AsyncClientProtocol, workspace: str, template_
         return f"Failed to fetch pipeline template '{template_name}': {e}"
 
 
+class PipelineTemplateSearchResult(BaseModel):
+    """Model representing a search result for pipeline templates."""
+
+    template: PipelineTemplate
+    "Pipeline template that matched the search criteria"
+    similarity_score: float
+    "Relevance score indicating how well the template matches the search"
+
+
+class PipelineTemplateSearchResults(BaseModel):
+    """Response model for pipeline template search results."""
+
+    results: list[PipelineTemplateSearchResult]
+    "List of pipeline templates matching the search criteria"
+    query: str
+    "Original search query string"
+    total_found: int
+    "Total number of templates found matching the search criteria"
+
+
 async def search_templates(
     *,
     client: AsyncClientProtocol,

test/unit/tools/test_pipeline_template.py

@@ -11,13 +11,13 @@
 from deepset_mcp.api.exceptions import ResourceNotFoundError, UnexpectedAPIError
 from deepset_mcp.api.pipeline_template.models import (
     PipelineTemplate,
-    PipelineTemplateSearchResults,
     PipelineTemplateTag,
     PipelineType,
 )
 from deepset_mcp.api.shared_models import PaginatedResponse
 from deepset_mcp.tools.model_protocol import ModelProtocol
 from deepset_mcp.tools.pipeline_template import (
+    PipelineTemplateSearchResults,
     get_template,
     list_templates,
     search_templates,