omnimancer-cli 0.1.0__py3-none-any.whl

omnimancer/utils/retry.py

@@ -0,0 +1,238 @@
+"""
+Retry logic utilities for Omnimancer.
+
+This module provides retry mechanisms with exponential backoff
+for handling transient errors in API calls and other operations.
+"""
+
+import asyncio
+import logging
+import random
+from typing import Any, Callable, Optional
+
+from .errors import NetworkError, RateLimitError
+
+logger = logging.getLogger(__name__)
+
+
+class RetryHandler:
+    """
+    Handles retry logic with exponential backoff.
+
+    This class provides configurable retry behavior for operations
+    that may fail due to transient errors like network issues or
+    rate limiting.
+    """
+
+    def __init__(
+        self,
+        max_retries: int = 3,
+        base_delay: float = 1.0,
+        max_delay: float = 60.0,
+        exponential_base: float = 2.0,
+        jitter: bool = True,
+    ):
+        """
+        Initialize the retry handler.
+
+        Args:
+            max_retries: Maximum number of retry attempts
+            base_delay: Initial delay between retries in seconds
+            max_delay: Maximum delay between retries in seconds
+            exponential_base: Base for exponential backoff calculation
+            jitter: Whether to add random jitter to delays
+        """
+        self.max_retries = max_retries
+        self.base_delay = base_delay
+        self.max_delay = max_delay
+        self.exponential_base = exponential_base
+        self.jitter = jitter
+
+    def should_retry(self, error: Exception, attempt: int) -> bool:
+        """
+        Determine if an operation should be retried.
+
+        Args:
+            error: The exception that occurred
+            attempt: Current attempt number (0-indexed)
+
+        Returns:
+            True if the operation should be retried, False otherwise
+        """
+        if attempt >= self.max_retries:
+            return False
+
+        # Always retry network errors
+        if isinstance(error, NetworkError):
+            return True
+
+        # Retry rate limit errors
+        if isinstance(error, RateLimitError):
+            return True
+
+        # Don't retry other types of errors by default
+        return False
+
+    def calculate_delay(
+        self, attempt: int, rate_limit_retry_after: Optional[int] = None
+    ) -> float:
+        """
+        Calculate the delay before the next retry attempt.
+
+        Args:
+            attempt: Current attempt number (0-indexed)
+            rate_limit_retry_after: Specific delay for rate limit errors
+
+        Returns:
+            Delay in seconds
+        """
+        if rate_limit_retry_after is not None:
+            # Use the specific delay provided by the API
+            delay = float(rate_limit_retry_after)
+        else:
+            # Calculate exponential backoff delay
+            delay = self.base_delay * (self.exponential_base**attempt)
+
+        # Cap the delay at max_delay
+        delay = min(delay, self.max_delay)
+
+        # Add jitter to avoid thundering herd
+        if self.jitter:
+            jitter_amount = delay * 0.1  # 10% jitter
+            delay += random.uniform(-jitter_amount, jitter_amount)
+
+        return max(0, delay)
+
+    async def execute_with_retry(
+        self, func: Callable[..., Any], *args, **kwargs
+    ) -> Any:
+        """
+        Execute a function with retry logic.
+
+        Args:
+            func: The function to execute
+            *args: Positional arguments for the function
+            **kwargs: Keyword arguments for the function
+
+        Returns:
+            The result of the function call
+
+        Raises:
+            The last exception if all retries are exhausted
+        """
+        last_exception = None
+
+        for attempt in range(self.max_retries + 1):
+            try:
+                # Execute the function
+                if asyncio.iscoroutinefunction(func):
+                    result = await func(*args, **kwargs)
+                else:
+                    result = func(*args, **kwargs)
+
+                # Success - log if this was a retry
+                if attempt > 0:
+                    logger.info(f"Operation succeeded after {attempt} retries")
+
+                return result
+
+            except Exception as e:
+                last_exception = e
+
+                # Check if we should retry
+                if not self.should_retry(e, attempt):
+                    logger.debug(f"Not retrying after attempt {attempt}: {e}")
+                    break
+
+                # Calculate delay
+                retry_after = None
+                if isinstance(e, RateLimitError):
+                    retry_after = e.retry_after
+
+                delay = self.calculate_delay(attempt, retry_after)
+
+                logger.warning(
+                    f"Attempt {attempt + 1} failed: {e}. "
+                    f"Retrying in {delay:.2f} seconds..."
+                )
+
+                # Wait before retrying
+                await asyncio.sleep(delay)
+
+        # All retries exhausted
+        logger.error(
+            f"All {self.max_retries} retries exhausted. Last error: {last_exception}"
+        )
+        raise last_exception
+
+    def execute_sync_with_retry(self, func: Callable[..., Any], *args, **kwargs) -> Any:
+        """
+        Execute a synchronous function with retry logic.
+
+        Args:
+            func: The synchronous function to execute
+            *args: Positional arguments for the function
+            **kwargs: Keyword arguments for the function
+
+        Returns:
+            The result of the function call
+
+        Raises:
+            The last exception if all retries are exhausted
+        """
+        import time
+
+        last_exception = None
+
+        for attempt in range(self.max_retries + 1):
+            try:
+                result = func(*args, **kwargs)
+
+                # Success - log if this was a retry
+                if attempt > 0:
+                    logger.info(f"Operation succeeded after {attempt} retries")
+
+                return result
+
+            except Exception as e:
+                last_exception = e
+
+                # Check if we should retry
+                if not self.should_retry(e, attempt):
+                    logger.debug(f"Not retrying after attempt {attempt}: {e}")
+                    break
+
+                # Calculate delay
+                retry_after = None
+                if isinstance(e, RateLimitError):
+                    retry_after = e.retry_after
+
+                delay = self.calculate_delay(attempt, retry_after)
+
+                logger.warning(
+                    f"Attempt {attempt + 1} failed: {e}. "
+                    f"Retrying in {delay:.2f} seconds..."
+                )
+
+                # Wait before retrying
+                time.sleep(delay)
+
+        # All retries exhausted
+        logger.error(
+            f"All {self.max_retries} retries exhausted. Last error: {last_exception}"
+        )
+        raise last_exception
+
+    def retry(self, func: Callable[..., Any], *args, **kwargs) -> Any:
+        """
+        Convenience method for synchronous retry execution.
+
+        Args:
+            func: The function to execute with retry
+            *args: Positional arguments for the function
+            **kwargs: Keyword arguments for the function
+
+        Returns:
+            The result of the function call
+        """
+        return self.execute_sync_with_retry(func, *args, **kwargs)

omnimancer_cli-0.1.0.dist-info/METADATA

@@ -0,0 +1,413 @@
+Metadata-Version: 2.4
+Name: omnimancer-cli
+Version: 0.1.0
+Summary: A unified command-line interface for multiple AI language models
+Author-email: Kellan Strong <kellan@jite.ai>
+License: MIT
+Project-URL: Homepage, https://gitlab.com/jite-ai/omnimancer
+Project-URL: Repository, https://gitlab.com/jite-ai/omnimancer
+Project-URL: Issues, https://gitlab.com/jite-ai/omnimancer/issues
+Keywords: ai,cli,chat,openai,claude,language-model
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Classifier: Topic :: Communications :: Chat
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: click>=8.0.0
+Requires-Dist: rich>=13.0.0
+Requires-Dist: httpx>=0.24.0
+Requires-Dist: pydantic>=2.0.0
+Requires-Dist: cryptography>=41.0.0
+Requires-Dist: python-dotenv>=1.0.0
+Requires-Dist: certifi>=2023.0.0
+Requires-Dist: aiofiles>=23.0.0
+Requires-Dist: aiohttp>=3.8.0
+Requires-Dist: psutil>=5.9.0
+Requires-Dist: beautifulsoup4>=4.12.0
+Requires-Dist: html2text>=2020.1.16
+Requires-Dist: jsonschema>=4.17.0
+Requires-Dist: readability-lxml>=0.8.1
+Requires-Dist: PyYAML>=6.0.0
+Requires-Dist: toml>=0.10.2
+Provides-Extra: dev
+Requires-Dist: pytest>=7.0.0; extra == "dev"
+Requires-Dist: pytest-cov>=4.0.0; extra == "dev"
+Requires-Dist: pytest-asyncio>=0.21.0; extra == "dev"
+Requires-Dist: pytest-mock>=3.10.0; extra == "dev"
+Requires-Dist: black>=23.0.0; extra == "dev"
+Requires-Dist: isort>=5.12.0; extra == "dev"
+Requires-Dist: flake8>=6.0.0; extra == "dev"
+Requires-Dist: mypy>=1.0.0; extra == "dev"
+Dynamic: license-file
+
+# Omnimancer CLI
+
+A unified command-line interface for multiple AI providers - chat with Claude, OpenAI, Gemini, and 10+ other AI models through a single, intuitive tool.
+
+## Quick Start
+
+### Installation
+
+**Using pipx (recommended):**
+```bash
+pipx install omnimancer-cli
+```
+
+**Using pip:**
+```bash
+pip install omnimancer-cli
+```
+
+### Available Commands
+
+After installation, use any of these commands:
+- **`omnimancer`** - Full command name
+- **`omn`** - Quick alias ⚡
+- **`omniman`** - Alternative alias
+
+### First Run
+
+```bash
+omn  # or omnimancer, or omniman
+```
+
+On first run, you'll be guided through setup:
+
+```
+🚀 Omnimancer Setup Wizard
+
+Select a provider to configure:
+1. Claude (Anthropic)
+2. OpenAI  
+3. Google Gemini
+4. Perplexity AI
+5. Ollama (Local)
+...
+
+Choose [1]: 1
+
+Enter your Claude API key: sk-ant-...
+✅ Configuration complete!
+
+>>> Hello! How can you help me today?
+🤖 Claude: I'm Claude, an AI assistant created by Anthropic...
+```
+
+## Basic Usage
+
+```bash
+# Start Omnimancer
+omn
+
+# Start chatting
+>>> What's the weather like?
+
+# Switch models mid-conversation  
+>>> /switch openai gpt-4o
+>>> Now using GPT-4. How are you different?
+
+# Check available providers and models
+>>> /providers
+>>> /models
+
+# Save conversations
+>>> /save my-chat
+
+# Load previous conversations
+>>> /load my-chat
+
+# Get help
+>>> /help
+```
+
+## Agent Mode & File Operations
+
+Omnimancer includes advanced agent capabilities that allow AI models to perform file operations with your explicit approval:
+
+### 🤖 **Autonomous Agent Features**
+- **File Creation**: Create new files with AI-generated content
+- **File Modification**: Edit existing files with intelligent changes
+- **Code Refactoring**: Restructure and improve existing code
+- **Documentation Generation**: Create comprehensive documentation
+- **Project Setup**: Initialize new projects with proper structure
+
+### 🔒 **Secure Approval System**
+Every file operation requires your explicit approval with:
+
+```bash
+🔍 File Operation Approval Required
+📄 Creating: data_analyzer.py
+📊 Risk Level: Low | 🟢 
+📏 Size: 1,247 bytes (45 lines)
+
+[Y] Approve  [N] Deny  [D] View Details  [Q] Quit
+```
+
+### 🎨 **Rich Visual Interface**
+- **Syntax Highlighting**: Code displayed with proper formatting
+- **Diff Views**: See exactly what changes before approval
+- **Risk Assessment**: Operations rated Low/Medium/High/Critical
+- **Batch Operations**: Handle multiple files efficiently
+
+### ⚡ **Quick Examples**
+
+```bash
+# Ask AI to create files
+>>> Create a Python script to analyze CSV data
+🔍 Shows preview → [Y] to approve → ✅ File created
+
+# Request code modifications  
+>>> Add error handling to this function
+🔍 Shows diff view → [Y] to approve → ✅ File updated
+
+# Batch project setup
+>>> Set up a Flask web application
+🔍 Shows 8 files → [A] approve all → ✅ Project ready
+```
+
+[**📖 Full Documentation**](docs/agent-approval-system.md) | [**🛡️ Security Guide**](docs/security.md)
+
+## Supported Providers
+
+| Provider | API Key Required | Best For |
+|----------|------------------|----------|
+| **Claude** | [Anthropic Console](https://console.anthropic.com/) | Complex reasoning, analysis |
+| **Claude Code** | Anthropic API key | IDE integration, coding |
+| **OpenAI** | [OpenAI Platform](https://platform.openai.com/) | General purpose, coding |
+| **Gemini** | [Google AI Studio](https://aistudio.google.com/) | Large context, research |
+| **Perplexity** | [Perplexity](https://www.perplexity.ai/) | Real-time web search |
+| **xAI (Grok)** | [xAI Console](https://console.x.ai/) | Creative tasks, real-time info |
+| **Mistral** | [Mistral Platform](https://mistral.ai/) | Code generation, efficiency |
+| **AWS Bedrock** | [AWS Console](https://console.aws.amazon.com/bedrock/) | AWS integration |
+| **Ollama** | No API key (local) | Privacy, offline use |
+| **Azure OpenAI** | Azure setup required | Enterprise |
+| **Vertex AI** | Google Cloud setup | Enterprise |
+| **OpenRouter** | [OpenRouter](https://openrouter.ai/) | 100+ models access |
+| **Cohere** | [Cohere Platform](https://cohere.com/) | Multilingual, embeddings |
+
+## Commands
+
+### Core Commands
+| Command | Description |
+|---------|-------------|
+| `/help` | Show all commands |
+| `/setup` | Run interactive setup wizard |
+| `/quit` | Exit Omnimancer |
+| `/clear` | Clear screen |
+
+### Model & Provider Management
+| Command | Description |
+|---------|-------------|
+| `/models` | List available models |
+| `/providers` | Show configured providers |
+| `/switch [provider] [model]` | Change provider/model |
+| `/validate [provider]` | Validate provider configurations |
+| `/health [provider]` | Check provider health status |
+| `/repair [provider]` | Repair provider issues |
+| `/diagnose [provider]` | Run diagnostic tests |
+
+### Conversation Management
+| Command | Description |
+|---------|-------------|
+| `/save [name]` | Save current conversation |
+| `/load [name]` | Load saved conversation |
+| `/list` | List saved conversations |
+| `/history` | Conversation history management |
+
+### Agent & File Operations
+| Command | Description |
+|---------|-------------|
+| `/agent` | Enable/disable agent mode |
+| `/agents` | Manage agent configurations |
+| `/approvals` | View/manage file operation approvals |
+| `/permissions` | Configure security permissions |
+
+### Tool Integration
+| Command | Description |
+|---------|-------------|
+| `/tools` | Show available tools |
+| `/mcp` | MCP server management |
+
+### Model Management
+| Command | Description |
+|---------|-------------|
+| `/add-model` | Add custom model |
+| `/remove-model` | Remove custom model |
+| `/list-custom-models` | List custom models |
+
+### System
+| Command | Description |
+|---------|-------------|
+| `/status` | Show system status |
+
+## Configuration
+
+Omnimancer stores encrypted configuration in `~/.omnimancer/config.json`.
+
+### Manual Configuration
+
+```bash
+# Add a provider
+omnimancer --config
+
+# Or edit configuration interactively
+>>> /config
+```
+
+### Environment Variables
+
+```bash
+export ANTHROPIC_API_KEY="your-key"
+export OPENAI_API_KEY="your-key"
+export GOOGLE_API_KEY="your-key"
+# ... then run omnimancer
+```
+
+## Local AI with Ollama
+
+For privacy and offline use:
+
+```bash
+# Install Ollama
+curl -fsSL https://ollama.ai/install.sh | sh
+
+# Start Ollama server
+ollama serve
+
+# Download a model
+ollama pull llama3.1
+
+# Configure Omnimancer
+omn
+>>> /switch ollama llama3.1
+```
+
+## Tool Integration (MCP)
+
+Enable AI tool calling for file operations, web search, and more:
+
+```bash
+# Install UV for MCP servers
+curl -LsSf https://astral.sh/uv/install.sh | sh
+
+# Check tool status
+>>> /tools
+>>> /mcp status
+```
+
+Popular MCP tools:
+- **Filesystem**: File operations
+- **Web Search**: Real-time search  
+- **Git**: Repository management
+
+## Examples
+
+### Basic Chat
+```
+>>> Explain quantum computing in simple terms
+🤖 Claude: Quantum computing is like having a super-powered calculator...
+
+>>> /switch openai gpt-4o  
+>>> How would you explain it differently?
+🤖 GPT-4: I'd compare quantum computing to exploring a maze...
+```
+
+### Code Generation
+```
+>>> Write a Python function to calculate fibonacci numbers
+🤖 Claude: Here's an efficient implementation using memoization:
+
+```python
+def fibonacci(n, memo={}):
+    if n in memo:
+        return memo[n]
+    if n <= 1:
+        return n
+    memo[n] = fibonacci(n-1, memo) + fibonacci(n-2, memo)
+    return memo[n]
+```
+
+### Model Comparison
+```
+>>> /switch claude claude-3-5-sonnet
+>>> What's 15 * 24?
+🤖 Claude: 15 × 24 = 360
+
+>>> /switch openai gpt-4o
+>>> What's 15 * 24?  
+🤖 GPT-4: 15 × 24 = 360
+```
+
+## Advanced Features
+
+- **Conversation Management**: Save/load chat history
+- **Model Switching**: Compare responses between providers
+- **Tool Calling**: AI can execute code, search web, manage files
+- **Health Monitoring**: Provider status and diagnostics
+- **Configuration Templates**: Pre-configured setups for different use cases
+
+## Development
+
+```bash
+git clone https://gitlab.com/jite-ai/omnimancer
+cd omnimancer
+pip install -e ".[dev]"
+pytest
+```
+
+## Troubleshooting
+
+### Common Issues
+
+**"No providers configured"**
+```bash
+omn  # Run setup wizard
+>>> /setup
+```
+
+**"Invalid API key"**
+- Check key format (Claude: `sk-ant-`, OpenAI: `sk-`, etc.)
+- Verify key at provider's website
+- Use `/validate` command to test configuration
+
+**"Ollama connection failed"**
+```bash
+ollama serve  # Start Ollama server
+ollama pull llama3.1  # Download a model
+```
+
+**Check system health:**
+```bash
+omn
+>>> /health  # Check all providers
+>>> /diagnose  # Run diagnostics
+>>> /validate  # Validate configurations
+```
+
+**Debug mode:**
+```bash
+export OMNIMANCER_DEBUG=1
+omn
+```
+
+## License
+
+MIT License - see [LICENSE](LICENSE) file.
+
+## Links
+
+- [GitHub Repository]https://gitlab.com/jite-ai/omnimancer)
+- [Issues](https://gitlab.com/jite-ai/omnimancer/issues)
+- [Documentation](https://gitlab.com/jite-ai/omnimancer/docs)
+
+---
+
+**Omnimancer CLI** - One tool, multiple AI providers, endless possibilities.