qwak-sdk 0.5.67__tar.gz → 0.5.69__tar.gz

{qwak_sdk-0.5.67 → qwak_sdk-0.5.69}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: qwak-sdk
-Version: 0.5.67
+Version: 0.5.69
 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.17)
 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 → qwak_sdk-0.5.69}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "qwak-sdk"
-version = "0.5.67"
+version = "0.5.69"
 description = "Qwak SDK and CLI for qwak models"
 authors = ["Qwak <info@qwak.com>"]
 readme = "README.md"
@@ -29,7 +29,7 @@ qwak = "qwak_sdk.main:qwak_cli"
 
 [tool.poetry.dependencies]
 python = ">=3.7.1,<3.12"
-qwak-core = "==0.4.8"
+qwak-core = "==0.4.17"
 qwak-inference = "~0.1.16"
 croniter = "==1.4.1"
 tabulate = ">=0.8.0"

{qwak_sdk-0.5.67 → qwak_sdk-0.5.69}/qwak_sdk/__init__.py

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

qwak_sdk-0.5.69/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-0.5.67 → qwak_sdk-0.5.69}/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 → qwak_sdk-0.5.69}/setup.py

@@ -134,7 +134,7 @@ install_requires = \
  'croniter==1.4.1',
  'gitpython>=2.1.0',
  'python-json-logger>=2.0.2',
- 'qwak-core==0.4.8',
+ 'qwak-core==0.4.17',
  'qwak-inference>=0.1.16,<0.2.0',
  'rich>=13.0.0',
  'tabulate>=0.8.0',
@@ -184,7 +184,7 @@ entry_points = \
 
 setup_kwargs = {
     'name': 'qwak-sdk',
-    'version': '0.5.67',
+    'version': '0.5.69',
     'description': 'Qwak SDK and CLI for qwak models',
     'long_description': '# Qwak SDK\n\nQwak is an end-to-end production ML platform designed to allow data scientists to build, deploy, and monitor their models in production with minimal engineering friction.\n',
     'author': 'Qwak',