frogml-cli 0.0.1__py3-none-any.whl → 0.0.2__py3-none-any.whl

frogml_cli/__init__.py

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

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/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/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.dist-info → frogml_cli-0.0.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: frogml-cli
-Version: 0.0.1
+Version: 0.0.2
 Summary: Frogml CLI for frogml models
 License: Apache-2.0
 Keywords: mlops,ml,deployment,serving,model,jfrog

{frogml_cli-0.0.1.dist-info → frogml_cli-0.0.2.dist-info}/RECORD

@@ -1,4 +1,4 @@
-frogml_cli/__init__.py,sha256=--BSCElfRpYapX7RjsaLuPosIsEKX_n43lfCZw2BI7s,157
+frogml_cli/__init__.py,sha256=XEZR5icFmwjSToyn9bnCF_RlgfF9KCdCKtbF-afHdwM,157
 frogml_cli/cli.py,sha256=Gh-KPvBf_X243Cz2TGx_mWd4K4NgXcSmpBxZfrFQSkc,1585
 frogml_cli/commands/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 frogml_cli/commands/_logic/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -205,7 +205,7 @@ frogml_cli/commands/models/init/_logic/template/credit_risk/{{cookiecutter.model
 frogml_cli/commands/models/init/_logic/template/credit_risk/{{cookiecutter.model_directory}}/main/data.csv,sha256=Qr47gqLlBzvVyiO84dHDFCa3j3LSDNiS86rKorowoHU,53393
 frogml_cli/commands/models/init/_logic/template/credit_risk/{{cookiecutter.model_directory}}/main/model.py,sha256=L5wmOy9O1ACyZFzB32ZT44fDy5wtGGELL8UkMnvDTj0,4005
 frogml_cli/commands/models/init/_logic/template/general/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-frogml_cli/commands/models/init/_logic/template/general/cookiecutter.json,sha256=dcBfmJgVzlG7z3k-WFLHNiH9fPirs-WU9TVgl4xN20A,139
+frogml_cli/commands/models/init/_logic/template/general/cookiecutter.json,sha256=Ddx20uknTUv-I8AnZ35T9B7zsNHmE6VtoIE0vRerx3k,139
 frogml_cli/commands/models/init/_logic/template/general/{{cookiecutter.model_directory}}/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 frogml_cli/commands/models/init/_logic/template/general/{{cookiecutter.model_directory}}/{{cookiecutter.main_directory}}/__init__.py,sha256=LG2LPM98oRS0v2uYDYs7hI7EIaui79YT--f91BbTqD4,120
 frogml_cli/commands/models/init/_logic/template/general/{{cookiecutter.model_directory}}/{{cookiecutter.main_directory}}/conda.yml,sha256=EDD9m5sTQS2Kg_e6IvxaAMrTN1J8xrdACVD-sVZm5LQ,140
@@ -230,7 +230,7 @@ frogml_cli/commands/models/init/_logic/template/titanic_poetry/{{cookiecutter.mo
 frogml_cli/commands/models/init/_logic/template/titanic_poetry/{{cookiecutter.model_directory}}/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 frogml_cli/commands/models/init/_logic/template/titanic_poetry/{{cookiecutter.model_directory}}/tests/it/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 frogml_cli/commands/models/init/_logic/template/titanic_poetry/{{cookiecutter.model_directory}}/tests/it/test_titanic.py,sha256=A75drbpCH-wNYwGZHS16xFBN7aApYbA7w_G1Hc2BwkU,761
-frogml_cli/commands/models/init/ui.py,sha256=zfg_qyt4G6KM-57Mq5d5jPiBuGju7iVb7t5j4H2sxaM,1925
+frogml_cli/commands/models/init/ui.py,sha256=p2VIxouKmaUraYD_wkdgLPeZmNGVG6p-uxoywz4r9iA,2063
 frogml_cli/commands/models/list/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 frogml_cli/commands/models/list/_logic.py,sha256=6ef3aBkN55b_4Z4VktNSzH6L0jXT2dFWz0LyYwWJ4MA,193
 frogml_cli/commands/models/list/ui.py,sha256=L5FotbJbeweTklDTKIA24_sdLNH3t53JHQFOx-JtBa4,1498
@@ -280,8 +280,8 @@ frogml_cli/tools/colors.py,sha256=7pui_GGjC4uZKYFsIyXaJjYsjLxJVHb4OrfTgr93hqo,28
 frogml_cli/tools/const.py,sha256=XDaMAVfScXoFc5fHbsNuqAefUD2MoZY-cAEMoh1xaYk,152
 frogml_cli/tools/files.py,sha256=AyKJTOy7NhvP3SrqwIw_lxYNCOy1CvLgMmSJpWZ0OKM,2257
 frogml_cli/tools/log_handling.py,sha256=QlgxbCmLLPK4wRyViWgAW8WLZ39Do1hYAvRyFWGbnFs,6362
-frogml_cli/tools/utils.py,sha256=dzv10KyWBl8lLh_snZN3L9SxnrHvk5QSVSLP2l2xQ40,1200
-frogml_cli-0.0.1.dist-info/METADATA,sha256=ls5JKVlleneD-J7HwZf1B9fmQkAmOlsGtXY58XqNg2Q,2202
-frogml_cli-0.0.1.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-frogml_cli-0.0.1.dist-info/entry_points.txt,sha256=2H5x0V_E73HeywIMBRRvZMgeRtmX5820KmSLiNFzDcA,53
-frogml_cli-0.0.1.dist-info/RECORD,,
+frogml_cli/tools/utils.py,sha256=Vliw-w6i-93dgZ3M09EgWjttHMuhT-3CH7tbWVKsf8s,1668
+frogml_cli-0.0.2.dist-info/METADATA,sha256=ZaO3QJi-64jkFgP3WdsFPOUPtur5tTePKxl9eLhWxtc,2202
+frogml_cli-0.0.2.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+frogml_cli-0.0.2.dist-info/entry_points.txt,sha256=2H5x0V_E73HeywIMBRRvZMgeRtmX5820KmSLiNFzDcA,53
+frogml_cli-0.0.2.dist-info/RECORD,,