@josephyan/qingflow-app-user-mcp 0.2.0-beta.2 → 0.2.0-beta.21

package/src/qingflow_mcp/tools/workflow_tools.py

@@ -2,6 +2,7 @@ from __future__ import annotations
 
 from mcp.server.fastmcp import FastMCP
 
+from ..backend_client import BackendRequestContext
 from ..config import DEFAULT_PROFILE
 from ..errors import QingflowApiError, raise_tool_error
 from ..json_types import JSONObject, JSONValue
@@ -70,13 +71,28 @@ class WorkflowTools(ToolBase):
     def workflow_add_node(self, *, profile: str, app_key: str, payload: JSONObject) -> JSONObject:
         self._require_app_key(app_key)
         body = self._require_dict(payload)
-        return self._request(profile, "POST", f"/app/{app_key}/auditNodes", app_key=app_key, json_body=body)
+        return self._request_with_post_fallbacks(
+            profile=profile,
+            app_key=app_key,
+            path=f"/app/{app_key}/auditNodes",
+            json_body=body,
+            alternate_paths=[f"/app/{app_key}/auditNode"],
+        )
 
     def workflow_update_node(self, *, profile: str, app_key: str, audit_node_id: int, payload: JSONObject) -> JSONObject:
         self._require_app_key(app_key)
         self._require_positive("audit_node_id", audit_node_id)
         body = self._require_dict(payload)
-        return self._request(profile, "POST", f"/app/{app_key}/auditNodes/{audit_node_id}", app_key=app_key, audit_node_id=audit_node_id, json_body=body, risk_operation="update", risk_target="workflow node configuration")
+        return self._request_with_post_fallbacks(
+            profile=profile,
+            app_key=app_key,
+            path=f"/app/{app_key}/auditNodes/{audit_node_id}",
+            json_body=body,
+            alternate_paths=[f"/app/{app_key}/auditNode/{audit_node_id}"],
+            risk_operation="update",
+            risk_target="workflow node configuration",
+            audit_node_id=audit_node_id,
+        )
 
     def workflow_delete_node(self, *, profile: str, app_key: str, payload: JSONObject) -> JSONObject:
         self._require_app_key(app_key)
@@ -110,12 +126,26 @@ class WorkflowTools(ToolBase):
     def workflow_update_global_settings(self, *, profile: str, app_key: str, payload: JSONObject) -> JSONObject:
         self._require_app_key(app_key)
         body = self._require_dict(payload)
-        return self._request(profile, "POST", f"/app/{app_key}/workflow/global/setting", app_key=app_key, json_body=body, risk_operation="update", risk_target="workflow global settings")
+        return self._request_with_post_fallbacks(
+            profile=profile,
+            app_key=app_key,
+            path=f"/app/{app_key}/workflow/global/setting",
+            json_body=body,
+            alternate_paths=[],
+            risk_operation="update",
+            risk_target="workflow global settings",
+        )
 
     def workflow_publish(self, *, profile: str, app_key: str, payload: JSONObject) -> JSONObject:
         self._require_app_key(app_key)
         body = self._require_dict(payload)
-        return self._request(profile, "POST", f"/app/{app_key}/publish", app_key=app_key, json_body=body)
+        return self._request_with_post_fallbacks(
+            profile=profile,
+            app_key=app_key,
+            path=f"/app/{app_key}/publish",
+            json_body=body,
+            alternate_paths=[],
+        )
 
     def workflow_get_future_nodes(self, *, profile: str, app_key: str, apply_id: int) -> JSONObject:
         self._require_app_key(app_key)
@@ -229,6 +259,50 @@ class WorkflowTools(ToolBase):
 
         return self._run(profile, runner)
 
+    def _request_with_post_fallbacks(
+        self,
+        *,
+        profile: str,
+        app_key: str,
+        path: str,
+        json_body: JSONObject,
+        alternate_paths: list[str],
+        risk_operation: str | None = None,
+        risk_target: str | None = None,
+        **extra: JSONValue,
+    ) -> JSONObject:
+        def runner(session_profile, context):
+            attempted_contexts = [context]
+            if context.qf_version is not None:
+                attempted_contexts.append(
+                    BackendRequestContext(
+                        base_url=context.base_url,
+                        token=context.token,
+                        ws_id=context.ws_id,
+                        qf_version=None,
+                        qf_version_source="workflow_retry_without_qf_version",
+                    )
+                )
+            paths = [path, *alternate_paths]
+            last_error: QingflowApiError | None = None
+            for call_context in attempted_contexts:
+                for candidate_path in paths:
+                    try:
+                        result = self.backend.request("POST", call_context, candidate_path, json_body=json_body)
+                        response: JSONObject = {"profile": profile, "ws_id": session_profile.selected_ws_id, "result": result}
+                        response.update(extra)
+                        if risk_operation and risk_target:
+                            return self._attach_human_review_notice(response, operation=risk_operation, target=risk_target)
+                        return response
+                    except QingflowApiError as error:
+                        last_error = error
+                        if error.http_status != 404:
+                            raise
+            assert last_error is not None
+            raise last_error
+
+        return self._run(profile, runner)
+
     def _require_app_key(self, app_key: str) -> None:
         if not app_key:
             raise_tool_error(QingflowApiError.config_error("app_key is required"))