Warn user before trying to add a stage to their pipeline

extension/src/pipeline/index.ts

@@ -17,6 +17,8 @@ import { getInput, getValidInput } from '../vscode/inputBox'
 import { Title } from '../vscode/title'
 import { quickPickOne, quickPickOneOrInput } from '../vscode/quickPick'
 import { pickFile } from '../vscode/resourcePicker'
+import { Modal } from '../vscode/modal'
+import { Response } from '../vscode/response'
 
 export enum ScriptCommand {
   JUPYTER = 'jupyter nbconvert --to notebook --inplace --execute',
@@ -150,6 +152,16 @@ export class Pipeline extends DeferredDisposable {
   }
 
   private async addPipeline() {
+    const response = await Modal.showInformation(
+      'To continue you must add a stage to the dvc.yaml file. Would you like to add a stage now?',
+      Response.YES,
+      Response.NO
+    )
+
+    if (response !== Response.YES) {
+      return
+    }
+
     const stageName = await this.askForStageName()
     if (!stageName) {
       return