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.
- _qwak_proto/qwak/administration/runtime_configuration/v0/external/databricks/auth_pb2.py +6 -4
- _qwak_proto/qwak/administration/runtime_configuration/v0/external/databricks/auth_pb2.pyi +27 -4
- _qwak_proto/qwak/administration/runtime_configuration/v0/hosting/azure/auth_pb2.py +5 -3
- _qwak_proto/qwak/administration/runtime_configuration/v0/hosting/azure/auth_pb2.pyi +21 -1
- _qwak_proto/qwak/batch_job/v1/batch_job_service_pb2.pyi +1 -1
- _qwak_proto/qwak/builds/build_pb2.py +42 -41
- _qwak_proto/qwak/builds/build_pb2.pyi +8 -1
- _qwak_proto/qwak/builds/build_values_pb2.py +76 -0
- _qwak_proto/qwak/builds/build_values_pb2.pyi +533 -0
- _qwak_proto/qwak/builds/build_values_pb2_grpc.py +4 -0
- _qwak_proto/qwak/kube_deployment_captain/batch_job_pb2.pyi +1 -1
- _qwak_proto/qwak/projects/projects_pb2.py +17 -15
- qwak/__init__.py +1 -1
- qwak/clients/model_management/client.py +0 -5
- qwak/clients/project/client.py +0 -7
- qwak/feature_store/_common/packaging.py +11 -5
- qwak/inner/const.py +0 -2
- qwak/inner/di_configuration/account.py +6 -66
- qwak/inner/tool/auth.py +0 -86
- qwak/inner/tool/grpc/grpc_auth.py +0 -32
- qwak/inner/tool/grpc/grpc_tools.py +2 -8
- qwak/inner/tool/grpc/grpc_try_wrapping.py +3 -1
- qwak/qwak_client/client.py +2 -8
- qwak/vector_store/rest_helpers.py +4 -16
- qwak_core-0.4.392.dist-info/METADATA +50 -0
- {qwak_core-0.4.366.dist-info → qwak_core-0.4.392.dist-info}/RECORD +27 -62
- frogml_storage/__init__.py +0 -1
- frogml_storage/artifactory/__init__.py +0 -1
- frogml_storage/artifactory/_artifactory_api.py +0 -315
- frogml_storage/authentication/login/__init__.py +0 -1
- frogml_storage/authentication/login/_login_cli.py +0 -239
- frogml_storage/authentication/login/_login_command.py +0 -74
- frogml_storage/authentication/models/__init__.py +0 -3
- frogml_storage/authentication/models/_auth.py +0 -24
- frogml_storage/authentication/models/_auth_config.py +0 -70
- frogml_storage/authentication/models/_login.py +0 -22
- frogml_storage/authentication/utils/__init__.py +0 -17
- frogml_storage/authentication/utils/_authentication_utils.py +0 -281
- frogml_storage/authentication/utils/_login_checks_utils.py +0 -114
- frogml_storage/base_storage.py +0 -140
- frogml_storage/constants.py +0 -56
- frogml_storage/exceptions/checksum_verification_error.py +0 -3
- frogml_storage/exceptions/validation_error.py +0 -4
- frogml_storage/frog_ml.py +0 -668
- frogml_storage/http/__init__.py +0 -1
- frogml_storage/http/http_client.py +0 -83
- frogml_storage/logging/__init__.py +0 -1
- frogml_storage/logging/_log_config.py +0 -45
- frogml_storage/logging/log_utils.py +0 -21
- frogml_storage/models/__init__.py +0 -1
- frogml_storage/models/_download_context.py +0 -54
- frogml_storage/models/dataset_manifest.py +0 -13
- frogml_storage/models/entity_manifest.py +0 -93
- frogml_storage/models/frogml_dataset_version.py +0 -21
- frogml_storage/models/frogml_entity_type_info.py +0 -50
- frogml_storage/models/frogml_entity_version.py +0 -34
- frogml_storage/models/frogml_model_version.py +0 -21
- frogml_storage/models/model_manifest.py +0 -60
- frogml_storage/models/serialization_metadata.py +0 -15
- frogml_storage/utils/__init__.py +0 -12
- frogml_storage/utils/_environment.py +0 -21
- frogml_storage/utils/_input_checks_utility.py +0 -104
- frogml_storage/utils/_storage_utils.py +0 -15
- frogml_storage/utils/_url_utils.py +0 -27
- qwak_core-0.4.366.dist-info/METADATA +0 -414
- {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\"\
|
|
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=
|
|
30
|
-
_globals['
|
|
31
|
-
_globals['
|
|
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
|
-
|
|
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:
|
|
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\"
|
|
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=
|
|
36
|
-
_globals['_AZUREMANAGEDIDENTITY']._serialized_end=
|
|
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
|
|
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\"\
|
|
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=
|
|
32
|
-
_globals['_MEMORYUNIT']._serialized_end=
|
|
33
|
-
_globals['_GPUTYPE']._serialized_start=
|
|
34
|
-
_globals['_GPUTYPE']._serialized_end=
|
|
35
|
-
_globals['_DATACOLUMNTYPE']._serialized_start=
|
|
36
|
-
_globals['_DATACOLUMNTYPE']._serialized_end=
|
|
37
|
-
_globals['_BASEDOCKERIMAGETYPE']._serialized_start=
|
|
38
|
-
_globals['_BASEDOCKERIMAGETYPE']._serialized_end=
|
|
39
|
-
_globals['_REMOTEBUILDSPEC']._serialized_start=
|
|
40
|
-
_globals['_REMOTEBUILDSPEC']._serialized_end=
|
|
41
|
-
_globals['_BUILDPROPERTIESV1']._serialized_start=
|
|
42
|
-
_globals['_BUILDPROPERTIESV1']._serialized_end=
|
|
43
|
-
_globals['_BUILDPROPERTIES']._serialized_start=
|
|
44
|
-
_globals['_BUILDPROPERTIES']._serialized_end=
|
|
45
|
-
_globals['_MODELURISPEC']._serialized_start=
|
|
46
|
-
_globals['_MODELURISPEC']._serialized_end=
|
|
47
|
-
_globals['_BUILDENV']._serialized_start=
|
|
48
|
-
_globals['_BUILDENV']._serialized_end=
|
|
49
|
-
_globals['_DOCKERENV']._serialized_start=
|
|
50
|
-
_globals['_DOCKERENV']._serialized_end=
|
|
51
|
-
_globals['_PYTHONENV']._serialized_start=
|
|
52
|
-
_globals['_PYTHONENV']._serialized_end=
|
|
53
|
-
_globals['_CPURESOURCES']._serialized_start=
|
|
54
|
-
_globals['_CPURESOURCES']._serialized_end=
|
|
55
|
-
_globals['_GPURESOURCES']._serialized_start=
|
|
56
|
-
_globals['_GPURESOURCES']._serialized_end=
|
|
57
|
-
_globals['_DATATABLEDEFINITION']._serialized_start=
|
|
58
|
-
_globals['_DATATABLEDEFINITION']._serialized_end=
|
|
59
|
-
_globals['_DATACOLUMNDEFINITION']._serialized_start=
|
|
60
|
-
_globals['_DATACOLUMNDEFINITION']._serialized_end=
|
|
61
|
-
_globals['_DATAFORMAT']._serialized_start=
|
|
62
|
-
_globals['_DATAFORMAT']._serialized_end=
|
|
63
|
-
_globals['_CSVFORMAT']._serialized_start=
|
|
64
|
-
_globals['_CSVFORMAT']._serialized_end=
|
|
65
|
-
_globals['_BUILDINITIATOR']._serialized_start=
|
|
66
|
-
_globals['_BUILDINITIATOR']._serialized_end=
|
|
67
|
-
_globals['_USERBUILDINITIATOR']._serialized_start=
|
|
68
|
-
_globals['_USERBUILDINITIATOR']._serialized_end=
|
|
69
|
-
_globals['_AUTOMATIONBUILDINITIATOR']._serialized_start=
|
|
70
|
-
_globals['_AUTOMATIONBUILDINITIATOR']._serialized_end=
|
|
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
|
|
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)
|