qwak-core 0.4.339__tar.gz → 0.4.407__tar.gz

qwak_core-0.4.407/PKG-INFO

@@ -0,0 +1,50 @@
+Metadata-Version: 2.3
+Name: qwak-core
+Version: 0.4.407
+Summary: Qwak Core contains the necessary objects and communication tools for using the Qwak Platform
+License: Apache-2.0
+Keywords: mlops,ml,deployment,serving,model
+Author: Qwak
+Author-email: info@qwak.com
+Requires-Python: >=3.9,<3.12
+Classifier: License :: OSI Approved :: Apache Software License
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.7
+Classifier: Programming Language :: Python :: 3.8
+Classifier: Programming Language :: Python :: Implementation :: CPython
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Provides-Extra: feature-store
+Requires-Dist: PyYAML (>=6.0.2)
+Requires-Dist: cachetools
+Requires-Dist: chevron (==0.14.0)
+Requires-Dist: cloudpickle (==2.2.1) ; extra == "feature-store"
+Requires-Dist: dacite (==1.9.2)
+Requires-Dist: dependency-injector (>=4.0)
+Requires-Dist: filelock
+Requires-Dist: grpcio (>=1.71.2)
+Requires-Dist: joblib (>=1.3.2,<2.0.0)
+Requires-Dist: marshmallow-dataclass (>=8.5.8,<9.0.0)
+Requires-Dist: protobuf (>=4.25.8,<5)
+Requires-Dist: pyarrow (>=20.0.0) ; extra == "feature-store"
+Requires-Dist: pyathena (>=2.2.0,!=2.18.0) ; extra == "feature-store"
+Requires-Dist: pydantic
+Requires-Dist: pyspark (==3.5.7) ; extra == "feature-store"
+Requires-Dist: python-jose[cryptography] (>=3.4.0)
+Requires-Dist: python-json-logger (>=2.0.2)
+Requires-Dist: requests
+Requires-Dist: retrying (==1.4.2)
+Requires-Dist: tqdm
+Requires-Dist: typeguard (>=2,<3)
+Requires-Dist: typer
+Project-URL: Home page, https://www.qwak.com/
+Description-Content-Type: text/markdown
+
+# Qwak Core
+
+Qwak is an end-to-end production ML platform designed to allow data scientists to build, deploy, and monitor their models in production with minimal engineering friction.
+Qwak Core contains all the objects and tools necessary to use the Qwak Platform
+

qwak_core-0.4.407/README.md

@@ -0,0 +1,4 @@
+# Qwak Core
+
+Qwak is an end-to-end production ML platform designed to allow data scientists to build, deploy, and monitor their models in production with minimal engineering friction.
+Qwak Core contains all the objects and tools necessary to use the Qwak Platform

{qwak_core-0.4.339 → qwak_core-0.4.407}/_qwak_proto/qwak/administration/authenticated_user/v1/credentials_pb2.pyi

@@ -47,7 +47,7 @@ class AwsTemporaryCredentials(google.protobuf.message.Message):
     """Aws session token"""
     @property
     def expiration_time(self) -> google.protobuf.timestamp_pb2.Timestamp:
-        """Expiration timestamp of the credentials"""
+        """Expiration timestamp of the  credentials"""
     def __init__(
         self,
         *,

qwak_core-0.4.407/_qwak_proto/qwak/administration/runtime_configuration/v0/external/databricks/auth_pb2.py

@@ -0,0 +1,34 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: qwak/administration/runtime_configuration/v0/external/databricks/auth.proto
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from _qwak_proto.qwak.administration.runtime_configuration.v0.creds import secret_pb2 as qwak_dot_administration_dot_runtime__configuration_dot_v0_dot_creds_dot_secret__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\nKqwak/administration/runtime_configuration/v0/external/databricks/auth.proto\x12@qwak.administration.runtime_configuration.v0.external.databricks\x1a?qwak/administration/runtime_configuration/v0/creds/secret.proto\"\x9f\x02\n\x0e\x44\x61tabricksAuth\x12r\n\x0f\x64\x62_access_token\x18\x01 \x01(\x0b\x32W.qwak.administration.runtime_configuration.v0.external.databricks.DatabricksAccessTokenH\x00\x12\x8b\x01\n\x18\x64\x62_principal_credentials\x18\x02 \x01(\x0b\x32g.qwak.administration.runtime_configuration.v0.external.databricks.DatabricksServicePrincipalCredentialsH\x00\x42\x0b\n\tauth_type\"\x9a\x03\n%DatabricksServicePrincipalCredentials\x12\\\n\tclient_id\x18\x01 \x01(\x0b\x32G.qwak.administration.runtime_configuration.v0.creds.secret.SecretKeyRefH\x00\x12`\n\rclient_secret\x18\x02 \x01(\x0b\x32G.qwak.administration.runtime_configuration.v0.creds.secret.SecretKeyRefH\x01\x12\x80\x01\n\x17\x64\x62_workspace_descriptor\x18\x03 \x01(\x0b\x32_.qwak.administration.runtime_configuration.v0.external.databricks.DatabricksWorkspaceDescriptorB\x14\n\x12\x63lient_id_locationB\x18\n\x16\x63lient_secret_location\":\n\x1d\x44\x61tabricksWorkspaceDescriptor\x12\x19\n\x11oauth_service_url\x18\x01 \x01(\t\"~\n\x15\x44\x61tabricksAccessToken\x12Y\n\x06secret\x18\x01 \x01(\x0b\x32G.qwak.administration.runtime_configuration.v0.creds.secret.SecretKeyRefH\x00\x42\n\n\x08locationB\xef\x01\n9com.jfrog.ml.runtime_configuration.v0.external.databricksP\x01Z\xaf\x01github.com/qwak-ai/qwak-platform/services/core/java/user-management/user-management-api/pb/qwak/administration/runtime_configuration/v0/external/databricks;external_databricksb\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.administration.runtime_configuration.v0.external.databricks.auth_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n9com.jfrog.ml.runtime_configuration.v0.external.databricksP\001Z\257\001github.com/qwak-ai/qwak-platform/services/core/java/user-management/user-management-api/pb/qwak/administration/runtime_configuration/v0/external/databricks;external_databricks'
+  _globals['_DATABRICKSAUTH']._serialized_start=211
+  _globals['_DATABRICKSAUTH']._serialized_end=498
+  _globals['_DATABRICKSSERVICEPRINCIPALCREDENTIALS']._serialized_start=501
+  _globals['_DATABRICKSSERVICEPRINCIPALCREDENTIALS']._serialized_end=911
+  _globals['_DATABRICKSWORKSPACEDESCRIPTOR']._serialized_start=913
+  _globals['_DATABRICKSWORKSPACEDESCRIPTOR']._serialized_end=971
+  _globals['_DATABRICKSACCESSTOKEN']._serialized_start=973
+  _globals['_DATABRICKSACCESSTOKEN']._serialized_end=1099
+# @@protoc_insertion_point(module_scope)

qwak_core-0.4.407/_qwak_proto/qwak/administration/runtime_configuration/v0/external/databricks/auth_pb2.pyi

@@ -0,0 +1,97 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+import builtins
+import google.protobuf.descriptor
+import google.protobuf.message
+import qwak.administration.runtime_configuration.v0.creds.secret_pb2
+import sys
+import typing
+
+if sys.version_info >= (3, 8):
+    import typing as typing_extensions
+else:
+    import typing_extensions
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+class DatabricksAuth(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    DB_ACCESS_TOKEN_FIELD_NUMBER: builtins.int
+    DB_PRINCIPAL_CREDENTIALS_FIELD_NUMBER: builtins.int
+    @property
+    def db_access_token(self) -> global___DatabricksAccessToken: ...
+    @property
+    def db_principal_credentials(self) -> global___DatabricksServicePrincipalCredentials: ...
+    def __init__(
+        self,
+        *,
+        db_access_token: global___DatabricksAccessToken | None = ...,
+        db_principal_credentials: global___DatabricksServicePrincipalCredentials | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["auth_type", b"auth_type", "db_access_token", b"db_access_token", "db_principal_credentials", b"db_principal_credentials"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["auth_type", b"auth_type", "db_access_token", b"db_access_token", "db_principal_credentials", b"db_principal_credentials"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["auth_type", b"auth_type"]) -> typing_extensions.Literal["db_access_token", "db_principal_credentials"] | None: ...
+
+global___DatabricksAuth = DatabricksAuth
+
+class DatabricksServicePrincipalCredentials(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    CLIENT_ID_FIELD_NUMBER: builtins.int
+    CLIENT_SECRET_FIELD_NUMBER: builtins.int
+    DB_WORKSPACE_DESCRIPTOR_FIELD_NUMBER: builtins.int
+    @property
+    def client_id(self) -> qwak.administration.runtime_configuration.v0.creds.secret_pb2.SecretKeyRef: ...
+    @property
+    def client_secret(self) -> qwak.administration.runtime_configuration.v0.creds.secret_pb2.SecretKeyRef: ...
+    @property
+    def db_workspace_descriptor(self) -> global___DatabricksWorkspaceDescriptor: ...
+    def __init__(
+        self,
+        *,
+        client_id: qwak.administration.runtime_configuration.v0.creds.secret_pb2.SecretKeyRef | None = ...,
+        client_secret: qwak.administration.runtime_configuration.v0.creds.secret_pb2.SecretKeyRef | None = ...,
+        db_workspace_descriptor: global___DatabricksWorkspaceDescriptor | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["client_id", b"client_id", "client_id_location", b"client_id_location", "client_secret", b"client_secret", "client_secret_location", b"client_secret_location", "db_workspace_descriptor", b"db_workspace_descriptor"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["client_id", b"client_id", "client_id_location", b"client_id_location", "client_secret", b"client_secret", "client_secret_location", b"client_secret_location", "db_workspace_descriptor", b"db_workspace_descriptor"]) -> None: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["client_id_location", b"client_id_location"]) -> typing_extensions.Literal["client_id"] | None: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["client_secret_location", b"client_secret_location"]) -> typing_extensions.Literal["client_secret"] | None: ...
+
+global___DatabricksServicePrincipalCredentials = DatabricksServicePrincipalCredentials
+
+class DatabricksWorkspaceDescriptor(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    OAUTH_SERVICE_URL_FIELD_NUMBER: builtins.int
+    oauth_service_url: builtins.str
+    def __init__(
+        self,
+        *,
+        oauth_service_url: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["oauth_service_url", b"oauth_service_url"]) -> None: ...
+
+global___DatabricksWorkspaceDescriptor = DatabricksWorkspaceDescriptor
+
+class DatabricksAccessToken(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    SECRET_FIELD_NUMBER: builtins.int
+    @property
+    def secret(self) -> qwak.administration.runtime_configuration.v0.creds.secret_pb2.SecretKeyRef: ...
+    def __init__(
+        self,
+        *,
+        secret: qwak.administration.runtime_configuration.v0.creds.secret_pb2.SecretKeyRef | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["location", b"location", "secret", b"secret"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["location", b"location", "secret", b"secret"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["location", b"location"]) -> typing_extensions.Literal["secret"] | None: ...
+
+global___DatabricksAccessToken = DatabricksAccessToken

{qwak_core-0.4.339 → qwak_core-0.4.407}/_qwak_proto/qwak/administration/runtime_configuration/v0/hosting/azure/auth_pb2.py

@@ -15,7 +15,7 @@ from _qwak_proto.qwak.administration.runtime_configuration.v0.creds import secre
 from _qwak_proto.qwak.administration.runtime_configuration.v0.external.databricks import auth_pb2 as qwak_dot_administration_dot_runtime__configuration_dot_v0_dot_external_dot_databricks_dot_auth__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\nEqwak/administration/runtime_configuration/v0/hosting/azure/auth.proto\x12:qwak.administration.runtime_configuration.v0.hosting.azure\x1a?qwak/administration/runtime_configuration/v0/creds/secret.proto\x1aKqwak/administration/runtime_configuration/v0/external/databricks/auth.proto\"\xa1\x01\n\x13\x41zureAuthentication\x12\x81\x01\n\x15\x61zure_databricks_auth\x18\x01 \x01(\x0b\x32`.qwak.administration.runtime_configuration.v0.hosting.azure.AzureDatabricksServiceAuthenticationH\x00\x42\x06\n\x04type\"\xfc\x01\n$AzureDatabricksServiceAuthentication\x12\x61\n\x07\x64\x62_auth\x18\x01 \x01(\x0b\x32P.qwak.administration.runtime_configuration.v0.external.databricks.DatabricksAuth\x12q\n\x17\x61zure_blob_storage_auth\x18\x02 \x01(\x0b\x32P.qwak.administration.runtime_configuration.v0.hosting.azure.AzureBlobStorageAuth\"\x83\x01\n\x14\x41zureBlobStorageAuth\x12^\n\tsas_token\x18\x01 \x01(\x0b\x32I.qwak.administration.runtime_configuration.v0.hosting.azure.AzureSASTokenH\x00\x42\x0b\n\tauth_type\"v\n\rAzureSASToken\x12Y\n\x06secret\x18\x01 \x01(\x0b\x32G.qwak.administration.runtime_configuration.v0.creds.secret.SecretKeyRefH\x00\x42\n\n\x08locationB\xdd\x01\n3com.jfrog.ml.runtime_configuration.v0.hosting.azureP\x01Z\xa3\x01github.com/qwak-ai/qwak-platform/services/core/java/user-management/user-management-api/pb/qwak/administration/runtime_configuration/v0/hosting/azure;hosting_azureb\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\nEqwak/administration/runtime_configuration/v0/hosting/azure/auth.proto\x12:qwak.administration.runtime_configuration.v0.hosting.azure\x1a?qwak/administration/runtime_configuration/v0/creds/secret.proto\x1aKqwak/administration/runtime_configuration/v0/external/databricks/auth.proto\"\xa1\x01\n\x13\x41zureAuthentication\x12\x81\x01\n\x15\x61zure_databricks_auth\x18\x01 \x01(\x0b\x32`.qwak.administration.runtime_configuration.v0.hosting.azure.AzureDatabricksServiceAuthenticationH\x00\x42\x06\n\x04type\"\xfc\x01\n$AzureDatabricksServiceAuthentication\x12\x61\n\x07\x64\x62_auth\x18\x01 \x01(\x0b\x32P.qwak.administration.runtime_configuration.v0.external.databricks.DatabricksAuth\x12q\n\x17\x61zure_blob_storage_auth\x18\x02 \x01(\x0b\x32P.qwak.administration.runtime_configuration.v0.hosting.azure.AzureBlobStorageAuth\"\xf1\x01\n\x14\x41zureBlobStorageAuth\x12^\n\tsas_token\x18\x01 \x01(\x0b\x32I.qwak.administration.runtime_configuration.v0.hosting.azure.AzureSASTokenH\x00\x12l\n\x10managed_identity\x18\x02 \x01(\x0b\x32P.qwak.administration.runtime_configuration.v0.hosting.azure.AzureManagedIdentityH\x00\x42\x0b\n\tauth_type\"v\n\rAzureSASToken\x12Y\n\x06secret\x18\x01 \x01(\x0b\x32G.qwak.administration.runtime_configuration.v0.creds.secret.SecretKeyRefH\x00\x42\n\n\x08location\"\xb9\x01\n\x14\x41zureManagedIdentity\x12\x11\n\tclient_id\x18\x01 \x01(\t\x12\x85\x01\n\x15\x66\x65\x64\x65rated_credentials\x18\x02 \x01(\x0b\x32\x64.qwak.administration.runtime_configuration.v0.hosting.azure.KubernetesFederatedCredentialsDescriptorH\x00\x42\x06\n\x04type\"H\n(KubernetesFederatedCredentialsDescriptor\x12\x1c\n\x14service_account_name\x18\x01 \x01(\tB\xdd\x01\n3com.jfrog.ml.runtime_configuration.v0.hosting.azureP\x01Z\xa3\x01github.com/qwak-ai/qwak-platform/services/core/java/user-management/user-management-api/pb/qwak/administration/runtime_configuration/v0/hosting/azure;hosting_azureb\x06proto3')
 
 _globals = globals()
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
@@ -29,7 +29,11 @@ if _descriptor._USE_C_DESCRIPTORS == False:
   _globals['_AZUREDATABRICKSSERVICEAUTHENTICATION']._serialized_start=440
   _globals['_AZUREDATABRICKSSERVICEAUTHENTICATION']._serialized_end=692
   _globals['_AZUREBLOBSTORAGEAUTH']._serialized_start=695
-  _globals['_AZUREBLOBSTORAGEAUTH']._serialized_end=826
-  _globals['_AZURESASTOKEN']._serialized_start=828
-  _globals['_AZURESASTOKEN']._serialized_end=946
+  _globals['_AZUREBLOBSTORAGEAUTH']._serialized_end=936
+  _globals['_AZURESASTOKEN']._serialized_start=938
+  _globals['_AZURESASTOKEN']._serialized_end=1056
+  _globals['_AZUREMANAGEDIDENTITY']._serialized_start=1059
+  _globals['_AZUREMANAGEDIDENTITY']._serialized_end=1244
+  _globals['_KUBERNETESFEDERATEDCREDENTIALSDESCRIPTOR']._serialized_start=1246
+  _globals['_KUBERNETESFEDERATEDCREDENTIALSDESCRIPTOR']._serialized_end=1318
 # @@protoc_insertion_point(module_scope)

{qwak_core-0.4.339 → qwak_core-0.4.407}/_qwak_proto/qwak/administration/runtime_configuration/v0/hosting/azure/auth_pb2.pyi

@@ -57,16 +57,20 @@ class AzureBlobStorageAuth(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
     SAS_TOKEN_FIELD_NUMBER: builtins.int
+    MANAGED_IDENTITY_FIELD_NUMBER: builtins.int
     @property
     def sas_token(self) -> global___AzureSASToken: ...
+    @property
+    def managed_identity(self) -> global___AzureManagedIdentity: ...
     def __init__(
         self,
         *,
         sas_token: global___AzureSASToken | None = ...,
+        managed_identity: global___AzureManagedIdentity | None = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["auth_type", b"auth_type", "sas_token", b"sas_token"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["auth_type", b"auth_type", "sas_token", b"sas_token"]) -> None: ...
-    def WhichOneof(self, oneof_group: typing_extensions.Literal["auth_type", b"auth_type"]) -> typing_extensions.Literal["sas_token"] | None: ...
+    def HasField(self, field_name: typing_extensions.Literal["auth_type", b"auth_type", "managed_identity", b"managed_identity", "sas_token", b"sas_token"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["auth_type", b"auth_type", "managed_identity", b"managed_identity", "sas_token", b"sas_token"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["auth_type", b"auth_type"]) -> typing_extensions.Literal["sas_token", "managed_identity"] | None: ...
 
 global___AzureBlobStorageAuth = AzureBlobStorageAuth
 
@@ -86,3 +90,37 @@ class AzureSASToken(google.protobuf.message.Message):
     def WhichOneof(self, oneof_group: typing_extensions.Literal["location", b"location"]) -> typing_extensions.Literal["secret"] | None: ...
 
 global___AzureSASToken = AzureSASToken
+
+class AzureManagedIdentity(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    CLIENT_ID_FIELD_NUMBER: builtins.int
+    FEDERATED_CREDENTIALS_FIELD_NUMBER: builtins.int
+    client_id: builtins.str
+    @property
+    def federated_credentials(self) -> global___KubernetesFederatedCredentialsDescriptor: ...
+    def __init__(
+        self,
+        *,
+        client_id: builtins.str = ...,
+        federated_credentials: global___KubernetesFederatedCredentialsDescriptor | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["federated_credentials", b"federated_credentials", "type", b"type"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["client_id", b"client_id", "federated_credentials", b"federated_credentials", "type", b"type"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["type", b"type"]) -> typing_extensions.Literal["federated_credentials"] | None: ...
+
+global___AzureManagedIdentity = AzureManagedIdentity
+
+class KubernetesFederatedCredentialsDescriptor(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    SERVICE_ACCOUNT_NAME_FIELD_NUMBER: builtins.int
+    service_account_name: builtins.str
+    def __init__(
+        self,
+        *,
+        service_account_name: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["service_account_name", b"service_account_name"]) -> None: ...
+
+global___KubernetesFederatedCredentialsDescriptor = KubernetesFederatedCredentialsDescriptor

{qwak_core-0.4.339 → qwak_core-0.4.407}/_qwak_proto/qwak/administration/runtime_configuration/v0/hosting_config_pb2.py

@@ -16,7 +16,7 @@ from _qwak_proto.qwak.administration.runtime_configuration.v0.hosting.aws import
 from _qwak_proto.qwak.administration.runtime_configuration.v0.hosting.azure import auth_pb2 as qwak_dot_administration_dot_runtime__configuration_dot_v0_dot_hosting_dot_azure_dot_auth__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\nAqwak/administration/runtime_configuration/v0/hosting_config.proto\x12;qwak.administration.runtime_configuration.v0.hosting_config\x1a?qwak/administration/runtime_configuration/v0/creds/secret.proto\x1a\x43qwak/administration/runtime_configuration/v0/hosting/aws/auth.proto\x1a\x45qwak/administration/runtime_configuration/v0/hosting/azure/auth.proto\"\x8a\x03\n\x14HostingConfiguration\x12v\n\x16\x61ws_host_configuration\x18\x01 \x01(\x0b\x32T.qwak.administration.runtime_configuration.v0.hosting_config.AWSHostingConfigurationH\x00\x12v\n\x16gcp_host_configuration\x18\x02 \x01(\x0b\x32T.qwak.administration.runtime_configuration.v0.hosting_config.GCPHostingConfigurationH\x00\x12z\n\x18\x61zure_host_configuration\x18\x03 \x01(\x0b\x32V.qwak.administration.runtime_configuration.v0.hosting_config.AzureHostingConfigurationH\x00\x42\x06\n\x04type\"\x8e\x02\n\x17\x41WSHostingConfiguration\x12m\n\x13iam_service_account\x18\x01 \x01(\x0b\x32N.qwak.administration.runtime_configuration.v0.hosting_config.IAMServiceAccountH\x00\x12l\n\x12scoped_permissions\x18\x03 \x01(\x0b\x32N.qwak.administration.runtime_configuration.v0.hosting.aws.AWSScopedPermissionsH\x00\x12\x0e\n\x06region\x18\x02 \x01(\tB\x06\n\x04\x61uth\"\xb8\x01\n\x17GCPHostingConfiguration\x12q\n\x13gcp_service_account\x18\x01 \x01(\x0b\x32R.qwak.administration.runtime_configuration.v0.hosting_config.GCPServiceAccountAuthH\x00\x12\x0e\n\x06region\x18\x02 \x01(\t\x12\x12\n\nproject_id\x18\x03 \x01(\tB\x06\n\x04\x61uth\"\xe7\x01\n\x15GCPServiceAccountAuth\x12g\n\x0fservice_account\x18\x01 \x01(\x0b\x32N.qwak.administration.runtime_configuration.v0.hosting_config.IAMServiceAccount\x12\x65\n\x14service_account_json\x18\x02 \x01(\x0b\x32G.qwak.administration.runtime_configuration.v0.creds.secret.SecretKeyRef\"&\n\x11IAMServiceAccount\x12\x11\n\trole_name\x18\x01 \x01(\t\"\x90\x01\n\x19\x41zureHostingConfiguration\x12\x63\n\nazure_auth\x18\x01 \x01(\x0b\x32O.qwak.administration.runtime_configuration.v0.hosting.azure.AzureAuthentication\x12\x0e\n\x06region\x18\x02 \x01(\tB\xdb\x01\n4com.jfrog.ml.runtime_configuration.v0.hosting_configP\x01Z\xa0\x01github.com/qwak-ai/qwak-platform/services/core/java/user-management/user-management-api/pb/qwak/administration/runtime_configuration/v0;runtime_configuration_v0b\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\nAqwak/administration/runtime_configuration/v0/hosting_config.proto\x12;qwak.administration.runtime_configuration.v0.hosting_config\x1a?qwak/administration/runtime_configuration/v0/creds/secret.proto\x1a\x43qwak/administration/runtime_configuration/v0/hosting/aws/auth.proto\x1a\x45qwak/administration/runtime_configuration/v0/hosting/azure/auth.proto\"\x8a\x03\n\x14HostingConfiguration\x12v\n\x16\x61ws_host_configuration\x18\x01 \x01(\x0b\x32T.qwak.administration.runtime_configuration.v0.hosting_config.AWSHostingConfigurationH\x00\x12v\n\x16gcp_host_configuration\x18\x02 \x01(\x0b\x32T.qwak.administration.runtime_configuration.v0.hosting_config.GCPHostingConfigurationH\x00\x12z\n\x18\x61zure_host_configuration\x18\x03 \x01(\x0b\x32V.qwak.administration.runtime_configuration.v0.hosting_config.AzureHostingConfigurationH\x00\x42\x06\n\x04type\"\x8e\x02\n\x17\x41WSHostingConfiguration\x12m\n\x13iam_service_account\x18\x01 \x01(\x0b\x32N.qwak.administration.runtime_configuration.v0.hosting_config.IAMServiceAccountH\x00\x12l\n\x12scoped_permissions\x18\x03 \x01(\x0b\x32N.qwak.administration.runtime_configuration.v0.hosting.aws.AWSScopedPermissionsH\x00\x12\x0e\n\x06region\x18\x02 \x01(\tB\x06\n\x04\x61uth\"\xb8\x01\n\x17GCPHostingConfiguration\x12q\n\x13gcp_service_account\x18\x01 \x01(\x0b\x32R.qwak.administration.runtime_configuration.v0.hosting_config.GCPServiceAccountAuthH\x00\x12\x0e\n\x06region\x18\x02 \x01(\t\x12\x12\n\nproject_id\x18\x03 \x01(\tB\x06\n\x04\x61uth\"\xe7\x01\n\x15GCPServiceAccountAuth\x12g\n\x0fservice_account\x18\x01 \x01(\x0b\x32N.qwak.administration.runtime_configuration.v0.hosting_config.IAMServiceAccount\x12\x65\n\x14service_account_json\x18\x02 \x01(\x0b\x32G.qwak.administration.runtime_configuration.v0.creds.secret.SecretKeyRef\"&\n\x11IAMServiceAccount\x12\x11\n\trole_name\x18\x01 \x01(\t\"\xa9\x01\n\x19\x41zureHostingConfiguration\x12\x63\n\nazure_auth\x18\x01 \x01(\x0b\x32O.qwak.administration.runtime_configuration.v0.hosting.azure.AzureAuthentication\x12\x0e\n\x06region\x18\x02 \x01(\t\x12\x17\n\x0f\x61zure_tenant_id\x18\x03 \x01(\tB\xdb\x01\n4com.jfrog.ml.runtime_configuration.v0.hosting_configP\x01Z\xa0\x01github.com/qwak-ai/qwak-platform/services/core/java/user-management/user-management-api/pb/qwak/administration/runtime_configuration/v0;runtime_configuration_v0b\x06proto3')
 
 _globals = globals()
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
@@ -36,5 +36,5 @@ if _descriptor._USE_C_DESCRIPTORS == False:
   _globals['_IAMSERVICEACCOUNT']._serialized_start=1426
   _globals['_IAMSERVICEACCOUNT']._serialized_end=1464
   _globals['_AZUREHOSTINGCONFIGURATION']._serialized_start=1467
-  _globals['_AZUREHOSTINGCONFIGURATION']._serialized_end=1611
+  _globals['_AZUREHOSTINGCONFIGURATION']._serialized_end=1636
 # @@protoc_insertion_point(module_scope)

{qwak_core-0.4.339 → qwak_core-0.4.407}/_qwak_proto/qwak/administration/runtime_configuration/v0/hosting_config_pb2.pyi

@@ -128,16 +128,19 @@ class AzureHostingConfiguration(google.protobuf.message.Message):
 
     AZURE_AUTH_FIELD_NUMBER: builtins.int
     REGION_FIELD_NUMBER: builtins.int
+    AZURE_TENANT_ID_FIELD_NUMBER: builtins.int
     @property
     def azure_auth(self) -> qwak.administration.runtime_configuration.v0.hosting.azure.auth_pb2.AzureAuthentication: ...
     region: builtins.str
+    azure_tenant_id: builtins.str
     def __init__(
         self,
         *,
         azure_auth: qwak.administration.runtime_configuration.v0.hosting.azure.auth_pb2.AzureAuthentication | None = ...,
         region: builtins.str = ...,
+        azure_tenant_id: builtins.str = ...,
     ) -> None: ...
     def HasField(self, field_name: typing_extensions.Literal["azure_auth", b"azure_auth"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["azure_auth", b"azure_auth", "region", b"region"]) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["azure_auth", b"azure_auth", "azure_tenant_id", b"azure_tenant_id", "region", b"region"]) -> None: ...
 
 global___AzureHostingConfiguration = AzureHostingConfiguration

qwak_core-0.4.407/_qwak_proto/qwak/batch_job/v1/batch_job_service_pb2.py

@@ -0,0 +1,167 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: qwak/batch_job/v1/batch_job_service.proto
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
+from _qwak_proto.qwak.user_application.common.v0 import resources_pb2 as qwak_dot_user__application_dot_common_dot_v0_dot_resources__pb2
+from _qwak_proto.qwak.batch_job.v1 import batch_job_events_pb2 as qwak_dot_batch__job_dot_v1_dot_batch__job__events__pb2
+from _qwak_proto.qwak.administration.authenticated_user.v1 import credentials_pb2 as qwak_dot_administration_dot_authenticated__user_dot_v1_dot_credentials__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n)qwak/batch_job/v1/batch_job_service.proto\x12\rqwak.batchjob\x1a\x1fgoogle/protobuf/timestamp.proto\x1a/qwak/user_application/common/v0/resources.proto\x1a(qwak/batch_job/v1/batch_job_events.proto\x1a;qwak/administration/authenticated_user/v1/credentials.proto\"\xb4\x01\n\x15StartWarmupJobRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x15\n\tbranch_id\x18\x02 \x01(\tB\x02\x18\x01\x12\x10\n\x08\x62uild_id\x18\x05 \x01(\t\x12\x16\n\x0ewarmup_timeout\x18\x03 \x01(\x05\x12H\n\x19\x62\x61tch_job_deployment_size\x18\x04 \x01(\x0b\x32%.qwak.batchjob.BatchJobDeploymentSize\"B\n\x16StartWarmupJobResponse\x12\x0f\n\x07success\x18\x01 \x01(\x08\x12\x17\n\x0f\x66\x61ilure_message\x18\x02 \x01(\t\"S\n\x16\x43\x61ncelWarmupJobRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x15\n\tbranch_id\x18\x02 \x01(\tB\x02\x18\x01\x12\x10\n\x08\x62uild_id\x18\x03 \x01(\t\"C\n\x17\x43\x61ncelWarmupJobResponse\x12\x0f\n\x07success\x18\x01 \x01(\x08\x12\x17\n\x0f\x66\x61ilure_message\x18\x02 \x01(\t\"\x89\x05\n\x14StartBatchJobRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x15\n\tbranch_id\x18\x02 \x01(\tB\x02\x18\x01\x12\x10\n\x08\x62uild_id\x18\x0f \x01(\t\x12\x15\n\rsource_bucket\x18\x03 \x01(\t\x12\x1a\n\x12\x64\x65stination_bucket\x18\x04 \x01(\t\x12\x15\n\rsource_folder\x18\x05 \x01(\t\x12\x1a\n\x12\x64\x65stination_folder\x18\x06 \x01(\t\x12H\n\x19\x62\x61tch_job_deployment_size\x18\x07 \x01(\x0b\x32%.qwak.batchjob.BatchJobDeploymentSize\x12\x13\n\x0bjob_timeout\x18\x08 \x01(\x05\x12\x14\n\x0ctask_timeout\x18\t \x01(\x05\x12\x35\n\x0finput_file_type\x18\n \x01(\x0e\x32\x1c.qwak.batchjob.InputFileType\x12\x37\n\x10output_file_type\x18\x0b \x01(\x0e\x32\x1d.qwak.batchjob.OutputFileType\x12\x14\n\x0ctoken_secret\x18\x0c \x01(\t\x12\x15\n\rsecret_secret\x18\r \x01(\t\x12M\n\x1b\x61\x64vanced_deployment_options\x18\x0e \x01(\x0b\x32(.qwak.batchjob.AdvancedDeploymentOptions\x12\x34\n\nparameters\x18\x10 \x03(\x0b\x32 .qwak.batchjob.BatchJobParameter\x12\x39\n\x11\x62\x61tch_job_request\x18\x11 \x01(\x0b\x32\x1e.qwak.batchjob.BatchJobRequest\"\xcb\x01\n\x0f\x42\x61tchJobRequest\x12:\n\rmodel_details\x18\x01 \x01(\x0b\x32#.qwak.batchjob.BatchJobModelDetails\x12\x38\n\x0c\x64\x61ta_details\x18\x02 \x01(\x0b\x32\".qwak.batchjob.BatchJobDataDetails\x12\x42\n\x11\x65xecution_details\x18\x04 \x01(\x0b\x32\'.qwak.batchjob.BatchJobExecutionDetails\":\n\x14\x42\x61tchJobModelDetails\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x10\n\x08\x62uild_id\x18\x0f \x01(\t\"\xfc\x01\n\x13\x42\x61tchJobDataDetails\x12\x36\n\x0bsource_path\x18\x01 \x01(\x0b\x32!.qwak.batchjob.BatchJobSourcePath\x12@\n\x10\x64\x65stination_path\x18\x02 \x01(\x0b\x32&.qwak.batchjob.BatchJobDestinationPath\x12\x14\n\x0ctoken_secret\x18\x0c \x01(\t\x12\x15\n\rsecret_secret\x18\r \x01(\t\x12\x15\n\rsession_token\x18\x03 \x01(\t\x12\'\n\x1fservice_account_json_key_secret\x18\x05 \x01(\t\"y\n\x12\x42\x61tchJobSourcePath\x12\x15\n\rsource_bucket\x18\x01 \x01(\t\x12\x15\n\rsource_folder\x18\x02 \x01(\t\x12\x35\n\x0finput_file_type\x18\x03 \x01(\x0e\x32\x1c.qwak.batchjob.InputFileType\"\x8a\x01\n\x17\x42\x61tchJobDestinationPath\x12\x1a\n\x12\x64\x65stination_bucket\x18\x01 \x01(\t\x12\x1a\n\x12\x64\x65stination_folder\x18\x02 \x01(\t\x12\x37\n\x10output_file_type\x18\x03 \x01(\x0e\x32\x1d.qwak.batchjob.OutputFileType\"\x94\x02\n\x18\x42\x61tchJobExecutionDetails\x12\x13\n\x0bjob_timeout\x18\x01 \x01(\x05\x12\x14\n\x0ctask_timeout\x18\x02 \x01(\x05\x12H\n\x19\x62\x61tch_job_deployment_size\x18\x03 \x01(\x0b\x32%.qwak.batchjob.BatchJobDeploymentSize\x12M\n\x1b\x61\x64vanced_deployment_options\x18\x04 \x01(\x0b\x32(.qwak.batchjob.AdvancedDeploymentOptions\x12\x34\n\nparameters\x18\x05 \x03(\x0b\x32 .qwak.batchjob.BatchJobParameter\"/\n\x11\x42\x61tchJobParameter\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t\"S\n\x15StartBatchJobResponse\x12\x10\n\x08\x62\x61tch_id\x18\x01 \x01(\t\x12\x0f\n\x07success\x18\x02 \x01(\x08\x12\x17\n\x0f\x66\x61ilure_message\x18\x03 \x01(\t\")\n\x15\x43\x61ncelBatchJobRequest\x12\x10\n\x08\x62\x61tch_id\x18\x01 \x01(\t\"B\n\x16\x43\x61ncelBatchJobResponse\x12\x0f\n\x07success\x18\x01 \x01(\x08\x12\x17\n\x0f\x66\x61ilure_message\x18\x02 \x01(\t\",\n\x18GetBatchJobStatusRequest\x12\x10\n\x08\x62\x61tch_id\x18\x01 \x01(\t\"\xac\x01\n\x19GetBatchJobStatusResponse\x12\x38\n\njob_status\x18\x01 \x01(\x0e\x32$.qwak.batchjob.BatchJobStatusMessage\x12\x0f\n\x07success\x18\x02 \x01(\x08\x12\x17\n\x0f\x66\x61ilure_message\x18\x03 \x01(\t\x12\x16\n\x0e\x66inished_files\x18\x04 \x01(\x05\x12\x13\n\x0btotal_files\x18\x05 \x01(\x05\",\n\x18GetBatchJobReportRequest\x12\x10\n\x08\x62\x61tch_id\x18\x01 \x01(\t\"a\n\x19GetBatchJobReportResponse\x12\x12\n\nsuccessful\x18\x01 \x01(\x08\x12\x17\n\x0f\x66\x61ilure_message\x18\x02 \x01(\t\x12\x17\n\x0freport_messages\x18\x03 \x03(\t\"`\n\x1dUpdateBatchTasksStatusRequest\x12?\n\x11\x62\x61tch_task_events\x18\x01 \x03(\x0b\x32$.qwak.batchjob.BatchTaskEventMessage\"I\n\x1dUpdateBatchTaskStatusResponse\x12\x0f\n\x07success\x18\x01 \x01(\x08\x12\x17\n\x0f\x66\x61ilure_message\x18\x02 \x01(\t\"z\n\x19\x41\x64vancedDeploymentOptions\x12\x1b\n\x13\x63ustom_iam_role_arn\x18\x01 \x01(\t\x12\x17\n\x0fpurchase_option\x18\x02 \x01(\t\x12\'\n\x1fservice_account_key_secret_name\x18\x03 \x01(\t\"\x85\x03\n\x16\x42\x61tchJobDeploymentSize\x12\x16\n\x0enumber_of_pods\x18\x01 \x01(\x05\x12\x0f\n\x03\x63pu\x18\x02 \x01(\x02\x42\x02\x18\x01\x12\x19\n\rmemory_amount\x18\x03 \x01(\x05\x42\x02\x18\x01\x12J\n\x0cmemory_units\x18\x04 \x01(\x0e\x32\x30.qwak.batchjob.BatchJobDeploymentSize.MemoryUnitB\x02\x18\x01\x12H\n\rgpu_resources\x18\x05 \x01(\x0b\x32-.qwak.user_application.common.v0.GpuResourcesB\x02\x18\x01\x12`\n\x1c\x63lient_pod_compute_resources\x18\x06 \x01(\x0b\x32:.qwak.user_application.common.v0.ClientPodComputeResources\"/\n\nMemoryUnit\x12\x0b\n\x07UNKNOWN\x10\x00\x12\x07\n\x03MIB\x10\x01\x12\x07\n\x03GIB\x10\x02\x1a\x02\x18\x01\"\x99\x01\n\x15\x42\x61tchJobExecutionSize\x12\x16\n\x0enumber_of_pods\x18\x01 \x01(\x05\x12S\n\x15pod_compute_resources\x18\x02 \x01(\x0b\x32\x34.qwak.user_application.common.v0.PodComputeResources\x12\x13\n\x0btemplate_id\x18\x03 \x01(\t\"Q\n\x14ListBatchJobsRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x15\n\tbranch_id\x18\x02 \x01(\tB\x02\x18\x01\x12\x10\n\x08\x62uild_id\x18\x03 \x01(\t\"K\n\x15ListBatchJobsResponse\x12\x32\n\nbatch_jobs\x18\x01 \x03(\x0b\x32\x1e.qwak.batchjob.BatchJobDetails\"\x81\x03\n\x0f\x42\x61tchJobDetails\x12\x10\n\x08\x62uild_id\x18\x01 \x01(\t\x12\x0e\n\x06job_id\x18\x02 \x01(\t\x12\x38\n\njob_status\x18\x03 \x01(\x0e\x32$.qwak.batchjob.BatchJobStatusMessage\x12.\n\nstart_time\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12,\n\x08\x65nd_time\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x17\n\x0f\x66\x61ilure_message\x18\x06 \x01(\t\x12;\n\x08job_size\x18\x07 \x01(\x0b\x32%.qwak.batchjob.BatchJobDeploymentSizeB\x02\x18\x01\x12\x46\n\x18\x62\x61tch_job_execution_size\x18\x08 \x01(\x0b\x32$.qwak.batchjob.BatchJobExecutionSize\x12\x16\n\x0e\x65nvironment_id\x18\t \x01(\t\"+\n\x19GetBatchJobDetailsRequest\x12\x0e\n\x06job_id\x18\x01 \x01(\t\"y\n\x1aGetBatchJobDetailsResponse\x12\x31\n\tbatch_job\x18\x01 \x01(\x0b\x32\x1e.qwak.batchjob.BatchJobMessage\x12\x0f\n\x07success\x18\x02 \x01(\x08\x12\x17\n\x0f\x66\x61ilure_message\x18\x03 \x01(\t\"\x84\x04\n\x0f\x42\x61tchJobMessage\x12\x0e\n\x06job_id\x18\x01 \x01(\t\x12\x10\n\x08model_id\x18\x02 \x01(\t\x12\x15\n\tbranch_id\x18\x03 \x01(\tB\x02\x18\x01\x12\x10\n\x08\x62uild_id\x18\x04 \x01(\t\x12.\n\nstart_time\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12,\n\x08\x65nd_time\x18\x06 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x38\n\njob_status\x18\x07 \x01(\x0e\x32$.qwak.batchjob.BatchJobStatusMessage\x12\x15\n\rtotal_records\x18\x08 \x01(\x05\x12\x37\n\x08job_size\x18\t \x01(\x0b\x32%.qwak.batchjob.BatchJobDeploymentSize\x12;\n\x0freport_messages\x18\n \x03(\x0b\x32\".qwak.batchjob.ExecutionReportLine\x12\x43\n\x16\x65xecution_file_details\x18\x0b \x01(\x0b\x32#.qwak.batchjob.ExecutionFileDetails\x12<\n\x0ftask_executions\x18\x0c \x03(\x0b\x32#.qwak.batchjob.TaskExecutionDetails\"M\n\x13\x45xecutionReportLine\x12(\n\x04time\x18\x01 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x0c\n\x04line\x18\x02 \x01(\t\"\x95\x02\n\x14\x45xecutionFileDetails\x12\x15\n\rsource_bucket\x18\x01 \x01(\t\x12\x1a\n\x12\x64\x65stination_bucket\x18\x02 \x01(\t\x12\x13\n\x0bsource_path\x18\x03 \x01(\t\x12\x18\n\x10\x64\x65stination_path\x18\x04 \x01(\t\x12\x35\n\x0finput_file_type\x18\x05 \x01(\x0e\x32\x1c.qwak.batchjob.InputFileType\x12\x37\n\x10output_file_type\x18\x06 \x01(\x0e\x32\x1d.qwak.batchjob.OutputFileType\x12\x16\n\x0e\x66inished_files\x18\x07 \x01(\x05\x12\x13\n\x0btotal_files\x18\x08 \x01(\x05\"\xfd\x01\n\x14TaskExecutionDetails\x12\x0f\n\x07task_id\x18\x01 \x01(\t\x12\x35\n\x06status\x18\x02 \x01(\x0e\x32%.qwak.batchjob.BatchTaskStatusMessage\x12.\n\nstart_time\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12,\n\x08\x65nd_time\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x16\n\x0enum_of_records\x18\x05 \x01(\x05\x12\x14\n\x08\x66ilename\x18\x06 \x01(\tB\x02\x18\x01\x12\x11\n\tfilenames\x18\x07 \x03(\t\"d\n$GetBatchJobPreSignedUploadUrlRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x17\n\x0fnumber_of_files\x18\x02 \x01(\x05\x12\x11\n\tfile_type\x18\x03 \x01(\t\"\x98\x01\n%GetBatchJobPreSignedUploadUrlResponse\x12\x12\n\ninput_path\x18\x01 \x01(\t\x12\x13\n\x0boutput_path\x18\x02 \x01(\t\x12\x0e\n\x06\x62ucket\x18\x03 \x01(\t\x12\x0c\n\x04urls\x18\x04 \x03(\t\x12\x0f\n\x07success\x18\x05 \x01(\x08\x12\x17\n\x0f\x66\x61ilure_message\x18\x06 \x01(\t\"8\n&GetBatchJobPreSignedDownloadUrlRequest\x12\x0e\n\x06job_id\x18\x01 \x01(\t\"a\n\'GetBatchJobPreSignedDownloadUrlResponse\x12\x0c\n\x04urls\x18\x01 \x03(\t\x12\x0f\n\x07success\x18\x02 \x01(\x08\x12\x17\n\x0f\x66\x61ilure_message\x18\x03 \x01(\t\"3\n\x1fGetBatchJobUploadDetailsRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\"\xb4\x01\n GetBatchJobUploadDetailsResponse\x12\x12\n\ninput_path\x18\x01 \x01(\t\x12\x13\n\x0boutput_path\x18\x02 \x01(\t\x12\x0e\n\x06\x62ucket\x18\x03 \x01(\t\x12W\n\x0b\x63redentials\x18\x04 \x01(\x0b\x32\x42.qwak.administration.authenticated_user.v1.AwsTemporaryCredentials\"3\n!GetBatchJobDownloadDetailsRequest\x12\x0e\n\x06job_id\x18\x01 \x01(\t\"\x9b\x01\n\"GetBatchJobDownloadDetailsResponse\x12\x0e\n\x06\x62ucket\x18\x01 \x01(\t\x12\x0c\n\x04keys\x18\x02 \x03(\t\x12W\n\x0b\x63redentials\x18\x03 \x01(\x0b\x32\x42.qwak.administration.authenticated_user.v1.AwsTemporaryCredentials\"\xd7\x01\n\x1aUpdateDefaultParamsRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x10\n\x08\x62uild_id\x18\x02 \x01(\t\x12\x46\n\x17\x64\x65\x66\x61ult_deployment_size\x18\x03 \x01(\x0b\x32%.qwak.batchjob.BatchJobDeploymentSize\x12M\n\x1b\x61\x64vanced_deployment_options\x18\x04 \x01(\x0b\x32(.qwak.batchjob.AdvancedDeploymentOptions\"\x1d\n\x1bUpdateDefaultParamsResponse\".\n\x1a\x44\x65leteDefaultParamsRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\"\x1d\n\x1b\x44\x65leteDefaultParamsResponse\"S\n&UpdateTaskIngestedRecordsAmountRequest\x12\x0f\n\x07task_id\x18\x01 \x01(\t\x12\x18\n\x10ingested_records\x18\x02 \x01(\x03\")\n\'UpdateTaskIngestedRecordsAmountResponse\"S\n\x19UpdateTasksDetailsRequest\x12\x36\n\rtasks_details\x18\x01 \x03(\x0b\x32\x1f.qwak.batchjob.BatchTaskDetails\"\x1c\n\x1aUpdateTasksDetailsResponse\"a\n\x10\x42\x61tchTaskDetails\x12\x0f\n\x07task_id\x18\x01 \x01(\t\x12<\n\x13input_files_details\x18\x02 \x03(\x0b\x32\x1f.qwak.batchjob.InputFileDetails\" \n\x10InputFileDetails\x12\x0c\n\x04path\x18\x01 \x01(\t*\x81\x01\n\rInputFileType\x12\x1d\n\x19UNDEFINED_INPUT_FILE_TYPE\x10\x00\x12\x17\n\x13\x43SV_INPUT_FILE_TYPE\x10\x01\x12\x1b\n\x17\x46\x45\x41THER_INPUT_FILE_TYPE\x10\x02\x12\x1b\n\x17PARQUET_INPUT_FILE_TYPE\x10\x03*\x86\x01\n\x0eOutputFileType\x12\x1e\n\x1aUNDEFINED_OUTPUT_FILE_TYPE\x10\x00\x12\x18\n\x14\x43SV_OUTPUT_FILE_TYPE\x10\x01\x12\x1c\n\x18\x46\x45\x41THER_OUTPUT_FILE_TYPE\x10\x02\x12\x1c\n\x18PARQUET_OUTPUT_FILE_TYPE\x10\x03*\x8d\x02\n\x15\x42\x61tchJobStatusMessage\x12\x1e\n\x1aUNDEFINED_BATCH_JOB_STATUS\x10\x00\x12\x1e\n\x1a\x42\x41TCH_JOB_COMMITTED_STATUS\x10\x01\x12\x1c\n\x18\x42\x41TCH_JOB_PENDING_STATUS\x10\x02\x12\x1c\n\x18\x42\x41TCH_JOB_RUNNING_STATUS\x10\x03\x12\x1d\n\x19\x42\x41TCH_JOB_FINISHED_STATUS\x10\x04\x12\x1b\n\x17\x42\x41TCH_JOB_FAILED_STATUS\x10\x05\x12\x1e\n\x1a\x42\x41TCH_JOB_CANCELLED_STATUS\x10\x06\x12\x1c\n\x18\x42\x41TCH_JOB_TIMEOUT_STATUS\x10\x07*\x9c\x02\n\x16\x42\x61tchTaskStatusMessage\x12\x1f\n\x1bUNDEFINED_BATCH_TASK_STATUS\x10\x00\x12\x1f\n\x1b\x42\x41TCH_TASK_COMMITTED_STATUS\x10\x01\x12#\n\x1f\x42\x41TCH_TASK_PENDING_START_STATUS\x10\x02\x12\x1d\n\x19\x42\x41TCH_TASK_RUNNING_STATUS\x10\x03\x12\x1e\n\x1a\x42\x41TCH_TASK_FINISHED_STATUS\x10\x04\x12\x1c\n\x18\x42\x41TCH_TASK_FAILED_STATUS\x10\x05\x12\x1f\n\x1b\x42\x41TCH_TASK_CANCELLED_STATUS\x10\x06\x12\x1d\n\x19\x42\x41TCH_TASK_TIMEOUT_STATUS\x10\x07\x32\x9f\x0f\n\x19\x42\x61tchJobManagementService\x12Z\n\rStartBatchJob\x12#.qwak.batchjob.StartBatchJobRequest\x1a$.qwak.batchjob.StartBatchJobResponse\x12]\n\x0e\x43\x61ncelBatchJob\x12$.qwak.batchjob.CancelBatchJobRequest\x1a%.qwak.batchjob.CancelBatchJobResponse\x12]\n\x0eStartWarmupJob\x12$.qwak.batchjob.StartWarmupJobRequest\x1a%.qwak.batchjob.StartWarmupJobResponse\x12`\n\x0f\x43\x61ncelWarmupJob\x12%.qwak.batchjob.CancelWarmupJobRequest\x1a&.qwak.batchjob.CancelWarmupJobResponse\x12\x66\n\x11GetBatchJobStatus\x12\'.qwak.batchjob.GetBatchJobStatusRequest\x1a(.qwak.batchjob.GetBatchJobStatusResponse\x12\x66\n\x11GetBatchJobReport\x12\'.qwak.batchjob.GetBatchJobReportRequest\x1a(.qwak.batchjob.GetBatchJobReportResponse\x12t\n\x16UpdateBatchTasksStatus\x12,.qwak.batchjob.UpdateBatchTasksStatusRequest\x1a,.qwak.batchjob.UpdateBatchTaskStatusResponse\x12Z\n\rListBatchJobs\x12#.qwak.batchjob.ListBatchJobsRequest\x1a$.qwak.batchjob.ListBatchJobsResponse\x12i\n\x12GetBatchJobDetails\x12(.qwak.batchjob.GetBatchJobDetailsRequest\x1a).qwak.batchjob.GetBatchJobDetailsResponse\x12\x8a\x01\n\x1dGetBatchJobPreSignedUploadUrl\x12\x33.qwak.batchjob.GetBatchJobPreSignedUploadUrlRequest\x1a\x34.qwak.batchjob.GetBatchJobPreSignedUploadUrlResponse\x12\x90\x01\n\x1fGetBatchJobPreSignedDownloadUrl\x12\x35.qwak.batchjob.GetBatchJobPreSignedDownloadUrlRequest\x1a\x36.qwak.batchjob.GetBatchJobPreSignedDownloadUrlResponse\x12{\n\x18GetBatchJobUploadDetails\x12..qwak.batchjob.GetBatchJobUploadDetailsRequest\x1a/.qwak.batchjob.GetBatchJobUploadDetailsResponse\x12\x81\x01\n\x1aGetBatchJobDownloadDetails\x12\x30.qwak.batchjob.GetBatchJobDownloadDetailsRequest\x1a\x31.qwak.batchjob.GetBatchJobDownloadDetailsResponse\x12l\n\x13UpdateDefaultParams\x12).qwak.batchjob.UpdateDefaultParamsRequest\x1a*.qwak.batchjob.UpdateDefaultParamsResponse\x12l\n\x13\x44\x65leteDefaultParams\x12).qwak.batchjob.DeleteDefaultParamsRequest\x1a*.qwak.batchjob.DeleteDefaultParamsResponse\x12\x90\x01\n\x1fUpdateTaskIngestedRecordsAmount\x12\x35.qwak.batchjob.UpdateTaskIngestedRecordsAmountRequest\x1a\x36.qwak.batchjob.UpdateTaskIngestedRecordsAmountResponse\x12i\n\x12UpdateTasksDetails\x12(.qwak.batchjob.UpdateTasksDetailsRequest\x1a).qwak.batchjob.UpdateTasksDetailsResponseB.\n\x19\x63om.qwak.ai.batch.job.apiP\x01Z\x0f./;inferencejobb\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.batch_job.v1.batch_job_service_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n\031com.qwak.ai.batch.job.apiP\001Z\017./;inferencejob'
+  _STARTWARMUPJOBREQUEST.fields_by_name['branch_id']._options = None
+  _STARTWARMUPJOBREQUEST.fields_by_name['branch_id']._serialized_options = b'\030\001'
+  _CANCELWARMUPJOBREQUEST.fields_by_name['branch_id']._options = None
+  _CANCELWARMUPJOBREQUEST.fields_by_name['branch_id']._serialized_options = b'\030\001'
+  _STARTBATCHJOBREQUEST.fields_by_name['branch_id']._options = None
+  _STARTBATCHJOBREQUEST.fields_by_name['branch_id']._serialized_options = b'\030\001'
+  _BATCHJOBDEPLOYMENTSIZE_MEMORYUNIT._options = None
+  _BATCHJOBDEPLOYMENTSIZE_MEMORYUNIT._serialized_options = b'\030\001'
+  _BATCHJOBDEPLOYMENTSIZE.fields_by_name['cpu']._options = None
+  _BATCHJOBDEPLOYMENTSIZE.fields_by_name['cpu']._serialized_options = b'\030\001'
+  _BATCHJOBDEPLOYMENTSIZE.fields_by_name['memory_amount']._options = None
+  _BATCHJOBDEPLOYMENTSIZE.fields_by_name['memory_amount']._serialized_options = b'\030\001'
+  _BATCHJOBDEPLOYMENTSIZE.fields_by_name['memory_units']._options = None
+  _BATCHJOBDEPLOYMENTSIZE.fields_by_name['memory_units']._serialized_options = b'\030\001'
+  _BATCHJOBDEPLOYMENTSIZE.fields_by_name['gpu_resources']._options = None
+  _BATCHJOBDEPLOYMENTSIZE.fields_by_name['gpu_resources']._serialized_options = b'\030\001'
+  _LISTBATCHJOBSREQUEST.fields_by_name['branch_id']._options = None
+  _LISTBATCHJOBSREQUEST.fields_by_name['branch_id']._serialized_options = b'\030\001'
+  _BATCHJOBDETAILS.fields_by_name['job_size']._options = None
+  _BATCHJOBDETAILS.fields_by_name['job_size']._serialized_options = b'\030\001'
+  _BATCHJOBMESSAGE.fields_by_name['branch_id']._options = None
+  _BATCHJOBMESSAGE.fields_by_name['branch_id']._serialized_options = b'\030\001'
+  _TASKEXECUTIONDETAILS.fields_by_name['filename']._options = None
+  _TASKEXECUTIONDETAILS.fields_by_name['filename']._serialized_options = b'\030\001'
+  _globals['_INPUTFILETYPE']._serialized_start=7238
+  _globals['_INPUTFILETYPE']._serialized_end=7367
+  _globals['_OUTPUTFILETYPE']._serialized_start=7370
+  _globals['_OUTPUTFILETYPE']._serialized_end=7504
+  _globals['_BATCHJOBSTATUSMESSAGE']._serialized_start=7507
+  _globals['_BATCHJOBSTATUSMESSAGE']._serialized_end=7776
+  _globals['_BATCHTASKSTATUSMESSAGE']._serialized_start=7779
+  _globals['_BATCHTASKSTATUSMESSAGE']._serialized_end=8063
+  _globals['_STARTWARMUPJOBREQUEST']._serialized_start=246
+  _globals['_STARTWARMUPJOBREQUEST']._serialized_end=426
+  _globals['_STARTWARMUPJOBRESPONSE']._serialized_start=428
+  _globals['_STARTWARMUPJOBRESPONSE']._serialized_end=494
+  _globals['_CANCELWARMUPJOBREQUEST']._serialized_start=496
+  _globals['_CANCELWARMUPJOBREQUEST']._serialized_end=579
+  _globals['_CANCELWARMUPJOBRESPONSE']._serialized_start=581
+  _globals['_CANCELWARMUPJOBRESPONSE']._serialized_end=648
+  _globals['_STARTBATCHJOBREQUEST']._serialized_start=651
+  _globals['_STARTBATCHJOBREQUEST']._serialized_end=1300
+  _globals['_BATCHJOBREQUEST']._serialized_start=1303
+  _globals['_BATCHJOBREQUEST']._serialized_end=1506
+  _globals['_BATCHJOBMODELDETAILS']._serialized_start=1508
+  _globals['_BATCHJOBMODELDETAILS']._serialized_end=1566
+  _globals['_BATCHJOBDATADETAILS']._serialized_start=1569
+  _globals['_BATCHJOBDATADETAILS']._serialized_end=1821
+  _globals['_BATCHJOBSOURCEPATH']._serialized_start=1823
+  _globals['_BATCHJOBSOURCEPATH']._serialized_end=1944
+  _globals['_BATCHJOBDESTINATIONPATH']._serialized_start=1947
+  _globals['_BATCHJOBDESTINATIONPATH']._serialized_end=2085
+  _globals['_BATCHJOBEXECUTIONDETAILS']._serialized_start=2088
+  _globals['_BATCHJOBEXECUTIONDETAILS']._serialized_end=2364
+  _globals['_BATCHJOBPARAMETER']._serialized_start=2366
+  _globals['_BATCHJOBPARAMETER']._serialized_end=2413
+  _globals['_STARTBATCHJOBRESPONSE']._serialized_start=2415
+  _globals['_STARTBATCHJOBRESPONSE']._serialized_end=2498
+  _globals['_CANCELBATCHJOBREQUEST']._serialized_start=2500
+  _globals['_CANCELBATCHJOBREQUEST']._serialized_end=2541
+  _globals['_CANCELBATCHJOBRESPONSE']._serialized_start=2543
+  _globals['_CANCELBATCHJOBRESPONSE']._serialized_end=2609
+  _globals['_GETBATCHJOBSTATUSREQUEST']._serialized_start=2611
+  _globals['_GETBATCHJOBSTATUSREQUEST']._serialized_end=2655
+  _globals['_GETBATCHJOBSTATUSRESPONSE']._serialized_start=2658
+  _globals['_GETBATCHJOBSTATUSRESPONSE']._serialized_end=2830
+  _globals['_GETBATCHJOBREPORTREQUEST']._serialized_start=2832
+  _globals['_GETBATCHJOBREPORTREQUEST']._serialized_end=2876
+  _globals['_GETBATCHJOBREPORTRESPONSE']._serialized_start=2878
+  _globals['_GETBATCHJOBREPORTRESPONSE']._serialized_end=2975
+  _globals['_UPDATEBATCHTASKSSTATUSREQUEST']._serialized_start=2977
+  _globals['_UPDATEBATCHTASKSSTATUSREQUEST']._serialized_end=3073
+  _globals['_UPDATEBATCHTASKSTATUSRESPONSE']._serialized_start=3075
+  _globals['_UPDATEBATCHTASKSTATUSRESPONSE']._serialized_end=3148
+  _globals['_ADVANCEDDEPLOYMENTOPTIONS']._serialized_start=3150
+  _globals['_ADVANCEDDEPLOYMENTOPTIONS']._serialized_end=3272
+  _globals['_BATCHJOBDEPLOYMENTSIZE']._serialized_start=3275
+  _globals['_BATCHJOBDEPLOYMENTSIZE']._serialized_end=3664
+  _globals['_BATCHJOBDEPLOYMENTSIZE_MEMORYUNIT']._serialized_start=3617
+  _globals['_BATCHJOBDEPLOYMENTSIZE_MEMORYUNIT']._serialized_end=3664
+  _globals['_BATCHJOBEXECUTIONSIZE']._serialized_start=3667
+  _globals['_BATCHJOBEXECUTIONSIZE']._serialized_end=3820
+  _globals['_LISTBATCHJOBSREQUEST']._serialized_start=3822
+  _globals['_LISTBATCHJOBSREQUEST']._serialized_end=3903
+  _globals['_LISTBATCHJOBSRESPONSE']._serialized_start=3905
+  _globals['_LISTBATCHJOBSRESPONSE']._serialized_end=3980
+  _globals['_BATCHJOBDETAILS']._serialized_start=3983
+  _globals['_BATCHJOBDETAILS']._serialized_end=4368
+  _globals['_GETBATCHJOBDETAILSREQUEST']._serialized_start=4370
+  _globals['_GETBATCHJOBDETAILSREQUEST']._serialized_end=4413
+  _globals['_GETBATCHJOBDETAILSRESPONSE']._serialized_start=4415
+  _globals['_GETBATCHJOBDETAILSRESPONSE']._serialized_end=4536
+  _globals['_BATCHJOBMESSAGE']._serialized_start=4539
+  _globals['_BATCHJOBMESSAGE']._serialized_end=5055
+  _globals['_EXECUTIONREPORTLINE']._serialized_start=5057
+  _globals['_EXECUTIONREPORTLINE']._serialized_end=5134
+  _globals['_EXECUTIONFILEDETAILS']._serialized_start=5137
+  _globals['_EXECUTIONFILEDETAILS']._serialized_end=5414
+  _globals['_TASKEXECUTIONDETAILS']._serialized_start=5417
+  _globals['_TASKEXECUTIONDETAILS']._serialized_end=5670
+  _globals['_GETBATCHJOBPRESIGNEDUPLOADURLREQUEST']._serialized_start=5672
+  _globals['_GETBATCHJOBPRESIGNEDUPLOADURLREQUEST']._serialized_end=5772
+  _globals['_GETBATCHJOBPRESIGNEDUPLOADURLRESPONSE']._serialized_start=5775
+  _globals['_GETBATCHJOBPRESIGNEDUPLOADURLRESPONSE']._serialized_end=5927
+  _globals['_GETBATCHJOBPRESIGNEDDOWNLOADURLREQUEST']._serialized_start=5929
+  _globals['_GETBATCHJOBPRESIGNEDDOWNLOADURLREQUEST']._serialized_end=5985
+  _globals['_GETBATCHJOBPRESIGNEDDOWNLOADURLRESPONSE']._serialized_start=5987
+  _globals['_GETBATCHJOBPRESIGNEDDOWNLOADURLRESPONSE']._serialized_end=6084
+  _globals['_GETBATCHJOBUPLOADDETAILSREQUEST']._serialized_start=6086
+  _globals['_GETBATCHJOBUPLOADDETAILSREQUEST']._serialized_end=6137
+  _globals['_GETBATCHJOBUPLOADDETAILSRESPONSE']._serialized_start=6140
+  _globals['_GETBATCHJOBUPLOADDETAILSRESPONSE']._serialized_end=6320
+  _globals['_GETBATCHJOBDOWNLOADDETAILSREQUEST']._serialized_start=6322
+  _globals['_GETBATCHJOBDOWNLOADDETAILSREQUEST']._serialized_end=6373
+  _globals['_GETBATCHJOBDOWNLOADDETAILSRESPONSE']._serialized_start=6376
+  _globals['_GETBATCHJOBDOWNLOADDETAILSRESPONSE']._serialized_end=6531
+  _globals['_UPDATEDEFAULTPARAMSREQUEST']._serialized_start=6534
+  _globals['_UPDATEDEFAULTPARAMSREQUEST']._serialized_end=6749
+  _globals['_UPDATEDEFAULTPARAMSRESPONSE']._serialized_start=6751
+  _globals['_UPDATEDEFAULTPARAMSRESPONSE']._serialized_end=6780
+  _globals['_DELETEDEFAULTPARAMSREQUEST']._serialized_start=6782
+  _globals['_DELETEDEFAULTPARAMSREQUEST']._serialized_end=6828
+  _globals['_DELETEDEFAULTPARAMSRESPONSE']._serialized_start=6830
+  _globals['_DELETEDEFAULTPARAMSRESPONSE']._serialized_end=6859
+  _globals['_UPDATETASKINGESTEDRECORDSAMOUNTREQUEST']._serialized_start=6861
+  _globals['_UPDATETASKINGESTEDRECORDSAMOUNTREQUEST']._serialized_end=6944
+  _globals['_UPDATETASKINGESTEDRECORDSAMOUNTRESPONSE']._serialized_start=6946
+  _globals['_UPDATETASKINGESTEDRECORDSAMOUNTRESPONSE']._serialized_end=6987
+  _globals['_UPDATETASKSDETAILSREQUEST']._serialized_start=6989
+  _globals['_UPDATETASKSDETAILSREQUEST']._serialized_end=7072
+  _globals['_UPDATETASKSDETAILSRESPONSE']._serialized_start=7074
+  _globals['_UPDATETASKSDETAILSRESPONSE']._serialized_end=7102
+  _globals['_BATCHTASKDETAILS']._serialized_start=7104
+  _globals['_BATCHTASKDETAILS']._serialized_end=7201
+  _globals['_INPUTFILEDETAILS']._serialized_start=7203
+  _globals['_INPUTFILEDETAILS']._serialized_end=7235
+  _globals['_BATCHJOBMANAGEMENTSERVICE']._serialized_start=8066
+  _globals['_BATCHJOBMANAGEMENTSERVICE']._serialized_end=10017
+# @@protoc_insertion_point(module_scope)

{qwak_core-0.4.339 → qwak_core-0.4.407}/_qwak_proto/qwak/batch_job/v1/batch_job_service_pb2.pyi

@@ -352,6 +352,7 @@ class BatchJobDataDetails(google.protobuf.message.Message):
     DESTINATION_PATH_FIELD_NUMBER: builtins.int
     TOKEN_SECRET_FIELD_NUMBER: builtins.int
     SECRET_SECRET_FIELD_NUMBER: builtins.int
+    SESSION_TOKEN_FIELD_NUMBER: builtins.int
     SERVICE_ACCOUNT_JSON_KEY_SECRET_FIELD_NUMBER: builtins.int
     @property
     def source_path(self) -> global___BatchJobSourcePath:
@@ -363,6 +364,8 @@ class BatchJobDataDetails(google.protobuf.message.Message):
     """The access token's secret name in secret service"""
     secret_secret: builtins.str
     """The access secret's secret name in secret service"""
+    session_token: builtins.str
+    """The session token required to use the bucket for batch inference."""
     service_account_json_key_secret: builtins.str
     """The service account json key secret name"""
     def __init__(
@@ -372,10 +375,11 @@ class BatchJobDataDetails(google.protobuf.message.Message):
         destination_path: global___BatchJobDestinationPath | None = ...,
         token_secret: builtins.str = ...,
         secret_secret: builtins.str = ...,
+        session_token: builtins.str = ...,
         service_account_json_key_secret: builtins.str = ...,
     ) -> None: ...
     def HasField(self, field_name: typing_extensions.Literal["destination_path", b"destination_path", "source_path", b"source_path"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["destination_path", b"destination_path", "secret_secret", b"secret_secret", "service_account_json_key_secret", b"service_account_json_key_secret", "source_path", b"source_path", "token_secret", b"token_secret"]) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["destination_path", b"destination_path", "secret_secret", b"secret_secret", "service_account_json_key_secret", b"service_account_json_key_secret", "session_token", b"session_token", "source_path", b"source_path", "token_secret", b"token_secret"]) -> None: ...
 
 global___BatchJobDataDetails = BatchJobDataDetails