hotdata 0.3.0__tar.gz → 0.4.0__tar.gz

{hotdata-0.3.0 → hotdata-0.4.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: hotdata
-Version: 0.3.0
+Version: 0.4.0
 Summary: Hotdata API
 Author-email: Hotdata <developers@hotdata.dev>
 License: MIT
@@ -11,10 +11,10 @@ Requires-Python: >=3.9
 Description-Content-Type: text/markdown
 Requires-Dist: urllib3<3.0.0,>=2.1.0
 Requires-Dist: python-dateutil>=2.8.2
-Requires-Dist: pydantic>=2
+Requires-Dist: pydantic>=2.4.0
 Requires-Dist: typing-extensions>=4.7.1
 Provides-Extra: arrow
-Requires-Dist: pyarrow>=14; extra == "arrow"
+Requires-Dist: pyarrow>=14.0.1; extra == "arrow"
 
 # hotdata
 

{hotdata-0.3.0 → hotdata-0.4.0}/hotdata/__init__.py

@@ -37,7 +37,6 @@ __all__ = [
     "QueryRunsApi",
     "RefreshApi",
     "ResultsApi",
-    "SandboxesApi",
     "SavedQueriesApi",
     "SecretsApi",
     "UploadsApi",
@@ -85,7 +84,6 @@ __all__ = [
     "CreateEmbeddingProviderRequest",
     "CreateEmbeddingProviderResponse",
     "CreateIndexRequest",
-    "CreateSandboxRequest",
     "CreateSavedQueryRequest",
     "CreateSecretRequest",
     "CreateSecretResponse",
@@ -105,7 +103,6 @@ __all__ = [
     "DatasetSourceOneOf4",
     "DatasetSummary",
     "DatasetVersionSummary",
-    "DeleteSandboxResponse",
     "DiscoveryStatus",
     "EmbeddingProviderResponse",
     "Error",
@@ -115,6 +112,7 @@ __all__ = [
     "GetDatasetResponse",
     "GetResultResponse",
     "GetSecretResponse",
+    "IndexEntryResponse",
     "IndexInfoResponse",
     "IndexStatus",
     "InformationSchemaResponse",
@@ -131,11 +129,11 @@ __all__ = [
     "ListDatasetVersionsResponse",
     "ListDatasetsResponse",
     "ListEmbeddingProvidersResponse",
+    "ListIndexesPageResponse",
     "ListIndexesResponse",
     "ListJobsResponse",
     "ListQueryRunsResponse",
     "ListResultsResponse",
-    "ListSandboxesResponse",
     "ListSavedQueriesResponse",
     "ListSavedQueryVersionsResponse",
     "ListSecretsResponse",
@@ -155,8 +153,6 @@ __all__ = [
     "RefreshWarning",
     "ResultInfo",
     "ResultsFormatQuery",
-    "Sandbox",
-    "SandboxResponse",
     "SavedQueryDatasetSource",
     "SavedQueryDetail",
     "SavedQuerySummary",
@@ -175,7 +171,6 @@ __all__ = [
     "UpdateDatasetResponse",
     "UpdateEmbeddingProviderRequest",
     "UpdateEmbeddingProviderResponse",
-    "UpdateSandboxRequest",
     "UpdateSavedQueryRequest",
     "UpdateSecretRequest",
     "UpdateSecretResponse",
@@ -203,7 +198,6 @@ from hotdata.api.query_api import QueryApi as QueryApi
 from hotdata.api.query_runs_api import QueryRunsApi as QueryRunsApi
 from hotdata.api.refresh_api import RefreshApi as RefreshApi
 from hotdata.api.results_api import ResultsApi as ResultsApi
-from hotdata.api.sandboxes_api import SandboxesApi as SandboxesApi
 from hotdata.api.saved_queries_api import SavedQueriesApi as SavedQueriesApi
 from hotdata.api.secrets_api import SecretsApi as SecretsApi
 from hotdata.api.uploads_api import UploadsApi as UploadsApi
@@ -255,7 +249,6 @@ from hotdata.models.create_dataset_response import CreateDatasetResponse as Crea
 from hotdata.models.create_embedding_provider_request import CreateEmbeddingProviderRequest as CreateEmbeddingProviderRequest
 from hotdata.models.create_embedding_provider_response import CreateEmbeddingProviderResponse as CreateEmbeddingProviderResponse
 from hotdata.models.create_index_request import CreateIndexRequest as CreateIndexRequest
-from hotdata.models.create_sandbox_request import CreateSandboxRequest as CreateSandboxRequest
 from hotdata.models.create_saved_query_request import CreateSavedQueryRequest as CreateSavedQueryRequest
 from hotdata.models.create_secret_request import CreateSecretRequest as CreateSecretRequest
 from hotdata.models.create_secret_response import CreateSecretResponse as CreateSecretResponse
@@ -275,7 +268,6 @@ from hotdata.models.dataset_source_one_of3 import DatasetSourceOneOf3 as Dataset
 from hotdata.models.dataset_source_one_of4 import DatasetSourceOneOf4 as DatasetSourceOneOf4
 from hotdata.models.dataset_summary import DatasetSummary as DatasetSummary
 from hotdata.models.dataset_version_summary import DatasetVersionSummary as DatasetVersionSummary
-from hotdata.models.delete_sandbox_response import DeleteSandboxResponse as DeleteSandboxResponse
 from hotdata.models.discovery_status import DiscoveryStatus as DiscoveryStatus
 from hotdata.models.embedding_provider_response import EmbeddingProviderResponse as EmbeddingProviderResponse
 from hotdata.models.error import Error as Error
@@ -285,6 +277,7 @@ from hotdata.models.get_database_context_response import GetDatabaseContextRespo
 from hotdata.models.get_dataset_response import GetDatasetResponse as GetDatasetResponse
 from hotdata.models.get_result_response import GetResultResponse as GetResultResponse
 from hotdata.models.get_secret_response import GetSecretResponse as GetSecretResponse
+from hotdata.models.index_entry_response import IndexEntryResponse as IndexEntryResponse
 from hotdata.models.index_info_response import IndexInfoResponse as IndexInfoResponse
 from hotdata.models.index_status import IndexStatus as IndexStatus
 from hotdata.models.information_schema_response import InformationSchemaResponse as InformationSchemaResponse
@@ -301,11 +294,11 @@ from hotdata.models.list_databases_response import ListDatabasesResponse as List
 from hotdata.models.list_dataset_versions_response import ListDatasetVersionsResponse as ListDatasetVersionsResponse
 from hotdata.models.list_datasets_response import ListDatasetsResponse as ListDatasetsResponse
 from hotdata.models.list_embedding_providers_response import ListEmbeddingProvidersResponse as ListEmbeddingProvidersResponse
+from hotdata.models.list_indexes_page_response import ListIndexesPageResponse as ListIndexesPageResponse
 from hotdata.models.list_indexes_response import ListIndexesResponse as ListIndexesResponse
 from hotdata.models.list_jobs_response import ListJobsResponse as ListJobsResponse
 from hotdata.models.list_query_runs_response import ListQueryRunsResponse as ListQueryRunsResponse
 from hotdata.models.list_results_response import ListResultsResponse as ListResultsResponse
-from hotdata.models.list_sandboxes_response import ListSandboxesResponse as ListSandboxesResponse
 from hotdata.models.list_saved_queries_response import ListSavedQueriesResponse as ListSavedQueriesResponse
 from hotdata.models.list_saved_query_versions_response import ListSavedQueryVersionsResponse as ListSavedQueryVersionsResponse
 from hotdata.models.list_secrets_response import ListSecretsResponse as ListSecretsResponse
@@ -325,8 +318,6 @@ from hotdata.models.refresh_response import RefreshResponse as RefreshResponse
 from hotdata.models.refresh_warning import RefreshWarning as RefreshWarning
 from hotdata.models.result_info import ResultInfo as ResultInfo
 from hotdata.models.results_format_query import ResultsFormatQuery as ResultsFormatQuery
-from hotdata.models.sandbox import Sandbox as Sandbox
-from hotdata.models.sandbox_response import SandboxResponse as SandboxResponse
 from hotdata.models.saved_query_dataset_source import SavedQueryDatasetSource as SavedQueryDatasetSource
 from hotdata.models.saved_query_detail import SavedQueryDetail as SavedQueryDetail
 from hotdata.models.saved_query_summary import SavedQuerySummary as SavedQuerySummary
@@ -345,7 +336,6 @@ from hotdata.models.update_dataset_request import UpdateDatasetRequest as Update
 from hotdata.models.update_dataset_response import UpdateDatasetResponse as UpdateDatasetResponse
 from hotdata.models.update_embedding_provider_request import UpdateEmbeddingProviderRequest as UpdateEmbeddingProviderRequest
 from hotdata.models.update_embedding_provider_response import UpdateEmbeddingProviderResponse as UpdateEmbeddingProviderResponse
-from hotdata.models.update_sandbox_request import UpdateSandboxRequest as UpdateSandboxRequest
 from hotdata.models.update_saved_query_request import UpdateSavedQueryRequest as UpdateSavedQueryRequest
 from hotdata.models.update_secret_request import UpdateSecretRequest as UpdateSecretRequest
 from hotdata.models.update_secret_response import UpdateSecretResponse as UpdateSecretResponse
@@ -358,3 +348,11 @@ from hotdata.models.url_dataset_source import UrlDatasetSource as UrlDatasetSour
 from hotdata.models.workspace_detail import WorkspaceDetail as WorkspaceDetail
 from hotdata.models.workspace_list_item import WorkspaceListItem as WorkspaceListItem
 
+
+# --- hand-applied: prefer the enhanced clients over the generated ones
+# (re-applied by scripts/patch_query_exports.py after regeneration).
+# hotdata.query.QueryApi adds 429 retry + truncation auto-follow;
+# hotdata.arrow.ResultsApi adds Arrow IPC result fetch. The raw generated
+# classes remain importable from hotdata.api.query_api / hotdata.api.results_api.
+from hotdata.query import QueryApi as QueryApi  # noqa: E402,F811
+from hotdata.arrow import ResultsApi as ResultsApi  # noqa: E402,F811

{hotdata-0.3.0 → hotdata-0.4.0}/hotdata/api/__init__.py

@@ -14,7 +14,6 @@ from hotdata.api.query_api import QueryApi
 from hotdata.api.query_runs_api import QueryRunsApi
 from hotdata.api.refresh_api import RefreshApi
 from hotdata.api.results_api import ResultsApi
-from hotdata.api.sandboxes_api import SandboxesApi
 from hotdata.api.saved_queries_api import SavedQueriesApi
 from hotdata.api.secrets_api import SecretsApi
 from hotdata.api.uploads_api import UploadsApi

{hotdata-0.3.0 → hotdata-0.4.0}/hotdata/api/indexes_api.py

@@ -17,9 +17,11 @@ from typing import Any, Dict, List, Optional, Tuple, Union
 from typing_extensions import Annotated
 
 from pydantic import Field, StrictStr
+from typing import Optional
 from typing_extensions import Annotated
 from hotdata.models.create_index_request import CreateIndexRequest
 from hotdata.models.index_info_response import IndexInfoResponse
+from hotdata.models.list_indexes_page_response import ListIndexesPageResponse
 from hotdata.models.list_indexes_response import ListIndexesResponse
 
 from hotdata.api_client import ApiClient, RequestSerialized
@@ -1825,3 +1827,376 @@ class IndexesApi:
         )
 
 
+
+
+    @validate_call
+    def list_indexes_collection(
+        self,
+        x_database_id: Annotated[StrictStr, Field(description="Database to scope to (required)")],
+        connection_id: Annotated[Optional[StrictStr], Field(description="Filter to one connection")] = None,
+        var_schema: Annotated[Optional[StrictStr], Field(description="Filter by schema name")] = None,
+        table: Annotated[Optional[StrictStr], Field(description="Filter by table name")] = None,
+        index_type: Annotated[Optional[StrictStr], Field(description="Filter by index type")] = None,
+        limit: Annotated[Optional[Annotated[int, Field(strict=True, ge=0)]], Field(description="Max indexes per page")] = None,
+        cursor: Annotated[Optional[StrictStr], Field(description="Pagination cursor")] = None,
+        _request_timeout: Union[
+            None,
+            Annotated[StrictFloat, Field(gt=0)],
+            Tuple[
+                Annotated[StrictFloat, Field(gt=0)],
+                Annotated[StrictFloat, Field(gt=0)]
+            ]
+        ] = None,
+        _request_auth: Optional[Dict[StrictStr, Any]] = None,
+        _content_type: Optional[StrictStr] = None,
+        _headers: Optional[Dict[StrictStr, Any]] = None,
+        _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
+    ) -> ListIndexesPageResponse:
+        """List indexes across tables in a database
+
+        List all indexes in the database identified by the required X-Database-Id header, paginated. Optional filters narrow by connection, schema, table, or index type.
+
+        :param x_database_id: Database to scope to (required) (required)
+        :type x_database_id: str
+        :param connection_id: Filter to one connection
+        :type connection_id: str
+        :param var_schema: Filter by schema name
+        :type var_schema: str
+        :param table: Filter by table name
+        :type table: str
+        :param index_type: Filter by index type
+        :type index_type: str
+        :param limit: Max indexes per page
+        :type limit: int
+        :param cursor: Pagination cursor
+        :type cursor: str
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :type _request_timeout: int, tuple(int, int), optional
+        :param _request_auth: set to override the auth_settings for an a single
+                              request; this effectively ignores the
+                              authentication in the spec for a single request.
+        :type _request_auth: dict, optional
+        :param _content_type: force content-type for the request.
+        :type _content_type: str, Optional
+        :param _headers: set to override the headers for a single
+                         request; this effectively ignores the headers
+                         in the spec for a single request.
+        :type _headers: dict, optional
+        :param _host_index: set to override the host_index for a single
+                            request; this effectively ignores the host_index
+                            in the spec for a single request.
+        :type _host_index: int, optional
+        :return: Returns the result object.
+        """ # noqa: E501
+
+        _param = self._list_indexes_collection_serialize(
+            x_database_id=x_database_id,
+            connection_id=connection_id,
+            var_schema=var_schema,
+            table=table,
+            index_type=index_type,
+            limit=limit,
+            cursor=cursor,
+            _request_auth=_request_auth,
+            _content_type=_content_type,
+            _headers=_headers,
+            _host_index=_host_index
+        )
+
+        _response_types_map: Dict[str, Optional[str]] = {
+            '200': "ListIndexesPageResponse",
+            '400': "ApiErrorResponse",
+            '404': "ApiErrorResponse",
+            '500': "ApiErrorResponse",
+        }
+        response_data = self.api_client.call_api(
+            *_param,
+            _request_timeout=_request_timeout
+        )
+        response_data.read()
+        return self.api_client.response_deserialize(
+            response_data=response_data,
+            response_types_map=_response_types_map,
+        ).data
+
+
+    @validate_call
+    def list_indexes_collection_with_http_info(
+        self,
+        x_database_id: Annotated[StrictStr, Field(description="Database to scope to (required)")],
+        connection_id: Annotated[Optional[StrictStr], Field(description="Filter to one connection")] = None,
+        var_schema: Annotated[Optional[StrictStr], Field(description="Filter by schema name")] = None,
+        table: Annotated[Optional[StrictStr], Field(description="Filter by table name")] = None,
+        index_type: Annotated[Optional[StrictStr], Field(description="Filter by index type")] = None,
+        limit: Annotated[Optional[Annotated[int, Field(strict=True, ge=0)]], Field(description="Max indexes per page")] = None,
+        cursor: Annotated[Optional[StrictStr], Field(description="Pagination cursor")] = None,
+        _request_timeout: Union[
+            None,
+            Annotated[StrictFloat, Field(gt=0)],
+            Tuple[
+                Annotated[StrictFloat, Field(gt=0)],
+                Annotated[StrictFloat, Field(gt=0)]
+            ]
+        ] = None,
+        _request_auth: Optional[Dict[StrictStr, Any]] = None,
+        _content_type: Optional[StrictStr] = None,
+        _headers: Optional[Dict[StrictStr, Any]] = None,
+        _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
+    ) -> ApiResponse[ListIndexesPageResponse]:
+        """List indexes across tables in a database
+
+        List all indexes in the database identified by the required X-Database-Id header, paginated. Optional filters narrow by connection, schema, table, or index type.
+
+        :param x_database_id: Database to scope to (required) (required)
+        :type x_database_id: str
+        :param connection_id: Filter to one connection
+        :type connection_id: str
+        :param var_schema: Filter by schema name
+        :type var_schema: str
+        :param table: Filter by table name
+        :type table: str
+        :param index_type: Filter by index type
+        :type index_type: str
+        :param limit: Max indexes per page
+        :type limit: int
+        :param cursor: Pagination cursor
+        :type cursor: str
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :type _request_timeout: int, tuple(int, int), optional
+        :param _request_auth: set to override the auth_settings for an a single
+                              request; this effectively ignores the
+                              authentication in the spec for a single request.
+        :type _request_auth: dict, optional
+        :param _content_type: force content-type for the request.
+        :type _content_type: str, Optional
+        :param _headers: set to override the headers for a single
+                         request; this effectively ignores the headers
+                         in the spec for a single request.
+        :type _headers: dict, optional
+        :param _host_index: set to override the host_index for a single
+                            request; this effectively ignores the host_index
+                            in the spec for a single request.
+        :type _host_index: int, optional
+        :return: Returns the result object.
+        """ # noqa: E501
+
+        _param = self._list_indexes_collection_serialize(
+            x_database_id=x_database_id,
+            connection_id=connection_id,
+            var_schema=var_schema,
+            table=table,
+            index_type=index_type,
+            limit=limit,
+            cursor=cursor,
+            _request_auth=_request_auth,
+            _content_type=_content_type,
+            _headers=_headers,
+            _host_index=_host_index
+        )
+
+        _response_types_map: Dict[str, Optional[str]] = {
+            '200': "ListIndexesPageResponse",
+            '400': "ApiErrorResponse",
+            '404': "ApiErrorResponse",
+            '500': "ApiErrorResponse",
+        }
+        response_data = self.api_client.call_api(
+            *_param,
+            _request_timeout=_request_timeout
+        )
+        response_data.read()
+        return self.api_client.response_deserialize(
+            response_data=response_data,
+            response_types_map=_response_types_map,
+        )
+
+
+    @validate_call
+    def list_indexes_collection_without_preload_content(
+        self,
+        x_database_id: Annotated[StrictStr, Field(description="Database to scope to (required)")],
+        connection_id: Annotated[Optional[StrictStr], Field(description="Filter to one connection")] = None,
+        var_schema: Annotated[Optional[StrictStr], Field(description="Filter by schema name")] = None,
+        table: Annotated[Optional[StrictStr], Field(description="Filter by table name")] = None,
+        index_type: Annotated[Optional[StrictStr], Field(description="Filter by index type")] = None,
+        limit: Annotated[Optional[Annotated[int, Field(strict=True, ge=0)]], Field(description="Max indexes per page")] = None,
+        cursor: Annotated[Optional[StrictStr], Field(description="Pagination cursor")] = None,
+        _request_timeout: Union[
+            None,
+            Annotated[StrictFloat, Field(gt=0)],
+            Tuple[
+                Annotated[StrictFloat, Field(gt=0)],
+                Annotated[StrictFloat, Field(gt=0)]
+            ]
+        ] = None,
+        _request_auth: Optional[Dict[StrictStr, Any]] = None,
+        _content_type: Optional[StrictStr] = None,
+        _headers: Optional[Dict[StrictStr, Any]] = None,
+        _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
+    ) -> RESTResponseType:
+        """List indexes across tables in a database
+
+        List all indexes in the database identified by the required X-Database-Id header, paginated. Optional filters narrow by connection, schema, table, or index type.
+
+        :param x_database_id: Database to scope to (required) (required)
+        :type x_database_id: str
+        :param connection_id: Filter to one connection
+        :type connection_id: str
+        :param var_schema: Filter by schema name
+        :type var_schema: str
+        :param table: Filter by table name
+        :type table: str
+        :param index_type: Filter by index type
+        :type index_type: str
+        :param limit: Max indexes per page
+        :type limit: int
+        :param cursor: Pagination cursor
+        :type cursor: str
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :type _request_timeout: int, tuple(int, int), optional
+        :param _request_auth: set to override the auth_settings for an a single
+                              request; this effectively ignores the
+                              authentication in the spec for a single request.
+        :type _request_auth: dict, optional
+        :param _content_type: force content-type for the request.
+        :type _content_type: str, Optional
+        :param _headers: set to override the headers for a single
+                         request; this effectively ignores the headers
+                         in the spec for a single request.
+        :type _headers: dict, optional
+        :param _host_index: set to override the host_index for a single
+                            request; this effectively ignores the host_index
+                            in the spec for a single request.
+        :type _host_index: int, optional
+        :return: Returns the result object.
+        """ # noqa: E501
+
+        _param = self._list_indexes_collection_serialize(
+            x_database_id=x_database_id,
+            connection_id=connection_id,
+            var_schema=var_schema,
+            table=table,
+            index_type=index_type,
+            limit=limit,
+            cursor=cursor,
+            _request_auth=_request_auth,
+            _content_type=_content_type,
+            _headers=_headers,
+            _host_index=_host_index
+        )
+
+        _response_types_map: Dict[str, Optional[str]] = {
+            '200': "ListIndexesPageResponse",
+            '400': "ApiErrorResponse",
+            '404': "ApiErrorResponse",
+            '500': "ApiErrorResponse",
+        }
+        response_data = self.api_client.call_api(
+            *_param,
+            _request_timeout=_request_timeout
+        )
+        return response_data.response
+
+
+    def _list_indexes_collection_serialize(
+        self,
+        x_database_id,
+        connection_id,
+        var_schema,
+        table,
+        index_type,
+        limit,
+        cursor,
+        _request_auth,
+        _content_type,
+        _headers,
+        _host_index,
+    ) -> RequestSerialized:
+
+        _host = None
+
+        _collection_formats: Dict[str, str] = {
+        }
+
+        _path_params: Dict[str, str] = {}
+        _query_params: List[Tuple[str, str]] = []
+        _header_params: Dict[str, Optional[str]] = _headers or {}
+        _form_params: List[Tuple[str, str]] = []
+        _files: Dict[
+            str, Union[str, bytes, List[str], List[bytes], List[Tuple[str, bytes]]]
+        ] = {}
+        _body_params: Optional[bytes] = None
+
+        # process the path parameters
+        # process the query parameters
+        if connection_id is not None:
+            
+            _query_params.append(('connection_id', connection_id))
+            
+        if var_schema is not None:
+            
+            _query_params.append(('schema', var_schema))
+            
+        if table is not None:
+            
+            _query_params.append(('table', table))
+            
+        if index_type is not None:
+            
+            _query_params.append(('index_type', index_type))
+            
+        if limit is not None:
+            
+            _query_params.append(('limit', limit))
+            
+        if cursor is not None:
+            
+            _query_params.append(('cursor', cursor))
+            
+        # process the header parameters
+        if x_database_id is not None:
+            _header_params['X-Database-Id'] = x_database_id
+        # process the form parameters
+        # process the body parameter
+
+
+        # set the HTTP header `Accept`
+        if 'Accept' not in _header_params:
+            _header_params['Accept'] = self.api_client.select_header_accept(
+                [
+                    'application/json'
+                ]
+            )
+
+
+        # authentication setting
+        _auth_settings: List[str] = [
+            'WorkspaceId', 
+            'BearerAuth'
+        ]
+
+        return self.api_client.param_serialize(
+            method='GET',
+            resource_path='/v1/indexes',
+            path_params=_path_params,
+            query_params=_query_params,
+            header_params=_header_params,
+            body=_body_params,
+            post_params=_form_params,
+            files=_files,
+            auth_settings=_auth_settings,
+            collection_formats=_collection_formats,
+            _host=_host,
+            _request_auth=_request_auth
+        )
+
+

{hotdata-0.3.0 → hotdata-0.4.0}/hotdata/api/query_api.py

@@ -102,6 +102,7 @@ class QueryApi:
             '202': "AsyncQueryResponse",
             '400': "ApiErrorResponse",
             '404': "ApiErrorResponse",
+            '429': "ApiErrorResponse",
             '500': "ApiErrorResponse",
         }
         response_data = self.api_client.call_api(
@@ -177,6 +178,7 @@ class QueryApi:
             '202': "AsyncQueryResponse",
             '400': "ApiErrorResponse",
             '404': "ApiErrorResponse",
+            '429': "ApiErrorResponse",
             '500': "ApiErrorResponse",
         }
         response_data = self.api_client.call_api(
@@ -252,6 +254,7 @@ class QueryApi:
             '202': "AsyncQueryResponse",
             '400': "ApiErrorResponse",
             '404': "ApiErrorResponse",
+            '429': "ApiErrorResponse",
             '500': "ApiErrorResponse",
         }
         response_data = self.api_client.call_api(

{hotdata-0.3.0 → hotdata-0.4.0}/hotdata/api_client.py

@@ -91,7 +91,7 @@ class ApiClient:
             self.default_headers[header_name] = header_value
         self.cookie = cookie
         # Set default User-Agent.
-        self.user_agent = 'OpenAPI-Generator/0.2.6/python'
+        self.user_agent = 'OpenAPI-Generator/0.3.4/python'
         self.client_side_validation = configuration.client_side_validation
 
     def __enter__(self):

{hotdata-0.3.0 → hotdata-0.4.0}/hotdata/arrow.py

@@ -50,6 +50,22 @@ class ResultNotReadyError(Exception):
         )
 
 
+def _release_conn(response: Any) -> None:
+    """Return a urllib3 connection to the pool without poisoning it.
+
+    pyarrow stops reading at the Arrow end-of-stream marker, which can leave
+    bytes unconsumed at the HTTP layer (e.g. the terminating zero-length chunk
+    of a chunked response). ``release_conn`` on a partially-read connection
+    hands a tainted socket back to the pool — the next request to reuse it
+    fails with ``http.client.ResponseNotReady`` ("Request-sent"). Draining any
+    unread bytes first makes the connection safe to reuse.
+    """
+    try:
+        response.drain_conn()
+    finally:
+        response.release_conn()
+
+
 def _import_pyarrow() -> Any:
     try:
         import pyarrow.ipc as ipc  # type: ignore[import-untyped]
@@ -95,7 +111,7 @@ class ResultsApi(_GeneratedResultsApi):
         try:
             return ipc.open_stream(response).read_all()
         finally:
-            response.release_conn()
+            _release_conn(response)
 
     @contextmanager
     def stream_result_arrow(
@@ -108,8 +124,10 @@ class ResultsApi(_GeneratedResultsApi):
     ) -> Iterator["pa.RecordBatchStreamReader"]:
         """Yield a :class:`pyarrow.RecordBatchStreamReader` for a ready result.
 
-        The HTTP connection is released when the context exits. Iterate the
-        reader to consume :class:`pyarrow.RecordBatch` messages, or call
+        The HTTP connection is drained and released when the context exits, so
+        exiting early (before the reader is exhausted) reads and discards the
+        remaining body to keep the connection reusable. Iterate the reader to
+        consume :class:`pyarrow.RecordBatch` messages, or call
         ``reader.read_all()`` for a full :class:`pyarrow.Table`.
 
         Example::
@@ -127,7 +145,7 @@ class ResultsApi(_GeneratedResultsApi):
         try:
             yield ipc.open_stream(response)
         finally:
-            response.release_conn()
+            _release_conn(response)
 
     def _call_arrow(
         self,