add heavy loggin
This commit is contained in:
@@ -34,10 +34,16 @@ def _encode_image(image: np.ndarray) -> str:
|
||||
class InferenceClient:
|
||||
"""HTTP client for the GPU inference server."""
|
||||
|
||||
def __init__(self, base_url: str | None = None, timeout: float = 60.0):
|
||||
def __init__(self, base_url: str | None = None, timeout: float = 60.0,
|
||||
job_id: str = "", log_level: str = "INFO"):
|
||||
self.base_url = (base_url or DEFAULT_URL).rstrip("/")
|
||||
self.timeout = timeout
|
||||
self.job_id = job_id
|
||||
self.log_level = log_level
|
||||
self.session = requests.Session()
|
||||
if job_id:
|
||||
self.session.headers["X-Job-Id"] = job_id
|
||||
self.session.headers["X-Log-Level"] = log_level
|
||||
|
||||
def health(self) -> ServerStatus:
|
||||
"""Check server health and loaded models."""
|
||||
|
||||
Reference in New Issue
Block a user