qwak-sdk 0.5.67__py3-none-any.whl → 0.5.72__py3-none-any.whl

qwak_sdk/__init__.py

@@ -1,6 +1,6 @@
 # fmt: off
 __author__ = '''Qwak.ai'''
-__version__ = '0.5.67'
+__version__ = '0.5.72'
 
 from qwak.inner import wire_dependencies
 

qwak_sdk/commands/models/build/_logic/wait_until_finished.py

@@ -0,0 +1,27 @@
+from logging import Logger
+import time
+from typing import List
+from _qwak_proto.qwak.builds.builds_pb2 import BuildStatus
+from qwak.clients.build_orchestrator import BuildOrchestratorClient
+
+
+def __get_current_status(build_id) -> BuildStatus:
+    return BuildOrchestratorClient().get_build(build_id).build.build_status
+
+
+def __end_state_statuses() -> List[BuildStatus.ValueType]:
+    return [BuildStatus.FAILED, BuildStatus.SUCCESSFUL, BuildStatus.REMOTE_BUILD_CANCELLED, BuildStatus.REMOTE_BUILD_TIMED_OUT]
+
+
+def wait_until_finished(build_id, log: Logger, pool_interval_seconds=10) -> None:
+    status = __get_current_status(build_id)
+    log.info(f"Waiting for build {build_id} to finish. Aborting this process will not stop the build!")
+    log.debug(f"Current status of build {build_id}: {BuildStatus.DESCRIPTOR.values_by_number[status].name}")
+    while status not in __end_state_statuses():
+        time.sleep(pool_interval_seconds)
+        status = __get_current_status(build_id)
+        log.debug(f"Current status of build {build_id}: {BuildStatus.DESCRIPTOR.values_by_number[status].name}")
+
+
+def is_final_status_successful(build_id) -> bool:
+    return __get_current_status(build_id) == BuildStatus.SUCCESSFUL

qwak_sdk/commands/models/build/ui.py

@@ -13,6 +13,7 @@ from qwak.inner.build_logic.run_handlers.programmatic_phase_run_handler import (
 from qwak.inner.const import QwakConstants
 
 from qwak_sdk.commands.models.build._logic.build_steps import create_pipeline
+from qwak_sdk.commands.models.build._logic.wait_until_finished import wait_until_finished, is_final_status_successful
 from qwak_sdk.commands.models.build._logic.client_logs.cli_phase_run_handler import (
     CLIPhaseRunHandler,
 )
@@ -238,6 +239,12 @@ from qwak_sdk.inner.tools.config_handler import config_handler
     default=False,
     is_flag=True,
 )
+@click.option(
+    "--sync",
+    help="Waits until the build is finished (successfully or not) and return the build status",
+    default=False,
+    is_flag=True,
+)
 @click.option(
     "--git-secret-ssh",
     metavar="NAME",
@@ -281,6 +288,7 @@ def build(
         return
     else:
         id_only = kwargs.get("id_only")
+        is_sync = kwargs.get("sync")
         pipeline, success_msg = create_pipeline(config)
         with get_build_logger(config=config, json_logs=json_logs) as (
             logger,
@@ -300,6 +308,12 @@ def build(
                 build_runner,
             )
 
+            if is_sync:
+                wait_until_finished(pipeline.context.build_id, logger)
+                if not is_final_status_successful(pipeline.context.build_id):
+                    click.echo(f"Build {pipeline.context.build_id} failed")
+                    exit(1)
+
             if id_only:
                 print(pipeline.context.build_id)
             else:

{qwak_sdk-0.5.67.dist-info → qwak_sdk-0.5.72.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: qwak-sdk
-Version: 0.5.67
+Version: 0.5.72
 Summary: Qwak SDK and CLI for qwak models
 License: Apache-2.0
 Keywords: mlops,ml,deployment,serving,model
@@ -37,7 +37,7 @@ Requires-Dist: pandas (>1.1.3) ; (python_version >= "3.9" and python_version < "
 Requires-Dist: pandas (>=1.4.0) ; (python_version >= "3.10" and python_version < "3.12") and (extra == "batch" or extra == "feedback")
 Requires-Dist: pyarrow (>=6.0.0,<11.0.0) ; extra == "batch"
 Requires-Dist: python-json-logger (>=2.0.2)
-Requires-Dist: qwak-core (==0.4.8)
+Requires-Dist: qwak-core (==0.4.38)
 Requires-Dist: qwak-inference (>=0.1.16,<0.2.0)
 Requires-Dist: rich (>=13.0.0)
 Requires-Dist: tabulate (>=0.8.0)

{qwak_sdk-0.5.67.dist-info → qwak_sdk-0.5.72.dist-info}/RECORD

@@ -1,4 +1,4 @@
-qwak_sdk/__init__.py,sha256=NNZZ1nm59YD8isM6cCdmZBmVnhSLrubQOw8xfd9VXpc,135
+qwak_sdk/__init__.py,sha256=6Bs9kviutNNAWsdI8ccThcuyIU588h-Fgapvs4KJYm8,135
 qwak_sdk/cli.py,sha256=FIK1dUNxR57ypb-CeD7fKSJnPJ02lrjR9G4aj2qMLPU,2458
 qwak_sdk/commands/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 qwak_sdk/commands/_logic/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -121,7 +121,8 @@ qwak_sdk/commands/models/build/_logic/util/__init__.py,sha256=47DEQpj8HBSa-_TImW
 qwak_sdk/commands/models/build/_logic/util/protobuf_factory.py,sha256=ar_oY38w_x0sxgVF7EBs5h7gchNsDntvtKK5sSYxb24,1686
 qwak_sdk/commands/models/build/_logic/util/step_decorator.py,sha256=HLZyCGdqe3Ir7SaPWp1YNRHJpjXG-e-bbAvnOFysAVM,1913
 qwak_sdk/commands/models/build/_logic/util/text.py,sha256=tH-v19Mt8l90sMVxku5XRtrderT0qdRqJ-jLijqannA,188
-qwak_sdk/commands/models/build/ui.py,sha256=qr-brINdA6wF9QVgFjE77qGi3KIYckHTrPQlr-d5jyo,9164
+qwak_sdk/commands/models/build/_logic/wait_until_finished.py,sha256=DxIyNK-MFjxSh9xe7dJx-znmz8ZOqelK-cJQvr7YI9g,1220
+qwak_sdk/commands/models/build/ui.py,sha256=ZfOFzZxMT3bSUz42dn9LF7Pept9Ed9afWu0-_0HES_w,9761
 qwak_sdk/commands/models/builds/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 qwak_sdk/commands/models/builds/builds_commands_group.py,sha256=0nSfTY8TracXG61rFboQWUTXJisHO6dgtJKeijy6ru8,491
 qwak_sdk/commands/models/builds/cancel/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -318,7 +319,7 @@ qwak_sdk/tools/colors.py,sha256=7pui_GGjC4uZKYFsIyXaJjYsjLxJVHb4OrfTgr93hqo,287
 qwak_sdk/tools/files.py,sha256=AyKJTOy7NhvP3SrqwIw_lxYNCOy1CvLgMmSJpWZ0OKM,2257
 qwak_sdk/tools/log_handling.py,sha256=Aa1EmxUPCX8YWiZRutUvnqPv6K_z1zoGMwIWsEv24mM,6327
 qwak_sdk/tools/utils.py,sha256=SHmU4r_m2ABZyFYMC03P17GvltPbYbmB39hvalIZEtI,1168
-qwak_sdk-0.5.67.dist-info/entry_points.txt,sha256=vSl0ELYDyj640oMM57u0AjBP87wtLYxCcGOendhEx80,47
-qwak_sdk-0.5.67.dist-info/WHEEL,sha256=vVCvjcmxuUltf8cYhJ0sJMRDLr1XsPuxEId8YDzbyCY,88
-qwak_sdk-0.5.67.dist-info/METADATA,sha256=xfnTJi0rGKt4gDYgthZkCDE43GNuPOkdz8KYgS3hvEM,2599
-qwak_sdk-0.5.67.dist-info/RECORD,,
+qwak_sdk-0.5.72.dist-info/entry_points.txt,sha256=vSl0ELYDyj640oMM57u0AjBP87wtLYxCcGOendhEx80,47
+qwak_sdk-0.5.72.dist-info/WHEEL,sha256=vVCvjcmxuUltf8cYhJ0sJMRDLr1XsPuxEId8YDzbyCY,88
+qwak_sdk-0.5.72.dist-info/METADATA,sha256=5AJVTJoHD3g69f6-EFQqadYu36k7-_ZgKDOax8nxE4A,2600
+qwak_sdk-0.5.72.dist-info/RECORD,,