janito 3.12.0__py3-none-any.whl → 3.12.2__py3-none-any.whl

janito/cli/cli_commands/list_drivers.py

@@ -1,162 +1,162 @@
-"""
-CLI Command: List available LLM drivers and their dependencies
-"""
-
-import importlib
-import sys
-from pathlib import Path
-from rich.console import Console
-from rich.table import Table
-from rich.panel import Panel
-from rich.text import Text
-
-console = Console()
-
-
-def _detect_dependencies_from_content(content, class_name):
-    """Detect dependencies from module content."""
-    dependencies = []
-
-    if "import openai" in content or "from openai" in content:
-        dependencies.append("openai")
-    if "import zai" in content or "from zai" in content:
-        dependencies.append("zai")
-    if "import anthropic" in content or "from anthropic" in content:
-        dependencies.append("anthropic")
-    if "import google" in content or "from google" in content:
-        dependencies.append("google-generativeai")
-
-    # Remove openai from zai driver dependencies
-    if "ZAIModelDriver" in class_name and "openai" in dependencies:
-        dependencies.remove("openai")
-
-    return dependencies
-
-
-def _check_dependency_status(dependencies):
-    """Check if dependencies are available."""
-    if not dependencies:
-        return ["No external dependencies"]
-
-    dep_status = []
-    for dep in dependencies:
-        try:
-            importlib.import_module(dep)
-            dep_status.append(f"✅ {dep}")
-        except ImportError:
-            dep_status.append(f"❌ {dep}")
-
-    return dep_status
-
-
-def _get_single_driver_info(module_path, class_name):
-    """Get information for a single driver."""
-    try:
-        module = importlib.import_module(module_path)
-        driver_class = getattr(module, class_name)
-
-        available = getattr(driver_class, "available", True)
-        unavailable_reason = getattr(driver_class, "unavailable_reason", None)
-
-        # Read module file to detect imports
-        module_file = Path(module.__file__)
-        with open(module_file, "r", encoding="utf-8") as f:
-            content = f.read()
-
-        dependencies = _detect_dependencies_from_content(content, class_name)
-        dep_status = _check_dependency_status(dependencies)
-
-        return {
-            "name": class_name,
-            "available": available,
-            "reason": unavailable_reason,
-            "dependencies": dep_status,
-        }
-
-    except (ImportError, AttributeError) as e:
-        return {
-            "name": class_name,
-            "module": module_path,
-            "available": False,
-            "reason": str(e),
-            "dependencies": ["❌ Module not found"],
-        }
-
-
-def get_driver_info():
-    """Get information about all available drivers."""
-    drivers = []
-
-    # Define known driver modules
-    driver_modules = [
-        ("janito.drivers.openai.driver", "OpenAIModelDriver"),
-        ("janito.drivers.azure_openai.driver", "AzureOpenAIModelDriver"),
-        ("janito.drivers.zai.driver", "ZAIModelDriver"),
-        ("janito.drivers.cerebras.driver", "CerebrasModelDriver"),
-    ]
-
-    for module_path, class_name in driver_modules:
-        driver_info = _get_single_driver_info(module_path, class_name)
-        drivers.append(driver_info)
-
-    return drivers
-
-
-def _create_driver_table(drivers):
-    """Create and populate the drivers table."""
-    table = Table(title="Available LLM Drivers")
-    table.add_column("Driver", style="cyan", no_wrap=True)
-    table.add_column("Status", style="bold")
-    table.add_column("Dependencies", style="yellow")
-
-    for driver in drivers:
-        name = driver["name"]
-
-        if driver["available"]:
-            status = "[green]✅ Available[/green]"
-            if driver["reason"]:
-                status = f"[yellow]⚠️ Available ({driver['reason']})[/yellow]"
-        else:
-            status = f"[red]❌ Unavailable[/red]"
-            if driver["reason"]:
-                status = f"[red]❌ {driver['reason']}[/red]"
-
-        deps = "\n".join(driver["dependencies"])
-        table.add_row(name, status, deps)
-
-    return table
-
-
-def _get_missing_dependencies(drivers):
-    """Get list of missing dependencies."""
-    missing_deps = []
-    for driver in drivers:
-        for dep_status in driver["dependencies"]:
-            if dep_status.startswith("❌"):
-                dep_name = dep_status.split()[1]
-                if dep_name not in missing_deps:
-                    missing_deps.append(dep_name)
-    return missing_deps
-
-
-def handle_list_drivers(args=None):
-    """List all available LLM drivers with their status and dependencies."""
-    drivers = get_driver_info()
-
-    if not drivers:
-        console.print("[red]No drivers found[/red]")
-        return
-
-    table = _create_driver_table(drivers)
-    console.print(table)
-
-    # Installation help - only show for missing dependencies
-    missing_deps = _get_missing_dependencies(drivers)
-    if missing_deps:
-        console.print(
-            f"\n[dim]💡 Install missing deps: pip install {' '.join(missing_deps)}[/dim]"
-        )
-
-
-if __name__ == "__main__":
-    handle_list_drivers()
+"""
+CLI Command: List available LLM drivers and their dependencies
+"""
+
+import importlib
+import sys
+from pathlib import Path
+from rich.console import Console
+from rich.table import Table
+from rich.panel import Panel
+from rich.text import Text
+
+console = Console()
+
+
+def _detect_dependencies_from_content(content, class_name):
+    """Detect dependencies from module content."""
+    dependencies = []
+
+    if "import openai" in content or "from openai" in content:
+        dependencies.append("openai")
+    if "import zai" in content or "from zai" in content:
+        dependencies.append("zai")
+    if "import anthropic" in content or "from anthropic" in content:
+        dependencies.append("anthropic")
+    if "import google" in content or "from google" in content:
+        dependencies.append("google-generativeai")
+
+    # Remove openai from zai driver dependencies
+    if "ZAIModelDriver" in class_name and "openai" in dependencies:
+        dependencies.remove("openai")
+
+    return dependencies
+
+
+def _check_dependency_status(dependencies):
+    """Check if dependencies are available."""
+    if not dependencies:
+        return ["No external dependencies"]
+
+    dep_status = []
+    for dep in dependencies:
+        try:
+            importlib.import_module(dep)
+            dep_status.append(f"✅ {dep}")
+        except ImportError:
+            dep_status.append(f"❌ {dep}")
+
+    return dep_status
+
+
+def _get_single_driver_info(module_path, class_name):
+    """Get information for a single driver."""
+    try:
+        module = importlib.import_module(module_path)
+        driver_class = getattr(module, class_name)
+
+        available = getattr(driver_class, "available", True)
+        unavailable_reason = getattr(driver_class, "unavailable_reason", None)
+
+        # Read module file to detect imports
+        module_file = Path(module.__file__)
+        with open(module_file, "r", encoding="utf-8") as f:
+            content = f.read()
+
+        dependencies = _detect_dependencies_from_content(content, class_name)
+        dep_status = _check_dependency_status(dependencies)
+
+        return {
+            "name": class_name,
+            "available": available,
+            "reason": unavailable_reason,
+            "dependencies": dep_status,
+        }
+
+    except (ImportError, AttributeError) as e:
+        return {
+            "name": class_name,
+            "module": module_path,
+            "available": False,
+            "reason": str(e),
+            "dependencies": ["❌ Module not found"],
+        }
+
+
+def get_driver_info():
+    """Get information about all available drivers."""
+    drivers = []
+
+    # Define known driver modules
+    driver_modules = [
+        ("janito.drivers.openai.driver", "OpenAIModelDriver"),
+        ("janito.drivers.azure_openai.driver", "AzureOpenAIModelDriver"),
+        ("janito.drivers.zai.driver", "ZAIModelDriver"),
+        ("janito.drivers.cerebras.driver", "CerebrasModelDriver"),
+    ]
+
+    for module_path, class_name in driver_modules:
+        driver_info = _get_single_driver_info(module_path, class_name)
+        drivers.append(driver_info)
+
+    return drivers
+
+
+def _create_driver_table(drivers):
+    """Create and populate the drivers table."""
+    table = Table(title="Available LLM Drivers")
+    table.add_column("Driver", style="cyan", no_wrap=True)
+    table.add_column("Status", style="bold")
+    table.add_column("Dependencies", style="yellow")
+
+    for driver in drivers:
+        name = driver["name"]
+
+        if driver["available"]:
+            status = "[green]✅ Available[/green]"
+            if driver["reason"]:
+                status = f"[yellow]⚠️ Available ({driver['reason']})[/yellow]"
+        else:
+            status = f"[red]❌ Unavailable[/red]"
+            if driver["reason"]:
+                status = f"[red]❌ {driver['reason']}[/red]"
+
+        deps = "\n".join(driver["dependencies"])
+        table.add_row(name, status, deps)
+
+    return table
+
+
+def _get_missing_dependencies(drivers):
+    """Get list of missing dependencies."""
+    missing_deps = []
+    for driver in drivers:
+        for dep_status in driver["dependencies"]:
+            if dep_status.startswith("❌"):
+                dep_name = dep_status.split()[1]
+                if dep_name not in missing_deps:
+                    missing_deps.append(dep_name)
+    return missing_deps
+
+
+def handle_list_drivers(args=None):
+    """List all available LLM drivers with their status and dependencies."""
+    drivers = get_driver_info()
+
+    if not drivers:
+        console.print("[red]No drivers found[/red]")
+        return
+
+    table = _create_driver_table(drivers)
+    console.print(table)
+
+    # Installation help - only show for missing dependencies
+    missing_deps = _get_missing_dependencies(drivers)
+    if missing_deps:
+        console.print(
+            f"\n[dim]💡 Install missing deps: uv pip install {' '.join(missing_deps)}[/dim]"
+        )
+
+
+if __name__ == "__main__":
+    handle_list_drivers()

janito/cli/cli_commands/list_profiles.py

@@ -1,107 +1,104 @@
-"""
-CLI Command: List available system prompt profiles (default and user-specific)
-"""
-
-from pathlib import Path
-import importlib.resources as resources
-from rich.console import Console
-from rich.table import Table
-
-
-_PREFIX = "system_prompt_template_"
-_SUFFIX = ".txt.j2"
-
-
-def _extract_profile_name(filename: str) -> str:
-    """Return the human-readable profile name from template file name."""
-    # Remove prefix & suffix and convert underscores back to spaces
-    if filename.startswith(_PREFIX):
-        filename = filename[len(_PREFIX) :]
-    if filename.endswith(_SUFFIX):
-        filename = filename[: -len(_SUFFIX)]
-
-    # Convert to title case for consistent capitalization, but handle common acronyms
-    name = filename.replace("_", " ")
-
-    # Convert to proper title case with consistent capitalization
-    name = filename.replace("_", " ")
-
-    # Handle special cases and acronyms
-    special_cases = {
-        "python": "Python",
-        "tools": "Tools",
-        "model": "Model",
-        "context": "Context",
-        "developer": "Developer",
-        "analyst": "Analyst",
-        "conversation": "Conversation",
-        "without": "Without",
-    }
-
-    words = name.split()
-    capitalized_words = []
-    for word in words:
-        lower_word = word.lower()
-        if lower_word in special_cases:
-            capitalized_words.append(special_cases[lower_word])
-        else:
-            capitalized_words.append(word.capitalize())
-
-    return " ".join(capitalized_words)
-
-
-def _gather_default_profiles():
-    """Return list of built-in profile names bundled with janito."""
-    profiles = []
-    try:
-        package_files = resources.files("janito.agent.templates.profiles")
-        for path in package_files.iterdir():
-            name = path.name
-            if name.startswith(_PREFIX) and name.endswith(_SUFFIX):
-                profiles.append(_extract_profile_name(name))
-    except Exception:
-        # If for some reason the resources are not available fall back to empty list
-        pass
-    return sorted(profiles, key=str.lower)
-
-
-def _gather_user_profiles():
-    """Return list of user-defined profile names from ~/.janito/profiles directory."""
-    user_dir = Path.home() / ".janito" / "profiles"
-    profiles = []
-    if user_dir.exists() and user_dir.is_dir():
-        for path in user_dir.iterdir():
-            if (
-                path.is_file()
-                and path.name.startswith(_PREFIX)
-                and path.name.endswith(_SUFFIX)
-            ):
-                profiles.append(_extract_profile_name(path.name))
-    return sorted(profiles, key=str.lower)
-
-
-def _print_profiles_table(default_profiles, user_profiles):
-    console = Console()
-    table = Table(title="Available System Prompt Profiles", box=None, show_lines=False)
-    table.add_column("Profile Name", style="cyan", no_wrap=False)
-    table.add_column("Source", style="magenta", no_wrap=True)
-
-    for p in default_profiles:
-        table.add_row(p, "default")
-    for p in user_profiles:
-        table.add_row(p, "user")
-
-    console.print(table)
-
-
-def handle_list_profiles(args=None):
-    """Entry point for the --list-profiles CLI flag."""
-    default_profiles = _gather_default_profiles()
-    user_profiles = _gather_user_profiles()
-
-    if not default_profiles and not user_profiles:
-        print("No profiles found.")
-        return
-
-    _print_profiles_table(default_profiles, user_profiles)
-    return
+"""
+CLI Command: List available system prompt profiles (default and user-specific)
+"""
+
+from pathlib import Path
+import importlib.resources as resources
+from rich.console import Console
+from rich.table import Table
+
+
+_PREFIX = "system_prompt_template_"
+_SUFFIX = ".txt.j2"
+
+
+def _extract_profile_name(filename: str) -> str:
+    """Return the human-readable profile name from template file name."""
+    # Remove prefix & suffix and convert underscores back to spaces
+    if filename.startswith(_PREFIX):
+        filename = filename[len(_PREFIX) :]
+    if filename.endswith(_SUFFIX):
+        filename = filename[: -len(_SUFFIX)]
+
+    # Convert to title case for consistent capitalization, but handle common acronyms
+    name = filename.replace("_", " ")
+
+    # Handle special cases and acronyms
+    special_cases = {
+        "python": "Python",
+        "tools": "Tools",
+        "model": "Model",
+        "context": "Context",
+        "developer": "Developer",
+        "analyst": "Analyst",
+        "conversation": "Conversation",
+        "without": "Without",
+    }
+
+    words = name.split()
+    capitalized_words = []
+    for word in words:
+        lower_word = word.lower()
+        if lower_word in special_cases:
+            capitalized_words.append(special_cases[lower_word])
+        else:
+            capitalized_words.append(word.capitalize())
+
+    return " ".join(capitalized_words)
+
+
+def _gather_default_profiles():
+    """Return list of built-in profile names bundled with janito."""
+    profiles = []
+    try:
+        package_files = resources.files("janito.agent.templates.profiles")
+        for path in package_files.iterdir():
+            name = path.name
+            if name.startswith(_PREFIX) and name.endswith(_SUFFIX):
+                profiles.append(_extract_profile_name(name))
+    except Exception:
+        # If for some reason the resources are not available fall back to empty list
+        pass
+    return sorted(profiles, key=str.lower)
+
+
+def _gather_user_profiles():
+    """Return list of user-defined profile names from ~/.janito/profiles directory."""
+    user_dir = Path.home() / ".janito" / "profiles"
+    profiles = []
+    if user_dir.exists() and user_dir.is_dir():
+        for path in user_dir.iterdir():
+            if (
+                path.is_file()
+                and path.name.startswith(_PREFIX)
+                and path.name.endswith(_SUFFIX)
+            ):
+                profiles.append(_extract_profile_name(path.name))
+    return sorted(profiles, key=str.lower)
+
+
+def _print_profiles_table(default_profiles, user_profiles):
+    console = Console()
+    table = Table(title="Available System Prompt Profiles", box=None, show_lines=False)
+    table.add_column("Profile Name", style="cyan", no_wrap=False)
+    table.add_column("Source", style="magenta", no_wrap=True)
+
+    for p in default_profiles:
+        table.add_row(p, "default")
+    for p in user_profiles:
+        table.add_row(p, "user")
+
+    console.print(table)
+
+
+def handle_list_profiles(args=None):
+    """Entry point for the --list-profiles CLI flag."""
+    default_profiles = _gather_default_profiles()
+    user_profiles = _gather_user_profiles()
+
+    if not default_profiles and not user_profiles:
+        print("No profiles found.")
+        return
+
+    _print_profiles_table(default_profiles, user_profiles)
+    return