commitmessagegenerator 1.4.1__tar.gz → 1.6.0__tar.gz

{commitmessagegenerator-1.4.1 → commitmessagegenerator-1.6.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: commitmessagegenerator
-Version: 1.4.1
+Version: 1.6.0
 Summary: Generate commit messages with AI (Google Gemini) automatically using `git diff`.
 Author-email: Gabriel Terceiro <gcarolinoterceiro@gmail.com>
 License: MIT
@@ -35,6 +35,11 @@ pip install commitmessagegenerator
 commitgen -cf
 ```
 
+This will prompt you for:
+
+1. Your Gemini API key
+2. Your preferred AI model (with options to choose from)
+
 ## Run this and type you API key to the terminal so the package creates the .env file and automatically adds it to the .gitignore
 
 Or do it manually:
@@ -45,6 +50,7 @@ Create a `.env` file in the directory where you will run commitgen (usually the
 
 ```
 GEMINI_API_KEY=your-gemini-api-key
+AI_MODEL=gemini-2.0-flash
 ```
 
 ## 🚀 Usage
@@ -58,9 +64,28 @@ commitgen (-c/-cp)
 The command will:
 
 - Read the git diff;
-- Send it to the Google Gemini API;
+- Send it to the Google Gemini API using your configured model;
 - Return a commit message suggestion directly in your terminal.
 
+### Available Commands
+
+- `commitgen` - Generate commit message only
+- `commitgen -c` - Generate and commit with the message
+- `commitgen -cp` - Generate, commit, and push
+- `commitgen -cf` - Configure API key and model
+- `commitgen -s` - Show current configuration status
+
+### Available Models
+
+When configuring with `-cf`, you can choose from:
+
+1. **gemini-2.0-flash** (default) - Fast and efficient
+2. **gemini-1.5-flash** - Good balance of speed and quality
+3. **gemini-1.5-pro** - Highest quality, slower
+4. **gemini-2.0-flash-exp** - Experimental version
+5. **gemini-2.5-flash** - Latest version, fast and efficient
+6. **gemini-2.5-pro** - Latest version, highest quality
+
 ## 🧩 Requisites
 
 - Python 3.8 or higher

{commitmessagegenerator-1.4.1 → commitmessagegenerator-1.6.0}/README.md

@@ -22,6 +22,11 @@ pip install commitmessagegenerator
 commitgen -cf
 ```
 
+This will prompt you for:
+
+1. Your Gemini API key
+2. Your preferred AI model (with options to choose from)
+
 ## Run this and type you API key to the terminal so the package creates the .env file and automatically adds it to the .gitignore
 
 Or do it manually:
@@ -32,6 +37,7 @@ Create a `.env` file in the directory where you will run commitgen (usually the
 
 ```
 GEMINI_API_KEY=your-gemini-api-key
+AI_MODEL=gemini-2.0-flash
 ```
 
 ## 🚀 Usage
@@ -45,9 +51,28 @@ commitgen (-c/-cp)
 The command will:
 
 - Read the git diff;
-- Send it to the Google Gemini API;
+- Send it to the Google Gemini API using your configured model;
 - Return a commit message suggestion directly in your terminal.
 
+### Available Commands
+
+- `commitgen` - Generate commit message only
+- `commitgen -c` - Generate and commit with the message
+- `commitgen -cp` - Generate, commit, and push
+- `commitgen -cf` - Configure API key and model
+- `commitgen -s` - Show current configuration status
+
+### Available Models
+
+When configuring with `-cf`, you can choose from:
+
+1. **gemini-2.0-flash** (default) - Fast and efficient
+2. **gemini-1.5-flash** - Good balance of speed and quality
+3. **gemini-1.5-pro** - Highest quality, slower
+4. **gemini-2.0-flash-exp** - Experimental version
+5. **gemini-2.5-flash** - Latest version, fast and efficient
+6. **gemini-2.5-pro** - Latest version, highest quality
+
 ## 🧩 Requisites
 
 - Python 3.8 or higher

commitmessagegenerator-1.6.0/commitmessagegenerator/cli.py

@@ -0,0 +1,78 @@
+import argparse
+import subprocess
+from .generator import gerar_mensagem_commit
+from .configure import api_key, get_configured_model
+import sys
+import getpass
+
+def main():
+    parser = argparse.ArgumentParser(description="Gerador de mensagens de commit com IA")
+    parser.add_argument("-c", "--commit", action="store_true", help="Commits with the generated message")
+    parser.add_argument("-cp", "--commitpush",  action="store_true", help="Commits and pushes with the generated message")
+    parser.add_argument("-cf", "--configure", action="store_true", help="Configures the GEMINI_API_KEY environment variable")
+    parser.add_argument("-s", "--status", action="store_true", help="Shows current configuration status")
+    args = parser.parse_args()
+
+    if args.status:
+        from .configure import get_configured_model
+        import os
+        from dotenv import load_dotenv
+        
+        load_dotenv()
+        key = os.getenv("GEMINI_API_KEY")
+        model = get_configured_model()
+        
+        print("\nCurrent Configuration:")
+        print(f"API Key: {'✓ Set' if key else '✗ Not set'}")
+        print(f"Model: {model}")
+        return
+
+    if not args.configure:
+        mensagem = gerar_mensagem_commit()
+
+        if "No changes detected" in mensagem:
+            print(mensagem)
+            return
+
+        print("\nGenerated commit message:\n" + mensagem)
+
+    if args.commit or args.commitpush:
+        print("\nCommitting changes...")
+        subprocess.run(["git", "commit", "-m", mensagem])
+
+    if args.commitpush:
+        print("\nPushing changes...")
+        subprocess.run(["git", "push"])
+    
+    if args.configure:
+        print("\nPlease input your API KEY\nThis is directly set in the .env file")
+        key = getpass.getpass()
+        
+        # Model selection
+        print("\nAvailable models:")
+        print("1. gemini-2.0-flash (default, fast and efficient)")
+        print("2. gemini-1.5-flash (good balance of speed and quality)")
+        print("3. gemini-1.5-pro (highest quality, slower)")
+        print("4. gemini-2.0-flash-exp (experimental)")
+        print("5. gemini-2.5-flash (latest, fast and efficient)")
+        print("6. gemini-2.5-pro (latest, highest quality)")
+        
+        model_choice = input("\nSelect model (1-6) or press Enter for default (1): ").strip()
+        
+        model_map = {
+            "1": "gemini-2.0-flash",
+            "2": "gemini-1.5-flash", 
+            "3": "gemini-1.5-pro",
+            "4": "gemini-2.0-flash-exp",
+            "5": "gemini-2.5-flash",
+            "6": "gemini-2.5-pro"
+        }
+        
+        selected_model = model_map.get(model_choice, "gemini-2.0-flash")
+        
+        api_key(key, selected_model)
+        print(f"\nAPI KEY and model ({selected_model}) saved in .env file\n")
+    
+    if len(sys.argv) == 1:
+        print("\nRemoving staged changes (git reset)...")
+        subprocess.run(["git", "reset"])

commitmessagegenerator-1.6.0/commitmessagegenerator/configure.py

@@ -0,0 +1,47 @@
+import os
+
+def api_key(key, model="gemini-2.0-flash"):
+    if os.path.exists(".env"):
+        with open(".env", "r+") as outfile:
+            lines = outfile.readlines()
+            gemini_api_key = next((line for line in lines if line.startswith("GEMINI_API_KEY=")), None)
+            model_line = next((line for line in lines if line.startswith("AI_MODEL=")), None)
+            
+            outfile.seek(0)
+            outfile.truncate()
+            
+            # Update or add GEMINI_API_KEY
+            if gemini_api_key:
+                outfile.writelines([line if not line.startswith("GEMINI_API_KEY=") else f"GEMINI_API_KEY={key}\n" for line in lines])
+            else:
+                outfile.writelines(lines)
+                outfile.write(f"GEMINI_API_KEY={key}\n")
+            
+            # Update or add AI_MODEL
+            if model_line:
+                outfile.seek(0)
+                outfile.truncate()
+                outfile.writelines([line if not line.startswith("AI_MODEL=") else f"AI_MODEL={model}\n" for line in outfile.readlines()])
+            else:
+                outfile.write(f"AI_MODEL={model}\n")
+    else:
+        with open(".env", "w") as outfile:
+            outfile.write(f"GEMINI_API_KEY={key}\n")
+            outfile.write(f"AI_MODEL={model}\n")
+
+    if os.path.exists(".gitignore"):
+        with open(".gitignore", "r+") as outfile:
+            lines = outfile.readlines()
+            env_in_gitignore = next((line for line in lines if line.strip() == ".env"), None)
+            if not env_in_gitignore:
+                outfile.write("\n.env")
+
+def get_configured_model():
+    """Get the currently configured AI model from .env file"""
+    if os.path.exists(".env"):
+        with open(".env", "r") as outfile:
+            lines = outfile.readlines()
+            model_line = next((line for line in lines if line.startswith("AI_MODEL=")), None)
+            if model_line:
+                return model_line.split("=", 1)[1].strip()
+    return "gemini-2.0-flash"  # Default fallback

{commitmessagegenerator-1.4.1 → commitmessagegenerator-1.6.0}/commitmessagegenerator/generator.py

@@ -2,6 +2,7 @@ import os
 from dotenv import load_dotenv
 from google import genai
 from git import Repo
+from .configure import get_configured_model
 
 def gerar_mensagem_commit():
     load_dotenv()
@@ -9,8 +10,8 @@ def gerar_mensagem_commit():
     if not key:
         raise RuntimeError("The GEMINI_API_KEY environment variable is not set.")
 
-    #genai.configure(api_key=key)
-    #model = genai.GenerativeModel("gemini-2.0-flash")
+    # Get the configured model
+    model = get_configured_model()
 
     client = genai.Client(api_key=key)
     
@@ -38,9 +39,8 @@ def gerar_mensagem_commit():
         '''
     )
 
-    response = client.generate_content(
-        model="gemini-2.0-flash",
-        contents=[{"role": "user", "parts": [prompt]}]
+    response = client.models.generate_content(
+        model=model,
+        contents=prompt
     )
-    return response.text.strip()
-    
+    return response.text.strip() if response.text else "Failed to generate commit message"

{commitmessagegenerator-1.4.1 → commitmessagegenerator-1.6.0}/commitmessagegenerator.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: commitmessagegenerator
-Version: 1.4.1
+Version: 1.6.0
 Summary: Generate commit messages with AI (Google Gemini) automatically using `git diff`.
 Author-email: Gabriel Terceiro <gcarolinoterceiro@gmail.com>
 License: MIT
@@ -35,6 +35,11 @@ pip install commitmessagegenerator
 commitgen -cf
 ```
 
+This will prompt you for:
+
+1. Your Gemini API key
+2. Your preferred AI model (with options to choose from)
+
 ## Run this and type you API key to the terminal so the package creates the .env file and automatically adds it to the .gitignore
 
 Or do it manually:
@@ -45,6 +50,7 @@ Create a `.env` file in the directory where you will run commitgen (usually the
 
 ```
 GEMINI_API_KEY=your-gemini-api-key
+AI_MODEL=gemini-2.0-flash
 ```
 
 ## 🚀 Usage
@@ -58,9 +64,28 @@ commitgen (-c/-cp)
 The command will:
 
 - Read the git diff;
-- Send it to the Google Gemini API;
+- Send it to the Google Gemini API using your configured model;
 - Return a commit message suggestion directly in your terminal.
 
+### Available Commands
+
+- `commitgen` - Generate commit message only
+- `commitgen -c` - Generate and commit with the message
+- `commitgen -cp` - Generate, commit, and push
+- `commitgen -cf` - Configure API key and model
+- `commitgen -s` - Show current configuration status
+
+### Available Models
+
+When configuring with `-cf`, you can choose from:
+
+1. **gemini-2.0-flash** (default) - Fast and efficient
+2. **gemini-1.5-flash** - Good balance of speed and quality
+3. **gemini-1.5-pro** - Highest quality, slower
+4. **gemini-2.0-flash-exp** - Experimental version
+5. **gemini-2.5-flash** - Latest version, fast and efficient
+6. **gemini-2.5-pro** - Latest version, highest quality
+
 ## 🧩 Requisites
 
 - Python 3.8 or higher

{commitmessagegenerator-1.4.1 → commitmessagegenerator-1.6.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "commitmessagegenerator"
-version = "1.4.1"
+version = "1.6.0"
 description = "Generate commit messages with AI (Google Gemini) automatically using `git diff`."
 readme = "README.md"
 license = {text = "MIT"}

commitmessagegenerator-1.4.1/commitmessagegenerator/cli.py

@@ -1,40 +0,0 @@
-import argparse
-import subprocess
-from .generator import gerar_mensagem_commit
-from .configure import api_key
-import sys
-import getpass
-
-def main():
-    parser = argparse.ArgumentParser(description="Gerador de mensagens de commit com IA")
-    parser.add_argument("-c", "--commit", action="store_true", help="Commits with the generated message")
-    parser.add_argument("-cp", "--commitpush",  action="store_true", help="Commits and pushes with the generated message")
-    parser.add_argument("-cf", "--configure", action="store_true", help="Configures the GEMINI_API_KEY environment variable")
-    args = parser.parse_args()
-
-    if not args.configure:
-        mensagem = gerar_mensagem_commit()
-
-        if "No changes detected" in mensagem:
-            print(mensagem)
-            return
-
-        print("\nGenerated commit message:\n" + mensagem)
-
-    if args.commit or args.commitpush:
-        print("\nCommitting changes...")
-        subprocess.run(["git", "commit", "-m", mensagem])
-
-    if args.commitpush:
-        print("\nPushing changes...")
-        subprocess.run(["git", "push"])
-    
-    if args.configure:
-        print("\nPlease input your API KEY\nThis is directly set in the .env file")
-        key = getpass.getpass()
-        api_key(key)
-        print("\nAPI KEY saved in .env file\n")
-    
-    if len(sys.argv) == 1:
-        print("\nRemoving staged changes (git reset)...")
-        subprocess.run(["git", "reset"])

commitmessagegenerator-1.4.1/commitmessagegenerator/configure.py

@@ -1,22 +0,0 @@
-import os
-def api_key(key):
-    if os.path.exists(".env"):
-        with open(".env", "r+") as outfile:
-            lines = outfile.readlines()
-            gemini_api_key = next((line for line in lines if line.startswith("GEMINI_API_KEY=")), None)
-            if gemini_api_key:
-                outfile.seek(0)
-                outfile.truncate()
-                outfile.writelines([line if not line.startswith("GEMINI_API_KEY=") else f"GEMINI_API_KEY={key}\n" for line in lines])
-            else:
-                outfile.write(f"GEMINI_API_KEY={key}\n")
-    else:
-        with open(".env", "w") as outfile:
-            outfile.write(f"GEMINI_API_KEY={key}\n")
-
-    if os.path.exists(".gitignore"):
-        with open(".gitignore", "r+") as outfile:
-            lines = outfile.readlines()
-            env_in_gitignore = next((line for line in lines if line.strip() == ".env"), None)
-            if not env_in_gitignore:
-                outfile.write("\n.env")