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

qwak_sdk/__init__.py

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

qwak_sdk/cli.py

@@ -4,6 +4,7 @@ import click
 from packaging import version
 from qwak.inner.di_configuration import UserAccountConfiguration
 from qwak.inner.di_configuration.account import UserAccount
+from qwak.inner.tool.auth import FrogMLAuthClient
 
 from qwak_sdk import __version__ as sdk_version
 from qwak_sdk.commands.admin.admin_commands_group import admin_commands_group
@@ -41,7 +42,32 @@ def create_qwak_cli():
 
     @qwak_cli.command("configure", short_help="Configure the Qwak environment")
     @click.option(
-        "--api-key", metavar="PASSWORD", required=False, help="Qwak assigned API key"
+        "--api-key",
+        metavar="QWAK_API_KEY",
+        required=False,
+        help="Qwak assigned API key",
+    )
+    @click.option(
+        "--url", metavar="BASE_URL", required=False, help="Artifactory base url"
+    )
+    @click.option(
+        "--username", metavar="USERNAME", required=False, help="The user's username"
+    )
+    @click.option(
+        "--password", metavar="PASSWORD", required=False, help="The user's password"
+    )
+    @click.option(
+        "--token",
+        metavar="TOKEN",
+        required=False,
+        help="Access token to authenticate",
+    )
+    @click.option(
+        "--interactive",
+        metavar="INTERACTIVE",
+        required=False,
+        help="Login with interactive flow",
+        default=False,
     )
     @click.option(
         "--environment",
@@ -51,15 +77,46 @@ def create_qwak_cli():
         is_eager=True,
         help="Qwak environment's name",
     )
+    @click.option(
+        "--type",
+        metavar="TYPE",
+        default="qwak",
+        required=False,
+        type=click.Choice(["qwak", "jfrog"], case_sensitive=False),
+        is_eager=True,
+        help="Login type (qwak/jfrog)",
+    )
     @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))
+    def set_configuration(
+        api_key: Optional[str],
+        url: Optional[str],
+        username: Optional[str],
+        password: Optional[str],
+        token: Optional[str],
+        interactive: bool,
+        environment: str,
+        type: str,
+        **_,
+    ):
+        if type.lower() == "jfrog":
+            account_config = UserAccountConfiguration(auth_client=FrogMLAuthClient)
+            account_config.configure_user(
+                UserAccount(
+                    url=url,
+                    username=username,
+                    password=password,
+                    api_key=token,
+                    anonymous=False,
+                    is_interactive=interactive,
+                )
+            )
+        elif type.lower() == "qwak":
+            if api_key is None:
+                api_key = click.prompt("Please enter your API key", type=str)
 
-        print(f"User successfully configured for the '{environment}' environment")
+            account_config = UserAccountConfiguration()
+            account_config.configure_user(UserAccount(api_key=api_key))
+            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/trigger/ui.py

@@ -1,4 +1,7 @@
 import click
+from _qwak_proto.qwak.feature_store.features.feature_set_service_pb2 import (
+    RunBatchFeatureSetResponse,
+)
 from qwak.clients.feature_store import FeatureRegistryClient
 from qwak.exceptions import QwakException
 
@@ -15,10 +18,12 @@ def trigger_feature_set(name, **kwargs):
     Trigger a batch feature set ingestion job
 
     Args:
-        name: feature set name that will be triggered
+        name: Feature set name that will be triggered.
     """
     try:
-        FeatureRegistryClient().run_feature_set(feature_set_name=name)
+        result: RunBatchFeatureSetResponse = FeatureRegistryClient().run_feature_set(
+            feature_set_name=name
+        )
     except Exception as e:
         print(
             f"{Color.RED} Failed to trigger a batch feature set ingestion for feature set {name} {Color.END}"
@@ -28,5 +33,7 @@ def trigger_feature_set(name, **kwargs):
         ) from e
 
     print(
-        f"{Color.GREEN}Successfully triggered a batch feature set ingestion for feature set {Color.YELLOW}{name}"
+        f"{Color.GREEN}Successfully triggered a batch feature set ingestion for feature set: {Color.YELLOW}{name}"
     )
+    if result.execution_id:
+        print(f"{Color.WHITE} Execution ID: {Color.YELLOW}{result.execution_id}")

qwak_sdk/commands/models/build/ui.py

@@ -132,8 +132,8 @@ from qwak_sdk.inner.tools.config_handler import config_handler
 )
 @click.option(
     "--cache/--no-cache",
-    default=None,
-    help="Disable docker build cache. [Default: Cache enabled]",
+    default=False,
+    help="Disable docker build cache. [Default: Cache disabled]",
 )
 @click.option(
     "-v",

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

@@ -1,16 +1,15 @@
 Metadata-Version: 2.1
 Name: qwak-sdk
-Version: 0.5.82
+Version: 0.5.86
 Summary: Qwak SDK and CLI for qwak models
 License: Apache-2.0
 Keywords: mlops,ml,deployment,serving,model
 Author: Qwak
 Author-email: info@qwak.com
-Requires-Python: >=3.7.1,<3.12
+Requires-Python: >=3.9,<3.12
 Classifier: License :: OSI Approved :: Apache Software License
 Classifier: Operating System :: OS Independent
 Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.8
 Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
@@ -41,7 +40,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.105)
+Requires-Dist: qwak-core (==0.4.165)
 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.82.dist-info → qwak_sdk-0.5.86.dist-info}/RECORD

@@ -1,5 +1,5 @@
-qwak_sdk/__init__.py,sha256=5WxdaiKUba1SxlVhvAfhCqKTvIv97ryFMadJWNOh6yw,135
-qwak_sdk/cli.py,sha256=AFUAqablunsHasfY1ikhj9GYUARbQ5dR-sblxqa1id4,2841
+qwak_sdk/__init__.py,sha256=9eYJ9BVyrth4Dyzyk3AYq0iWC7Z31TrbdZLJm9iJbrY,135
+qwak_sdk/cli.py,sha256=frMefF0AmLI1g5A9aADNueUkMDjRHikKlmXIA2i4kYA,4510
 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
@@ -97,7 +97,7 @@ qwak_sdk/commands/feature_store/register/ui.py,sha256=F9bCsV13ZdmzHw470Bfso48Lzj
 qwak_sdk/commands/feature_store/resume/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 qwak_sdk/commands/feature_store/resume/ui.py,sha256=nI87xvA30qNQVJnT67lYJgwKGBtvZAurC6XX9ttpCZA,700
 qwak_sdk/commands/feature_store/trigger/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-qwak_sdk/commands/feature_store/trigger/ui.py,sha256=1VJBqzxMwUWcytmZL0ymf9QlYa7oX6RDLmKsLmYVW_4,1015
+qwak_sdk/commands/feature_store/trigger/ui.py,sha256=NvFfnZn-Idddt5fkO8nWvu-nP8PazN5tAb7vHyRgeck,1298
 qwak_sdk/commands/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 qwak_sdk/commands/models/build/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 qwak_sdk/commands/models/build/_logic/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -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=eQWoaA5zA3KsRUfNMA3XanLboVc2rO50l6FzdDomd1c,9869
+qwak_sdk/commands/models/build/ui.py,sha256=lBuRa-DpEKnHBz6xs2ZGWr6JzZkwGSXDNeK8LDJCM4A,9871
 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.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,,
+qwak_sdk-0.5.86.dist-info/entry_points.txt,sha256=vSl0ELYDyj640oMM57u0AjBP87wtLYxCcGOendhEx80,47
+qwak_sdk-0.5.86.dist-info/WHEEL,sha256=vVCvjcmxuUltf8cYhJ0sJMRDLr1XsPuxEId8YDzbyCY,88
+qwak_sdk-0.5.86.dist-info/METADATA,sha256=LP4ousAyUlpzuorwgdCGPtAVLw9r_a4W53Oc7uVloyM,2926
+qwak_sdk-0.5.86.dist-info/RECORD,,