chore: export tools (#192)

Author: mathislucka

src/deepset_mcp/tools/__init__.py

@@ -2,3 +2,60 @@
 #
 # SPDX-License-Identifier: Apache-2.0
 
+from .custom_components import get_latest_custom_component_installation_logs, list_custom_component_installations
+from .doc_search import search_docs
+from .haystack_service import (
+    get_component_definition,
+    get_custom_components,
+    list_component_families,
+    run_component,
+    search_component_definition,
+)
+from .indexes import create_index, deploy_index, get_index, list_indexes, update_index
+from .object_store import create_get_from_object_store, create_get_slice_from_object_store
+from .pipeline import (
+    create_pipeline,
+    deploy_pipeline,
+    get_pipeline,
+    get_pipeline_logs,
+    list_pipelines,
+    search_pipeline,
+    update_pipeline,
+    validate_pipeline,
+)
+from .pipeline_template import get_template, list_templates, search_templates
+from .secrets import get_secret, list_secrets
+from .workspace import get_workspace, list_workspaces
+
+__all__ = [
+    "list_custom_component_installations",
+    "get_latest_custom_component_installation_logs",
+    "search_docs",
+    "run_component",
+    "get_custom_components",
+    "get_component_definition",
+    "search_component_definition",
+    "list_component_families",
+    "list_indexes",
+    "deploy_index",
+    "update_index",
+    "create_index",
+    "get_index",
+    "create_get_from_object_store",
+    "create_get_slice_from_object_store",
+    "list_pipelines",
+    "get_pipeline",
+    "get_pipeline_logs",
+    "deploy_pipeline",
+    "search_pipeline",
+    "create_pipeline",
+    "update_pipeline",
+    "validate_pipeline",
+    "list_templates",
+    "get_template",
+    "search_templates",
+    "get_secret",
+    "list_secrets",
+    "list_workspaces",
+    "get_workspace",
+]

src/deepset_mcp/tools/haystack_service.py

@@ -27,11 +27,9 @@
 def extract_component_texts(*, component_def: dict[str, Any]) -> tuple[str, str]:
     """Extracts the component name and description for embedding.
 
-    Args:
-        component_def: The component definition
+    :param component_def: The component definition
 
-    Returns:
-        A tuple containing the component name and description
+    :returns: A tuple containing the component name and description
     """
     component_type = component_def["properties"]["type"]["const"]
     name = component_def.get("title", "")
@@ -170,13 +168,11 @@ async def _build_component_definition(
 async def get_component_definition(*, client: AsyncClientProtocol, component_type: str) -> ComponentDefinition | str:
     """Returns the definition of a specific Haystack component.
 
-    Args:
-        client: The API client to use
-        component_type: Fully qualified component type
-            (e.g. haystack.components.routers.conditional_router.ConditionalRouter)
+    :param client: The API client to use
+    :param component_type: Fully qualified component type
+        (e.g. haystack.components.routers.conditional_router.ConditionalRouter)
 
-    Returns:
-        ComponentDefinition model or error message string
+    :returns: ComponentDefinition model or error message string
     """
     haystack_service = client.haystack_service()
 
@@ -207,14 +203,12 @@ async def search_component_definition(
 ) -> ComponentSearchResults | str:
     """Searches for components based on name or description using semantic similarity.
 
-    Args:
-        client: The API client to use
-        query: The search query
-        model: The model to use for computing embeddings
-        top_k: Maximum number of results to return (default: 5)
+    :param client: The API client to use
+    :param query: The search query
+    :param model: The model to use for computing embeddings
+    :param top_k: Maximum number of results to return (default: 5)
 
-    Returns:
-        ComponentSearchResults model or error message string
+    :returns: ComponentSearchResults model or error message string
     """
     haystack_service = client.haystack_service()
 
@@ -273,11 +267,9 @@ async def search_component_definition(
 async def list_component_families(*, client: AsyncClientProtocol) -> ComponentFamilyList | str:
     """Lists all Haystack component families that are available on deepset.
 
-    Args:
-        client: The API client to use
+    :param client: The API client to use
 
-    Returns:
-        ComponentFamilyList model or error message string
+    :returns: ComponentFamilyList model or error message string
     """
     haystack_service = client.haystack_service()