gac 1.13.0__py3-none-any.whl → 3.6.0__py3-none-any.whl

gac/providers/__init__.py

@@ -1,42 +1,88 @@
 """AI provider implementations for commit message generation."""
 
 from .anthropic import call_anthropic_api
+from .azure_openai import call_azure_openai_api
 from .cerebras import call_cerebras_api
 from .chutes import call_chutes_api
+from .claude_code import call_claude_code_api
 from .custom_anthropic import call_custom_anthropic_api
 from .custom_openai import call_custom_openai_api
 from .deepseek import call_deepseek_api
 from .fireworks import call_fireworks_api
 from .gemini import call_gemini_api
 from .groq import call_groq_api
+from .kimi_coding import call_kimi_coding_api
 from .lmstudio import call_lmstudio_api
 from .minimax import call_minimax_api
+from .mistral import call_mistral_api
+from .moonshot import call_moonshot_api
 from .ollama import call_ollama_api
 from .openai import call_openai_api
 from .openrouter import call_openrouter_api
+from .replicate import call_replicate_api
 from .streamlake import call_streamlake_api
 from .synthetic import call_synthetic_api
 from .together import call_together_api
 from .zai import call_zai_api, call_zai_coding_api
 
+# Provider registry - single source of truth for all providers
+PROVIDER_REGISTRY = {
+    "anthropic": call_anthropic_api,
+    "azure-openai": call_azure_openai_api,
+    "cerebras": call_cerebras_api,
+    "claude-code": call_claude_code_api,
+    "chutes": call_chutes_api,
+    "custom-anthropic": call_custom_anthropic_api,
+    "custom-openai": call_custom_openai_api,
+    "deepseek": call_deepseek_api,
+    "fireworks": call_fireworks_api,
+    "gemini": call_gemini_api,
+    "groq": call_groq_api,
+    "kimi-coding": call_kimi_coding_api,
+    "lm-studio": call_lmstudio_api,
+    "minimax": call_minimax_api,
+    "mistral": call_mistral_api,
+    "moonshot": call_moonshot_api,
+    "ollama": call_ollama_api,
+    "openai": call_openai_api,
+    "openrouter": call_openrouter_api,
+    "replicate": call_replicate_api,
+    "streamlake": call_streamlake_api,
+    "synthetic": call_synthetic_api,
+    "together": call_together_api,
+    "zai": call_zai_api,
+    "zai-coding": call_zai_coding_api,
+}
+
+# List of supported provider names - derived from registry keys
+SUPPORTED_PROVIDERS = sorted(PROVIDER_REGISTRY.keys())
+
 __all__ = [
     "call_anthropic_api",
+    "call_azure_openai_api",
     "call_cerebras_api",
     "call_chutes_api",
+    "call_claude_code_api",
     "call_custom_anthropic_api",
     "call_custom_openai_api",
     "call_deepseek_api",
     "call_fireworks_api",
     "call_gemini_api",
     "call_groq_api",
+    "call_kimi_coding_api",
     "call_lmstudio_api",
     "call_minimax_api",
+    "call_mistral_api",
+    "call_moonshot_api",
     "call_ollama_api",
     "call_openai_api",
     "call_openrouter_api",
+    "call_replicate_api",
     "call_streamlake_api",
     "call_synthetic_api",
     "call_together_api",
     "call_zai_api",
     "call_zai_coding_api",
+    "PROVIDER_REGISTRY",
+    "SUPPORTED_PROVIDERS",
 ]

gac/providers/azure_openai.py

@@ -0,0 +1,97 @@
+"""Azure OpenAI provider for gac.
+
+This provider provides native support for Azure OpenAI Service with proper
+endpoint construction and API version handling.
+"""
+
+import json
+import logging
+import os
+
+import httpx
+
+from gac.errors import AIError
+
+logger = logging.getLogger(__name__)
+
+
+def call_azure_openai_api(model: str, messages: list[dict], temperature: float, max_tokens: int) -> str:
+    """Call Azure OpenAI Service API.
+
+    Environment variables:
+        AZURE_OPENAI_API_KEY: Azure OpenAI API key (required)
+        AZURE_OPENAI_ENDPOINT: Azure OpenAI endpoint URL (required)
+            Example: https://your-resource.openai.azure.com
+        AZURE_OPENAI_API_VERSION: Azure OpenAI API version (required)
+            Example: 2025-01-01-preview
+            Example: 2024-02-15-preview
+
+    Args:
+        model: The deployment name in Azure OpenAI (e.g., 'gpt-4o', 'gpt-35-turbo')
+        messages: List of message dictionaries with 'role' and 'content' keys
+        temperature: Controls randomness (0.0-1.0)
+        max_tokens: Maximum tokens in the response
+
+    Returns:
+        The generated commit message
+
+    Raises:
+        AIError: If authentication fails, API errors occur, or response is invalid
+    """
+    api_key = os.getenv("AZURE_OPENAI_API_KEY")
+    if not api_key:
+        raise AIError.authentication_error("AZURE_OPENAI_API_KEY environment variable not set")
+
+    endpoint = os.getenv("AZURE_OPENAI_ENDPOINT")
+    if not endpoint:
+        raise AIError.model_error("AZURE_OPENAI_ENDPOINT environment variable not set")
+
+    api_version = os.getenv("AZURE_OPENAI_API_VERSION")
+    if not api_version:
+        raise AIError.model_error("AZURE_OPENAI_API_VERSION environment variable not set")
+
+    # Build Azure OpenAI URL with proper structure
+    endpoint = endpoint.rstrip("/")
+    url = f"{endpoint}/openai/deployments/{model}/chat/completions?api-version={api_version}"
+
+    headers = {"api-key": api_key, "Content-Type": "application/json"}
+
+    data = {"messages": messages, "temperature": temperature, "max_tokens": max_tokens}
+
+    try:
+        response = httpx.post(url, headers=headers, json=data, timeout=120)
+        response.raise_for_status()
+        response_data = response.json()
+
+        try:
+            content = response_data["choices"][0]["message"]["content"]
+        except (KeyError, IndexError, TypeError) as e:
+            logger.error(f"Unexpected response format from Azure OpenAI API. Response: {json.dumps(response_data)}")
+            raise AIError.model_error(
+                f"Azure OpenAI API returned unexpected format. Expected response with "
+                f"'choices[0].message.content', but got: {type(e).__name__}. Check logs for full response structure."
+            ) from e
+
+        if content is None:
+            raise AIError.model_error("Azure OpenAI API returned null content")
+        if content == "":
+            raise AIError.model_error("Azure OpenAI API returned empty content")
+        return content
+    except httpx.ConnectError as e:
+        raise AIError.connection_error(f"Azure OpenAI API connection failed: {str(e)}") from e
+    except httpx.HTTPStatusError as e:
+        status_code = e.response.status_code
+        error_text = e.response.text
+
+        if status_code == 401:
+            raise AIError.authentication_error(f"Azure OpenAI API authentication failed: {error_text}") from e
+        elif status_code == 429:
+            raise AIError.rate_limit_error(f"Azure OpenAI API rate limit exceeded: {error_text}") from e
+        else:
+            raise AIError.model_error(f"Azure OpenAI API error: {status_code} - {error_text}") from e
+    except httpx.TimeoutException as e:
+        raise AIError.timeout_error(f"Azure OpenAI API request timed out: {str(e)}") from e
+    except AIError:
+        raise
+    except Exception as e:
+        raise AIError.model_error(f"Error calling Azure OpenAI API: {str(e)}") from e

gac/providers/claude_code.py

@@ -0,0 +1,102 @@
+"""Claude Code provider implementation.
+
+This provider allows users with Claude Code subscriptions to use their OAuth tokens
+instead of paying for the expensive Anthropic API.
+"""
+
+import os
+
+import httpx
+
+from gac.errors import AIError
+
+
+def call_claude_code_api(model: str, messages: list[dict], temperature: float, max_tokens: int) -> str:
+    """Call Claude Code API using OAuth token.
+
+    This provider uses the Claude Code subscription OAuth token instead of the Anthropic API key.
+    It authenticates using Bearer token authentication with the special anthropic-beta header.
+
+    Environment variables:
+        CLAUDE_CODE_ACCESS_TOKEN: OAuth access token from Claude Code authentication
+
+    Args:
+        model: Model name (e.g., 'claude-sonnet-4-5')
+        messages: List of message dictionaries with 'role' and 'content' keys
+        temperature: Sampling temperature (0.0-1.0)
+        max_tokens: Maximum tokens in response
+
+    Returns:
+        Generated text response
+
+    Raises:
+        AIError: If authentication fails or API call fails
+    """
+    access_token = os.getenv("CLAUDE_CODE_ACCESS_TOKEN")
+    if not access_token:
+        raise AIError.authentication_error(
+            "CLAUDE_CODE_ACCESS_TOKEN not found in environment variables. "
+            "Please authenticate with Claude Code and set this token."
+        )
+
+    url = "https://api.anthropic.com/v1/messages"
+    headers = {
+        "Authorization": f"Bearer {access_token}",
+        "anthropic-version": "2023-06-01",
+        "anthropic-beta": "oauth-2025-04-20",
+        "content-type": "application/json",
+    }
+
+    # Convert messages to Anthropic format
+    # IMPORTANT: Claude Code OAuth tokens require the system message to be EXACTLY
+    # "You are Claude Code, Anthropic's official CLI for Claude." with NO additional content.
+    # Any other instructions must be moved to the user message.
+    anthropic_messages = []
+    system_instructions = ""
+
+    for msg in messages:
+        if msg["role"] == "system":
+            system_instructions = msg["content"]
+        else:
+            anthropic_messages.append({"role": msg["role"], "content": msg["content"]})
+
+    # Claude Code requires this exact system message, nothing more
+    system_message = "You are Claude Code, Anthropic's official CLI for Claude."
+
+    # Move any system instructions into the first user message
+    if system_instructions and anthropic_messages:
+        # Prepend system instructions to the first user message
+        first_user_msg = anthropic_messages[0]
+        first_user_msg["content"] = f"{system_instructions}\n\n{first_user_msg['content']}"
+
+    data = {
+        "model": model,
+        "messages": anthropic_messages,
+        "temperature": temperature,
+        "max_tokens": max_tokens,
+        "system": system_message,
+    }
+
+    try:
+        response = httpx.post(url, headers=headers, json=data, timeout=120)
+        response.raise_for_status()
+        response_data = response.json()
+        content = response_data["content"][0]["text"]
+        if content is None:
+            raise AIError.model_error("Claude Code API returned null content")
+        if content == "":
+            raise AIError.model_error("Claude Code API returned empty content")
+        return content
+    except httpx.HTTPStatusError as e:
+        if e.response.status_code == 401:
+            raise AIError.authentication_error(
+                f"Claude Code authentication failed: {e.response.text}. "
+                "Your token may have expired. Please re-authenticate."
+            ) from e
+        if e.response.status_code == 429:
+            raise AIError.rate_limit_error(f"Claude Code API rate limit exceeded: {e.response.text}") from e
+        raise AIError.model_error(f"Claude Code API error: {e.response.status_code} - {e.response.text}") from e
+    except httpx.TimeoutException as e:
+        raise AIError.timeout_error(f"Claude Code API request timed out: {str(e)}") from e
+    except Exception as e:
+        raise AIError.model_error(f"Error calling Claude Code API: {str(e)}") from e

gac/providers/custom_anthropic.py

@@ -30,7 +30,7 @@ def call_custom_anthropic_api(model: str, messages: list[dict], temperature: flo
         CUSTOM_ANTHROPIC_VERSION: API version header (optional, defaults to '2023-06-01')
 
     Args:
-        model: The model to use (e.g., 'claude-3-5-sonnet-20241022', 'claude-3-5-haiku-latest')
+        model: The model to use (e.g., 'claude-sonnet-4-5', 'claude-haiku-4-5')
         messages: List of message dictionaries with 'role' and 'content' keys
         temperature: Controls randomness (0.0-1.0)
         max_tokens: Maximum tokens in the response

gac/providers/custom_openai.py

@@ -19,7 +19,6 @@ def call_custom_openai_api(model: str, messages: list[dict], temperature: float,
     """Call a custom OpenAI-compatible API endpoint.
 
     This provider is useful for:
-    - Azure OpenAI Service
     - OpenAI-compatible proxies or gateways
     - Self-hosted OpenAI-compatible services
     - Other services implementing the OpenAI Chat Completions API
@@ -27,8 +26,8 @@ def call_custom_openai_api(model: str, messages: list[dict], temperature: float,
     Environment variables:
         CUSTOM_OPENAI_API_KEY: API key for authentication (required)
         CUSTOM_OPENAI_BASE_URL: Base URL for the API endpoint (required)
-            Example: https://your-endpoint.openai.azure.com
             Example: https://your-proxy.example.com/v1
+            Example: https://your-custom-endpoint.com
 
     Args:
         model: The model to use (e.g., 'gpt-4', 'gpt-3.5-turbo')

gac/providers/kimi_coding.py

@@ -0,0 +1,63 @@
+"""Kimi Coding AI provider implementation."""
+
+import json
+import logging
+import os
+
+import httpx
+
+from gac.errors import AIError
+
+logger = logging.getLogger(__name__)
+
+
+def call_kimi_coding_api(model: str, messages: list[dict], temperature: float, max_tokens: int) -> str:
+    """Call Kimi Coding API using OpenAI-compatible endpoint."""
+    api_key = os.getenv("KIMI_CODING_API_KEY")
+    if not api_key:
+        raise AIError.authentication_error("KIMI_CODING_API_KEY not found in environment variables")
+
+    base_url = "https://api.kimi.com/coding/v1"
+    url = f"{base_url}/chat/completions"
+    headers = {"Authorization": f"Bearer {api_key}", "Content-Type": "application/json"}
+
+    # Use standard OpenAI format - no message conversion needed
+    data = {"model": model, "messages": messages, "temperature": temperature, "max_completion_tokens": max_tokens}
+
+    try:
+        response = httpx.post(url, headers=headers, json=data, timeout=120)
+        response.raise_for_status()
+        response_data = response.json()
+
+        try:
+            content = response_data["choices"][0]["message"]["content"]
+        except (KeyError, IndexError, TypeError) as e:
+            logger.error(f"Unexpected response format from Kimi Coding API. Response: {json.dumps(response_data)}")
+            raise AIError.model_error(
+                f"Kimi Coding API returned unexpected format. Expected OpenAI-compatible response with "
+                f"'choices[0].message.content', but got: {type(e).__name__}. Check logs for full response structure."
+            ) from e
+
+        if content is None:
+            raise AIError.model_error("Kimi Coding API returned null content")
+        if content == "":
+            raise AIError.model_error("Kimi Coding API returned empty content")
+        return content
+    except httpx.ConnectError as e:
+        raise AIError.connection_error(f"Kimi Coding API connection failed: {str(e)}") from e
+    except httpx.HTTPStatusError as e:
+        status_code = e.response.status_code
+        error_text = e.response.text
+
+        if status_code == 401:
+            raise AIError.authentication_error(f"Kimi Coding API authentication failed: {error_text}") from e
+        elif status_code == 429:
+            raise AIError.rate_limit_error(f"Kimi Coding API rate limit exceeded: {error_text}") from e
+        else:
+            raise AIError.model_error(f"Kimi Coding API error: {status_code} - {error_text}") from e
+    except httpx.TimeoutException as e:
+        raise AIError.timeout_error(f"Kimi Coding API request timed out: {str(e)}") from e
+    except AIError:
+        raise
+    except Exception as e:
+        raise AIError.model_error(f"Error calling Kimi Coding API: {str(e)}") from e

gac/providers/mistral.py

@@ -0,0 +1,38 @@
+"""Mistral API provider for gac."""
+
+import os
+
+import httpx
+
+from gac.errors import AIError
+
+
+def call_mistral_api(model: str, messages: list[dict], temperature: float, max_tokens: int) -> str:
+    """Call Mistral API directly."""
+    api_key = os.getenv("MISTRAL_API_KEY")
+    if not api_key:
+        raise AIError.authentication_error("MISTRAL_API_KEY not found in environment variables")
+
+    url = "https://api.mistral.ai/v1/chat/completions"
+    headers = {"Authorization": f"Bearer {api_key}", "Content-Type": "application/json"}
+
+    data = {"model": model, "messages": messages, "temperature": temperature, "max_tokens": max_tokens}
+
+    try:
+        response = httpx.post(url, headers=headers, json=data, timeout=120)
+        response.raise_for_status()
+        response_data = response.json()
+        content = response_data["choices"][0]["message"]["content"]
+        if content is None:
+            raise AIError.model_error("Mistral API returned null content")
+        if content == "":
+            raise AIError.model_error("Mistral API returned empty content")
+        return content
+    except httpx.HTTPStatusError as e:
+        if e.response.status_code == 429:
+            raise AIError.rate_limit_error(f"Mistral API rate limit exceeded: {e.response.text}") from e
+        raise AIError.model_error(f"Mistral API error: {e.response.status_code} - {e.response.text}") from e
+    except httpx.TimeoutException as e:
+        raise AIError.timeout_error(f"Mistral API request timed out: {str(e)}") from e
+    except Exception as e:
+        raise AIError.model_error(f"Error calling Mistral API: {str(e)}") from e

gac/providers/moonshot.py

@@ -0,0 +1,38 @@
+"""Moonshot AI provider for gac."""
+
+import os
+
+import httpx
+
+from gac.errors import AIError
+
+
+def call_moonshot_api(model: str, messages: list[dict], temperature: float, max_tokens: int) -> str:
+    """Call Moonshot AI API directly."""
+    api_key = os.getenv("MOONSHOT_API_KEY")
+    if not api_key:
+        raise AIError.authentication_error("MOONSHOT_API_KEY not found in environment variables")
+
+    url = "https://api.moonshot.ai/v1/chat/completions"
+    headers = {"Authorization": f"Bearer {api_key}", "Content-Type": "application/json"}
+
+    data = {"model": model, "messages": messages, "temperature": temperature, "max_tokens": max_tokens}
+
+    try:
+        response = httpx.post(url, headers=headers, json=data, timeout=120)
+        response.raise_for_status()
+        response_data = response.json()
+        content = response_data["choices"][0]["message"]["content"]
+        if content is None:
+            raise AIError.model_error("Moonshot AI API returned null content")
+        if content == "":
+            raise AIError.model_error("Moonshot AI API returned empty content")
+        return content
+    except httpx.HTTPStatusError as e:
+        if e.response.status_code == 429:
+            raise AIError.rate_limit_error(f"Moonshot AI API rate limit exceeded: {e.response.text}") from e
+        raise AIError.model_error(f"Moonshot AI API error: {e.response.status_code} - {e.response.text}") from e
+    except httpx.TimeoutException as e:
+        raise AIError.timeout_error(f"Moonshot AI API request timed out: {str(e)}") from e
+    except Exception as e:
+        raise AIError.model_error(f"Error calling Moonshot AI API: {str(e)}") from e

gac/providers/replicate.py

@@ -0,0 +1,98 @@
+"""Replicate API provider for gac."""
+
+import os
+
+import httpx
+
+from gac.errors import AIError
+
+
+def call_replicate_api(model: str, messages: list[dict], temperature: float, max_tokens: int) -> str:
+    """Call Replicate API directly."""
+    api_key = os.getenv("REPLICATE_API_TOKEN")
+    if not api_key:
+        raise AIError.authentication_error("REPLICATE_API_TOKEN not found in environment variables")
+
+    # Replicate uses a different endpoint for language models
+    url = "https://api.replicate.com/v1/predictions"
+    headers = {"Authorization": f"Token {api_key}", "Content-Type": "application/json"}
+
+    # Convert messages to a single prompt for Replicate
+    prompt_parts = []
+    system_message = None
+
+    for message in messages:
+        role = message.get("role")
+        content = message.get("content", "")
+
+        if role == "system":
+            system_message = content
+        elif role == "user":
+            prompt_parts.append(f"Human: {content}")
+        elif role == "assistant":
+            prompt_parts.append(f"Assistant: {content}")
+
+    # Add system message at the beginning if present
+    if system_message:
+        prompt_parts.insert(0, f"System: {system_message}")
+
+    # Add final assistant prompt
+    prompt_parts.append("Assistant:")
+    full_prompt = "\n\n".join(prompt_parts)
+
+    # Replicate prediction payload
+    data = {
+        "version": model,  # Replicate uses version string as model identifier
+        "input": {
+            "prompt": full_prompt,
+            "temperature": temperature,
+            "max_tokens": max_tokens,
+        },
+    }
+
+    try:
+        # Create prediction
+        response = httpx.post(url, headers=headers, json=data, timeout=120)
+        response.raise_for_status()
+        prediction_data = response.json()
+
+        # Get the prediction URL to check status
+        get_url = f"https://api.replicate.com/v1/predictions/{prediction_data['id']}"
+
+        # Poll for completion (Replicate predictions are async)
+        max_wait_time = 120
+        wait_interval = 2
+        elapsed_time = 0
+
+        while elapsed_time < max_wait_time:
+            get_response = httpx.get(get_url, headers=headers, timeout=120)
+            get_response.raise_for_status()
+            status_data = get_response.json()
+
+            if status_data["status"] == "succeeded":
+                content = status_data["output"]
+                if not content:
+                    raise AIError.model_error("Replicate API returned empty content")
+                return content
+            elif status_data["status"] == "failed":
+                raise AIError.model_error(f"Replicate prediction failed: {status_data.get('error', 'Unknown error')}")
+            elif status_data["status"] in ["starting", "processing"]:
+                import time
+
+                time.sleep(wait_interval)
+                elapsed_time += wait_interval
+            else:
+                raise AIError.model_error(f"Replicate API returned unknown status: {status_data['status']}")
+
+        raise AIError.timeout_error("Replicate API prediction timed out")
+
+    except httpx.HTTPStatusError as e:
+        if e.response.status_code == 429:
+            raise AIError.rate_limit_error(f"Replicate API rate limit exceeded: {e.response.text}") from e
+        elif e.response.status_code == 401:
+            raise AIError.authentication_error(f"Replicate API authentication failed: {e.response.text}") from e
+        raise AIError.model_error(f"Replicate API error: {e.response.status_code} - {e.response.text}") from e
+    except httpx.TimeoutException as e:
+        raise AIError.timeout_error(f"Replicate API request timed out: {str(e)}") from e
+    except Exception as e:
+        raise AIError.model_error(f"Error calling Replicate API: {str(e)}") from e

gac/security.py

@@ -54,7 +54,7 @@ class SecretPatterns:
     PRIVATE_KEY = re.compile(r"-----BEGIN (?:RSA |EC |OPENSSH )?PRIVATE KEY-----")
 
     # Bearer Tokens (require actual token with specific characteristics)
-    BEARER_TOKEN = re.compile(r"Bearer\s+[A-Za-z0-9]{20,}[/=]*\s", re.IGNORECASE)
+    BEARER_TOKEN = re.compile(r"Bearer\s+[A-Za-z0-9]{20,}[/=]*(?:\s|$)", re.IGNORECASE)
 
     # JWT Tokens
     JWT_TOKEN = re.compile(r"eyJ[A-Za-z0-9_-]+\.eyJ[A-Za-z0-9_-]+\.[A-Za-z0-9_-]+")