~h?ddlmZddlZddlZddlZddlZddlmZmZddl m Z ddl m Z m Z mZddlZddlmZddlmZe rddlmZe dd Zdd Ze d d  dd Zy)) annotationsN)Callable Generator)contextmanager) TYPE_CHECKINGAnyLiteral)settings)find_available_port)FastMCPc +Ktjtj} |j D]5\}}t t|st d|dtt||7d|D]'}t t|stt|||)y#|D]'}t t|stt|||)wxYww)a Temporarily override ControlFlow setting values. Args: **kwargs: The settings to override, including nested settings. Example: Temporarily override a setting: ```python import fastmcp from fastmcp.utilities.tests import temporary_settings with temporary_settings(log_level='DEBUG'): assert fastmcp.settings.log_level == 'DEBUG' assert fastmcp.settings.log_level == 'INFO' ``` zSetting z does not exist.N)copydeepcopyr model_dumpitemshasattrAttributeErrorsetattr)kwargs old_settingsattrvalues S/opt/mcp/mcp-sentiment/venv/lib/python3.12/site-packages/fastmcp/utilities/tests.pytemporary_settingsrs&==!4!4!67L <!<<> +KD%8T*$xv5E%FGG HdE * +  |j|jd|j r tdyy#1swYnxYwnD#t$r8|dkrtjd ntjd |d z }YnwxYw||ks|j rDw)a] Context manager that runs a FastMCP server in a separate process and returns the server URL. When the context manager is exited, the server process is killed. Args: server_fn: The function that runs a FastMCP server. FastMCP servers are not pickleable, so we need a function that creates and runs one. *args: Arguments to pass to the server function. provide_host_and_port: Whether to provide the host and port to the server function as kwargs. **kwargs: Keyword arguments to pass to the server function. Returns: The server URL. r)r!r"T)targetargsrdaemon rNg{Gz?g?zServer failed to start after z attemptszhttp://:)timeoutz2Server process failed to terminate even after kill)r multiprocessingProcessstartis_alivesocketAF_INET SOCK_STREAMconnectConnectionRefusedErrortimesleep RuntimeError terminatejoinkill) server_fnr/r2rr!r"proc max_attemptsattemptss rrun_server_in_processrOIs* D  D4..  " "tF4 D JJLLG L T]]_ v~~v/A/AB a 4,'  :<. RSS D64& !!NNIIaI }}  !  ==?ST T  !   & { 4  3 qLG   L T]]_sOAF 2ED6#E+B F 6D?;EF >FF FF  F )rr)r+r rzLiteral['sse']r"intreturnNone)rJzCallable[..., None]r/boolrQzGenerator[str, None, None]) __future__rrr;r?rDcollections.abcrr contextlibrtypingrrr r'fastmcpr fastmcp.utilities.httpr fastmcp.server.serverr rr-rOr.rr\s"  /%..6- <