Update create_output signature

src/prompt_toolkit/output/defaults.py

@@ -1,7 +1,7 @@
 from __future__ import annotations
 
 import sys
-from typing import TextIO, cast
+from typing import TYPE_CHECKING, TextIO, cast
 
 from prompt_toolkit.utils import (
  get_bell_environment_variable,
@@ -13,13 +13,17 @@
 from .color_depth import ColorDepth
 from .plain_text import PlainTextOutput
 
+if TYPE_CHECKING:
+ from prompt_toolkit.patch_stdout import StdoutProxy
+
+
 __all__ = [
  "create_output",
 ]
 
 
 def create_output(
- stdout: TextIO | None = None, always_prefer_tty: bool = False
+ stdout: TextIO | StdoutProxy | None = None, always_prefer_tty: bool = False
 ) -> Output:
  """
  Return an :class:`~prompt_toolkit.output.Output` instance for the command
@@ -45,7 +49,7 @@ def create_output(
  if stdout is None:
  # By default, render to stdout. If the output is piped somewhere else,
  # render to stderr.
- stdout = sys.__stdout__
+ stdout = sys.stdout
 
  if always_prefer_tty:
  for io in [sys.stdout, sys.stderr]:
@@ -57,8 +61,6 @@ def create_output(
  # If the patch_stdout context manager has been used, then sys.stdout is
  # replaced by this proxy. For prompt_toolkit applications, we want to use
  # the real stdout.
- from prompt_toolkit.patch_stdout import StdoutProxy
-
  while isinstance(stdout, StdoutProxy):
  stdout = stdout.original_stdout