@josephyan/qingflow-cli 0.2.0-beta.55 → 0.2.0-beta.57

package/src/qingflow_mcp/server_app_builder.py

@@ -4,12 +4,9 @@ from mcp.server.fastmcp import FastMCP
 
 from .backend_client import BackendClient
 from .config import DEFAULT_PROFILE
+from .ops.base import mcp_result_from_operation
+from .ops.context import build_operations_runtime
 from .session_store import SessionStore
-from .tools.ai_builder_tools import AiBuilderTools
-from .tools.auth_tools import AuthTools
-from .tools.feedback_tools import FeedbackTools
-from .tools.file_tools import FileTools
-from .tools.workspace_tools import WorkspaceTools
 
 
 def build_builder_server() -> FastMCP:
@@ -33,11 +30,7 @@ def build_builder_server() -> FastMCP:
     )
     sessions = SessionStore()
     backend = BackendClient()
-    auth = AuthTools(sessions, backend)
-    workspace = WorkspaceTools(sessions, backend)
-    files = FileTools(sessions, backend)
-    ai_builder = AiBuilderTools(sessions, backend)
-    feedback = FeedbackTools(backend, mcp_side="App Builder MCP")
+    runtime = build_operations_runtime(sessions=sessions, backend=backend, feedback_mcp_side="App Builder MCP")
 
     @server.tool()
     def auth_login(
@@ -48,13 +41,15 @@ def build_builder_server() -> FastMCP:
         password: str = "",
         persist: bool = True,
     ) -> dict:
-        return auth.auth_login(
-            profile=profile,
-            base_url=base_url,
-            qf_version=qf_version,
-            email=email,
-            password=password,
-            persist=persist,
+        return mcp_result_from_operation(
+            runtime.auth.login(
+                profile=profile,
+                base_url=base_url,
+                qf_version=qf_version,
+                email=email,
+                password=password,
+                persist=persist,
+            )
         )
 
     @server.tool()
@@ -66,22 +61,24 @@ def build_builder_server() -> FastMCP:
         ws_id: int | None = None,
         persist: bool = False,
     ) -> dict:
-        return auth.auth_use_token(
-            profile=profile,
-            base_url=base_url,
-            qf_version=qf_version,
-            token=token,
-            ws_id=ws_id,
-            persist=persist,
+        return mcp_result_from_operation(
+            runtime.auth.use_token(
+                profile=profile,
+                base_url=base_url,
+                qf_version=qf_version,
+                token=token,
+                ws_id=ws_id,
+                persist=persist,
+            )
         )
 
     @server.tool()
     def auth_whoami(profile: str = DEFAULT_PROFILE) -> dict:
-        return auth.auth_whoami(profile=profile)
+        return mcp_result_from_operation(runtime.auth.me(profile=profile))
 
     @server.tool()
     def auth_logout(profile: str = DEFAULT_PROFILE, forget_persisted: bool = False) -> dict:
-        return auth.auth_logout(profile=profile, forget_persisted=forget_persisted)
+        return mcp_result_from_operation(runtime.auth.logout(profile=profile, forget_persisted=forget_persisted))
 
     @server.tool()
     def workspace_list(
@@ -90,16 +87,18 @@ def build_builder_server() -> FastMCP:
         page_size: int = 20,
         include_external: bool = False,
     ) -> dict:
-        return workspace.workspace_list(
-            profile=profile,
-            page_num=page_num,
-            page_size=page_size,
-            include_external=include_external,
+        return mcp_result_from_operation(
+            runtime.workspace.list(
+                profile=profile,
+                page_num=page_num,
+                page_size=page_size,
+                include_external=include_external,
+            )
         )
 
     @server.tool()
     def workspace_select(profile: str = DEFAULT_PROFILE, ws_id: int = 0) -> dict:
-        return workspace.workspace_select(profile=profile, ws_id=ws_id)
+        return mcp_result_from_operation(runtime.workspace.use(profile=profile, ws_id=ws_id))
 
     @server.tool()
     def file_upload_local(
@@ -112,34 +111,64 @@ def build_builder_server() -> FastMCP:
         path_id: int | None = None,
         file_related_url: str | None = None,
     ) -> dict:
-        return files.file_upload_local(
-            profile=profile,
-            upload_kind=upload_kind,
-            file_path=file_path,
-            upload_mark=upload_mark,
-            content_type=content_type,
-            bucket_type=bucket_type,
-            path_id=path_id,
-            file_related_url=file_related_url,
+        return mcp_result_from_operation(
+            runtime.files.upload_local(
+                profile=profile,
+                upload_kind=upload_kind,
+                file_path=file_path,
+                upload_mark=upload_mark,
+                content_type=content_type,
+                bucket_type=bucket_type,
+                path_id=path_id,
+                file_related_url=file_related_url,
+            )
         )
 
-    feedback.register(server)
+    @server.tool()
+    def feedback_submit(
+        category: str = "",
+        title: str = "",
+        description: str = "",
+        expected_behavior: str | None = None,
+        actual_behavior: str | None = None,
+        impact_scope: str | None = None,
+        tool_name: str | None = None,
+        app_key: str | None = None,
+        record_id: str | int | None = None,
+        workflow_node_id: str | int | None = None,
+        note: str | None = None,
+    ) -> dict:
+        return mcp_result_from_operation(
+            runtime.feedback.submit(
+                category=category,
+                title=title,
+                description=description,
+                expected_behavior=expected_behavior,
+                actual_behavior=actual_behavior,
+                impact_scope=impact_scope,
+                tool_name=tool_name,
+                app_key=app_key,
+                record_id=record_id,
+                workflow_node_id=workflow_node_id,
+                note=note,
+            )
+        )
 
     @server.tool()
     def package_list(profile: str = DEFAULT_PROFILE, trial_status: str = "all") -> dict:
-        return ai_builder.package_list(profile=profile, trial_status=trial_status)
+        return mcp_result_from_operation(runtime.builder.package_list(profile=profile, trial_status=trial_status))
 
     @server.tool()
     def package_resolve(profile: str = DEFAULT_PROFILE, package_name: str = "") -> dict:
-        return ai_builder.package_resolve(profile=profile, package_name=package_name)
+        return mcp_result_from_operation(runtime.builder.package_resolve(profile=profile, package_name=package_name))
 
     @server.tool()
     def builder_tool_contract(tool_name: str = "") -> dict:
-        return ai_builder.builder_tool_contract(tool_name=tool_name)
+        return mcp_result_from_operation(runtime.builder.builder_tool_contract(tool_name=tool_name))
 
     @server.tool()
     def package_create(profile: str = DEFAULT_PROFILE, package_name: str = "") -> dict:
-        return ai_builder.package_create(profile=profile, package_name=package_name)
+        return mcp_result_from_operation(runtime.builder.package_create(profile=profile, package_name=package_name))
 
     @server.tool()
     def member_search(
@@ -149,12 +178,14 @@ def build_builder_server() -> FastMCP:
         page_size: int = 20,
         contain_disable: bool = False,
     ) -> dict:
-        return ai_builder.member_search(
-            profile=profile,
-            query=query,
-            page_num=page_num,
-            page_size=page_size,
-            contain_disable=contain_disable,
+        return mcp_result_from_operation(
+            runtime.builder.member_search(
+                profile=profile,
+                query=query,
+                page_num=page_num,
+                page_size=page_size,
+                contain_disable=contain_disable,
+            )
         )
 
     @server.tool()
@@ -164,7 +195,9 @@ def build_builder_server() -> FastMCP:
         page_num: int = 1,
         page_size: int = 20,
     ) -> dict:
-        return ai_builder.role_search(profile=profile, keyword=keyword, page_num=page_num, page_size=page_size)
+        return mcp_result_from_operation(
+            runtime.builder.role_search(profile=profile, keyword=keyword, page_num=page_num, page_size=page_size)
+        )
 
     @server.tool()
     def role_create(
@@ -175,13 +208,15 @@ def build_builder_server() -> FastMCP:
         member_names: list[str] | None = None,
         role_icon: str = "ex-user-outlined",
     ) -> dict:
-        return ai_builder.role_create(
-            profile=profile,
-            role_name=role_name,
-            member_uids=member_uids or [],
-            member_emails=member_emails or [],
-            member_names=member_names or [],
-            role_icon=role_icon,
+        return mcp_result_from_operation(
+            runtime.builder.role_create(
+                profile=profile,
+                role_name=role_name,
+                member_uids=member_uids or [],
+                member_emails=member_emails or [],
+                member_names=member_names or [],
+                role_icon=role_icon,
+            )
         )
 
     @server.tool()
@@ -191,7 +226,9 @@ def build_builder_server() -> FastMCP:
         app_key: str = "",
         app_title: str = "",
     ) -> dict:
-        return ai_builder.package_attach_app(profile=profile, tag_id=tag_id, app_key=app_key, app_title=app_title)
+        return mcp_result_from_operation(
+            runtime.builder.package_attach_app(profile=profile, tag_id=tag_id, app_key=app_key, app_title=app_title)
+        )
 
     @server.tool()
     def app_release_edit_lock_if_mine(
@@ -200,11 +237,13 @@ def build_builder_server() -> FastMCP:
         lock_owner_email: str = "",
         lock_owner_name: str = "",
     ) -> dict:
-        return ai_builder.app_release_edit_lock_if_mine(
-            profile=profile,
-            app_key=app_key,
-            lock_owner_email=lock_owner_email,
-            lock_owner_name=lock_owner_name,
+        return mcp_result_from_operation(
+            runtime.builder.release_edit_lock_if_mine(
+                profile=profile,
+                app_key=app_key,
+                lock_owner_email=lock_owner_email,
+                lock_owner_name=lock_owner_name,
+            )
         )
 
     @server.tool()
@@ -214,31 +253,38 @@ def build_builder_server() -> FastMCP:
         app_name: str = "",
         package_tag_id: int | None = None,
     ) -> dict:
-        return ai_builder.app_resolve(profile=profile, app_key=app_key, app_name=app_name, package_tag_id=package_tag_id)
+        return mcp_result_from_operation(
+            runtime.builder.app_resolve(
+                profile=profile,
+                app_key=app_key,
+                app_name=app_name,
+                package_tag_id=package_tag_id,
+            )
+        )
 
     @server.tool()
     def app_read_summary(profile: str = DEFAULT_PROFILE, app_key: str = "") -> dict:
-        return ai_builder.app_read_summary(profile=profile, app_key=app_key)
+        return mcp_result_from_operation(runtime.builder.app_show(profile=profile, app_key=app_key))
 
     @server.tool()
     def app_read_fields(profile: str = DEFAULT_PROFILE, app_key: str = "") -> dict:
-        return ai_builder.app_read_fields(profile=profile, app_key=app_key)
+        return mcp_result_from_operation(runtime.builder.fields_show(profile=profile, app_key=app_key))
 
     @server.tool()
     def app_read_layout_summary(profile: str = DEFAULT_PROFILE, app_key: str = "") -> dict:
-        return ai_builder.app_read_layout_summary(profile=profile, app_key=app_key)
+        return mcp_result_from_operation(runtime.builder.layout_show(profile=profile, app_key=app_key))
 
     @server.tool()
     def app_read_views_summary(profile: str = DEFAULT_PROFILE, app_key: str = "") -> dict:
-        return ai_builder.app_read_views_summary(profile=profile, app_key=app_key)
+        return mcp_result_from_operation(runtime.builder.views_show(profile=profile, app_key=app_key))
 
     @server.tool()
     def app_read_flow_summary(profile: str = DEFAULT_PROFILE, app_key: str = "") -> dict:
-        return ai_builder.app_read_flow_summary(profile=profile, app_key=app_key)
+        return mcp_result_from_operation(runtime.builder.flow_show(profile=profile, app_key=app_key))
 
     @server.tool()
     def app_read_charts_summary(profile: str = DEFAULT_PROFILE, app_key: str = "") -> dict:
-        return ai_builder.app_read_charts_summary(profile=profile, app_key=app_key)
+        return mcp_result_from_operation(runtime.builder.charts_show(profile=profile, app_key=app_key))
 
     @server.tool()
     def portal_read_summary(
@@ -246,7 +292,9 @@ def build_builder_server() -> FastMCP:
         dash_key: str = "",
         being_draft: bool = True,
     ) -> dict:
-        return ai_builder.portal_read_summary(profile=profile, dash_key=dash_key, being_draft=being_draft)
+        return mcp_result_from_operation(
+            runtime.builder.portal_show(profile=profile, dash_key=dash_key, being_draft=being_draft)
+        )
 
     @server.tool()
     def app_schema_apply(
@@ -261,17 +309,19 @@ def build_builder_server() -> FastMCP:
         update_fields: list[dict] | None = None,
         remove_fields: list[dict] | None = None,
     ) -> dict:
-        return ai_builder.app_schema_apply(
-            profile=profile,
-            app_key=app_key,
-            package_tag_id=package_tag_id,
-            app_name=app_name,
-            app_title=app_title,
-            create_if_missing=create_if_missing,
-            publish=publish,
-            add_fields=add_fields or [],
-            update_fields=update_fields or [],
-            remove_fields=remove_fields or [],
+        return mcp_result_from_operation(
+            runtime.builder.fields_apply(
+                profile=profile,
+                app_key=app_key,
+                package_tag_id=package_tag_id,
+                app_name=app_name,
+                app_title=app_title,
+                create_if_missing=create_if_missing,
+                publish=publish,
+                add_fields=add_fields or [],
+                update_fields=update_fields or [],
+                remove_fields=remove_fields or [],
+            )
         )
 
     @server.tool()
@@ -282,7 +332,15 @@ def build_builder_server() -> FastMCP:
         publish: bool = True,
         sections: list[dict] | None = None,
     ) -> dict:
-        return ai_builder.app_layout_apply(profile=profile, app_key=app_key, mode=mode, publish=publish, sections=sections or [])
+        return mcp_result_from_operation(
+            runtime.builder.layout_apply(
+                profile=profile,
+                app_key=app_key,
+                mode=mode,
+                publish=publish,
+                sections=sections or [],
+            )
+        )
 
     @server.tool()
     def app_flow_apply(
@@ -293,13 +351,15 @@ def build_builder_server() -> FastMCP:
         nodes: list[dict] | None = None,
         transitions: list[dict] | None = None,
     ) -> dict:
-        return ai_builder.app_flow_apply(
-            profile=profile,
-            app_key=app_key,
-            mode=mode,
-            publish=publish,
-            nodes=nodes or [],
-            transitions=transitions or [],
+        return mcp_result_from_operation(
+            runtime.builder.flow_apply(
+                profile=profile,
+                app_key=app_key,
+                mode=mode,
+                publish=publish,
+                nodes=nodes or [],
+                transitions=transitions or [],
+            )
         )
 
     @server.tool()
@@ -310,12 +370,14 @@ def build_builder_server() -> FastMCP:
         upsert_views: list[dict] | None = None,
         remove_views: list[str] | None = None,
     ) -> dict:
-        return ai_builder.app_views_apply(
-            profile=profile,
-            app_key=app_key,
-            publish=publish,
-            upsert_views=upsert_views or [],
-            remove_views=remove_views or [],
+        return mcp_result_from_operation(
+            runtime.builder.views_apply(
+                profile=profile,
+                app_key=app_key,
+                publish=publish,
+                upsert_views=upsert_views or [],
+                remove_views=remove_views or [],
+            )
         )
 
     @server.tool()
@@ -326,12 +388,14 @@ def build_builder_server() -> FastMCP:
         remove_chart_ids: list[str] | None = None,
         reorder_chart_ids: list[str] | None = None,
     ) -> dict:
-        return ai_builder.app_charts_apply(
-            profile=profile,
-            app_key=app_key,
-            upsert_charts=upsert_charts or [],
-            remove_chart_ids=remove_chart_ids or [],
-            reorder_chart_ids=reorder_chart_ids or [],
+        return mcp_result_from_operation(
+            runtime.builder.charts_apply(
+                profile=profile,
+                app_key=app_key,
+                upsert_charts=upsert_charts or [],
+                remove_chart_ids=remove_chart_ids or [],
+                reorder_chart_ids=reorder_chart_ids or [],
+            )
         )
 
     @server.tool()
@@ -349,19 +413,21 @@ def build_builder_server() -> FastMCP:
         dash_global_config: dict | None = None,
         config: dict | None = None,
     ) -> dict:
-        return ai_builder.portal_apply(
-            profile=profile,
-            dash_key=dash_key,
-            dash_name=dash_name,
-            package_tag_id=package_tag_id,
-            publish=publish,
-            sections=sections or [],
-            auth=auth,
-            icon=icon,
-            color=color,
-            hide_copyright=hide_copyright,
-            dash_global_config=dash_global_config,
-            config=config or {},
+        return mcp_result_from_operation(
+            runtime.builder.portal_apply(
+                profile=profile,
+                dash_key=dash_key,
+                dash_name=dash_name,
+                package_tag_id=package_tag_id,
+                publish=publish,
+                sections=sections or [],
+                auth=auth,
+                icon=icon,
+                color=color,
+                hide_copyright=hide_copyright,
+                dash_global_config=dash_global_config,
+                config=config or {},
+            )
         )
 
     @server.tool()
@@ -370,10 +436,12 @@ def build_builder_server() -> FastMCP:
         app_key: str = "",
         expected_package_tag_id: int | None = None,
     ) -> dict:
-        return ai_builder.app_publish_verify(
-            profile=profile,
-            app_key=app_key,
-            expected_package_tag_id=expected_package_tag_id,
+        return mcp_result_from_operation(
+            runtime.builder.publish_verify(
+                profile=profile,
+                app_key=app_key,
+                expected_package_tag_id=expected_package_tag_id,
+            )
         )
 
     return server