SK-2353: Update generated code with updated openapi spec file.

.github/workflows/shared-build-and-deploy.yml

@@ -27,6 +27,11 @@ jobs:
           python -m pip install --upgrade pip
           pip install setuptools wheel twine
 
+      - name: Build and install skyflow package
+        run: |
+          python setup.py sdist bdist_wheel
+          pip install dist/skyflow-*.whl
+
       - name: Resolve Branch for the Tagged Commit
         id: resolve-branch
         if: ${{ inputs.tag == 'beta' || inputs.tag == 'public' }} 

.github/workflows/shared-tests.yml

@@ -23,6 +23,12 @@ jobs:
         with:
           name: "credentials.json"
           json: ${{ secrets.VALID_SKYFLOW_CREDS_TEST }}
+
+      - name: Build and install skyflow package
+        run: |
+          pip install --upgrade pip setuptools wheel
+          python setup.py sdist bdist_wheel
+          pip install dist/skyflow-*.whl
 
       - name: 'Run Tests'
         run: |

skyflow/generated/rest/__init__.py

@@ -3,49 +3,69 @@
 # isort: skip_file
 
 from .types import (
-    AllowRegex,
     AuditEventAuditResourceType,
     AuditEventContext,
     AuditEventData,
     AuditEventHttpInfo,
     BatchRecordMethod,
-    CheckGuardrailsResponse,
-    CheckGuardrailsResponseValidation,
-    ConfigurationId,
     ContextAccessType,
     ContextAuthMode,
-    DeidentifyFileOutput,
-    DeidentifyFileOutputProcessedFileType,
+    DeidentifiedFileOutput,
+    DeidentifiedFileOutputProcessedFileExtension,
+    DeidentifiedFileOutputProcessedFileType,
     DeidentifyFileResponse,
-    DeidentifyStatusResponse,
-    DeidentifyStatusResponseOutputType,
-    DeidentifyStatusResponseStatus,
     DeidentifyStringResponse,
-    DetectedEntity,
+    DetectGuardrailsResponse,
+    DetectGuardrailsResponseValidation,
+    DetectRunsResponse,
+    DetectRunsResponseOutputType,
+    DetectRunsResponseStatus,
     DetokenizeRecordResponseValueType,
-    EntityLocation,
-    EntityType,
-    EntityTypes,
     ErrorResponse,
     ErrorResponseError,
-    ErrorString,
+    FileData,
+    FileDataDataFormat,
+    FileDataDeidentifyAudio,
+    FileDataDeidentifyAudioDataFormat,
+    FileDataDeidentifyDocument,
+    FileDataDeidentifyDocumentDataFormat,
+    FileDataDeidentifyImage,
+    FileDataDeidentifyImageDataFormat,
+    FileDataDeidentifyPdf,
+    FileDataDeidentifyPresentation,
+    FileDataDeidentifyPresentationDataFormat,
+    FileDataDeidentifySpreadsheet,
+    FileDataDeidentifySpreadsheetDataFormat,
+    FileDataDeidentifyStructuredText,
+    FileDataDeidentifyStructuredTextDataFormat,
+    FileDataDeidentifyText,
+    FileDataReidentifyFile,
+    FileDataReidentifyFileDataFormat,
+    Format,
+    FormatMaskedItem,
+    FormatPlaintextItem,
+    FormatRedactedItem,
     GooglerpcStatus,
+    HttpCode,
+    IdentifyResponse,
     ProtobufAny,
     RedactionEnumRedaction,
+    ReidentifiedFileOutput,
+    ReidentifiedFileOutputProcessedFileExtension,
     ReidentifyFileResponse,
-    ReidentifyFileResponseOutput,
+    ReidentifyFileResponseOutputType,
     ReidentifyFileResponseStatus,
-    ReidentifyStringResponse,
     RequestActionType,
     ResourceId,
-    RestrictRegex,
-    TokenType,
-    TokenTypeDefault,
-    TokenTypeWithoutVault,
-    TokenTypeWithoutVaultDefault,
+    ShiftDates,
+    ShiftDatesEntityTypesItem,
+    StringResponseEntities,
+    TokenTypeMapping,
+    TokenTypeMappingDefault,
+    TokenTypeMappingEntityOnlyItem,
+    TokenTypeMappingEntityUnqCounterItem,
+    TokenTypeMappingVaultTokenItem,
     Transformations,
-    TransformationsShiftDates,
-    TransformationsShiftDatesEntityTypesItem,
     UploadFileV2Response,
     Uuid,
     V1AuditAfterOptions,
@@ -79,7 +99,7 @@
     V1UpdateRecordResponse,
     V1VaultFieldMapping,
     V1VaultSchemaConfig,
-    VaultId,
+    WordCharacterCount,
 )
 from .errors import BadRequestError, InternalServerError, NotFoundError, UnauthorizedError
 from . import audit, authentication, bin_lookup, files, guardrails, query, records, strings, tokens
@@ -94,38 +114,27 @@
 from .client import AsyncSkyflow, Skyflow
 from .environment import SkyflowEnvironment
 from .files import (
-    DeidentifyAudioRequestFile,
-    DeidentifyAudioRequestFileDataFormat,
-    DeidentifyAudioRequestOutputTranscription,
-    DeidentifyDocumentRequestFile,
-    DeidentifyDocumentRequestFileDataFormat,
-    DeidentifyFileRequestFile,
-    DeidentifyFileRequestFileDataFormat,
-    DeidentifyImageRequestFile,
-    DeidentifyImageRequestFileDataFormat,
-    DeidentifyImageRequestMaskingMethod,
-    DeidentifyPdfRequestFile,
-    DeidentifyPresentationRequestFile,
-    DeidentifyPresentationRequestFileDataFormat,
-    DeidentifySpreadsheetRequestFile,
-    DeidentifySpreadsheetRequestFileDataFormat,
-    DeidentifyStructuredTextRequestFile,
-    DeidentifyStructuredTextRequestFileDataFormat,
-    DeidentifyTextRequestFile,
-    ReidentifyFileRequestFile,
-    ReidentifyFileRequestFileDataFormat,
-    ReidentifyFileRequestFormat,
+    DeidentifyFileAudioRequestDeidentifyAudioEntityTypesItem,
+    DeidentifyFileAudioRequestDeidentifyAudioOutputTranscription,
+    DeidentifyFileDocumentPdfRequestDeidentifyPdfEntityTypesItem,
+    DeidentifyFileImageRequestDeidentifyImageEntityTypesItem,
+    DeidentifyFileImageRequestDeidentifyImageMaskingMethod,
+    DeidentifyFileRequestDeidentifyDocumentEntityTypesItem,
+    DeidentifyFileRequestDeidentifyPresentationEntityTypesItem,
+    DeidentifyFileRequestDeidentifySpreadsheetEntityTypesItem,
+    DeidentifyFileRequestDeidentifyStructuredTextEntityTypesItem,
+    DeidentifyFileRequestDeidentifyTextEntityTypesItem,
+    DeidentifyFileRequestEntityTypesItem,
 )
 from .records import (
     RecordServiceBulkGetRecordRequestOrderBy,
     RecordServiceBulkGetRecordRequestRedaction,
     RecordServiceGetRecordRequestRedaction,
 )
-from .strings import ReidentifyStringRequestFormat
+from .strings import DeidentifyStringRequestEntityTypesItem
 from .version import __version__
 
 __all__ = [
-    "AllowRegex",
     "AsyncSkyflow",
     "AuditEventAuditResourceType",
     "AuditEventContext",
@@ -139,72 +148,83 @@
     "AuditServiceListAuditEventsRequestSortOpsOrderBy",
     "BadRequestError",
     "BatchRecordMethod",
-    "CheckGuardrailsResponse",
-    "CheckGuardrailsResponseValidation",
-    "ConfigurationId",
     "ContextAccessType",
     "ContextAuthMode",
-    "DeidentifyAudioRequestFile",
-    "DeidentifyAudioRequestFileDataFormat",
-    "DeidentifyAudioRequestOutputTranscription",
-    "DeidentifyDocumentRequestFile",
-    "DeidentifyDocumentRequestFileDataFormat",
-    "DeidentifyFileOutput",
-    "DeidentifyFileOutputProcessedFileType",
-    "DeidentifyFileRequestFile",
-    "DeidentifyFileRequestFileDataFormat",
+    "DeidentifiedFileOutput",
+    "DeidentifiedFileOutputProcessedFileExtension",
+    "DeidentifiedFileOutputProcessedFileType",
+    "DeidentifyFileAudioRequestDeidentifyAudioEntityTypesItem",
+    "DeidentifyFileAudioRequestDeidentifyAudioOutputTranscription",
+    "DeidentifyFileDocumentPdfRequestDeidentifyPdfEntityTypesItem",
+    "DeidentifyFileImageRequestDeidentifyImageEntityTypesItem",
+    "DeidentifyFileImageRequestDeidentifyImageMaskingMethod",
+    "DeidentifyFileRequestDeidentifyDocumentEntityTypesItem",
+    "DeidentifyFileRequestDeidentifyPresentationEntityTypesItem",
+    "DeidentifyFileRequestDeidentifySpreadsheetEntityTypesItem",
+    "DeidentifyFileRequestDeidentifyStructuredTextEntityTypesItem",
+    "DeidentifyFileRequestDeidentifyTextEntityTypesItem",
+    "DeidentifyFileRequestEntityTypesItem",
     "DeidentifyFileResponse",
-    "DeidentifyImageRequestFile",
-    "DeidentifyImageRequestFileDataFormat",
-    "DeidentifyImageRequestMaskingMethod",
-    "DeidentifyPdfRequestFile",
-    "DeidentifyPresentationRequestFile",
-    "DeidentifyPresentationRequestFileDataFormat",
-    "DeidentifySpreadsheetRequestFile",
-    "DeidentifySpreadsheetRequestFileDataFormat",
-    "DeidentifyStatusResponse",
-    "DeidentifyStatusResponseOutputType",
-    "DeidentifyStatusResponseStatus",
+    "DeidentifyStringRequestEntityTypesItem",
     "DeidentifyStringResponse",
-    "DeidentifyStructuredTextRequestFile",
-    "DeidentifyStructuredTextRequestFileDataFormat",
-    "DeidentifyTextRequestFile",
-    "DetectedEntity",
+    "DetectGuardrailsResponse",
+    "DetectGuardrailsResponseValidation",
+    "DetectRunsResponse",
+    "DetectRunsResponseOutputType",
+    "DetectRunsResponseStatus",
     "DetokenizeRecordResponseValueType",
-    "EntityLocation",
-    "EntityType",
-    "EntityTypes",
     "ErrorResponse",
     "ErrorResponseError",
-    "ErrorString",
+    "FileData",
+    "FileDataDataFormat",
+    "FileDataDeidentifyAudio",
+    "FileDataDeidentifyAudioDataFormat",
+    "FileDataDeidentifyDocument",
+    "FileDataDeidentifyDocumentDataFormat",
+    "FileDataDeidentifyImage",
+    "FileDataDeidentifyImageDataFormat",
+    "FileDataDeidentifyPdf",
+    "FileDataDeidentifyPresentation",
+    "FileDataDeidentifyPresentationDataFormat",
+    "FileDataDeidentifySpreadsheet",
+    "FileDataDeidentifySpreadsheetDataFormat",
+    "FileDataDeidentifyStructuredText",
+    "FileDataDeidentifyStructuredTextDataFormat",
+    "FileDataDeidentifyText",
+    "FileDataReidentifyFile",
+    "FileDataReidentifyFileDataFormat",
+    "Format",
+    "FormatMaskedItem",
+    "FormatPlaintextItem",
+    "FormatRedactedItem",
     "GooglerpcStatus",
+    "HttpCode",
+    "IdentifyResponse",
     "InternalServerError",
     "NotFoundError",
     "ProtobufAny",
     "RecordServiceBulkGetRecordRequestOrderBy",
     "RecordServiceBulkGetRecordRequestRedaction",
     "RecordServiceGetRecordRequestRedaction",
     "RedactionEnumRedaction",
-    "ReidentifyFileRequestFile",
-    "ReidentifyFileRequestFileDataFormat",
-    "ReidentifyFileRequestFormat",
+    "ReidentifiedFileOutput",
+    "ReidentifiedFileOutputProcessedFileExtension",
     "ReidentifyFileResponse",
-    "ReidentifyFileResponseOutput",
+    "ReidentifyFileResponseOutputType",
     "ReidentifyFileResponseStatus",
-    "ReidentifyStringRequestFormat",
-    "ReidentifyStringResponse",
     "RequestActionType",
     "ResourceId",
-    "RestrictRegex",
+    "ShiftDates",
+    "ShiftDatesEntityTypesItem",
     "Skyflow",
     "SkyflowEnvironment",
-    "TokenType",
-    "TokenTypeDefault",
-    "TokenTypeWithoutVault",
-    "TokenTypeWithoutVaultDefault",
+    "StringResponseEntities",
+    "TokenTypeMapping",
+    "TokenTypeMappingDefault",
+    "TokenTypeMappingEntityOnlyItem",
+    "TokenTypeMappingEntityUnqCounterItem",
+    "TokenTypeMappingVaultTokenItem",
     "Transformations",
-    "TransformationsShiftDates",
-    "TransformationsShiftDatesEntityTypesItem",
     "UnauthorizedError",
     "UploadFileV2Response",
     "Uuid",
@@ -239,7 +259,7 @@
     "V1UpdateRecordResponse",
     "V1VaultFieldMapping",
     "V1VaultSchemaConfig",
-    "VaultId",
+    "WordCharacterCount",
     "__version__",
     "audit",
     "authentication",

skyflow/generated/rest/audit/client.py

@@ -205,7 +205,39 @@ def audit_service_list_audit_events(
             token="YOUR_TOKEN",
         )
         client.audit.audit_service_list_audit_events(
+            filter_ops_context_change_id="filterOps.context.changeID",
+            filter_ops_context_request_id="filterOps.context.requestID",
+            filter_ops_context_trace_id="filterOps.context.traceID",
+            filter_ops_context_session_id="filterOps.context.sessionID",
+            filter_ops_context_actor="filterOps.context.actor",
+            filter_ops_context_actor_type="NONE",
+            filter_ops_context_access_type="ACCESS_NONE",
+            filter_ops_context_ip_address="filterOps.context.ipAddress",
+            filter_ops_context_origin="filterOps.context.origin",
+            filter_ops_context_auth_mode="AUTH_NONE",
+            filter_ops_context_jwt_id="filterOps.context.jwtID",
+            filter_ops_context_bearer_token_context_id="filterOps.context.bearerTokenContextID",
+            filter_ops_parent_account_id="filterOps.parentAccountID",
             filter_ops_account_id="filterOps.accountID",
+            filter_ops_workspace_id="filterOps.workspaceID",
+            filter_ops_vault_id="filterOps.vaultID",
+            filter_ops_resource_i_ds="filterOps.resourceIDs",
+            filter_ops_action_type="NONE",
+            filter_ops_resource_type="NONE_API",
+            filter_ops_tags="filterOps.tags",
+            filter_ops_response_code=1,
+            filter_ops_start_time="filterOps.startTime",
+            filter_ops_end_time="filterOps.endTime",
+            filter_ops_api_name="filterOps.apiName",
+            filter_ops_response_message="filterOps.responseMessage",
+            filter_ops_http_method="filterOps.httpMethod",
+            filter_ops_http_uri="filterOps.httpURI",
+            sort_ops_sort_by="sortOps.sortBy",
+            sort_ops_order_by="ASCENDING",
+            after_ops_timestamp="afterOps.timestamp",
+            after_ops_change_id="afterOps.changeID",
+            limit=1000000,
+            offset=1000000,
         )
         """
         _response = self._raw_client.audit_service_list_audit_events(
@@ -431,7 +463,39 @@ async def audit_service_list_audit_events(
 
         async def main() -> None:
             await client.audit.audit_service_list_audit_events(
+                filter_ops_context_change_id="filterOps.context.changeID",
+                filter_ops_context_request_id="filterOps.context.requestID",
+                filter_ops_context_trace_id="filterOps.context.traceID",
+                filter_ops_context_session_id="filterOps.context.sessionID",
+                filter_ops_context_actor="filterOps.context.actor",
+                filter_ops_context_actor_type="NONE",
+                filter_ops_context_access_type="ACCESS_NONE",
+                filter_ops_context_ip_address="filterOps.context.ipAddress",
+                filter_ops_context_origin="filterOps.context.origin",
+                filter_ops_context_auth_mode="AUTH_NONE",
+                filter_ops_context_jwt_id="filterOps.context.jwtID",
+                filter_ops_context_bearer_token_context_id="filterOps.context.bearerTokenContextID",
+                filter_ops_parent_account_id="filterOps.parentAccountID",
                 filter_ops_account_id="filterOps.accountID",
+                filter_ops_workspace_id="filterOps.workspaceID",
+                filter_ops_vault_id="filterOps.vaultID",
+                filter_ops_resource_i_ds="filterOps.resourceIDs",
+                filter_ops_action_type="NONE",
+                filter_ops_resource_type="NONE_API",
+                filter_ops_tags="filterOps.tags",
+                filter_ops_response_code=1,
+                filter_ops_start_time="filterOps.startTime",
+                filter_ops_end_time="filterOps.endTime",
+                filter_ops_api_name="filterOps.apiName",
+                filter_ops_response_message="filterOps.responseMessage",
+                filter_ops_http_method="filterOps.httpMethod",
+                filter_ops_http_uri="filterOps.httpURI",
+                sort_ops_sort_by="sortOps.sortBy",
+                sort_ops_order_by="ASCENDING",
+                after_ops_timestamp="afterOps.timestamp",
+                after_ops_change_id="afterOps.changeID",
+                limit=1000000,
+                offset=1000000,
             )