Skip to content

Commit 214ec10

Browse files
committed
fix #22 : convert LLM provider selection to switch statement
- Replaced if-else with switch statement to explicitly handle all providers (gemini, chatgpt, claude, grok) - Used existing commitLLM variable instead of multiple os.Getenv calls
1 parent e4248b2 commit 214ec10

1 file changed

Lines changed: 7 additions & 4 deletions

File tree

cmd/commit-msg/main.go

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -112,13 +112,16 @@ func main() {
112112
}
113113

114114
var commitMsg string
115-
if os.Getenv("COMMIT_LLM") == "gemini" {
115+
switch commitLLM {
116+
case "gemini":
116117
commitMsg, err = gemini.GenerateCommitMessage(config, changes, apiKey)
117-
} else if os.Getenv("COMMIT_LLM") == "chatgpt" {
118+
case "chatgpt":
118119
commitMsg, err = chatgpt.GenerateCommitMessage(config, changes, apiKey)
119-
} else if os.Getenv("COMMIT_LLM") == "claude" {
120+
case "claude":
120121
commitMsg, err = claude.GenerateCommitMessage(config, changes, apiKey)
121-
} else {
122+
case "grok":
123+
commitMsg, err = grok.GenerateCommitMessage(config, changes, apiKey)
124+
default:
122125
commitMsg, err = grok.GenerateCommitMessage(config, changes, apiKey)
123126
}
124127

0 commit comments

Comments
 (0)