dograh/api/mcp_server/server.py

from fastmcp import FastMCP

from api.mcp_server.instructions import DOGRAH_MCP_INSTRUCTIONS
from api.mcp_server.tools.catalog import (
    list_credentials,
    list_documents,
    list_recordings,
    list_tools,
)
from api.mcp_server.tools.create_workflow import create_workflow
from api.mcp_server.tools.get_workflow_code import get_workflow_code
from api.mcp_server.tools.node_types import get_node_type, list_node_types
from api.mcp_server.tools.save_workflow import save_workflow
from api.mcp_server.tools.workflows import get_workflow, list_workflows

mcp = FastMCP("dograh", instructions=DOGRAH_MCP_INSTRUCTIONS)

for _tool in (
    create_workflow,
    get_node_type,
    get_workflow,
    get_workflow_code,
    list_credentials,
    list_documents,
    list_node_types,
    list_recordings,
    list_tools,
    list_workflows,
    save_workflow,
):
    mcp.tool(_tool)