gitai-cli 0.1.0__py3-none-any.whl

gitai/__init__.py

@@ -0,0 +1 @@
+__version__ = "0.1.0"

gitai/ai.py

@@ -0,0 +1,52 @@
+import litellm
+from litellm.exceptions import APIConnectionError, AuthenticationError
+from gitai.config import load_config
+
+def get_commit_suggestions(prompt: str) -> list[str]:
+    config = load_config()
+
+    provider = config["provider"].lower()
+    model = config["model"]
+    model_string = f"{provider}/{model}"
+
+    kwargs = {
+        "model": model_string,
+        "messages": [{"role": "user", "content": prompt}],
+    }
+
+    if provider == "ollama":
+        kwargs["api_base"] = config["ollama_url"]
+
+    try:
+        response = litellm.completion(**kwargs)
+    except APIConnectionError:
+        if provider == "ollama":
+            raise SystemExit(
+                f"[gitai] Could not connect to Ollama at {config['ollama_url']}. "
+                "Is Ollama running? Try: ollama serve"
+            )
+        raise SystemExit(
+            f"[gitai] Could not connect to the {provider} API. "
+            "Check your network connection and try again."
+        )
+    except AuthenticationError:
+        key_map = {
+            "openai": "OPENAI_API_KEY",
+            "anthropic": "ANTHROPIC_API_KEY",
+            "gemini": "GEMINI_API_KEY",
+        }
+        env_var = key_map.get(provider, f"{provider.upper()}_API_KEY")
+        raise SystemExit(
+            f"[gitai] Authentication failed for {provider}. "
+            f"Set the {env_var} environment variable and try again."
+        )
+
+    text = response.choices[0].message.content.strip()
+
+    suggestions = []
+    for line in text.splitlines():
+        line = line.strip()
+        if line and line[0].isdigit() and "." in line:
+            suggestions.append(line)
+
+    return suggestions

gitai/cli.py

@@ -0,0 +1,132 @@
+from typing import Optional
+import typer
+import questionary
+import subprocess
+from gitai.config import load_config, save_config
+from gitai.git import get_staged_diff, get_repo_name, is_diff_meaningful
+from gitai.prompt import build_commit_prompt
+from gitai.ai import get_commit_suggestions
+from gitai import __version__
+
+VALID_PROVIDERS = {"ollama", "openai", "anthropic", "gemini"}
+VALID_COMMIT_STYLES = {"conventional", "free-form"}
+
+app = typer.Typer()
+
+
+def _version_callback(value: bool):
+    if value:
+        typer.echo(f"gitai {__version__}")
+        raise typer.Exit()
+
+
+@app.callback()
+def main(
+    version: Optional[bool] = typer.Option(
+        None, "--version", "-V",
+        callback=_version_callback,
+        is_eager=True,
+        help="Show version and exit.",
+    ),
+):
+    """AI-powered git commit message generator."""
+
+
+@app.command()
+def commit():
+    """Generate AI-powered commit message suggestions for staged changes."""
+    typer.echo("🔍 Reading your git diff...")
+
+    diff = get_staged_diff()
+    if not diff:
+        typer.echo("No staged changes found. Please stage your changes before committing.")
+        raise typer.Exit(code=1)
+
+    if not is_diff_meaningful(diff):
+        typer.echo("Staged file appears to be empty or has no content changes.")
+        chosen = typer.prompt("Enter your commit message")
+        subprocess.run(["git", "commit", "-m", chosen])
+        typer.echo(f"\n Committed: {chosen}")
+        raise typer.Exit()
+
+    config = load_config()
+    repo_name = get_repo_name()
+    prompt = build_commit_prompt(diff, repo_name, emoji=config["emoji"], commit_style=config["commit_style"])
+
+    typer.echo("Generating commit message suggestions...")
+    suggestions = get_commit_suggestions(prompt)
+
+    if not suggestions:
+        typer.echo("No suggestions generated. Please try again.")
+        raise typer.Exit()
+
+    clean = [s.split(". ", 1)[1] if ". " in s else s for s in suggestions]
+    clean.append("Write my own")
+
+    chosen = questionary.select(
+        "Choose a commit message:",
+        choices=clean
+    ).ask()
+
+    if chosen is None:
+        typer.echo("Aborted.")
+        raise typer.Exit()
+
+    if chosen == "Write my own":
+        chosen = typer.prompt("Enter your custom commit message")
+
+    subprocess.run(["git", "commit", "-m", chosen])
+    typer.echo(f"\n Committed: {chosen}")
+
+
+@app.command()
+def config():
+    """View and update gitai settings."""
+    current = load_config()
+
+    typer.echo("Current configuration:\n")
+    for key, value in current.items():
+        typer.echo(f"  {key}: {value}")
+
+    typer.echo("")
+    if not typer.confirm("Do you want to change any settings?"):
+        raise typer.Exit()
+
+    provider = typer.prompt(
+        "Provider (ollama, openai, anthropic, gemini)",
+        default=current["provider"],
+    )
+    if provider not in VALID_PROVIDERS:
+        typer.echo(f"[gitai] Unknown provider '{provider}'. Choose from: {', '.join(sorted(VALID_PROVIDERS))}")
+        raise typer.Exit(code=1)
+
+    model = typer.prompt("Model name", default=current["model"])
+
+    ollama_url = current["ollama_url"]
+    if provider == "ollama":
+        ollama_url = typer.prompt("Ollama URL", default=current["ollama_url"])
+
+    commit_style = typer.prompt(
+        "Commit style (conventional, free-form)",
+        default=current["commit_style"],
+    )
+    if commit_style not in VALID_COMMIT_STYLES:
+        typer.echo(f"[gitai] Unknown commit style '{commit_style}'. Choose from: {', '.join(sorted(VALID_COMMIT_STYLES))}")
+        raise typer.Exit(code=1)
+
+    emoji = typer.confirm("Use emojis in commit messages?", default=current["emoji"])
+
+    new_config = {
+        "model": model,
+        "provider": provider,
+        "ollama_url": ollama_url,
+        "commit_style": commit_style,
+        "emoji": emoji,
+    }
+
+    save_config(new_config)
+    typer.echo("\n✅ Config saved to ~/.gitai.toml")
+
+
+if __name__ == "__main__":
+    app()

gitai/config.py

@@ -0,0 +1,27 @@
+import tomllib
+import tomli_w
+from pathlib import Path
+
+CONFIG_PATH = Path.home() / ".gitai.toml"
+
+DEFAULT_CONFIG = {
+    "model": "llama3.2",
+    "provider": "ollama",
+    "ollama_url": "http://localhost:11434",
+    "commit_style": "conventional",
+    "emoji": False,
+}
+
+def load_config() -> dict:
+    if not CONFIG_PATH.exists():
+        save_config(DEFAULT_CONFIG)
+        return DEFAULT_CONFIG.copy()
+    
+    with open(CONFIG_PATH, "rb") as f:
+        user_config = tomllib.load(f)
+
+    return {**DEFAULT_CONFIG, **user_config}
+
+def save_config(config: dict) -> None:
+    with open(CONFIG_PATH, "wb") as f:
+        tomli_w.dump(config, f)

gitai/git.py

@@ -0,0 +1,24 @@
+import subprocess
+from pathlib import Path
+
+def get_staged_diff() -> str:
+    result = subprocess.run(
+        ["git", "diff", "--cached"],
+        capture_output=True,
+        text=True,
+        encoding="utf-8",
+    )
+    return result.stdout
+
+def get_repo_name() -> str:
+    path = Path.cwd()
+    return path.name
+
+def is_diff_meaningful(diff: str) -> bool:
+    meaningful_lines = [
+        line for line in diff.splitlines()
+        if line.startswith(("+", "-"))
+        and not line.startswith(("+++", "---"))
+        and line.strip() not in ("+", "-", "")
+    ]
+    return len(meaningful_lines) > 0

gitai/prompt.py

@@ -0,0 +1,73 @@
+def build_commit_prompt(diff: str, repo_name: str, emoji: bool = False, commit_style: str = "conventional") -> str:
+    format_rules = _build_format_rules(commit_style, emoji)
+    example = _build_example(commit_style, emoji)
+
+    return f"""You are an expert developer generating git commit messages.
+
+Repository: {repo_name}
+
+Analyze the git diff below and return exactly 3 commit message suggestions.
+
+Rules:
+{format_rules}
+- Each suggestion must offer a meaningfully different angle — vary the type, scope, or emphasis
+- Base suggestions strictly on what you see in the diff — never invent context
+- Output: a numbered list of exactly 3 lines, no intro, no explanation, nothing else
+
+Example output:
+{example}
+
+Git diff:
+{diff}
+"""
+
+
+def _build_format_rules(commit_style: str, emoji: bool) -> str:
+    if commit_style == "free-form":
+        rules = (
+            "- Format: a single clear imperative sentence, no type prefix required\n"
+            "- Keep the message under 72 characters\n"
+            '- Imperative mood ("add" not "added"), lowercase, no trailing period'
+        )
+        if emoji:
+            rules += "\n- Prefix each message with a relevant emoji that reflects the nature of the change"
+    else:  # conventional (default)
+        rules = (
+            "- Format: type(scope): description\n"
+            "- Types: feat, fix, refactor, chore, docs, style, test, perf, ci, build\n"
+            "- Scope: infer from the file paths or module names changed; omit if unclear\n"
+            '- Description: imperative mood ("add" not "added"), lowercase, no trailing period, under 72 characters'
+        )
+        if emoji:
+            rules += (
+                "\n- Prefix each message with the matching gitmoji before the type: "
+                "✨ feat, 🐛 fix, ♻️ refactor, 🔧 chore, 📝 docs, 🎨 style, ✅ test, ⚡ perf, 👷 ci, 📦 build"
+            )
+    return rules
+
+
+def _build_example(commit_style: str, emoji: bool) -> str:
+    if commit_style == "free-form":
+        if emoji:
+            return (
+                "1. ✨ add refresh token rotation on session expiry\n"
+                "2. 🐛 prevent reuse of invalidated tokens\n"
+                "3. ♻️ extract token validation into dedicated service"
+            )
+        return (
+            "1. add refresh token rotation on session expiry\n"
+            "2. prevent reuse of invalidated tokens\n"
+            "3. extract token validation into dedicated service"
+        )
+    else:  # conventional
+        if emoji:
+            return (
+                "1. ✨ feat(auth): add refresh token rotation on expiry\n"
+                "2. 🐛 fix(auth): prevent reuse of invalidated tokens\n"
+                "3. ♻️ refactor(auth): extract token validation into dedicated service"
+            )
+        return (
+            "1. feat(auth): add refresh token rotation on expiry\n"
+            "2. fix(auth): prevent reuse of invalidated tokens\n"
+            "3. refactor(auth): extract token validation into dedicated service"
+        )

gitai_cli-0.1.0.dist-info/METADATA

@@ -0,0 +1,139 @@
+Metadata-Version: 2.4
+Name: gitai-cli
+Version: 0.1.0
+Summary: AI-powered git commit message generator
+Project-URL: Homepage, https://github.com/Jeranguz/gitai
+Project-URL: Repository, https://github.com/Jeranguz/gitai
+Project-URL: Issues, https://github.com/Jeranguz/gitai/issues
+Author: Jeranguz
+License: MIT
+License-File: LICENSE
+Keywords: ai,cli,commit,conventional-commits,git,llm
+Classifier: Development Status :: 3 - Alpha
+Classifier: Environment :: Console
+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.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Topic :: Software Development :: Version Control :: Git
+Classifier: Topic :: Utilities
+Requires-Python: >=3.11
+Requires-Dist: gitpython
+Requires-Dist: litellm
+Requires-Dist: questionary
+Requires-Dist: rich
+Requires-Dist: tomli-w
+Requires-Dist: typer
+Provides-Extra: dev
+Requires-Dist: pytest; extra == 'dev'
+Description-Content-Type: text/markdown
+
+# gitai
+
+<p align="center">
+  <img src="assets/commit-genie.png" alt="The Commit Genie" width="200"/>
+</p>
+
+<p align="center">
+  <a href="https://pypi.org/project/gitai"><img src="https://img.shields.io/pypi/v/gitai" alt="PyPI version"/></a>
+  <a href="https://pypi.org/project/gitai"><img src="https://img.shields.io/pypi/pyversions/gitai" alt="Python versions"/></a>
+  <a href="LICENSE"><img src="https://img.shields.io/badge/license-MIT-blue" alt="MIT license"/></a>
+</p>
+
+AI-powered git commit message generator. Analyzes your staged changes and suggests meaningful commit messages — using any LLM you already have access to.
+
+## Features
+
+- Reads your staged `git diff` and generates 3 commit message suggestions
+- Interactive selection: pick a suggestion or write your own
+- Supports multiple providers: Ollama (local), OpenAI, Anthropic, Gemini, and [more](https://docs.litellm.ai/docs/providers)
+- Two commit styles: [Conventional Commits](https://www.conventionalcommits.org/) or free-form
+- Optional emoji (gitmoji) support
+
+## Installation
+
+```bash
+pip install gitai
+```
+
+Requires Python 3.11+.
+
+## Quick start
+
+```bash
+# 1. Stage your changes
+git add .
+
+# 2. Run gitai
+gitai commit
+```
+
+gitai reads the diff, calls your configured LLM, and presents 3 suggestions to choose from.
+
+## Usage
+
+```
+gitai commit      Generate commit message suggestions for staged changes
+gitai config      View and update settings
+gitai --version   Show version
+gitai --help      Show help
+```
+
+## Configuration
+
+Run `gitai config` to update settings interactively. Settings are stored in `~/.gitai.toml`.
+
+| Key | Default | Description |
+|---|---|---|
+| `provider` | `ollama` | LLM provider |
+| `model` | `llama3.2` | Model name |
+| `ollama_url` | `http://localhost:11434` | Ollama API base URL (Ollama only) |
+| `commit_style` | `conventional` | `conventional` or `free-form` |
+| `emoji` | `false` | Prefix suggestions with gitmoji |
+
+### Supported providers
+
+| Provider | `provider` value | Example `model` value | API key env var |
+|---|---|---|---|
+| Ollama (local) | `ollama` | `llama3.2`, `mistral` | — |
+| Anthropic | `anthropic` | `claude-sonnet-4-6`, `claude-haiku-4-5-20251001` | `ANTHROPIC_API_KEY` |
+| OpenAI | `openai` | `gpt-4o`, `gpt-4o-mini` | `OPENAI_API_KEY` |
+| Gemini | `gemini` | `gemini-2.0-flash` | `GEMINI_API_KEY` |
+
+For cloud providers, set the API key in your shell profile:
+
+**bash/zsh** (`~/.bashrc` or `~/.zshrc`):
+```bash
+export ANTHROPIC_API_KEY=sk-ant-...
+```
+
+**PowerShell** (`$PROFILE`):
+```powershell
+$env:ANTHROPIC_API_KEY="sk-ant-..."
+```
+
+### Example `~/.gitai.toml`
+
+```toml
+provider = "anthropic"
+model = "claude-haiku-4-5-20251001"
+commit_style = "conventional"
+emoji = false
+ollama_url = "http://localhost:11434"
+```
+
+## Local setup (Ollama)
+
+If you want to run fully offline with Ollama:
+
+1. Install [Ollama](https://ollama.com/)
+2. Pull a model: `ollama pull llama3.2`
+3. Run `gitai commit` — no API key needed
+
+## TODO
+
+- [ ] Allow configuring the number of suggestions generated
+- [ ] Add `--push` flag to commit and push in one step
+- [ ] Support unstaged changes with an optional `--all` flag

gitai_cli-0.1.0.dist-info/RECORD

@@ -0,0 +1,11 @@
+gitai/__init__.py,sha256=kUR5RAFc7HCeiqdlX36dZOHkUI5wI6V_43RpEcD8b-0,22
+gitai/ai.py,sha256=wx_yYA865VsQB8SsrzkEKd1hLF7SmdsWf1j-cbWeemU,1656
+gitai/cli.py,sha256=pyiJS7jm9IRgeYdsJX4-v77zramPo_BXF9hghchmLi4,3949
+gitai/config.py,sha256=LqmaFN9JBpHL0MGbn476sHBGUgfhfmEeQYD-7J60yxE,669
+gitai/git.py,sha256=iDv09764q21uRBVJk784-qo9Gl3Nrpk0rpW31ha9_p0,628
+gitai/prompt.py,sha256=Hj61wFDOrDztmnQxLNWagjCJ6OXfqWreS93z0KfTE_o,3057
+gitai_cli-0.1.0.dist-info/METADATA,sha256=HS5B9TjSQ8OTbx_Yfbn6BXM16ew3H7KQPadtmtJB-Go,4257
+gitai_cli-0.1.0.dist-info/WHEEL,sha256=QccIxa26bgl1E6uMy58deGWi-0aeIkkangHcxk2kWfw,87
+gitai_cli-0.1.0.dist-info/entry_points.txt,sha256=Cv684a3aG7n1URVEr6skYx27B37CJoev2fZx2irdUiw,40
+gitai_cli-0.1.0.dist-info/licenses/LICENSE,sha256=RjOVI-Vu4fAe5qNurmnVBMVROuK_xPePzpCSEEEtKq4,1065
+gitai_cli-0.1.0.dist-info/RECORD,,

gitai_cli-0.1.0.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.29.0
+Root-Is-Purelib: true
+Tag: py3-none-any

gitai_cli-0.1.0.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+gitai = gitai.cli:app

gitai_cli-0.1.0.dist-info/licenses/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2024 Jeranguz
+
+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.