mirror of
https://github.com/dograh-hq/dograh.git
synced 2026-06-07 07:55:16 +02:00
00a1a22b74
* refactor: carve out extraction panel * refactor: create spec versions for node types * refactor: create a GenericNode and remove custom nodes * feat: add python and typescript sdk * add dograh sdk * fix: fetch draft workflow definition over published one * fix: fix routes of SDKs to use code gen * chore: remove doclink dependency to reduce image size * chore: format files * chore: bump pipecat * feat: let mcp fetch archived workflows on demand * chore: fix tests * feat: add sdk documentation * chore: change banner and add badge
71 lines
2.5 KiB
Python
71 lines
2.5 KiB
Python
"""MCP tool that returns a workflow as SDK TypeScript code.
|
|
|
|
Companion to `save_workflow`: the LLM calls `get_workflow_code` to see
|
|
the current state of a workflow as editable code, mutates it, and calls
|
|
`save_workflow` with the new code. Storage stays JSON; the TS form is
|
|
an ephemeral projection for the LLM edit loop.
|
|
|
|
Selection priority: latest draft → latest published → legacy
|
|
`workflow.workflow_definition`. That matches the UI's "whichever is the
|
|
working copy" behavior so the LLM sees what a human editor would see.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import Any
|
|
|
|
from fastapi import HTTPException
|
|
|
|
from api.db import db_client
|
|
from api.mcp_server.auth import authenticate_mcp_request
|
|
from api.mcp_server.server import mcp
|
|
from api.mcp_server.tracing import traced_tool
|
|
from api.mcp_server.ts_bridge import TsBridgeError, generate_code
|
|
|
|
|
|
@mcp.tool
|
|
@traced_tool
|
|
async def get_workflow_code(workflow_id: int) -> dict[str, Any]:
|
|
"""Return the workflow as SDK TypeScript code the LLM can edit.
|
|
|
|
Output shape:
|
|
{"code": "<TS source>", "workflow_id": int, "version": "draft" | "published" | "legacy"}
|
|
|
|
The LLM edits `code`, then calls `save_workflow(workflow_id, code)`.
|
|
"""
|
|
user = await authenticate_mcp_request()
|
|
|
|
workflow = await db_client.get_workflow(
|
|
workflow_id, organization_id=user.selected_organization_id
|
|
)
|
|
if not workflow:
|
|
raise HTTPException(status_code=404, detail=f"Workflow {workflow_id} not found")
|
|
|
|
# Draft wins over published — editing a draft is the normal flow.
|
|
# `current_definition` (is_current=True) is the published row, so we
|
|
# fetch the draft explicitly. If the latest draft was just published,
|
|
# no draft row exists and we fall through to `released_definition`.
|
|
draft = await db_client.get_draft_version(workflow_id)
|
|
released = workflow.released_definition
|
|
|
|
if draft is not None and draft.workflow_json:
|
|
payload = draft.workflow_json
|
|
source = "draft"
|
|
elif released is not None and released.workflow_json:
|
|
payload = released.workflow_json
|
|
source = "published"
|
|
else:
|
|
payload = workflow.workflow_definition or {}
|
|
source = "legacy"
|
|
|
|
try:
|
|
code = await generate_code(payload, workflow_name=workflow.name or "")
|
|
except TsBridgeError as e:
|
|
raise HTTPException(status_code=500, detail=f"Failed to generate code: {e}")
|
|
|
|
return {
|
|
"workflow_id": workflow_id,
|
|
"name": workflow.name or "",
|
|
"version": source,
|
|
"code": code,
|
|
}
|