Merge pull request #31 from aneeshsunganahalli/fix-inverted-else-clause

Fixed Inverted else clause

Author: DFanso

src/claude/claude.go

@@ -60,7 +60,6 @@ func GenerateCommitMessage(config *types.Config, changes string, apiKey string)
 	req.Header.Set("x-api-key", apiKey)
 	req.Header.Set("anthropic-version", "2023-06-01")
 
-	client := &http.Client{}
 	client := &http.Client{}
 	resp, err := client.Do(req)
 	if err != nil {

src/main.go

@@ -6,6 +6,7 @@ import (
 
 	"github.com/atotto/clipboard"
 	"github.com/dfanso/commit-msg/src/chatgpt"
+	"github.com/dfanso/commit-msg/src/claude"
 	"github.com/dfanso/commit-msg/src/gemini"
 	"github.com/dfanso/commit-msg/src/grok"
 	"github.com/dfanso/commit-msg/src/internal/display"
@@ -17,30 +18,34 @@ import (
 
 // main is the entry point of the commit message generator
 func main() {
-	// Get API key from environment variables
-	var apiKey string
-	if os.Getenv("COMMIT_LLM") == "google" {
-		apiKey = os.Getenv("GOOGLE_API_KEY")
-		if apiKey == "" {
-			log.Fatalf("GOOGLE_API_KEY is not set")
-		}
-	} else if os.Getenv("COMMIT_LLM") == "grok" {
-		apiKey = os.Getenv("GROK_API_KEY")
-		if apiKey == "" {
-			log.Fatalf("GROK_API_KEY is not set")
-		}
-	} else if os.Getenv("COMMIT_LLM") == "chatgpt" {
-		apiKey = os.Getenv("OPENAI_API_KEY")
-		if apiKey == "" {
-			log.Fatalf("OPENAI_API_KEY is not set")
-		}
-	} else if os.Getenv("COMMIT_LLM") == "claude" {
-		apiKey = os.Getenv("CLAUDE_API_KEY")
-		if apiKey == "" {
-			log.Fatalf("CLAUDE_API_KEY is not set")
-		} else {
-			log.Fatalf("Invalid COMMIT_LLM value: %s", os.Getenv("COMMIT_LLM"))
-		}
+    // Validate COMMIT_LLM and required API keys
+    commitLLM := os.Getenv("COMMIT_LLM")
+    var apiKey string
+
+    switch commitLLM {
+    case "google":
+        apiKey = os.Getenv("GOOGLE_API_KEY")
+        if apiKey == "" {
+            log.Fatalf("GOOGLE_API_KEY is not set")
+        }
+    case "grok":
+        apiKey = os.Getenv("GROK_API_KEY")
+        if apiKey == "" {
+            log.Fatalf("GROK_API_KEY is not set")
+        }
+    case "chatgpt":
+        apiKey = os.Getenv("OPENAI_API_KEY")
+        if apiKey == "" {
+            log.Fatalf("OPENAI_API_KEY is not set")
+        }
+    case "claude":
+        apiKey = os.Getenv("CLAUDE_API_KEY")
+        if apiKey == "" {
+            log.Fatalf("CLAUDE_API_KEY is not set")
+        }
+    default:
+        log.Fatalf("Invalid COMMIT_LLM value: %s", commitLLM)
+    }
 
 		// Get current directory
 		currentDir, err := os.Getwd()
@@ -111,6 +116,8 @@ func main() {
 			commitMsg, err = gemini.GenerateCommitMessage(config, changes, apiKey)
 		} else if os.Getenv("COMMIT_LLM") == "chatgpt" {
 			commitMsg, err = chatgpt.GenerateCommitMessage(config, changes, apiKey)
+		} else if os.Getenv("COMMIT_LLM") == "claude" {
+			commitMsg, err = claude.GenerateCommitMessage(config, changes, apiKey)
 		} else {
 			commitMsg, err = grok.GenerateCommitMessage(config, changes, apiKey)
 		}
@@ -139,5 +146,4 @@ func main() {
 
 		// Display changes preview
 		display.ShowChangesPreview(fileStats)
-	}
 }