git-commit-msg-ai 1.1.0__py3-none-any.whl → 1.3.0__py3-none-any.whl

git_commit_msg_ai/ai_client.py

@@ -1,28 +1,42 @@
-from typing import cast
+import textwrap
+from typing import Final, cast
+
 import anthropic
 
-SYSTEM_PROMPT = (
-    'You are a Git commit message generator. Output only the commit message, nothing else. '
-    'Follow the Conventional Commits specification:\n'
-    '- First line: <type>(<optional scope>)<!>: <short subject> (50 chars max); '
-    'append ! before the colon if the commit introduces a breaking change\n'
-    '- Blank line\n'
-    '- Bullet-point body explaining WHY the changes were made, not what\n'
-    '- If there is a breaking change, add a blank line after the body followed by '
-    '"BREAKING CHANGE: <description of what breaks and why>\n'
-    'Types: feat, fix, docs, style, refactor, test, chore'
-)
+from git_commit_msg_ai.exceptions import AIError
+
+SYSTEM_PROMPT: Final[str] = textwrap.dedent("""\
+    You are a Git commit message generator. Output only the commit message, nothing else. Follow the Conventional Commits specification:
+    - First line: <type>(<optional scope>)<!>: <short subject> (50 chars max); append ! before the colon if the commit introduces a breaking change
+    - Blank line
+    - Bullet-point body explaining WHY the changes were made, not what
+    - If there is a breaking change, add a blank line after the body followed by "BREAKING CHANGE: <description of what breaks and why>
+    Types: feat, fix, docs, style, refactor, test, chore""")
 
-MODEL = 'claude-haiku-4-5-20251001'
-MAX_TOKENS = 1024
+MODEL: Final[str] = 'claude-haiku-4-5-20251001'
+MAX_TOKENS: Final[int] = 1024
 
 
 def generate_commit_message(diff: str) -> str:
-    client = anthropic.Anthropic()
-    response = client.messages.create(
-        model=MODEL,
-        max_tokens=MAX_TOKENS,
-        system=SYSTEM_PROMPT,
-        messages=[{'role': 'user', 'content': diff}],
-    )
-    return cast(anthropic.types.TextBlock, response.content[0]).text.strip()
+    try:
+        anthropic_client = anthropic.Anthropic()
+
+        anthropic_api_response = anthropic_client.messages.create(
+            model=MODEL,
+            max_tokens=MAX_TOKENS,
+            system=SYSTEM_PROMPT,
+            messages=[{'role': 'user', 'content': diff}],
+        )
+    except anthropic.AuthenticationError:
+        raise AIError('Anthropic API key is missing or invalid. Set the ANTHROPIC_API_KEY environment variable.')
+    except anthropic.RateLimitError:
+        raise AIError('Anthropic API rate limit reached. Wait a moment and try again.')
+    except anthropic.APIConnectionError:
+        raise AIError('Could not reach the Anthropic API. Check your network connection.')
+    except anthropic.APIStatusError as error:
+        raise AIError(f'Anthropic API returned an error: {error.status_code}.')
+
+    anthropic_api_response_message = anthropic_api_response.content[0]
+    commit_message = cast(anthropic.types.TextBlock, anthropic_api_response_message).text.strip()
+
+    return commit_message

git_commit_msg_ai/cli.py

@@ -1,21 +1,35 @@
-from git_commit_msg_ai import ai_client, editor, git_ops
-
-
-def main() -> None:
-    diff = git_ops.get_staged_diff()
-    msg = ai_client.generate_commit_message(diff)
-    print(msg)
-
-    choice = input('\n[a]ccept / [e]dit / [r]eject: ').strip().lower()
-
-    if choice == 'a':
-        git_ops.commit(msg)
-    elif choice == 'e':
-        msg = editor.open_in_editor(msg)
-        git_ops.commit(msg)
-    elif choice == 'r':
-        print('User rejected the generated commit message. No commit made.')
-
-
-if __name__ == '__main__':
-    main()
+import sys
+
+from git_commit_msg_ai import ai_client, editor, git_ops
+from git_commit_msg_ai.exceptions import GitCommitAIError
+
+
+def main() -> None:
+    try:
+        diff = git_ops.get_staged_diff()
+        commit_message = ai_client.generate_commit_message(diff)
+        print(commit_message)
+
+        print()
+        user_selection = input('[a]ccept / [e]dit / [r]eject: ').strip().lower()
+
+        if user_selection == 'a':
+            git_ops.commit(commit_message)
+        elif user_selection == 'e':
+            updated_commit_message = editor.open_in_editor(commit_message)
+            git_ops.commit(updated_commit_message)
+        elif user_selection == 'r':
+            print('User rejected the generated commit message. No commit made.')
+        else:
+            print('Invalid selection.')
+            sys.exit(1)
+    except GitCommitAIError as error:
+        print(str(error))
+        sys.exit(1)
+    except (KeyboardInterrupt, EOFError):
+        print('Aborted.')
+        sys.exit(1)
+
+
+if __name__ == '__main__':  # pragma: no cover
+    main()

git_commit_msg_ai/editor.py

@@ -2,13 +2,33 @@ import os
 import subprocess
 import tempfile
 
+from git_commit_msg_ai.exceptions import EditorError
+
 
 def open_in_editor(initial_text: str) -> str:
-    with tempfile.NamedTemporaryFile(suffix='.txt', mode='w', delete=False) as f:
-        f.write(initial_text)
-        tmp = f.name
-    subprocess.run([os.environ.get('EDITOR', 'notepad'), tmp])
-    with open(tmp) as f:
-        result = f.read().strip()
-    os.unlink(tmp)
-    return result
+    try:
+        with tempfile.NamedTemporaryFile(suffix='.txt', mode='w', delete=False) as temp_file:
+            temp_file.write(initial_text)
+            temp_file_path = temp_file.name
+    except OSError:
+        raise EditorError('Could not create a temporary file.')
+
+    editor_command = os.environ.get('EDITOR', 'notepad')
+
+    try:
+        try:
+            subprocess.run([editor_command, temp_file_path], check=True)
+        except FileNotFoundError:
+            raise EditorError(f'Editor "{editor_command}" was not found. Set the EDITOR environment variable to a valid editor.')
+        except subprocess.CalledProcessError:
+            raise EditorError(f'Editor "{editor_command}" exited with an error.')
+
+        try:
+            with open(temp_file_path) as temp_file:
+                edited_text = temp_file.read().strip()
+        except OSError:
+            raise EditorError('Could not read the edited commit message from the temporary file.')
+    finally:
+        os.unlink(temp_file_path)
+
+    return edited_text

git_commit_msg_ai/exceptions.py

@@ -0,0 +1,14 @@
+class GitCommitAIError(Exception):
+    """Base exception for the application."""
+
+
+class GitError(GitCommitAIError):
+    """Raised when a git operation fails."""
+
+
+class AIError(GitCommitAIError):
+    """Raised when the Anthropic API call fails."""
+
+
+class EditorError(GitCommitAIError):
+    """Raised when opening or reading from the editor fails."""

git_commit_msg_ai/git_ops.py

@@ -1,9 +1,28 @@
 import subprocess
 
+from git_commit_msg_ai.exceptions import GitError
+
 
 def get_staged_diff() -> str:
-    return subprocess.check_output(['git', 'diff', '--cached']).decode()
+    try:
+        staged_diff = subprocess.check_output(['git', 'diff', '--cached']).decode()
+    except FileNotFoundError:
+        raise GitError('git is not installed or not on PATH.')
+    except subprocess.CalledProcessError:
+        raise GitError('Failed to get staged diff. Are you inside a git repository?')
+    except UnicodeDecodeError:
+        raise GitError('Staged diff contains bytes that could not be decoded as UTF-8.')
+
+    if not staged_diff:
+        raise GitError('No staged changes found. Stage files with git add before running.')
+
+    return staged_diff
 
 
 def commit(message: str) -> None:
-    subprocess.run(['git', 'commit', '-m', message])
+    try:
+        subprocess.run(['git', 'commit', '-m', message], check=True)
+    except FileNotFoundError:
+        raise GitError('git is not installed or not on PATH.')
+    except subprocess.CalledProcessError:
+        raise GitError('git commit failed.')

{git_commit_msg_ai-1.1.0.dist-info → git_commit_msg_ai-1.3.0.dist-info}/METADATA

@@ -1,11 +1,16 @@
 Metadata-Version: 2.4
 Name: git-commit-msg-ai
-Version: 1.1.0
+Version: 1.3.0
 Summary: AI-powered git commit message generator following Conventional Commits
 License-Expression: MIT
 Requires-Python: >=3.9
 Description-Content-Type: text/markdown
 Requires-Dist: anthropic
+Provides-Extra: dev
+Requires-Dist: mypy; extra == "dev"
+Requires-Dist: ruff; extra == "dev"
+Requires-Dist: pytest; extra == "dev"
+Requires-Dist: pytest-cov; extra == "dev"
 
 # git-commit-msg-ai
 

git_commit_msg_ai-1.3.0.dist-info/RECORD

@@ -0,0 +1,11 @@
+git_commit_msg_ai/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+git_commit_msg_ai/ai_client.py,sha256=RnzgIx1ieuiiH6w2VijA6_F-cs6tIQR57W1x3avg8lE,1887
+git_commit_msg_ai/cli.py,sha256=s94orvxin-38vqUYWAO-j5_JZTzxBXbZSVOHYZ5KO_k,1059
+git_commit_msg_ai/editor.py,sha256=5X4cKFz2ry70BTtWQopSuDpGJGP3JwMcpa7vJpPbjyA,1180
+git_commit_msg_ai/exceptions.py,sha256=7Hwluf3zHMjs4lpGktWS-Lwgo8y_4Xbb1WqzPQHkkUA,352
+git_commit_msg_ai/git_ops.py,sha256=UJr3FUHPg8OTATy5Z5kbnXpy4Kh7fuORKR5HZV774NQ,970
+git_commit_msg_ai-1.3.0.dist-info/METADATA,sha256=cIlbCpC92sAcvU1LLA4GsqgqbKLylX1l-2Pgl8BMBf8,2008
+git_commit_msg_ai-1.3.0.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
+git_commit_msg_ai-1.3.0.dist-info/entry_points.txt,sha256=KTu6wUhl0p3nf27k8L4vpSH_hpeRQpwzMPSmKv7K5Cs,65
+git_commit_msg_ai-1.3.0.dist-info/top_level.txt,sha256=XYQC2BXvrcREGKEG7sm9nbwO7ifqcUSVgU7SW8BTURs,18
+git_commit_msg_ai-1.3.0.dist-info/RECORD,,

git_commit_msg_ai-1.1.0.dist-info/RECORD

@@ -1,10 +0,0 @@
-git_commit_msg_ai/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-git_commit_msg_ai/ai_client.py,sha256=A6EukpLUldU5IZRCajIQwex06vpZSdioJTiR-1HqfU0,1067
-git_commit_msg_ai/cli.py,sha256=t0MMBmGmclGOKVQZi8IqexZfinrHKpWBmqwli0mwDKo,559
-git_commit_msg_ai/editor.py,sha256=P5I2HQIdsoZCc5PuUdE6CvpepenWYJTieVkONzPgLCs,384
-git_commit_msg_ai/git_ops.py,sha256=s6i0JPjQAXfBkX41KPk4oXbRhLyK-HtIFzMF3dQouv0,212
-git_commit_msg_ai-1.1.0.dist-info/METADATA,sha256=BenF9whnrOE4pLFVk33c71BD_C6S2qFjW0QhY5BoETw,1831
-git_commit_msg_ai-1.1.0.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
-git_commit_msg_ai-1.1.0.dist-info/entry_points.txt,sha256=KTu6wUhl0p3nf27k8L4vpSH_hpeRQpwzMPSmKv7K5Cs,65
-git_commit_msg_ai-1.1.0.dist-info/top_level.txt,sha256=XYQC2BXvrcREGKEG7sm9nbwO7ifqcUSVgU7SW8BTURs,18
-git_commit_msg_ai-1.1.0.dist-info/RECORD,,