terminal-sherpa 0.3.0__py3-none-any.whl → 0.4.0__py3-none-any.whl

ask/config.py

@@ -2,7 +2,7 @@
 
 import os
 from pathlib import Path
-from typing import Any, Optional
+from typing import Any
 
 import toml
 
@@ -16,7 +16,7 @@ SYSTEM_PROMPT = (
 )
 
 
-def get_config_path() -> Optional[Path]:
+def get_config_path() -> Path | None:
     """Find config file using XDG standard."""
     # Primary location: $XDG_CONFIG_HOME/ask/config.toml
     xdg_config_home = os.environ.get("XDG_CONFIG_HOME")
@@ -77,13 +77,13 @@ def get_provider_config(
     return provider_name, merged_config
 
 
-def get_default_model(config: dict[str, Any]) -> Optional[str]:
+def get_default_model(config: dict[str, Any]) -> str | None:
     """Get default model from configuration."""
     global_config = config.get("ask", {})
     return global_config.get("default_model")
 
 
-def get_default_provider() -> Optional[str]:
+def get_default_provider() -> str | None:
     """Determine fallback provider from environment variables."""
     # Check for API keys in order of preference: claude -> openai
     if os.environ.get("ANTHROPIC_API_KEY"):

ask/providers/__init__.py

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

ask/providers/anthropic.py

@@ -1,7 +1,7 @@
 """Anthropic provider implementation."""
 
 import os
-from typing import Any, Optional
+from typing import Any
 
 import anthropic
 
@@ -16,7 +16,7 @@ class AnthropicProvider(ProviderInterface):
     def __init__(self, config: dict[str, Any]):
         """Initialize Anthropic provider with configuration."""
         super().__init__(config)
-        self.client: Optional[anthropic.Anthropic] = None
+        self.client: anthropic.Anthropic | None = None
 
     def get_bash_command(self, prompt: str) -> str:
         """Generate bash command from natural language prompt."""

ask/providers/gemini.py

@@ -1,7 +1,7 @@
 """Anthropic provider implementation."""
 
 import os
-from typing import Any, Optional
+from typing import Any
 
 from google import genai
 from google.genai.types import GenerateContentConfig, GenerateContentResponse
@@ -17,7 +17,7 @@ class GeminiProvider(ProviderInterface):
     def __init__(self, config: dict[str, Any]):
         """Initialize Gemini provider with configuration."""
         super().__init__(config)
-        self.client: Optional[genai.Client] = None
+        self.client: genai.Client | None = None
 
     def _parse_response(self, response: GenerateContentResponse) -> str:
         """Parse response from Gemini API."""

ask/providers/grok.py

@@ -0,0 +1,118 @@
+"""Grok provider implementation using official xAI SDK."""
+
+import os
+import re
+from typing import Any, NoReturn
+
+from xai_sdk import Client
+from xai_sdk.chat import system, user
+
+from ask.config import SYSTEM_PROMPT
+from ask.exceptions import APIError, AuthenticationError, RateLimitError
+from ask.providers.base import ProviderInterface
+
+
+class GrokProvider(ProviderInterface):
+    """Grok provider implementation using official xAI SDK."""
+
+    def __init__(self, config: dict[str, Any]):
+        """Initialize Grok provider with configuration.
+
+        Args:
+            config: The configuration for the Grok provider
+        """
+        super().__init__(config)
+        self.client: Client | None = None
+
+    def get_bash_command(self, prompt: str) -> str:
+        """Generate bash command from natural language prompt.
+
+        Args:
+            prompt: The natural language prompt to generate a bash command for
+
+        Returns:
+            The generated bash command
+        """
+        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:
+            model_name = self.config.get("model_name", "grok-3-fast")
+            system_prompt = self.config.get("system_prompt", SYSTEM_PROMPT)
+
+            # Create chat using xAI SDK workflow
+            chat = self.client.chat.create(model=model_name)
+            chat.append(system(system_prompt))
+            chat.append(user(prompt))
+
+            # Get response
+            response = chat.sample()
+            content = response.content
+
+            if content is None:
+                raise APIError("Error: API returned empty response")
+
+            # Remove ```bash and ``` from the content if present
+            re_match = re.search(r"```bash\n(.*)\n```", content, re.DOTALL)
+            if re_match is None:
+                return content.strip()
+            else:
+                return re_match.group(1).strip()
+
+        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", "XAI_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"
+            )
+
+        # Initialize xAI SDK client
+        self.client = Client(api_key=api_key)
+
+    def _handle_api_error(self, error: Exception) -> NoReturn:
+        """Handle API errors and map them to standard exceptions.
+
+        Args:
+            error: The exception to handle
+
+        Raises:
+            AuthenticationError: If the API key is invalid
+            RateLimitError: If the API rate limit is exceeded
+        """
+        error_str = str(error).lower()
+
+        if (
+            "authentication" in error_str
+            or "unauthorized" in error_str
+            or "invalid api key" in error_str
+        ):
+            raise AuthenticationError("Error: Invalid API key")
+        elif (
+            "rate limit" in error_str
+            or "quota" in error_str
+            or "too many requests" 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 Grok provider."""
+        return {
+            "model_name": "grok-3-fast",
+            "max_tokens": 150,
+            "api_key_env": "XAI_API_KEY",
+            "temperature": 0.0,
+            "system_prompt": SYSTEM_PROMPT,
+        }

ask/providers/openai.py

@@ -2,7 +2,7 @@
 
 import os
 import re
-from typing import Any, NoReturn, Optional
+from typing import Any, NoReturn
 
 import openai
 
@@ -21,7 +21,7 @@ class OpenAIProvider(ProviderInterface):
             config: The configuration for the OpenAI provider
         """
         super().__init__(config)
-        self.client: Optional[openai.OpenAI] = None
+        self.client: openai.OpenAI | None = None
 
     def get_bash_command(self, prompt: str) -> str:
         """Generate bash command from natural language prompt.

{terminal_sherpa-0.3.0.dist-info → terminal_sherpa-0.4.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: terminal-sherpa
-Version: 0.3.0
+Version: 0.4.0
 Summary: AI-powered bash command generator
 Project-URL: Homepage, https://github.com/lcford2/terminal-sherpa
 Project-URL: Issues, https://github.com/lcford2/terminal-sherpa/issues
@@ -19,7 +19,7 @@ Classifier: Programming Language :: Python :: 3.13
 Classifier: Programming Language :: Python :: 3 :: Only
 Classifier: Topic :: Utilities
 Classifier: Topic :: Software Development :: Libraries
-Requires-Python: >=3.9
+Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: anthropic>=0.7.0
@@ -28,6 +28,7 @@ Requires-Dist: loguru>=0.7.0
 Requires-Dist: openai>=1.0.0
 Requires-Dist: setuptools>=80.9.0
 Requires-Dist: toml>=0.10.0
+Requires-Dist: xai-sdk
 Dynamic: license-file
 
 # terminal-sherpa
@@ -67,7 +68,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), OpenAI (GPT), and Google (Gemini) models
+- **Multiple AI provider support** - Choose between Anthropic (Claude), OpenAI (GPT), Google (Gemini), and xAI (Grok) 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
@@ -77,7 +78,7 @@ find . -name "*.py" -mtime -7
 ### Requirements
 
 - Python 3.9+
-- API key for Anthropic or OpenAI
+- API key for Anthropic, OpenAI, Google, or xAI
 
 ### Install Methods
 
@@ -112,11 +113,15 @@ ask "your natural language prompt"
 
 ### Command Options
 
-| Option                   | Description                | Example                                             |
-| ------------------------ | -------------------------- | --------------------------------------------------- |
-| `--model provider:model` | Specify provider and model | `ask --model anthropic:claude-3-haiku "list files"` |
-|                          |                            | `ask --model gemini:gemini-2.5-flash "list files"` |
-| `--verbose`              | Enable verbose logging     | `ask --verbose "compress this folder"`              |
+| Option                   | Description                | Example                                     |
+| ------------------------ | -------------------------- | ------------------------------------------- |
+| `--model provider:model` | Specify provider and model | `ask --model anthropic "list files"`        |
+|                          |                            | `ask --model anthropic:sonnet "list files"` |
+|                          |                            | `ask --model openai "list files"`           |
+|                          |                            | `ask --model gemini "list files"`           |
+|                          |                            | `ask --model gemini:pro "list files"`       |
+|                          |                            | `ask --model grok "list files"`             |
+| `--verbose`              | Enable verbose logging     | `ask --verbose "compress this folder"`      |
 
 ### Practical Examples
 
@@ -186,6 +191,7 @@ Ask follows XDG Base Directory Specification:
 export ANTHROPIC_API_KEY="your-anthropic-key"
 export OPENAI_API_KEY="your-openai-key"
 export GEMINI_API_KEY="your-gemini-key"
+export XAI_API_KEY="your-xai-key"
 ```
 
 ### Example Configuration File
@@ -201,7 +207,7 @@ model = "claude-3-haiku-20240307"
 max_tokens = 512
 
 [anthropic.sonnet]
-model = "claude-3-5-sonnet-20241022"
+model = "claude-sonnet-4-20250514"
 max_tokens = 1024
 
 [openai]
@@ -209,12 +215,17 @@ model = "gpt-4o"
 max_tokens = 1024
 
 [gemini]
-model = "gemini-2.5-flash"
+model = "gemini-2.5-flash-lite-preview-06-17"
 max_tokens = 150
 
 [gemini.pro]
 model = "gemini-2.5-pro"
 max_tokens = 1024
+
+[grok]
+model = "grok-3-fast"
+max_tokens = 150
+temperature = 0.0
 ```
 
 ## 🤖 Supported Providers
@@ -222,31 +233,15 @@ max_tokens = 1024
 - Anthropic (Claude)
 - OpenAI (GPT)
 - Google (Gemini)
+- xAI (Grok)
 
-> **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/)
+> **Note:** Get API keys from [Anthropic Console](https://console.anthropic.com/), [OpenAI Platform](https://platform.openai.com/), [Google AI Studio](https://aistudio.google.com/), or [xAI Console](https://x.ai/console)
 
 ## 🛣️ 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
-
+- [ ] Additional providers (Cohere, Mistral)
 - [ ] Local model support (Ollama, llama.cpp)
-- [ ] Learning from user preferences
-- [ ] Advanced safety and sandboxing
-- [ ] GUI and web interface options
 
 ## 🔧 Development
 
@@ -255,7 +250,7 @@ max_tokens = 1024
 ```bash
 git clone https://github.com/lcford2/terminal-sherpa.git
 cd ask
-uv sync
+uv sync --all-groups
 uv run pre-commit install
 ```
 
@@ -271,16 +266,13 @@ uv run python -m pytest
 2. Create a feature branch
 3. Make your changes
 4. Run pre-commit checks: `uv run pre-commit run --all-files`
-5. Submit a pull request
+5. Run tests: `uv run task test`
+6. Submit a pull request
 
 ## License
 
 This project is licensed under the MIT License - see the LICENSE file for details.
 
-## Contributing
-
-Contributions are welcome! Please see our [Contributing Guidelines](CONTRIBUTING.md) for details.
-
 ## Issues
 
 Found a bug or have a feature request? Please open an issue on [GitHub Issues](https://github.com/lcford2/ask/issues).

terminal_sherpa-0.4.0.dist-info/RECORD

@@ -0,0 +1,25 @@
+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=WRcLAqAK6dEbRM9aF5hYWw60_sjCtLlSBiyf8VFb4tA,1255
+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/grok.py,sha256=Lbue8dgcGoaSAtQ-Jme5e5EQ9hp608W78QEPjBP_vBk,3824
+ask/providers/openai.py,sha256=9PS1AgMr6Nb-OcYYiNYNrG384wNUS8m2lVT04K3hFV8,3683
+terminal_sherpa-0.4.0.dist-info/licenses/LICENSE,sha256=xLe81eIrf0X6CnEDDJXmoXuDzkdMYM3Eq1BgHUpG1JQ,1067
+test/conftest.py,sha256=aRJL41pj0ITemxvXO1FqANhF8zi4T1T8ehKNbZKBSao,1724
+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_grok.py,sha256=udlLJCCZ9NUYz7Szpy4bifTPtOR9-QMQ_S4ZDLNWhw8,8372
+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.4.0.dist-info/METADATA,sha256=_WxCjbuTNgns8yq5VFYkFs0LD_2dB50aaoBuUsIzGKk,7637
+terminal_sherpa-0.4.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+terminal_sherpa-0.4.0.dist-info/entry_points.txt,sha256=LxG9-J__nMGmeEIi47WVGYC1LLJo1GaADH21hfxEK70,38
+terminal_sherpa-0.4.0.dist-info/top_level.txt,sha256=Y7k5b2NSCkKiA_XPU-4fT_GYangD6JVDug5xwfXvmuQ,9
+terminal_sherpa-0.4.0.dist-info/RECORD,,

test/conftest.py

@@ -52,6 +52,13 @@ def mock_openai_key():
         yield
 
 
+@pytest.fixture
+def mock_grok_key():
+    """Mock Grok API key in environment."""
+    with patch.dict(os.environ, {"XAI_API_KEY": "test-grok-key"}, clear=True):
+        yield
+
+
 @pytest.fixture
 def mock_both_keys():
     """Mock both API keys in environment."""

test/test_grok.py

@@ -0,0 +1,247 @@
+"""Tests for Grok provider using xAI SDK."""
+
+import os
+from unittest.mock import MagicMock, patch
+
+import pytest
+
+from ask.config import SYSTEM_PROMPT
+from ask.exceptions import APIError, AuthenticationError, RateLimitError
+from ask.providers.grok import GrokProvider
+
+
+def test_grok_provider_init():
+    """Test provider initialization."""
+    config = {"model_name": "grok-3-fast"}
+    provider = GrokProvider(config)
+
+    assert provider.config == config
+    assert provider.client is None
+
+
+def test_validate_config_success(mock_grok_key):
+    """Test successful config validation."""
+    config = {"api_key_env": "XAI_API_KEY"}
+    provider = GrokProvider(config)
+
+    with patch("ask.providers.grok.Client") as mock_client_class:
+        mock_client = MagicMock()
+        mock_client_class.return_value = mock_client
+
+        provider.validate_config()
+
+        assert provider.client == mock_client
+        mock_client_class.assert_called_once_with(api_key="test-grok-key")
+
+
+def test_validate_config_missing_key(mock_env_vars):
+    """Test missing API key error."""
+    config = {"api_key_env": "XAI_API_KEY"}
+    provider = GrokProvider(config)
+
+    with pytest.raises(
+        AuthenticationError, match="XAI_API_KEY environment variable is required"
+    ):
+        provider.validate_config()
+
+
+def test_validate_config_custom_env():
+    """Test custom environment variable."""
+    config = {"api_key_env": "CUSTOM_XAI_KEY"}
+    provider = GrokProvider(config)
+
+    with patch.dict(os.environ, {"CUSTOM_XAI_KEY": "custom-xai-key"}):
+        with patch("ask.providers.grok.Client") as mock_client_class:
+            mock_client = MagicMock()
+            mock_client_class.return_value = mock_client
+
+            provider.validate_config()
+
+            assert provider.client == mock_client
+            mock_client_class.assert_called_once_with(api_key="custom-xai-key")
+
+
+@patch("ask.providers.grok.system")
+@patch("ask.providers.grok.user")
+def test_get_bash_command_success(mock_user, mock_system, mock_grok_key):
+    """Test successful bash command generation."""
+    config = {"model_name": "grok-beta", "max_tokens": 150}
+    provider = GrokProvider(config)
+
+    # Mock the xAI SDK workflow components
+    mock_chat = MagicMock()
+    mock_response = MagicMock()
+    mock_response.content = "ls -la"
+    mock_chat.sample.return_value = mock_response
+
+    mock_system_msg = MagicMock()
+    mock_user_msg = MagicMock()
+    mock_system.return_value = mock_system_msg
+    mock_user.return_value = mock_user_msg
+
+    with patch("ask.providers.grok.Client") as mock_client_class:
+        mock_client = MagicMock()
+        mock_client.chat.create.return_value = mock_chat
+        mock_client_class.return_value = mock_client
+        provider.client = mock_client
+
+        result = provider.get_bash_command("list files")
+
+        assert result == "ls -la"
+        mock_client.chat.create.assert_called_once_with(model="grok-beta")
+        mock_system.assert_called_once_with(SYSTEM_PROMPT)
+        mock_user.assert_called_once_with("list files")
+        assert mock_chat.append.call_count == 2
+        mock_chat.append.assert_any_call(mock_system_msg)
+        mock_chat.append.assert_any_call(mock_user_msg)
+        mock_chat.sample.assert_called_once()
+
+
+@patch("ask.providers.grok.system")
+@patch("ask.providers.grok.user")
+def test_get_bash_command_with_code_block(mock_user, mock_system, mock_grok_key):
+    """Test bash command extraction from code block."""
+    config = {}
+    provider = GrokProvider(config)
+
+    # Mock the xAI SDK workflow components
+    mock_chat = MagicMock()
+    mock_response = MagicMock()
+    mock_response.content = "```bash\nls -la\n```"
+    mock_chat.sample.return_value = mock_response
+
+    mock_system_msg = MagicMock()
+    mock_user_msg = MagicMock()
+    mock_system.return_value = mock_system_msg
+    mock_user.return_value = mock_user_msg
+
+    with patch("ask.providers.grok.Client") as mock_client_class:
+        mock_client = MagicMock()
+        mock_client.chat.create.return_value = mock_chat
+        mock_client_class.return_value = mock_client
+        provider.client = mock_client
+
+        result = provider.get_bash_command("list files")
+
+        assert result == "ls -la"
+
+
+@patch("ask.providers.grok.system")
+@patch("ask.providers.grok.user")
+def test_get_bash_command_empty_response(mock_user, mock_system, mock_grok_key):
+    """Test handling of empty API response."""
+    config = {}
+    provider = GrokProvider(config)
+
+    # Mock the xAI SDK workflow components
+    mock_chat = MagicMock()
+    mock_response = MagicMock()
+    mock_response.content = None
+    mock_chat.sample.return_value = mock_response
+
+    mock_system_msg = MagicMock()
+    mock_user_msg = MagicMock()
+    mock_system.return_value = mock_system_msg
+    mock_user.return_value = mock_user_msg
+
+    with patch("ask.providers.grok.Client") as mock_client_class:
+        mock_client = MagicMock()
+        mock_client.chat.create.return_value = mock_chat
+        mock_client_class.return_value = mock_client
+        provider.client = mock_client
+
+        with pytest.raises(APIError, match="API returned empty response"):
+            provider.get_bash_command("list files")
+
+
+def test_handle_authentication_error(mock_grok_key):
+    """Test authentication error handling."""
+    config = {}
+    provider = GrokProvider(config)
+
+    with patch("ask.providers.grok.Client") as mock_client_class:
+        mock_client = MagicMock()
+        mock_client.chat.create.side_effect = Exception("Authentication failed")
+        mock_client_class.return_value = mock_client
+        provider.client = mock_client
+
+        with pytest.raises(AuthenticationError, match="Invalid API key"):
+            provider.get_bash_command("test")
+
+
+def test_handle_rate_limit_error(mock_grok_key):
+    """Test rate limit error handling."""
+    config = {}
+    provider = GrokProvider(config)
+
+    with patch("ask.providers.grok.Client") as mock_client_class:
+        mock_client = MagicMock()
+        mock_client.chat.create.side_effect = Exception("Rate limit exceeded")
+        mock_client_class.return_value = mock_client
+        provider.client = mock_client
+
+        with pytest.raises(RateLimitError, match="rate limit exceeded"):
+            provider.get_bash_command("test")
+
+
+def test_handle_general_api_error(mock_grok_key):
+    """Test general API error handling."""
+    config = {}
+    provider = GrokProvider(config)
+
+    with patch("ask.providers.grok.Client") as mock_client_class:
+        mock_client = MagicMock()
+        mock_client.chat.create.side_effect = Exception("Unknown error")
+        mock_client_class.return_value = mock_client
+        provider.client = mock_client
+
+        with pytest.raises(APIError, match="API request failed - Unknown error"):
+            provider.get_bash_command("test")
+
+
+def test_get_default_config():
+    """Test default configuration."""
+    default_config = GrokProvider.get_default_config()
+
+    expected_config = {
+        "model_name": "grok-3-fast",
+        "max_tokens": 150,
+        "api_key_env": "XAI_API_KEY",
+        "temperature": 0.0,
+        "system_prompt": SYSTEM_PROMPT,
+    }
+
+    assert default_config == expected_config
+
+
+@patch("ask.providers.grok.system")
+@patch("ask.providers.grok.user")
+def test_get_bash_command_multiline_code_block(mock_user, mock_system, mock_grok_key):
+    """Test bash command extraction from multiline code block."""
+    config = {}
+    provider = GrokProvider(config)
+
+    # Mock the xAI SDK workflow components
+    mock_chat = MagicMock()
+    mock_response = MagicMock()
+    mock_response.content = (
+        "```bash\nfind . -name '*.py' \\\n  "
+        "-type f \\\n  -exec grep -l 'test' {} \\;\n```"
+    )
+    mock_chat.sample.return_value = mock_response
+
+    mock_system_msg = MagicMock()
+    mock_user_msg = MagicMock()
+    mock_system.return_value = mock_system_msg
+    mock_user.return_value = mock_user_msg
+
+    with patch("ask.providers.grok.Client") as mock_client_class:
+        mock_client = MagicMock()
+        mock_client.chat.create.return_value = mock_chat
+        mock_client_class.return_value = mock_client
+        provider.client = mock_client
+
+        result = provider.get_bash_command("find Python files with test")
+
+        expected = "find . -name '*.py' \\\n  -type f \\\n  -exec grep -l 'test' {} \\;"
+        assert result == expected

terminal_sherpa-0.3.0.dist-info/RECORD

@@ -1,23 +0,0 @@
-ask/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ask/config.py,sha256=12YKck6Q9vaSESfuI7kKFbs1hkxzts2FOsxR9eK96MY,2899
-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=M6bcOxDbbWQrJ6kWWtYqsVk6NsZOqX7PzxyiI_yji1Q,2738
-ask/providers/base.py,sha256=91ZbVORYWckSHNwNPiTmgfqQN0FLO9AgV6mptuAkIU0,769
-ask/providers/gemini.py,sha256=taFZYEygiEf05XCm4AKSKL2F_BQwLrU4Y5Ac-WN5owk,3421
-ask/providers/openai.py,sha256=jVyRH4FRdF_91iuK5Tga4as9zbyGKPPFW90ewGG5s5k,3696
-terminal_sherpa-0.3.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.3.0.dist-info/METADATA,sha256=r0ARy4RTO3vnlxn6tizVE-aEDLj79MyXLTl5ZxLp0Yk,7521
-terminal_sherpa-0.3.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-terminal_sherpa-0.3.0.dist-info/entry_points.txt,sha256=LxG9-J__nMGmeEIi47WVGYC1LLJo1GaADH21hfxEK70,38
-terminal_sherpa-0.3.0.dist-info/top_level.txt,sha256=Y7k5b2NSCkKiA_XPU-4fT_GYangD6JVDug5xwfXvmuQ,9
-terminal_sherpa-0.3.0.dist-info/RECORD,,