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

package/src/qingflow_mcp/server_app_builder.py

@@ -4,9 +4,12 @@ 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:
@@ -30,7 +33,11 @@ def build_builder_server() -> FastMCP:
     )
     sessions = SessionStore()
     backend = BackendClient()
-    runtime = build_operations_runtime(sessions=sessions, backend=backend, feedback_mcp_side="App Builder MCP")
+    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")
 
     @server.tool()
     def auth_login(
@@ -41,15 +48,13 @@ def build_builder_server() -> FastMCP:
         password: str = "",
         persist: bool = True,
     ) -> dict:
-        return mcp_result_from_operation(
-            runtime.auth.login(
-                profile=profile,
-                base_url=base_url,
-                qf_version=qf_version,
-                email=email,
-                password=password,
-                persist=persist,
-            )
+        return auth.auth_login(
+            profile=profile,
+            base_url=base_url,
+            qf_version=qf_version,
+            email=email,
+            password=password,
+            persist=persist,
         )
 
     @server.tool()
@@ -61,24 +66,22 @@ def build_builder_server() -> FastMCP:
         ws_id: int | None = None,
         persist: bool = False,
     ) -> dict:
-        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,
-            )
+        return auth.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 mcp_result_from_operation(runtime.auth.me(profile=profile))
+        return auth.auth_whoami(profile=profile)
 
     @server.tool()
     def auth_logout(profile: str = DEFAULT_PROFILE, forget_persisted: bool = False) -> dict:
-        return mcp_result_from_operation(runtime.auth.logout(profile=profile, forget_persisted=forget_persisted))
+        return auth.auth_logout(profile=profile, forget_persisted=forget_persisted)
 
     @server.tool()
     def workspace_list(
@@ -87,18 +90,16 @@ def build_builder_server() -> FastMCP:
         page_size: int = 20,
         include_external: bool = False,
     ) -> dict:
-        return mcp_result_from_operation(
-            runtime.workspace.list(
-                profile=profile,
-                page_num=page_num,
-                page_size=page_size,
-                include_external=include_external,
-            )
+        return workspace.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 mcp_result_from_operation(runtime.workspace.use(profile=profile, ws_id=ws_id))
+        return workspace.workspace_select(profile=profile, ws_id=ws_id)
 
     @server.tool()
     def file_upload_local(
@@ -111,64 +112,34 @@ def build_builder_server() -> FastMCP:
         path_id: int | None = None,
         file_related_url: str | None = None,
     ) -> dict:
-        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,
-            )
+        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,
         )
 
-    @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,
-            )
-        )
+    feedback.register(server)
 
     @server.tool()
     def package_list(profile: str = DEFAULT_PROFILE, trial_status: str = "all") -> dict:
-        return mcp_result_from_operation(runtime.builder.package_list(profile=profile, trial_status=trial_status))
+        return ai_builder.package_list(profile=profile, trial_status=trial_status)
 
     @server.tool()
     def package_resolve(profile: str = DEFAULT_PROFILE, package_name: str = "") -> dict:
-        return mcp_result_from_operation(runtime.builder.package_resolve(profile=profile, package_name=package_name))
+        return ai_builder.package_resolve(profile=profile, package_name=package_name)
 
     @server.tool()
     def builder_tool_contract(tool_name: str = "") -> dict:
-        return mcp_result_from_operation(runtime.builder.builder_tool_contract(tool_name=tool_name))
+        return ai_builder.builder_tool_contract(tool_name=tool_name)
 
     @server.tool()
     def package_create(profile: str = DEFAULT_PROFILE, package_name: str = "") -> dict:
-        return mcp_result_from_operation(runtime.builder.package_create(profile=profile, package_name=package_name))
+        return ai_builder.package_create(profile=profile, package_name=package_name)
 
     @server.tool()
     def member_search(
@@ -178,14 +149,12 @@ def build_builder_server() -> FastMCP:
         page_size: int = 20,
         contain_disable: bool = False,
     ) -> dict:
-        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,
-            )
+        return ai_builder.member_search(
+            profile=profile,
+            query=query,
+            page_num=page_num,
+            page_size=page_size,
+            contain_disable=contain_disable,
         )
 
     @server.tool()
@@ -195,9 +164,7 @@ def build_builder_server() -> FastMCP:
         page_num: int = 1,
         page_size: int = 20,
     ) -> dict:
-        return mcp_result_from_operation(
-            runtime.builder.role_search(profile=profile, keyword=keyword, page_num=page_num, page_size=page_size)
-        )
+        return ai_builder.role_search(profile=profile, keyword=keyword, page_num=page_num, page_size=page_size)
 
     @server.tool()
     def role_create(
@@ -208,15 +175,13 @@ def build_builder_server() -> FastMCP:
         member_names: list[str] | None = None,
         role_icon: str = "ex-user-outlined",
     ) -> dict:
-        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,
-            )
+        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,
         )
 
     @server.tool()
@@ -226,9 +191,7 @@ def build_builder_server() -> FastMCP:
         app_key: str = "",
         app_title: str = "",
     ) -> dict:
-        return mcp_result_from_operation(
-            runtime.builder.package_attach_app(profile=profile, tag_id=tag_id, app_key=app_key, app_title=app_title)
-        )
+        return ai_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(
@@ -237,13 +200,11 @@ def build_builder_server() -> FastMCP:
         lock_owner_email: str = "",
         lock_owner_name: str = "",
     ) -> dict:
-        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,
-            )
+        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,
         )
 
     @server.tool()
@@ -253,38 +214,31 @@ def build_builder_server() -> FastMCP:
         app_name: str = "",
         package_tag_id: int | None = None,
     ) -> dict:
-        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,
-            )
-        )
+        return ai_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 mcp_result_from_operation(runtime.builder.app_show(profile=profile, app_key=app_key))
+        return ai_builder.app_read_summary(profile=profile, app_key=app_key)
 
     @server.tool()
     def app_read_fields(profile: str = DEFAULT_PROFILE, app_key: str = "") -> dict:
-        return mcp_result_from_operation(runtime.builder.fields_show(profile=profile, app_key=app_key))
+        return ai_builder.app_read_fields(profile=profile, app_key=app_key)
 
     @server.tool()
     def app_read_layout_summary(profile: str = DEFAULT_PROFILE, app_key: str = "") -> dict:
-        return mcp_result_from_operation(runtime.builder.layout_show(profile=profile, app_key=app_key))
+        return ai_builder.app_read_layout_summary(profile=profile, app_key=app_key)
 
     @server.tool()
     def app_read_views_summary(profile: str = DEFAULT_PROFILE, app_key: str = "") -> dict:
-        return mcp_result_from_operation(runtime.builder.views_show(profile=profile, app_key=app_key))
+        return ai_builder.app_read_views_summary(profile=profile, app_key=app_key)
 
     @server.tool()
     def app_read_flow_summary(profile: str = DEFAULT_PROFILE, app_key: str = "") -> dict:
-        return mcp_result_from_operation(runtime.builder.flow_show(profile=profile, app_key=app_key))
+        return ai_builder.app_read_flow_summary(profile=profile, app_key=app_key)
 
     @server.tool()
     def app_read_charts_summary(profile: str = DEFAULT_PROFILE, app_key: str = "") -> dict:
-        return mcp_result_from_operation(runtime.builder.charts_show(profile=profile, app_key=app_key))
+        return ai_builder.app_read_charts_summary(profile=profile, app_key=app_key)
 
     @server.tool()
     def portal_read_summary(
@@ -292,9 +246,7 @@ def build_builder_server() -> FastMCP:
         dash_key: str = "",
         being_draft: bool = True,
     ) -> dict:
-        return mcp_result_from_operation(
-            runtime.builder.portal_show(profile=profile, dash_key=dash_key, being_draft=being_draft)
-        )
+        return ai_builder.portal_read_summary(profile=profile, dash_key=dash_key, being_draft=being_draft)
 
     @server.tool()
     def app_schema_apply(
@@ -309,19 +261,17 @@ def build_builder_server() -> FastMCP:
         update_fields: list[dict] | None = None,
         remove_fields: list[dict] | None = None,
     ) -> dict:
-        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 [],
-            )
+        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 [],
         )
 
     @server.tool()
@@ -332,15 +282,7 @@ def build_builder_server() -> FastMCP:
         publish: bool = True,
         sections: list[dict] | None = None,
     ) -> dict:
-        return mcp_result_from_operation(
-            runtime.builder.layout_apply(
-                profile=profile,
-                app_key=app_key,
-                mode=mode,
-                publish=publish,
-                sections=sections or [],
-            )
-        )
+        return ai_builder.app_layout_apply(profile=profile, app_key=app_key, mode=mode, publish=publish, sections=sections or [])
 
     @server.tool()
     def app_flow_apply(
@@ -351,15 +293,13 @@ def build_builder_server() -> FastMCP:
         nodes: list[dict] | None = None,
         transitions: list[dict] | None = None,
     ) -> dict:
-        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 [],
-            )
+        return ai_builder.app_flow_apply(
+            profile=profile,
+            app_key=app_key,
+            mode=mode,
+            publish=publish,
+            nodes=nodes or [],
+            transitions=transitions or [],
         )
 
     @server.tool()
@@ -370,14 +310,12 @@ def build_builder_server() -> FastMCP:
         upsert_views: list[dict] | None = None,
         remove_views: list[str] | None = None,
     ) -> dict:
-        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 [],
-            )
+        return ai_builder.app_views_apply(
+            profile=profile,
+            app_key=app_key,
+            publish=publish,
+            upsert_views=upsert_views or [],
+            remove_views=remove_views or [],
         )
 
     @server.tool()
@@ -388,14 +326,12 @@ def build_builder_server() -> FastMCP:
         remove_chart_ids: list[str] | None = None,
         reorder_chart_ids: list[str] | None = None,
     ) -> dict:
-        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 [],
-            )
+        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 [],
         )
 
     @server.tool()
@@ -413,21 +349,19 @@ def build_builder_server() -> FastMCP:
         dash_global_config: dict | None = None,
         config: dict | None = None,
     ) -> dict:
-        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 {},
-            )
+        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 {},
         )
 
     @server.tool()
@@ -436,12 +370,10 @@ def build_builder_server() -> FastMCP:
         app_key: str = "",
         expected_package_tag_id: int | None = None,
     ) -> dict:
-        return mcp_result_from_operation(
-            runtime.builder.publish_verify(
-                profile=profile,
-                app_key=app_key,
-                expected_package_tag_id=expected_package_tag_id,
-            )
+        return ai_builder.app_publish_verify(
+            profile=profile,
+            app_key=app_key,
+            expected_package_tag_id=expected_package_tag_id,
         )
 
     return server