qwak-sdk 0.5.80__py3-none-any.whl → 0.5.82__py3-none-any.whl

qwak_sdk/__init__.py

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

qwak_sdk/cli.py

@@ -1,3 +1,5 @@
+from typing import Optional
+
 import click
 from packaging import version
 from qwak.inner.di_configuration import UserAccountConfiguration
@@ -19,6 +21,7 @@ from qwak_sdk.commands.secrets.secrets_commands_group import secrets_commands_gr
 from qwak_sdk.commands.workspaces.workspaces_commands_group import (
     workspaces_commands_group,
 )
+from qwak_sdk.inner.tools.cli_tools import profile_setter_wrapper
 from qwak_sdk.inner.tools.logger import setup_qwak_logger
 
 version_option_kwargs = {}
@@ -40,14 +43,23 @@ def create_qwak_cli():
     @click.option(
         "--api-key", metavar="PASSWORD", required=False, help="Qwak assigned API key"
     )
-    def set_configuration(api_key, **kwargs):
+    @click.option(
+        "--environment",
+        metavar="ENVIRONMENT",
+        default="default",
+        required=False,
+        is_eager=True,
+        help="Qwak environment's name",
+    )
+    @profile_setter_wrapper
+    def set_configuration(api_key: Optional[str], environment: str, **_):
         if api_key is None:
             api_key = click.prompt("Please enter your API key", type=str)
 
         account_config = UserAccountConfiguration()
         account_config.configure_user(UserAccount(api_key=api_key))
 
-        print("User successfully configured")
+        print(f"User successfully configured for the '{environment}' environment")
 
     qwak_cli.add_command(projects_command_group)
     qwak_cli.add_command(models_command_group)

qwak_sdk/commands/feature_store/register/_logic.py

@@ -1,3 +1,4 @@
+from pathlib import Path
 from typing import List, Optional, Tuple, cast
 
 from _qwak_proto.qwak.feature_store.entities.entity_pb2 import (
@@ -89,7 +90,9 @@ def _register_data_sources(
     with qwak_spinner(
         begin_text="Finding Data Sources to register", print_callback=print
     ):
-        qwak_sources = extract_class_objects(qwak_python_files, BaseSource)
+        qwak_sources: List[Tuple[BaseSource, str]] = extract_class_objects(
+            qwak_python_files, BaseSource
+        )
 
     print(f"👀 Found {len(qwak_sources)} Data Sources")
     for data_source, source_file_path in qwak_sources:
@@ -115,7 +118,10 @@ def _register_data_sources(
                 f"Update existing Data Source '{data_source.name}' from source file '{source_file_path}'?",
                 force,
             ):
-                data_source_proto, _ = data_source._prepare_and_get(artifact_url)
+                data_source_proto, _ = data_source._prepare_and_get(
+                    artifact_url=artifact_url,
+                    source_definition_path=Path(source_file_path),
+                )
                 registry.update_data_source(
                     existing_source.data_source.data_source_definition.data_source_id,
                     data_source_proto,
@@ -125,7 +131,10 @@ def _register_data_sources(
                 f"Create Data Source '{data_source.name}' from source file '{source_file_path}'?",
                 force,
             ):
-                data_source_proto, _ = data_source._prepare_and_get(artifact_url)
+                data_source_proto, _ = data_source._prepare_and_get(
+                    artifact_url=artifact_url,
+                    source_definition_path=Path(source_file_path),
+                )
                 registry.create_data_source(data_source_proto)
     print(DELIMITER)
 

qwak_sdk/commands/models/build/ui.py

@@ -264,6 +264,13 @@ from qwak_sdk.inner.tools.config_handler import config_handler
     help="Timeout in minutes for the provision instance step",
     default=120,
 )
+@click.option(
+    "-N",
+    "--name",
+    "build_name",
+    required=False,
+    help="The build's name",
+)
 @click.argument("uri", required=False)
 def models_build(**kwargs):
     return build(**kwargs)

{qwak_sdk-0.5.80.dist-info → qwak_sdk-0.5.82.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: qwak-sdk
-Version: 0.5.80
+Version: 0.5.82
 Summary: Qwak SDK and CLI for qwak models
 License: Apache-2.0
 Keywords: mlops,ml,deployment,serving,model
@@ -41,7 +41,7 @@ Requires-Dist: pyarrow (>=6.0.0,<11.0.0) ; extra == "batch" or extra == "feature
 Requires-Dist: pyathena (>=2.2.0,!=2.18.0) ; extra == "feature-store"
 Requires-Dist: pyspark (==3.4.2) ; extra == "feature-store"
 Requires-Dist: python-json-logger (>=2.0.2)
-Requires-Dist: qwak-core (==0.4.85)
+Requires-Dist: qwak-core (==0.4.105)
 Requires-Dist: qwak-inference (>=0.1.18,<0.2.0)
 Requires-Dist: rich (>=13.0.0)
 Requires-Dist: tabulate (>=0.8.0)

{qwak_sdk-0.5.80.dist-info → qwak_sdk-0.5.82.dist-info}/RECORD

@@ -1,5 +1,5 @@
-qwak_sdk/__init__.py,sha256=rkSR8F1Hley9NFLHspeV9WmPLZL6jXsWklMvicZsowc,135
-qwak_sdk/cli.py,sha256=FIK1dUNxR57ypb-CeD7fKSJnPJ02lrjR9G4aj2qMLPU,2458
+qwak_sdk/__init__.py,sha256=5WxdaiKUba1SxlVhvAfhCqKTvIv97ryFMadJWNOh6yw,135
+qwak_sdk/cli.py,sha256=AFUAqablunsHasfY1ikhj9GYUARbQ5dR-sblxqa1id4,2841
 qwak_sdk/commands/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 qwak_sdk/commands/_logic/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 qwak_sdk/commands/_logic/tools.py,sha256=uJOSfpifONXjFJjaHvm_7ivTs8K0PO56YoxhU0dktPU,239
@@ -92,7 +92,7 @@ qwak_sdk/commands/feature_store/list/ui.py,sha256=NuXnQ3cdXSDCFAcC2jmgx4sAdjNuyI
 qwak_sdk/commands/feature_store/pause/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 qwak_sdk/commands/feature_store/pause/ui.py,sha256=C89yl4ly-RpyZub30q5xV5g1UDnI_GmHSndvXAkXkw4,695
 qwak_sdk/commands/feature_store/register/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-qwak_sdk/commands/feature_store/register/_logic.py,sha256=xCMHnyTYA4UbPm0sxiJ-j1bej00YVHZfUAgJMdv1K9s,13153
+qwak_sdk/commands/feature_store/register/_logic.py,sha256=C_clkccTr36fUAYrjOG6-4m0qz4sc2nx351qdHtFbRI,13468
 qwak_sdk/commands/feature_store/register/ui.py,sha256=F9bCsV13ZdmzHw470Bfso48LzjABLmYx5p8sISJNPh0,2997
 qwak_sdk/commands/feature_store/resume/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 qwak_sdk/commands/feature_store/resume/ui.py,sha256=nI87xvA30qNQVJnT67lYJgwKGBtvZAurC6XX9ttpCZA,700
@@ -122,7 +122,7 @@ qwak_sdk/commands/models/build/_logic/util/protobuf_factory.py,sha256=ar_oY38w_x
 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/_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/build/ui.py,sha256=eQWoaA5zA3KsRUfNMA3XanLboVc2rO50l6FzdDomd1c,9869
 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
@@ -319,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.80.dist-info/entry_points.txt,sha256=vSl0ELYDyj640oMM57u0AjBP87wtLYxCcGOendhEx80,47
-qwak_sdk-0.5.80.dist-info/WHEEL,sha256=vVCvjcmxuUltf8cYhJ0sJMRDLr1XsPuxEId8YDzbyCY,88
-qwak_sdk-0.5.80.dist-info/METADATA,sha256=ITCJuku9CrL4QB93Ntg1DU0igReI3p-QCzf89i1GktI,2977
-qwak_sdk-0.5.80.dist-info/RECORD,,
+qwak_sdk-0.5.82.dist-info/entry_points.txt,sha256=vSl0ELYDyj640oMM57u0AjBP87wtLYxCcGOendhEx80,47
+qwak_sdk-0.5.82.dist-info/WHEEL,sha256=vVCvjcmxuUltf8cYhJ0sJMRDLr1XsPuxEId8YDzbyCY,88
+qwak_sdk-0.5.82.dist-info/METADATA,sha256=v3Tkqf9G6556gBOX_jfQCW4ygXdgSb0N7D2S8Liw4fE,2978
+qwak_sdk-0.5.82.dist-info/RECORD,,