terminal-sherpa 0.1.0__py3-none-any.whl → 0.2.0__py3-none-any.whl

ask/main.py

@@ -76,10 +76,11 @@ def resolve_provider(args, config_data) -> ProviderInterface:
             # Use default provider from environment variables
             default_provider = config.get_default_provider()
             if not default_provider:
+                keys = ["GEMINI_API_KEY", "ANTHROPIC_API_KEY", "OPENAI_API_KEY"]
                 logger.error(
                     "No default model configured and no API keys found. "
-                    "Please set ANTHROPIC_API_KEY or OPENAI_API_KEY environment "
-                    "variable, or set a default_provider in your config file."
+                    f"Please set one or more of {keys} environment variables, "
+                    "or set a default_provider in your config file."
                 )
                 sys.exit(1)
             logger.debug(f"Using default provider from environment: {default_provider}")

ask/providers/__init__.py

@@ -2,6 +2,7 @@
 
 from .anthropic import AnthropicProvider
 from .base import ProviderInterface
+from .gemini import GeminiProvider
 from .openai import OpenAIProvider
 
 # Provider registry - maps provider names to their classes
@@ -33,3 +34,4 @@ def list_providers() -> list[str]:
 
 register_provider("anthropic", AnthropicProvider)
 register_provider("openai", OpenAIProvider)
+register_provider("gemini", GeminiProvider)

ask/providers/anthropic.py

@@ -37,6 +37,7 @@ class AnthropicProvider(ProviderInterface):
             return response.content[0].text
         except Exception as e:
             self._handle_api_error(e)
+            return ""
 
     def validate_config(self) -> None:
         """Validate provider configuration and API key."""

ask/providers/gemini.py

@@ -0,0 +1,89 @@
+"""Anthropic provider implementation."""
+
+import os
+from typing import Any
+
+from google import genai
+from google.genai.types import GenerateContentConfig, GenerateContentResponse
+
+from ask.config import SYSTEM_PROMPT
+from ask.exceptions import APIError, AuthenticationError, RateLimitError
+from ask.providers.base import ProviderInterface
+
+
+class GeminiProvider(ProviderInterface):
+    """Gemini AI provider implementation."""
+
+    def __init__(self, config: dict[str, Any]):
+        """Initialize Gemini provider with configuration."""
+        super().__init__(config)
+        self.client: genai.Client | None = None
+
+    def _parse_response(self, response: GenerateContentResponse) -> str:
+        """Parse response from Gemini API."""
+        if response.candidates is None or len(response.candidates) == 0:
+            return ""
+        parts = response.candidates[0].content.parts
+        if parts is None:
+            return ""
+        return "".join([part.text for part in parts])
+
+    def get_bash_command(self, prompt: str) -> str:
+        """Generate bash command from natural language prompt."""
+        if self.client is None:
+            self.validate_config()
+
+        # After validate_config(), client should be set
+        assert self.client is not None, "Client should be initialized after validation"
+
+        try:
+            # max_tokens=self.config.get("max_tokens", 150),
+            # temperature=self.config.get("temperature", 0.0),
+            # system=self.config.get("system_prompt", SYSTEM_PROMPT),
+            response = self.client.models.generate_content(
+                model=self.config.get("model_name", "gemini-2.5-flash"),
+                contents=prompt,
+                config=GenerateContentConfig(
+                    max_output_tokens=self.config.get("max_tokens", 150),
+                    temperature=self.config.get("temperature", 0.0),
+                    system_instruction=self.config.get("system_prompt", SYSTEM_PROMPT),
+                ),
+            )
+            return self._parse_response(response)
+        except Exception as e:
+            self._handle_api_error(e)
+            return ""
+
+    def validate_config(self) -> None:
+        """Validate provider configuration and API key."""
+        api_key_env = self.config.get("api_key_env", "GEMINI_API_KEY")
+        api_key = os.environ.get(api_key_env)
+
+        if not api_key:
+            raise AuthenticationError(
+                f"Error: {api_key_env} environment variable is required"
+            )
+
+        self.client = genai.Client(api_key=api_key)
+
+    def _handle_api_error(self, error: Exception):
+        """Handle API errors and map them to standard exceptions."""
+        error_str = str(error).lower()
+
+        if "authentication" in error_str or "unauthorized" in error_str:
+            raise AuthenticationError("Error: Invalid API key")
+        elif "rate limit" in error_str:
+            raise RateLimitError("Error: API rate limit exceeded")
+        else:
+            raise APIError(f"Error: API request failed - {error}")
+
+    @classmethod
+    def get_default_config(cls) -> dict[str, Any]:
+        """Return default configuration for Gemini provider."""
+        return {
+            "model_name": "gemini-2.5-flash",
+            "max_tokens": 150,
+            "api_key_env": "GEMINI_API_KEY",
+            "temperature": 0.0,
+            "system_prompt": SYSTEM_PROMPT,
+        }

ask/providers/openai.py

@@ -62,6 +62,7 @@ class OpenAIProvider(ProviderInterface):
                 return re_match.group(1)
         except Exception as e:
             self._handle_api_error(e)
+            return ""
 
     def validate_config(self) -> None:
         """Validate provider configuration and API key."""

{terminal_sherpa-0.1.0.dist-info → terminal_sherpa-0.2.0.dist-info}/METADATA

@@ -1,35 +1,42 @@
 Metadata-Version: 2.4
 Name: terminal-sherpa
-Version: 0.1.0
+Version: 0.2.0
 Summary: AI-powered bash command generator
 Requires-Python: >=3.13
 Description-Content-Type: text/markdown
+License-File: LICENSE
 Requires-Dist: anthropic>=0.7.0
 Requires-Dist: black>=25.1.0
+Requires-Dist: cosmic-ray>=8.3.4
+Requires-Dist: google-genai>=1.26.0
 Requires-Dist: loguru>=0.7.0
+Requires-Dist: mutatest>=3.1.0
 Requires-Dist: openai>=1.0.0
 Requires-Dist: pytest>=8.0.0
 Requires-Dist: pytest-cov>=4.0.0
 Requires-Dist: pytest-mock>=3.12.0
 Requires-Dist: ruff>=0.12.3
+Requires-Dist: setuptools>=80.9.0
+Requires-Dist: taskipy>=1.14.1
 Requires-Dist: toml>=0.10.0
+Dynamic: license-file
 
-# ask
+# terminal-sherpa
 
 A lightweight AI chat interface for fellow terminal dwellers.
 
 Turn natural language into bash commands instantly.
 Stop googling syntax and start asking.
 
-[![codecov](https://codecov.io/github/lcford2/ask/graph/badge.svg?token=2MXHNL3RHE)](https://codecov.io/github/lcford2/ask)
+[![codecov](https://codecov.io/github/lcford2/terminal-sherpa/graph/badge.svg?token=2MXHNL3RHE)](https://codecov.io/github/lcford2/terminal-sherpa)
 
 ## 🚀 Getting Started
 
 Get up and running:
 
 ```bash
-# Install ask
-uv tool install ask
+# Install terminal-sherpa
+pip install terminal-sherpa # installs the `ask` CLI tool
 
 # Set your API key
 export ANTHROPIC_API_KEY="your-key-here"
@@ -39,6 +46,7 @@ ask "find all .py files modified in the last week"
 ```
 
 **Example output:**
+
 ```bash
 find . -name "*.py" -mtime -7
 ```
@@ -46,7 +54,7 @@ find . -name "*.py" -mtime -7
 ## ✨ Features
 
 - **Natural language to bash conversion** - Describe what you want, get the command
-- **Multiple AI provider support** - Choose between Anthropic (Claude) and OpenAI (GPT) models
+- **Multiple AI provider support** - Choose between Anthropic (Claude), OpenAI (GPT), and Google (Gemini) models
 - **Flexible configuration system** - Set defaults, customize models, and manage API keys
 - **XDG-compliant config files** - Follows standard configuration file locations
 - **Verbose logging support** - Debug and understand what's happening under the hood
@@ -54,30 +62,29 @@ find . -name "*.py" -mtime -7
 ## 📦 Installation
 
 ### Requirements
+
 - Python 3.13+
 - API key for Anthropic or OpenAI
 
 ### Install Methods
 
-**Recommended (uv):**
-```bash
-uv tool install ask
-```
-
 **Using pip:**
+
 ```bash
-pip install ask
+pip install terminal-sherpa
 ```
 
 **From source:**
+
 ```bash
-git clone https://github.com/lcford2/ask.git
-cd ask
+git clone https://github.com/lcford2/terminal-sherpa.git
+cd terminal-sherpa
 uv sync
 uv run ask "your prompt here"
 ```
 
 **Verify installation:**
+
 ```bash
 ask --help
 ```
@@ -85,20 +92,23 @@ ask --help
 ## 💡 Usage
 
 ### Basic Syntax
+
 ```bash
 ask "your natural language prompt"
 ```
 
 ### Command Options
 
-| Option | Description | Example |
-|--------|-------------|---------|
+| Option                   | Description                | Example                                             |
+| ------------------------ | -------------------------- | --------------------------------------------------- |
 | `--model provider:model` | Specify provider and model | `ask --model anthropic:claude-3-haiku "list files"` |
-| `--verbose` | Enable verbose logging | `ask --verbose "compress this folder"` |
+|                          |                            | `ask --model gemini:gemini-2.5-flash "list files"` |
+| `--verbose`              | Enable verbose logging     | `ask --verbose "compress this folder"`              |
 
 ### Practical Examples
 
 **File Operations:**
+
 ```bash
 ask "find all files larger than 100MB"
 # Example output: find . -size +100M
@@ -108,6 +118,7 @@ ask "create a backup of config.txt with timestamp"
 ```
 
 **Git Commands:**
+
 ```bash
 ask "show git log for last 5 commits with one line each"
 # Example output: git log --oneline -5
@@ -117,6 +128,7 @@ ask "delete all local branches that have been merged"
 ```
 
 **System Administration:**
+
 ```bash
 ask "check disk usage of current directory sorted by size"
 # Example output: du -sh * | sort -hr
@@ -126,6 +138,7 @@ ask "find processes using port 8080"
 ```
 
 **Text Processing:**
+
 ```bash
 ask "count lines in all Python files"
 # Example output: find . -name "*.py" -exec wc -l {} + | tail -1
@@ -135,6 +148,7 @@ ask "replace all tabs with spaces in file.txt"
 ```
 
 **Network Operations:**
+
 ```bash
 ask "download file from URL and save to downloads folder"
 # Example output: curl -o ~/Downloads/filename "https://example.com/file"
@@ -146,6 +160,7 @@ ask "check if port 443 is open on example.com"
 ## ⚙️ Configuration
 
 ### Configuration File Locations
+
 Ask follows XDG Base Directory Specification:
 
 1. `$XDG_CONFIG_HOME/ask/config.toml`
@@ -153,12 +168,15 @@ Ask follows XDG Base Directory Specification:
 1. `~/.ask/config.toml` (fallback)
 
 ### Environment Variables
+
 ```bash
 export ANTHROPIC_API_KEY="your-anthropic-key"
 export OPENAI_API_KEY="your-openai-key"
+export GEMINI_API_KEY="your-gemini-key"
 ```
 
 ### Example Configuration File
+
 Create `~/.config/ask/config.toml`:
 
 ```toml
@@ -166,7 +184,6 @@ Create `~/.config/ask/config.toml`:
 default_model = "anthropic"
 
 [anthropic]
-api_key = "your-anthropic-key"
 model = "claude-3-haiku-20240307"
 max_tokens = 512
 
@@ -175,33 +192,44 @@ model = "claude-3-5-sonnet-20241022"
 max_tokens = 1024
 
 [openai]
-api_key = "your-openai-key"
 model = "gpt-4o"
 max_tokens = 1024
+
+[gemini]
+model = "gemini-2.5-flash"
+max_tokens = 150
+
+[gemini.pro]
+model = "gemini-2.5-pro"
+max_tokens = 1024
 ```
 
 ## 🤖 Supported Providers
 
 - Anthropic (Claude)
 - OpenAI (GPT)
+- Google (Gemini)
 
-> **Note:** Get API keys from [Anthropic Console](https://console.anthropic.com/) or [OpenAI Platform](https://platform.openai.com/)
+> **Note:** Get API keys from [Anthropic Console](https://console.anthropic.com/), [OpenAI Platform](https://platform.openai.com/), or [Google AI Studio](https://aistudio.google.com/)
 
 ## 🛣️ Roadmap
 
 ### Near-term
+
 - [ ] Shell integration and auto-completion
 - [ ] Command history and favorites
 - [ ] Safety features (command preview/confirmation)
 - [ ] Output formatting options
 
 ### Medium-term
+
 - [ ] Additional providers (Google, Cohere, Mistral)
 - [ ] Interactive mode for complex tasks
 - [ ] Plugin system for custom providers
 - [ ] Command validation and testing
 
 ### Long-term
+
 - [ ] Local model support (Ollama, llama.cpp)
 - [ ] Learning from user preferences
 - [ ] Advanced safety and sandboxing
@@ -210,19 +238,22 @@ max_tokens = 1024
 ## 🔧 Development
 
 ### Setup
+
 ```bash
-git clone https://github.com/lcford2/ask.git
+git clone https://github.com/lcford2/terminal-sherpa.git
 cd ask
 uv sync
 uv run pre-commit install
 ```
 
 ### Testing
+
 ```bash
 uv run python -m pytest
 ```
 
 ### Contributing
+
 1. Fork the repository
 2. Create a feature branch
 3. Make your changes

terminal_sherpa-0.2.0.dist-info/RECORD

@@ -0,0 +1,23 @@
+ask/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+ask/config.py,sha256=iHIiMKePia80Sno_XlARqa7pEyW3eZm_Bf5SlUMidiQ,2880
+ask/exceptions.py,sha256=0RLMSbw6j49BEhJN7C8MYaKpuhVeitsBhTGjZmaiHis,434
+ask/main.py,sha256=9mVXwncU2P4OQxE7Oxcqi376A06xluC76kiIoCCqNSc,3936
+ask/providers/__init__.py,sha256=Y0NswA6O8PpE_PDWa-GZ1FNmSXrwReZ9-roUoTOksXU,1184
+ask/providers/anthropic.py,sha256=3bB335ZxRu4a_j7U0tox_6sQxTf4X287GhU2-gr5ctU,2725
+ask/providers/base.py,sha256=91ZbVORYWckSHNwNPiTmgfqQN0FLO9AgV6mptuAkIU0,769
+ask/providers/gemini.py,sha256=c1uQg6ShNkw2AkwDJhjfki7hTBl4vOkT9v1O4ewRfbg,3408
+ask/providers/openai.py,sha256=9PS1AgMr6Nb-OcYYiNYNrG384wNUS8m2lVT04K3hFV8,3683
+terminal_sherpa-0.2.0.dist-info/licenses/LICENSE,sha256=xLe81eIrf0X6CnEDDJXmoXuDzkdMYM3Eq1BgHUpG1JQ,1067
+test/conftest.py,sha256=pjDI0SbIhHxDqJW-BdL7s6lTqM2f8hItxWY8EjC-dL8,1548
+test/test_anthropic.py,sha256=S5OQ67qIZ4VO38eJwAAwJa4JBylJhKCtmcGjCWA8WLY,5687
+test/test_config.py,sha256=FrJ6bsZ6mK46e-8fQfkFGx9GgwHrNfnoI8211R0V9K8,5565
+test/test_exceptions.py,sha256=tw-spMitAdYj9uW_8TjnlyVKKXFC06FR3610WGR-494,1754
+test/test_gemini.py,sha256=sV8FkaU5rLfuu3lGeQdxsa-ZmNnLUYpODaKhayrasSo,8000
+test/test_main.py,sha256=3gZ83nVHMSEmgHSF2UJoELfK028a4vgxLpIk2P1cH1Y,7745
+test/test_openai.py,sha256=KAGQWFrXeu4P9umij7XDoxnKQ2cApv6ImuL8EiG_5W8,8388
+test/test_providers.py,sha256=SejQvCZSEQ5RAfVTCtPZ-39fXnfV17n4gaSxjiHA5UM,2140
+terminal_sherpa-0.2.0.dist-info/METADATA,sha256=V3EAri2cVmsjv3zxWnJxhpEK_2kXnBLxWa5QsNUALc4,6392
+terminal_sherpa-0.2.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+terminal_sherpa-0.2.0.dist-info/entry_points.txt,sha256=LxG9-J__nMGmeEIi47WVGYC1LLJo1GaADH21hfxEK70,38
+terminal_sherpa-0.2.0.dist-info/top_level.txt,sha256=Y7k5b2NSCkKiA_XPU-4fT_GYangD6JVDug5xwfXvmuQ,9
+terminal_sherpa-0.2.0.dist-info/RECORD,,

terminal_sherpa-0.2.0.dist-info/licenses/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 Lucas Ford
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

test/conftest.py

@@ -38,6 +38,13 @@ def mock_anthropic_key():
         yield
 
 
+@pytest.fixture
+def mock_gemini_key():
+    """Mock Gemini API key in environment."""
+    with patch.dict(os.environ, {"GEMINI_API_KEY": "test-gemini-key"}, clear=True):
+        yield
+
+
 @pytest.fixture
 def mock_openai_key():
     """Mock OpenAI API key in environment."""

test/test_gemini.py

@@ -0,0 +1,244 @@
+"""Tests for Gemini provider."""
+
+import os
+from unittest.mock import MagicMock, patch
+
+import pytest
+from google.genai.types import GenerateContentConfig
+
+from ask.config import SYSTEM_PROMPT
+from ask.exceptions import APIError, AuthenticationError, RateLimitError
+from ask.providers.gemini import GeminiProvider
+
+
+def test_gemini_provider_init():
+    """Test provider initialization."""
+    config = {"model_name": "gemini-2.5-flash"}
+    provider = GeminiProvider(config)
+
+    assert provider.config == config
+    assert provider.client is None
+
+
+def test_validate_config_success(mock_gemini_key):
+    """Test successful config validation."""
+    config = {"api_key_env": "GEMINI_API_KEY"}
+    provider = GeminiProvider(config)
+
+    with patch("google.genai.Client") as mock_genai:
+        mock_client = MagicMock()
+        mock_genai.return_value = mock_client
+
+        provider.validate_config()
+
+        assert provider.client == mock_client
+        mock_genai.assert_called_once_with(api_key="test-gemini-key")
+
+
+def test_validate_config_missing_key(mock_env_vars):
+    """Test missing API key error."""
+    config = {"api_key_env": "GEMINI_API_KEY"}
+    provider = GeminiProvider(config)
+
+    with pytest.raises(
+        AuthenticationError, match="GEMINI_API_KEY environment variable is required"
+    ):
+        provider.validate_config()
+
+
+def test_validate_config_custom_env():
+    """Test custom environment variable."""
+    config = {"api_key_env": "CUSTOM_GEMINI_KEY"}
+    provider = GeminiProvider(config)
+
+    with patch.dict(os.environ, {"CUSTOM_GEMINI_KEY": "custom-key"}):
+        with patch("google.genai.Client") as mock_genai:
+            mock_client = MagicMock()
+            mock_genai.return_value = mock_client
+
+            provider.validate_config()
+
+            mock_genai.assert_called_once_with(api_key="custom-key")
+
+
+def test_get_default_config():
+    """Test default configuration values."""
+    default_config = GeminiProvider.get_default_config()
+
+    assert default_config["model_name"] == "gemini-2.5-flash"
+    assert default_config["max_tokens"] == 150
+    assert default_config["api_key_env"] == "GEMINI_API_KEY"
+    assert default_config["temperature"] == 0.0
+    assert default_config["system_prompt"] == SYSTEM_PROMPT
+
+
+def test_parse_response_empty_candidates():
+    """Test _parse_response with empty candidates."""
+    provider = GeminiProvider({})
+
+    # Test with None candidates
+    mock_response = MagicMock()
+    mock_response.candidates = None
+    result = provider._parse_response(mock_response)
+    assert result == ""
+
+    # Test with empty candidates list
+    mock_response.candidates = []
+    result = provider._parse_response(mock_response)
+    assert result == ""
+
+
+def test_parse_response_none_parts():
+    """Test _parse_response with None parts."""
+    provider = GeminiProvider({})
+
+    mock_response = MagicMock()
+    mock_response.candidates = [MagicMock()]
+    mock_response.candidates[0].content = MagicMock()
+    mock_response.candidates[0].content.parts = None
+
+    result = provider._parse_response(mock_response)
+    assert result == ""
+
+
+def test_parse_response_success():
+    """Test _parse_response with successful response."""
+    provider = GeminiProvider({})
+
+    mock_response = MagicMock()
+    mock_response.candidates = [MagicMock()]
+    mock_response.candidates[0].content = MagicMock()
+    mock_response.candidates[0].content.parts = [
+        MagicMock(text="part1"),
+        MagicMock(text="part2"),
+    ]
+
+    result = provider._parse_response(mock_response)
+    assert result == "part1part2"
+
+
+def test_get_bash_command_success(mock_gemini_key):
+    """Test successful command generation."""
+    config = {"model_name": "gemini-2.5-flash", "max_tokens": 150}
+    provider = GeminiProvider(config)
+
+    mock_response = MagicMock()
+    mock_response.candidates = [MagicMock()]
+    mock_response.candidates[0].content = MagicMock()
+    mock_response.candidates[0].content.parts = [MagicMock(text="ls -la")]
+
+    with patch("google.genai.Client") as mock_genai:
+        mock_client = MagicMock()
+        mock_client.models.generate_content.return_value = mock_response
+        mock_genai.return_value = mock_client
+
+        result = provider.get_bash_command("list files")
+
+        assert result == "ls -la"
+        mock_client.models.generate_content.assert_called_once_with(
+            model="gemini-2.5-flash",
+            contents="list files",
+            config=GenerateContentConfig(
+                max_output_tokens=150,
+                temperature=0.0,
+                system_instruction=SYSTEM_PROMPT,
+            ),
+        )
+
+
+def test_get_bash_command_api_exception(mock_gemini_key):
+    """Test get_bash_command with API exception."""
+    config = {"model_name": "gemini-2.5-flash"}
+    provider = GeminiProvider(config)
+
+    with patch("google.genai.Client") as mock_genai:
+        mock_client = MagicMock()
+        mock_client.models.generate_content.side_effect = Exception("API error")
+        mock_genai.return_value = mock_client
+
+        # This should trigger the exception handling and return empty string
+        with pytest.raises(APIError, match="API request failed"):
+            provider.get_bash_command("test prompt")
+
+
+def test_get_bash_command_auto_validate(mock_gemini_key):
+    """Test auto-validation behavior."""
+    config = {}
+    provider = GeminiProvider(config)
+
+    mock_response = MagicMock()
+    mock_response.candidates = [MagicMock()]
+    mock_response.candidates[0].content = MagicMock()
+    mock_response.candidates[0].content.parts = [MagicMock(text="ls -la")]
+
+    with patch("google.genai.Client") as mock_genai:
+        mock_client = MagicMock()
+        mock_client.models.generate_content.return_value = mock_response
+        mock_genai.return_value = mock_client
+
+        # Client should be None initially
+        assert provider.client is None
+
+        result = provider.get_bash_command("list files")
+
+        # Client should be set after auto-validation
+        assert provider.client is not None
+        assert result == "ls -la"
+
+
+def test_handle_api_error_auth():
+    """Test authentication error mapping."""
+    provider = GeminiProvider({})
+
+    with pytest.raises(AuthenticationError, match="Invalid API key"):
+        provider._handle_api_error(Exception("authentication failed"))
+
+
+def test_handle_api_error_rate_limit():
+    """Test rate limit error mapping."""
+    provider = GeminiProvider({})
+
+    with pytest.raises(RateLimitError, match="API rate limit exceeded"):
+        provider._handle_api_error(Exception("rate limit exceeded"))
+
+
+def test_handle_api_error_generic():
+    """Test generic API error mapping."""
+    provider = GeminiProvider({})
+
+    with pytest.raises(APIError, match="API request failed"):
+        provider._handle_api_error(Exception("unexpected error"))
+
+
+def test_config_parameter_usage(mock_gemini_key):
+    """Test configuration parameter usage."""
+    config = {
+        "model_name": "gemini-2.5-pro",
+        "max_tokens": 1024,
+        "temperature": 0.5,
+        "system_prompt": "Custom system prompt",
+    }
+    provider = GeminiProvider(config)
+
+    mock_response = MagicMock()
+    mock_response.candidates = [MagicMock()]
+    mock_response.candidates[0].content = MagicMock()
+    mock_response.candidates[0].content.parts = [MagicMock(text="custom response")]
+
+    with patch("google.genai.Client") as mock_genai:
+        mock_client = MagicMock()
+        mock_client.models.generate_content.return_value = mock_response
+        mock_genai.return_value = mock_client
+
+        result = provider.get_bash_command("test prompt")
+
+        assert result == "custom response"
+        mock_client.models.generate_content.assert_called_once_with(
+            model="gemini-2.5-pro",
+            contents="test prompt",
+            config=GenerateContentConfig(
+                max_output_tokens=1024,
+                temperature=0.5,
+                system_instruction="Custom system prompt",
+            ),
+        )

test/test_openai.py

@@ -196,14 +196,6 @@ def test_handle_api_error_rate_limit():
         provider._handle_api_error(Exception("rate limit exceeded"))
 
 
-def test_handle_api_error_quota():
-    """Test quota error mapping."""
-    provider = OpenAIProvider({})
-
-    with pytest.raises(RateLimitError, match="API rate limit exceeded"):
-        provider._handle_api_error(Exception("quota exceeded"))
-
-
 def test_handle_api_error_generic():
     """Test generic API error mapping."""
     provider = OpenAIProvider({})

terminal_sherpa-0.1.0.dist-info/RECORD

@@ -1,20 +0,0 @@
-ask/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ask/config.py,sha256=iHIiMKePia80Sno_XlARqa7pEyW3eZm_Bf5SlUMidiQ,2880
-ask/exceptions.py,sha256=0RLMSbw6j49BEhJN7C8MYaKpuhVeitsBhTGjZmaiHis,434
-ask/main.py,sha256=SJ084NXNvd-pO2hY8oFO__NFR1F07YxeOacBuv65crc,3867
-ask/providers/__init__.py,sha256=IINO0hNGarFpf62lCuVtIqMeAOC1CtR_wDWUL6_iWzI,1105
-ask/providers/anthropic.py,sha256=ZFlnQZaxPGHLYjAacapE-63LoSu43xYnZ30Ajmrmgw0,2703
-ask/providers/base.py,sha256=91ZbVORYWckSHNwNPiTmgfqQN0FLO9AgV6mptuAkIU0,769
-ask/providers/openai.py,sha256=R-UgVArtlpn8F4qkliQ7unNk11ekTPL0hFZCfubGYpg,3661
-test/conftest.py,sha256=V9ebLC-soz0hHocZLZAibzsVdvzZh4-elcTtKmZ2FyA,1363
-test/test_anthropic.py,sha256=S5OQ67qIZ4VO38eJwAAwJa4JBylJhKCtmcGjCWA8WLY,5687
-test/test_config.py,sha256=FrJ6bsZ6mK46e-8fQfkFGx9GgwHrNfnoI8211R0V9K8,5565
-test/test_exceptions.py,sha256=tw-spMitAdYj9uW_8TjnlyVKKXFC06FR3610WGR-494,1754
-test/test_main.py,sha256=3gZ83nVHMSEmgHSF2UJoELfK028a4vgxLpIk2P1cH1Y,7745
-test/test_openai.py,sha256=3dDwlxnKGwl5aJcKHyNgrwrciJBZ96a51fHuxr-V8FA,8633
-test/test_providers.py,sha256=SejQvCZSEQ5RAfVTCtPZ-39fXnfV17n4gaSxjiHA5UM,2140
-terminal_sherpa-0.1.0.dist-info/METADATA,sha256=hh4NTZNv3yisUjLUcsJxlOBHUiK70l0u82CQIxJrjJI,5610
-terminal_sherpa-0.1.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-terminal_sherpa-0.1.0.dist-info/entry_points.txt,sha256=LxG9-J__nMGmeEIi47WVGYC1LLJo1GaADH21hfxEK70,38
-terminal_sherpa-0.1.0.dist-info/top_level.txt,sha256=Y7k5b2NSCkKiA_XPU-4fT_GYangD6JVDug5xwfXvmuQ,9
-terminal_sherpa-0.1.0.dist-info/RECORD,,