qwak-core 0.4.245__py3-none-any.whl → 0.4.246__py3-none-any.whl

qwak/__init__.py

@@ -1,7 +1,7 @@
 """Top-level package for qwak-core."""
 
 __author__ = "Qwak.ai"
-__version__ = "0.4.245"
+__version__ = "0.4.246"
 
 from qwak.inner.di_configuration import wire_dependencies
 from qwak.model.experiment_tracking import log_metric, log_param

qwak/inner/build_logic/phases/phase_020_remote_register_qwak_build/upload_step.py

@@ -3,15 +3,14 @@ from __future__ import annotations
 import re
 from pathlib import Path
 from typing import Any, List, Tuple, Union, Optional
-from retrying import retry
 
 import joblib
 import requests
+from retrying import retry
 
 from _qwak_proto.qwak.builds.build_url_pb2 import BuildVersioningTagsType
 from _qwak_proto.qwak.builds.builds_orchestrator_service_pb2 import (
     GetBuildVersioningUploadURLResponse,
-    AuthenticationDetail,
 )
 from qwak.exceptions import QwakException
 from qwak.exceptions import QwakGeneralBuildException
@@ -158,15 +157,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")
@@ -179,7 +180,7 @@ class UploadStep(Step):
         try:
             self.build_logger.debug(f"Getting pre-signed url for upload - tag {tag}")
 
-            pre_signed_url_response = (
+            pre_signed_url_response: GetBuildVersioningUploadURLResponse = (
                 self.context.client_builds_orchestrator.get_build_versioning_upload_url(
                     build_id=self.context.build_id,
                     model_id=self.context.model_id,
@@ -203,22 +204,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"
@@ -236,8 +231,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(
@@ -247,8 +248,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]:

{qwak_core-0.4.245.dist-info → qwak_core-0.4.246.dist-info}/METADATA

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

{qwak_core-0.4.245.dist-info → qwak_core-0.4.246.dist-info}/RECORD

@@ -576,7 +576,7 @@ _qwak_proto/qwak/workspace/workspace_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXH
 _qwak_proto/qwak/workspace/workspace_service_pb2.py,sha256=AB3C9S_AbOD7Nx1Ni4j1rW6PNtYTV1zjiqFQk-goQ74,21429
 _qwak_proto/qwak/workspace/workspace_service_pb2.pyi,sha256=nKKCHwnovZhsy8TSVmdz-Vtl0nviOOoX56HD-41Xo08,13726
 _qwak_proto/qwak/workspace/workspace_service_pb2_grpc.py,sha256=yKGuexxTBza99Ihe0DSTniV2ZSd_AG47inHenqfi890,27193
-qwak/__init__.py,sha256=NY9pmuaNHtKMgBcn4hujiqzSsH5CwkeNGVui3rFInyI,587
+qwak/__init__.py,sha256=NI8Z0miVc9RaYzA7QH7AlrhoEjQOmUcaFlF7EcOmxO0,587
 qwak/automations/__init__.py,sha256=qFZRvCxUUn8gcxkJR0v19ulHW2oJ0x6-Rif7HiheDP4,1522
 qwak/automations/automation_executions.py,sha256=5MeH_epYYWb8NKXgAozwT_jPyyUDednBHG7izloi7RY,3228
 qwak/automations/automations.py,sha256=3yx8e2v0uSKDnXbqyknasyEoQ5vxGni6K40Hbi1_zkk,12599
@@ -810,7 +810,7 @@ qwak/inner/build_logic/phases/phase_010_fetch_model/set_version_step.py,sha256=M
 qwak/inner/build_logic/phases/phase_020_remote_register_qwak_build/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 qwak/inner/build_logic/phases/phase_020_remote_register_qwak_build/cleanup_step.py,sha256=Nxv3wAbuCocvsC1V7-gGSTRLOEKdrlViME4_BUTedCY,618
 qwak/inner/build_logic/phases/phase_020_remote_register_qwak_build/start_remote_build_step.py,sha256=3k8wLGh-tANo0AYJstPHIACpSul-h7gYbxGgFQZzRZQ,1673
-qwak/inner/build_logic/phases/phase_020_remote_register_qwak_build/upload_step.py,sha256=Ncv4-Qx3co1mvSf-bG5npFybwDw7FHhYvqLyrx6EgSI,9553
+qwak/inner/build_logic/phases/phase_020_remote_register_qwak_build/upload_step.py,sha256=tj42f3h1uFl4FYiwp0uBxfRVqFyiGQLEIJ4iz1M_IJc,9323
 qwak/inner/build_logic/phases/phases_pipeline.py,sha256=KveRNEYji1FbzP1tswJWReMqoCnuHfDLmYFeS4YBJpE,1244
 qwak/inner/build_logic/run_handlers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 qwak/inner/build_logic/run_handlers/programmatic_phase_run_handler.py,sha256=nDR0e357sc-Lr-36uCUzxCK3LORzqYyoutHQ3rmXo1w,3484
@@ -1052,6 +1052,6 @@ qwak_services_mock/mocks/workspace_manager_service_mock.py,sha256=O9ZSwln4T4kHVk
 qwak_services_mock/services_mock.py,sha256=CABZ2teXoSJ-X7CD3iRo5ht-w3jhW-Cvpb47APUHXEA,19089
 qwak_services_mock/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 qwak_services_mock/utils/service_utils.py,sha256=ZlB0CnB1J6oBn6_m7fQO2U8tKoboHdUa6ljjkRMYNXU,265
-qwak_core-0.4.245.dist-info/METADATA,sha256=LfCbNAOmCd87wxO5GQBw35GQvP_ctyHLr4jBkfKxEfw,2150
-qwak_core-0.4.245.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
-qwak_core-0.4.245.dist-info/RECORD,,
+qwak_core-0.4.246.dist-info/METADATA,sha256=HOEV8pRohcWvk7huZnH3R9NpzwIT5pLppjfhfr2bc2I,2150
+qwak_core-0.4.246.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
+qwak_core-0.4.246.dist-info/RECORD,,