flock-core 0.4.542__py3-none-any.whl → 0.5.0__py3-none-any.whl

flock/cli/load_flock.py

@@ -1,192 +0,0 @@
-"""Load a Flock from a file."""
-
-from pathlib import Path
-
-import questionary
-from rich.console import Console
-from rich.markdown import Markdown
-
-from flock.cli.loaded_flock_cli import start_loaded_flock_cli
-from flock.core.flock import Flock
-from flock.core.logging.logging import get_logger
-
-logger = get_logger("cli.load_flock")
-
-
-def filter(file_path) -> bool:
-    """Filter function for file selection."""
-    path = Path(file_path)
-    if path.is_dir():
-        return True
-    return path.is_file() and (
-        path.suffix == ".flock"
-        or path.suffix == ".yaml"
-        or path.suffix == ".yml"
-    )
-
-
-def load_flock():
-    """Load a Flock from a file."""
-    console = Console()
-
-    console.print(
-        "\nPlease select a *.flock, *.yaml, or *.yml file\n", style="bold green"
-    )
-
-    result = questionary.path("", file_filter=filter).ask()
-
-    if not result:
-        return
-
-    selected_file = Path(result)
-    if selected_file.is_file():
-        console.print(f"Selected file: {selected_file}", style="bold green")
-
-        try:
-            # Try loading with detailed error handling
-            try:
-                logger.info(f"Attempting to load Flock from: {result}")
-                flock = Flock.load_from_file(result)
-            except ImportError as e:
-                # Handle missing module path errors
-                if "No module named" in str(e):
-                    console.print(
-                        f"[yellow]Warning: Module import failed: {e}[/]"
-                    )
-                    console.print(
-                        "[yellow]Trying file path fallback mechanism...[/]"
-                    )
-                    # Re-try loading with fallback
-                    flock = Flock.load_from_file(result)
-                else:
-                    raise  # Re-raise if it's not a missing module error
-            except KeyError as e:
-                # This could be caused by missing tool references
-                if "__callable_ref__" in str(e):
-                    console.print(
-                        f"[yellow]Warning: Tool reference error: {e}[/]"
-                    )
-                    console.print(
-                        "[yellow]This may be due to missing tool registrations. Attempting to scan for tools...[/]"
-                    )
-                    # Scan for tools and retry
-                    from flock.cli.registry_management import (
-                        auto_registration_scanner,
-                    )
-
-                    auto_registration_scanner()
-                    # Try loading again
-                    flock = Flock.load_from_file(result)
-                else:
-                    raise  # Re-raise if it's not a tool reference error
-
-            console.line()
-            console.print(
-                Markdown("# Flock loaded successfully"), style="bold green"
-            )
-            console.line()
-
-            # Instead of just running the Flock, start our enhanced CLI
-            start_loaded_flock_cli(
-                flock, server_name=f"Flock - {selected_file.name}"
-            )
-
-        except Exception as e:
-            console.print(f"Error loading Flock: {e!s}", style="bold red")
-            logger.error(f"Failed to load Flock: {e}", exc_info=True)
-
-            # Add more detailed error information for specific errors
-            if "No module named" in str(e):
-                console.print(
-                    "\n[yellow]This error might be due to missing module paths.[/]"
-                )
-                console.print(
-                    "[yellow]Component references may need to be updated with file paths.[/]"
-                )
-
-                # Show the option to scan the directory for components
-                fix_paths = questionary.confirm(
-                    "Would you like to scan directories for components to fix missing imports?",
-                    default=True,
-                ).ask()
-
-                if fix_paths:
-                    from flock.cli.registry_management import (
-                        auto_registration_scanner,
-                    )
-
-                    auto_registration_scanner()
-
-                    # Try loading again
-                    console.print(
-                        "\n[yellow]Attempting to load Flock again...[/]"
-                    )
-                    try:
-                        flock = Flock.load_from_file(result)
-                        console.line()
-                        console.print(
-                            Markdown(
-                                "# Flock loaded successfully after component scan"
-                            ),
-                            style="bold green",
-                        )
-                        console.line()
-
-                        start_loaded_flock_cli(
-                            flock, server_name=f"Flock - {selected_file.name}"
-                        )
-                        return
-                    except Exception as e2:
-                        console.print(
-                            f"Error loading Flock after scan: {e2!s}",
-                            style="bold red",
-                        )
-
-            # Handle tool reference issues
-            elif "__callable_ref__" in str(e) or "callable" in str(e).lower():
-                console.print(
-                    "\n[yellow]This error might be due to missing tool registrations.[/]"
-                )
-
-                # Show the option to scan the directory for tools
-                fix_tools = questionary.confirm(
-                    "Would you like to scan directories for tools to fix missing references?",
-                    default=True,
-                ).ask()
-
-                if fix_tools:
-                    from flock.cli.registry_management import (
-                        auto_registration_scanner,
-                    )
-
-                    console.print(
-                        "\n[yellow]Scanning for tools and callables...[/]"
-                    )
-                    auto_registration_scanner()
-
-                    # Try loading again
-                    console.print(
-                        "\n[yellow]Attempting to load Flock again...[/]"
-                    )
-                    try:
-                        flock = Flock.load_from_file(result)
-                        console.line()
-                        console.print(
-                            Markdown(
-                                "# Flock loaded successfully after tool scan"
-                            ),
-                            style="bold green",
-                        )
-                        console.line()
-
-                        start_loaded_flock_cli(
-                            flock, server_name=f"Flock - {selected_file.name}"
-                        )
-                        return
-                    except Exception as e2:
-                        console.print(
-                            f"Error loading Flock after tool scan: {e2!s}",
-                            style="bold red",
-                        )
-
-            input("\nPress Enter to continue...")

flock/cli/load_release_notes.py

@@ -1,20 +0,0 @@
-from pathlib import Path
-
-
-def load_release_notes():
-    """Load release notes."""
-    from rich.console import Console
-    from rich.markdown import Markdown
-
-    from flock.core.util.cli_helper import display_hummingbird, init_console
-
-    console = Console()
-    file_path = Path(__file__).parent / "assets" / "release_notes.md"
-
-    init_console()
-    console.print(Markdown("# *'Hummingbird'* Release Notes"))
-    display_hummingbird()
-    with open(file_path) as file:
-        release_notes = file.read()
-
-    console.print(Markdown(release_notes))

flock/cli/loaded_flock_cli.py

@@ -1,254 +0,0 @@
-"""CLI interface for working with a loaded Flock instance.
-
-This module provides a CLI interface for a Flock instance that has already been loaded,
-allowing users to execute, edit, or manage agents from the existing configuration.
-"""
-
-import questionary
-from rich.console import Console
-from rich.panel import Panel
-
-from flock.cli.constants import (
-    CLI_REGISTRY_MANAGEMENT,
-    CLI_SETTINGS,
-)
-from flock.core.flock import Flock
-from flock.core.logging.logging import get_logger
-from flock.core.util.cli_helper import init_console
-
-# Import future modules we'll create
-# These will be implemented later
-try:
-    from flock.cli.yaml_editor import yaml_editor
-
-    yaml_editor_available = True
-except ImportError:
-    yaml_editor_available = False
-
-try:
-    from flock.cli.manage_agents import manage_agents
-
-    manage_agents_available = True
-except ImportError:
-    manage_agents_available = False
-
-try:
-    from flock.cli.execute_flock import execute_flock, execute_flock_batch
-
-    execute_flock_available = True
-except ImportError:
-    execute_flock_available = False
-
-try:
-    from flock.cli.view_results import view_results
-
-    view_results_available = True
-except ImportError:
-    view_results_available = False
-
-# Create console instance
-console = Console()
-logger = get_logger("cli.loaded_flock")
-
-
-def start_loaded_flock_cli(
-    flock: Flock,
-    server_name: str = "Flock CLI",
-    show_results: bool = False,
-    edit_mode: bool = False,
-) -> None:
-    """Start a CLI interface with a loaded Flock instance.
-
-    Args:
-        flock: The loaded Flock instance
-        server_name: Optional name for the CLI interface
-        show_results: Whether to initially show results of previous runs
-        edit_mode: Whether to open directly in edit mode
-    """
-    if not flock:
-        console.print("[bold red]Error: No Flock instance provided.[/]")
-        return
-
-    agent_names = list(flock._agents.keys())
-
-    # Directly go to specific modes if requested
-    if edit_mode and yaml_editor_available:
-        yaml_editor(flock)
-        return
-
-    if show_results and view_results_available:
-        view_results(flock)
-        return
-
-    # Main CLI loop
-    while True:
-        # Initialize console for each loop iteration
-        init_console()
-
-        # Display header with Flock information
-        console.print(Panel(f"[bold green]{server_name}[/]"), justify="center")
-        console.print(
-            f"Flock loaded with [bold cyan]{len(agent_names)}[/] agents: {', '.join(agent_names)}"
-        )
-        console.line()
-
-        # Main menu choices
-        choices = [
-            questionary.Separator(line=" "),
-            "Execute Flock",
-            "Execute Flock - Batch Mode",
-            "Start Web Server",
-            "Start Web Server with UI",
-            "Manage Agents",
-            "View Results of Past Runs",
-        ]
-
-        # Add YAML Editor option if available
-        if yaml_editor_available:
-            choices.append("Edit YAML Configurations")
-
-        # Add remaining options
-        choices.extend([questionary.Separator(), CLI_REGISTRY_MANAGEMENT])
-        choices.extend(
-            [
-                questionary.Separator(),
-                CLI_SETTINGS,
-                questionary.Separator(),
-                "Exit",
-            ]
-        )
-
-        # Display menu and get choice
-        choice = questionary.select(
-            "What would you like to do?",
-            choices=choices,
-        ).ask()
-
-        # Handle menu selection
-        if choice == "Execute Flock":
-            if execute_flock_available:
-                execute_flock(flock)
-            else:
-                console.print(
-                    "[yellow]Execute Flock functionality not yet implemented.[/]"
-                )
-                input("\nPress Enter to continue...")
-
-        elif choice == "Execute Flock - Batch Mode":
-            if execute_flock_available:
-                execute_flock_batch(flock)
-            else:
-                console.print(
-                    "[yellow]Batch execution functionality not yet implemented.[/]"
-                )
-                input("\nPress Enter to continue...")
-
-        elif choice == "Start Web Server":
-            _start_web_server(flock, create_ui=False)
-
-        elif choice == "Start Web Server with UI":
-            _start_web_server(flock, create_ui=True)
-
-        elif choice == "Manage Agents":
-            if manage_agents_available:
-                manage_agents(flock)
-            else:
-                console.print(
-                    "[yellow]Manage Agents functionality not yet implemented.[/]"
-                )
-                input("\nPress Enter to continue...")
-
-        elif choice == "View Results of Past Runs":
-            if view_results_available:
-                view_results(flock)
-            else:
-                console.print(
-                    "[yellow]View Results functionality not yet implemented.[/]"
-                )
-                input("\nPress Enter to continue...")
-
-        elif choice == CLI_REGISTRY_MANAGEMENT:
-            from flock.cli.registry_management import manage_registry
-
-            manage_registry()
-
-        elif choice == "Edit YAML Configurations" and yaml_editor_available:
-            yaml_editor(flock)
-
-        elif choice == CLI_SETTINGS:
-            from flock.cli.settings import settings_editor
-
-            settings_editor()
-
-        elif choice == "Exit":
-            break
-
-        # Pause after each action unless we're exiting
-        if choice != "Exit" and not choice.startswith("Start Web Server"):
-            input("\nPress Enter to continue...")
-
-
-def _start_web_server(flock: Flock, create_ui: bool = False) -> None:
-    """Start a web server with the loaded Flock instance."""
-    host = "127.0.0.1"
-    port = 8344
-    # server_name = flock.name + " API" # Use flock name by default for server_name
-
-    console.print("\n[bold]Web Server Configuration[/]")
-
-    host_input = questionary.text(
-        "Host (default: 127.0.0.1):", default=host
-    ).ask()
-    if host_input:
-        host = host_input
-
-    port_input = questionary.text(
-        "Port (default: 8344):", default=str(port)
-    ).ask()
-    if port_input and port_input.isdigit():
-        port = int(port_input)
-
-    server_name_input = questionary.text(
-        "Server name (default: FlockName API):", default=f"{flock.name or 'Flock'} API"
-    ).ask()
-    # if server_name_input: # server_name will be set by flock.start_api if not passed, or use its default
-    # server_name = server_name_input
-
-    ui_theme_to_pass = None
-    if create_ui:
-        try:
-            from flock.webapp.app.config import (
-                DEFAULT_THEME_NAME,
-                list_available_themes,
-            )
-            available_themes = list_available_themes()
-            theme_choices = ["default (current environment setting)", "random"] + sorted(available_themes)
-
-            selected_theme_choice = questionary.select(
-                "Select UI theme:",
-                choices=theme_choices,
-                default="default (current environment setting)"
-            ).ask()
-
-            if selected_theme_choice == "random":
-                ui_theme_to_pass = "random"
-            elif selected_theme_choice and selected_theme_choice != "default (current environment setting)":
-                ui_theme_to_pass = selected_theme_choice
-            # If "default" or None, ui_theme_to_pass remains None, Flock.start_api will use its logic
-
-        except ImportError:
-            logger.warning("Could not import webapp theme configuration for CLI selection. Theme will use default.")
-            ui_theme_to_pass = None # Fallback if webapp components not there
-
-    console.print(
-        f"\nStarting web server on {host}:{port} {'with UI' if create_ui else 'without UI'}{' (Theme: ' + (ui_theme_to_pass if ui_theme_to_pass else 'default') + ')' if create_ui else ''}..."
-    )
-
-    # Call the Flock instance's own start_api method directly
-    flock.start_api(
-        host=host,
-        port=port,
-        server_name=server_name_input, # Pass the chosen server name
-        create_ui=create_ui,
-        ui_theme=ui_theme_to_pass,
-    )