more clean up

Signed-off-by: Vikrant Puppala <vikrant.puppala@databricks.com>

Author: vikrantpuppala

src/databricks/sql/result_set.py

@@ -244,7 +244,7 @@ def __init__(
                 session_id_hex=connection.get_session_id_hex(),
                 statement_id=execute_response.command_id.to_hex_guid(),
                 chunk_id=self.num_chunks,
-                http_client=connection.session.http_client,
+                http_client=connection.http_client,
             )
             if t_row_set.resultLinks:
                 self.num_chunks += len(t_row_set.resultLinks)

src/databricks/sql/telemetry/telemetry_client.py

@@ -257,18 +257,6 @@ def _send_with_unified_client(self, url, data, headers, timeout=900):
             response = self._http_client.request(
                 HttpMethod.POST, url, body=data, headers=headers, timeout=timeout
             )
-            # Convert urllib3 response to requests-like response for compatibility
-            response.status_code = response.status
-            response.ok = 200 <= response.status < 300
-            response.json = (
-                lambda: json.loads(response.data.decode()) if response.data else {}
-            )
-            # Add raise_for_status method
-            def raise_for_status():
-                if not response.ok:
-                    raise Exception(f"HTTP {response.status_code}")
-
-            response.raise_for_status = raise_for_status
             return response
         except Exception as e:
             logger.error("Failed to send telemetry with unified client: %s", e)
@@ -279,14 +267,18 @@ def _telemetry_request_callback(self, future, sent_count: int):
         try:
             response = future.result()
 
-            if not response.ok:
+            # Check if response is successful (urllib3 uses response.status)
+            is_success = 200 <= response.status < 300
+            if not is_success:
                 logger.debug(
                     "Telemetry request failed with status code: %s, response: %s",
-                    response.status_code,
-                    response.text,
+                    response.status,
+                    response.data.decode() if response.data else "",
                 )
 
-            telemetry_response = TelemetryResponse(**response.json())
+            # Parse JSON response (urllib3 uses response.data)
+            response_data = json.loads(response.data.decode()) if response.data else {}
+            telemetry_response = TelemetryResponse(**response_data)
 
             logger.debug(
                 "Pushed Telemetry logs with success count: %s, error count: %s",