janito 0.11.0__py3-none-any.whl → 0.13.0__py3-none-any.whl

janito/cli/commands.py

@@ -0,0 +1,329 @@
+"""
+Command handling logic for Janito CLI.
+"""
+import sys
+import os
+from typing import Optional, Tuple, Any
+from pathlib import Path
+import typer
+from rich.console import Console
+
+from janito.config import get_config, Config
+
+console = Console()
+
+def validate_parameters(temperature: float) -> None:
+    """
+    Validate temperature parameter.
+    
+    Args:
+        temperature: Temperature value for model generation
+    """
+    try:
+        if temperature < 0.0 or temperature > 1.0:
+            raise ValueError("Temperature must be between 0.0 and 1.0")
+            
+        # We'll use this value directly in the agent initialization but we don't save it to config
+        # Temperature display is hidden
+    except ValueError as e:
+        console.print(f"[bold red]Error:[/bold red] {str(e)}")
+        sys.exit(1)
+
+def handle_reset_config(reset_config: bool, ctx: typer.Context, query: Optional[str]) -> bool:
+    """
+    Handle the --reset-config parameter.
+    
+    Args:
+        reset_config: Whether to reset the configuration
+        ctx: Typer context
+        query: Query string
+        
+    Returns:
+        bool: True if the program should exit after this operation
+    """
+    if reset_config:
+        try:
+            config_path = Path(get_config().workspace_dir) / ".janito" / "config.json"
+            if get_config().reset_config():
+                console.print(f"[bold green]✅ Configuration file removed: {config_path}[/bold green]")
+            else:
+                console.print(f"[bold yellow]⚠️ Configuration file does not exist: {config_path}[/bold yellow]")
+        except Exception as e:
+            console.print(f"[bold red]Error removing configuration file:[/bold red] {str(e)}")
+        
+        # Exit after resetting config if no other operation is requested
+        return ctx.invoked_subcommand is None and not query
+    
+    return False
+
+def handle_workspace(workspace: Optional[str]) -> bool:
+    """
+    Handle the --workspace parameter.
+    
+    Args:
+        workspace: Workspace directory path
+        
+    Returns:
+        bool: True if the program should exit after this operation
+    """
+    if workspace:
+        try:
+            console.print(f"[bold]📂 Setting workspace directory to: {workspace}[/bold]")
+            get_config().workspace_dir = workspace
+            console.print(f"[bold green]✅ Workspace directory set to: {get_config().workspace_dir}[/bold green]")
+        except ValueError as e:
+            console.print(f"[bold red]Error:[/bold red] {str(e)}")
+            sys.exit(1)
+    
+    return False
+
+def handle_show_config(show_config: bool, ctx: typer.Context, query: Optional[str]) -> bool:
+    """
+    Handle the --show-config parameter.
+    
+    Args:
+        show_config: Whether to show the configuration
+        ctx: Typer context
+        query: Query string
+        
+    Returns:
+        bool: True if the program should exit after this operation
+    """
+    if show_config:
+        config = get_config()
+        console.print("[bold blue]⚙️  Current Configuration:[/bold blue]")
+        console.print(f"[bold]📁 Local Configuration File:[/bold] .janito/config.json")
+        console.print(f"[bold]🏠 Global Configuration File:[/bold] {Path.home() / '.janito' / 'config.json'}")
+        
+        # Show API key status
+        api_key_global = Config.get_api_key()
+        api_key_env = os.environ.get("ANTHROPIC_API_KEY")
+        if api_key_global:
+            console.print(f"[bold]🔑 API Key:[/bold] [green]Set in global config[/green]")
+        elif api_key_env:
+            console.print(f"[bold]🔑 API Key:[/bold] [yellow]Set in environment variable[/yellow]")
+        else:
+            console.print(f"[bold]🔑 API Key:[/bold] [red]Not set[/red]")
+            
+        console.print(f"[bold]🔊 Verbose Mode:[/bold] {'Enabled' if config.verbose else 'Disabled'}")
+        console.print(f"[bold]❓ Ask Mode:[/bold] {'Enabled' if config.ask_mode else 'Disabled'}")
+
+        console.print(f"[bold]👤 Role:[/bold] {config.role}")
+        
+        # Show profile information if one is set
+        if config.profile:
+            profile_data = config.get_available_profiles()[config.profile]
+            console.print(f"[bold]📋 Active Profile:[/bold] {config.profile} - {profile_data['description']}")
+        
+        # Show available profiles
+        profiles = config.get_available_profiles()
+        if profiles:
+            console.print("\n[bold blue]📋 Available Parameter Profiles:[/bold blue]")
+            for name, data in profiles.items():
+                console.print(f"[bold]🔹 {name}[/bold] - {data['description']}")
+            
+        # Exit if this was the only operation requested
+        return ctx.invoked_subcommand is None and not query
+    
+    return False
+
+def handle_profile(profile: Optional[str], ctx: typer.Context, query: Optional[str]) -> bool:
+    """
+    Handle the --profile parameter.
+    
+    Args:
+        profile: Profile name
+        ctx: Typer context
+        query: Query string
+        
+    Returns:
+        bool: True if the program should exit after this operation
+    """
+    if profile is not None:
+        try:
+            # Apply profile without saving to config
+            config = get_config()
+            profile_data = config.get_available_profiles()[profile.lower()]
+            
+            # Set values directly without saving
+            config._temperature = profile_data["temperature"]
+            config._profile = profile.lower()
+            
+            console.print(f"[bold green]✅ Profile '{profile.lower()}' applied for this session only[/bold green]")
+            console.print(f"[dim]📝 Description: {profile_data['description']}[/dim]")
+            
+            # Exit after applying profile if no other operation is requested
+            return ctx.invoked_subcommand is None and not query
+        except ValueError as e:
+            console.print(f"[bold red]Error:[/bold red] {str(e)}")
+            sys.exit(1)
+    
+    return False
+
+def handle_role(role: Optional[str], ctx: typer.Context, query: Optional[str]) -> bool:
+    """
+    Handle the --role parameter.
+    
+    Args:
+        role: Role name
+        ctx: Typer context
+        query: Query string
+        
+    Returns:
+        bool: True if the program should exit after this operation
+    """
+    if role is not None:
+        try:
+            # Set role directly without saving to config
+            config = get_config()
+            config._role = role
+            
+            console.print(f"[bold green]✅ Role '{role}' applied for this session only[/bold green]")
+            
+            # Exit after applying role if no other operation is requested
+            return ctx.invoked_subcommand is None and not query
+        except Exception as e:
+            console.print(f"[bold red]Error:[/bold red] {str(e)}")
+            sys.exit(1)
+    
+    return False
+
+def handle_set_api_key(set_api_key: Optional[str], ctx: typer.Context, query: Optional[str]) -> bool:
+    """
+    Handle the --set-api-key parameter.
+    
+    Args:
+        set_api_key: API key
+        ctx: Typer context
+        query: Query string
+        
+    Returns:
+        bool: True if the program should exit after this operation
+    """
+    if set_api_key is not None:
+        try:
+            Config.set_api_key(set_api_key)
+            console.print(f"[bold green]✅ API key saved to global configuration[/bold green]")
+            console.print(f"[dim]📁 Location: {Path.home() / '.janito' / 'config.json'}[/dim]")
+            
+            # Exit after setting API key if no other operation is requested
+            return ctx.invoked_subcommand is None and not query
+        except Exception as e:
+            console.print(f"[bold red]Error:[/bold red] {str(e)}")
+            sys.exit(1)
+    
+    return False
+
+def handle_set_config(config_str: Optional[str], ctx: typer.Context, query: Optional[str]) -> bool:
+    """
+    Handle the --set-config parameter.
+    
+    Args:
+        config_str: Configuration string in format 'key=value'
+        ctx: Typer context
+        query: Query string
+        
+    Returns:
+        bool: True if the program should exit after this operation
+    """
+    if config_str is not None:
+        try:
+            # Parse the config string
+            config_parts = config_str.split("=", 1)
+            if len(config_parts) != 2:
+                console.print(f"[bold red]Error:[/bold red] Invalid configuration format. Use 'key=value' format.")
+                return ctx.invoked_subcommand is None and not query
+                
+            key = config_parts[0].strip()
+            value = config_parts[1].strip()
+            
+            # Remove quotes if present
+            if (value.startswith("'") and value.endswith("'")) or \
+               (value.startswith('"') and value.endswith('"')):
+                value = value[1:-1]
+                
+            if key == "profile":
+                try:
+                    get_config().set_profile(value)
+                    profile_data = get_config().get_available_profiles()[value.lower()]
+                    console.print(f"[bold green]✅ Profile set to '{value.lower()}'[/bold green]")
+                    console.print(f"[dim]📝 Description: {profile_data['description']}[/dim]")
+                except ValueError as e:
+                    console.print(f"[bold red]Error:[/bold red] {str(e)}")
+            elif key == "temperature":
+                try:
+                    temp_value = float(value)
+                    if temp_value < 0.0 or temp_value > 1.0:
+                        console.print("[bold red]Error:[/bold red] Temperature must be between 0.0 and 1.0")
+                        return ctx.invoked_subcommand is None and not query
+                    
+                    get_config().temperature = temp_value
+                    console.print(f"[bold green]✅ Temperature set to {temp_value} and saved to configuration[/bold green]")
+                except ValueError:
+                    console.print(f"[bold red]Error:[/bold red] Invalid temperature value: {value}. Must be a float between 0.0 and 1.0.")
+            # top_k and top_p are now only accessible through profiles
+            elif key == "role":
+                get_config().role = value
+                console.print(f"[bold green]✅ Role set to '{value}' and saved to configuration[/bold green]")
+            else:
+                console.print(f"[bold yellow]Warning:[/bold yellow] Unsupported configuration key: {key}")
+            
+            # Exit after applying config changes if no other operation is requested
+            return ctx.invoked_subcommand is None and not query
+        except Exception as e:
+            console.print(f"[bold red]Error:[/bold red] {str(e)}")
+            
+    return False
+
+def handle_config_commands(
+    ctx: typer.Context,
+    reset_config: bool,
+    workspace: Optional[str],
+    show_config: bool,
+    profile: Optional[str],
+    role: Optional[str],
+    set_api_key: Optional[str],
+    config_str: Optional[str],
+    query: Optional[str],
+    continue_conversation: Optional[str] = None
+) -> bool:
+    """
+    Handle all configuration-related commands.
+    
+    Args:
+        ctx: Typer context
+        reset_config: Whether to reset the configuration
+        workspace: Workspace directory path
+        show_config: Whether to show the configuration
+        profile: Profile name
+        role: Role name
+        set_api_key: API key
+        config_str: Configuration string in format 'key=value'
+        query: Query string
+        continue_conversation: Optional message ID to continue a specific conversation
+        
+    Returns:
+        bool: True if the program should exit after these operations
+    """
+    # Handle each command and check if we should exit after it
+    if handle_reset_config(reset_config, ctx, query):
+        return True
+        
+    handle_workspace(workspace)
+    
+    if handle_show_config(show_config, ctx, query):
+        return True
+        
+    if handle_profile(profile, ctx, query):
+        return True
+        
+    if handle_role(role, ctx, query):
+        return True
+        
+    if handle_set_api_key(set_api_key, ctx, query):
+        return True
+        
+    if handle_set_config(config_str, ctx, query):
+        return True
+        
+    return False

janito/cli/output.py

@@ -0,0 +1,29 @@
+"""
+Output formatting and display for Janito CLI.
+"""
+from rich.console import Console
+from janito.config import get_config
+
+console = Console()
+
+def display_generation_params(
+    temp_to_use: float, 
+    profile_data: dict = None,
+    temperature: float = 0.0
+) -> None:
+    """
+    Display generation parameters in verbose mode.
+    
+    Args:
+        temp_to_use: The temperature value being used
+        profile_data: The profile data if a profile is being used
+        temperature: The temperature value from command line
+    """
+    # Show profile information if one is active
+    config = get_config()
+    if config.profile:
+        if not profile_data:
+            profile_data = config.get_available_profiles()[config.profile]
+        console.print(f"[dim]👤 Using profile: {config.profile} - {profile_data['description']}[/dim]")
+    
+    # Temperature, top_k, and top_p information is hidden

janito/cli/utils.py

@@ -0,0 +1,22 @@
+"""
+Utility functions for the CLI module.
+"""
+import platform
+from rich.console import Console
+
+console = Console()
+
+def get_stdin_termination_hint():
+    """
+    Returns a user-friendly message about how to terminate stdin input
+    based on the current platform.
+    
+    Returns:
+        str: A message with the key sequence to terminate stdin input
+    """
+    system = platform.system()
+    
+    if system == "Windows":
+        return "[bold yellow]Press Ctrl+Z followed by Enter to terminate input[/bold yellow]"
+    else:  # Unix-like systems (Linux, macOS)
+        return "[bold yellow]Press Ctrl+D to terminate input[/bold yellow]"