@@ -368,14 +368,8 @@ def initialize_telemetry_client(
368368 try:
369369
370370 with TelemetryClientFactory._lock:
371-
372371 TelemetryClientFactory._initialize()
373372 if session_id_hex not in TelemetryClientFactory._clients:
374- logger.debug(
375- "Session ID not in clients: %s",
376- session_id_hex,
377- flush=True,
378- )
379373 logger.debug(
380374 "Creating new TelemetryClient for connection %s",
381375 session_id_hex,
@@ -391,13 +385,9 @@ def initialize_telemetry_client(
391385 executor=TelemetryClientFactory._executor,
392386 )
393387 else:
394- logger.debug("Telemetry disabled: %s", session_id_hex)
395388 TelemetryClientFactory._clients[
396389 session_id_hex
397390 ] = NoopTelemetryClient()
398- logger.debug(
399- "Noop Telemetry client initialized: %s", session_id_hex
400- )
401391
402392 except Exception as e:
403393 logger.debug("Failed to initialize telemetry client: %s", e)
@@ -423,6 +413,7 @@ def get_telemetry_client(session_id_hex):
423413 @staticmethod
424414 def close(session_id_hex):
425415 """Close and remove the telemetry client for a specific connection"""
416+
426417 with TelemetryClientFactory._lock:
427418 if (
428419 telemetry_client := TelemetryClientFactory._clients.pop(
@@ -432,7 +423,6 @@ def close(session_id_hex):
432423 logger.debug(
433424 "Removing telemetry client for connection %s", session_id_hex
434425 )
435- telemetry_client = TelemetryClientFactory._clients.pop(session_id_hex)
436426 telemetry_client.close()
437427
438428 # Shutdown executor if no more clients
0 commit comments