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
|
@@ -46,8 +46,8 @@ _qwak_proto/qwak/administration/runtime_configuration/v0/creds/secret_pb2_grpc.p
|
|
|
46
46
|
_qwak_proto/qwak/administration/runtime_configuration/v0/data_catalog_config_pb2.py,sha256=qhcszHQcNgbDHdsC2U0E0f0s6Z6K9ZCSQNKkaeaIMbU,2888
|
|
47
47
|
_qwak_proto/qwak/administration/runtime_configuration/v0/data_catalog_config_pb2.pyi,sha256=itqSZkHGzVDfHMNTK6rlgXkU19ZncRNYYeO8-lHIpXU,3399
|
|
48
48
|
_qwak_proto/qwak/administration/runtime_configuration/v0/data_catalog_config_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
|
|
49
|
-
_qwak_proto/qwak/administration/runtime_configuration/v0/external/databricks/auth_pb2.py,sha256=
|
|
50
|
-
_qwak_proto/qwak/administration/runtime_configuration/v0/external/databricks/auth_pb2.pyi,sha256=
|
|
49
|
+
_qwak_proto/qwak/administration/runtime_configuration/v0/external/databricks/auth_pb2.py,sha256=Xsr3QoqcR-5ktyX6ks595efBZ8BXxgaRXXuZ6MEmHk8,3537
|
|
50
|
+
_qwak_proto/qwak/administration/runtime_configuration/v0/external/databricks/auth_pb2.pyi,sha256=IYgeqMFvIzl4rJrrJh0SSlGsQcqCGoQ5IXl_IwBy33g,5125
|
|
51
51
|
_qwak_proto/qwak/administration/runtime_configuration/v0/external/databricks/auth_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
|
|
52
52
|
_qwak_proto/qwak/administration/runtime_configuration/v0/external/elasticsearch_config_pb2.py,sha256=KbF15euA3Jrc4LfCVSjWAEM1dIcV5KCqSuo_w7RIo44,2308
|
|
53
53
|
_qwak_proto/qwak/administration/runtime_configuration/v0/external/elasticsearch_config_pb2.pyi,sha256=kn55-29574hUvrodpivp5TLXyOaJ8VuGzyUIt73syAc,1525
|
|
@@ -70,8 +70,8 @@ _qwak_proto/qwak/administration/runtime_configuration/v0/feature_store_config_pb
|
|
|
70
70
|
_qwak_proto/qwak/administration/runtime_configuration/v0/hosting/aws/auth_pb2.py,sha256=7cnXQm-0nz1N9ABllcUYWAP877989cA9V64hdgFebp4,4634
|
|
71
71
|
_qwak_proto/qwak/administration/runtime_configuration/v0/hosting/aws/auth_pb2.pyi,sha256=ShrRAP2q9cAiK-DGgCrOY-yx8HNEc3faRAOW8sGrTso,7529
|
|
72
72
|
_qwak_proto/qwak/administration/runtime_configuration/v0/hosting/aws/auth_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
|
|
73
|
-
_qwak_proto/qwak/administration/runtime_configuration/v0/hosting/azure/auth_pb2.py,sha256=
|
|
74
|
-
_qwak_proto/qwak/administration/runtime_configuration/v0/hosting/azure/auth_pb2.pyi,sha256=
|
|
73
|
+
_qwak_proto/qwak/administration/runtime_configuration/v0/hosting/azure/auth_pb2.py,sha256=I5fgDuhWQJngaEzPNcqJ90re1eAW9NY2tvBh2If_kcA,4219
|
|
74
|
+
_qwak_proto/qwak/administration/runtime_configuration/v0/hosting/azure/auth_pb2.pyi,sha256=rkgJJYJc_QbMllvphZOwrm_Jla-KPpZ2Rt6Lb1bqWcQ,6127
|
|
75
75
|
_qwak_proto/qwak/administration/runtime_configuration/v0/hosting/azure/auth_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
|
|
76
76
|
_qwak_proto/qwak/administration/runtime_configuration/v0/hosting_config_pb2.py,sha256=PdbXyznN2L1xFXs1P9CiHXIhI6Iy3xve46iiXM-PnQM,4776
|
|
77
77
|
_qwak_proto/qwak/administration/runtime_configuration/v0/hosting_config_pb2.pyi,sha256=lFhK2Vh0aZXfCBlzL-EKVSzEKR5yQ9MVHI_NnKXosek,7326
|
|
@@ -182,7 +182,7 @@ _qwak_proto/qwak/batch_job/v1/batch_job_resources_pb2.py,sha256=EvcGo4P-K376Uh-Y
|
|
|
182
182
|
_qwak_proto/qwak/batch_job/v1/batch_job_resources_pb2.pyi,sha256=e4WFKs7LbWS2M5XLyLrMQROhfSIpmh9Nj-qcSbEv9E0,1977
|
|
183
183
|
_qwak_proto/qwak/batch_job/v1/batch_job_resources_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
|
|
184
184
|
_qwak_proto/qwak/batch_job/v1/batch_job_service_pb2.py,sha256=-uAQzkZrHjW1fwMSvRdrScHjR489XkQqEQicCvFe0As,25710
|
|
185
|
-
_qwak_proto/qwak/batch_job/v1/batch_job_service_pb2.pyi,sha256=
|
|
185
|
+
_qwak_proto/qwak/batch_job/v1/batch_job_service_pb2.pyi,sha256=B3qHYud6RZA_j8Rjhv_tl-Zf_83rBGFX8jULLCk_S2U,62659
|
|
186
186
|
_qwak_proto/qwak/batch_job/v1/batch_job_service_pb2_grpc.py,sha256=IQVdGiiHA7l4ovHyOM5ODIq7Riy8u7zJWFvwirU9B6w,33921
|
|
187
187
|
_qwak_proto/qwak/build/v1/build_api_pb2.py,sha256=HnzPqfCpNEhTO8CmEpdzf8Al9E1p1PU8jD9G5Ke2atw,15506
|
|
188
188
|
_qwak_proto/qwak/build/v1/build_api_pb2.pyi,sha256=zgi8Gy3_Ollk48HxcbQYQfEaKwqiNagfALwcH4GuyOM,26751
|
|
@@ -196,12 +196,15 @@ _qwak_proto/qwak/build_settings/build_settings_api_pb2_grpc.py,sha256=I8qxd_DuYr
|
|
|
196
196
|
_qwak_proto/qwak/build_settings/build_settings_pb2.py,sha256=KycMr1SY_3UPDhXm2IONYur2gT0egCEcAeO_DGHBqyg,3258
|
|
197
197
|
_qwak_proto/qwak/build_settings/build_settings_pb2.pyi,sha256=MMYamilVLTgHn-WmsIqRCI8K9QJSehtxAclPeP75CWA,7049
|
|
198
198
|
_qwak_proto/qwak/build_settings/build_settings_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
|
|
199
|
-
_qwak_proto/qwak/builds/build_pb2.py,sha256=
|
|
200
|
-
_qwak_proto/qwak/builds/build_pb2.pyi,sha256=
|
|
199
|
+
_qwak_proto/qwak/builds/build_pb2.py,sha256=VucN5JInr7_b-rWxOM17dtFcaAXwpFRE1YpcBOpyamQ,9410
|
|
200
|
+
_qwak_proto/qwak/builds/build_pb2.pyi,sha256=yuwHuPN-i6iQ470SwNyhl4WP-z8mgFgk96X6787s3Ds,27079
|
|
201
201
|
_qwak_proto/qwak/builds/build_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
|
|
202
202
|
_qwak_proto/qwak/builds/build_url_pb2.py,sha256=G7EUD7nYmhEVsStEYPnmqiESsV6sKaWjmcuyvahI0nk,4611
|
|
203
203
|
_qwak_proto/qwak/builds/build_url_pb2.pyi,sha256=qaYyuLRHN2hOqYXlN9mKJo9gpggUnPmTVZIu-zNA0dI,11372
|
|
204
204
|
_qwak_proto/qwak/builds/build_url_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
|
|
205
|
+
_qwak_proto/qwak/builds/build_values_pb2.py,sha256=L-bkUAQIZFC1NcJcpsnNDUlBO8Br2N52yest-pZ3ZNA,8860
|
|
206
|
+
_qwak_proto/qwak/builds/build_values_pb2.pyi,sha256=T-WvC45AaYIuQbYbs2QysVVHVaOqsaG1HyTz8NxQaek,24158
|
|
207
|
+
_qwak_proto/qwak/builds/build_values_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
|
|
205
208
|
_qwak_proto/qwak/builds/builds_orchestrator_service_pb2.py,sha256=nvh6ANn8rNVjWByiB0xu-8Kt45FKh-C8YJaw5_I_N9M,14575
|
|
206
209
|
_qwak_proto/qwak/builds/builds_orchestrator_service_pb2.pyi,sha256=Vl8UmQpIEslUhln3_xizdaHAWfBMBuaL8-aNAbdNBQs,24711
|
|
207
210
|
_qwak_proto/qwak/builds/builds_orchestrator_service_pb2_grpc.py,sha256=nNTst6BicQAkC-z2Iz2OXI6_3oXCVL7S0zrwEWgKjuE,27004
|
|
@@ -476,7 +479,7 @@ _qwak_proto/qwak/kube_deployment_captain/alerting_pb2.py,sha256=l9xheVe-_YJH56ip
|
|
|
476
479
|
_qwak_proto/qwak/kube_deployment_captain/alerting_pb2.pyi,sha256=AGXtot0uXW-mGd2xuixCCJB6eTt0UuF3b2oAhb4IsNc,3759
|
|
477
480
|
_qwak_proto/qwak/kube_deployment_captain/alerting_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
|
|
478
481
|
_qwak_proto/qwak/kube_deployment_captain/batch_job_pb2.py,sha256=A24aCVTWKQF-fe-ZQDB9Bw2-SMAiuAG3e6CU783fKU8,15828
|
|
479
|
-
_qwak_proto/qwak/kube_deployment_captain/batch_job_pb2.pyi,sha256=
|
|
482
|
+
_qwak_proto/qwak/kube_deployment_captain/batch_job_pb2.pyi,sha256=v6G0GeQhpE6cGHm7dgAJxxHoNGkIXrGadt4izyClhBw,39789
|
|
480
483
|
_qwak_proto/qwak/kube_deployment_captain/batch_job_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
|
|
481
484
|
_qwak_proto/qwak/kube_deployment_captain/deployment_pb2.py,sha256=urxClvH3ip8kk3p7HN7y6sOK5hvd-iI5WR3KuutxyrU,8188
|
|
482
485
|
_qwak_proto/qwak/kube_deployment_captain/deployment_pb2.pyi,sha256=Sz11iTrL3BlUEl1jXNf0Q1SDe1SX_MjjyB9bMrtsOVg,24090
|
|
@@ -541,7 +544,7 @@ _qwak_proto/qwak/offline/serving/v1/population_pb2_grpc.py,sha256=1oboBPFxaTEXt9
|
|
|
541
544
|
_qwak_proto/qwak/projects/jfrog_project_spec_pb2.py,sha256=W4eUydLc6G50n8M56_b2BVdn39Pq3H0qfemHaf4mCTI,1310
|
|
542
545
|
_qwak_proto/qwak/projects/jfrog_project_spec_pb2.pyi,sha256=Ovtqc19hCDGuOaaXKfMplDFnuMjVnzI7GX1lep8NrjA,1279
|
|
543
546
|
_qwak_proto/qwak/projects/jfrog_project_spec_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
|
|
544
|
-
_qwak_proto/qwak/projects/projects_pb2.py,sha256=
|
|
547
|
+
_qwak_proto/qwak/projects/projects_pb2.py,sha256=3PbNFlXi5_qM2joyhBcC2XdNjQRQubCItRZoKg-hqvk,8846
|
|
545
548
|
_qwak_proto/qwak/projects/projects_pb2.pyi,sha256=JG3ORdann5kvmsPfOH_ECRNJQluTpeQyv7DtHXqHy9I,20118
|
|
546
549
|
_qwak_proto/qwak/projects/projects_pb2_grpc.py,sha256=bM4s9M89p8YSodLUwWTPR_IpB3bGw6E6NQxXoONxxMA,9642
|
|
547
550
|
_qwak_proto/qwak/prompt/v1/prompt/prompt_manager_service_pb2.py,sha256=UlQf0aTOQzdo9SeFIpv7Uytdm2j8-dovKewLIKZWBHI,9021
|
|
@@ -619,45 +622,7 @@ _qwak_proto/qwak/workspace/workspace_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXH
|
|
|
619
622
|
_qwak_proto/qwak/workspace/workspace_service_pb2.py,sha256=93uNm83VVrkLG_XVsuBOBTPs4UJF91YD1xJTOEbRyig,9239
|
|
620
623
|
_qwak_proto/qwak/workspace/workspace_service_pb2.pyi,sha256=nKKCHwnovZhsy8TSVmdz-Vtl0nviOOoX56HD-41Xo08,13726
|
|
621
624
|
_qwak_proto/qwak/workspace/workspace_service_pb2_grpc.py,sha256=yKGuexxTBza99Ihe0DSTniV2ZSd_AG47inHenqfi890,27193
|
|
622
|
-
|
|
623
|
-
frogml_storage/artifactory/__init__.py,sha256=C02rcm7kqsZBVA6c6Gztxamj96hn8Aj6BuzYWFRmWbQ,71
|
|
624
|
-
frogml_storage/artifactory/_artifactory_api.py,sha256=z8YX90bCy82BWqMNUdpbbig1qF5CarkIOfP6V96uYYg,11188
|
|
625
|
-
frogml_storage/authentication/login/__init__.py,sha256=ch8UhQwh3o5ddyoJykQ0Bnog5-8qKHmeDpwqny8xMzM,46
|
|
626
|
-
frogml_storage/authentication/login/_login_cli.py,sha256=9Lyt0poCYVYPFaiWWtcX0kO4MncXeZnkQ-GfEnamnfo,8414
|
|
627
|
-
frogml_storage/authentication/login/_login_command.py,sha256=dj9pOanaYg4PMNbF1EbgMXeoIMo1lfoHp5JFNUZFZ5g,2849
|
|
628
|
-
frogml_storage/authentication/models/__init__.py,sha256=-nG08ZWSD2nIa5lqGtn9pidXM_-XLj4ASG_jHATOK84,113
|
|
629
|
-
frogml_storage/authentication/models/_auth.py,sha256=uKo6hxD1jjDGmYeQIJSnKJCOZIE6iTA8LJJs__BTufA,600
|
|
630
|
-
frogml_storage/authentication/models/_auth_config.py,sha256=uDcPXaPRXKDEtAhHEuN-GQTeDg8AUL8FE4aiAByHU_I,2024
|
|
631
|
-
frogml_storage/authentication/models/_login.py,sha256=jcFdiWRqjaVAGG9hiYpPHF0WUNT4DqMqVp1FPFGuwOM,723
|
|
632
|
-
frogml_storage/authentication/utils/__init__.py,sha256=NBT-KjHJrqXYS3sZvWENjq1gmLiyKSN3fXWDl0-rh8k,430
|
|
633
|
-
frogml_storage/authentication/utils/_authentication_utils.py,sha256=WTQRclyGKWTKLfFQEV3Xgq1_aCsKVDZsuhM3WkxeC1k,9730
|
|
634
|
-
frogml_storage/authentication/utils/_login_checks_utils.py,sha256=OQZSJHN9C0ujCaAjYuttBBxyoOrWo0KGLKOw5et__ds,2993
|
|
635
|
-
frogml_storage/base_storage.py,sha256=gbkDRhXhLLJrr72A2vGcQqEHEjCDEk9-n7ye98F3_2k,5606
|
|
636
|
-
frogml_storage/constants.py,sha256=Fx-Dyzkl9e28O2qCvIgyEf2aBuhftXAdbn2BIfXFaGw,1542
|
|
637
|
-
frogml_storage/exceptions/checksum_verification_error.py,sha256=t1muLRYvqza7Q4jo1mYHck9ZoDwlGE7WxcheAFAnIQs,159
|
|
638
|
-
frogml_storage/exceptions/validation_error.py,sha256=_twu_xcL-O7D26qkskuuR-_NcnPPTADIUO9HGY4g0x0,156
|
|
639
|
-
frogml_storage/frog_ml.py,sha256=RO6YVLv1eBAZWs1noWkNqyWcI8yNoHP3mnGIr67qCkk,24182
|
|
640
|
-
frogml_storage/http/__init__.py,sha256=BOXGzjq1gJXlihldaZvMQNGJIdCy-UHWsk3u-bXEZnQ,36
|
|
641
|
-
frogml_storage/http/http_client.py,sha256=jYwe4aHhMRsTjl5fZDlJxaTFuY3RnIcw7P1Wz8Ykz4g,2420
|
|
642
|
-
frogml_storage/logging/__init__.py,sha256=JaBPWQQt3pjoLhUcZtgqPydGlKAWYpSvfPkt2z1Wi90,32
|
|
643
|
-
frogml_storage/logging/_log_config.py,sha256=ytwyRLYr3SyRlo4UpTjLBtlNqVApP9URTVO5m5DN3uU,1162
|
|
644
|
-
frogml_storage/logging/log_utils.py,sha256=MLuQvBrmysiWAeKKq1lHFt3o6DJPyafFWRJNTg9ECaY,711
|
|
645
|
-
frogml_storage/models/__init__.py,sha256=ZoWMjtQ4Bn-ARJqKEUp-XDRbHgaSQJG_1M2S5z91jbQ,47
|
|
646
|
-
frogml_storage/models/_download_context.py,sha256=8Jc5JmQG5mwObQp-CXF-D6dP3xUbZcyHUSdwM-p-8s0,1667
|
|
647
|
-
frogml_storage/models/dataset_manifest.py,sha256=w5ARGQSkRSP7mofsbxOcc9bXORKXIMo5iF1yFGd8Nyw,303
|
|
648
|
-
frogml_storage/models/entity_manifest.py,sha256=BXwJz1q3NIH6QFccnGDG51R3HnHO9-oCeO9qZqaxQg4,2787
|
|
649
|
-
frogml_storage/models/frogml_dataset_version.py,sha256=Uwe50e8eD9XqwARCsa4-iJJqyMQsDqs9E-iuAInvJuA,637
|
|
650
|
-
frogml_storage/models/frogml_entity_type_info.py,sha256=gIoWcm-nqKQRpVg8DJcq8jJE-e7qgR_-EUv6R8wKCFs,1448
|
|
651
|
-
frogml_storage/models/frogml_entity_version.py,sha256=jiDGk3xcZ-OaaTacs6VLshpyUNKYs8qgBOPk1NUJFkY,928
|
|
652
|
-
frogml_storage/models/frogml_model_version.py,sha256=2pVHW-BhowBcIScLZzWj5SH0abAirP3cDvXeq334lJI,631
|
|
653
|
-
frogml_storage/models/model_manifest.py,sha256=NEyUTt0ZbLbv-47nb_yJJrA0xnTCk67SdEUcZ09tjr8,2200
|
|
654
|
-
frogml_storage/models/serialization_metadata.py,sha256=KOJ9Yj0XGbYudpcUBMh67Hd7VohmoWsYOvhD-L2yzyY,347
|
|
655
|
-
frogml_storage/utils/__init__.py,sha256=HQUWfuGUIPZY7kfS795TRW8BQ4WmNqrNjS7lUrbxCdk,299
|
|
656
|
-
frogml_storage/utils/_environment.py,sha256=NEnRxaFdRDi3UA33IVTSfQxqc9ZB6nV9YlHDmT5bIkI,650
|
|
657
|
-
frogml_storage/utils/_input_checks_utility.py,sha256=ZUvkxhcTcbSKtpSo5ePWha3Ca5xohxIpiAlZlpcQ3xk,3221
|
|
658
|
-
frogml_storage/utils/_storage_utils.py,sha256=HB2g7uY5A3b33yIcAUM1OjHb5jWsnpESsiDrEviQwrI,366
|
|
659
|
-
frogml_storage/utils/_url_utils.py,sha256=NUEfz9Fp1iE8b676-A5wrMlSTsJVRKrUhcUItOFAJD8,821
|
|
660
|
-
qwak/__init__.py,sha256=ltT02TCcHd-OCbe8R9-Vv6aqO-v77_184q82hlva3zw,587
|
|
625
|
+
qwak/__init__.py,sha256=pzn7kqlQYa5blclBgjgbXKm8XngWu_7svRRNBDg1DSA,587
|
|
661
626
|
qwak/automations/__init__.py,sha256=qFZRvCxUUn8gcxkJR0v19ulHW2oJ0x6-Rif7HiheDP4,1522
|
|
662
627
|
qwak/automations/automation_executions.py,sha256=5MeH_epYYWb8NKXgAozwT_jPyyUDednBHG7izloi7RY,3228
|
|
663
628
|
qwak/automations/automations.py,sha256=2Wyrgj2mW5VmJzTKAXj3vQM_svYCuq_Bsu3_Vu9SdR4,12732
|
|
@@ -728,9 +693,9 @@ qwak/clients/location_discovery/client.py,sha256=O-wjMGiSTJ5qGjD0VhofE6Mv7NqvmwC
|
|
|
728
693
|
qwak/clients/logging_client/__init__.py,sha256=1OCHnigQBYThBwGbxCreYA0BgP0HcuLFzNEWd3Yxh-c,34
|
|
729
694
|
qwak/clients/logging_client/client.py,sha256=bLTxX2JFF1DKWjxtqoZexS10pN1BHSen3IRSbWjX5HQ,4746
|
|
730
695
|
qwak/clients/model_management/__init__.py,sha256=vjWVP8MjmK4_A70WOgJqa6x24AeLK-ABjGJtogGzw9w,43
|
|
731
|
-
qwak/clients/model_management/client.py,sha256=
|
|
696
|
+
qwak/clients/model_management/client.py,sha256=TdfWm_sjWWJHe2bBMwSF773MR3Pmh-ZYaUQtkXGaVV0,4471
|
|
732
697
|
qwak/clients/project/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
733
|
-
qwak/clients/project/client.py,sha256=
|
|
698
|
+
qwak/clients/project/client.py,sha256=15HNp5yAdS8FLNxK-NLOKbRZFnqU_4F22kHywKFZkl0,2180
|
|
734
699
|
qwak/clients/prompt_manager/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
735
700
|
qwak/clients/prompt_manager/model_descriptor_mapper.py,sha256=QDLPImRIPv9Jrdtg03oL7zgKvrWDXmzK0QXkUFAo1bs,7395
|
|
736
701
|
qwak/clients/prompt_manager/prompt_manager_client.py,sha256=Mg1uqJjwj57jFfNKx6tomzfOe6C8keTKpxvXFSPo0yA,6952
|
|
@@ -767,7 +732,7 @@ qwak/feature_store/_common/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJ
|
|
|
767
732
|
qwak/feature_store/_common/artifact_utils.py,sha256=9N_hFk5eguOWUHR4xw93bCghJyG3oc7zNfb40ypGQy8,1933
|
|
768
733
|
qwak/feature_store/_common/feature_set_utils.py,sha256=dhtPWnwv1nB3h3EA8FAYklx1qGkO_Dj1jecaBGNNpew,8634
|
|
769
734
|
qwak/feature_store/_common/functions.py,sha256=kSNYJ7dy48NN09HG9asm4ibQh0JaCGcZYsRDjRWlUHE,659
|
|
770
|
-
qwak/feature_store/_common/packaging.py,sha256=
|
|
735
|
+
qwak/feature_store/_common/packaging.py,sha256=dUXwg3dFc5asv0tqeepnfCE6sUnlu_g8wcUUMNOqWaM,8754
|
|
771
736
|
qwak/feature_store/_common/source_code_spec.py,sha256=UmpuJxte8OtztkXuN5d4TFQk_f-MLnjHm_T7W4DF3N4,3762
|
|
772
737
|
qwak/feature_store/_common/source_code_spec_factory.py,sha256=F4aKvUnDHvRiB-jQpeH2kTNpUsZlGUC1kAVIz9xWIMY,3326
|
|
773
738
|
qwak/feature_store/_common/value.py,sha256=6aN3101DvEDUJalNL1LrzH32wi9SVhl0ZpsGHcFJZzg,323
|
|
@@ -898,9 +863,9 @@ qwak/inner/build_logic/tools/files.py,sha256=druKkTp640Zy3x1MLjuTuXY56r5blEnH8dS
|
|
|
898
863
|
qwak/inner/build_logic/tools/ignore_files.py,sha256=E65s3m46pmSX4q4hX2-R5HYHDO4VCQHjip8Xn_yOBRg,750
|
|
899
864
|
qwak/inner/build_logic/tools/text.py,sha256=tH-v19Mt8l90sMVxku5XRtrderT0qdRqJ-jLijqannA,188
|
|
900
865
|
qwak/inner/build_logic/trigger_build_context.py,sha256=n5s7MRk41h8cxPI3IVuqMQUVuCc-D8A7Hhq35YuuAps,200
|
|
901
|
-
qwak/inner/const.py,sha256=
|
|
866
|
+
qwak/inner/const.py,sha256=x4QzsCKr80c7CfYI2TkgIVdvX7vKrTq3mmHp7GFVF34,1131
|
|
902
867
|
qwak/inner/di_configuration/__init__.py,sha256=_QzRdBIKha032PIx5qqJ2bI6K1Z7ceAqVUXmBJtXvJ8,1709
|
|
903
|
-
qwak/inner/di_configuration/account.py,sha256=
|
|
868
|
+
qwak/inner/di_configuration/account.py,sha256=V5NOVOsxhe11-oD4i4UFskWMhq0EOyJ0rDbNSAv7GfA,5338
|
|
904
869
|
qwak/inner/di_configuration/config.yml,sha256=GUvaZMWIDIR_d7hFcPVG_kHdCwpERKH1AFDakG3vqI4,242
|
|
905
870
|
qwak/inner/di_configuration/containers.py,sha256=dK-cfd8ajT6yujlEhPhcRYB-avdYDcreQz_2kTIVDRQ,1252
|
|
906
871
|
qwak/inner/di_configuration/session.py,sha256=Pwqapcu-0dsbb5dDn7ZGewVk4bYRN9sWk6ity72IDqY,452
|
|
@@ -912,11 +877,11 @@ qwak/inner/runtime_di/__init__.py,sha256=ylaHwo2Hlq3pmeziA0Ip87DMnkcuBffKKi0tTH7
|
|
|
912
877
|
qwak/inner/runtime_di/containers.py,sha256=-AREM-IUAd9JhrhQablqixxpj4L9Tk1iQkK1fA5YbrA,545
|
|
913
878
|
qwak/inner/singleton_meta.py,sha256=1cU99I0f9tjuMQLMJyLsK1oK3fZJMsO5-TbRHAMXqds,627
|
|
914
879
|
qwak/inner/tool/__init__.py,sha256=DDoMZk7LIZNfu63Ft0mLE5wflwlQoghAJ1ZmMAcAeVE,74
|
|
915
|
-
qwak/inner/tool/auth.py,sha256=
|
|
880
|
+
qwak/inner/tool/auth.py,sha256=_78ufvtkNQ3ogcFwkotVySZgdS1yMTuIDMLOBsICkSU,4483
|
|
916
881
|
qwak/inner/tool/grpc/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
917
|
-
qwak/inner/tool/grpc/grpc_auth.py,sha256=
|
|
918
|
-
qwak/inner/tool/grpc/grpc_tools.py,sha256=
|
|
919
|
-
qwak/inner/tool/grpc/grpc_try_wrapping.py,sha256=
|
|
882
|
+
qwak/inner/tool/grpc/grpc_auth.py,sha256=_DvwaRruUu12l4K6ACvmzsTbIDlKl1kuKjm-ustQ5ns,1139
|
|
883
|
+
qwak/inner/tool/grpc/grpc_tools.py,sha256=lY7SkXnnE0MrZOCFmg2KZZ5iZ7YJu_Z5F-R-6ezm4cU,6897
|
|
884
|
+
qwak/inner/tool/grpc/grpc_try_wrapping.py,sha256=4RScvo2jiCr5XwZvaH516OJWqbMSxC8MWIarrD-5HnA,5654
|
|
920
885
|
qwak/inner/tool/protobuf_factory.py,sha256=ar_oY38w_x0sxgVF7EBs5h7gchNsDntvtKK5sSYxb24,1686
|
|
921
886
|
qwak/inner/tool/retry_utils.py,sha256=KcSFJuj02RKF-H9INpCmdiTNXlywEMJ2ClBa00N9aNM,435
|
|
922
887
|
qwak/inner/tool/run_config/__init__.py,sha256=AIuy7zhaF9Zvlp6tuLonnCfjp7vIwH1YOl8P5qT6YGE,275
|
|
@@ -1051,7 +1016,7 @@ qwak/qwak_client/builds/build.py,sha256=1dBRZkvZV04uw8RI-7iOs_yvfgUzglh9ZyzgHJyU
|
|
|
1051
1016
|
qwak/qwak_client/builds/filters/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
1052
1017
|
qwak/qwak_client/builds/filters/metric_filter.py,sha256=i0ZsjqN1iw-cX_uxvbJzRT-EYAOEqejKkukLiyYmYqE,1185
|
|
1053
1018
|
qwak/qwak_client/builds/filters/parameter_filter.py,sha256=IY-J6bAIEM_KyVzESMDx0xv_fAScRp_AsltdMJk0IDY,1088
|
|
1054
|
-
qwak/qwak_client/client.py,sha256
|
|
1019
|
+
qwak/qwak_client/client.py,sha256=-GMcSRRvuPH6oYasFK3RO3_bQ7GcWQ-y7KoyvMBRxcQ,28246
|
|
1055
1020
|
qwak/qwak_client/data_versioning/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
1056
1021
|
qwak/qwak_client/data_versioning/data_tag.py,sha256=B7_TsBrp0LZK-yLqGpodVhoh67dpWBsiyEgWq3RCKcA,806
|
|
1057
1022
|
qwak/qwak_client/deployments/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
@@ -1077,7 +1042,7 @@ qwak/vector_store/client.py,sha256=UVc3UK2ytyhhXPF8rAznENXBWhjYpxEl8i5JIjWgO_8,6
|
|
|
1077
1042
|
qwak/vector_store/collection.py,sha256=3HWXU5DwcK30-eJ1ABs-r0S3UsUAPmDjS_aR1DgSHSM,16916
|
|
1078
1043
|
qwak/vector_store/filters.py,sha256=y549rUHf2pRSDqNnslxb7DgUj-eNu_i755-CuTt8apo,8209
|
|
1079
1044
|
qwak/vector_store/inference_client.py,sha256=Zyqx3nmhvyiNDBlaL8J1egDiZuB__ecsfvSTgNxrkT0,3711
|
|
1080
|
-
qwak/vector_store/rest_helpers.py,sha256=
|
|
1045
|
+
qwak/vector_store/rest_helpers.py,sha256=pKp77Qt8NPJDuampmOWhJgt6zgMLDq6YwahHwMcehME,2323
|
|
1081
1046
|
qwak/vector_store/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
1082
1047
|
qwak/vector_store/utils/filter_utils.py,sha256=fK01Ag2UxdXXqCVWWQOibh2OAGvm6Lvdl1dBWk3vVlg,837
|
|
1083
1048
|
qwak/vector_store/utils/upsert_utils.py,sha256=DEjJlX_h1NM1_AhakH_IyfOJPOxWnyTPG9DOrEH7QIw,7459
|
|
@@ -1130,6 +1095,6 @@ qwak_services_mock/mocks/workspace_manager_service_mock.py,sha256=O9ZSwln4T4kHVk
|
|
|
1130
1095
|
qwak_services_mock/services_mock.py,sha256=0BWwS2re9ryO3JrJJgSNyEQ0lDOjyrpV36oa8t7Pd7A,19163
|
|
1131
1096
|
qwak_services_mock/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
1132
1097
|
qwak_services_mock/utils/service_utils.py,sha256=ZlB0CnB1J6oBn6_m7fQO2U8tKoboHdUa6ljjkRMYNXU,265
|
|
1133
|
-
qwak_core-0.4.
|
|
1134
|
-
qwak_core-0.4.
|
|
1135
|
-
qwak_core-0.4.
|
|
1098
|
+
qwak_core-0.4.392.dist-info/METADATA,sha256=lVSyywf9wGSkGK1G6jQ1P1UPQE3Ewbal-Wp0puxOHes,2088
|
|
1099
|
+
qwak_core-0.4.392.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
|
|
1100
|
+
qwak_core-0.4.392.dist-info/RECORD,,
|
frogml_storage/__init__.py
DELETED
|
@@ -1 +0,0 @@
|
|
|
1
|
-
__version__ = "0.4.366"
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
from ._artifactory_api import ArtifactoryApi, StartTransactionResponse
|
|
@@ -1,315 +0,0 @@
|
|
|
1
|
-
import json
|
|
2
|
-
import os
|
|
3
|
-
from typing import Optional
|
|
4
|
-
|
|
5
|
-
from requests import Response
|
|
6
|
-
from tqdm.auto import tqdm
|
|
7
|
-
from tqdm.utils import CallbackIOWrapper
|
|
8
|
-
from urllib3 import Retry
|
|
9
|
-
|
|
10
|
-
from frogml_storage.logging import logger
|
|
11
|
-
from frogml_storage.utils import join_url
|
|
12
|
-
from frogml_storage.constants import CHECKSUM_SHA2_HEADER
|
|
13
|
-
from frogml_storage.models.entity_manifest import Checksums, EntityManifest
|
|
14
|
-
from frogml_storage.models.frogml_entity_type_info import FrogMLEntityTypeInfo
|
|
15
|
-
from frogml_storage.http import HTTPClient
|
|
16
|
-
from frogml_storage.models import DownloadContext
|
|
17
|
-
|
|
18
|
-
|
|
19
|
-
class StartTransactionResponse:
|
|
20
|
-
files_upload_path: str
|
|
21
|
-
lead_upload_path: str
|
|
22
|
-
dependencies_upload_path: str
|
|
23
|
-
code_upload_path: str
|
|
24
|
-
transaction_id: str
|
|
25
|
-
|
|
26
|
-
def __init__(
|
|
27
|
-
self,
|
|
28
|
-
files_upload_path,
|
|
29
|
-
lead_upload_path,
|
|
30
|
-
dependencies_upload_path,
|
|
31
|
-
code_upload_path,
|
|
32
|
-
transaction_id,
|
|
33
|
-
):
|
|
34
|
-
self.files_upload_path = files_upload_path
|
|
35
|
-
self.lead_upload_path = lead_upload_path
|
|
36
|
-
self.dependencies_upload_path = dependencies_upload_path
|
|
37
|
-
self.code_upload_path = code_upload_path
|
|
38
|
-
self.transaction_id = transaction_id
|
|
39
|
-
|
|
40
|
-
|
|
41
|
-
class ArtifactoryApi:
|
|
42
|
-
def __init__(self, uri, auth=None, http_client=None):
|
|
43
|
-
self.uri = uri
|
|
44
|
-
if http_client is not None:
|
|
45
|
-
self.http_client = http_client
|
|
46
|
-
else:
|
|
47
|
-
self.auth = auth
|
|
48
|
-
self.http_client = HTTPClient(auth=auth)
|
|
49
|
-
|
|
50
|
-
def start_transaction(
|
|
51
|
-
self,
|
|
52
|
-
entity_type_info: FrogMLEntityTypeInfo,
|
|
53
|
-
repository: str,
|
|
54
|
-
entity_name: str,
|
|
55
|
-
version: Optional[str],
|
|
56
|
-
) -> StartTransactionResponse:
|
|
57
|
-
"""
|
|
58
|
-
Initializes an upload. Returns transaction ID and upload path
|
|
59
|
-
"""
|
|
60
|
-
if version is None:
|
|
61
|
-
start_transaction_url = (
|
|
62
|
-
f"{self.uri}/api/machinelearning/{repository}/"
|
|
63
|
-
f"{entity_type_info.entity_type}/{entity_name}/start-transaction"
|
|
64
|
-
)
|
|
65
|
-
else:
|
|
66
|
-
start_transaction_url = (
|
|
67
|
-
f"{self.uri}/api/machinelearning/{repository}/{entity_type_info.entity_type}"
|
|
68
|
-
f"/{entity_name}/start-transaction/{version}"
|
|
69
|
-
)
|
|
70
|
-
try:
|
|
71
|
-
response = self.http_client.post(start_transaction_url)
|
|
72
|
-
response.raise_for_status()
|
|
73
|
-
files_upload_path = response.json()["filesUploadPath"]
|
|
74
|
-
lead_upload_path = response.json()["leadUploadPath"]
|
|
75
|
-
dependencies_upload_path = response.json()["dependenciesUploadPath"]
|
|
76
|
-
code_upload_path = response.json()["codeUploadPath"]
|
|
77
|
-
transaction_id = response.json()["transactionId"]
|
|
78
|
-
except Exception as exception:
|
|
79
|
-
err = (
|
|
80
|
-
f"Error occurred while trying to start an upload transaction for "
|
|
81
|
-
f"{entity_type_info.entity_type}: '{entity_name}'"
|
|
82
|
-
f" Error: '{exception}'"
|
|
83
|
-
)
|
|
84
|
-
logger.error(err, exc_info=False)
|
|
85
|
-
raise exception
|
|
86
|
-
return StartTransactionResponse(
|
|
87
|
-
files_upload_path=files_upload_path,
|
|
88
|
-
lead_upload_path=lead_upload_path,
|
|
89
|
-
dependencies_upload_path=dependencies_upload_path,
|
|
90
|
-
code_upload_path=code_upload_path,
|
|
91
|
-
transaction_id=transaction_id,
|
|
92
|
-
)
|
|
93
|
-
|
|
94
|
-
def end_transaction(
|
|
95
|
-
self,
|
|
96
|
-
entity_type: FrogMLEntityTypeInfo,
|
|
97
|
-
repository: str,
|
|
98
|
-
entity_name: str,
|
|
99
|
-
entity_manifest: EntityManifest,
|
|
100
|
-
transaction_id: str,
|
|
101
|
-
version: str,
|
|
102
|
-
properties: Optional[dict[str, str]],
|
|
103
|
-
) -> None:
|
|
104
|
-
"""
|
|
105
|
-
Upload model-manifest.json | dataset-manifest.json file, makes the model | dataset available in the repository
|
|
106
|
-
"""
|
|
107
|
-
filename = entity_type.metadata_file_name
|
|
108
|
-
|
|
109
|
-
url = join_url(
|
|
110
|
-
self.uri,
|
|
111
|
-
"api",
|
|
112
|
-
"machinelearning",
|
|
113
|
-
repository,
|
|
114
|
-
entity_type.entity_type,
|
|
115
|
-
"entity-manifest",
|
|
116
|
-
entity_name,
|
|
117
|
-
version,
|
|
118
|
-
transaction_id,
|
|
119
|
-
)
|
|
120
|
-
|
|
121
|
-
json_entity_manifest = entity_manifest.to_json()
|
|
122
|
-
self.upload_entity_manifest(
|
|
123
|
-
entity_type=entity_type,
|
|
124
|
-
filename=filename,
|
|
125
|
-
payload=json_entity_manifest,
|
|
126
|
-
url=url,
|
|
127
|
-
properties=properties,
|
|
128
|
-
)
|
|
129
|
-
|
|
130
|
-
def download_file(self, args: DownloadContext) -> None:
|
|
131
|
-
filename = os.path.basename(args.target_path)
|
|
132
|
-
try:
|
|
133
|
-
url = f"{self.uri}/{args.repo_key}/{args.source_url}"
|
|
134
|
-
with self.http_client.get(url=url, stream=True) as response:
|
|
135
|
-
response.raise_for_status()
|
|
136
|
-
total_size = int(response.headers.get("content-length", 0))
|
|
137
|
-
with open(args.target_path, "wb") as file:
|
|
138
|
-
with self.__initialize_progress_bar(total_size, filename) as pbar:
|
|
139
|
-
for chunk in response.iter_content(chunk_size=8192):
|
|
140
|
-
if chunk:
|
|
141
|
-
file.write(chunk)
|
|
142
|
-
pbar.update(len(chunk))
|
|
143
|
-
|
|
144
|
-
except Exception as exception:
|
|
145
|
-
self.__handle_download_exception(exception, args.target_path, filename)
|
|
146
|
-
|
|
147
|
-
def get_entity_manifest(
|
|
148
|
-
self,
|
|
149
|
-
entity_type_info: FrogMLEntityTypeInfo,
|
|
150
|
-
repository: str,
|
|
151
|
-
entity_name: str,
|
|
152
|
-
namespace: Optional[str],
|
|
153
|
-
version: Optional[str],
|
|
154
|
-
) -> dict:
|
|
155
|
-
url = join_url(
|
|
156
|
-
self.uri,
|
|
157
|
-
"api",
|
|
158
|
-
"machinelearning",
|
|
159
|
-
repository,
|
|
160
|
-
entity_type_info.entity_type,
|
|
161
|
-
"entity-manifest",
|
|
162
|
-
namespace,
|
|
163
|
-
entity_name,
|
|
164
|
-
version,
|
|
165
|
-
)
|
|
166
|
-
try:
|
|
167
|
-
with self.http_client.get(url=url) as r:
|
|
168
|
-
r.raise_for_status()
|
|
169
|
-
return r.json()
|
|
170
|
-
except Exception as exception:
|
|
171
|
-
err = f"Error occurred while trying to get {entity_type_info.entity_type} info file. Error: '{exception}'"
|
|
172
|
-
logger.error(err, exc_info=False)
|
|
173
|
-
raise exception
|
|
174
|
-
|
|
175
|
-
@staticmethod
|
|
176
|
-
def __handle_download_exception(
|
|
177
|
-
exception: Exception, target_path: str, filename: str
|
|
178
|
-
) -> None:
|
|
179
|
-
if os.path.exists(target_path):
|
|
180
|
-
os.remove(target_path)
|
|
181
|
-
err = f"Error occurred while trying to download file: '{filename}' Error: '{exception}'"
|
|
182
|
-
logger.error(err, exc_info=False)
|
|
183
|
-
raise exception
|
|
184
|
-
|
|
185
|
-
def get_artifact_checksum(self, download_context: DownloadContext) -> str:
|
|
186
|
-
url = f"{self.uri}/{download_context.repo_key}/{download_context.source_url}"
|
|
187
|
-
try:
|
|
188
|
-
with self.http_client.head(url=url, stream=True) as response:
|
|
189
|
-
response.raise_for_status()
|
|
190
|
-
return response.headers.get(CHECKSUM_SHA2_HEADER)
|
|
191
|
-
|
|
192
|
-
except Exception as exception:
|
|
193
|
-
logger.error(exception.__cause__, exc_info=False)
|
|
194
|
-
raise exception
|
|
195
|
-
|
|
196
|
-
def upload_entity_manifest(
|
|
197
|
-
self,
|
|
198
|
-
entity_type: FrogMLEntityTypeInfo,
|
|
199
|
-
filename: str,
|
|
200
|
-
payload: str,
|
|
201
|
-
url: str,
|
|
202
|
-
properties: Optional[dict[str, str]],
|
|
203
|
-
stream: bool = False,
|
|
204
|
-
) -> None:
|
|
205
|
-
body_part_name = f"{entity_type.body_part_stream}"
|
|
206
|
-
|
|
207
|
-
try:
|
|
208
|
-
files = {
|
|
209
|
-
f"{body_part_name}": (
|
|
210
|
-
f"{body_part_name}",
|
|
211
|
-
payload,
|
|
212
|
-
"application/octet-stream",
|
|
213
|
-
), # Include the InputStream
|
|
214
|
-
"additionalData": (
|
|
215
|
-
"additionalData",
|
|
216
|
-
json.dumps(properties),
|
|
217
|
-
"application/octet-stream",
|
|
218
|
-
), # Include the object
|
|
219
|
-
}
|
|
220
|
-
with self.http_client.put(url=url, files=files, stream=stream) as response:
|
|
221
|
-
response.raise_for_status()
|
|
222
|
-
except Exception as exception:
|
|
223
|
-
err = f"Error occurred while trying to upload file: '{filename}' Error: '{exception}'"
|
|
224
|
-
logger.error(err, exc_info=False)
|
|
225
|
-
raise exception
|
|
226
|
-
|
|
227
|
-
def upload_file(self, file_path: str, url: str) -> None:
|
|
228
|
-
wrapped_file = None
|
|
229
|
-
try:
|
|
230
|
-
file_size = os.stat(file_path).st_size
|
|
231
|
-
with (
|
|
232
|
-
self.__initialize_progress_bar(file_size, file_path) as pbar,
|
|
233
|
-
open(file_path, "rb") as file,
|
|
234
|
-
):
|
|
235
|
-
wrapped_file = CallbackIOWrapper(pbar.update, file, "read")
|
|
236
|
-
with self.http_client.put(url=url, payload=wrapped_file) as response:
|
|
237
|
-
response.raise_for_status()
|
|
238
|
-
except Exception as exception:
|
|
239
|
-
err = f"Error occurred while trying to upload file: '{file_path}' Error: '{exception}'"
|
|
240
|
-
logger.error(err, exc_info=False)
|
|
241
|
-
raise type(exception)(f"{err} File: {file_path}") from exception
|
|
242
|
-
finally:
|
|
243
|
-
if wrapped_file is not None:
|
|
244
|
-
wrapped_file.close()
|
|
245
|
-
|
|
246
|
-
def checksum_deployment(
|
|
247
|
-
self,
|
|
248
|
-
checksum: Checksums,
|
|
249
|
-
url: str,
|
|
250
|
-
full_path: str,
|
|
251
|
-
stream: bool = False,
|
|
252
|
-
) -> bool:
|
|
253
|
-
response = self.http_client.put(
|
|
254
|
-
url=url,
|
|
255
|
-
headers={"X-Checksum-Sha256": checksum.sha2, "X-Checksum-Deploy": "true"},
|
|
256
|
-
stream=stream,
|
|
257
|
-
)
|
|
258
|
-
if response.status_code != 200 and response.status_code != 201:
|
|
259
|
-
return False
|
|
260
|
-
else:
|
|
261
|
-
file_size = os.path.getsize(full_path)
|
|
262
|
-
pbar = self.__initialize_progress_bar(file_size, full_path)
|
|
263
|
-
pbar.update(file_size)
|
|
264
|
-
pbar.close()
|
|
265
|
-
return True
|
|
266
|
-
|
|
267
|
-
@staticmethod
|
|
268
|
-
def __initialize_progress_bar(total_size: int, filename: str) -> tqdm:
|
|
269
|
-
return tqdm(
|
|
270
|
-
total=total_size, unit="B", unit_scale=True, desc=filename, initial=0
|
|
271
|
-
)
|
|
272
|
-
|
|
273
|
-
def encrypt_password(self) -> Response:
|
|
274
|
-
"""
|
|
275
|
-
returns encrypted password as text
|
|
276
|
-
"""
|
|
277
|
-
return self.http_client.get(
|
|
278
|
-
url=join_url(self.uri, "/api/security/encryptedPassword")
|
|
279
|
-
)
|
|
280
|
-
|
|
281
|
-
def ping(self) -> Response:
|
|
282
|
-
"""
|
|
283
|
-
Sends a ping to Artifactory to validate login status
|
|
284
|
-
"""
|
|
285
|
-
url = join_url(self.uri, "api/system/ping")
|
|
286
|
-
return self.http_client.get(url=url)
|
|
287
|
-
|
|
288
|
-
def get_artifactory_version(self) -> Response:
|
|
289
|
-
return self.http_client.get(url=join_url(self.uri, "/api/system/version"))
|
|
290
|
-
|
|
291
|
-
def create_machinelearning_local_repo(self, repo_name: str) -> Response:
|
|
292
|
-
data = {
|
|
293
|
-
"rclass": "local",
|
|
294
|
-
"packageType": "machinelearning",
|
|
295
|
-
}
|
|
296
|
-
return self.http_client.put(
|
|
297
|
-
url=join_url(self.uri, "/api/repositories/" + repo_name), json=data
|
|
298
|
-
)
|
|
299
|
-
|
|
300
|
-
def delete_frogml_local_repo(self, repo_name: str) -> Response:
|
|
301
|
-
return self.http_client.delete(
|
|
302
|
-
url=join_url(self.uri, "/api/repositories/" + repo_name)
|
|
303
|
-
)
|
|
304
|
-
|
|
305
|
-
|
|
306
|
-
class RetryWithLog(Retry):
|
|
307
|
-
"""
|
|
308
|
-
Adding extra logs before making a retry request
|
|
309
|
-
"""
|
|
310
|
-
|
|
311
|
-
def __init__(self, *args, **kwargs):
|
|
312
|
-
history = kwargs.get("history")
|
|
313
|
-
if history is not None:
|
|
314
|
-
logger.debug(f"Error: ${history[-1].error}\nretrying...")
|
|
315
|
-
super().__init__(*args, **kwargs)
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
from ._login_cli import login as frogml_login
|