qwak-core 0.4.366__py3-none-any.whl → 0.4.392__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.


This version of qwak-core might be problematic. Click here for more details.

Files changed (66) hide show
  1. _qwak_proto/qwak/administration/runtime_configuration/v0/external/databricks/auth_pb2.py +6 -4
  2. _qwak_proto/qwak/administration/runtime_configuration/v0/external/databricks/auth_pb2.pyi +27 -4
  3. _qwak_proto/qwak/administration/runtime_configuration/v0/hosting/azure/auth_pb2.py +5 -3
  4. _qwak_proto/qwak/administration/runtime_configuration/v0/hosting/azure/auth_pb2.pyi +21 -1
  5. _qwak_proto/qwak/batch_job/v1/batch_job_service_pb2.pyi +1 -1
  6. _qwak_proto/qwak/builds/build_pb2.py +42 -41
  7. _qwak_proto/qwak/builds/build_pb2.pyi +8 -1
  8. _qwak_proto/qwak/builds/build_values_pb2.py +76 -0
  9. _qwak_proto/qwak/builds/build_values_pb2.pyi +533 -0
  10. _qwak_proto/qwak/builds/build_values_pb2_grpc.py +4 -0
  11. _qwak_proto/qwak/kube_deployment_captain/batch_job_pb2.pyi +1 -1
  12. _qwak_proto/qwak/projects/projects_pb2.py +17 -15
  13. qwak/__init__.py +1 -1
  14. qwak/clients/model_management/client.py +0 -5
  15. qwak/clients/project/client.py +0 -7
  16. qwak/feature_store/_common/packaging.py +11 -5
  17. qwak/inner/const.py +0 -2
  18. qwak/inner/di_configuration/account.py +6 -66
  19. qwak/inner/tool/auth.py +0 -86
  20. qwak/inner/tool/grpc/grpc_auth.py +0 -32
  21. qwak/inner/tool/grpc/grpc_tools.py +2 -8
  22. qwak/inner/tool/grpc/grpc_try_wrapping.py +3 -1
  23. qwak/qwak_client/client.py +2 -8
  24. qwak/vector_store/rest_helpers.py +4 -16
  25. qwak_core-0.4.392.dist-info/METADATA +50 -0
  26. {qwak_core-0.4.366.dist-info → qwak_core-0.4.392.dist-info}/RECORD +27 -62
  27. frogml_storage/__init__.py +0 -1
  28. frogml_storage/artifactory/__init__.py +0 -1
  29. frogml_storage/artifactory/_artifactory_api.py +0 -315
  30. frogml_storage/authentication/login/__init__.py +0 -1
  31. frogml_storage/authentication/login/_login_cli.py +0 -239
  32. frogml_storage/authentication/login/_login_command.py +0 -74
  33. frogml_storage/authentication/models/__init__.py +0 -3
  34. frogml_storage/authentication/models/_auth.py +0 -24
  35. frogml_storage/authentication/models/_auth_config.py +0 -70
  36. frogml_storage/authentication/models/_login.py +0 -22
  37. frogml_storage/authentication/utils/__init__.py +0 -17
  38. frogml_storage/authentication/utils/_authentication_utils.py +0 -281
  39. frogml_storage/authentication/utils/_login_checks_utils.py +0 -114
  40. frogml_storage/base_storage.py +0 -140
  41. frogml_storage/constants.py +0 -56
  42. frogml_storage/exceptions/checksum_verification_error.py +0 -3
  43. frogml_storage/exceptions/validation_error.py +0 -4
  44. frogml_storage/frog_ml.py +0 -668
  45. frogml_storage/http/__init__.py +0 -1
  46. frogml_storage/http/http_client.py +0 -83
  47. frogml_storage/logging/__init__.py +0 -1
  48. frogml_storage/logging/_log_config.py +0 -45
  49. frogml_storage/logging/log_utils.py +0 -21
  50. frogml_storage/models/__init__.py +0 -1
  51. frogml_storage/models/_download_context.py +0 -54
  52. frogml_storage/models/dataset_manifest.py +0 -13
  53. frogml_storage/models/entity_manifest.py +0 -93
  54. frogml_storage/models/frogml_dataset_version.py +0 -21
  55. frogml_storage/models/frogml_entity_type_info.py +0 -50
  56. frogml_storage/models/frogml_entity_version.py +0 -34
  57. frogml_storage/models/frogml_model_version.py +0 -21
  58. frogml_storage/models/model_manifest.py +0 -60
  59. frogml_storage/models/serialization_metadata.py +0 -15
  60. frogml_storage/utils/__init__.py +0 -12
  61. frogml_storage/utils/_environment.py +0 -21
  62. frogml_storage/utils/_input_checks_utility.py +0 -104
  63. frogml_storage/utils/_storage_utils.py +0 -15
  64. frogml_storage/utils/_url_utils.py +0 -27
  65. qwak_core-0.4.366.dist-info/METADATA +0 -414
  66. {qwak_core-0.4.366.dist-info → qwak_core-0.4.392.dist-info}/WHEEL +0 -0
@@ -14,7 +14,7 @@ _sym_db = _symbol_database.Default()
14
14
  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
15
15
 
16
16
 
17
- 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\"\xb6\x01\n%DatabricksServicePrincipalCredentials\x12\x11\n\tclient_id\x18\x01 \x01(\t\x12`\n\rclient_secret\x18\x02 \x01(\x0b\x32G.qwak.administration.runtime_configuration.v0.creds.secret.SecretKeyRefH\x00\x42\x18\n\x16\x63lient_secret_location\"~\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')
17
+ 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')
18
18
 
19
19
  _globals = globals()
20
20
  _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
@@ -26,7 +26,9 @@ if _descriptor._USE_C_DESCRIPTORS == False:
26
26
  _globals['_DATABRICKSAUTH']._serialized_start=211
27
27
  _globals['_DATABRICKSAUTH']._serialized_end=498
28
28
  _globals['_DATABRICKSSERVICEPRINCIPALCREDENTIALS']._serialized_start=501
29
- _globals['_DATABRICKSSERVICEPRINCIPALCREDENTIALS']._serialized_end=683
30
- _globals['_DATABRICKSACCESSTOKEN']._serialized_start=685
31
- _globals['_DATABRICKSACCESSTOKEN']._serialized_end=811
29
+ _globals['_DATABRICKSSERVICEPRINCIPALCREDENTIALS']._serialized_end=911
30
+ _globals['_DATABRICKSWORKSPACEDESCRIPTOR']._serialized_start=913
31
+ _globals['_DATABRICKSWORKSPACEDESCRIPTOR']._serialized_end=971
32
+ _globals['_DATABRICKSACCESSTOKEN']._serialized_start=973
33
+ _globals['_DATABRICKSACCESSTOKEN']._serialized_end=1099
32
34
  # @@protoc_insertion_point(module_scope)
@@ -7,6 +7,7 @@ import google.protobuf.descriptor
7
7
  import google.protobuf.message
8
8
  import qwak.administration.runtime_configuration.v0.creds.secret_pb2
9
9
  import sys
10
+ import typing
10
11
 
11
12
  if sys.version_info >= (3, 8):
12
13
  import typing as typing_extensions
@@ -41,21 +42,43 @@ class DatabricksServicePrincipalCredentials(google.protobuf.message.Message):
41
42
 
42
43
  CLIENT_ID_FIELD_NUMBER: builtins.int
43
44
  CLIENT_SECRET_FIELD_NUMBER: builtins.int
44
- client_id: builtins.str
45
+ DB_WORKSPACE_DESCRIPTOR_FIELD_NUMBER: builtins.int
46
+ @property
47
+ def client_id(self) -> qwak.administration.runtime_configuration.v0.creds.secret_pb2.SecretKeyRef: ...
45
48
  @property
46
49
  def client_secret(self) -> qwak.administration.runtime_configuration.v0.creds.secret_pb2.SecretKeyRef: ...
50
+ @property
51
+ def db_workspace_descriptor(self) -> global___DatabricksWorkspaceDescriptor: ...
47
52
  def __init__(
48
53
  self,
49
54
  *,
50
- client_id: builtins.str = ...,
55
+ client_id: qwak.administration.runtime_configuration.v0.creds.secret_pb2.SecretKeyRef | None = ...,
51
56
  client_secret: qwak.administration.runtime_configuration.v0.creds.secret_pb2.SecretKeyRef | None = ...,
57
+ db_workspace_descriptor: global___DatabricksWorkspaceDescriptor | None = ...,
52
58
  ) -> None: ...
53
- def HasField(self, field_name: typing_extensions.Literal["client_secret", b"client_secret", "client_secret_location", b"client_secret_location"]) -> builtins.bool: ...
54
- def ClearField(self, field_name: typing_extensions.Literal["client_id", b"client_id", "client_secret", b"client_secret", "client_secret_location", b"client_secret_location"]) -> None: ...
59
+ 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: ...
60
+ 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: ...
61
+ @typing.overload
62
+ def WhichOneof(self, oneof_group: typing_extensions.Literal["client_id_location", b"client_id_location"]) -> typing_extensions.Literal["client_id"] | None: ...
63
+ @typing.overload
55
64
  def WhichOneof(self, oneof_group: typing_extensions.Literal["client_secret_location", b"client_secret_location"]) -> typing_extensions.Literal["client_secret"] | None: ...
56
65
 
57
66
  global___DatabricksServicePrincipalCredentials = DatabricksServicePrincipalCredentials
58
67
 
68
+ class DatabricksWorkspaceDescriptor(google.protobuf.message.Message):
69
+ DESCRIPTOR: google.protobuf.descriptor.Descriptor
70
+
71
+ OAUTH_SERVICE_URL_FIELD_NUMBER: builtins.int
72
+ oauth_service_url: builtins.str
73
+ def __init__(
74
+ self,
75
+ *,
76
+ oauth_service_url: builtins.str = ...,
77
+ ) -> None: ...
78
+ def ClearField(self, field_name: typing_extensions.Literal["oauth_service_url", b"oauth_service_url"]) -> None: ...
79
+
80
+ global___DatabricksWorkspaceDescriptor = DatabricksWorkspaceDescriptor
81
+
59
82
  class DatabricksAccessToken(google.protobuf.message.Message):
60
83
  DESCRIPTOR: google.protobuf.descriptor.Descriptor
61
84
 
@@ -15,7 +15,7 @@ from _qwak_proto.qwak.administration.runtime_configuration.v0.creds import secre
15
15
  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
16
16
 
17
17
 
18
- 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\")\n\x14\x41zureManagedIdentity\x12\x11\n\tclient_id\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')
18
+ 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')
19
19
 
20
20
  _globals = globals()
21
21
  _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
@@ -32,6 +32,8 @@ if _descriptor._USE_C_DESCRIPTORS == False:
32
32
  _globals['_AZUREBLOBSTORAGEAUTH']._serialized_end=936
33
33
  _globals['_AZURESASTOKEN']._serialized_start=938
34
34
  _globals['_AZURESASTOKEN']._serialized_end=1056
35
- _globals['_AZUREMANAGEDIDENTITY']._serialized_start=1058
36
- _globals['_AZUREMANAGEDIDENTITY']._serialized_end=1099
35
+ _globals['_AZUREMANAGEDIDENTITY']._serialized_start=1059
36
+ _globals['_AZUREMANAGEDIDENTITY']._serialized_end=1244
37
+ _globals['_KUBERNETESFEDERATEDCREDENTIALSDESCRIPTOR']._serialized_start=1246
38
+ _globals['_KUBERNETESFEDERATEDCREDENTIALSDESCRIPTOR']._serialized_end=1318
37
39
  # @@protoc_insertion_point(module_scope)
@@ -95,12 +95,32 @@ class AzureManagedIdentity(google.protobuf.message.Message):
95
95
  DESCRIPTOR: google.protobuf.descriptor.Descriptor
96
96
 
97
97
  CLIENT_ID_FIELD_NUMBER: builtins.int
98
+ FEDERATED_CREDENTIALS_FIELD_NUMBER: builtins.int
98
99
  client_id: builtins.str
100
+ @property
101
+ def federated_credentials(self) -> global___KubernetesFederatedCredentialsDescriptor: ...
99
102
  def __init__(
100
103
  self,
101
104
  *,
102
105
  client_id: builtins.str = ...,
106
+ federated_credentials: global___KubernetesFederatedCredentialsDescriptor | None = ...,
103
107
  ) -> None: ...
104
- def ClearField(self, field_name: typing_extensions.Literal["client_id", b"client_id"]) -> None: ...
108
+ def HasField(self, field_name: typing_extensions.Literal["federated_credentials", b"federated_credentials", "type", b"type"]) -> builtins.bool: ...
109
+ def ClearField(self, field_name: typing_extensions.Literal["client_id", b"client_id", "federated_credentials", b"federated_credentials", "type", b"type"]) -> None: ...
110
+ def WhichOneof(self, oneof_group: typing_extensions.Literal["type", b"type"]) -> typing_extensions.Literal["federated_credentials"] | None: ...
105
111
 
106
112
  global___AzureManagedIdentity = AzureManagedIdentity
113
+
114
+ class KubernetesFederatedCredentialsDescriptor(google.protobuf.message.Message):
115
+ DESCRIPTOR: google.protobuf.descriptor.Descriptor
116
+
117
+ SERVICE_ACCOUNT_NAME_FIELD_NUMBER: builtins.int
118
+ service_account_name: builtins.str
119
+ def __init__(
120
+ self,
121
+ *,
122
+ service_account_name: builtins.str = ...,
123
+ ) -> None: ...
124
+ def ClearField(self, field_name: typing_extensions.Literal["service_account_name", b"service_account_name"]) -> None: ...
125
+
126
+ global___KubernetesFederatedCredentialsDescriptor = KubernetesFederatedCredentialsDescriptor
@@ -365,7 +365,7 @@ class BatchJobDataDetails(google.protobuf.message.Message):
365
365
  secret_secret: builtins.str
366
366
  """The access secret's secret name in secret service"""
367
367
  session_token: builtins.str
368
- """The session token required to use the bucket for batch inference"""
368
+ """The session token required to use the bucket for batch inference."""
369
369
  service_account_json_key_secret: builtins.str
370
370
  """The service account json key secret name"""
371
371
  def __init__(
@@ -13,9 +13,10 @@ _sym_db = _symbol_database.Default()
13
13
 
14
14
  from _qwak_proto.qwak.user_application.common.v0 import resources_pb2 as qwak_dot_user__application_dot_common_dot_v0_dot_resources__pb2
15
15
  from _qwak_proto.qwak.fitness_service import fitness_pb2 as qwak_dot_fitness__service_dot_fitness__pb2
16
+ from _qwak_proto.qwak.builds import build_values_pb2 as qwak_dot_builds_dot_build__values__pb2
16
17
 
17
18
 
18
- DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x17qwak/builds/build.proto\x12\x18qwak.builds.orchestrator\x1a/qwak/user_application/common/v0/resources.proto\x1a\"qwak/fitness_service/fitness.proto\"\xbf\x06\n\x0fRemoteBuildSpec\x12\x43\n\x10\x62uild_properties\x18\x01 \x01(\x0b\x32).qwak.builds.orchestrator.BuildProperties\x12\x35\n\tbuild_env\x18\x02 \x01(\x0b\x32\".qwak.builds.orchestrator.BuildEnv\x12\x43\n\rcpu_resources\x18\x03 \x01(\x0b\x32&.qwak.builds.orchestrator.CpuResourcesB\x02\x18\x01H\x00\x12\x43\n\rgpu_resources\x18\x07 \x01(\x0b\x32&.qwak.builds.orchestrator.GpuResourcesB\x02\x18\x01H\x00\x12\x62\n\x1c\x63lient_pod_compute_resources\x18\r \x01(\x0b\x32:.qwak.user_application.common.v0.ClientPodComputeResourcesH\x00\x12\x0f\n\x07verbose\x18\x04 \x01(\x05\x12\x17\n\x0f\x62uild_code_path\x18\x05 \x01(\t\x12\x14\n\x0c\x62uild_config\x18\x06 \x01(\t\x12\x15\n\rbuild_v1_flag\x18\x08 \x01(\x08\x12H\n\x13\x62uild_properties_v1\x18\t \x01(\x0b\x32+.qwak.builds.orchestrator.BuildPropertiesV1\x12\x13\n\x0b\x62uild_steps\x18\n \x03(\t\x12\x16\n\x0e\x65nvironment_id\x18\x0b \x01(\t\x12\x13\n\x0bsdk_version\x18\x0c \x01(\t\x12=\n\x0fpurchase_option\x18\x0e \x01(\x0e\x32$.qwak.fitness.service.PurchaseOption\x12\x1c\n\x14\x62uild_destined_image\x18\x0f \x01(\t\x12\"\n\x1aprovision_instance_timeout\x18\x10 \x01(\x05\x12\x16\n\x0ejfrog_token_id\x18\x11 \x01(\t\x12\x1f\n\x17\x61rtifactory_project_key\x18\x12 \x01(\t\x12\x18\n\x10model_version_id\x18\x13 \x01(\tB\x0b\n\tResources\"g\n\x11\x42uildPropertiesV1\x12\x18\n\x10\x62uild_config_url\x18\x01 \x01(\t\x12\x1a\n\x12qwak_sdk_wheel_url\x18\x02 \x01(\t\x12\x1c\n\x14qwak_sdk_version_url\x18\x03 \x01(\t\"\xba\x01\n\x0f\x42uildProperties\x12\x10\n\x08\x62uild_id\x18\x01 \x01(\t\x12\x0e\n\x06\x62ranch\x18\x02 \x01(\t\x12\x10\n\x08model_id\x18\x03 \x01(\t\x12\x0c\n\x04tags\x18\x04 \x03(\t\x12\x39\n\tmodel_uri\x18\x05 \x01(\x0b\x32&.qwak.builds.orchestrator.ModelUriSpec\x12\x16\n\x0egpu_compatible\x18\x06 \x01(\x08\x12\x12\n\nbuild_name\x18\x07 \x01(\t\"\xb2\x01\n\x0cModelUriSpec\x12\x0b\n\x03uri\x18\x01 \x01(\t\x12\x12\n\ngit_branch\x18\x02 \x01(\t\x12\x17\n\x0fgit_credentials\x18\x03 \x01(\t\x12\x1e\n\x16git_credentials_secret\x18\x04 \x01(\t\x12\x11\n\tcommit_id\x18\x05 \x01(\t\x12\x10\n\x08main_dir\x18\x06 \x01(\t\x12#\n\x1b\x64\x65pendency_required_folders\x18\x07 \x03(\t\"|\n\x08\x42uildEnv\x12\x37\n\ndocker_env\x18\x01 \x01(\x0b\x32#.qwak.builds.orchestrator.DockerEnv\x12\x37\n\npython_env\x18\x02 \x01(\x0b\x32#.qwak.builds.orchestrator.PythonEnv\"\x8a\x01\n\tDockerEnv\x12\x12\n\nbase_image\x18\x01 \x01(\t\x12\x10\n\x08\x65nv_vars\x18\x02 \x03(\t\x12\x10\n\x08no_cache\x18\x03 \x01(\x08\x12\x1c\n\x14\x61ssumed_iam_role_arn\x18\x04 \x01(\t\x12\'\n\x1fservice_account_key_secret_name\x18\x05 \x01(\t\"\xb8\x01\n\tPythonEnv\x12\x17\n\x0fgit_credentials\x18\x01 \x01(\t\x12\x1e\n\x16git_credentials_secret\x18\x02 \x01(\t\x12\x18\n\x10qwak_sdk_version\x18\x03 \x01(\t\x12\x1c\n\x14qwak_sdk_extra_index\x18\x04 \x01(\t\x12\x1a\n\x12\x66rogml_cli_version\x18\x05 \x01(\t\x12\x1e\n\x16\x66rogml_cli_extra_index\x18\x06 \x01(\t\"n\n\x0c\x43puResources\x12\x0b\n\x03\x63pu\x18\x01 \x01(\x02\x12\x15\n\rmemory_amount\x18\x02 \x01(\x05\x12:\n\x0cmemory_units\x18\x03 \x01(\x0e\x32$.qwak.builds.orchestrator.MemoryUnit\"W\n\x0cGpuResources\x12\x33\n\x08gpu_type\x18\x01 \x01(\x0e\x32!.qwak.builds.orchestrator.GpuType\x12\x12\n\ngpu_amount\x18\x02 \x01(\x05\"\x91\x01\n\x13\x44\x61taTableDefinition\x12?\n\x07\x63olumns\x18\x01 \x03(\x0b\x32..qwak.builds.orchestrator.DataColumnDefinition\x12\x39\n\x0b\x64\x61ta_format\x18\x02 \x01(\x0b\x32$.qwak.builds.orchestrator.DataFormat\"\\\n\x14\x44\x61taColumnDefinition\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x36\n\x04type\x18\x02 \x01(\x0e\x32(.qwak.builds.orchestrator.DataColumnType\"J\n\nDataFormat\x12\x32\n\x03\x63sv\x18\x01 \x01(\x0b\x32#.qwak.builds.orchestrator.CsvFormatH\x00\x42\x08\n\x06\x66ormat\"G\n\tCsvFormat\x12\x11\n\tdelimiter\x18\x01 \x01(\t\x12\x12\n\nquote_char\x18\x02 \x01(\t\x12\x13\n\x0b\x65scape_char\x18\x03 \x01(\t\"\xa5\x01\n\x0e\x42uildInitiator\x12<\n\x04user\x18\x01 \x01(\x0b\x32,.qwak.builds.orchestrator.UserBuildInitiatorH\x00\x12H\n\nautomation\x18\x02 \x01(\x0b\x32\x32.qwak.builds.orchestrator.AutomationBuildInitiatorH\x00\x42\x0b\n\tInitiator\"\x14\n\x12UserBuildInitiator\"`\n\x18\x41utomationBuildInitiator\x12\x15\n\rautomation_id\x18\x01 \x01(\t\x12\x14\n\x0c\x65xecution_id\x18\x02 \x01(\t\x12\x17\n\x0f\x61utomation_name\x18\x03 \x01(\t*7\n\nMemoryUnit\x12\x17\n\x13UNKNOWN_MEMORY_UNIT\x10\x00\x12\x07\n\x03MIB\x10\x01\x12\x07\n\x03GIB\x10\x02*\x82\x03\n\x07GpuType\x12\x0b\n\x07INVALID\x10\x00\x12\x0e\n\nNVIDIA_K80\x10\x01\x12\x0f\n\x0bNVIDIA_V100\x10\x02\x12\x0f\n\x0bNVIDIA_A100\x10\x03\x12\r\n\tNVIDIA_T4\x10\x04\x12\x0f\n\x0bNVIDIA_A10G\x10\x05\x12\r\n\tNVIDIA_L4\x10\x06\x12\x14\n\x10NVIDIA_T4_1_4_15\x10\x07\x12\x14\n\x10NVIDIA_T4_1_8_30\x10\x08\x12\x15\n\x11NVIDIA_T4_1_16_60\x10\t\x12\x1e\n\x1aNVIDIA_A100_80GB_8_96_1360\x10\n\x12\x16\n\x12NVIDIA_V100_1_8_52\x10\x0b\x12\x18\n\x14NVIDIA_V100_4_32_208\x10\x0c\x12\x13\n\x0fNVIDIA_H100_NVL\x10\r\x12\x0e\n\nNVIDIA_A10\x10\x0e\x12\x13\n\x0fNVIDIA_H100_SXM\x10\x0f\x12\x14\n\x10NVIDIA_H100_40GB\x10\x10\x12\x14\n\x10NVIDIA_H100_80GB\x10\x11\x12\x0e\n\nNVIDIA_M60\x10\x12*d\n\x0e\x44\x61taColumnType\x12\x17\n\x13INVALID_COLUMN_TYPE\x10\x00\x12\n\n\x06OBJECT\x10\x01\x12\x07\n\x03INT\x10\x02\x12\t\n\x05\x46LOAT\x10\x03\x12\x0b\n\x07\x42OOLEAN\x10\x04\x12\x0c\n\x08\x44\x41TETIME\x10\x05*K\n\x13\x42\x61seDockerImageType\x12\"\n\x1eUNKNOWN_BASE_DOCKER_IMAGE_TYPE\x10\x00\x12\x07\n\x03\x43PU\x10\x01\x12\x07\n\x03GPU\x10\x02\x42\'\n#com.qwak.ai.builds.orchestrator.apiP\x01\x62\x06proto3')
19
+ DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x17qwak/builds/build.proto\x12\x18qwak.builds.orchestrator\x1a/qwak/user_application/common/v0/resources.proto\x1a\"qwak/fitness_service/fitness.proto\x1a\x1eqwak/builds/build_values.proto\"\xbf\x06\n\x0fRemoteBuildSpec\x12\x43\n\x10\x62uild_properties\x18\x01 \x01(\x0b\x32).qwak.builds.orchestrator.BuildProperties\x12\x35\n\tbuild_env\x18\x02 \x01(\x0b\x32\".qwak.builds.orchestrator.BuildEnv\x12\x43\n\rcpu_resources\x18\x03 \x01(\x0b\x32&.qwak.builds.orchestrator.CpuResourcesB\x02\x18\x01H\x00\x12\x43\n\rgpu_resources\x18\x07 \x01(\x0b\x32&.qwak.builds.orchestrator.GpuResourcesB\x02\x18\x01H\x00\x12\x62\n\x1c\x63lient_pod_compute_resources\x18\r \x01(\x0b\x32:.qwak.user_application.common.v0.ClientPodComputeResourcesH\x00\x12\x0f\n\x07verbose\x18\x04 \x01(\x05\x12\x17\n\x0f\x62uild_code_path\x18\x05 \x01(\t\x12\x14\n\x0c\x62uild_config\x18\x06 \x01(\t\x12\x15\n\rbuild_v1_flag\x18\x08 \x01(\x08\x12H\n\x13\x62uild_properties_v1\x18\t \x01(\x0b\x32+.qwak.builds.orchestrator.BuildPropertiesV1\x12\x13\n\x0b\x62uild_steps\x18\n \x03(\t\x12\x16\n\x0e\x65nvironment_id\x18\x0b \x01(\t\x12\x13\n\x0bsdk_version\x18\x0c \x01(\t\x12=\n\x0fpurchase_option\x18\x0e \x01(\x0e\x32$.qwak.fitness.service.PurchaseOption\x12\x1c\n\x14\x62uild_destined_image\x18\x0f \x01(\t\x12\"\n\x1aprovision_instance_timeout\x18\x10 \x01(\x05\x12\x16\n\x0ejfrog_token_id\x18\x11 \x01(\t\x12\x1f\n\x17\x61rtifactory_project_key\x18\x12 \x01(\t\x12\x18\n\x10model_version_id\x18\x13 \x01(\tB\x0b\n\tResources\"g\n\x11\x42uildPropertiesV1\x12\x18\n\x10\x62uild_config_url\x18\x01 \x01(\t\x12\x1a\n\x12qwak_sdk_wheel_url\x18\x02 \x01(\t\x12\x1c\n\x14qwak_sdk_version_url\x18\x03 \x01(\t\"\xba\x01\n\x0f\x42uildProperties\x12\x10\n\x08\x62uild_id\x18\x01 \x01(\t\x12\x0e\n\x06\x62ranch\x18\x02 \x01(\t\x12\x10\n\x08model_id\x18\x03 \x01(\t\x12\x0c\n\x04tags\x18\x04 \x03(\t\x12\x39\n\tmodel_uri\x18\x05 \x01(\x0b\x32&.qwak.builds.orchestrator.ModelUriSpec\x12\x16\n\x0egpu_compatible\x18\x06 \x01(\x08\x12\x12\n\nbuild_name\x18\x07 \x01(\t\"\xb2\x01\n\x0cModelUriSpec\x12\x0b\n\x03uri\x18\x01 \x01(\t\x12\x12\n\ngit_branch\x18\x02 \x01(\t\x12\x17\n\x0fgit_credentials\x18\x03 \x01(\t\x12\x1e\n\x16git_credentials_secret\x18\x04 \x01(\t\x12\x11\n\tcommit_id\x18\x05 \x01(\t\x12\x10\n\x08main_dir\x18\x06 \x01(\t\x12#\n\x1b\x64\x65pendency_required_folders\x18\x07 \x03(\t\"|\n\x08\x42uildEnv\x12\x37\n\ndocker_env\x18\x01 \x01(\x0b\x32#.qwak.builds.orchestrator.DockerEnv\x12\x37\n\npython_env\x18\x02 \x01(\x0b\x32#.qwak.builds.orchestrator.PythonEnv\"\xcc\x01\n\tDockerEnv\x12\x12\n\nbase_image\x18\x01 \x01(\t\x12\x10\n\x08\x65nv_vars\x18\x02 \x03(\t\x12\x10\n\x08no_cache\x18\x03 \x01(\x08\x12\x1c\n\x14\x61ssumed_iam_role_arn\x18\x04 \x01(\t\x12\'\n\x1fservice_account_key_secret_name\x18\x05 \x01(\t\x12@\n\x10\x63ontainer_engine\x18\x06 \x01(\x0b\x32&.qwak.build.management.ContainerEngine\"\xb8\x01\n\tPythonEnv\x12\x17\n\x0fgit_credentials\x18\x01 \x01(\t\x12\x1e\n\x16git_credentials_secret\x18\x02 \x01(\t\x12\x18\n\x10qwak_sdk_version\x18\x03 \x01(\t\x12\x1c\n\x14qwak_sdk_extra_index\x18\x04 \x01(\t\x12\x1a\n\x12\x66rogml_cli_version\x18\x05 \x01(\t\x12\x1e\n\x16\x66rogml_cli_extra_index\x18\x06 \x01(\t\"n\n\x0c\x43puResources\x12\x0b\n\x03\x63pu\x18\x01 \x01(\x02\x12\x15\n\rmemory_amount\x18\x02 \x01(\x05\x12:\n\x0cmemory_units\x18\x03 \x01(\x0e\x32$.qwak.builds.orchestrator.MemoryUnit\"W\n\x0cGpuResources\x12\x33\n\x08gpu_type\x18\x01 \x01(\x0e\x32!.qwak.builds.orchestrator.GpuType\x12\x12\n\ngpu_amount\x18\x02 \x01(\x05\"\x91\x01\n\x13\x44\x61taTableDefinition\x12?\n\x07\x63olumns\x18\x01 \x03(\x0b\x32..qwak.builds.orchestrator.DataColumnDefinition\x12\x39\n\x0b\x64\x61ta_format\x18\x02 \x01(\x0b\x32$.qwak.builds.orchestrator.DataFormat\"\\\n\x14\x44\x61taColumnDefinition\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x36\n\x04type\x18\x02 \x01(\x0e\x32(.qwak.builds.orchestrator.DataColumnType\"J\n\nDataFormat\x12\x32\n\x03\x63sv\x18\x01 \x01(\x0b\x32#.qwak.builds.orchestrator.CsvFormatH\x00\x42\x08\n\x06\x66ormat\"G\n\tCsvFormat\x12\x11\n\tdelimiter\x18\x01 \x01(\t\x12\x12\n\nquote_char\x18\x02 \x01(\t\x12\x13\n\x0b\x65scape_char\x18\x03 \x01(\t\"\xa5\x01\n\x0e\x42uildInitiator\x12<\n\x04user\x18\x01 \x01(\x0b\x32,.qwak.builds.orchestrator.UserBuildInitiatorH\x00\x12H\n\nautomation\x18\x02 \x01(\x0b\x32\x32.qwak.builds.orchestrator.AutomationBuildInitiatorH\x00\x42\x0b\n\tInitiator\"\x14\n\x12UserBuildInitiator\"`\n\x18\x41utomationBuildInitiator\x12\x15\n\rautomation_id\x18\x01 \x01(\t\x12\x14\n\x0c\x65xecution_id\x18\x02 \x01(\t\x12\x17\n\x0f\x61utomation_name\x18\x03 \x01(\t*7\n\nMemoryUnit\x12\x17\n\x13UNKNOWN_MEMORY_UNIT\x10\x00\x12\x07\n\x03MIB\x10\x01\x12\x07\n\x03GIB\x10\x02*\x82\x03\n\x07GpuType\x12\x0b\n\x07INVALID\x10\x00\x12\x0e\n\nNVIDIA_K80\x10\x01\x12\x0f\n\x0bNVIDIA_V100\x10\x02\x12\x0f\n\x0bNVIDIA_A100\x10\x03\x12\r\n\tNVIDIA_T4\x10\x04\x12\x0f\n\x0bNVIDIA_A10G\x10\x05\x12\r\n\tNVIDIA_L4\x10\x06\x12\x14\n\x10NVIDIA_T4_1_4_15\x10\x07\x12\x14\n\x10NVIDIA_T4_1_8_30\x10\x08\x12\x15\n\x11NVIDIA_T4_1_16_60\x10\t\x12\x1e\n\x1aNVIDIA_A100_80GB_8_96_1360\x10\n\x12\x16\n\x12NVIDIA_V100_1_8_52\x10\x0b\x12\x18\n\x14NVIDIA_V100_4_32_208\x10\x0c\x12\x13\n\x0fNVIDIA_H100_NVL\x10\r\x12\x0e\n\nNVIDIA_A10\x10\x0e\x12\x13\n\x0fNVIDIA_H100_SXM\x10\x0f\x12\x14\n\x10NVIDIA_H100_40GB\x10\x10\x12\x14\n\x10NVIDIA_H100_80GB\x10\x11\x12\x0e\n\nNVIDIA_M60\x10\x12*d\n\x0e\x44\x61taColumnType\x12\x17\n\x13INVALID_COLUMN_TYPE\x10\x00\x12\n\n\x06OBJECT\x10\x01\x12\x07\n\x03INT\x10\x02\x12\t\n\x05\x46LOAT\x10\x03\x12\x0b\n\x07\x42OOLEAN\x10\x04\x12\x0c\n\x08\x44\x41TETIME\x10\x05*K\n\x13\x42\x61seDockerImageType\x12\"\n\x1eUNKNOWN_BASE_DOCKER_IMAGE_TYPE\x10\x00\x12\x07\n\x03\x43PU\x10\x01\x12\x07\n\x03GPU\x10\x02\x42\'\n#com.qwak.ai.builds.orchestrator.apiP\x01\x62\x06proto3')
19
20
 
20
21
  _globals = globals()
21
22
  _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
@@ -28,44 +29,44 @@ if _descriptor._USE_C_DESCRIPTORS == False:
28
29
  _REMOTEBUILDSPEC.fields_by_name['cpu_resources']._serialized_options = b'\030\001'
29
30
  _REMOTEBUILDSPEC.fields_by_name['gpu_resources']._options = None
30
31
  _REMOTEBUILDSPEC.fields_by_name['gpu_resources']._serialized_options = b'\030\001'
31
- _globals['_MEMORYUNIT']._serialized_start=2781
32
- _globals['_MEMORYUNIT']._serialized_end=2836
33
- _globals['_GPUTYPE']._serialized_start=2839
34
- _globals['_GPUTYPE']._serialized_end=3225
35
- _globals['_DATACOLUMNTYPE']._serialized_start=3227
36
- _globals['_DATACOLUMNTYPE']._serialized_end=3327
37
- _globals['_BASEDOCKERIMAGETYPE']._serialized_start=3329
38
- _globals['_BASEDOCKERIMAGETYPE']._serialized_end=3404
39
- _globals['_REMOTEBUILDSPEC']._serialized_start=139
40
- _globals['_REMOTEBUILDSPEC']._serialized_end=970
41
- _globals['_BUILDPROPERTIESV1']._serialized_start=972
42
- _globals['_BUILDPROPERTIESV1']._serialized_end=1075
43
- _globals['_BUILDPROPERTIES']._serialized_start=1078
44
- _globals['_BUILDPROPERTIES']._serialized_end=1264
45
- _globals['_MODELURISPEC']._serialized_start=1267
46
- _globals['_MODELURISPEC']._serialized_end=1445
47
- _globals['_BUILDENV']._serialized_start=1447
48
- _globals['_BUILDENV']._serialized_end=1571
49
- _globals['_DOCKERENV']._serialized_start=1574
50
- _globals['_DOCKERENV']._serialized_end=1712
51
- _globals['_PYTHONENV']._serialized_start=1715
52
- _globals['_PYTHONENV']._serialized_end=1899
53
- _globals['_CPURESOURCES']._serialized_start=1901
54
- _globals['_CPURESOURCES']._serialized_end=2011
55
- _globals['_GPURESOURCES']._serialized_start=2013
56
- _globals['_GPURESOURCES']._serialized_end=2100
57
- _globals['_DATATABLEDEFINITION']._serialized_start=2103
58
- _globals['_DATATABLEDEFINITION']._serialized_end=2248
59
- _globals['_DATACOLUMNDEFINITION']._serialized_start=2250
60
- _globals['_DATACOLUMNDEFINITION']._serialized_end=2342
61
- _globals['_DATAFORMAT']._serialized_start=2344
62
- _globals['_DATAFORMAT']._serialized_end=2418
63
- _globals['_CSVFORMAT']._serialized_start=2420
64
- _globals['_CSVFORMAT']._serialized_end=2491
65
- _globals['_BUILDINITIATOR']._serialized_start=2494
66
- _globals['_BUILDINITIATOR']._serialized_end=2659
67
- _globals['_USERBUILDINITIATOR']._serialized_start=2661
68
- _globals['_USERBUILDINITIATOR']._serialized_end=2681
69
- _globals['_AUTOMATIONBUILDINITIATOR']._serialized_start=2683
70
- _globals['_AUTOMATIONBUILDINITIATOR']._serialized_end=2779
32
+ _globals['_MEMORYUNIT']._serialized_start=2879
33
+ _globals['_MEMORYUNIT']._serialized_end=2934
34
+ _globals['_GPUTYPE']._serialized_start=2937
35
+ _globals['_GPUTYPE']._serialized_end=3323
36
+ _globals['_DATACOLUMNTYPE']._serialized_start=3325
37
+ _globals['_DATACOLUMNTYPE']._serialized_end=3425
38
+ _globals['_BASEDOCKERIMAGETYPE']._serialized_start=3427
39
+ _globals['_BASEDOCKERIMAGETYPE']._serialized_end=3502
40
+ _globals['_REMOTEBUILDSPEC']._serialized_start=171
41
+ _globals['_REMOTEBUILDSPEC']._serialized_end=1002
42
+ _globals['_BUILDPROPERTIESV1']._serialized_start=1004
43
+ _globals['_BUILDPROPERTIESV1']._serialized_end=1107
44
+ _globals['_BUILDPROPERTIES']._serialized_start=1110
45
+ _globals['_BUILDPROPERTIES']._serialized_end=1296
46
+ _globals['_MODELURISPEC']._serialized_start=1299
47
+ _globals['_MODELURISPEC']._serialized_end=1477
48
+ _globals['_BUILDENV']._serialized_start=1479
49
+ _globals['_BUILDENV']._serialized_end=1603
50
+ _globals['_DOCKERENV']._serialized_start=1606
51
+ _globals['_DOCKERENV']._serialized_end=1810
52
+ _globals['_PYTHONENV']._serialized_start=1813
53
+ _globals['_PYTHONENV']._serialized_end=1997
54
+ _globals['_CPURESOURCES']._serialized_start=1999
55
+ _globals['_CPURESOURCES']._serialized_end=2109
56
+ _globals['_GPURESOURCES']._serialized_start=2111
57
+ _globals['_GPURESOURCES']._serialized_end=2198
58
+ _globals['_DATATABLEDEFINITION']._serialized_start=2201
59
+ _globals['_DATATABLEDEFINITION']._serialized_end=2346
60
+ _globals['_DATACOLUMNDEFINITION']._serialized_start=2348
61
+ _globals['_DATACOLUMNDEFINITION']._serialized_end=2440
62
+ _globals['_DATAFORMAT']._serialized_start=2442
63
+ _globals['_DATAFORMAT']._serialized_end=2516
64
+ _globals['_CSVFORMAT']._serialized_start=2518
65
+ _globals['_CSVFORMAT']._serialized_end=2589
66
+ _globals['_BUILDINITIATOR']._serialized_start=2592
67
+ _globals['_BUILDINITIATOR']._serialized_end=2757
68
+ _globals['_USERBUILDINITIATOR']._serialized_start=2759
69
+ _globals['_USERBUILDINITIATOR']._serialized_end=2779
70
+ _globals['_AUTOMATIONBUILDINITIATOR']._serialized_start=2781
71
+ _globals['_AUTOMATIONBUILDINITIATOR']._serialized_end=2877
71
72
  # @@protoc_insertion_point(module_scope)
@@ -8,6 +8,7 @@ import google.protobuf.descriptor
8
8
  import google.protobuf.internal.containers
9
9
  import google.protobuf.internal.enum_type_wrapper
10
10
  import google.protobuf.message
11
+ import qwak.builds.build_values_pb2
11
12
  import qwak.fitness_service.fitness_pb2
12
13
  import qwak.user_application.common.v0.resources_pb2
13
14
  import sys
@@ -380,6 +381,7 @@ class DockerEnv(google.protobuf.message.Message):
380
381
  NO_CACHE_FIELD_NUMBER: builtins.int
381
382
  ASSUMED_IAM_ROLE_ARN_FIELD_NUMBER: builtins.int
382
383
  SERVICE_ACCOUNT_KEY_SECRET_NAME_FIELD_NUMBER: builtins.int
384
+ CONTAINER_ENGINE_FIELD_NUMBER: builtins.int
383
385
  base_image: builtins.str
384
386
  """Base image to use for serving"""
385
387
  @property
@@ -393,6 +395,9 @@ class DockerEnv(google.protobuf.message.Message):
393
395
  """Custom IAM role to be assumed by model"""
394
396
  service_account_key_secret_name: builtins.str
395
397
  """Service account key secret name for gcp"""
398
+ @property
399
+ def container_engine(self) -> qwak.builds.build_values_pb2.ContainerEngine:
400
+ """Container engine to use for building and pushing the image (e.g., Docker, Podman)"""
396
401
  def __init__(
397
402
  self,
398
403
  *,
@@ -401,8 +406,10 @@ class DockerEnv(google.protobuf.message.Message):
401
406
  no_cache: builtins.bool = ...,
402
407
  assumed_iam_role_arn: builtins.str = ...,
403
408
  service_account_key_secret_name: builtins.str = ...,
409
+ container_engine: qwak.builds.build_values_pb2.ContainerEngine | None = ...,
404
410
  ) -> None: ...
405
- def ClearField(self, field_name: typing_extensions.Literal["assumed_iam_role_arn", b"assumed_iam_role_arn", "base_image", b"base_image", "env_vars", b"env_vars", "no_cache", b"no_cache", "service_account_key_secret_name", b"service_account_key_secret_name"]) -> None: ...
411
+ def HasField(self, field_name: typing_extensions.Literal["container_engine", b"container_engine"]) -> builtins.bool: ...
412
+ def ClearField(self, field_name: typing_extensions.Literal["assumed_iam_role_arn", b"assumed_iam_role_arn", "base_image", b"base_image", "container_engine", b"container_engine", "env_vars", b"env_vars", "no_cache", b"no_cache", "service_account_key_secret_name", b"service_account_key_secret_name"]) -> None: ...
406
413
 
407
414
  global___DockerEnv = DockerEnv
408
415
 
@@ -0,0 +1,76 @@
1
+ # -*- coding: utf-8 -*-
2
+ # Generated by the protocol buffer compiler. DO NOT EDIT!
3
+ # source: qwak/builds/build_values.proto
4
+ """Generated protocol buffer code."""
5
+ from google.protobuf import descriptor as _descriptor
6
+ from google.protobuf import descriptor_pool as _descriptor_pool
7
+ from google.protobuf import symbol_database as _symbol_database
8
+ from google.protobuf.internal import builder as _builder
9
+ # @@protoc_insertion_point(imports)
10
+
11
+ _sym_db = _symbol_database.Default()
12
+
13
+
14
+ from _qwak_proto.qwak.user_application.common.v0 import resources_pb2 as qwak_dot_user__application_dot_common_dot_v0_dot_resources__pb2
15
+
16
+
17
+ DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1eqwak/builds/build_values.proto\x12\x15qwak.build.management\x1a/qwak/user_application/common/v0/resources.proto\"\xc3\x03\n\x10ModelBuildValues\x12\x34\n\nbuild_info\x18\x01 \x01(\x0b\x32 .qwak.build.management.BuildInfo\x12<\n\x0emodel_metadata\x18\x02 \x01(\x0b\x32$.qwak.build.management.ModelMetadata\x12\x45\n\x13model_code_location\x18\x03 \x01(\x0b\x32(.qwak.build.management.ModelCodeLocation\x12G\n\x14\x62uild_execution_spec\x18\x04 \x01(\x0b\x32).qwak.build.management.BuildExecutionSpec\x12\x30\n\x08sdk_spec\x18\x05 \x01(\x0b\x32\x1e.qwak.build.management.SdkSpec\x12G\n\x14jfrog_ecosystem_spec\x18\x06 \x01(\x0b\x32).qwak.build.management.JfrogEcosystemSpec\x12\x30\n\x08job_spec\x18\x07 \x01(\x0b\x32\x1e.qwak.build.management.JobSpec\"\xa3\x01\n\tBuildInfo\x12\x10\n\x08\x62uild_id\x18\x01 \x01(\t\x12\x13\n\x0b\x62ranch_name\x18\x02 \x01(\t\x12\x12\n\nbuild_tags\x18\x03 \x03(\t\x12\x14\n\x0c\x62uild_config\x18\x04 \x01(\t\x12\x45\n\x13\x62uild_properties_v1\x18\x05 \x01(\x0b\x32(.qwak.build.management.BuildPropertiesV1\"]\n\x11\x42uildPropertiesV1\x12\x18\n\x10\x62uild_config_url\x18\x01 \x01(\t\x12\x15\n\rsdk_wheel_url\x18\x02 \x01(\t\x12\x17\n\x0fsdk_version_url\x18\x03 \x01(\t\"\x8e\x01\n\rModelMetadata\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x12\n\naccount_id\x18\x02 \x01(\t\x12\x0f\n\x07user_id\x18\x03 \x01(\t\x12\x46\n\x13\x61rtifactory_details\x18\x04 \x01(\x0b\x32).qwak.build.management.ArtifactoryDetails\")\n\x12\x41rtifactoryDetails\x12\x13\n\x0bproject_key\x18\x01 \x01(\t\"\xa0\x01\n\x11ModelCodeLocation\x12\x1c\n\x14main_model_directory\x18\x01 \x01(\t\x12\x34\n\tlocal_uri\x18\x02 \x01(\x0b\x32\x1f.qwak.build.management.LocalUriH\x00\x12\x30\n\x07git_uri\x18\x03 \x01(\x0b\x32\x1d.qwak.build.management.GitUriH\x00\x42\x05\n\x03uri\"\x18\n\x08LocalUri\x12\x0c\n\x04path\x18\x01 \x01(\t\"M\n\x06GitUri\x12\x0b\n\x03url\x18\x01 \x01(\t\x12\x36\n\x0bgit_details\x18\x02 \x01(\x0b\x32!.qwak.build.management.GitDetails\"|\n\nGitDetails\x12\x0e\n\x06\x62ranch\x18\x01 \x01(\t\x12\x11\n\tcommit_id\x18\x02 \x01(\t\x12\x15\n\x0b\x63redentials\x18\x03 \x01(\tH\x00\x12!\n\x17\x63redentials_secret_name\x18\x04 \x01(\tH\x00\x42\x11\n\x0fgit_credentials\"\xd0\x02\n\x12\x42uildExecutionSpec\x12\x1f\n\x17\x62uild_image_destination\x18\x01 \x01(\t\x12\"\n\x1ashould_build_without_cache\x18\x02 \x01(\x08\x12G\n\x14logs_verbosity_level\x18\x03 \x01(\x0e\x32).qwak.build.management.LogsVerbosityLevel\x12\x12\n\nbase_image\x18\x04 \x01(\t\x12\x1b\n\x13\x63ustom_iam_role_arn\x18\x05 \x01(\t\x12\x10\n\x08\x65nv_vars\x18\x06 \x03(\t\x12\'\n\x1fremote_builder_auth_secret_name\x18\x07 \x01(\t\x12@\n\x10\x63ontainer_engine\x18\x08 \x01(\x0b\x32&.qwak.build.management.ContainerEngine\"a\n\x07SdkSpec\x12\x13\n\x0bsdk_version\x18\x01 \x01(\t\x12\x1b\n\x13sdk_extra_index_url\x18\x02 \x01(\t\x12$\n\x1csdk_runtime_version_override\x18\x03 \x01(\t\"k\n\x12JfrogEcosystemSpec\x12\x0f\n\x07\x62\x61seUrl\x18\x01 \x01(\t\x12\x1d\n\x15\x61\x63\x63\x65ssTokenSecretName\x18\x02 \x01(\t\x12%\n\x1d\x61rtifactorySettingsSecretName\x18\x03 \x01(\t\"\x9d\x03\n\x07JobSpec\x12\x11\n\tnamespace\x18\x01 \x01(\t\x12\"\n\x1aprovision_instance_timeout\x18\x02 \x01(\x05\x12>\n\x0fpurchase_option\x18\x03 \x01(\x0e\x32%.qwak.build.management.PurchaseOption\x12$\n\x1cis_node_optimization_enabled\x18\x04 \x01(\x08\x12\"\n\x1ashould_use_build_isolation\x18\x05 \x01(\x08\x12#\n\x1bis_multi_tenant_environment\x18\x06 \x01(\x08\x12\x1a\n\x12is_self_hosted_jpd\x18\n \x01(\x08\x12 \n\x18\x62uild_executor_image_url\x18\x07 \x01(\t\x12<\n\x0e\x63loud_provider\x18\x08 \x01(\x0e\x32$.qwak.build.management.CloudProvider\x12\x30\n\x08job_size\x18\t \x01(\x0b\x32\x1e.qwak.build.management.JobSize\"\x9e\x02\n\x07JobSize\x12\x0f\n\x03\x63pu\x18\x01 \x01(\x02\x42\x02\x18\x01\x12\x19\n\rmemory_amount\x18\x02 \x01(\x05\x42\x02\x18\x01\x12;\n\x0cmemory_units\x18\x03 \x01(\x0e\x32!.qwak.build.management.MemoryUnitB\x02\x18\x01\x12H\n\rgpu_resources\x18\x04 \x01(\x0b\x32-.qwak.user_application.common.v0.GpuResourcesB\x02\x18\x01\x12`\n\x1c\x63lient_pod_compute_resources\x18\x05 \x01(\x0b\x32:.qwak.user_application.common.v0.ClientPodComputeResources\"\x99\x01\n\x0f\x43ontainerEngine\x12>\n\x06\x64ocker\x18\x01 \x01(\x0b\x32,.qwak.build.management.DockerContainerEngineH\x00\x12>\n\x06podman\x18\x02 \x01(\x0b\x32,.qwak.build.management.PodmanContainerEngineH\x00\x42\x06\n\x04type\"\x17\n\x15\x44ockerContainerEngine\"\x17\n\x15PodmanContainerEngine*e\n\x0ePurchaseOption\x12\x1b\n\x17INVALID_PURCHASE_OPTION\x10\x00\x12\x18\n\x14SPOT_PURCHASE_OPTION\x10\x01\x12\x1c\n\x18ONDEMAND_PURCHASE_OPTION\x10\x02*H\n\rCloudProvider\x12\x1a\n\x16INVALID_CLOUD_PROVIDER\x10\x00\x12\x07\n\x03\x41WS\x10\x01\x12\x07\n\x03GCP\x10\x02\x12\t\n\x05\x41ZURE\x10\x03*7\n\nMemoryUnit\x12\x17\n\x13INVALID_MEMORY_UNIT\x10\x00\x12\x07\n\x03MIB\x10\x01\x12\x07\n\x03GIB\x10\x02*S\n\x12LogsVerbosityLevel\x12\x1a\n\x16INVALID_LOGS_VERBOSITY\x10\x00\x12\t\n\x05\x44\x45\x42UG\x10\n\x12\x08\n\x04INFO\x10\x14\x12\x0c\n\x08\x43RITICAL\x10\x32\x42\x36\n com.qwak.ai.management.build.apiP\x01Z\x10qwak/build;buildb\x06proto3')
18
+
19
+ _globals = globals()
20
+ _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
21
+ _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.builds.build_values_pb2', _globals)
22
+ if _descriptor._USE_C_DESCRIPTORS == False:
23
+
24
+ DESCRIPTOR._options = None
25
+ DESCRIPTOR._serialized_options = b'\n com.qwak.ai.management.build.apiP\001Z\020qwak/build;build'
26
+ _JOBSIZE.fields_by_name['cpu']._options = None
27
+ _JOBSIZE.fields_by_name['cpu']._serialized_options = b'\030\001'
28
+ _JOBSIZE.fields_by_name['memory_amount']._options = None
29
+ _JOBSIZE.fields_by_name['memory_amount']._serialized_options = b'\030\001'
30
+ _JOBSIZE.fields_by_name['memory_units']._options = None
31
+ _JOBSIZE.fields_by_name['memory_units']._serialized_options = b'\030\001'
32
+ _JOBSIZE.fields_by_name['gpu_resources']._options = None
33
+ _JOBSIZE.fields_by_name['gpu_resources']._serialized_options = b'\030\001'
34
+ _globals['_PURCHASEOPTION']._serialized_start=2861
35
+ _globals['_PURCHASEOPTION']._serialized_end=2962
36
+ _globals['_CLOUDPROVIDER']._serialized_start=2964
37
+ _globals['_CLOUDPROVIDER']._serialized_end=3036
38
+ _globals['_MEMORYUNIT']._serialized_start=3038
39
+ _globals['_MEMORYUNIT']._serialized_end=3093
40
+ _globals['_LOGSVERBOSITYLEVEL']._serialized_start=3095
41
+ _globals['_LOGSVERBOSITYLEVEL']._serialized_end=3178
42
+ _globals['_MODELBUILDVALUES']._serialized_start=107
43
+ _globals['_MODELBUILDVALUES']._serialized_end=558
44
+ _globals['_BUILDINFO']._serialized_start=561
45
+ _globals['_BUILDINFO']._serialized_end=724
46
+ _globals['_BUILDPROPERTIESV1']._serialized_start=726
47
+ _globals['_BUILDPROPERTIESV1']._serialized_end=819
48
+ _globals['_MODELMETADATA']._serialized_start=822
49
+ _globals['_MODELMETADATA']._serialized_end=964
50
+ _globals['_ARTIFACTORYDETAILS']._serialized_start=966
51
+ _globals['_ARTIFACTORYDETAILS']._serialized_end=1007
52
+ _globals['_MODELCODELOCATION']._serialized_start=1010
53
+ _globals['_MODELCODELOCATION']._serialized_end=1170
54
+ _globals['_LOCALURI']._serialized_start=1172
55
+ _globals['_LOCALURI']._serialized_end=1196
56
+ _globals['_GITURI']._serialized_start=1198
57
+ _globals['_GITURI']._serialized_end=1275
58
+ _globals['_GITDETAILS']._serialized_start=1277
59
+ _globals['_GITDETAILS']._serialized_end=1401
60
+ _globals['_BUILDEXECUTIONSPEC']._serialized_start=1404
61
+ _globals['_BUILDEXECUTIONSPEC']._serialized_end=1740
62
+ _globals['_SDKSPEC']._serialized_start=1742
63
+ _globals['_SDKSPEC']._serialized_end=1839
64
+ _globals['_JFROGECOSYSTEMSPEC']._serialized_start=1841
65
+ _globals['_JFROGECOSYSTEMSPEC']._serialized_end=1948
66
+ _globals['_JOBSPEC']._serialized_start=1951
67
+ _globals['_JOBSPEC']._serialized_end=2364
68
+ _globals['_JOBSIZE']._serialized_start=2367
69
+ _globals['_JOBSIZE']._serialized_end=2653
70
+ _globals['_CONTAINERENGINE']._serialized_start=2656
71
+ _globals['_CONTAINERENGINE']._serialized_end=2809
72
+ _globals['_DOCKERCONTAINERENGINE']._serialized_start=2811
73
+ _globals['_DOCKERCONTAINERENGINE']._serialized_end=2834
74
+ _globals['_PODMANCONTAINERENGINE']._serialized_start=2836
75
+ _globals['_PODMANCONTAINERENGINE']._serialized_end=2859
76
+ # @@protoc_insertion_point(module_scope)