dataiku_mcp/__init__.py

# Copyright 2026 Dataiku SAS
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

"""Dataiku MCP server package.

Exposes Dataiku operations through FastMCP tools.
"""

from .server import mcp, run_http_server, run_stdio_server

# Import all modules to register tools and resources
from .tools import (  # noqa: F401
    agent_reviews,
    agents,
    cobuild,
    code_environments,
    connections,
    container_exec,
    cross_project_sharing,
    dashboards,
    data_collections,
    data_quality,
    datasets,
    evaluation_stores,
    flow,
    general_settings,
    groups,
    insights,
    instances,
    jobs,
    licensing,
    llms_and_knowledge_banks,
    managed_folders,
    plugins,
    project_folders,
    project_libraries,
    projects,
    recipes,
    scenarios,
    semantic_models,
    users,
    webapps,
    wikis,
)
from .tools.machine_learning import (  # noqa: F401
    analyses,
    saved_models,
)


__all__ = ["mcp", "run_stdio_server", "run_http_server"]