1 files changed, 9 insertions, 23 deletions
diff --git a/remote/test/puppeteer/packages/puppeteer-core/src/bidi/Dialog.ts b/remote/test/puppeteer/packages/puppeteer-core/src/bidi/Dialog.ts
index ce22223461..1774a29f6b 100644
--- a/remote/test/puppeteer/packages/puppeteer-core/src/bidi/Dialog.ts
+++ b/remote/test/puppeteer/packages/puppeteer-core/src/bidi/Dialog.ts
@@ -4,40 +4,26 @@
  * SPDX-License-Identifier: Apache-2.0
  */
 
-import type * as Bidi from 'chromium-bidi/lib/cjs/protocol/protocol.js';
-
 import {Dialog} from '../api/Dialog.js';
 
-import type {BrowsingContext} from './BrowsingContext.js';
+import type {UserPrompt} from './core/UserPrompt.js';
 
-/**
- * @internal
- */
 export class BidiDialog extends Dialog {
-  #context: BrowsingContext;
+  static from(prompt: UserPrompt): BidiDialog {
+    return new BidiDialog(prompt);
+  }
 
-  /**
-   * @internal
-   */
-  constructor(
-    context: BrowsingContext,
-    type: Bidi.BrowsingContext.UserPromptOpenedParameters['type'],
-    message: string,
-    defaultValue?: string
-  ) {
-    super(type, message, defaultValue);
-    this.#context = context;
+  #prompt: UserPrompt;
+  private constructor(prompt: UserPrompt) {
+    super(prompt.info.type, prompt.info.message, prompt.info.defaultValue);
+    this.#prompt = prompt;
   }
 
-  /**
-   * @internal
-   */
   override async handle(options: {
     accept: boolean;
     text?: string;
   }): Promise<void> {
-    await this.#context.connection.send('browsingContext.handleUserPrompt', {
-      context: this.#context.id,
+    await this.#prompt.handle({
       accept: options.accept,
       userText: options.text,
     });