frogml-cli 0.0.1__tar.gz → 0.0.3__tar.gz

{frogml_cli-0.0.1 → frogml_cli-0.0.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: frogml-cli
-Version: 0.0.1
+Version: 0.0.3
 Summary: Frogml CLI for frogml models
 License: Apache-2.0
 Keywords: mlops,ml,deployment,serving,model,jfrog
@@ -27,8 +27,8 @@ Requires-Dist: click (==8.1.8)
 Requires-Dist: cloudpickle (==2.2.1) ; extra == "feature-store"
 Requires-Dist: cookiecutter
 Requires-Dist: croniter (==1.4.1)
-Requires-Dist: frogml (>=1.1.24)
-Requires-Dist: frogml-inference (>=0.0.1,<0.0.2)
+Requires-Dist: frogml (>=1.1.61)
+Requires-Dist: frogml-inference (>=0.0.2,<0.0.3)
 Requires-Dist: gitpython (>=2.1.0)
 Requires-Dist: joblib (>=1.1.0,<2.0.0) ; extra == "batch" or extra == "feedback"
 Requires-Dist: numpy (<2.0.0)

{frogml_cli-0.0.1 → frogml_cli-0.0.3}/frogml_cli/__init__.py

@@ -1,6 +1,6 @@
 # fmt: off
 __author__ = '''JFrog Ltd.'''
-__version__ = '0.0.1'
+__version__ = '0.0.3'
 __name__ = 'frogml_cli'
 
 from frogml import wire_dependencies

{frogml_cli-0.0.1 → frogml_cli-0.0.3}/frogml_cli/commands/models/build/_logic/phase/c_deploy/deploy_build.py

@@ -35,7 +35,7 @@ class DeployBuildStep(Step):
                 "model_id": self.context.model_id,
                 "instance": template_id,
             }
-            deploy_realtime(from_file=None, out_conf=False, sync=False, **deploy_config)
+            deploy_realtime(from_file=None, out_conf=False, local=False, sync=False, **deploy_config)
             self.build_logger.info(f"Finished deploying build {self.context.build_id}")
         except Exception as e:
             raise FrogmlDeployNewBuildFailedException(

{frogml_cli-0.0.1 → frogml_cli-0.0.3}/frogml_cli/commands/models/build/ui.py

@@ -206,8 +206,8 @@ from frogml_cli.tools.const import GPU_TYPES
 @click.option(
     "--deploy",
     help="Whether you want to deploy the build if it finishes successfully. "
-    "Choosing this will follow the build process in the terminal and will trigger a deployment when the "
-    "build finishes.",
+    "Choosing this will follow the build process in the terminal and will trigger a deployment in JFrogML's platform "
+    "when the build finishes.",
     default=False,
     is_flag=True,
 )

{frogml_cli-0.0.1 → frogml_cli-0.0.3}/frogml_cli/commands/models/create/ui.py

@@ -12,7 +12,7 @@ from frogml_cli.tools.colors import Color
     "--project-key",
     metavar="ID",
     required=False,
-    help="An existing jfrog project key",
+    help="An existing jfrog project key (3-44 characters)",
 )
 @click.option(
     "--description",

{frogml_cli-0.0.1 → frogml_cli-0.0.3}/frogml_cli/commands/models/init/_logic/template/general/cookiecutter.json

@@ -2,5 +2,5 @@
     "model_directory": "model",
     "main_directory": "main",
     "test_directory": "tests",
-    "model_class_name": "modelPrediction"
+    "model_class_name": "ModelPrediction"
 }

{frogml_cli-0.0.1 → frogml_cli-0.0.3}/frogml_cli/commands/models/init/ui.py

@@ -1,3 +1,5 @@
+from typing import List
+
 import click
 from frogml.core.tools.logger.logger import get_frogml_logger
 
@@ -5,10 +7,14 @@ from frogml_cli.commands.models.init._logic.initialize_model_structure import (
     initialize_model_structure,
 )
 from frogml_cli.inner.tools.cli_tools import FrogMLCommand
+from frogml_cli.tools.utils import get_models_init_example_choices
 
 logger = get_frogml_logger()
 
 
+MODELS_INIT_EXAMPLE_CHOICES: List[str] = get_models_init_example_choices()
+
+
 @click.command("init", cls=FrogMLCommand)
 @click.option(
     "--model-directory", metavar="NAME", required=False, help="folder for model content"
@@ -23,10 +29,8 @@ logger = get_frogml_logger()
     "--example",
     metavar="NAME",
     required=False,
-    type=click.Choice(
-        ["titanic", "credit_risk", "churn", "titanic_poetry"], case_sensitive=True
-    ),
-    help="""Generate a fully functioning example of a Qwak based model. Options: titanic / credit_risk / churn""",
+    type=click.Choice(MODELS_INIT_EXAMPLE_CHOICES, case_sensitive=True),
+    help=f"Generate a fully functioning example of a Qwak based model. Options: {' / '.join(MODELS_INIT_EXAMPLE_CHOICES)}",
 )
 @click.argument("uri", metavar="URI", required=True)
 def model_init(

{frogml_cli-0.0.1 → frogml_cli-0.0.3}/frogml_cli/tools/utils.py

@@ -1,7 +1,8 @@
 import logging
 import sys
 from contextlib import contextmanager
-from typing import Callable, Optional
+from pathlib import Path
+from typing import Callable, List, Optional
 
 from frogml.core.tools.logger.logger import (
     get_frogml_logger,
@@ -43,3 +44,21 @@ def frogml_spinner(
             if end_text:
                 sp.text = end_text
             sp.ok("✅")
+
+
+def get_models_init_example_choices() -> List[str]:
+    """Dynamically get available template choices from the template directory."""
+    template_dir = (
+        Path(__file__).parent.parent
+        / "commands"
+        / "models"
+        / "init"
+        / "_logic"
+        / "template"
+    )
+
+    return [
+        item.name
+        for item in template_dir.iterdir()
+        if item.is_dir() and not item.name.startswith(".")
+    ]

{frogml_cli-0.0.1 → frogml_cli-0.0.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "frogml-cli"
-version = "0.0.1"
+version = "0.0.3"
 description = "Frogml CLI for frogml models"
 authors = ["JFrog Ltd. <contact@jfrog.com>"]
 readme = "README.md"
@@ -48,8 +48,8 @@ cloudpickle = {version = "2.2.1", optional = true} # feature-store dependencies
 pyspark = {version = "3.4.2", optional = true} # feature-store dependencies
 click = "8.1.8"
 catboost = ">=1.1.11"
-frogml-inference = "^0.0.1"
-frogml = ">=1.1.24"
+frogml-inference = "^0.0.2"
+frogml = ">=1.1.61"
 
 [tool.poetry.group.dev.dependencies]
 pytest = "7.2.1"