logging

Author: MikeBirdTech

.gitignore

@@ -163,3 +163,7 @@ cython_debug/
 
 .idea/
 .vscode/
+
+
+# Testing files
+gitdiff.txt

CODE_OF_CONDUCT.md

@@ -46,7 +46,7 @@ This blueprint guides you to easily generate AI-powered git commit messages base
    ollama pull llama3.1
    ```
 
-4. Make changes to your git repo and run:
+4. After adding your changes to your git repo, run:
 
    ```bash
    lcg

README.md

@@ -1,10 +1,12 @@
 """AI Service module for interacting with different LLM providers."""
 
-import os
-from typing import Dict, Optional
+from typing import Optional
 
 import requests
 
+JAN_BASE_URL = "http://localhost:1337/v1/chat/completions"
+OLLAMA_BASE_URL = "http://localhost:11434/api/generate"
+
 
 class AIService:
     """Service for interacting with different AI providers."""
@@ -21,8 +23,8 @@ def __init__(self, service_type: str, model: Optional[str] = None):
 
         # Set up base URLs for services
         self.base_urls = {
-            "ollama": "http://localhost:11434/api/generate",
-            "jan": "http://localhost:1337/v1/chat/completions",
+            "ollama": OLLAMA_BASE_URL,
+            "jan": JAN_BASE_URL,
         }
 
         if service_type not in self.base_urls:

src/blueprint/ai_service.py

@@ -12,11 +12,14 @@
     create_commit,
 )
 
+DEFAULT_OLLAMA_MODEL = "llama3.1"
+DEFAULT_JAN_MODEL = "llama3.1-8b-instruct"
+
 
 def main():
     """Main entry point for the CLI application."""
-    OLLAMA_MODEL = os.getenv("OLLAMA_MODEL", "llama3.1")
-    JAN_MODEL = os.getenv("JAN_MODEL", "llama 3.1")
+    OLLAMA_MODEL = os.getenv("OLLAMA_MODEL", DEFAULT_OLLAMA_MODEL)
+    JAN_MODEL = os.getenv("JAN_MODEL", DEFAULT_JAN_MODEL)
 
     parser = argparse.ArgumentParser(
         description="Generate git commit messages using LLMs."
@@ -67,7 +70,7 @@ def main():
 
     # Show analytics if requested
     if args.analytics:
-        print(f"\nAnalytics:")
+        print("\nAnalytics:")
         print(
             f"Time taken to generate commit messages: {end_time - start_time:.2f} seconds"
         )
@@ -101,7 +104,7 @@ def main():
             end_time = time.time()
 
             if args.analytics:
-                print(f"\nRegeneration Analytics:")
+                print("\nRegeneration Analytics:")
                 print(
                     f"Time taken to regenerate commit messages: {end_time - start_time:.2f} seconds"
                 )

src/blueprint/cli.py

@@ -1,9 +1,7 @@
 """Git commit message generator using AI models."""
 
-import os
 import subprocess
 import sys
-import time
 from typing import List, Optional
 
 from blueprint.ai_service import AIService
@@ -73,7 +71,11 @@ def parse_commit_messages(response: str) -> List[str]:
     messages = []
     for line in response.split("\n"):
         if line.strip().startswith(("1.", "2.", "3.")):
-            messages.append(line.split(".", 1)[1].strip())
+            message = line.split(".", 1)[1].strip()
+            # Strip surrounding single quotes if present
+            if message.startswith("'") and message.endswith("'"):
+                message = message[1:-1]
+            messages.append(message)
     return messages