snowflake-connector-python 5.0.0b1__tar.gz → 5.0.0b2__tar.gz

{snowflake_connector_python-5.0.0b1 → snowflake_connector_python-5.0.0b2}/Cargo.lock

@@ -905,6 +905,55 @@ dependencies = [
  "tracing",
 ]
 
+[[package]]
+name = "axum"
+version = "0.8.9"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "31b698c5f9a010f6573133b09e0de5408834d0c82f8d7475a89fc1867a71cd90"
+dependencies = [
+ "axum-core",
+ "bytes",
+ "form_urlencoded",
+ "futures-util",
+ "http 1.3.1",
+ "http-body 1.0.1",
+ "http-body-util",
+ "hyper 1.7.0",
+ "hyper-util",
+ "itoa",
+ "matchit",
+ "memchr",
+ "mime",
+ "percent-encoding",
+ "pin-project-lite",
+ "serde_core",
+ "serde_path_to_error",
+ "serde_urlencoded",
+ "sync_wrapper",
+ "tokio",
+ "tower",
+ "tower-layer",
+ "tower-service",
+]
+
+[[package]]
+name = "axum-core"
+version = "0.5.6"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "08c78f31d7b1291f7ee735c1c6780ccde7785daae9a9206026862dab7d8792d1"
+dependencies = [
+ "bytes",
+ "futures-core",
+ "http 1.3.1",
+ "http-body 1.0.1",
+ "http-body-util",
+ "mime",
+ "pin-project-lite",
+ "sync_wrapper",
+ "tower-layer",
+ "tower-service",
+]
+
 [[package]]
 name = "backtrace"
 version = "0.3.76"
@@ -1377,7 +1426,7 @@ dependencies = [
  "crc",
  "digest",
  "libc",
- "rand",
+ "rand 0.9.2",
  "regex",
 ]
 
@@ -2652,7 +2701,7 @@ dependencies = [
 name = "jdbc_bridge"
 version = "0.1.0"
 dependencies = [
- "jni",
+ "jni 0.21.1",
  "proto_utils",
  "sf_core",
  "tokio",
@@ -2694,7 +2743,7 @@ dependencies = [
  "cfg-if",
  "combine",
  "java-locator",
- "jni-sys",
+ "jni-sys 0.3.0",
  "libloading 0.7.4",
  "log",
  "thiserror 1.0.69",
@@ -2702,12 +2751,61 @@ dependencies = [
  "windows-sys 0.45.0",
 ]
 
+[[package]]
+name = "jni"
+version = "0.22.4"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "5efd9a482cf3a427f00d6b35f14332adc7902ce91efb778580e180ff90fa3498"
+dependencies = [
+ "cfg-if",
+ "combine",
+ "jni-macros",
+ "jni-sys 0.4.1",
+ "log",
+ "simd_cesu8",
+ "thiserror 2.0.16",
+ "walkdir",
+ "windows-link 0.2.0",
+]
+
+[[package]]
+name = "jni-macros"
+version = "0.22.4"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "a00109accc170f0bdb141fed3e393c565b6f5e072365c3bd58f5b062591560a3"
+dependencies = [
+ "proc-macro2",
+ "quote",
+ "rustc_version",
+ "simd_cesu8",
+ "syn 2.0.106",
+]
+
 [[package]]
 name = "jni-sys"
 version = "0.3.0"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "8eaf4bc02d17cbdd7ff4c7438cafcdf7fb9a4613313ad11b4f8fefe7d3fa0130"
 
+[[package]]
+name = "jni-sys"
+version = "0.4.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "c6377a88cb3910bee9b0fa88d4f42e1d2da8e79915598f65fb0c7ee14c878af2"
+dependencies = [
+ "jni-sys-macros",
+]
+
+[[package]]
+name = "jni-sys-macros"
+version = "0.4.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "38c0b942f458fe50cdac086d2f946512305e5631e720728f2a61aabcd47a6264"
+dependencies = [
+ "quote",
+ "syn 2.0.106",
+]
+
 [[package]]
 name = "jobserver"
 version = "0.1.34"
@@ -2931,6 +3029,12 @@ dependencies = [
  "regex-automata",
 ]
 
+[[package]]
+name = "matchit"
+version = "0.8.4"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "47e1ffaa40ddd1f3ed91f717a33c8c0ee23fff369e3aa8772b9605cc1d22f4c3"
+
 [[package]]
 name = "md-5"
 version = "0.10.6"
@@ -3031,6 +3135,12 @@ dependencies = [
  "tempfile",
 ]
 
+[[package]]
+name = "ndk-context"
+version = "0.1.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "27b02d87554356db9e9a873add8782d4ea6e3e58ea071a9adb9a2e8ddb884a8b"
+
 [[package]]
 name = "nix"
 version = "0.26.4"
@@ -3173,6 +3283,26 @@ dependencies = [
  "libc",
 ]
 
+[[package]]
+name = "oauth2"
+version = "5.0.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "51e219e79014df21a225b1860a479e2dcd7cbd9130f4defd4bd0e191ea31d67d"
+dependencies = [
+ "base64 0.22.1",
+ "chrono",
+ "getrandom 0.2.16",
+ "http 1.3.1",
+ "rand 0.8.6",
+ "reqwest",
+ "serde",
+ "serde_json",
+ "serde_path_to_error",
+ "sha2",
+ "thiserror 1.0.69",
+ "url",
+]
+
 [[package]]
 name = "objc2"
 version = "0.6.4"
@@ -3355,9 +3485,12 @@ dependencies = [
  "parking_lot",
  "proto_utils",
  "rust-ini",
+ "rustc_version",
  "serde_json",
  "sf_core",
  "snafu 0.8.9",
+ "strum",
+ "strum_macros",
  "test-case",
  "tokio",
  "tokio-util",
@@ -3540,7 +3673,7 @@ dependencies = [
  "futures-util",
  "opentelemetry",
  "percent-encoding",
- "rand",
+ "rand 0.9.2",
  "serde_json",
  "thiserror 2.0.16",
 ]
@@ -3985,7 +4118,7 @@ dependencies = [
  "bytes",
  "getrandom 0.3.3",
  "lru-slab",
- "rand",
+ "rand 0.9.2",
  "ring",
  "rustc-hash",
  "rustls 0.23.32",
@@ -4035,16 +4168,37 @@ dependencies = [
  "ptr_meta",
 ]
 
+[[package]]
+name = "rand"
+version = "0.8.6"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "5ca0ecfa931c29007047d1bc58e623ab12e5590e8c7cc53200d5202b69266d8a"
+dependencies = [
+ "libc",
+ "rand_chacha 0.3.1",
+ "rand_core 0.6.4",
+]
+
 [[package]]
 name = "rand"
 version = "0.9.2"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "6db2770f06117d490610c7488547d543617b21bfa07796d7a12f6f1bd53850d1"
 dependencies = [
- "rand_chacha",
+ "rand_chacha 0.9.0",
  "rand_core 0.9.3",
 ]
 
+[[package]]
+name = "rand_chacha"
+version = "0.3.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "e6c10a63a0fa32252be49d21e7709d4d4baf8d231c2dbce1eaa8141b9b127d88"
+dependencies = [
+ "ppv-lite86",
+ "rand_core 0.6.4",
+]
+
 [[package]]
 name = "rand_chacha"
 version = "0.9.0"
@@ -4609,6 +4763,17 @@ dependencies = [
  "serde_core",
 ]
 
+[[package]]
+name = "serde_path_to_error"
+version = "0.1.20"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "10a9ff822e371bb5403e391ecd83e182e0e77ba7f6fe0160b795797109d1b457"
+dependencies = [
+ "itoa",
+ "serde",
+ "serde_core",
+]
+
 [[package]]
 name = "serde_spanned"
 version = "0.6.9"
@@ -4664,6 +4829,7 @@ dependencies = [
  "aws-lc-rs",
  "aws-sdk-s3",
  "aws-sdk-sts",
+ "axum",
  "base64 0.22.1",
  "brotli",
  "bzip2",
@@ -4681,6 +4847,7 @@ dependencies = [
  "glob",
  "hex",
  "html-escape",
+ "http 1.3.1",
  "infer",
  "jwt",
  "keyring",
@@ -4688,6 +4855,7 @@ dependencies = [
  "lru",
  "memchr",
  "num-traits",
+ "oauth2",
  "once_cell",
  "openssl",
  "opentelemetry",
@@ -4699,7 +4867,7 @@ dependencies = [
  "prost 0.14.1",
  "proto_generator",
  "proto_utils",
- "rand",
+ "rand 0.9.2",
  "rcgen",
  "reqwest",
  "rust-ini",
@@ -4733,6 +4901,7 @@ dependencies = [
  "url",
  "urlencoding",
  "uuid",
+ "webbrowser",
  "wiremock",
  "x509-cert",
  "x509-parser",
@@ -4816,6 +4985,16 @@ version = "2.2.0"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "77549399552de45a898a580c1b41d445bf730df867cc44e6c0233bbc4b8329de"
 
+[[package]]
+name = "simd_cesu8"
+version = "1.1.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "94f90157bb87cddf702797c5dadfa0be7d266cdf49e22da2fcaa32eff75b2c33"
+dependencies = [
+ "rustc_version",
+ "simdutf8",
+]
+
 [[package]]
 name = "simdutf8"
 version = "0.1.5"
@@ -4983,6 +5162,25 @@ version = "0.11.1"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "7da8b5736845d9f2fcb837ea5d9e2628564b3b043a70948a3f0b778838c5fb4f"
 
+[[package]]
+name = "strum"
+version = "0.26.3"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "8fec0f0aef304996cf250b31b5a10dee7980c85da9d759361292b8bca5a18f06"
+
+[[package]]
+name = "strum_macros"
+version = "0.26.4"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "4c6bee85a5a24955dc440386795aa378cd9cf82acd5f764469152d2270e581be"
+dependencies = [
+ "heck 0.5.0",
+ "proc-macro2",
+ "quote",
+ "rustversion",
+ "syn 2.0.106",
+]
+
 [[package]]
 name = "subtle"
 version = "2.6.1"
@@ -5885,6 +6083,22 @@ dependencies = [
  "wasm-bindgen",
 ]
 
+[[package]]
+name = "webbrowser"
+version = "1.2.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "0fc95580916af1e68ff6a7be07446fc5db73ebf71cf092de939bbf5f7e189f72"
+dependencies = [
+ "core-foundation 0.10.1",
+ "jni 0.22.4",
+ "log",
+ "ndk-context",
+ "objc2",
+ "objc2-foundation",
+ "url",
+ "web-sys",
+]
+
 [[package]]
 name = "webpki-roots"
 version = "1.0.2"

{snowflake_connector_python-5.0.0b1 → snowflake_connector_python-5.0.0b2}/PKG-INFO

@@ -1,7 +1,7 @@
 Metadata-Version: 2.4
 Name: snowflake-connector-python
-Version: 5.0.0b1
-Summary: A PEP 249 Database API 2.0 implementation for Snowflake
+Version: 5.0.0b2
+Summary: Snowflake DB driver for Python (Private Preview)
 Project-URL: Homepage, https://github.com/snowflakedb/universal-driver
 Project-URL: Bug Reports, https://github.com/snowflakedb/universal-driver/issues
 Project-URL: Source, https://github.com/snowflakedb/universal-driver
@@ -19,7 +19,7 @@ Classifier: Programming Language :: Python :: 3.13
 Classifier: Programming Language :: Python :: 3.14
 Classifier: Topic :: Database
 Classifier: Topic :: Software Development :: Libraries :: Python Modules
-Requires-Python: >=3.9
+Requires-Python: >=3.10
 Requires-Dist: cryptography>=44.0.1
 Requires-Dist: protobuf>=6.32.1
 Requires-Dist: pytz
@@ -56,14 +56,18 @@ Requires-Dist: tzdata; extra == 'test'
 Requires-Dist: zstandard; extra == 'test'
 Description-Content-Type: text/markdown
 
-# PEP 249 Database API 2.0 Implementation
+# ⚠️ PRIVATE PREVIEW DISCLAIMER ⚠️
 
-A Python library that implements [PEP 249 (Python Database API Specification 2.0)](https://peps.python.org/pep-0249/) with empty interface implementations. This library provides a complete skeleton implementation that follows the PEP 249 specification, making it an ideal starting point for creating new database drivers or for testing database API compliance.
+**IMPORTANT:** This release contains a **Private Preview** version of the Snowflake DB driver. By downloading or using this software, you acknowledge and agree to the following conditions:
+
+- **Restricted Access:** Usage of this driver is strictly limited to participants who are actively enrolled in the official Private Preview program for this specific driver. If you have not been explicitly invited and authorized to participate in this preview, do not download, install, or use this software.
+- **Testing Environments Only:** For authorized Private Preview partners, this driver is provided solely for evaluation, feedback, and testing purposes. It must be deployed exclusively in isolated, non-production environments.
+- **"As-Is" Software:** As a preview release, this driver is still under active development. It may contain bugs, lack features, or undergo significant breaking changes before general availability. It is provided "as-is" without any warranties, service level agreements (SLAs), or official support commitments.
 
 ## Development
 
 ### Prerequisites
-- Python 3.9+
+- Python 3.10+
 - [uv](https://docs.astral.sh/uv/) package manager
 - [Hatch](https://hatch.pypa.io/) build tool
 - Rust toolchain (for building core library)
@@ -102,7 +106,7 @@ The `dev` environment is designed for **human developers** during active develop
 - Installs from sources (`skip-install = false`)
 - Editable mode enabled (`dev-mode = true`)
 - Changes to source code are immediately available
-- Supports Python matrix: 3.9, 3.10, 3.11, 3.12, 3.13
+- Supports Python matrix: 3.10, 3.11, 3.12, 3.13, 3.14
 
 **Usage:**
 ```bash
@@ -165,8 +169,10 @@ hatch run test.py3.12:all
 hatch run precommit:check        # Run all checks (format, lint, type)
 hatch run precommit:fix          # Auto-fix formatting and linting issues
 
-# Reference connector tests (for compatibility testing)
-PYTHON_REFERENCE_DRIVER_VERSION=3.17.2 hatch run reference:test
+# Reference connector tests (for compatibility testing, installs latest v4.x)
+hatch run reference:test
+# Pin a specific version if needed:
+PYTHON_REFERENCE_DRIVER_VERSION='==4.3.0' hatch run reference:test
 ```
 
 ## References

{snowflake_connector_python-5.0.0b1 → snowflake_connector_python-5.0.0b2}/README.md

@@ -1,11 +1,15 @@
-# PEP 249 Database API 2.0 Implementation
+# ⚠️ PRIVATE PREVIEW DISCLAIMER ⚠️
 
-A Python library that implements [PEP 249 (Python Database API Specification 2.0)](https://peps.python.org/pep-0249/) with empty interface implementations. This library provides a complete skeleton implementation that follows the PEP 249 specification, making it an ideal starting point for creating new database drivers or for testing database API compliance.
+**IMPORTANT:** This release contains a **Private Preview** version of the Snowflake DB driver. By downloading or using this software, you acknowledge and agree to the following conditions:
+
+- **Restricted Access:** Usage of this driver is strictly limited to participants who are actively enrolled in the official Private Preview program for this specific driver. If you have not been explicitly invited and authorized to participate in this preview, do not download, install, or use this software.
+- **Testing Environments Only:** For authorized Private Preview partners, this driver is provided solely for evaluation, feedback, and testing purposes. It must be deployed exclusively in isolated, non-production environments.
+- **"As-Is" Software:** As a preview release, this driver is still under active development. It may contain bugs, lack features, or undergo significant breaking changes before general availability. It is provided "as-is" without any warranties, service level agreements (SLAs), or official support commitments.
 
 ## Development
 
 ### Prerequisites
-- Python 3.9+
+- Python 3.10+
 - [uv](https://docs.astral.sh/uv/) package manager
 - [Hatch](https://hatch.pypa.io/) build tool
 - Rust toolchain (for building core library)
@@ -44,7 +48,7 @@ The `dev` environment is designed for **human developers** during active develop
 - Installs from sources (`skip-install = false`)
 - Editable mode enabled (`dev-mode = true`)
 - Changes to source code are immediately available
-- Supports Python matrix: 3.9, 3.10, 3.11, 3.12, 3.13
+- Supports Python matrix: 3.10, 3.11, 3.12, 3.13, 3.14
 
 **Usage:**
 ```bash
@@ -107,8 +111,10 @@ hatch run test.py3.12:all
 hatch run precommit:check        # Run all checks (format, lint, type)
 hatch run precommit:fix          # Auto-fix formatting and linting issues
 
-# Reference connector tests (for compatibility testing)
-PYTHON_REFERENCE_DRIVER_VERSION=3.17.2 hatch run reference:test
+# Reference connector tests (for compatibility testing, installs latest v4.x)
+hatch run reference:test
+# Pin a specific version if needed:
+PYTHON_REFERENCE_DRIVER_VERSION='==4.3.0' hatch run reference:test
 ```
 
 ## References

{snowflake_connector_python-5.0.0b1 → snowflake_connector_python-5.0.0b2}/protobuf/database_driver_v1.proto

@@ -150,8 +150,6 @@ message QueryStats {
 }
 
 // Describes one result set from a (possibly multi-statement) execution.
-// For single statements, returned inline in the execute response.
-// For multi-statement children, fetched lazily via StatementGetResultSet.
 message ResultSetDescriptor {
   string query_id = 1;
   repeated ColumnMetadata columns = 2;
@@ -161,6 +159,11 @@ message ResultSetDescriptor {
   optional QueryStats stats = 6;
 }
 
+message ResultSetResponse {
+  ResultSetHandle result_set_handle = 1;
+  ResultSetDescriptor result_descriptor = 2;
+}
+
 // Multi-statement execution: child query IDs and their statement types.
 message MultiStatementResult {
   ResultSetDescriptor parent = 1;
@@ -195,6 +198,12 @@ message StatementHandle {
   int64 magic = 2;
 }
 
+// ResultSet handle
+message ResultSetHandle {
+  int64 id = 1;
+  int64 magic = 2;
+}
+
 // Arrow array stream pointer
 message ArrowArrayStreamPtr {
   bytes value = 1;
@@ -405,6 +414,14 @@ message ConnectionGetParameterResponse {
   optional string value = 1;
 }
 
+message ConnectionGetAllParametersRequest {
+  ConnectionHandle conn_handle = 1;
+}
+
+message ConnectionGetAllParametersResponse {
+  map<string, string> parameters = 1;
+}
+
 message ConnectionValidateOptionsRequest {
   ConnectionHandle conn_handle = 1;
 }
@@ -462,11 +479,6 @@ message ConnectionGetQueryResultRequest {
   string query_id = 2;
 }
 
-message ResultSetResponse {
-  ResultSetDescriptor result_descriptor = 1;
-  ArrowArrayStreamPtr stream = 2;
-}
-
 message ConnectionGetResultSetRequest {
   ConnectionHandle conn_handle = 1;
   string query_id = 2;
@@ -544,24 +556,18 @@ message StatementGetParameterSchemaResponse {
 message StatementExecuteQueryRequest {
   StatementHandle stmt_handle = 1;
   optional QueryBindings bindings = 2;  // None = no bindings
+  optional uint32 timeout_seconds = 3;  // 0 / absent = no timeout override
 }
 
 message ExecuteQueryResponse {
   oneof result {
-    // Single statement — metadata available immediately.
-    ResultSetDescriptor single = 1;
+    // Single statement — descriptor + result_set_handle for stream/chunks access.
+    ResultSetResponse single = 1;
     // Multi-statement — child query IDs only, fetch metadata lazily.
     MultiStatementResult multi = 2;
   }
 }
 
-// Fetch a result set by query_id: returns metadata + Arrow stream.
-// For multi-statement children, this triggers a lazy fetch in sf_core.
-message StatementGetResultSetRequest {
-  StatementHandle stmt_handle = 1;
-  string query_id = 2;
-}
-
 // Async query execution: submit and return immediately with query_id
 message StatementExecuteAsyncRequest {
   StatementHandle stmt_handle = 1;
@@ -617,45 +623,43 @@ message ResultChunk {
   int32 row_count = 4;
 }
 
-message ResultChunksResult {
-  repeated ResultChunk chunks = 1;
-  // Result column metadata. Required for CHUNK_FORMAT_JSON (forward verbatim on DatabaseFetchChunkRequest); ignored for CHUNK_FORMAT_ARROW_IPC.
-  repeated ColumnMetadata columns = 2;
+
+message DatabaseFetchChunkRequest {
+  DatabaseHandle db_handle = 1;
+  ResultChunk chunk = 2;
+  // Required when chunk.format == CHUNK_FORMAT_JSON (forward from ResultSetGetChunksResponse.columns); ignored for CHUNK_FORMAT_ARROW_IPC.
+  repeated ColumnMetadata columns = 3;
 }
 
-// Fetch raw result chunks for a query executed on this statement.
-// query_id identifies which query's chunks to return; for single-statement
-// executions this is the query_id from ExecuteQueryResponse, for
-// multi-statement children it is one of the query_ids from MultiStatementResult.
-message StatementResultChunksRequest {
-  StatementHandle stmt_handle = 1;
-  string query_id = 2;
+message DatabaseFetchChunkResponse {
+  ArrowArrayStreamPtr stream = 1;
 }
 
-message StatementResultChunksResponse {
-  ResultChunksResult result = 1;
+// ResultSet RPCs
+
+message ResultSetGetStreamRequest {
+  ResultSetHandle result_set_handle = 1;
 }
 
-// Fetch raw result chunks by query_id using the connection (no statement handle required).
-// Useful for multi-statement children after the batch statement handle has been released.
-message ConnectionResultChunksRequest {
-  ConnectionHandle conn_handle = 1;
-  string query_id = 2;
+message ResultSetGetStreamResponse {
+  ArrowArrayStreamPtr stream = 1;
 }
 
-message ConnectionResultChunksResponse {
-  ResultChunksResult result = 1;
+message ResultSetGetChunksRequest {
+  ResultSetHandle result_set_handle = 1;
 }
 
-message DatabaseFetchChunkRequest {
-  DatabaseHandle db_handle = 1;
-  ResultChunk chunk = 2;
-  // Required when chunk.format == CHUNK_FORMAT_JSON (forward verbatim from ResultChunksResult.columns); ignored for CHUNK_FORMAT_ARROW_IPC.
-  repeated ColumnMetadata columns = 3;
+message ResultSetGetChunksResponse {
+  repeated ResultChunk chunks = 1;
+  // Column metadata. Required for CHUNK_FORMAT_JSON (forward on DatabaseFetchChunkRequest); ignored for CHUNK_FORMAT_ARROW_IPC.
+  repeated ColumnMetadata columns = 2;
 }
 
-message DatabaseFetchChunkResponse {
-  ArrowArrayStreamPtr stream = 1;
+message ResultSetReleaseRequest {
+  ResultSetHandle result_set_handle = 1;
+}
+
+message ResultSetReleaseResponse {
 }
 
 // Validation issue reported during set or validate operations.
@@ -806,6 +810,8 @@ service DatabaseDriver {
   // gets cached session parameter. Cache is updated with subset of connection parameters
   // (chosen by server, included in HTTP response) after each query
   rpc ConnectionGetParameter(ConnectionGetParameterRequest) returns (ConnectionGetParameterResponse);
+  // gets all resolved session parameters at once
+  rpc ConnectionGetAllParameters(ConnectionGetAllParametersRequest) returns (ConnectionGetAllParametersResponse);
   // pre-flight validation of connection options - reports all issues at once
   rpc ConnectionValidateOptions(ConnectionValidateOptionsRequest) returns (ConnectionValidateOptionsResponse);
   // Fetch the result of a previously executed query by Snowflake Query ID
@@ -820,6 +826,7 @@ service DatabaseDriver {
   rpc ConnectionRequestToken(ConnectionTokenRequest) returns (ConnectionTokenResponse);
   // Heartbeat to validate connection and session are still alive
   rpc ConnectionHeartbeat(ConnectionHeartbeatRequest) returns (ConnectionHeartbeatResponse);
+
   // Statement operations
   rpc StatementNew(StatementNewRequest) returns (StatementNewResponse);
   rpc StatementRelease(StatementReleaseRequest) returns (StatementReleaseResponse);
@@ -829,13 +836,15 @@ service DatabaseDriver {
   rpc StatementSetOptions(StatementSetOptionsRequest) returns (StatementSetOptionsResponse);
   rpc StatementGetParameterSchema(StatementGetParameterSchemaRequest) returns (StatementGetParameterSchemaResponse);
   rpc StatementExecuteQuery(StatementExecuteQueryRequest) returns (ExecuteQueryResponse);
-  rpc StatementGetResultSet(StatementGetResultSetRequest) returns (ResultSetResponse);
   rpc StatementExecutePartitions(StatementExecutePartitionsRequest) returns (StatementExecutePartitionsResponse);
   rpc StatementReadPartition(StatementReadPartitionRequest) returns (StatementReadPartitionResponse);
-  rpc StatementResultChunks(StatementResultChunksRequest) returns (StatementResultChunksResponse);
-  rpc ConnectionResultChunks(ConnectionResultChunksRequest) returns (ConnectionResultChunksResponse);
   rpc StatementExecuteAsync(StatementExecuteAsyncRequest) returns (StatementExecuteAsyncResponse);
 
+  // ResultSet operations
+  rpc ResultSetGetStream(ResultSetGetStreamRequest) returns (ResultSetGetStreamResponse);
+  rpc ResultSetGetChunks(ResultSetGetChunksRequest) returns (ResultSetGetChunksResponse);
+  rpc ResultSetRelease(ResultSetReleaseRequest) returns (ResultSetReleaseResponse);
+
   // Config operations
   rpc ConfigLoadAllSections(ConfigLoadAllSectionsRequest) returns (ConfigLoadAllSectionsResponse);
   rpc ConfigGetPaths(ConfigGetPathsRequest) returns (ConfigGetPathsResponse);