Function ModifyPromptDialog

site/src/pages/TaskPage/ModifyPromptDialog.tsx:30–158

({
	task,
	workspace,
	open,
	onOpenChange,
})

};

export const ModifyPromptDialog: FC<ModifyPromptDialogProps> = ({
	task,
	workspace,
	open,
	onOpenChange,
}) => {
	const formId = useId();
	const formik = useFormik({
		initialValues: {
			prompt: task.initial_prompt,
		},
		onSubmit: (values) => {
			updatePromptMutation.mutate(values.prompt);
		},
	});

	const queryClient = useQueryClient();

	const buildParametersQuery = useQuery(
		workspaceBuildParameters(workspace.latest_build.id),
	);

	const updatePromptMutation = useMutation({
		mutationFn: async (prompt: string) => {
			const currentBuild = await API.getWorkspaceBuildByNumber(
				workspace.owner_name,
				workspace.name,
				workspace.latest_build.build_number,
			);

			if (currentBuild.status !== "stopped") {
				if (!currentBuild.job.completed_at) {
					await API.cancelWorkspaceBuild(currentBuild.id);
					await API.waitForBuild(currentBuild);
				}

				const stopBuild = await API.stopWorkspace(workspace.id);
				await API.waitForBuild(stopBuild);
			}

			await API.updateTaskInput(task.owner_name, task.id, prompt);
			await API.startWorkspace(
				workspace.id,
				task.template_version_id,
				undefined,
				buildParametersQuery.data,
			);
		},
		onSuccess: () => {
			queryClient.invalidateQueries({
				queryKey: ["tasks", task.owner_name, task.id],
			});
			queryClient.invalidateQueries({
				queryKey: workspaceByOwnerAndNameKey(
					workspace.owner_name,
					workspace.name,
				),
			});