mcli-framework 7.0.3__py3-none-any.whl → 7.0.5__py3-none-any.whl

mcli/app/completion_helpers.py

@@ -137,8 +137,12 @@ LAZY_COMMAND_COMPLETIONS = {
         "subcommands": [
             "download",
             "list",
-            "remove",
-            "info"
+            "start",
+            "stop",
+            "pull",
+            "delete",
+            "recommend",
+            "status"
         ]
     },
     "cron-test": {

mcli/app/model_cmd.py

@@ -7,6 +7,7 @@ from pathlib import Path
 from typing import Optional
 
 import click
+import psutil
 
 from mcli.lib.logger.logger import get_logger
 from mcli.workflow.model_service.lightweight_model_server import (
@@ -223,5 +224,106 @@ def status(port: int):
         click.echo(f"❌ Error checking server: {e}")
 
 
+@model.command()
+@click.option("--port", "-p", default=8080, help="Port where server is running")
+def stop(port: int):
+    """Stop the lightweight model server."""
+    import requests
+    import psutil
+
+    try:
+        # First check if server is running
+        try:
+            response = requests.get(f"http://localhost:{port}/health", timeout=2)
+            if response.status_code != 200:
+                click.echo(f"❌ No server running on port {port}")
+                return
+        except requests.exceptions.ConnectionError:
+            click.echo(f"❌ No server running on port {port}")
+            return
+
+        # Find and kill the process using the port
+        for proc in psutil.process_iter(["pid", "name", "connections"]):
+            try:
+                connections = proc.info.get("connections")
+                if connections:
+                    for conn in connections:
+                        if hasattr(conn, "laddr") and conn.laddr.port == port:
+                            click.echo(f"🛑 Stopping server (PID: {proc.pid})...")
+                            proc.terminate()
+                            proc.wait(timeout=5)
+                            click.echo("✅ Server stopped successfully")
+                            return
+            except (psutil.NoSuchProcess, psutil.AccessDenied, psutil.TimeoutExpired):
+                continue
+
+        click.echo("⚠️  Could not find server process")
+
+    except Exception as e:
+        click.echo(f"❌ Error stopping server: {e}")
+
+
+@model.command()
+@click.argument("model_name")
+def pull(model_name: str):
+    """Pull (download) a specific lightweight model."""
+    if model_name not in LIGHTWEIGHT_MODELS:
+        click.echo(f"❌ Model '{model_name}' not found.")
+        click.echo("Available models:")
+        for key in LIGHTWEIGHT_MODELS.keys():
+            click.echo(f"  • {key}")
+        sys.exit(1)
+
+    server = LightweightModelServer()
+
+    click.echo(f"Pulling model: {model_name}")
+    success = server.download_and_load_model(model_name)
+
+    if success:
+        click.echo(f"✅ Successfully pulled {model_name}")
+    else:
+        click.echo(f"❌ Failed to pull {model_name}")
+        sys.exit(1)
+
+
+@model.command()
+@click.argument("model_name")
+@click.option("--force", "-f", is_flag=True, help="Force deletion without confirmation")
+def delete(model_name: str, force: bool):
+    """Delete a downloaded lightweight model."""
+    server = LightweightModelServer()
+    downloaded_models = server.downloader.get_downloaded_models()
+
+    if model_name not in downloaded_models:
+        click.echo(f"❌ Model '{model_name}' not found.")
+        click.echo("Downloaded models:")
+        if downloaded_models:
+            for model in downloaded_models:
+                click.echo(f"  • {model}")
+        else:
+            click.echo("  (none)")
+        sys.exit(1)
+
+    # Confirm deletion unless --force is used
+    if not force:
+        model_info = LIGHTWEIGHT_MODELS.get(model_name, {})
+        name = model_info.get("name", model_name)
+        size = model_info.get("size_mb", "unknown")
+        click.echo(f"⚠️  About to delete:")
+        click.echo(f"   Model: {name}")
+        click.echo(f"   Size: {size} MB")
+        if not click.confirm("Are you sure you want to delete this model?"):
+            click.echo("❌ Deletion cancelled")
+            return
+
+    success = server.delete_model(model_name)
+
+    if success:
+        click.echo(f"✅ Successfully deleted {model_name}")
+    else:
+        click.echo(f"❌ Failed to delete {model_name}")
+        sys.exit(1)
+
+
 if __name__ == "__main__":
     model()

mcli/self/self_cmd.py

@@ -774,18 +774,15 @@ def logs(type: str, lines: int, follow: bool, date: str, grep: str, level: str):
     from datetime import datetime
     from pathlib import Path
 
-    # Find the logs directory
-    current_file = Path(__file__)
-    # Go up 5 levels: file -> self -> mcli -> src -> repo_root
-    project_root = current_file.parents[4]
-    logs_dir = project_root / "logs"
+    # Import get_logs_dir to get the correct logs directory
+    from mcli.lib.paths import get_logs_dir
 
-    # Alternative: try current working directory first
-    if not logs_dir.exists():
-        logs_dir = Path.cwd() / "logs"
+    # Get the logs directory (creates it if it doesn't exist)
+    logs_dir = get_logs_dir()
 
     if not logs_dir.exists():
         click.echo("❌ Logs directory not found", err=True)
+        click.echo(f"Expected location: {logs_dir}", err=True)
         return
 
     # Determine which log files to read

mcli/workflow/model_service/lightweight_model_server.py

@@ -566,6 +566,43 @@ class LightweightModelServer:
         else:
             return "distilbert-base-uncased"  # Standard small model
 
+    def stop_server(self) -> bool:
+        """Stop the lightweight server"""
+        if not self.running:
+            print("⚠️  Server is not running")
+            return False
+
+        try:
+            self.running = False
+            print("🛑 Server stopped")
+            return True
+        except Exception as e:
+            print(f"❌ Error stopping server: {e}")
+            return False
+
+    def delete_model(self, model_key: str) -> bool:
+        """Delete a downloaded model"""
+        try:
+            model_dir = self.models_dir / model_key
+
+            if not model_dir.exists():
+                print(f"⚠️  Model '{model_key}' not found")
+                return False
+
+            # Remove from loaded models if present
+            if model_key in self.loaded_models:
+                del self.loaded_models[model_key]
+                print(f"✅ Unloaded model: {model_key}")
+
+            # Delete the model directory
+            shutil.rmtree(model_dir)
+            print(f"✅ Deleted model: {model_key}")
+            return True
+
+        except Exception as e:
+            print(f"❌ Error deleting model {model_key}: {e}")
+            return False
+
 
 def create_simple_client():
     """Create a simple client script for testing"""

{mcli_framework-7.0.3.dist-info → mcli_framework-7.0.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mcli-framework
-Version: 7.0.3
+Version: 7.0.5
 Summary: 🚀 High-performance CLI framework with Rust extensions, AI chat, and stunning visuals
 Author-email: Luis Fernandez de la Vara <luis@lefv.io>
 Maintainer-email: Luis Fernandez de la Vara <luis@lefv.io>

{mcli_framework-7.0.3.dist-info → mcli_framework-7.0.5.dist-info}/RECORD

@@ -6,11 +6,11 @@ mcli/app/__init__.py,sha256=Qy_ttkXt_XKuSQKgQx3H7NqVy8c7otSFtBVZaatCdZc,816
 mcli/app/chat_cmd.py,sha256=OJK91iX-f-CL5-m3ECG4IlcbLbsWQpPp7XWuGnb0cVQ,1552
 mcli/app/commands_cmd.py,sha256=5MccxYKNkzIisDeBPJj6K4ZwWeZFB2a4P8AS9B1_EDs,8441
 mcli/app/completion_cmd.py,sha256=8gtYDddviBCD6Gk57lkLKOpim5QAMMcugYPmwfraBeo,7862
-mcli/app/completion_helpers.py,sha256=GJvkxJ5lU_Gh4K35Q275QW1fN39poasyYjDO5-5kxM0,8809
+mcli/app/completion_helpers.py,sha256=PR66qgVNC5_st-CBiD4uuGfO3Zs7Y3QmJ2GJjpx5N6g,8897
 mcli/app/cron_test_cmd.py,sha256=Ai4Smg2WxULeiMD5s2m_S_fXdMAAQsKHpSc4iJGSnwI,26156
 mcli/app/logs_cmd.py,sha256=xkGjUiuV1a1-2j8rkffp0OwnG9nS39-i2jpCnKUn9f0,13777
 mcli/app/main.py,sha256=iA9HdqhBaOnOJ2--edLPD7iAatAzhIl5NAcNhw9qJAw,19010
-mcli/app/model_cmd.py,sha256=wRs2xfDbWrdzmVH1hbHr8jypucNUKpXvNdYJAlvk5xc,8153
+mcli/app/model_cmd.py,sha256=xJqD4xBLFqi3sBsgAeyXIC8n5SR3czl8wIWx8za31vA,11756
 mcli/app/redis_cmd.py,sha256=Cl0LQ3Mqt27gLeb542_xw6bJBbIE-CBmWyMmaUTSk8c,9426
 mcli/app/visual_cmd.py,sha256=jXighahHxeM9HANQ2Brk6nKFgi2ZuQBOBH7PE5xhebk,9428
 mcli/app/model/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -159,7 +159,7 @@ mcli/public/public.py,sha256=t9BkO1XV7s3YcoH0bbIpyjZ05UX_vBjaKtKkuDX7wZ0,114
 mcli/public/commands/__init__.py,sha256=0WGm1i4t1uifkC_5FmbQ2r03q53yrZ3KnoWOd2M7dEA,103
 mcli/public/oi/oi.py,sha256=SQabQWQ1pE67pWYEHwIDc3R93DARJfB6VHk7qxWx9xo,308
 mcli/self/__init__.py,sha256=7hCrgaRb4oAgAf-kcyzqhJ5LFpW19jwF5bxowR4LwjY,41
-mcli/self/self_cmd.py,sha256=FM7LqGi3ZXRZCjFBhxIMRce_--8L7GT2UjH3sgKP3UI,47491
+mcli/self/self_cmd.py,sha256=KbgavTt9uXVGKbuX1o4aD4Ll0Ssz96qDHy6WSP7nf88,47419
 mcli/workflow/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 mcli/workflow/lsh_integration.py,sha256=khwmMPsdYdkmmLxlZi_UqUo2p0Nf-6GF6PPbtOrmoYQ,13290
 mcli/workflow/workflow.py,sha256=t58OVXmU9uQCJnyXuIbMAm8lihSzJ_jI10vXPNpZspk,928
@@ -191,7 +191,7 @@ mcli/workflow/interview/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG
 mcli/workflow/model_service/client.py,sha256=wWYde_zR6_o6SP6oePwkHKmEVBtELrP8uKyZNTBWwVc,20888
 mcli/workflow/model_service/download_and_run_efficient_models.py,sha256=UlKs_Oq_1Vn5qFP28ALCF6CVMTqcRT8hG5quHeGHLIc,9898
 mcli/workflow/model_service/lightweight_embedder.py,sha256=82B0tYOYVTB3rdC-7XYEMnkw4Q_K0-p3LeyvdaP6ro8,14165
-mcli/workflow/model_service/lightweight_model_server.py,sha256=NYhIBDBSa1UwnsF6cEwDuNrcOLYcEIntw19GeMGVF5U,29883
+mcli/workflow/model_service/lightweight_model_server.py,sha256=VyBrUba1maPHNDzO_o7FhLPgN_ncz5EOqrHorzDV5rc,31063
 mcli/workflow/model_service/lightweight_test.py,sha256=J1--5Q8YvJT3Wzhqf0ElDAostRVA2yDs1LkVwkDMf5M,7330
 mcli/workflow/model_service/model_service.py,sha256=pAxxHw4tShkrnWWFBnzsrGGKEFa8ZEv4Mza_C8clK38,69729
 mcli/workflow/model_service/ollama_efficient_runner.py,sha256=fFZLoIfmgbQnMLwnXuH2ya5o_Tu6ww3ERtXg58HbamE,14340
@@ -236,9 +236,9 @@ mcli/workflow/videos/__init__.py,sha256=aV3DEoO7qdKJY4odWKoQbOKDQq4ludTeCLnZcupO
 mcli/workflow/videos/videos.py,sha256=C47ViVv6qqqkSKQz6YXjzhok4UrqFbya8w5k_x7hToM,8360
 mcli/workflow/wakatime/__init__.py,sha256=wKG8cVIHVtMPhNRFGFtX43bRnocHqOMMkFMkmW-M6pU,2626
 mcli/workflow/wakatime/wakatime.py,sha256=sEjsUKa3-XyE8Ni6sAb_D3GAY5jDcA30KknW9YTbLTA,142
-mcli_framework-7.0.3.dist-info/licenses/LICENSE,sha256=sahwAMfrJv2-V66HNPTp7A9UmMjxtyejwTZZoWQvEcI,1075
-mcli_framework-7.0.3.dist-info/METADATA,sha256=T2hlb-nQrSmCvzSNg6QcRYrpEGJYyGjW6LUHvL6JbIs,14307
-mcli_framework-7.0.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-mcli_framework-7.0.3.dist-info/entry_points.txt,sha256=dYrZbDIm-KUPsl1wfv600Kx_8sMy89phMkCihbDRgP8,261
-mcli_framework-7.0.3.dist-info/top_level.txt,sha256=_bnO8J2EUkliWivey_1le0UrnocFKmyVMQjbQ8iVXjc,5
-mcli_framework-7.0.3.dist-info/RECORD,,
+mcli_framework-7.0.5.dist-info/licenses/LICENSE,sha256=sahwAMfrJv2-V66HNPTp7A9UmMjxtyejwTZZoWQvEcI,1075
+mcli_framework-7.0.5.dist-info/METADATA,sha256=qds7VDa4wmRAfUlfVpr3y9bwS1gHHKD5uQ193Ol89qA,14307
+mcli_framework-7.0.5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+mcli_framework-7.0.5.dist-info/entry_points.txt,sha256=dYrZbDIm-KUPsl1wfv600Kx_8sMy89phMkCihbDRgP8,261
+mcli_framework-7.0.5.dist-info/top_level.txt,sha256=_bnO8J2EUkliWivey_1le0UrnocFKmyVMQjbQ8iVXjc,5
+mcli_framework-7.0.5.dist-info/RECORD,,