diff --git a/apps/mobile/src/features/threads/NewTaskDraftScreen.tsx b/apps/mobile/src/features/threads/NewTaskDraftScreen.tsx index e0ca8fee3249..853335ace06d 100644 --- a/apps/mobile/src/features/threads/NewTaskDraftScreen.tsx +++ b/apps/mobile/src/features/threads/NewTaskDraftScreen.tsx @@ -95,6 +95,7 @@ import { resolveProjectThreadCreationBranch } from "./projectThreadCreationValid import { useCreateProjectThread } from "./use-project-actions"; import { resolveDraftProjectSelection } from "./new-task-project-selection"; import { + applyNonGitWorkspaceFallback, resolveNewTaskBranchLabel, resolveNewTaskWorkspaceLabel, } from "./new-task-context-presentation"; @@ -870,10 +871,18 @@ export function NewTaskDraftScreen(props: { selectedEnvironmentServerConfig, draft.modelSelection ?? null, ) ?? flow.selectedModel; - const workspaceMode = draft.workspaceSelection?.mode ?? flow.workspaceMode; - const selectedBranchName = draft.workspaceSelection?.branch ?? flow.selectedBranchName; - const selectedWorktreePath = - draft.workspaceSelection?.worktreePath ?? flow.selectedWorktreePath; + const workspaceSelection = applyNonGitWorkspaceFallback({ + selection: draft.workspaceSelection, + fallback: { + mode: flow.workspaceMode, + branch: flow.selectedBranchName, + worktreePath: flow.selectedWorktreePath, + }, + isGitRepo: flow.isGitRepo, + }); + const workspaceMode = workspaceSelection.mode; + const selectedBranchName = workspaceSelection.branch; + const selectedWorktreePath = workspaceSelection.worktreePath; const startFromOrigin = draft.workspaceSelection?.startFromOrigin ?? flow.startFromOrigin; const runtimeMode = draft.runtimeMode ?? flow.runtimeMode; const interactionMode = flow.planModeEnabled @@ -1168,8 +1177,8 @@ export function NewTaskDraftScreen(props: { ); - const workspaceControls = ( - + const workspaceControls = flow.isGitRepo ? ( + {flow.submitting && environmentConnected && flow.workspaceMode === "worktree" ? ( )} - ); + ) : null; const composerDock = ( @@ -1229,7 +1238,7 @@ export function NewTaskDraftScreen(props: { /> ) : null} - {workspaceControls} + {workspaceControls}