IOStream

def print(self, *objects: Any, sep: str = " ", end: str = "\n", flush: bool = False) -> None:
    """Print data to the output stream.

    Args:
        objects (any): The data to print.
        sep (str, optional): The separator between objects. Defaults to " ".
        end (str, optional): The end of the output. Defaults to "\n".
        flush (bool, optional): Whether to flush the output. Defaults to False.
    """
    ...  # pragma: no cover

def send(self, message: BaseMessage) -> None:
    """Send data to the output stream.

    Args:
        message (BaseMessage): BaseMessage from autogen.messages.base_message
    """
    ...

def input(self, prompt: str = "", *, password: bool = False) -> str:
    """Read a line from the input stream.

    Args:
        prompt (str, optional): The prompt to display. Defaults to "".
        password (bool, optional): Whether to read a password. Defaults to False.

    Returns:
        str: The line read from the input stream.

    """
    ...  # pragma: no cover

@staticmethod
def set_global_default(stream: "IOStream") -> None:
    """Set the default input/output stream.

    Args:
        stream (IOStream): The input/output stream to set as the default.
    """
    IOStream._global_default = stream

@staticmethod
def get_global_default() -> "IOStream":
    """Get the default input/output stream.

    Returns:
        IOStream: The default input/output stream.
    """
    if IOStream._global_default is None:
        raise RuntimeError("No global default IOStream has been set")
    return IOStream._global_default

@staticmethod
def get_default() -> "IOStream":
    """Get the default input/output stream.

    Returns:
        IOStream: The default input/output stream.
    """
    iostream = IOStream._default_io_stream.get()
    if iostream is None:
        iostream = IOStream.get_global_default()
        # Set the default IOStream of the current context (thread/cooroutine)
        IOStream.set_default(iostream)
    return iostream

@staticmethod
@contextmanager
def set_default(stream: Optional["IOStream"]) -> Iterator[None]:
    """Set the default input/output stream.

    Args:
        stream (IOStream): The input/output stream to set as the default.
    """
    global _default_io_stream
    try:
        token = IOStream._default_io_stream.set(stream)
        yield
    finally:
        IOStream._default_io_stream.reset(token)

    return