j ddlZddlmZmZddlZddlmZmZeje Z GddeZ GddZ y)N) NamedTupleOptional) httpx_client url_helpersc(eZdZUeed<eeed<y) ProbeResult is_healthy error_messageN)__name__ __module__ __qualname__bool__annotations__rstrz/Users/manta/Documents/Projects/TheRoad-I1/backend/.venv/lib/python3.12/site-packages/opik/healthcheck/connection_probe.pyrr sC= rrcDeZdZdZdedej fdZdede fdZ y) ConnectionProbea Provides functionality to probe and verify the connection to a remote server. This class is designed to check the availability and responsiveness of a server by performing a lightweight connection test to a specific endpoint. It encapsulates the logic to send a request and interpret its response, ensuring it is simple to evaluate the connection status programmatically. base_urlclientcFtj||_||_y)aB Initializes an instance with a base URL and an HTTP client. This constructor sets up the necessary URLs and initializes the client needed to make requests. Args: base_url: The base URL for API requests. client: The HTTP client instance to make API calls. N)rget_is_alive_ping_url _ping_url_client)selfrrs r__init__zConnectionProbe.__init__s%::8D rtimeoutreturnc |jj|j|}|jdk(r t ddSt dd|jS#t j t jf$r}t dd|cYd}~Sd}~wt$r2}tjd |d t dd |cYd}~Sd}~wwxYw) a Checks the connection to a specified server by sending a ping request and evaluates the response. This method performs a basic health check by sending an HTTP GET request to the server's ping URL. If the response status code is 200, it indicates the server is healthy, otherwise it is considered unhealthy. Handles timeouts and other unexpected exceptions during the connection attempt. Args: timeout: The maximum duration, in seconds, to wait for the server's response before timing out. Returns: An object indicating whether the server is healthy and, if not, the associated error message. )rTN)r r FzUnexpected status code: zConnection error: z8Unexpected error while checking connection to server: %s)exc_infozUnexpected error: ) rgetr status_coderhttpx ConnectErrorTimeoutException ExceptionLOGGER exception)rrresponsees rcheck_connectionz ConnectionProbe.check_connection)s Y||'''HH##s*"d$GG"$$r5s=' *   8 $!*! 7Y7Yr