pygeai-orchestration 0.1.0b2__py3-none-any.whl

pygeai_orchestration/__init__.py

@@ -0,0 +1,99 @@
+import logging
+
+__author__ = "Globant"
+__version__ = "0.1.0b1"
+
+logger = logging.getLogger("pygeai-orchestration")
+logger.addHandler(logging.NullHandler())
+
+from .core import (
+    BaseAgent,
+    AgentConfig,
+    BaseOrchestrator,
+    OrchestratorConfig,
+    BasePattern,
+    PatternConfig,
+    PatternResult,
+    PatternType,
+    BaseTool,
+    ToolConfig,
+    ToolResult,
+    ToolCategory,
+    GEAIAgent,
+    GEAIOrchestrator,
+    Message,
+    MessageRole,
+    Conversation,
+    Context,
+    ContextManager,
+    State,
+    StateStatus,
+    StateManager,
+    Memory,
+    MemoryEntry,
+    MemoryStore,
+    get_logger,
+    get_config,
+    OrchestrationError,
+    PatternExecutionError,
+    PatternConfigurationError,
+    AgentError,
+    ToolExecutionError,
+    StateError,
+    ValidationError,
+)
+
+from .patterns import (
+    ReflectionPattern,
+    ToolUsePattern,
+    ReActPattern,
+    PlanningPattern,
+    PlanStep,
+    MultiAgentPattern,
+    AgentRole,
+)
+
+__all__ = [
+    "__version__",
+    "BaseAgent",
+    "AgentConfig",
+    "BaseOrchestrator",
+    "OrchestratorConfig",
+    "BasePattern",
+    "PatternConfig",
+    "PatternResult",
+    "PatternType",
+    "BaseTool",
+    "ToolConfig",
+    "ToolResult",
+    "ToolCategory",
+    "GEAIAgent",
+    "GEAIOrchestrator",
+    "Message",
+    "MessageRole",
+    "Conversation",
+    "Context",
+    "ContextManager",
+    "State",
+    "StateStatus",
+    "StateManager",
+    "Memory",
+    "MemoryEntry",
+    "MemoryStore",
+    "get_logger",
+    "get_config",
+    "OrchestrationError",
+    "PatternExecutionError",
+    "PatternConfigurationError",
+    "AgentError",
+    "ToolExecutionError",
+    "StateError",
+    "ValidationError",
+    "ReflectionPattern",
+    "ToolUsePattern",
+    "ReActPattern",
+    "PlanningPattern",
+    "PlanStep",
+    "MultiAgentPattern",
+    "AgentRole",
+]

pygeai_orchestration/cli/__init__.py

@@ -0,0 +1,7 @@
+"""
+CLI module for PyGEAI-Orchestration.
+
+Provides the geai-orch command-line interface for orchestration patterns.
+"""
+
+__all__ = []

pygeai_orchestration/cli/__main__.py

@@ -0,0 +1,11 @@
+import sys
+
+from pygeai_orchestration.cli.geai_orch import main as geai_orch
+
+
+def main():
+    sys.exit(geai_orch())
+
+
+if __name__ == "__main__":
+    main()

pygeai_orchestration/cli/commands/__init__.py

@@ -0,0 +1,13 @@
+"""
+CLI commands module.
+
+Re-exports command infrastructure from PyGEAI to avoid code duplication.
+"""
+
+from pygeai.cli.commands import Command, Option, ArgumentsEnum
+
+__all__ = [
+    "Command",
+    "Option",
+    "ArgumentsEnum",
+]

pygeai_orchestration/cli/commands/base.py

@@ -0,0 +1,192 @@
+"""
+Base commands for the geai-orch CLI.
+"""
+
+from typing import List, Tuple
+
+from pygeai_orchestration import __version__
+from pygeai_orchestration.cli.commands import Command, Option, ArgumentsEnum
+from pygeai_orchestration.cli.texts.help import (
+    CLI_USAGE,
+    VERSION_TEXT,
+    HELP_REFLECTION,
+    HELP_TOOL_USE,
+    HELP_REACT,
+    HELP_PLANNING,
+    HELP_MULTI_AGENT,
+)
+
+
+def show_help(option_list: List[Tuple[Option, str]] = None):
+    """
+    Display help information for geai-orch CLI.
+
+    :param option_list: Optional list of options (unused, for compatibility)
+    """
+    print(CLI_USAGE)
+
+
+def show_version(option_list: List[Tuple[Option, str]] = None):
+    """
+    Display version information.
+
+    :param option_list: Optional list of options (unused, for compatibility)
+    """
+    print(VERSION_TEXT.format(version=__version__))
+
+
+def show_reflection_help(option_list: List[Tuple[Option, str]] = None):
+    """Display help for reflection command."""
+    print(HELP_REFLECTION)
+
+
+def show_tool_use_help(option_list: List[Tuple[Option, str]] = None):
+    """Display help for tool-use command."""
+    print(HELP_TOOL_USE)
+
+
+def show_react_help(option_list: List[Tuple[Option, str]] = None):
+    """Display help for react command."""
+    print(HELP_REACT)
+
+
+def show_planning_help(option_list: List[Tuple[Option, str]] = None):
+    """Display help for planning command."""
+    print(HELP_PLANNING)
+
+
+def show_multi_agent_help(option_list: List[Tuple[Option, str]] = None):
+    """Display help for multi-agent command."""
+    print(HELP_MULTI_AGENT)
+
+
+def placeholder_handler(option_list: List[Tuple[Option, str]] = None):
+    """
+    Placeholder handler for commands not yet implemented.
+
+    :param option_list: List of parsed options
+    """
+    print("⚠️  This command is not yet implemented.")
+    print("This feature is under development and will be available soon.")
+
+
+help_command = Command(
+    name="help",
+    identifiers=["help", "-h", "--help"],
+    description="Show help information",
+    action=show_help,
+    additional_args=ArgumentsEnum.NOT_AVAILABLE,
+    subcommands=[],
+    options=[],
+)
+
+version_command = Command(
+    name="version",
+    identifiers=["version", "-V", "--version"],
+    description="Show version information",
+    action=show_version,
+    additional_args=ArgumentsEnum.NOT_AVAILABLE,
+    subcommands=[],
+    options=[],
+)
+
+reflection_command = Command(
+    name="reflection",
+    identifiers=["reflection", "reflect"],
+    description="Run reflection pattern for iterative improvement",
+    action=placeholder_handler,
+    additional_args=ArgumentsEnum.OPTIONAL,
+    subcommands=[],
+    options=[
+        Option("agent", ["-a", "--agent"], "Agent ID to use for reflection", True),
+        Option("iterations", ["-i", "--iterations"], "Number of reflection iterations", True),
+        Option("task", ["-t", "--task"], "Task or text to improve", True),
+        Option("output", ["-o", "--output"], "Output file for results", True),
+        Option("verbose", ["-v", "--verbose"], "Show detailed output", False),
+        Option("help", ["-h", "--help"], "Show help for this command", False),
+    ],
+)
+
+tool_use_command = Command(
+    name="tool-use",
+    identifiers=["tool-use", "tools"],
+    description="Execute tool use pattern with function calling",
+    action=placeholder_handler,
+    additional_args=ArgumentsEnum.OPTIONAL,
+    subcommands=[],
+    options=[
+        Option("agent", ["-a", "--agent"], "Agent ID to use", True),
+        Option("tools", ["-t", "--tools"], "Tools configuration file", True),
+        Option("task", ["--task"], "Task description", True),
+        Option("output", ["-o", "--output"], "Output file for results", True),
+        Option("verbose", ["-v", "--verbose"], "Show detailed output", False),
+        Option("help", ["-h", "--help"], "Show help for this command", False),
+    ],
+)
+
+react_command = Command(
+    name="react",
+    identifiers=["react"],
+    description="Run ReAct (Reasoning + Acting) pattern",
+    action=placeholder_handler,
+    additional_args=ArgumentsEnum.OPTIONAL,
+    subcommands=[],
+    options=[
+        Option("agent", ["-a", "--agent"], "Agent ID to use", True),
+        Option("task", ["-t", "--task"], "Task to solve", True),
+        Option("max_steps", ["--max-steps"], "Maximum reasoning-acting steps", True),
+        Option("output", ["-o", "--output"], "Output file for results", True),
+        Option("verbose", ["-v", "--verbose"], "Show detailed reasoning trace", False),
+        Option("help", ["-h", "--help"], "Show help for this command", False),
+    ],
+)
+
+planning_command = Command(
+    name="planning",
+    identifiers=["planning", "plan"],
+    description="Execute planning pattern for multi-step tasks",
+    action=placeholder_handler,
+    additional_args=ArgumentsEnum.OPTIONAL,
+    subcommands=[],
+    options=[
+        Option("agent", ["-a", "--agent"], "Planner agent ID", True),
+        Option("task", ["-t", "--task"], "Task to plan and execute", True),
+        Option("plan_only", ["--plan-only"], "Only create plan, don't execute", False),
+        Option("output", ["-o", "--output"], "Output file for results", True),
+        Option("verbose", ["-v", "--verbose"], "Show detailed execution trace", False),
+        Option("help", ["-h", "--help"], "Show help for this command", False),
+    ],
+)
+
+multi_agent_command = Command(
+    name="multi-agent",
+    identifiers=["multi-agent", "multi"],
+    description="Run multi-agent coordination pattern",
+    action=placeholder_handler,
+    additional_args=ArgumentsEnum.OPTIONAL,
+    subcommands=[],
+    options=[
+        Option("config", ["-c", "--config"], "Agent configuration file", True),
+        Option("task", ["-t", "--task"], "Task for agents to collaborate on", True),
+        Option("strategy", ["--strategy"], "Coordination strategy", True),
+        Option("output", ["-o", "--output"], "Output file for results", True),
+        Option("verbose", ["-v", "--verbose"], "Show agent communication details", False),
+        Option("help", ["-h", "--help"], "Show help for this command", False),
+    ],
+)
+
+base_commands = [
+    help_command,
+    version_command,
+    reflection_command,
+    tool_use_command,
+    react_command,
+    planning_command,
+    multi_agent_command,
+]
+
+base_options = [
+    Option("alias", ["-a", "--alias"], "Use a specific credentials profile", True),
+    Option("credentials", ["--credentials", "--creds"], "Path to custom credentials file", True),
+    Option("verbose", ["-v", "--verbose"], "Enable verbose logging", False),
+]

pygeai_orchestration/cli/error_handler.py

@@ -0,0 +1,123 @@
+"""
+Error handling for the geai-orch CLI.
+"""
+
+from typing import List
+from enum import IntEnum
+
+
+class ExitCode(IntEnum):
+    """Exit codes for the CLI."""
+
+    SUCCESS = 0
+    USER_INPUT_ERROR = 1
+    MISSING_REQUIREMENT = 2
+    SERVICE_ERROR = 3
+    ORCHESTRATION_ERROR = 4
+    KEYBOARD_INTERRUPT = 130
+    UNEXPECTED_ERROR = 255
+
+
+class ErrorHandler:
+    """Handles error formatting and display for the CLI."""
+
+    @staticmethod
+    def format_error(title: str, message: str) -> str:
+        """
+        Format an error message with a title.
+
+        :param title: str - Error title
+        :param message: str - Error message
+        :return: str - Formatted error message
+        """
+        return f"\n❌ {title}\n{message}\n"
+
+    @staticmethod
+    def handle_unknown_command(arg: str, available_commands: List) -> str:
+        """
+        Handle unknown command error.
+
+        :param arg: str - The unknown argument
+        :param available_commands: List - List of available commands
+        :return: str - Formatted error message
+        """
+        commands_str = ", ".join([cmd.name for cmd in available_commands])
+        message = (
+            f"'{arg}' is not a recognized command.\n\n"
+            f"Available commands: {commands_str}\n\n"
+            f"Use 'geai-orch help' for more information."
+        )
+        return ErrorHandler.format_error("Unknown Command", message)
+
+    @staticmethod
+    def handle_unknown_option(arg: str, available_options: List) -> str:
+        """
+        Handle unknown option error.
+
+        :param arg: str - The unknown argument
+        :param available_options: List - List of available options
+        :return: str - Formatted error message
+        """
+        options_str = ", ".join(["/".join(opt.identifiers) for opt in available_options])
+        message = (
+            f"'{arg}' is not a recognized option.\n\n"
+            f"Available options: {options_str}\n\n"
+            f"Use 'geai-orch help' for more information."
+        )
+        return ErrorHandler.format_error("Unknown Option", message)
+
+    @staticmethod
+    def handle_wrong_argument(message: str, usage: str) -> str:
+        """
+        Handle wrong argument error.
+
+        :param message: str - Error message
+        :param usage: str - Usage information
+        :return: str - Formatted error message
+        """
+        full_message = f"{message}\n\n{usage}"
+        return ErrorHandler.format_error("Invalid Argument", full_message)
+
+    @staticmethod
+    def handle_missing_requirement(message: str) -> str:
+        """
+        Handle missing requirement error.
+
+        :param message: str - Error message
+        :return: str - Formatted error message
+        """
+        return ErrorHandler.format_error("Missing Requirement", message)
+
+    @staticmethod
+    def handle_orchestration_error(message: str) -> str:
+        """
+        Handle orchestration-specific errors.
+
+        :param message: str - Error message
+        :return: str - Formatted error message
+        """
+        return ErrorHandler.format_error("Orchestration Error", message)
+
+    @staticmethod
+    def handle_keyboard_interrupt() -> str:
+        """
+        Handle keyboard interrupt (Ctrl+C).
+
+        :return: str - Formatted message
+        """
+        return "\n\n⚠️  Operation cancelled by user.\n"
+
+    @staticmethod
+    def handle_unexpected_error(exception: Exception) -> str:
+        """
+        Handle unexpected errors.
+
+        :param exception: Exception - The exception that occurred
+        :return: str - Formatted error message
+        """
+        message = (
+            f"An unexpected error occurred: {str(exception)}\n\n"
+            f"Please report this issue at:\n"
+            f"https://github.com/genexus-books/pygeai-orchestration/issues"
+        )
+        return ErrorHandler.format_error("Unexpected Error", message)