mcli-framework 7.12.1__py3-none-any.whl → 7.12.3__py3-none-any.whl

mcli/workflow/model_service/test_example.py

@@ -9,12 +9,10 @@ This script demonstrates how to:
 4. Test different model types
 """
 
-import json
 import os
 import subprocess
 import sys
 import time
-from pathlib import Path
 
 import requests
 
@@ -25,16 +23,16 @@ from client import ModelServiceClient
 
 
 def check_service_running(url: str = "http://localhost:8000") -> bool:
-    """Check if the model service is running"""
+    """Check if the model service is running."""
     try:
         response = requests.get(f"{url}/health", timeout=5)
         return response.status_code == 200
-    except:
+    except Exception:
         return False
 
 
 def start_service():
-    """Start the model service if not running"""
+    """Start the model service if not running."""
     if check_service_running():
         print("✅ Model service is already running")
         return True
@@ -42,7 +40,7 @@ def start_service():
     print("🚀 Starting model service...")
     try:
         # Start the service in the background
-        process = subprocess.Popen(
+        _process = subprocess.Popen(  # noqa: F841
             [sys.executable, "model_service.py", "start"],
             stdout=subprocess.PIPE,
             stderr=subprocess.PIPE,
@@ -64,7 +62,7 @@ def start_service():
 
 
 def test_text_generation():
-    """Test text generation with a simple model"""
+    """Test text generation with a simple model."""
     print("\n" + "=" * 60)
     print("🧪 Testing Text Generation")
     print("=" * 60)
@@ -106,7 +104,7 @@ def test_text_generation():
 
 
 def test_text_classification():
-    """Test text classification with a sentiment model"""
+    """Test text classification with a sentiment model."""
     print("\n" + "=" * 60)
     print("🧪 Testing Text Classification")
     print("=" * 60)
@@ -148,7 +146,7 @@ def test_text_classification():
 
 
 def test_translation():
-    """Test translation with a translation model"""
+    """Test translation with a translation model."""
     print("\n" + "=" * 60)
     print("🧪 Testing Translation")
     print("=" * 60)
@@ -186,7 +184,7 @@ def test_translation():
 
 
 def test_batch_operations():
-    """Test batch operations and performance"""
+    """Test batch operations and performance."""
     print("\n" + "=" * 60)
     print("🧪 Testing Batch Operations")
     print("=" * 60)
@@ -231,7 +229,7 @@ def test_batch_operations():
         total_time = time.time() - start_time
 
         # Display results
-        print(f"\n📊 Batch Results:")
+        print("\n📊 Batch Results:")
         print(f"Total time: {total_time:.2f} seconds")
         print(f"Average time per request: {total_time/len(prompts):.2f} seconds")
 
@@ -249,7 +247,7 @@ def test_batch_operations():
 
 
 def test_service_management():
-    """Test service management functions"""
+    """Test service management functions."""
     print("\n" + "=" * 60)
     print("🧪 Testing Service Management")
     print("=" * 60)
@@ -286,7 +284,7 @@ def test_service_management():
 
 
 def main():
-    """Main test function"""
+    """Main test function."""
     print("🚀 MCLI Model Service Test Suite")
     print("=" * 60)
 

mcli/workflow/model_service/test_integration.py

@@ -3,9 +3,7 @@
 Test script for lightweight model server integration with MCLI model service
 """
 
-import json
 import sys
-import time
 from pathlib import Path
 
 import requests
@@ -17,7 +15,7 @@ from mcli.workflow.model_service.model_service import LIGHTWEIGHT_MODELS, ModelS
 
 
 def test_lightweight_integration():
-    """Test the lightweight server integration"""
+    """Test the lightweight server integration."""
     print("🧪 Testing Lightweight Model Server Integration")
     print("=" * 60)
 
@@ -86,7 +84,7 @@ def test_lightweight_integration():
 
 
 def test_cli_commands():
-    """Test CLI commands"""
+    """Test CLI commands."""
     print("\n🧪 Testing CLI Commands")
     print("=" * 40)
 
@@ -100,7 +98,7 @@ def test_cli_commands():
 
 
 def main():
-    """Main test function"""
+    """Main test function."""
     print("🚀 MCLI Lightweight Model Server Integration Test")
     print("=" * 70)
 

mcli/workflow/model_service/test_new_features.py

@@ -5,18 +5,17 @@ Test script for the new model service features:
 2. Add model from URL functionality
 """
 
-import os
 import sys
 from pathlib import Path
 
 # Add the parent directory to the path so we can import the model service
 sys.path.insert(0, str(Path(__file__).parent.parent.parent.parent))
 
-from mcli.workflow.model_service.model_service import ModelManager, ModelService
+from mcli.workflow.model_service.model_service import ModelService
 
 
 def test_list_models():
-    """Test the list models functionality"""
+    """Test the list models functionality."""
     print("🧪 Testing list models functionality...")
 
     try:
@@ -35,7 +34,7 @@ def test_list_models():
 
         # Test summary
         summary = service.model_manager.get_models_summary()
-        print(f"\n📊 Summary:")
+        print("\n📊 Summary:")
         print(f"  Total models: {summary['total_models']}")
         print(f"  Loaded models: {summary['loaded_models']}")
         print(f"  Total memory: {summary['total_memory_mb']:.1f} MB")
@@ -50,7 +49,7 @@ def test_list_models():
 
 
 def test_add_model_from_url():
-    """Test the add model from URL functionality"""
+    """Test the add model from URL functionality."""
     print("\n🧪 Testing add model from URL functionality...")
 
     try:
@@ -60,7 +59,7 @@ def test_add_model_from_url():
         test_model_url = (
             "https://huggingface.co/microsoft/DialoGPT-small/resolve/main/pytorch_model.bin"
         )
-        test_tokenizer_url = (
+        _test_tokenizer_url = (  # noqa: F841
             "https://huggingface.co/microsoft/DialoGPT-small/resolve/main/tokenizer.json"
         )
 
@@ -92,7 +91,7 @@ def test_add_model_from_url():
 
 
 def test_api_endpoints():
-    """Test that the new API endpoints are properly defined"""
+    """Test that the new API endpoints are properly defined."""
     print("\n🧪 Testing API endpoints...")
 
     try:
@@ -120,7 +119,7 @@ def test_api_endpoints():
 
 
 def main():
-    """Run all tests"""
+    """Run all tests."""
     print("🚀 Testing new model service features...")
     print("=" * 50)
 

mcli/workflow/notebook/converter.py

@@ -47,7 +47,7 @@ class WorkflowConverter:
 
         if len(segments) > 1:
             # Found cell markers
-            for i, segment in enumerate(segments):
+            for _i, segment in enumerate(segments):
                 if segment.strip():
                     cells.append(
                         NotebookCell(

mcli/workflow/notebook/notebook_cmd.py

@@ -27,8 +27,7 @@ logger = get_logger()
 
 @click.group(name="notebook")
 def notebook():
-    """Workflow notebook management commands"""
-    pass
+    """Workflow notebook management commands."""
 
 
 @notebook.command(name="convert")
@@ -153,7 +152,7 @@ def migrate(directory: Optional[str], backup: bool, in_place: bool, dry_run: boo
     try:
         results = WorkflowConverter.migrate_directory(target_dir, backup=backup, in_place=in_place)
 
-        success(f"\nMigration complete:")
+        success("\nMigration complete:")
         info(f"  Total files: {results['total']}")
         success(f"  Converted: {results['converted']}")
         warning(f"  Skipped: {results['skipped']}")
@@ -203,7 +202,7 @@ def validate(notebook_file: str, schema: bool, syntax: bool, validate_all: bool)
         # Run all validations
         mcli workflow notebook validate notebook.json --all
     """
-    if validate_all:
+    if validate_all:  # noqa: SIM114
         schema = syntax = True
     elif not (schema or syntax):
         # Default to all if no specific validation requested
@@ -300,7 +299,7 @@ def notebook_info(notebook_file: str, output_json: bool):
                 info(f"Group: {notebook.metadata.mcli.group}")
             info(f"Version: {notebook.metadata.mcli.version}")
             info(f"Language: {notebook.metadata.mcli.language.value}")
-            info(f"\nCells:")
+            info("\nCells:")
             info(f"  Total: {len(notebook.cells)}")
             info(f"  Code: {len(notebook.code_cells)}")
             info(f"  Markdown: {len(notebook.markdown_cells)}")
@@ -368,7 +367,7 @@ def create(name: str, description: str, group: Optional[str], language: str, out
     # Add welcome markdown cell
     notebook.add_markdown_cell(
         f"# {name}\n\n{description}\n\n"
-        f"This is a workflow notebook. Add code cells below to define your workflow."
+        "This is a workflow notebook. Add code cells below to define your workflow."
     )
 
     # Add example code cell

mcli/workflow/notebook/schema.py

@@ -7,7 +7,6 @@ with existing MCLI workflow JSON format.
 """
 
 from dataclasses import dataclass, field
-from datetime import datetime
 from enum import Enum
 from typing import Any, Dict, List, Optional, Union
 

mcli/workflow/notebook/validator.py

@@ -12,7 +12,7 @@ import ast
 import subprocess
 import tempfile
 from pathlib import Path
-from typing import Any, Dict, List, Optional
+from typing import Dict, List
 
 from mcli.lib.logger.logger import get_logger
 
@@ -105,7 +105,11 @@ class NotebookValidator:
                 if not self._validate_python_syntax(code, i):
                     all_valid = False
 
-            elif language in (CellLanguage.SHELL, CellLanguage.BASH, CellLanguage.ZSH):
+            elif language in (
+                CellLanguage.SHELL,
+                CellLanguage.BASH,
+                CellLanguage.ZSH,
+            ):  # noqa: SIM102
                 if not self._validate_shell_syntax(code, i):
                     all_valid = False
 
@@ -281,7 +285,7 @@ class CodeLinter:
             try:
                 # Try to use shellcheck
                 result = subprocess.run(
-                    ["shellcheck", "-f", "json", temp_path],
+                    ["shellcheck", "-", "json", temp_path],
                     capture_output=True,
                     text=True,
                     timeout=10,

mcli/workflow/openai/openai.py

@@ -43,7 +43,7 @@ class OpenAI:
 
         See https://platform.openai.com/docs/guides/moderation/quickstart for details.
         """
-        allowed_categories = {"violence"}  # Can be triggered by some AI safety terms
+        _allowed_categories = {"violence"}  # Can be triggered by some AI safety terms  # noqa: F841
 
         response = None
         try:
@@ -79,7 +79,6 @@ class OpenAI:
 @click.group(name="openai")
 def openai():
     """OpenAI CLI command group."""
-    pass
 
 
 @openai.command(name="is_text_risky")

mcli/workflow/registry/registry.py

@@ -7,6 +7,10 @@ import requests
 warnings.filterwarnings("ignore", message="Using slow pure-python SequenceMatcher")
 from fuzzywuzzy import process
 
+from mcli.lib.logger.logger import get_logger
+
+logger = get_logger(__name__)
+
 """_summary_
     ssh -o GatewayPorts=yes -o ServerAliveInterval=60 -o ProxyCommand="ssh -W %h:%p myuser@my-proxy-server" -L80::80 -L443::443 myuser@localhost
 Returns:
@@ -56,7 +60,6 @@ class DockerClient:
                     logger.info(f"{repo}: {matching_tuples}")
             except Exception as e:
                 logger.info(e)
-                pass
         return response
 
     def count_images(self, repository):

mcli/workflow/repo/repo.py

@@ -14,14 +14,14 @@ logger = get_logger(__name__)
 
 @click.group(name="repo")
 def repo():
-    """repo utility - use this to interact with git and relevant utilities"""
+    """repo utility - use this to interact with git and relevant utilities."""
     click.echo("repo")
 
 
 @repo.command()
 @click.argument("path")
 def analyze(path: str):
-    """Provides a source lines of code analysis for a given pkg path"""
+    """Provides a source lines of code analysis for a given pkg path."""
     _analyze(path)
 
 
@@ -68,7 +68,6 @@ def _analyze(path: str):
                         max_length = len(cell.value)
                 except Exception as e:
                     logger.info(e)
-                    pass
             adjusted_width = max_length + 2
             worksheet.column_dimensions[column].width = adjusted_width
             for cell in col:
@@ -147,20 +146,20 @@ def _analyze(path: str):
 
 @repo.command(name="wt")
 def worktree():
-    """Create and manage worktrees"""
+    """Create and manage worktrees."""
     scripts_path = get_shell_script_path("repo", __file__)
     shell_exec(scripts_path, "wt")
 
 
 @repo.command(name="commit")
 def commit():
-    """Edit commits to a repository"""
+    """Edit commits to a repository."""
     click.echo("commit")
 
 
 @repo.command(name="revert")
 def revert():
-    """Create and manage worktrees"""
+    """Create and manage worktrees."""
     scripts_path = get_shell_script_path("repo", __file__)
     shell_exec(scripts_path, "revert")
 
@@ -169,7 +168,7 @@ def revert():
 @click.argument("branch-a")
 @click.argument("branch-b")
 def loe(branch_a: str, branch_b: str):
-    """Create and manage worktrees"""
+    """Create and manage worktrees."""
     scripts_path = get_shell_script_path("repo", __file__)
     result = shell_exec(scripts_path, "migration_loe", branch_a, branch_b)
     # Assume result['result'] contains the output from the shell script

mcli/workflow/scheduler/cron_parser.py

@@ -7,7 +7,6 @@ Supports standard cron expressions with some extensions:
 - Special: @reboot (run at scheduler start)
 """
 
-import re
 from datetime import datetime, timedelta
 from typing import List, Optional, Set
 
@@ -17,13 +16,11 @@ logger = get_logger(__name__)
 
 
 class CronParseError(Exception):
-    """Exception raised when cron expression cannot be parsed"""
-
-    pass
+    """Exception raised when cron expression cannot be parsed."""
 
 
 class CronExpression:
-    """Parser and calculator for cron expressions"""
+    """Parser and calculator for cron expressions."""
 
     # Predefined cron shortcuts
     SHORTCUTS = {
@@ -47,7 +44,7 @@ class CronExpression:
             self._validate_fields()
 
     def _normalize_expression(self, expression: str) -> str:
-        """Convert shortcuts to standard cron format"""
+        """Convert shortcuts to standard cron format."""
         expression = expression.strip().lower()
 
         if expression in self.SHORTCUTS:
@@ -56,7 +53,7 @@ class CronExpression:
         return expression
 
     def _parse_expression(self) -> List[Set[int]]:
-        """Parse cron expression into field sets"""
+        """Parse cron expression into field sets."""
         if self.is_reboot:
             return []
 
@@ -73,14 +70,14 @@ class CronExpression:
             (0, 6),  # weekday (0=Sunday)
         ]
 
-        for i, (part, (min_val, max_val)) in enumerate(zip(parts, ranges)):
+        for _i, (part, (min_val, max_val)) in enumerate(zip(parts, ranges)):
             field_values = self._parse_field(part, min_val, max_val)
             fields.append(field_values)
 
         return fields
 
     def _parse_field(self, field: str, min_val: int, max_val: int) -> Set[int]:
-        """Parse a single cron field"""
+        """Parse a single cron field."""
         if field == "*":
             return set(range(min_val, max_val + 1))
 
@@ -121,12 +118,12 @@ class CronExpression:
         return values
 
     def _validate_fields(self):
-        """Validate parsed cron fields"""
+        """Validate parsed cron fields."""
         if len(self.fields) != 5:
             raise CronParseError("Invalid number of parsed fields")
 
     def get_next_run_time(self, from_time: Optional[datetime] = None) -> Optional[datetime]:
-        """Calculate the next time this cron expression should run"""
+        """Calculate the next time this cron expression should run."""
         if self.is_reboot:
             return None  # @reboot jobs run only at scheduler start
 
@@ -151,7 +148,7 @@ class CronExpression:
         return None
 
     def _matches_time(self, dt: datetime) -> bool:
-        """Check if datetime matches this cron expression"""
+        """Check if datetime matches this cron expression."""
         if self.is_reboot:
             return False
 
@@ -166,11 +163,11 @@ class CronExpression:
         )
 
     def matches_now(self) -> bool:
-        """Check if cron expression matches current time"""
+        """Check if cron expression matches current time."""
         return self._matches_time(datetime.now())
 
     def get_description(self) -> str:
-        """Get human-readable description of cron expression"""
+        """Get human-readable description of cron expression."""
         if self.is_reboot:
             return "Run at scheduler startup"
 
@@ -191,12 +188,12 @@ class CronExpression:
             return f"Custom schedule: {self.original_expression}"
 
     def is_valid(self) -> bool:
-        """Check if cron expression is valid"""
+        """Check if cron expression is valid."""
         try:
             if self.is_reboot:
                 return True
             return len(self.fields) == 5
-        except:
+        except Exception:
             return False
 
     def __str__(self) -> str:
@@ -207,16 +204,16 @@ class CronExpression:
 
 
 def validate_cron_expression(expression: str) -> bool:
-    """Validate a cron expression without creating a full object"""
+    """Validate a cron expression without creating a full object."""
     try:
         cron = CronExpression(expression)
         return cron.is_valid()
-    except:
+    except Exception:
         return False
 
 
 def get_next_run_times(expression: str, count: int = 5) -> List[datetime]:
-    """Get the next N run times for a cron expression"""
+    """Get the next N run times for a cron expression."""
     try:
         cron = CronExpression(expression)
         if cron.is_reboot:

mcli/workflow/scheduler/job.py

@@ -6,7 +6,7 @@ import json
 import uuid
 from datetime import datetime, timedelta
 from enum import Enum
-from typing import Any, Dict, List, Optional
+from typing import Any, Dict, Optional
 
 from mcli.lib.logger.logger import get_logger
 
@@ -14,7 +14,7 @@ logger = get_logger(__name__)
 
 
 class JobStatus(Enum):
-    """Job execution status"""
+    """Job execution status."""
 
     PENDING = "pending"
     RUNNING = "running"
@@ -25,7 +25,7 @@ class JobStatus(Enum):
 
 
 class JobType(Enum):
-    """Types of jobs that can be scheduled"""
+    """Types of jobs that can be scheduled."""
 
     COMMAND = "command"  # Execute shell commands
     PYTHON = "python"  # Execute Python code
@@ -36,7 +36,7 @@ class JobType(Enum):
 
 
 class ScheduledJob:
-    """Represents a scheduled job with all its metadata"""
+    """Represents a scheduled job with all its metadata."""
 
     def __init__(
         self,
@@ -84,7 +84,7 @@ class ScheduledJob:
         self.current_retry = 0
 
     def to_dict(self) -> Dict[str, Any]:
-        """Convert job to dictionary for serialization"""
+        """Convert job to dictionary for serialization."""
         return {
             "id": self.id,
             "name": self.name,
@@ -115,7 +115,7 @@ class ScheduledJob:
 
     @classmethod
     def from_dict(cls, data: Dict[str, Any]) -> "ScheduledJob":
-        """Create job from dictionary"""
+        """Create job from dictionary."""
         job = cls(
             name=data["name"],
             cron_expression=data["cron_expression"],
@@ -149,7 +149,7 @@ class ScheduledJob:
         return job
 
     def update_status(self, status: JobStatus, output: str = "", error: str = ""):
-        """Update job status and related metadata"""
+        """Update job status and related metadata."""
         self.status = status
         self.last_output = output
         self.last_error = error
@@ -164,15 +164,15 @@ class ScheduledJob:
             self.failure_count += 1
 
     def should_retry(self) -> bool:
-        """Check if job should be retried after failure"""
+        """Check if job should be retried after failure."""
         return self.status == JobStatus.FAILED and self.current_retry < self.retry_count
 
     def get_next_retry_time(self) -> datetime:
-        """Calculate next retry time"""
+        """Calculate next retry time."""
         return datetime.now() + timedelta(seconds=self.retry_delay)
 
     def to_json(self) -> str:
-        """Convert job to JSON string"""
+        """Convert job to JSON string."""
         return json.dumps(self.to_dict(), indent=2)
 
     def __str__(self) -> str: