sima-cli 0.0.13__py3-none-any.whl → 0.0.15__py3-none-any.whl

sima_cli/__version__.py

@@ -1,2 +1,2 @@
 # sima_cli/__version__.py
-__version__ = "0.0.13"
+__version__ = "0.0.15"

sima_cli/auth/basic_auth.py

@@ -0,0 +1,132 @@
+import os
+import click
+import getpass
+import requests
+import json
+from http.cookiejar import MozillaCookieJar
+
+HOME_DIR = os.path.expanduser("~/.sima-cli")
+COOKIE_JAR_PATH = os.path.join(HOME_DIR, ".sima-cli-cookies.txt")
+CSRF_PATH = os.path.join(HOME_DIR, ".sima-cli-csrf.json")
+
+CSRF_URL = "https://developer.sima.ai/session/csrf"
+LOGIN_URL = "https://developer.sima.ai/session"
+DUMMY_CHECK_URL = "https://docs.sima.ai/pkg_downloads/dummy"
+
+HEADERS = {
+    "User-Agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/137.0.0.0 Safari/537.36",
+    "X-Requested-With": "XMLHttpRequest",
+    "Referer": "https://developer.sima.ai/login",
+    "Origin": "https://developer.sima.ai",
+    "Content-Type": "application/x-www-form-urlencoded; charset=UTF-8",
+    "sec-ch-ua": '"Google Chrome";v="137", "Chromium";v="137", "Not/A)Brand";v="24"',
+    "sec-ch-ua-mobile": "?0",
+    "sec-ch-ua-platform": '"macOS"',
+}
+
+
+def _is_session_valid(session: requests.Session) -> bool:
+    try:
+        response = session.get(DUMMY_CHECK_URL, allow_redirects=False)
+        return response.status_code == 200
+    except Exception:
+        return False
+
+def _delete_auth_files():
+    for path in [COOKIE_JAR_PATH, CSRF_PATH]:
+        if os.path.exists(path):
+            try:
+                os.remove(path)
+            except Exception as e:
+                click.echo(f"⚠️ Could not delete {path}: {e}")
+
+
+def _save_cookie_jar(session: requests.Session):
+    cj = MozillaCookieJar(COOKIE_JAR_PATH)
+    for c in session.cookies:
+        cj.set_cookie(c)
+    cj.save(ignore_discard=True)
+
+
+def _load_cookie_jar(session: requests.Session):
+    if os.path.exists(COOKIE_JAR_PATH):
+        cj = MozillaCookieJar()
+        cj.load(COOKIE_JAR_PATH, ignore_discard=True)
+        session.cookies.update(cj)
+
+
+def _load_csrf_token() -> str:
+    if os.path.exists(CSRF_PATH):
+        with open(CSRF_PATH, "r") as f:
+            data = json.load(f)
+            return data.get("csrf", "")
+    return ""
+
+
+def _fetch_and_store_csrf_token(session: requests.Session) -> str:
+    try:
+        resp = session.get(CSRF_URL)
+        resp.raise_for_status()
+        csrf_token = resp.json().get("csrf")
+        if csrf_token:
+            with open(CSRF_PATH, "w") as f:
+                json.dump({"csrf": csrf_token}, f)
+        return csrf_token
+    except Exception as e:
+        click.echo(f"❌ Failed to fetch CSRF token: {e}")
+        return ""
+
+
+def login_external():
+    """Interactive login workflow with CSRF token, cookie caching, and dummy session validation."""
+    for attempt in range(1, 4):
+        session = requests.Session()
+        session.headers.update(HEADERS)
+
+        _load_cookie_jar(session)
+        csrf_token = _load_csrf_token()
+
+        if not csrf_token:
+            csrf_token = _fetch_and_store_csrf_token(session)
+
+        if not csrf_token:
+            click.echo("❌ CSRF token is missing or invalid.")
+            continue
+
+        session.headers["X-CSRF-Token"] = csrf_token
+
+        if _is_session_valid(session):
+            click.echo("🚀 Reusing existing session.")
+            return session
+
+        # Prompt user login
+        _delete_auth_files()
+        click.echo(f"🔐 Login attempt {attempt}/3")
+        username = click.prompt("Email or Username")
+        password = getpass.getpass("Password: ")
+
+        login_data = {
+            "login": username,
+            "password": password,
+            "second_factor_method": "1"
+        }
+
+        try:
+            resp = session.post(LOGIN_URL, data=login_data)
+            name = resp.json().get('users')[0]['name'] if 'users' in resp.json() else ''
+            if resp.status_code != 200:
+                click.echo(f"⚠️ Login request returned status {resp.status_code}")
+                continue
+        except Exception as e:
+            click.echo(f"❌ Login request failed: {e}")
+            continue
+
+        if _is_session_valid(session):
+            _save_cookie_jar(session)
+            click.echo(f"✅ Login successful. Welcome to Sima Developer Portal, {name}!")
+            return session
+        else:
+            click.echo("❌ Login failed.")
+
+    click.echo("❌ Login failed after 3 attempts.")
+    raise SystemExit(1)

sima_cli/auth/login.py

@@ -2,8 +2,9 @@ import click
 import getpass
 import requests
 from sima_cli.utils.config import set_auth_token, get_auth_token
-from sima_cli.utils.config_loader import load_resource_config
+from sima_cli.utils.config_loader import load_resource_config, artifactory_url
 from sima_cli.utils.artifactory import exchange_identity_token, validate_token
+from sima_cli.auth.basic_auth import login_external
 
 def login(method: str = "external"):
     """
@@ -30,7 +31,7 @@ def login_internal():
 
     cfg = load_resource_config()
     auth_cfg = cfg.get("internal", {}).get("auth", {})
-    base_url = cfg.get("internal", {}).get("artifactory", {}).get("url", {})
+    base_url = artifactory_url()
     validate_url = auth_cfg.get("validate_url")
     internal_url = auth_cfg.get("internal_url")
     validate_url = f"{base_url}/{validate_url}"
@@ -67,13 +68,14 @@ def login_internal():
     click.echo(f"💾 Short-lived access token saved successfully for {user_name} (valid for 7 days).")
 
 
-def login_external():
+def _login_external():
     """
     External login using Developer Portal endpoint defined in the 'public' section of YAML config.
     Prompts for username/password and retrieves access token.
     """
+    
     cfg = load_resource_config()
-    auth_url = cfg.get("public", {}).get("auth", {}).get("external_url")
+    auth_url = cfg.get("public", {}).get("auth", {}).get("auth_url")
 
     if not auth_url:
         click.echo("❌ External auth URL not configured in YAML.")

sima_cli/cli.py

@@ -2,8 +2,9 @@ import os
 import click
 from sima_cli.utils.env import get_environment_type
 from sima_cli.update.updater import perform_update
-from sima_cli.model_zoo.model import list_models, download_model
+from sima_cli.model_zoo.model import list_models, download_model, describe_model
 from sima_cli.utils.config_loader import internal_resource_exists
+from sima_cli.mla.meminfo import monitor_simaai_mem_chart
 
 # Entry point for the CLI tool using Click's command group decorator
 @click.group()
@@ -29,7 +30,6 @@ def main(ctx, internal):
 
     ctx.obj["internal"] = internal
 
-    from sima_cli.utils.env import get_environment_type
     env_type, env_subtype = get_environment_type()
 
     if internal:
@@ -37,10 +37,6 @@ def main(ctx, internal):
     else:
         click.echo(f"🔧 Environment: {env_type} ({env_subtype})")
 
-    if not internal:
-        click.echo(f"external environment is not supported yet..")
-        exit(0)
-
 # ----------------------
 # Authentication Command
 # ----------------------
@@ -92,20 +88,14 @@ def download(ctx, url, dest):
 @main.command(name="update")
 @click.argument('version_or_url')
 @click.option('--ip', help="Target device IP address for remote firmware update.")
-@click.option(
-    '--board',
-    default='davinci',
-    type=click.Choice(['davinci', 'modalix'], case_sensitive=False),
-    show_default=True,
-    help="Target board type (davinci or modalix)."
-)
+
 @click.option(
     '--passwd',
     default='edgeai',
     help="Optional SSH password for remote board (default is 'edgeai')."
 )
 @click.pass_context
-def update(ctx, version_or_url, ip, board, passwd):
+def update(ctx, version_or_url, ip, passwd):
     """
     Run system update across different environments.
     Downloads and applies firmware updates for PCIe host or SiMa board.
@@ -113,7 +103,11 @@ def update(ctx, version_or_url, ip, board, passwd):
     version_or_url: The version string (e.g. '1.5.0') or a direct URL to the firmware package.
     """
     internal = ctx.obj.get("internal", False)
-    perform_update(version_or_url, ip, board.lower(), internal, passwd=passwd)
+    if not internal:
+        click.echo(f"external environment is not supported yet..")
+        exit(0)
+ 
+    perform_update(version_or_url, ip, internal, passwd=passwd)
 
 # ----------------------
 # Model Zoo Subcommands
@@ -125,6 +119,11 @@ def modelzoo(ctx, ver):
     """Access models from the Model Zoo."""
     ctx.ensure_object(dict)
     ctx.obj['ver'] = ver
+    internal = ctx.obj.get("internal", False)
+    if not internal:
+        click.echo(f"external environment is not supported yet..")
+        exit(0)
+
     pass
 
 @modelzoo.command("list")
@@ -146,6 +145,35 @@ def get_model(ctx, model_name):
     click.echo(f"Getting model '{model_name}' for version: {ver}")
     download_model(internal, ver, model_name)
 
+@modelzoo.command("describe")
+@click.argument('model_name') 
+@click.pass_context
+def get_model(ctx, model_name):
+    """Download a specific model."""
+    ver = ctx.obj.get("ver")
+    internal = ctx.obj.get("internal", False)
+    click.echo(f"Getting model '{model_name}' for version: {ver}")
+    describe_model(internal, ver, model_name)
+
+# ----------------------
+# Authentication Command
+# ----------------------
+@main.group()
+@click.pass_context
+def mla(ctx):
+    """Machine Learning Accelerator Utilities."""
+    env_type, _ = get_environment_type()
+    if env_type != 'board':
+        click.echo("❌ This command can only be executed on the SiMa board.")
+    pass
+
+@mla.command("meminfo")
+@click.pass_context
+def show_mla_memory_usage(ctx):
+    """Show MLA Memory usage overtime."""
+    monitor_simaai_mem_chart()
+    pass
+
 # ----------------------
 # App Zoo Subcommands
 # ----------------------

sima_cli/data/resources_internal.yaml

@@ -0,0 +1,9 @@
+artifactory:
+  url: "https://artifacts.eng.sima.ai"
+
+auth:
+  internal_url: "access/api/v1/tokens"
+  validate_url: "artifactory/api/security/encryptedPassword"
+
+download:
+  download_url: "artifactory/"

sima_cli/data/resources_public.yaml

@@ -0,0 +1,9 @@
+artifactory:
+  url: "https://developer.sima.ai"
+
+auth:
+  auth_url: "/login"
+  validate_url: "artifactory/api/security/encryptedPassword"
+
+download:
+  download_url: "artifactory/"

sima_cli/mla/meminfo.py

@@ -0,0 +1,47 @@
+import time
+import re
+import sys
+import select
+import subprocess
+import plotext as plt
+
+def monitor_simaai_mem_chart(sample_interval_sec=5, max_samples=100):
+    sizes = []
+
+    def read_allocated_size():
+        try:
+            output = subprocess.check_output(['sudo', 'cat', '/dev/simaai-mem'], text=True)
+            match = re.search(r"Total allocated size:\s+0x([0-9a-fA-F]+)", output)
+            if match:
+                size_bytes = int(match.group(1), 16)
+                return size_bytes / (1024 * 1024)
+        except Exception as e:
+            print(f"Error reading /dev/simaai-mem: {e}")
+        return None
+
+    print("📈 Monitoring MLA memory usage... (Press 'Ctrl+C' to quit)")
+
+    while True:
+        # Check for quit key
+        if sys.stdin in select.select([sys.stdin], [], [], 0)[0]:
+            key = sys.stdin.read(1)
+            if key.lower() == 'q':
+                print("\n❌ Exiting memory monitor...")
+                break
+
+        size_mb = read_allocated_size()
+        if size_mb is not None:
+            sizes.append(size_mb)
+            sizes = sizes[-max_samples:]
+
+        if sizes:
+            plt.clear_data()
+            plt.clc()  
+            plt.title("SIMA MLA Memory Usage (MB)")
+            plt.xlabel("Seconds")
+            plt.ylabel("Memory (MB)")
+            plt.plot(sizes)
+            plt.ylim(min(sizes) * 0.95, max(sizes) * 1.05)
+            plt.show()
+        
+        time.sleep(sample_interval_sec)

sima_cli/model_zoo/model.py

@@ -3,12 +3,16 @@
 import requests
 import click
 import os
+import yaml
 from urllib.parse import urlparse
-
+from rich import print
+from rich.table import Table
+from rich.panel import Panel
 from sima_cli.utils.config import get_auth_token
+from sima_cli.utils.config_loader import artifactory_url
 from sima_cli.download import download_file_from_url
 
-ARTIFACTORY_BASE_URL = "https://artifacts.eng.sima.ai:443/artifactory"
+ARTIFACTORY_BASE_URL = artifactory_url() + '/artifactory'
 
 def _is_valid_url(url: str) -> bool:
     try:
@@ -17,6 +21,109 @@ def _is_valid_url(url: str) -> bool:
     except:
         return False
 
+def _describe_model_internal(ver: str, model_name: str):
+    repo = "sima-qa-releases"
+    base_path = f"SiMaCLI-SDK-Releases/{ver}-Release/modelzoo_edgematic/{model_name}"
+    aql_query = f"""
+                items.find({{
+                    "repo": "{repo}",
+                    "path": "{base_path}",
+                    "$or": [
+                        {{ "name": {{ "$match": "*.yaml" }} }},
+                        {{ "name": {{ "$match": "*.yml" }} }}
+                    ],
+                    "type": "file"
+                }}).include("name", "path", "repo")
+                """.strip()
+
+    headers = {
+        "Content-Type": "text/plain",
+        "Authorization": f"Bearer {get_auth_token(internal=True)}"
+    }
+
+    aql_url = f"{ARTIFACTORY_BASE_URL}/api/search/aql"
+    response = requests.post(aql_url, data=aql_query, headers=headers)
+    if response.status_code != 200:
+        click.echo(f"❌ Failed to list model files. Status: {response.status_code}")
+        click.echo(response.text)
+        return
+
+    files = response.json().get("results", [])
+    yaml_file = next((f for f in files if f["name"].endswith((".yaml", ".yml"))), None)
+
+    if not yaml_file:
+        click.echo(f"⚠️ No .yaml or .yml file found under: {base_path}")
+        return
+
+    # Download the YAML file
+    yaml_url = f"{ARTIFACTORY_BASE_URL}/{repo}/{yaml_file['path']}/{yaml_file['name']}"
+    response = requests.get(yaml_url, headers={"Authorization": f"Bearer {get_auth_token(internal=True)}"})
+
+    if response.status_code != 200:
+        click.echo(f"❌ Failed to fetch YAML: {response.status_code}")
+        return
+
+    try:
+        data = yaml.safe_load(response.text)
+    except yaml.YAMLError as e:
+        click.echo(f"❌ Failed to parse YAML: {e}")
+        return
+
+    model = data.get("model", {})
+    pipeline = data.get("pipeline", {})
+
+    print(Panel.fit(f"[bold green]{model.get('name', 'Unknown')}[/bold green] - {model.get('task', 'Unknown Task')}",
+                    subtitle=f"Status: [yellow]{model.get('status', 'n/a')}[/yellow]"))
+
+    desc_table = Table(title="Description", show_header=False)
+    for k, v in (model.get("description") or {}).items():
+        desc_table.add_row(k.capitalize(), v or "-")
+    print(desc_table)
+
+    dataset = model.get("dataset", {})
+    dataset_table = Table(title="Dataset", header_style="bold magenta")
+    dataset_table.add_column("Key")
+    dataset_table.add_column("Value")
+    dataset_table.add_row("Name", dataset.get("name", "-"))
+    for k, v in (dataset.get("params") or {}).items():
+        dataset_table.add_row(k, str(v))
+    dataset_table.add_row("Accuracy", dataset.get("accuracy", "-"))
+    dataset_table.add_row("Calibration", dataset.get("calibration", "-"))
+    print(dataset_table)
+
+    if qm := model.get("quality_metric"):
+        print(Panel.fit(f"Quality Metric: [cyan]{qm.get('name')}[/cyan]"))
+
+    q = model.get("quantization_settings", {})
+    q_table = Table(title="Quantization Settings", header_style="bold blue")
+    q_table.add_column("Setting")
+    q_table.add_column("Value")
+
+    q_table.add_row("Calibration Samples", str(q.get("calibration_num_samples", "-")))
+    q_table.add_row("Calibration Method", q.get("calibration_method", "-"))
+    q_table.add_row("Requantization Mode", q.get("requantization_mode", "-"))
+    q_table.add_row("Bias Correction", str(q.get("bias_correction", "-")))
+
+    aq = q.get("activation_quantization_scheme", {})
+    wq = q.get("weight_quantization_scheme", {})
+
+    q_table.add_row("Activation Quant", f"Asym: {aq.get('asymmetric')}, PerCh: {aq.get('per_channel')}, Bits: {aq.get('bits')}")
+    q_table.add_row("Weight Quant", f"Asym: {wq.get('asymmetric')}, PerCh: {wq.get('per_channel')}, Bits: {wq.get('bits')}")
+    print(q_table)
+
+    transforms = pipeline.get("transforms", [])
+    t_table = Table(title="Pipeline Transforms", header_style="bold green")
+    t_table.add_column("Name")
+    t_table.add_column("Params")
+
+    for step in transforms:
+        name = step.get("name")
+        params = step.get("params", {})
+        param_str = ", ".join(f"{k}={v}" for k, v in params.items()) if params else "-"
+        t_table.add_row(name, param_str)
+
+    print(t_table)
+
 def _download_model_internal(ver: str, model_name: str):
     repo = "sima-qa-releases"
     base_path = f"SiMaCLI-SDK-Releases/{ver}-Release/modelzoo_edgematic/{model_name}"
@@ -92,6 +199,7 @@ def _list_available_models_internal(version: str):
                 """.strip()
 
     aql_url = f"{ARTIFACTORY_BASE_URL}/api/search/aql"
+    print(aql_url)
     headers = {
         "Content-Type": "text/plain",
         "Authorization": f"Bearer {get_auth_token(internal=True)}"
@@ -138,6 +246,13 @@ def download_model(internal, ver, model_name):
     else:
         print('External model zoo not supported yet')
 
+def describe_model(internal, ver, model_name):
+    if internal:
+        click.echo("Model Zoo Source : SiMa Artifactory...")
+        return _describe_model_internal(ver, model_name)
+    else:
+        print('External model zoo not supported yet')
+
 # Module CLI tests
 if __name__ == "__main__":
     import sys

sima_cli/update/query.py

@@ -0,0 +1,64 @@
+import requests
+from sima_cli.utils.config_loader import load_resource_config, artifactory_url
+from sima_cli.utils.config import get_auth_token
+
+ARTIFACTORY_BASE_URL = artifactory_url() + '/artifactory'
+
+def _list_available_firmware_versions_internal(board: str, match_keyword: str = None):
+    fw_path = f"{board}" 
+    aql_query = f"""
+                items.find({{
+                    "repo": "soc-images",
+                    "path": {{
+                        "$match": "{fw_path}/*"
+                    }},
+                    "type": "folder"
+                }}).include("repo", "path", "name")
+                """.strip()
+
+    aql_url = f"{ARTIFACTORY_BASE_URL}/api/search/aql"
+    headers = {
+        "Content-Type": "text/plain",
+        "Authorization": f"Bearer {get_auth_token(internal=True)}"
+    }
+
+    response = requests.post(aql_url, data=aql_query, headers=headers)
+    if response.status_code != 200:
+        return None
+
+    results = response.json().get("results", [])
+
+    # Reconstruct full paths and remove board prefix
+    full_paths = {
+        f"{item['path']}/{item['name']}".replace(fw_path + "/", "")
+        for item in results
+    }
+
+    # Extract top-level folders
+    top_level_folders = sorted({path.split("/")[0] for path in full_paths})
+
+    if match_keyword:
+        match_keyword = match_keyword.lower()
+        top_level_folders = [
+            f for f in top_level_folders if match_keyword in f.lower()
+        ]
+
+    return top_level_folders
+
+
+def list_available_firmware_versions(board: str, match_keyword: str = None, internal: bool = False):
+    """
+    Public interface to list available firmware versions.
+
+    Parameters:
+    - board: str – Name of the board (e.g. 'davinci')
+    - match_keyword: str – Optional keyword to filter versions (case-insensitive)
+    - internal: bool – Must be True to access internal Artifactory
+
+    Returns:
+    - List[str] of firmware version folder names, or None if access is not allowed
+    """
+    if not internal:
+        raise PermissionError("Internal access required to list firmware versions.")
+
+    return _list_available_firmware_versions_internal(board, match_keyword)

sima_cli/update/updater.py

@@ -5,6 +5,7 @@ import time
 import tempfile
 import tarfile
 import subprocess
+from InquirerPy import inquirer
 from urllib.parse import urlparse
 from typing import List
 from sima_cli.utils.env import get_environment_type
@@ -12,6 +13,7 @@ from sima_cli.download import download_file_from_url
 from sima_cli.utils.config_loader import load_resource_config
 from sima_cli.update.remote import push_and_update_remote_board, get_remote_board_info, reboot_remote_board
 from sima_cli.update.local import get_local_board_info, push_and_update_local_board
+from sima_cli.update.query import list_available_firmware_versions
 
 def _resolve_firmware_url(version_or_url: str, board: str, internal: bool = False) -> str:
     """
@@ -44,6 +46,42 @@ def _resolve_firmware_url(version_or_url: str, board: str, internal: bool = Fals
     download_url = url.rstrip("/") + f"/soc-images/{board}/{version_or_url}/artifacts/release.tar.gz"
     return download_url
 
+def _pick_from_available_versions(board: str, version_or_url: str, internal: bool) -> str:
+    """
+    Presents an interactive menu (with search) for selecting a firmware version.
+    """
+
+    if "http" in version_or_url:
+        return version_or_url
+
+    available_versions = list_available_firmware_versions(board, version_or_url, internal=True)
+
+    if len(available_versions) > 1:
+        click.echo("Multiple firmware versions found matching your input:")
+
+        selected_version = inquirer.fuzzy(
+            message="Select a version:",
+            choices=available_versions,
+            max_height="70%",  # scrollable
+            instruction="(Use ↑↓ to navigate, / to search, Enter to select)"
+        ).execute()
+
+        if not selected_version:
+            click.echo("No selection made. Exiting.", err=True)
+            raise SystemExit(1)
+
+        return selected_version
+
+    elif len(available_versions) == 1:
+        return available_versions[0]
+
+    else:
+        click.echo(
+            f"No firmware versions found matching keyword '{version_or_url}' for board '{board}'.",
+            err=True
+        )
+        raise SystemExit(1)
+
 def _sanitize_url_to_filename(url: str) -> str:
     """
     Convert a URL to a safe filename by replacing slashes and removing protocol.
@@ -289,7 +327,7 @@ def _update_remote(extracted_paths: List[str], ip: str, board: str, passwd: str,
     return script_path
 
 
-def perform_update(version_or_url: str, ip: str = None, board: str = "davinci", internal: bool = False, passwd: str = "edgeai"):
+def perform_update(version_or_url: str, ip: str = None, internal: bool = False, passwd: str = "edgeai"):
     r"""
     Update the system based on environment and input.
 
@@ -304,36 +342,45 @@ def perform_update(version_or_url: str, ip: str = None, board: str = "davinci",
         board (str): Board type, must be 'davinci' or 'modalix'.
         passwd : non-default password in case user has changed the password of the board user `sima`
     """
-    board = board.lower()
-    if board not in ("davinci", "modalix"):
-        click.echo(f"❌ Invalid board type '{board}'. Must be 'davinci' or 'modalix'.")
-        return
-
     try:
+        board = ''
         env_type, env_subtype = get_environment_type()
         click.echo(f"🔄 Running update for environment: {env_type} ({env_subtype})")
         click.echo(f"🔧 Requested version or URL: {version_or_url}")
-        click.echo(f"🔧 Target board: {board}")
-
-        extracted_paths = _download_image(version_or_url, board, internal)
-        click.echo("⚠️  DO NOT INTERRUPT THE UPDATE PROCESS...")
-
-        if len(extracted_paths) > 0:
-            if env_type == "host" and env_subtype == 'linux':
-                # Always update the remote device first then update the host driver, otherwise the host would 
-                # not be able to connect to the board
-                click.echo("👉 Updating PCIe host driver and downloading firmware...")        
-                script_path = _update_remote(extracted_paths, ip, board, passwd, reboot_and_wait = False)
-                _update_host(script_path, board, ip, passwd)
-            elif env_type == "board":
-                _update_board(extracted_paths, board, passwd)
-            elif env_type == "sdk":
-                click.echo("👉 Updating firmware from within the Palette SDK...: Not implemented yet")
-            elif ip:
-                click.echo(f"👉 Updating firmware on remote board at {ip}...")
-                _update_remote(extracted_paths, ip, board, passwd, reboot_and_wait = True)
-            else:
-                click.echo("❌ Unknown environment. Use --ip to specify target device.")
+
+        if env_type == 'board':
+            board, version = get_local_board_info()
+        else:
+            board, version = get_remote_board_info(ip, passwd)
+
+        if board in ['davinci', 'modalix']:
+            click.echo(f"🔧 Target board: {board}, board currently running: {version}")
+                        
+            if 'http' not in version_or_url:
+                version_or_url = _pick_from_available_versions(board, version_or_url, internal)
+
+            extracted_paths = _download_image(version_or_url, board, internal)
+            click.echo("⚠️  DO NOT INTERRUPT THE UPDATE PROCESS...")
+
+            if len(extracted_paths) > 0:
+                if env_type == "host" and env_subtype == 'linux':
+                    # Always update the remote device first then update the host driver, otherwise the host would 
+                    # not be able to connect to the board
+                    click.echo("👉 Updating PCIe host driver and downloading firmware...")        
+                    script_path = _update_remote(extracted_paths, ip, board, passwd, reboot_and_wait = False)
+                    _update_host(script_path, board, ip, passwd)
+                elif env_type == "board":
+                    _update_board(extracted_paths, board, passwd)
+                elif env_type == "sdk":
+                    click.echo("👉 Updating firmware from within the Palette SDK...: Not implemented yet")
+                elif ip:
+                    click.echo(f"👉 Updating firmware on remote board at {ip}...")
+                    _update_remote(extracted_paths, ip, board, passwd, reboot_and_wait = True)
+                else:
+                    click.echo("❌ Unknown environment. Use --ip to specify target device.")
+        else:
+            click.echo("❌ Unable to retrieve target board information")
+
     except Exception as e:
         click.echo(f"❌ Update failed {e}")
 

sima_cli/utils/config_loader.py

@@ -13,7 +13,6 @@ def load_resource_config():
         "internal": {}
     }
 
-    # Public config (bundled with the package)
     public_path = os.path.abspath(
         os.path.join(os.path.dirname(__file__), "..", "data", "resources_public.yaml")
     )
@@ -21,11 +20,14 @@ def load_resource_config():
         with open(public_path, "r") as f:
             config["public"] = yaml.safe_load(f) or {}
 
-    # Internal config (~/.sima-cli/resources_internal.yaml)
-    internal_path = os.path.join(os.path.expanduser("~"), ".sima-cli", "resources_internal.yaml")
+    internal_path = os.path.abspath(
+        os.path.join(os.path.dirname(__file__), "..", "data", "resources_internal.yaml")
+    )
     if os.path.exists(internal_path):
         with open(internal_path, "r") as f:
             config["internal"] = yaml.safe_load(f) or {}
+    else:
+        print(f"Internal resource map not found... {internal_path}")
 
     return config
 
@@ -36,5 +38,28 @@ def internal_resource_exists():
     Returns:
         bool: True if the internal resource file exists, False otherwise.
     """
-    internal_path = os.path.join(os.path.expanduser("~"), ".sima-cli", "resources_internal.yaml")
-    return os.path.exists(internal_path)
+    internal_path = os.path.abspath(
+        os.path.join(os.path.dirname(__file__), "..", "data", "resources_internal.yaml")
+    )
+    return os.path.exists(internal_path)
+
+def artifactory_url():
+    """
+    Retrieve the Artifactory base URL from the internal resource configuration.
+
+    Returns:
+        str or None: The Artifactory URL if found, otherwise None.
+
+    Notes:
+        - Expects 'load_resource_config()' to return a dictionary with an 'internal' section
+          containing an 'artifactory' section with a 'url' field.
+        - If any error occurs (e.g., missing fields, file issues), prints an error message
+          and returns None.
+    """
+    try:
+        cfg = load_resource_config()
+        return cfg.get("internal", {}).get("artifactory", {}).get("url", {})
+    except Exception as e:
+        print('Unable to retrieve Artifactory URL')
+        return None
+

{sima_cli-0.0.13.dist-info → sima_cli-0.0.15.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sima-cli
-Version: 0.0.13
+Version: 0.0.15
 Summary: CLI tool for SiMa Developer Portal to download models, firmware, and apps.
 Home-page: https://developer.sima.ai/
 Author: SiMa.ai
@@ -21,6 +21,8 @@ Requires-Dist: click
 Requires-Dist: tqdm
 Requires-Dist: pyyaml
 Requires-Dist: paramiko
+Requires-Dist: plotext
+Requires-Dist: rich
 Dynamic: author
 Dynamic: license-file
 Dynamic: requires-python

{sima_cli-0.0.13.dist-info → sima_cli-0.0.15.dist-info}/RECORD

@@ -1,26 +1,31 @@
 sima_cli/__init__.py,sha256=Nb2jSg9-CX1XvSc1c21U9qQ3atINxphuNkNfmR-9P3o,332
 sima_cli/__main__.py,sha256=ehzD6AZ7zGytC2gLSvaJatxeD0jJdaEvNJvwYeGsWOg,69
-sima_cli/__version__.py,sha256=HIATn2M6NOw71vakibWXUOPNt1MzrZdtVUqxl8ScDyU,49
-sima_cli/cli.py,sha256=hXWL2K7PHC2ezWEdsw2O6FPqjqLUC9sCv53YX7JXK5w,6114
+sima_cli/__version__.py,sha256=rZ8VyuhqgF22Ho9wHPc5IESWFORxhypVEyKqL_-8IvU,49
+sima_cli/cli.py,sha256=yd4ClOInhKAFA2ffMX1lUPRvWVziMZLHToSTAr_eyAA,6894
 sima_cli/app_zoo/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sima_cli/app_zoo/app.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sima_cli/auth/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-sima_cli/auth/login.py,sha256=lnTfJrwrRznNY6zWEl2AwAveGJoCduPiOHLiGxTIyJ4,3660
+sima_cli/auth/basic_auth.py,sha256=xS_EC0iYyozDSrZEoz44lU5dLBk6covxiCv09WiQECw,4264
+sima_cli/auth/login.py,sha256=Tpe5ZRau4QrlQ6Zxg1rBco4qZ2nr4rkfZAohiTPBbVk,3687
+sima_cli/data/resources_internal.yaml,sha256=zlQD4cSnZK86bLtTWuvEudZTARKiuIKmB--Jv4ajL8o,200
+sima_cli/data/resources_public.yaml,sha256=ZT1CzPrGMfSLVWiRJfW2-jb-ilmh7yNlkqyO5Fvqk58,178
 sima_cli/download/__init__.py,sha256=6y4O2FOCYFR2jdnQoVi3hRtEoZ0Gw6rydlTy1SGJ5FE,218
 sima_cli/download/downloader.py,sha256=zL8daM7Fqj1evzfQ9EHL1CVbuYL0_aQNhromqm7LkE0,4863
+sima_cli/mla/meminfo.py,sha256=ndc8kQJmWGEIdvNh6iIhATGdrkqM2pbddr_eHxaPNfg,1466
 sima_cli/model_zoo/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-sima_cli/model_zoo/model.py,sha256=WwtxyPiL0XAOFwH1JUK4aWnZpROHULTwfsnv0Y5_DpM,5223
+sima_cli/model_zoo/model.py,sha256=q91Nrg62j1TqwPO8HiX4nlEFCCmzNEFcyFTBVMbJm8w,9836
 sima_cli/update/__init__.py,sha256=0P-z-rSaev40IhfJXytK3AFWv2_sdQU4Ry6ei2sEus0,66
 sima_cli/update/local.py,sha256=jiGrwuU2Z1HV_RT1_dYuI_Ish-f818AvCEk7sAM3l94,3032
+sima_cli/update/query.py,sha256=eOTC2ZAWbFFf_0h8D-MO1HrIsQYRc7fu5OyeFNEAv48,2168
 sima_cli/update/remote.py,sha256=ePlnvlGHrASMMjYGM9w-6hMeDMgGiJu_BlHLamU1YtI,8420
-sima_cli/update/updater.py,sha256=lrXYV1Iy2u7leY043GUSIfP0g9MaKTjSoo1hrMyiChk,13572
+sima_cli/update/updater.py,sha256=GtYOyhZj_Xk99iI-Zj4hGywAnitkah2MuMoE9NEa9V0,15228
 sima_cli/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sima_cli/utils/artifactory.py,sha256=6YyVpzVm8ATy7NEwT9nkWx-wptkXrvG7Wl_zDT6jmLs,2390
 sima_cli/utils/config.py,sha256=wE-cPQqY_gOqaP8t01xsRHD9tBUGk9MgBUm2GYYxI3E,1616
-sima_cli/utils/config_loader.py,sha256=MTNQXK7hRy3n5SgzVsjXyjIHInZH9RdAcQtsLM6f3RU,1250
+sima_cli/utils/config_loader.py,sha256=7I5we1yiCai18j9R9jvhfUzAmT3OjAqVK35XSLuUw8c,2005
 sima_cli/utils/env.py,sha256=LJy2eO8cfEYsLuC7p3BT_FAoaZc9emtq6NYhHRBpiBE,5512
 sima_cli/utils/network.py,sha256=UvqxbqbWUczGFyO-t1SybG7Q-x9kjUVRNIn_D6APzy8,1252
-sima_cli-0.0.13.dist-info/licenses/LICENSE,sha256=a260OFuV4SsMZ6sQCkoYbtws_4o2deFtbnT9kg7Rfd4,1082
+sima_cli-0.0.15.dist-info/licenses/LICENSE,sha256=a260OFuV4SsMZ6sQCkoYbtws_4o2deFtbnT9kg7Rfd4,1082
 tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tests/test_app_zoo.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tests/test_auth.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -29,8 +34,8 @@ tests/test_download.py,sha256=t87DwxlHs26_ws9rpcHGwr_OrcRPd3hz6Zmm0vRee2U,4465
 tests/test_firmware.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tests/test_model_zoo.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tests/test_utils.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-sima_cli-0.0.13.dist-info/METADATA,sha256=ffFCnatscJoVHDiYr3UclBVrHc_MbkPUBwwDtCMhXZo,3562
-sima_cli-0.0.13.dist-info/WHEEL,sha256=SmOxYU7pzNKBqASvQJ7DjX3XGUF92lrGhMb3R6_iiqI,91
-sima_cli-0.0.13.dist-info/entry_points.txt,sha256=xRYrDq1nCs6R8wEdB3c1kKuimxEjWJkHuCzArQPT0Xk,47
-sima_cli-0.0.13.dist-info/top_level.txt,sha256=FtrbAUdHNohtEPteOblArxQNwoX9_t8qJQd59fagDlc,15
-sima_cli-0.0.13.dist-info/RECORD,,
+sima_cli-0.0.15.dist-info/METADATA,sha256=Gfm1zm3xrURxtFf9DMeq5d5vjkko8Mnk8sWVpvW4EU8,3605
+sima_cli-0.0.15.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+sima_cli-0.0.15.dist-info/entry_points.txt,sha256=xRYrDq1nCs6R8wEdB3c1kKuimxEjWJkHuCzArQPT0Xk,47
+sima_cli-0.0.15.dist-info/top_level.txt,sha256=FtrbAUdHNohtEPteOblArxQNwoX9_t8qJQd59fagDlc,15
+sima_cli-0.0.15.dist-info/RECORD,,

{sima_cli-0.0.13.dist-info → sima_cli-0.0.15.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (79.0.1)
+Generator: setuptools (80.9.0)
 Root-Is-Purelib: true
 Tag: py3-none-any