TelemetryResponse, send_count

Signed-off-by: Sai Shree Pradhan <saishree.pradhan@databricks.com>

Author: saishreeeee

src/databricks/sql/telemetry/models/endpoint_models.py

@@ -36,3 +36,4 @@ class TelemetryResponse(JsonSerializableMixin):
     errors: List[str]
     numSuccess: int
     numProtoSuccess: int
+    numRealtimeSuccess: int

src/databricks/sql/telemetry/telemetry_client.py

@@ -207,7 +207,9 @@ def _send_telemetry(self, events):
                 headers=headers,
                 timeout=900,
             )
-            future.add_done_callback(self._telemetry_request_callback, sent_count)
+            future.add_done_callback(
+                lambda fut: self._telemetry_request_callback(fut, sent_count=sent_count)
+            )
         except Exception as e:
             logger.debug("Failed to submit telemetry request: %s", e)
 
@@ -223,7 +225,7 @@ def _telemetry_request_callback(self, future, sent_count: int):
                     response.text,
                 )
 
-            telemetry_response = TelemetryResponse.from_json(**response.json())
+            telemetry_response = TelemetryResponse(**response.json())
 
             logger.debug(
                 "Pushed Telemetry logs with success count: %s, error count: %s",