feat: add draft delete command
This commit is contained in:
@@ -79,6 +79,16 @@ class RpcDraftClientMixin:
|
||||
{"workspace_id": workspace_id},
|
||||
)
|
||||
|
||||
async def delete_draft_workspace(
|
||||
self: RpcCaller,
|
||||
*,
|
||||
workspace_id: str,
|
||||
) -> dict[str, Any]:
|
||||
return await self._call(
|
||||
"workflow.draft_workspaces.delete",
|
||||
{"workspace_id": workspace_id},
|
||||
)
|
||||
|
||||
async def create_artifact_from_workspace(
|
||||
self: RpcCaller,
|
||||
*,
|
||||
|
||||
@@ -11,6 +11,7 @@ from ..models import (
|
||||
CreateArtifactFromWorkspaceParams,
|
||||
CreateDraftFromCapabilityParams,
|
||||
CreateWrapperFromWorkspaceParams,
|
||||
DeleteDraftWorkspaceParams,
|
||||
GetDraftWorkspaceParams,
|
||||
ListDraftWorkspacesParams,
|
||||
PatchDraftParams,
|
||||
@@ -117,6 +118,19 @@ def register_methods(
|
||||
except (ValueError, KeyError, LookupError, FileNotFoundError) as exc:
|
||||
raise_workflow_rpc_error(exc)
|
||||
|
||||
@entrypoint.method(
|
||||
name="workflow.draft_workspaces.delete", errors=[WorkflowRpcError]
|
||||
)
|
||||
async def workflow_draft_workspaces_delete(
|
||||
params: DeleteDraftWorkspaceParams = RpcParams(),
|
||||
) -> dict[str, Any]:
|
||||
try:
|
||||
return await server.api.delete_draft_workspace(
|
||||
workspace_id=params.workspace_id,
|
||||
)
|
||||
except (ValueError, KeyError, LookupError, FileNotFoundError) as exc:
|
||||
raise_workflow_rpc_error(exc)
|
||||
|
||||
@entrypoint.method(
|
||||
name="workflow.draft_workspaces.create_artifact", errors=[WorkflowRpcError]
|
||||
)
|
||||
|
||||
@@ -111,6 +111,10 @@ class ValidateDraftWorkspaceParams(RpcParamsModel):
|
||||
workspace_id: str = Field(min_length=1)
|
||||
|
||||
|
||||
class DeleteDraftWorkspaceParams(RpcParamsModel):
|
||||
workspace_id: str = Field(min_length=1)
|
||||
|
||||
|
||||
class CreateArtifactFromWorkspaceParams(RpcParamsModel):
|
||||
workspace_id: str = Field(min_length=1)
|
||||
artifact_id: str = Field(min_length=1)
|
||||
|
||||
Reference in New Issue
Block a user