feat: add create workflow tool in MCP

This commit is contained in:
Abhishek Kumar 2026-04-25 17:38:38 +05:30
parent d6567eef28
commit 3e3773f400
13 changed files with 245 additions and 94 deletions

View file

@ -1,13 +1,31 @@
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)
from api.mcp_server.tools import catalog as _catalog # noqa: E402, F401
from api.mcp_server.tools import (
get_workflow_code as _get_workflow_code, # noqa: E402, F401
)
from api.mcp_server.tools import node_types as _node_types # noqa: E402, F401
from api.mcp_server.tools import save_workflow as _save_workflow # noqa: E402, F401
from api.mcp_server.tools import workflows as _workflows # noqa: E402, F401
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)