Skip to content

Commit d1929fe

Browse files
authored
Merge pull request #41 from adeeshperera/main
fix #22 : convert LLM provider selection to switch statement
2 parents e4248b2 + 4934d22 commit d1929fe

1 file changed

Lines changed: 6 additions & 5 deletions

File tree

cmd/commit-msg/main.go

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -112,13 +112,14 @@ 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-
commitMsg, err = claude.GenerateCommitMessage(config, changes, apiKey)
121-
} else {
120+
case "claude":
121+
commitMsg, err = claude.GenerateCommitMessage(config, changes, apiKey)
122+
default:
122123
commitMsg, err = grok.GenerateCommitMessage(config, changes, apiKey)
123124
}
124125

0 commit comments

Comments
 (0)