diff --git a/.openapi-generator/FILES b/.openapi-generator/FILES index 5572cf2..7a0363a 100644 --- a/.openapi-generator/FILES +++ b/.openapi-generator/FILES @@ -237,7 +237,6 @@ docs/ValidateQueryResponse.md docs/View.md docs/ViewsApi.md docs/VirtualInstance.md -docs/VirtualInstanceRrn.md docs/VirtualInstanceStats.md docs/VirtualInstancesApi.md docs/Workspace.md @@ -488,7 +487,6 @@ rockset/model/user.py rockset/model/validate_query_response.py rockset/model/view.py rockset/model/virtual_instance.py -rockset/model/virtual_instance_rrn.py rockset/model/virtual_instance_stats.py rockset/model/workspace.py rockset/model/xml_params.py @@ -735,7 +733,6 @@ test/test_validate_query_response.py test/test_view.py test/test_views_api.py test/test_virtual_instance.py -test/test_virtual_instance_rrn.py test/test_virtual_instance_stats.py test/test_virtual_instances_api.py test/test_workspace.py diff --git a/HISTORY.md b/HISTORY.md index 8b7af61..1a61ce6 100644 --- a/HISTORY.md +++ b/HISTORY.md @@ -1,3 +1,6 @@ +### v2.2.0 + - TODO + ### v2.1.0 - Get Collection Offset APIs diff --git a/README.md b/README.md index 8fa992d..4cde0bb 100644 --- a/README.md +++ b/README.md @@ -474,7 +474,6 @@ Class | Method | HTTP request | Description - [ValidateQueryResponse](docs/ValidateQueryResponse.md) - [View](docs/View.md) - [VirtualInstance](docs/VirtualInstance.md) - - [VirtualInstanceRrn](docs/VirtualInstanceRrn.md) - [VirtualInstanceStats](docs/VirtualInstanceStats.md) - [Workspace](docs/Workspace.md) - [XmlParams](docs/XmlParams.md) diff --git a/docs/CreateIntegrationRequest.md b/docs/CreateIntegrationRequest.md index 5b855f2..1d9b090 100644 --- a/docs/CreateIntegrationRequest.md +++ b/docs/CreateIntegrationRequest.md @@ -11,6 +11,7 @@ Name | Type | Description | Notes **description** | **str** | Longer explanation for the integration. | [optional] **dynamodb** | [**DynamodbIntegration**](DynamodbIntegration.md) | | [optional] **gcs** | [**GcsIntegration**](GcsIntegration.md) | | [optional] +**is_write_enabled** | **bool** | is write access enabled for this integration. | [optional] **kafka** | [**KafkaIntegration**](KafkaIntegration.md) | | [optional] **kinesis** | [**KinesisIntegration**](KinesisIntegration.md) | | [optional] **mongodb** | [**MongoDbIntegration**](MongoDbIntegration.md) | | [optional] diff --git a/docs/Integration.md b/docs/Integration.md index dd96a78..22f6bdc 100644 --- a/docs/Integration.md +++ b/docs/Integration.md @@ -16,6 +16,7 @@ Name | Type | Description | Notes **description** | **str** | Longer explanation for the integration. | [optional] **dynamodb** | [**DynamodbIntegration**](DynamodbIntegration.md) | | [optional] **gcs** | [**GcsIntegration**](GcsIntegration.md) | | [optional] +**is_write_enabled** | **bool** | is write access enabled for this integration | [optional] **kafka** | [**KafkaIntegration**](KafkaIntegration.md) | | [optional] **kinesis** | [**KinesisIntegration**](KinesisIntegration.md) | | [optional] **mongodb** | [**MongoDbIntegration**](MongoDbIntegration.md) | | [optional] diff --git a/docs/IntegrationsApi.md b/docs/IntegrationsApi.md index 02067f0..2017327 100644 --- a/docs/IntegrationsApi.md +++ b/docs/IntegrationsApi.md @@ -1241,6 +1241,7 @@ SharedAccessSignature=''', service_account_key_file_json="service_account_key_file_json_example", ), ), + is_write_enabled=True, kafka=KafkaIntegration( aws_role=AwsRole( aws_external_id="external id of aws", @@ -1334,6 +1335,7 @@ Name | Type | Description | Notes **description** | **str** | Longer explanation for the integration. | [optional] **dynamodb** | [**DynamodbIntegration**](DynamodbIntegration.md) | | [optional] **gcs** | [**GcsIntegration**](GcsIntegration.md) | | [optional] + **is_write_enabled** | **bool** | is write access enabled for this integration. | [optional] **kafka** | [**KafkaIntegration**](KafkaIntegration.md) | | [optional] **kinesis** | [**KinesisIntegration**](KinesisIntegration.md) | | [optional] **mongodb** | [**MongoDbIntegration**](MongoDbIntegration.md) | | [optional] diff --git a/docs/ScheduledLambda.md b/docs/ScheduledLambda.md index 599c868..9f82fa1 100644 --- a/docs/ScheduledLambda.md +++ b/docs/ScheduledLambda.md @@ -16,7 +16,6 @@ Name | Type | Description | Notes **tag** | **str** | The query lambda tag. | [optional] **total_times_to_execute** | **int** | The number of times to execute this scheduled query lambda. Once this scheduled query lambda has been executed this many times, it will no longer be executed. | [optional] **version** | **str** | The version of the associated query lambda. | [optional] -**vi_rrn** | [**VirtualInstanceRrn**](VirtualInstanceRrn.md) | | [optional] **webhook_payload** | **str** | The payload that should be sent to the webhook. | [optional] **webhook_url** | **str** | The URL of the webhook that should be triggered after this scheduled query lambda completes. | [optional] **workspace** | **str** | Workspace of the associated query lambda. | [optional] diff --git a/docs/Source.md b/docs/Source.md index e2a742b..3377868 100644 --- a/docs/Source.md +++ b/docs/Source.md @@ -25,6 +25,7 @@ Name | Type | Description | Notes **status** | **bool, date, datetime, dict, float, int, list, str, none_type** | | [optional] [readonly] **suspended_at** | **str** | ISO-8601 date when source was suspended, if suspended | [optional] [readonly] **system** | [**SourceSystem**](SourceSystem.md) | | [optional] +**write_api** | **{str: (bool, date, datetime, dict, float, int, list, str, none_type)}** | | [optional] **any string name** | **bool, date, datetime, dict, float, int, list, str, none_type** | any string name can be used but the value must be the correct type | [optional] [[Back to Model list]](../README.md#documentation-for-models) [[Back to API list]](../README.md#documentation-for-api-endpoints) [[Back to README]](../README.md) diff --git a/docs/Stats.md b/docs/Stats.md index ab4b91a..90bd460 100644 --- a/docs/Stats.md +++ b/docs/Stats.md @@ -7,6 +7,7 @@ Name | Type | Description | Notes **elapsed_time_ms** | **int** | Total execution time (including time queued) of the query, in milliseconds. | [optional] **result_set_bytes_size** | **int** | Number of bytes in the query result set. Only populated if `status` is `COMPLETE`. Not populated for INSERT INTO queries. | [optional] **result_set_document_count** | **int** | Number of documents returned by the query. Only populated if `status` is `COMPLETE`. | [optional] +**result_set_file_count** | **int** | Number of files written by by the query. Only populated if `status` is `COMPLETE` and the query is an export query. | [optional] **throttled_time_ms** | **int** | Time query spent queued, in milliseconds. | [optional] **any string name** | **bool, date, datetime, dict, float, int, list, str, none_type** | any string name can be used but the value must be the correct type | [optional] diff --git a/docs/UpdateIntegrationRequest.md b/docs/UpdateIntegrationRequest.md index 44c7df4..bfdaebe 100644 --- a/docs/UpdateIntegrationRequest.md +++ b/docs/UpdateIntegrationRequest.md @@ -10,6 +10,7 @@ Name | Type | Description | Notes **description** | **str** | Longer explanation for the integration. | [optional] **dynamodb** | [**DynamodbIntegration**](DynamodbIntegration.md) | | [optional] **gcs** | [**GcsIntegration**](GcsIntegration.md) | | [optional] +**is_write_enabled** | **bool** | is write access enabled for this integration. | [optional] **kafka** | [**KafkaIntegration**](KafkaIntegration.md) | | [optional] **kinesis** | [**KinesisIntegration**](KinesisIntegration.md) | | [optional] **mongodb** | [**MongoDbIntegration**](MongoDbIntegration.md) | | [optional] diff --git a/rockset/api/integrations_api.py b/rockset/api/integrations_api.py index 4428cd0..cf894f4 100644 --- a/rockset/api/integrations_api.py +++ b/rockset/api/integrations_api.py @@ -1881,6 +1881,7 @@ def update( description: str = None, dynamodb: DynamodbIntegration = None, gcs: GcsIntegration = None, + is_write_enabled: bool = None, kafka: KafkaIntegration = None, kinesis: KinesisIntegration = None, mongodb: MongoDbIntegration = None, @@ -1925,6 +1926,7 @@ def update( service_account_key_file_json="service_account_key_file_json_example", ), ), + is_write_enabled=True, kafka=KafkaIntegration( aws_role=AwsRole( aws_external_id="external id of aws", @@ -2010,6 +2012,7 @@ def update( description (str): Longer explanation for the integration.. [optional] dynamodb (DynamodbIntegration): [optional] gcs (GcsIntegration): [optional] + is_write_enabled (bool): is write access enabled for this integration.. [optional] kafka (KafkaIntegration): [optional] kinesis (KinesisIntegration): [optional] mongodb (MongoDbIntegration): [optional] diff --git a/rockset/model/create_integration_request.py b/rockset/model/create_integration_request.py index bb0efd4..7400641 100644 --- a/rockset/model/create_integration_request.py +++ b/rockset/model/create_integration_request.py @@ -112,6 +112,7 @@ def openapi_types(): 'description': (str, none_type), # noqa: E501 'dynamodb': (DynamodbIntegration, none_type), # noqa: E501 'gcs': (GcsIntegration, none_type), # noqa: E501 + 'is_write_enabled': (bool, none_type), # noqa: E501 'kafka': (KafkaIntegration, none_type), # noqa: E501 'kinesis': (KinesisIntegration, none_type), # noqa: E501 'mongodb': (MongoDbIntegration, none_type), # noqa: E501 @@ -132,6 +133,7 @@ def discriminator(): 'description': 'description', # noqa: E501 'dynamodb': 'dynamodb', # noqa: E501 'gcs': 'gcs', # noqa: E501 + 'is_write_enabled': 'is_write_enabled', # noqa: E501 'kafka': 'kafka', # noqa: E501 'kinesis': 'kinesis', # noqa: E501 'mongodb': 'mongodb', # noqa: E501 @@ -189,6 +191,7 @@ def _from_openapi_data(cls, name, *args, **kwargs): # noqa: E501 description (str): Longer explanation for the integration.. [optional] # noqa: E501 dynamodb (DynamodbIntegration): [optional] # noqa: E501 gcs (GcsIntegration): [optional] # noqa: E501 + is_write_enabled (bool): is write access enabled for this integration.. [optional] # noqa: E501 kafka (KafkaIntegration): [optional] # noqa: E501 kinesis (KinesisIntegration): [optional] # noqa: E501 mongodb (MongoDbIntegration): [optional] # noqa: E501 @@ -253,6 +256,7 @@ def __init__(self, *, name, **kwargs): # noqa: E501 description (str): Longer explanation for the integration.. [optional] # noqa: E501 dynamodb (DynamodbIntegration): [optional] # noqa: E501 gcs (GcsIntegration): [optional] # noqa: E501 + is_write_enabled (bool): is write access enabled for this integration.. [optional] # noqa: E501 kafka (KafkaIntegration): [optional] # noqa: E501 kinesis (KinesisIntegration): [optional] # noqa: E501 mongodb (MongoDbIntegration): [optional] # noqa: E501 diff --git a/rockset/model/integration.py b/rockset/model/integration.py index 116c2aa..7ec6ff8 100644 --- a/rockset/model/integration.py +++ b/rockset/model/integration.py @@ -118,6 +118,7 @@ def openapi_types(): 'description': (str, none_type), # noqa: E501 'dynamodb': (DynamodbIntegration, none_type), # noqa: E501 'gcs': (GcsIntegration, none_type), # noqa: E501 + 'is_write_enabled': (bool, none_type), # noqa: E501 'kafka': (KafkaIntegration, none_type), # noqa: E501 'kinesis': (KinesisIntegration, none_type), # noqa: E501 'mongodb': (MongoDbIntegration, none_type), # noqa: E501 @@ -143,6 +144,7 @@ def discriminator(): 'description': 'description', # noqa: E501 'dynamodb': 'dynamodb', # noqa: E501 'gcs': 'gcs', # noqa: E501 + 'is_write_enabled': 'is_write_enabled', # noqa: E501 'kafka': 'kafka', # noqa: E501 'kinesis': 'kinesis', # noqa: E501 'mongodb': 'mongodb', # noqa: E501 @@ -205,6 +207,7 @@ def _from_openapi_data(cls, created_by, name, *args, **kwargs): # noqa: E501 description (str): Longer explanation for the integration.. [optional] # noqa: E501 dynamodb (DynamodbIntegration): [optional] # noqa: E501 gcs (GcsIntegration): [optional] # noqa: E501 + is_write_enabled (bool): is write access enabled for this integration. [optional] # noqa: E501 kafka (KafkaIntegration): [optional] # noqa: E501 kinesis (KinesisIntegration): [optional] # noqa: E501 mongodb (MongoDbIntegration): [optional] # noqa: E501 @@ -275,6 +278,7 @@ def __init__(self, *, created_by, name, **kwargs): # noqa: E501 description (str): Longer explanation for the integration.. [optional] # noqa: E501 dynamodb (DynamodbIntegration): [optional] # noqa: E501 gcs (GcsIntegration): [optional] # noqa: E501 + is_write_enabled (bool): is write access enabled for this integration. [optional] # noqa: E501 kafka (KafkaIntegration): [optional] # noqa: E501 kinesis (KinesisIntegration): [optional] # noqa: E501 mongodb (MongoDbIntegration): [optional] # noqa: E501 diff --git a/rockset/model/privilege.py b/rockset/model/privilege.py index 6decf5a..d515c2a 100644 --- a/rockset/model/privilege.py +++ b/rockset/model/privilege.py @@ -114,7 +114,7 @@ class Privilege(ModelNormal): 'CREATE_NETWORK_CONFIGURATION_GLOBAL': "CREATE_NETWORK_CONFIGURATION_GLOBAL", 'DELETE_NETWORK_CONFIGURATION_GLOBAL': "DELETE_NETWORK_CONFIGURATION_GLOBAL", 'LIST_NETWORK_CONFIGURATIONS_GLOBAL': "LIST_NETWORK_CONFIGURATIONS_GLOBAL", - 'UPDATE_ORG_QUERY_ROUTING': "UPDATE_ORG_QUERY_ROUTING", + 'UPDATE_DEPLOYMENT_QUERY_ROUTING': "UPDATE_DEPLOYMENT_QUERY_ROUTING", }, } diff --git a/rockset/model/scheduled_lambda.py b/rockset/model/scheduled_lambda.py index 4f644a2..f4ef4bf 100644 --- a/rockset/model/scheduled_lambda.py +++ b/rockset/model/scheduled_lambda.py @@ -31,9 +31,7 @@ def lazy_import(): from rockset.model.execution_status import ExecutionStatus - from rockset.model.virtual_instance_rrn import VirtualInstanceRrn globals()['ExecutionStatus'] = ExecutionStatus - globals()['VirtualInstanceRrn'] = VirtualInstanceRrn class ScheduledLambda(ModelNormal): @@ -101,7 +99,6 @@ def openapi_types(): 'tag': (str, none_type), # noqa: E501 'total_times_to_execute': (int, none_type), # noqa: E501 'version': (str, none_type), # noqa: E501 - 'vi_rrn': (VirtualInstanceRrn, none_type), # noqa: E501 'webhook_payload': (str, none_type), # noqa: E501 'webhook_url': (str, none_type), # noqa: E501 'workspace': (str, none_type), # noqa: E501 @@ -125,7 +122,6 @@ def discriminator(): 'tag': 'tag', # noqa: E501 'total_times_to_execute': 'total_times_to_execute', # noqa: E501 'version': 'version', # noqa: E501 - 'vi_rrn': 'vi_rrn', # noqa: E501 'webhook_payload': 'webhook_payload', # noqa: E501 'webhook_url': 'webhook_url', # noqa: E501 'workspace': 'workspace', # noqa: E501 @@ -184,7 +180,6 @@ def _from_openapi_data(cls, *args, **kwargs): # noqa: E501 tag (str): The query lambda tag.. [optional] # noqa: E501 total_times_to_execute (int): The number of times to execute this scheduled query lambda. Once this scheduled query lambda has been executed this many times, it will no longer be executed.. [optional] # noqa: E501 version (str): The version of the associated query lambda.. [optional] # noqa: E501 - vi_rrn (VirtualInstanceRrn): [optional] # noqa: E501 webhook_payload (str): The payload that should be sent to the webhook.. [optional] # noqa: E501 webhook_url (str): The URL of the webhook that should be triggered after this scheduled query lambda completes.. [optional] # noqa: E501 workspace (str): Workspace of the associated query lambda.. [optional] # noqa: E501 @@ -251,7 +246,6 @@ def __init__(self, **kwargs): # noqa: E501 tag (str): The query lambda tag.. [optional] # noqa: E501 total_times_to_execute (int): The number of times to execute this scheduled query lambda. Once this scheduled query lambda has been executed this many times, it will no longer be executed.. [optional] # noqa: E501 version (str): The version of the associated query lambda.. [optional] # noqa: E501 - vi_rrn (VirtualInstanceRrn): [optional] # noqa: E501 webhook_payload (str): The payload that should be sent to the webhook.. [optional] # noqa: E501 webhook_url (str): The URL of the webhook that should be triggered after this scheduled query lambda completes.. [optional] # noqa: E501 workspace (str): Workspace of the associated query lambda.. [optional] # noqa: E501 diff --git a/rockset/model/source.py b/rockset/model/source.py index 6d22c1f..6f0ae7d 100644 --- a/rockset/model/source.py +++ b/rockset/model/source.py @@ -137,6 +137,7 @@ def openapi_types(): 'status': (bool, date, datetime, dict, float, int, list, str, none_type, none_type), # noqa: E501 'suspended_at': (str, none_type), # noqa: E501 'system': (SourceSystem, none_type), # noqa: E501 + 'write_api': ({str: (bool, date, datetime, dict, float, int, list, str, none_type)}, none_type), # noqa: E501 } @cached_property @@ -165,6 +166,7 @@ def discriminator(): 'status': 'status', # noqa: E501 'suspended_at': 'suspended_at', # noqa: E501 'system': 'system', # noqa: E501 + 'write_api': 'write_api', # noqa: E501 } read_only_vars = { @@ -233,6 +235,7 @@ def _from_openapi_data(cls, *args, **kwargs): # noqa: E501 status (bool, date, datetime, dict, float, int, list, str, none_type): [optional] # noqa: E501 suspended_at (str): ISO-8601 date when source was suspended, if suspended. [optional] # noqa: E501 system (SourceSystem): [optional] # noqa: E501 + write_api ({str: (bool, date, datetime, dict, float, int, list, str, none_type)}): [optional] # noqa: E501 """ _check_type = kwargs.pop('_check_type', True) @@ -299,6 +302,7 @@ def __init__(self, **kwargs): # noqa: E501 snapshot (SourceSnapshot): [optional] # noqa: E501 snowflake (SourceSnowflake): [optional] # noqa: E501 system (SourceSystem): [optional] # noqa: E501 + write_api ({str: (bool, date, datetime, dict, float, int, list, str, none_type)}): [optional] # noqa: E501 _check_type (bool): if True, values for parameters in openapi_types will be type checked and a TypeError will be raised if the wrong type is input. diff --git a/rockset/model/stats.py b/rockset/model/stats.py index 23a441b..dc162af 100644 --- a/rockset/model/stats.py +++ b/rockset/model/stats.py @@ -84,6 +84,7 @@ def openapi_types(): 'elapsed_time_ms': (int, none_type), # noqa: E501 'result_set_bytes_size': (int, none_type), # noqa: E501 'result_set_document_count': (int, none_type), # noqa: E501 + 'result_set_file_count': (int, none_type), # noqa: E501 'throttled_time_ms': (int, none_type), # noqa: E501 } @@ -96,6 +97,7 @@ def discriminator(): 'elapsed_time_ms': 'elapsed_time_ms', # noqa: E501 'result_set_bytes_size': 'result_set_bytes_size', # noqa: E501 'result_set_document_count': 'result_set_document_count', # noqa: E501 + 'result_set_file_count': 'result_set_file_count', # noqa: E501 'throttled_time_ms': 'throttled_time_ms', # noqa: E501 } @@ -143,6 +145,7 @@ def _from_openapi_data(cls, *args, **kwargs): # noqa: E501 elapsed_time_ms (int): Total execution time (including time queued) of the query, in milliseconds.. [optional] # noqa: E501 result_set_bytes_size (int): Number of bytes in the query result set. Only populated if `status` is `COMPLETE`. Not populated for INSERT INTO queries.. [optional] # noqa: E501 result_set_document_count (int): Number of documents returned by the query. Only populated if `status` is `COMPLETE`.. [optional] # noqa: E501 + result_set_file_count (int): Number of files written by by the query. Only populated if `status` is `COMPLETE` and the query is an export query.. [optional] # noqa: E501 throttled_time_ms (int): Time query spent queued, in milliseconds.. [optional] # noqa: E501 """ @@ -198,6 +201,7 @@ def __init__(self, **kwargs): # noqa: E501 elapsed_time_ms (int): Total execution time (including time queued) of the query, in milliseconds.. [optional] # noqa: E501 result_set_bytes_size (int): Number of bytes in the query result set. Only populated if `status` is `COMPLETE`. Not populated for INSERT INTO queries.. [optional] # noqa: E501 result_set_document_count (int): Number of documents returned by the query. Only populated if `status` is `COMPLETE`.. [optional] # noqa: E501 + result_set_file_count (int): Number of files written by by the query. Only populated if `status` is `COMPLETE` and the query is an export query.. [optional] # noqa: E501 throttled_time_ms (int): Time query spent queued, in milliseconds.. [optional] # noqa: E501 _check_type (bool): if True, values for parameters in openapi_types will be type checked and a TypeError will be diff --git a/rockset/model/update_integration_request.py b/rockset/model/update_integration_request.py index 99c7ef9..b7d47a4 100644 --- a/rockset/model/update_integration_request.py +++ b/rockset/model/update_integration_request.py @@ -111,6 +111,7 @@ def openapi_types(): 'description': (str, none_type), # noqa: E501 'dynamodb': (DynamodbIntegration, none_type), # noqa: E501 'gcs': (GcsIntegration, none_type), # noqa: E501 + 'is_write_enabled': (bool, none_type), # noqa: E501 'kafka': (KafkaIntegration, none_type), # noqa: E501 'kinesis': (KinesisIntegration, none_type), # noqa: E501 'mongodb': (MongoDbIntegration, none_type), # noqa: E501 @@ -130,6 +131,7 @@ def discriminator(): 'description': 'description', # noqa: E501 'dynamodb': 'dynamodb', # noqa: E501 'gcs': 'gcs', # noqa: E501 + 'is_write_enabled': 'is_write_enabled', # noqa: E501 'kafka': 'kafka', # noqa: E501 'kinesis': 'kinesis', # noqa: E501 'mongodb': 'mongodb', # noqa: E501 @@ -184,6 +186,7 @@ def _from_openapi_data(cls, *args, **kwargs): # noqa: E501 description (str): Longer explanation for the integration.. [optional] # noqa: E501 dynamodb (DynamodbIntegration): [optional] # noqa: E501 gcs (GcsIntegration): [optional] # noqa: E501 + is_write_enabled (bool): is write access enabled for this integration.. [optional] # noqa: E501 kafka (KafkaIntegration): [optional] # noqa: E501 kinesis (KinesisIntegration): [optional] # noqa: E501 mongodb (MongoDbIntegration): [optional] # noqa: E501 @@ -246,6 +249,7 @@ def __init__(self, **kwargs): # noqa: E501 description (str): Longer explanation for the integration.. [optional] # noqa: E501 dynamodb (DynamodbIntegration): [optional] # noqa: E501 gcs (GcsIntegration): [optional] # noqa: E501 + is_write_enabled (bool): is write access enabled for this integration.. [optional] # noqa: E501 kafka (KafkaIntegration): [optional] # noqa: E501 kinesis (KinesisIntegration): [optional] # noqa: E501 mongodb (MongoDbIntegration): [optional] # noqa: E501 diff --git a/rockset/models/__init__.py b/rockset/models/__init__.py index 2bd043f..873b060 100644 --- a/rockset/models/__init__.py +++ b/rockset/models/__init__.py @@ -233,7 +233,6 @@ from rockset.model.validate_query_response import ValidateQueryResponse from rockset.model.view import View from rockset.model.virtual_instance import VirtualInstance -from rockset.model.virtual_instance_rrn import VirtualInstanceRrn from rockset.model.virtual_instance_stats import VirtualInstanceStats from rockset.model.workspace import Workspace from rockset.model.xml_params import XmlParams diff --git a/swagger/modified-openapi.yaml b/swagger/modified-openapi.yaml index 83a9d94..c013dd8 100644 --- a/swagger/modified-openapi.yaml +++ b/swagger/modified-openapi.yaml @@ -990,6 +990,9 @@ components: gcs: $ref: "#/components/schemas/GcsIntegration" description: GCS details. + is_write_enabled: + description: is write access enabled for this integration. + type: boolean kafka: $ref: "#/components/schemas/KafkaIntegration" kinesis: @@ -2034,6 +2037,9 @@ components: gcs: $ref: "#/components/schemas/GcsIntegration" description: GCS details. + is_write_enabled: + description: is write access enabled for this integration + type: boolean kafka: $ref: "#/components/schemas/KafkaIntegration" description: Kafka details. @@ -2769,7 +2775,7 @@ components: - CREATE_NETWORK_CONFIGURATION_GLOBAL - DELETE_NETWORK_CONFIGURATION_GLOBAL - LIST_NETWORK_CONFIGURATIONS_GLOBAL - - UPDATE_ORG_QUERY_ROUTING + - UPDATE_DEPLOYMENT_QUERY_ROUTING example: CREATE_COLLECTION_WS type: string cluster: @@ -3397,12 +3403,6 @@ components: description: The version of the associated query lambda. example: abcde1234 type: string - vi_rrn: - $ref: "#/components/schemas/VirtualInstanceRrn" - description: - The RRN of the virtual instance that should be used to execute - this scheduled query lambda. - example: rrn:vi:use1a1:123e4567-e89b-12d3-a456-556642440000 webhook_payload: description: The payload that should be sent to the webhook. example: @@ -3639,6 +3639,8 @@ components: type: string system: $ref: "#/components/schemas/SourceSystem" + write_api: + $ref: "#/components/schemas/SourceWriteApi" type: object SourceAzBlobStorageBase: properties: @@ -4093,6 +4095,8 @@ components: example: commons type: string type: object + SourceWriteApi: + type: object SqlExpression: properties: sql: @@ -4123,6 +4127,13 @@ components: example: 1235 format: int64 type: integer + result_set_file_count: + description: + Number of files written by by the query. Only populated if + `status` is `COMPLETE` and the query is an export query. + example: 1235 + format: int64 + type: integer throttled_time_ms: description: Time query spent queued, in milliseconds. example: 16 @@ -4554,6 +4565,9 @@ components: gcs: $ref: "#/components/schemas/GcsIntegration" description: GCS details. + is_write_enabled: + description: is write access enabled for this integration. + type: boolean kafka: $ref: "#/components/schemas/KafkaIntegration" kinesis: @@ -5013,65 +5027,6 @@ components: required: - name type: object - VirtualInstanceRrn: - properties: - cluster: - enum: - - APT2A1 - - DEV - - DEV_USW2A1 - - EUC1A1 - - EUW1A1 - - GLOBAL - - TEST - - UNIT_TEST - - UNKNOWN - - USE1A1 - - USW2A1 - type: string - resourceType: - enum: - - ALIAS - - CLUSTER - - COLLECTION - - COLLECTION_STATS - - COMPACTION_INFO - - DEPLOYMENT_STATS - - FIELD_SCHEMA - - INTEGRATION - - IP_ALLOWLIST - - LEAF - - LEAF_GROUP - - MOUNT - - NETWORK_CONFIGURATION - - ORG - - ORG_MIGRATION - - PENDING_SNAPSHOTS - - PREVIEW_SOURCE - - QUERY_LAMBDA - - QUERY_LAMBDA_TAG - - QUERY_LAMBDA_VERSION - - ROCKSET_DEPLOYMENT - - ROLE - - ROOT - - SCHEDULED_QUERY_LAMBDA - - SHARD - - SHARD_BATCH - - SIMILARITY_INDEX - - SNAPSHOT - - SOURCE - - USER - - VIEW - - VIRTUAL_INSTANCE - - WORKSPACE - type: string - uuid: - format: uuid - type: string - version: - format: int32 - type: integer - type: object VirtualInstanceStats: properties: last_queried_ms: @@ -18336,6 +18291,7 @@ tags: - name: API Keys - name: Collections - name: Custom Roles + - name: Deployment Settings - name: Documents - name: Integrations - name: Organizations diff --git a/swagger/openapi-generated.yaml b/swagger/openapi-generated.yaml index 59ecc6e..8b5613e 100644 --- a/swagger/openapi-generated.yaml +++ b/swagger/openapi-generated.yaml @@ -11319,6 +11319,9 @@ components: gcs: $ref: "#/components/schemas/GcsIntegration" description: GCS details. + is_write_enabled: + description: is write access enabled for this integration. + type: boolean kafka: $ref: "#/components/schemas/KafkaIntegration" kinesis: @@ -12179,6 +12182,9 @@ components: gcs: $ref: "#/components/schemas/GcsIntegration" description: GCS details. + is_write_enabled: + description: is write access enabled for this integration + type: boolean kafka: $ref: "#/components/schemas/KafkaIntegration" description: Kafka details. @@ -12650,7 +12656,7 @@ components: - CREATE_NETWORK_CONFIGURATION_GLOBAL - DELETE_NETWORK_CONFIGURATION_GLOBAL - LIST_NETWORK_CONFIGURATIONS_GLOBAL - - UPDATE_ORG_QUERY_ROUTING + - UPDATE_DEPLOYMENT_QUERY_ROUTING example: CREATE_COLLECTION_WS type: string cluster: @@ -13193,12 +13199,6 @@ components: description: The version of the associated query lambda. example: abcde1234 type: string - vi_rrn: - $ref: "#/components/schemas/VirtualInstanceRrn" - description: >- - The RRN of the virtual instance that should be used to execute this - scheduled query lambda. - example: "rrn:vi:use1a1:123e4567-e89b-12d3-a456-556642440000" webhook_payload: description: The payload that should be sent to the webhook. example: >- @@ -13345,6 +13345,8 @@ components: type: string system: $ref: "#/components/schemas/SourceSystem" + write_api: + $ref: "#/components/schemas/SourceWriteApi" type: object SourceAzBlobStorageBase: properties: @@ -13787,6 +13789,8 @@ components: example: commons type: string type: object + SourceWriteApi: + type: object SqlExpression: properties: sql: @@ -13817,6 +13821,13 @@ components: example: 1235 format: int64 type: integer + result_set_file_count: + description: >- + Number of files written by by the query. Only populated if `status` + is `COMPLETE` and the query is an export query. + example: 1235 + format: int64 + type: integer throttled_time_ms: description: "Time query spent queued, in milliseconds." example: 16 @@ -14243,6 +14254,9 @@ components: gcs: $ref: "#/components/schemas/GcsIntegration" description: GCS details. + is_write_enabled: + description: is write access enabled for this integration. + type: boolean kafka: $ref: "#/components/schemas/KafkaIntegration" kinesis: @@ -14694,65 +14708,6 @@ components: required: - name type: object - VirtualInstanceRrn: - properties: - cluster: - enum: - - APT2A1 - - DEV - - DEV_USW2A1 - - EUC1A1 - - EUW1A1 - - GLOBAL - - TEST - - UNIT_TEST - - UNKNOWN - - USE1A1 - - USW2A1 - type: string - resourceType: - enum: - - ALIAS - - CLUSTER - - COLLECTION - - COLLECTION_STATS - - COMPACTION_INFO - - DEPLOYMENT_STATS - - FIELD_SCHEMA - - INTEGRATION - - IP_ALLOWLIST - - LEAF - - LEAF_GROUP - - MOUNT - - NETWORK_CONFIGURATION - - ORG - - ORG_MIGRATION - - PENDING_SNAPSHOTS - - PREVIEW_SOURCE - - QUERY_LAMBDA - - QUERY_LAMBDA_TAG - - QUERY_LAMBDA_VERSION - - ROCKSET_DEPLOYMENT - - ROLE - - ROOT - - SCHEDULED_QUERY_LAMBDA - - SHARD - - SHARD_BATCH - - SIMILARITY_INDEX - - SNAPSHOT - - SOURCE - - USER - - VIEW - - VIRTUAL_INSTANCE - - WORKSPACE - type: string - uuid: - format: uuid - type: string - version: - format: int32 - type: integer - type: object VirtualInstanceStats: properties: last_queried_ms: @@ -14816,6 +14771,7 @@ tags: - name: API Keys - name: Collections - name: Custom Roles + - name: Deployment Settings - name: Documents - name: Integrations - name: Organizations diff --git a/test/test_integrations_api.py b/test/test_integrations_api.py index 721f92a..82cfa93 100644 --- a/test/test_integrations_api.py +++ b/test/test_integrations_api.py @@ -285,6 +285,7 @@ def test_update(get_client, mock_request, request_validator): service_account_key_file_json="service_account_key_file_json_example", ), ), + is_write_enabled=True, kafka=KafkaIntegration( aws_role=AwsRole( aws_external_id="external id of aws", diff --git a/test/test_virtual_instance_rrn.py b/test/test_virtual_instance_rrn.py deleted file mode 100644 index 20b773f..0000000 --- a/test/test_virtual_instance_rrn.py +++ /dev/null @@ -1,16 +0,0 @@ -""" - REST API - - Rockset's REST API allows for creating and managing all resources in Rockset. Each supported endpoint is documented below. All requests must be authorized with a Rockset API key, which can be created in the [Rockset console](https://console.rockset.com). The API key must be provided as `ApiKey ` in the `Authorization` request header. For example: ``` Authorization: ApiKey aB35kDjg93J5nsf4GjwMeErAVd832F7ad4vhsW1S02kfZiab42sTsfW5Sxt25asT ``` All endpoints are only accessible via https. Build something awesome! # noqa: E501 - - The version of the OpenAPI document: v1 - Generated by: https://openapi-generator.tech -""" - - -import sys -from rockset.models import * - - -def test_virtual_instance_rrn_init(): - VirtualInstanceRrn()