qwak-core 0.4.362__py3-none-any.whl → 0.5.4__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 (88) hide show
  1. _qwak_proto/qwak/administration/account/v1/account_pb2.py +20 -18
  2. _qwak_proto/qwak/administration/account/v1/account_pb2.pyi +21 -2
  3. _qwak_proto/qwak/administration/runtime_configuration/v0/external/databricks/auth_pb2.py +6 -4
  4. _qwak_proto/qwak/administration/runtime_configuration/v0/external/databricks/auth_pb2.pyi +27 -4
  5. _qwak_proto/qwak/administration/runtime_configuration/v0/hosting/azure/auth_pb2.py +5 -3
  6. _qwak_proto/qwak/administration/runtime_configuration/v0/hosting/azure/auth_pb2.pyi +21 -1
  7. _qwak_proto/qwak/admiral/secret/v0/secret_pb2.py +16 -14
  8. _qwak_proto/qwak/admiral/secret/v0/secret_pb2.pyi +21 -2
  9. _qwak_proto/qwak/batch_job/v1/batch_job_service_pb2.py +100 -100
  10. _qwak_proto/qwak/batch_job/v1/batch_job_service_pb2.pyi +5 -1
  11. _qwak_proto/qwak/builds/build_pb2.py +42 -41
  12. _qwak_proto/qwak/builds/build_pb2.pyi +32 -1
  13. _qwak_proto/qwak/builds/build_values_pb2.py +82 -0
  14. _qwak_proto/qwak/builds/build_values_pb2.pyi +553 -0
  15. _qwak_proto/qwak/builds/build_values_pb2_grpc.py +4 -0
  16. _qwak_proto/qwak/execution/v1/streaming_aggregation_pb2.py +18 -11
  17. _qwak_proto/qwak/execution/v1/streaming_aggregation_pb2.pyi +71 -1
  18. _qwak_proto/qwak/feature_store/features/feature_set_pb2.py +4 -4
  19. _qwak_proto/qwak/feature_store/features/feature_set_pb2.pyi +4 -0
  20. _qwak_proto/qwak/feature_store/features/feature_set_types_pb2.py +60 -58
  21. _qwak_proto/qwak/feature_store/features/feature_set_types_pb2.pyi +7 -2
  22. _qwak_proto/qwak/fitness_service/constructs_pb2.py +2 -2
  23. _qwak_proto/qwak/fitness_service/constructs_pb2.pyi +24 -0
  24. _qwak_proto/qwak/kube_deployment_captain/batch_job_pb2.py +40 -40
  25. _qwak_proto/qwak/kube_deployment_captain/batch_job_pb2.pyi +12 -2
  26. _qwak_proto/qwak/projects/projects_pb2.py +17 -15
  27. _qwak_proto/qwak/secret_service/secret_service_pb2.pyi +1 -1
  28. qwak/__init__.py +1 -1
  29. qwak/clients/model_management/client.py +0 -5
  30. qwak/clients/project/client.py +0 -7
  31. qwak/exceptions/__init__.py +1 -0
  32. qwak/exceptions/qwak_grpc_address_exception.py +9 -0
  33. qwak/feature_store/_common/packaging.py +11 -5
  34. qwak/inner/const.py +2 -8
  35. qwak/inner/di_configuration/__init__.py +1 -67
  36. qwak/inner/di_configuration/account.py +6 -66
  37. qwak/inner/di_configuration/dependency_wiring.py +98 -0
  38. qwak/inner/tool/auth.py +0 -86
  39. qwak/inner/tool/grpc/grpc_auth.py +0 -32
  40. qwak/inner/tool/grpc/grpc_tools.py +125 -11
  41. qwak/inner/tool/grpc/grpc_try_wrapping.py +3 -1
  42. qwak/llmops/generation/chat/openai/types/chat/chat_completion.py +24 -6
  43. qwak/llmops/generation/chat/openai/types/chat/chat_completion_chunk.py +44 -8
  44. qwak/llmops/generation/chat/openai/types/chat/chat_completion_message.py +6 -3
  45. qwak/qwak_client/client.py +2 -8
  46. qwak/vector_store/rest_helpers.py +4 -16
  47. qwak_core-0.5.4.dist-info/METADATA +48 -0
  48. {qwak_core-0.4.362.dist-info → qwak_core-0.5.4.dist-info}/RECORD +49 -82
  49. frogml_storage/__init__.py +0 -1
  50. frogml_storage/artifactory/__init__.py +0 -1
  51. frogml_storage/artifactory/_artifactory_api.py +0 -315
  52. frogml_storage/authentication/login/__init__.py +0 -1
  53. frogml_storage/authentication/login/_login_cli.py +0 -239
  54. frogml_storage/authentication/login/_login_command.py +0 -74
  55. frogml_storage/authentication/models/__init__.py +0 -3
  56. frogml_storage/authentication/models/_auth.py +0 -24
  57. frogml_storage/authentication/models/_auth_config.py +0 -70
  58. frogml_storage/authentication/models/_login.py +0 -22
  59. frogml_storage/authentication/utils/__init__.py +0 -17
  60. frogml_storage/authentication/utils/_authentication_utils.py +0 -281
  61. frogml_storage/authentication/utils/_login_checks_utils.py +0 -114
  62. frogml_storage/base_storage.py +0 -140
  63. frogml_storage/constants.py +0 -56
  64. frogml_storage/exceptions/checksum_verification_error.py +0 -3
  65. frogml_storage/exceptions/validation_error.py +0 -4
  66. frogml_storage/frog_ml.py +0 -668
  67. frogml_storage/http/__init__.py +0 -1
  68. frogml_storage/http/http_client.py +0 -83
  69. frogml_storage/logging/__init__.py +0 -1
  70. frogml_storage/logging/_log_config.py +0 -45
  71. frogml_storage/logging/log_utils.py +0 -21
  72. frogml_storage/models/__init__.py +0 -1
  73. frogml_storage/models/_download_context.py +0 -54
  74. frogml_storage/models/dataset_manifest.py +0 -13
  75. frogml_storage/models/entity_manifest.py +0 -93
  76. frogml_storage/models/frogml_dataset_version.py +0 -21
  77. frogml_storage/models/frogml_entity_type_info.py +0 -50
  78. frogml_storage/models/frogml_entity_version.py +0 -34
  79. frogml_storage/models/frogml_model_version.py +0 -21
  80. frogml_storage/models/model_manifest.py +0 -60
  81. frogml_storage/models/serialization_metadata.py +0 -15
  82. frogml_storage/utils/__init__.py +0 -12
  83. frogml_storage/utils/_environment.py +0 -21
  84. frogml_storage/utils/_input_checks_utility.py +0 -104
  85. frogml_storage/utils/_storage_utils.py +0 -15
  86. frogml_storage/utils/_url_utils.py +0 -27
  87. qwak_core-0.4.362.dist-info/METADATA +0 -414
  88. {qwak_core-0.4.362.dist-info → qwak_core-0.5.4.dist-info}/WHEEL +0 -0
@@ -0,0 +1,553 @@
1
+ """
2
+ @generated by mypy-protobuf. Do not edit manually!
3
+ isort:skip_file
4
+ """
5
+ import builtins
6
+ import collections.abc
7
+ import google.protobuf.descriptor
8
+ import google.protobuf.internal.containers
9
+ import google.protobuf.internal.enum_type_wrapper
10
+ import google.protobuf.message
11
+ import qwak.user_application.common.v0.resources_pb2
12
+ import sys
13
+ import typing
14
+
15
+ if sys.version_info >= (3, 10):
16
+ import typing as typing_extensions
17
+ else:
18
+ import typing_extensions
19
+
20
+ DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
21
+
22
+ class _PurchaseOption:
23
+ ValueType = typing.NewType("ValueType", builtins.int)
24
+ V: typing_extensions.TypeAlias = ValueType
25
+
26
+ class _PurchaseOptionEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_PurchaseOption.ValueType], builtins.type): # noqa: F821
27
+ DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
28
+ INVALID_PURCHASE_OPTION: _PurchaseOption.ValueType # 0
29
+ SPOT_PURCHASE_OPTION: _PurchaseOption.ValueType # 1
30
+ ONDEMAND_PURCHASE_OPTION: _PurchaseOption.ValueType # 2
31
+
32
+ class PurchaseOption(_PurchaseOption, metaclass=_PurchaseOptionEnumTypeWrapper): ...
33
+
34
+ INVALID_PURCHASE_OPTION: PurchaseOption.ValueType # 0
35
+ SPOT_PURCHASE_OPTION: PurchaseOption.ValueType # 1
36
+ ONDEMAND_PURCHASE_OPTION: PurchaseOption.ValueType # 2
37
+ global___PurchaseOption = PurchaseOption
38
+
39
+ class _CloudProvider:
40
+ ValueType = typing.NewType("ValueType", builtins.int)
41
+ V: typing_extensions.TypeAlias = ValueType
42
+
43
+ class _CloudProviderEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_CloudProvider.ValueType], builtins.type): # noqa: F821
44
+ DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
45
+ INVALID_CLOUD_PROVIDER: _CloudProvider.ValueType # 0
46
+ AWS: _CloudProvider.ValueType # 1
47
+ GCP: _CloudProvider.ValueType # 2
48
+ AZURE: _CloudProvider.ValueType # 3
49
+
50
+ class CloudProvider(_CloudProvider, metaclass=_CloudProviderEnumTypeWrapper): ...
51
+
52
+ INVALID_CLOUD_PROVIDER: CloudProvider.ValueType # 0
53
+ AWS: CloudProvider.ValueType # 1
54
+ GCP: CloudProvider.ValueType # 2
55
+ AZURE: CloudProvider.ValueType # 3
56
+ global___CloudProvider = CloudProvider
57
+
58
+ class _MemoryUnit:
59
+ ValueType = typing.NewType("ValueType", builtins.int)
60
+ V: typing_extensions.TypeAlias = ValueType
61
+
62
+ class _MemoryUnitEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_MemoryUnit.ValueType], builtins.type): # noqa: F821
63
+ DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
64
+ INVALID_MEMORY_UNIT: _MemoryUnit.ValueType # 0
65
+ MIB: _MemoryUnit.ValueType # 1
66
+ GIB: _MemoryUnit.ValueType # 2
67
+
68
+ class MemoryUnit(_MemoryUnit, metaclass=_MemoryUnitEnumTypeWrapper): ...
69
+
70
+ INVALID_MEMORY_UNIT: MemoryUnit.ValueType # 0
71
+ MIB: MemoryUnit.ValueType # 1
72
+ GIB: MemoryUnit.ValueType # 2
73
+ global___MemoryUnit = MemoryUnit
74
+
75
+ class _LogsVerbosityLevel:
76
+ ValueType = typing.NewType("ValueType", builtins.int)
77
+ V: typing_extensions.TypeAlias = ValueType
78
+
79
+ class _LogsVerbosityLevelEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_LogsVerbosityLevel.ValueType], builtins.type): # noqa: F821
80
+ DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
81
+ INVALID_LOGS_VERBOSITY: _LogsVerbosityLevel.ValueType # 0
82
+ DEBUG: _LogsVerbosityLevel.ValueType # 10
83
+ INFO: _LogsVerbosityLevel.ValueType # 20
84
+ CRITICAL: _LogsVerbosityLevel.ValueType # 50
85
+
86
+ class LogsVerbosityLevel(_LogsVerbosityLevel, metaclass=_LogsVerbosityLevelEnumTypeWrapper): ...
87
+
88
+ INVALID_LOGS_VERBOSITY: LogsVerbosityLevel.ValueType # 0
89
+ DEBUG: LogsVerbosityLevel.ValueType # 10
90
+ INFO: LogsVerbosityLevel.ValueType # 20
91
+ CRITICAL: LogsVerbosityLevel.ValueType # 50
92
+ global___LogsVerbosityLevel = LogsVerbosityLevel
93
+
94
+ class ModelBuildValues(google.protobuf.message.Message):
95
+ """Represents all values required for initiating a model build job.
96
+ Mirrors the Helm chart values used by Admiral to deploy model builds.
97
+ """
98
+
99
+ DESCRIPTOR: google.protobuf.descriptor.Descriptor
100
+
101
+ BUILD_INFO_FIELD_NUMBER: builtins.int
102
+ MODEL_METADATA_FIELD_NUMBER: builtins.int
103
+ MODEL_CODE_LOCATION_FIELD_NUMBER: builtins.int
104
+ BUILD_EXECUTION_SPEC_FIELD_NUMBER: builtins.int
105
+ SDK_SPEC_FIELD_NUMBER: builtins.int
106
+ JFROG_ECOSYSTEM_SPEC_FIELD_NUMBER: builtins.int
107
+ JOB_SPEC_FIELD_NUMBER: builtins.int
108
+ @property
109
+ def build_info(self) -> global___BuildInfo:
110
+ """Build identifiers, tags, parameters, and serialized configuration."""
111
+ @property
112
+ def model_metadata(self) -> global___ModelMetadata:
113
+ """Identifiers and ownership metadata."""
114
+ @property
115
+ def model_code_location(self) -> global___ModelCodeLocation:
116
+ """Location of model code"""
117
+ @property
118
+ def build_execution_spec(self) -> global___BuildExecutionSpec:
119
+ """Execution parameters for the build process."""
120
+ @property
121
+ def sdk_spec(self) -> global___SdkSpec:
122
+ """Environment and SDK-specific execution config."""
123
+ @property
124
+ def jfrog_ecosystem_spec(self) -> global___JfrogEcosystemSpec:
125
+ """Access settings for JFrog ecosystem."""
126
+ @property
127
+ def job_spec(self) -> global___JobSpec:
128
+ """Infrastructure and runtime settings for the build job."""
129
+ def __init__(
130
+ self,
131
+ *,
132
+ build_info: global___BuildInfo | None = ...,
133
+ model_metadata: global___ModelMetadata | None = ...,
134
+ model_code_location: global___ModelCodeLocation | None = ...,
135
+ build_execution_spec: global___BuildExecutionSpec | None = ...,
136
+ sdk_spec: global___SdkSpec | None = ...,
137
+ jfrog_ecosystem_spec: global___JfrogEcosystemSpec | None = ...,
138
+ job_spec: global___JobSpec | None = ...,
139
+ ) -> None: ...
140
+ def HasField(self, field_name: typing_extensions.Literal["build_execution_spec", b"build_execution_spec", "build_info", b"build_info", "jfrog_ecosystem_spec", b"jfrog_ecosystem_spec", "job_spec", b"job_spec", "model_code_location", b"model_code_location", "model_metadata", b"model_metadata", "sdk_spec", b"sdk_spec"]) -> builtins.bool: ...
141
+ def ClearField(self, field_name: typing_extensions.Literal["build_execution_spec", b"build_execution_spec", "build_info", b"build_info", "jfrog_ecosystem_spec", b"jfrog_ecosystem_spec", "job_spec", b"job_spec", "model_code_location", b"model_code_location", "model_metadata", b"model_metadata", "sdk_spec", b"sdk_spec"]) -> None: ...
142
+
143
+ global___ModelBuildValues = ModelBuildValues
144
+
145
+ class BuildInfo(google.protobuf.message.Message):
146
+ """Build identifiers, tags, parameters, and serialized configuration."""
147
+
148
+ DESCRIPTOR: google.protobuf.descriptor.Descriptor
149
+
150
+ BUILD_ID_FIELD_NUMBER: builtins.int
151
+ BRANCH_NAME_FIELD_NUMBER: builtins.int
152
+ BUILD_TAGS_FIELD_NUMBER: builtins.int
153
+ BUILD_CONFIG_FIELD_NUMBER: builtins.int
154
+ BUILD_PROPERTIES_V1_FIELD_NUMBER: builtins.int
155
+ build_id: builtins.str
156
+ branch_name: builtins.str
157
+ @property
158
+ def build_tags(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]: ...
159
+ build_config: builtins.str
160
+ """Serialized build configuration."""
161
+ @property
162
+ def build_properties_v1(self) -> global___BuildPropertiesV1: ...
163
+ def __init__(
164
+ self,
165
+ *,
166
+ build_id: builtins.str = ...,
167
+ branch_name: builtins.str = ...,
168
+ build_tags: collections.abc.Iterable[builtins.str] | None = ...,
169
+ build_config: builtins.str = ...,
170
+ build_properties_v1: global___BuildPropertiesV1 | None = ...,
171
+ ) -> None: ...
172
+ def HasField(self, field_name: typing_extensions.Literal["build_properties_v1", b"build_properties_v1"]) -> builtins.bool: ...
173
+ def ClearField(self, field_name: typing_extensions.Literal["branch_name", b"branch_name", "build_config", b"build_config", "build_id", b"build_id", "build_properties_v1", b"build_properties_v1", "build_tags", b"build_tags"]) -> None: ...
174
+
175
+ global___BuildInfo = BuildInfo
176
+
177
+ class BuildPropertiesV1(google.protobuf.message.Message):
178
+ """Build artifact urls set at upload time"""
179
+
180
+ DESCRIPTOR: google.protobuf.descriptor.Descriptor
181
+
182
+ BUILD_CONFIG_URL_FIELD_NUMBER: builtins.int
183
+ SDK_WHEEL_URL_FIELD_NUMBER: builtins.int
184
+ SDK_VERSION_URL_FIELD_NUMBER: builtins.int
185
+ build_config_url: builtins.str
186
+ sdk_wheel_url: builtins.str
187
+ sdk_version_url: builtins.str
188
+ def __init__(
189
+ self,
190
+ *,
191
+ build_config_url: builtins.str = ...,
192
+ sdk_wheel_url: builtins.str = ...,
193
+ sdk_version_url: builtins.str = ...,
194
+ ) -> None: ...
195
+ def ClearField(self, field_name: typing_extensions.Literal["build_config_url", b"build_config_url", "sdk_version_url", b"sdk_version_url", "sdk_wheel_url", b"sdk_wheel_url"]) -> None: ...
196
+
197
+ global___BuildPropertiesV1 = BuildPropertiesV1
198
+
199
+ class ModelMetadata(google.protobuf.message.Message):
200
+ """Metadata identifying the model and its ownership."""
201
+
202
+ DESCRIPTOR: google.protobuf.descriptor.Descriptor
203
+
204
+ MODEL_ID_FIELD_NUMBER: builtins.int
205
+ ACCOUNT_ID_FIELD_NUMBER: builtins.int
206
+ USER_ID_FIELD_NUMBER: builtins.int
207
+ ARTIFACTORY_DETAILS_FIELD_NUMBER: builtins.int
208
+ model_id: builtins.str
209
+ account_id: builtins.str
210
+ user_id: builtins.str
211
+ @property
212
+ def artifactory_details(self) -> global___ArtifactoryDetails: ...
213
+ def __init__(
214
+ self,
215
+ *,
216
+ model_id: builtins.str = ...,
217
+ account_id: builtins.str = ...,
218
+ user_id: builtins.str = ...,
219
+ artifactory_details: global___ArtifactoryDetails | None = ...,
220
+ ) -> None: ...
221
+ def HasField(self, field_name: typing_extensions.Literal["artifactory_details", b"artifactory_details"]) -> builtins.bool: ...
222
+ def ClearField(self, field_name: typing_extensions.Literal["account_id", b"account_id", "artifactory_details", b"artifactory_details", "model_id", b"model_id", "user_id", b"user_id"]) -> None: ...
223
+
224
+ global___ModelMetadata = ModelMetadata
225
+
226
+ class ArtifactoryDetails(google.protobuf.message.Message):
227
+ DESCRIPTOR: google.protobuf.descriptor.Descriptor
228
+
229
+ PROJECT_KEY_FIELD_NUMBER: builtins.int
230
+ project_key: builtins.str
231
+ def __init__(
232
+ self,
233
+ *,
234
+ project_key: builtins.str = ...,
235
+ ) -> None: ...
236
+ def ClearField(self, field_name: typing_extensions.Literal["project_key", b"project_key"]) -> None: ...
237
+
238
+ global___ArtifactoryDetails = ArtifactoryDetails
239
+
240
+ class ModelCodeLocation(google.protobuf.message.Message):
241
+ """References to the model's storage location and source control info."""
242
+
243
+ DESCRIPTOR: google.protobuf.descriptor.Descriptor
244
+
245
+ MAIN_MODEL_DIRECTORY_FIELD_NUMBER: builtins.int
246
+ LOCAL_URI_FIELD_NUMBER: builtins.int
247
+ GIT_URI_FIELD_NUMBER: builtins.int
248
+ main_model_directory: builtins.str
249
+ @property
250
+ def local_uri(self) -> global___LocalUri: ...
251
+ @property
252
+ def git_uri(self) -> global___GitUri: ...
253
+ def __init__(
254
+ self,
255
+ *,
256
+ main_model_directory: builtins.str = ...,
257
+ local_uri: global___LocalUri | None = ...,
258
+ git_uri: global___GitUri | None = ...,
259
+ ) -> None: ...
260
+ def HasField(self, field_name: typing_extensions.Literal["git_uri", b"git_uri", "local_uri", b"local_uri", "uri", b"uri"]) -> builtins.bool: ...
261
+ def ClearField(self, field_name: typing_extensions.Literal["git_uri", b"git_uri", "local_uri", b"local_uri", "main_model_directory", b"main_model_directory", "uri", b"uri"]) -> None: ...
262
+ def WhichOneof(self, oneof_group: typing_extensions.Literal["uri", b"uri"]) -> typing_extensions.Literal["local_uri", "git_uri"] | None: ...
263
+
264
+ global___ModelCodeLocation = ModelCodeLocation
265
+
266
+ class LocalUri(google.protobuf.message.Message):
267
+ DESCRIPTOR: google.protobuf.descriptor.Descriptor
268
+
269
+ PATH_FIELD_NUMBER: builtins.int
270
+ path: builtins.str
271
+ def __init__(
272
+ self,
273
+ *,
274
+ path: builtins.str = ...,
275
+ ) -> None: ...
276
+ def ClearField(self, field_name: typing_extensions.Literal["path", b"path"]) -> None: ...
277
+
278
+ global___LocalUri = LocalUri
279
+
280
+ class GitUri(google.protobuf.message.Message):
281
+ DESCRIPTOR: google.protobuf.descriptor.Descriptor
282
+
283
+ URL_FIELD_NUMBER: builtins.int
284
+ GIT_DETAILS_FIELD_NUMBER: builtins.int
285
+ url: builtins.str
286
+ @property
287
+ def git_details(self) -> global___GitDetails: ...
288
+ def __init__(
289
+ self,
290
+ *,
291
+ url: builtins.str = ...,
292
+ git_details: global___GitDetails | None = ...,
293
+ ) -> None: ...
294
+ def HasField(self, field_name: typing_extensions.Literal["git_details", b"git_details"]) -> builtins.bool: ...
295
+ def ClearField(self, field_name: typing_extensions.Literal["git_details", b"git_details", "url", b"url"]) -> None: ...
296
+
297
+ global___GitUri = GitUri
298
+
299
+ class GitDetails(google.protobuf.message.Message):
300
+ """Git source control information used for the build."""
301
+
302
+ DESCRIPTOR: google.protobuf.descriptor.Descriptor
303
+
304
+ BRANCH_FIELD_NUMBER: builtins.int
305
+ COMMIT_ID_FIELD_NUMBER: builtins.int
306
+ CREDENTIALS_FIELD_NUMBER: builtins.int
307
+ CREDENTIALS_SECRET_NAME_FIELD_NUMBER: builtins.int
308
+ branch: builtins.str
309
+ commit_id: builtins.str
310
+ credentials: builtins.str
311
+ credentials_secret_name: builtins.str
312
+ def __init__(
313
+ self,
314
+ *,
315
+ branch: builtins.str = ...,
316
+ commit_id: builtins.str = ...,
317
+ credentials: builtins.str = ...,
318
+ credentials_secret_name: builtins.str = ...,
319
+ ) -> None: ...
320
+ def HasField(self, field_name: typing_extensions.Literal["credentials", b"credentials", "credentials_secret_name", b"credentials_secret_name", "git_credentials", b"git_credentials"]) -> builtins.bool: ...
321
+ def ClearField(self, field_name: typing_extensions.Literal["branch", b"branch", "commit_id", b"commit_id", "credentials", b"credentials", "credentials_secret_name", b"credentials_secret_name", "git_credentials", b"git_credentials"]) -> None: ...
322
+ def WhichOneof(self, oneof_group: typing_extensions.Literal["git_credentials", b"git_credentials"]) -> typing_extensions.Literal["credentials", "credentials_secret_name"] | None: ...
323
+
324
+ global___GitDetails = GitDetails
325
+
326
+ class BuildExecutionSpec(google.protobuf.message.Message):
327
+ """Specifies how the build should be executed."""
328
+
329
+ DESCRIPTOR: google.protobuf.descriptor.Descriptor
330
+
331
+ BUILD_IMAGE_DESTINATION_FIELD_NUMBER: builtins.int
332
+ SHOULD_BUILD_WITHOUT_CACHE_FIELD_NUMBER: builtins.int
333
+ LOGS_VERBOSITY_LEVEL_FIELD_NUMBER: builtins.int
334
+ BASE_IMAGE_FIELD_NUMBER: builtins.int
335
+ CUSTOM_IAM_ROLE_ARN_FIELD_NUMBER: builtins.int
336
+ ENV_VARS_FIELD_NUMBER: builtins.int
337
+ REMOTE_BUILDER_AUTH_SECRET_NAME_FIELD_NUMBER: builtins.int
338
+ CONTAINER_ENGINE_FIELD_NUMBER: builtins.int
339
+ build_image_destination: builtins.str
340
+ """Destination image to be built and pushed."""
341
+ should_build_without_cache: builtins.bool
342
+ """If true, disables build caching."""
343
+ logs_verbosity_level: global___LogsVerbosityLevel.ValueType
344
+ """Verbosity level for build logs."""
345
+ base_image: builtins.str
346
+ """Base Docker image used in the build."""
347
+ custom_iam_role_arn: builtins.str
348
+ """Custom IAM role used during the build."""
349
+ @property
350
+ def env_vars(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]:
351
+ """Env var parameters used in the build process"""
352
+ remote_builder_auth_secret_name: builtins.str
353
+ """Secret name for the remote builder, used for remote build registry authentication"""
354
+ @property
355
+ def container_engine(self) -> global___ContainerEngine:
356
+ """Container engine to use for the build (e.g., Docker, Podman)"""
357
+ def __init__(
358
+ self,
359
+ *,
360
+ build_image_destination: builtins.str = ...,
361
+ should_build_without_cache: builtins.bool = ...,
362
+ logs_verbosity_level: global___LogsVerbosityLevel.ValueType = ...,
363
+ base_image: builtins.str = ...,
364
+ custom_iam_role_arn: builtins.str = ...,
365
+ env_vars: collections.abc.Iterable[builtins.str] | None = ...,
366
+ remote_builder_auth_secret_name: builtins.str = ...,
367
+ container_engine: global___ContainerEngine | None = ...,
368
+ ) -> None: ...
369
+ def HasField(self, field_name: typing_extensions.Literal["container_engine", b"container_engine"]) -> builtins.bool: ...
370
+ def ClearField(self, field_name: typing_extensions.Literal["base_image", b"base_image", "build_image_destination", b"build_image_destination", "container_engine", b"container_engine", "custom_iam_role_arn", b"custom_iam_role_arn", "env_vars", b"env_vars", "logs_verbosity_level", b"logs_verbosity_level", "remote_builder_auth_secret_name", b"remote_builder_auth_secret_name", "should_build_without_cache", b"should_build_without_cache"]) -> None: ...
371
+
372
+ global___BuildExecutionSpec = BuildExecutionSpec
373
+
374
+ class SdkSpec(google.protobuf.message.Message):
375
+ """Specifies the SDK environment and configuration used to execute the build."""
376
+
377
+ DESCRIPTOR: google.protobuf.descriptor.Descriptor
378
+
379
+ SDK_VERSION_FIELD_NUMBER: builtins.int
380
+ SDK_EXTRA_INDEX_URL_FIELD_NUMBER: builtins.int
381
+ SDK_RUNTIME_VERSION_OVERRIDE_FIELD_NUMBER: builtins.int
382
+ sdk_version: builtins.str
383
+ sdk_extra_index_url: builtins.str
384
+ sdk_runtime_version_override: builtins.str
385
+ def __init__(
386
+ self,
387
+ *,
388
+ sdk_version: builtins.str = ...,
389
+ sdk_extra_index_url: builtins.str = ...,
390
+ sdk_runtime_version_override: builtins.str = ...,
391
+ ) -> None: ...
392
+ def ClearField(self, field_name: typing_extensions.Literal["sdk_extra_index_url", b"sdk_extra_index_url", "sdk_runtime_version_override", b"sdk_runtime_version_override", "sdk_version", b"sdk_version"]) -> None: ...
393
+
394
+ global___SdkSpec = SdkSpec
395
+
396
+ class JfrogEcosystemSpec(google.protobuf.message.Message):
397
+ """JFrog ecosystem configuration, used for authentication and configuration"""
398
+
399
+ DESCRIPTOR: google.protobuf.descriptor.Descriptor
400
+
401
+ BASEURL_FIELD_NUMBER: builtins.int
402
+ ACCESSTOKENSECRETNAME_FIELD_NUMBER: builtins.int
403
+ ARTIFACTORYSETTINGSSECRETNAME_FIELD_NUMBER: builtins.int
404
+ baseUrl: builtins.str
405
+ accessTokenSecretName: builtins.str
406
+ artifactorySettingsSecretName: builtins.str
407
+ def __init__(
408
+ self,
409
+ *,
410
+ baseUrl: builtins.str = ...,
411
+ accessTokenSecretName: builtins.str = ...,
412
+ artifactorySettingsSecretName: builtins.str = ...,
413
+ ) -> None: ...
414
+ def ClearField(self, field_name: typing_extensions.Literal["accessTokenSecretName", b"accessTokenSecretName", "artifactorySettingsSecretName", b"artifactorySettingsSecretName", "baseUrl", b"baseUrl"]) -> None: ...
415
+
416
+ global___JfrogEcosystemSpec = JfrogEcosystemSpec
417
+
418
+ class JobSpec(google.protobuf.message.Message):
419
+ """Describes the runtime and infrastructure settings for a build job."""
420
+
421
+ DESCRIPTOR: google.protobuf.descriptor.Descriptor
422
+
423
+ class BuildExecutorEnvVarsEntry(google.protobuf.message.Message):
424
+ DESCRIPTOR: google.protobuf.descriptor.Descriptor
425
+
426
+ KEY_FIELD_NUMBER: builtins.int
427
+ VALUE_FIELD_NUMBER: builtins.int
428
+ key: builtins.str
429
+ value: builtins.str
430
+ def __init__(
431
+ self,
432
+ *,
433
+ key: builtins.str = ...,
434
+ value: builtins.str = ...,
435
+ ) -> None: ...
436
+ def ClearField(self, field_name: typing_extensions.Literal["key", b"key", "value", b"value"]) -> None: ...
437
+
438
+ NAMESPACE_FIELD_NUMBER: builtins.int
439
+ PROVISION_INSTANCE_TIMEOUT_FIELD_NUMBER: builtins.int
440
+ PURCHASE_OPTION_FIELD_NUMBER: builtins.int
441
+ IS_NODE_OPTIMIZATION_ENABLED_FIELD_NUMBER: builtins.int
442
+ SHOULD_USE_BUILD_ISOLATION_FIELD_NUMBER: builtins.int
443
+ IS_MULTI_TENANT_ENVIRONMENT_FIELD_NUMBER: builtins.int
444
+ IS_SELF_HOSTED_JPD_FIELD_NUMBER: builtins.int
445
+ BUILD_EXECUTOR_IMAGE_URL_FIELD_NUMBER: builtins.int
446
+ CLOUD_PROVIDER_FIELD_NUMBER: builtins.int
447
+ JOB_SIZE_FIELD_NUMBER: builtins.int
448
+ BUILD_EXECUTOR_ENV_VARS_FIELD_NUMBER: builtins.int
449
+ namespace: builtins.str
450
+ """Kubernetes namespace in which to run the build job."""
451
+ provision_instance_timeout: builtins.int
452
+ purchase_option: global___PurchaseOption.ValueType
453
+ is_node_optimization_enabled: builtins.bool
454
+ should_use_build_isolation: builtins.bool
455
+ is_multi_tenant_environment: builtins.bool
456
+ is_self_hosted_jpd: builtins.bool
457
+ build_executor_image_url: builtins.str
458
+ """Image used for the build executor."""
459
+ cloud_provider: global___CloudProvider.ValueType
460
+ @property
461
+ def job_size(self) -> global___JobSize: ...
462
+ @property
463
+ def build_executor_env_vars(self) -> google.protobuf.internal.containers.ScalarMap[builtins.str, builtins.str]:
464
+ """Environment variables to set on the build executor container."""
465
+ def __init__(
466
+ self,
467
+ *,
468
+ namespace: builtins.str = ...,
469
+ provision_instance_timeout: builtins.int = ...,
470
+ purchase_option: global___PurchaseOption.ValueType = ...,
471
+ is_node_optimization_enabled: builtins.bool = ...,
472
+ should_use_build_isolation: builtins.bool = ...,
473
+ is_multi_tenant_environment: builtins.bool = ...,
474
+ is_self_hosted_jpd: builtins.bool = ...,
475
+ build_executor_image_url: builtins.str = ...,
476
+ cloud_provider: global___CloudProvider.ValueType = ...,
477
+ job_size: global___JobSize | None = ...,
478
+ build_executor_env_vars: collections.abc.Mapping[builtins.str, builtins.str] | None = ...,
479
+ ) -> None: ...
480
+ def HasField(self, field_name: typing_extensions.Literal["job_size", b"job_size"]) -> builtins.bool: ...
481
+ def ClearField(self, field_name: typing_extensions.Literal["build_executor_env_vars", b"build_executor_env_vars", "build_executor_image_url", b"build_executor_image_url", "cloud_provider", b"cloud_provider", "is_multi_tenant_environment", b"is_multi_tenant_environment", "is_node_optimization_enabled", b"is_node_optimization_enabled", "is_self_hosted_jpd", b"is_self_hosted_jpd", "job_size", b"job_size", "namespace", b"namespace", "provision_instance_timeout", b"provision_instance_timeout", "purchase_option", b"purchase_option", "should_use_build_isolation", b"should_use_build_isolation"]) -> None: ...
482
+
483
+ global___JobSpec = JobSpec
484
+
485
+ class JobSize(google.protobuf.message.Message):
486
+ """Defines the CPU, memory, and GPU resources allocated to a build job."""
487
+
488
+ DESCRIPTOR: google.protobuf.descriptor.Descriptor
489
+
490
+ CPU_FIELD_NUMBER: builtins.int
491
+ MEMORY_AMOUNT_FIELD_NUMBER: builtins.int
492
+ MEMORY_UNITS_FIELD_NUMBER: builtins.int
493
+ GPU_RESOURCES_FIELD_NUMBER: builtins.int
494
+ CLIENT_POD_COMPUTE_RESOURCES_FIELD_NUMBER: builtins.int
495
+ cpu: builtins.float
496
+ memory_amount: builtins.int
497
+ memory_units: global___MemoryUnit.ValueType
498
+ @property
499
+ def gpu_resources(self) -> qwak.user_application.common.v0.resources_pb2.GpuResources: ...
500
+ @property
501
+ def client_pod_compute_resources(self) -> qwak.user_application.common.v0.resources_pb2.ClientPodComputeResources: ...
502
+ def __init__(
503
+ self,
504
+ *,
505
+ cpu: builtins.float = ...,
506
+ memory_amount: builtins.int = ...,
507
+ memory_units: global___MemoryUnit.ValueType = ...,
508
+ gpu_resources: qwak.user_application.common.v0.resources_pb2.GpuResources | None = ...,
509
+ client_pod_compute_resources: qwak.user_application.common.v0.resources_pb2.ClientPodComputeResources | None = ...,
510
+ ) -> None: ...
511
+ def HasField(self, field_name: typing_extensions.Literal["client_pod_compute_resources", b"client_pod_compute_resources", "gpu_resources", b"gpu_resources"]) -> builtins.bool: ...
512
+ def ClearField(self, field_name: typing_extensions.Literal["client_pod_compute_resources", b"client_pod_compute_resources", "cpu", b"cpu", "gpu_resources", b"gpu_resources", "memory_amount", b"memory_amount", "memory_units", b"memory_units"]) -> None: ...
513
+
514
+ global___JobSize = JobSize
515
+
516
+ class ContainerEngine(google.protobuf.message.Message):
517
+ DESCRIPTOR: google.protobuf.descriptor.Descriptor
518
+
519
+ DOCKER_FIELD_NUMBER: builtins.int
520
+ PODMAN_FIELD_NUMBER: builtins.int
521
+ @property
522
+ def docker(self) -> global___DockerContainerEngine: ...
523
+ @property
524
+ def podman(self) -> global___PodmanContainerEngine: ...
525
+ def __init__(
526
+ self,
527
+ *,
528
+ docker: global___DockerContainerEngine | None = ...,
529
+ podman: global___PodmanContainerEngine | None = ...,
530
+ ) -> None: ...
531
+ def HasField(self, field_name: typing_extensions.Literal["docker", b"docker", "podman", b"podman", "type", b"type"]) -> builtins.bool: ...
532
+ def ClearField(self, field_name: typing_extensions.Literal["docker", b"docker", "podman", b"podman", "type", b"type"]) -> None: ...
533
+ def WhichOneof(self, oneof_group: typing_extensions.Literal["type", b"type"]) -> typing_extensions.Literal["docker", "podman"] | None: ...
534
+
535
+ global___ContainerEngine = ContainerEngine
536
+
537
+ class DockerContainerEngine(google.protobuf.message.Message):
538
+ DESCRIPTOR: google.protobuf.descriptor.Descriptor
539
+
540
+ def __init__(
541
+ self,
542
+ ) -> None: ...
543
+
544
+ global___DockerContainerEngine = DockerContainerEngine
545
+
546
+ class PodmanContainerEngine(google.protobuf.message.Message):
547
+ DESCRIPTOR: google.protobuf.descriptor.Descriptor
548
+
549
+ def __init__(
550
+ self,
551
+ ) -> None: ...
552
+
553
+ global___PodmanContainerEngine = PodmanContainerEngine
@@ -0,0 +1,4 @@
1
+ # Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
2
+ """Client and server classes corresponding to protobuf-defined services."""
3
+ import grpc
4
+
@@ -11,9 +11,12 @@ from google.protobuf.internal import builder as _builder
11
11
  _sym_db = _symbol_database.Default()
12
12
 
13
13
 
14
+ from _qwak_proto.qwak.feature_store.features import feature_set_types_pb2 as qwak_dot_feature__store_dot_features_dot_feature__set__types__pb2
15
+ from _qwak_proto.qwak.feature_store.features import execution_pb2 as qwak_dot_feature__store_dot_features_dot_execution__pb2
16
+ from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
14
17
 
15
18
 
16
- DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n-qwak/execution/v1/streaming_aggregation.proto\x12\x1fqwak.feature.store.execution.v1\"@\n%RowLevelStreamingAggregationIngestion\x12\x17\n\x0f\x66\x65\x61tureset_name\x18\x01 \x01(\t\"\xf8\x01\n\'CompactionStreamingAggregationIngestion\x12\x17\n\x0f\x66\x65\x61tureset_name\x18\x01 \x01(\t\x12T\n\x14scheduled_compaction\x18\x02 \x01(\x0b\x32\x34.qwak.feature.store.execution.v1.ScheduledCompactionH\x00\x12N\n\x11manual_compaction\x18\x03 \x01(\x0b\x32\x31.qwak.feature.store.execution.v1.ManualCompactionH\x00\x42\x0e\n\x0ctrigger_type\"\x15\n\x13ScheduledCompaction\"\x12\n\x10ManualCompaction\"@\n%StreamingAggregationBackfillIngestion\x12\x17\n\x0f\x66\x65\x61tureset_name\x18\x01 \x01(\tB)\n%com.qwak.ai.features.execution.api.v1P\x01\x62\x06proto3')
19
+ DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n-qwak/execution/v1/streaming_aggregation.proto\x12\x1fqwak.feature.store.execution.v1\x1a\x33qwak/feature_store/features/feature_set_types.proto\x1a+qwak/feature_store/features/execution.proto\x1a\x1fgoogle/protobuf/timestamp.proto\"@\n%RowLevelStreamingAggregationIngestion\x12\x17\n\x0f\x66\x65\x61tureset_name\x18\x01 \x01(\t\"\xf8\x01\n\'CompactionStreamingAggregationIngestion\x12\x17\n\x0f\x66\x65\x61tureset_name\x18\x01 \x01(\t\x12T\n\x14scheduled_compaction\x18\x02 \x01(\x0b\x32\x34.qwak.feature.store.execution.v1.ScheduledCompactionH\x00\x12N\n\x11manual_compaction\x18\x03 \x01(\x0b\x32\x31.qwak.feature.store.execution.v1.ManualCompactionH\x00\x42\x0e\n\x0ctrigger_type\"\x15\n\x13ScheduledCompaction\"\x12\n\x10ManualCompaction\"\x8b\x03\n%StreamingAggregationBackfillIngestion\x12\x17\n\x0f\x66\x65\x61tureset_name\x18\x01 \x01(\t\x12L\n\x0e\x65xecution_spec\x18\x02 \x01(\x0b\x32\x34.qwak.feature.store.features.execution.ExecutionSpec\x12\x33\n\x0fstart_timestamp\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x31\n\rend_timestamp\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x43\n\x0etransformation\x18\x05 \x01(\x0b\x32+.qwak.feature.store.features.Transformation\x12N\n\x11\x64\x61ta_source_specs\x18\x06 \x03(\x0b\x32\x33.qwak.feature.store.execution.v1.BackfillDataSource\"n\n\x12\x42\x61\x63kfillDataSource\x12\x18\n\x10\x64\x61ta_source_name\x18\x01 \x01(\t\x12>\n\ntime_range\x18\x02 \x01(\x0b\x32*.qwak.feature.store.execution.v1.TimeRange\"\x99\x01\n\tTimeRange\x12\x35\n\x0fstart_timestamp\x18\x01 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00\x12\x33\n\rend_timestamp\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x01\x42\x10\n\x0estart_optionalB\x0e\n\x0c\x65nd_optionalB)\n%com.qwak.ai.features.execution.api.v1P\x01\x62\x06proto3')
17
20
 
18
21
  _globals = globals()
19
22
  _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
@@ -22,14 +25,18 @@ if _descriptor._USE_C_DESCRIPTORS == False:
22
25
 
23
26
  DESCRIPTOR._options = None
24
27
  DESCRIPTOR._serialized_options = b'\n%com.qwak.ai.features.execution.api.v1P\001'
25
- _globals['_ROWLEVELSTREAMINGAGGREGATIONINGESTION']._serialized_start=82
26
- _globals['_ROWLEVELSTREAMINGAGGREGATIONINGESTION']._serialized_end=146
27
- _globals['_COMPACTIONSTREAMINGAGGREGATIONINGESTION']._serialized_start=149
28
- _globals['_COMPACTIONSTREAMINGAGGREGATIONINGESTION']._serialized_end=397
29
- _globals['_SCHEDULEDCOMPACTION']._serialized_start=399
30
- _globals['_SCHEDULEDCOMPACTION']._serialized_end=420
31
- _globals['_MANUALCOMPACTION']._serialized_start=422
32
- _globals['_MANUALCOMPACTION']._serialized_end=440
33
- _globals['_STREAMINGAGGREGATIONBACKFILLINGESTION']._serialized_start=442
34
- _globals['_STREAMINGAGGREGATIONBACKFILLINGESTION']._serialized_end=506
28
+ _globals['_ROWLEVELSTREAMINGAGGREGATIONINGESTION']._serialized_start=213
29
+ _globals['_ROWLEVELSTREAMINGAGGREGATIONINGESTION']._serialized_end=277
30
+ _globals['_COMPACTIONSTREAMINGAGGREGATIONINGESTION']._serialized_start=280
31
+ _globals['_COMPACTIONSTREAMINGAGGREGATIONINGESTION']._serialized_end=528
32
+ _globals['_SCHEDULEDCOMPACTION']._serialized_start=530
33
+ _globals['_SCHEDULEDCOMPACTION']._serialized_end=551
34
+ _globals['_MANUALCOMPACTION']._serialized_start=553
35
+ _globals['_MANUALCOMPACTION']._serialized_end=571
36
+ _globals['_STREAMINGAGGREGATIONBACKFILLINGESTION']._serialized_start=574
37
+ _globals['_STREAMINGAGGREGATIONBACKFILLINGESTION']._serialized_end=969
38
+ _globals['_BACKFILLDATASOURCE']._serialized_start=971
39
+ _globals['_BACKFILLDATASOURCE']._serialized_end=1081
40
+ _globals['_TIMERANGE']._serialized_start=1084
41
+ _globals['_TIMERANGE']._serialized_end=1237
35
42
  # @@protoc_insertion_point(module_scope)