frogml-core 0.0.71__py3-none-any.whl → 0.0.72__py3-none-any.whl

frogml_core/__init__.py

@@ -1,7 +1,7 @@
 """Top-level package for frogml."""
 
 __author__ = "jfrog"
-__version__ = "0.0.71"
+__version__ = "0.0.72"
 
 from frogml_core.inner.di_configuration import wire_dependencies
 from frogml_core.model.model_version_tracking import (  # noqa: F401,E501

frogml_core/inner/build_logic/phases/phase_020_remote_register_frogml_build/upload_step.py

@@ -10,7 +10,6 @@ from retrying import retry
 
 from frogml_proto.qwak.builds.build_url_pb2 import BuildVersioningTagsType
 from frogml_proto.qwak.builds.builds_orchestrator_service_pb2 import (
-    AuthenticationDetail,
     GetBuildVersioningUploadURLResponse,
 )
 from frogml_core.exceptions import FrogmlException, FrogmlGeneralBuildException
@@ -159,15 +158,17 @@ class UploadStep(Step):
     ):
         self.build_logger.debug(f"Upload file {file}")
 
-        pre_signed_url_response = self.get_pre_signed_upload_url(
-            tag=tag, tag_type=BuildVersioningTagsType.FILE_TAG_TYPE
+        pre_signed_url_response: GetBuildVersioningUploadURLResponse = (
+            self.get_pre_signed_upload_url(
+                tag=tag, tag_type=BuildVersioningTagsType.FILE_TAG_TYPE
+            )
         )
         self.upload_file_to_remote_storge(
             upload_url=pre_signed_url_response.upload_url,
             file=file,
             all_files_size_to_upload=all_files_size_to_upload,
             read_so_far=read_so_far,
-            authentication_details=pre_signed_url_response.authentication_detail,
+            headers=pre_signed_url_response.headers,
         )
 
         self.build_logger.debug(f"Upload file {file} completed")
@@ -204,22 +205,16 @@ class UploadStep(Step):
         file: Path,
         all_files_size_to_upload: int,
         read_so_far: int,
-        authentication_details: Optional[AuthenticationDetail] = None,
+        headers: Optional[dict] = None,
     ):
+        if not headers:
+            headers = {}
+
         try:
             self.build_logger.debug(f"Upload file {file} to Qwak storage")
-            auth: Optional[Tuple[str, str]] = None
-            if (
-                authentication_details.WhichOneof("integration_type")
-                == "jfrog_authentication_detail"
-            ):
-                auth = (
-                    authentication_details.jfrog_authentication_detail.username,
-                    authentication_details.jfrog_authentication_detail.token,
-                )
 
             self.send_request(
-                upload_url, file, all_files_size_to_upload, read_so_far, auth
+                upload_url, file, all_files_size_to_upload, read_so_far, headers
             )
             self.build_logger.debug(
                 f"File {file} uploaded to Qwak storage successfully"
@@ -237,8 +232,14 @@ class UploadStep(Step):
         file: Path,
         all_files_size_to_upload: int,
         read_so_far: int,
-        auth: Optional[Tuple[str, str]],
+        headers: Optional[dict],
     ):
+        if not headers:
+            headers = {}
+
+        # Adding to the current headers the content-type
+        headers["content-type"] = "text/plain"
+
         http_response = requests.put(  # nosec B113
             url=upload_url,
             data=UploadInChunks(
@@ -248,8 +249,7 @@ class UploadStep(Step):
                 all_files_size_to_upload=all_files_size_to_upload,
                 read_so_far=read_so_far,
             ),
-            auth=auth,
-            headers={"content-type": "text/plain"},
+            headers=headers,
         )
 
         if http_response.status_code not in [200, 201]:

{frogml_core-0.0.71.dist-info → frogml_core-0.0.72.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: frogml-core
-Version: 0.0.71
+Version: 0.0.72
 Summary: frogml Core contains the necessary objects and communication tools for using the Jfrog ml Platform
 License: Apache-2.0
 Keywords: mlops,ml,deployment,serving,model

{frogml_core-0.0.71.dist-info → frogml_core-0.0.72.dist-info}/RECORD

@@ -1,4 +1,4 @@
-frogml_core/__init__.py,sha256=ulzYRqlreb83b240lIjMo6hT8Vweg-Pr8UncNGR-d_4,777
+frogml_core/__init__.py,sha256=3K96265rqyrsUYDxxsCrsjPeUs7MIy5XlgMXY9_BO5A,777
 frogml_core/automations/__init__.py,sha256=j2gD15MN-xVWhI5rAFsDwhL0CIyICLNT0scXsKvNBkU,1547
 frogml_core/automations/automation_executions.py,sha256=xpOb9Dq8gPPGNQDJTvBBZbNz4woZDRZY0HqnLSu7pwU,3230
 frogml_core/automations/automations.py,sha256=GKEQyQMi8sxX5oZn62PaxPi0zD8IaJRjBkhczRJxHNs,13070
@@ -260,7 +260,7 @@ frogml_core/inner/build_logic/phases/phase_010_fetch_model/set_version_step.py,s
 frogml_core/inner/build_logic/phases/phase_020_remote_register_frogml_build/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 frogml_core/inner/build_logic/phases/phase_020_remote_register_frogml_build/cleanup_step.py,sha256=HmQ0iQ9pJxLZTDzUvwVwwWO2CG-4fHtxwmye-xL0Q9U,632
 frogml_core/inner/build_logic/phases/phase_020_remote_register_frogml_build/start_remote_build_step.py,sha256=UQWLfJpj9lm0fJfKj0Kc2N99x6ioUVfKPXrf1u70rJY,1691
-frogml_core/inner/build_logic/phases/phase_020_remote_register_frogml_build/upload_step.py,sha256=TbNonL_FR0yycEiPqAGYldPD5hY_ooXDXo3JgvMdWC8,9617
+frogml_core/inner/build_logic/phases/phase_020_remote_register_frogml_build/upload_step.py,sha256=bbNSttwy2vEZv9yfE1Ndxf_rv03KP7QuM4Q9rM9BEUI,9350
 frogml_core/inner/build_logic/phases/phases_pipeline.py,sha256=Ei70zbiiT6Be2aFLV7Z5zcVykm2peOsd-UFLG47AR_c,1258
 frogml_core/inner/build_logic/run_handlers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 frogml_core/inner/build_logic/run_handlers/programmatic_phase_run_handler.py,sha256=XE1afXZfA1AQOa9bV2CtmK--gcX0YKRhLwGdkfb3cto,3547
@@ -1082,6 +1082,6 @@ frogml_services_mock/mocks/workspace_manager_service_mock.py,sha256=WbOiWgOyr-xT
 frogml_services_mock/services_mock.py,sha256=_34z6rFCHFwcSni-9eyJlrH264xsL_QDNmG-EBv0zMc,20281
 frogml_services_mock/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 frogml_services_mock/utils/service_utils.py,sha256=ZlB0CnB1J6oBn6_m7fQO2U8tKoboHdUa6ljjkRMYNXU,265
-frogml_core-0.0.71.dist-info/METADATA,sha256=OMEQfxHj-4aK09h3A6A-1ak4g6BkW3H_1-GGyOZXkmE,2004
-frogml_core-0.0.71.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
-frogml_core-0.0.71.dist-info/RECORD,,
+frogml_core-0.0.72.dist-info/METADATA,sha256=F0pOBIedX8mbtzKDvMg0xLciGRTXE-mjzhMMrQAjb3I,2004
+frogml_core-0.0.72.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
+frogml_core-0.0.72.dist-info/RECORD,,