Skip to content

Commit ce1d73e

Browse files
authored
Merge pull request #2 from adeeshperera/LLM-provider-selection-to-switch-statement
fix #22 : convert LLM provider selection to switch statement
2 parents ea4a548 + 214ec10 commit ce1d73e

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)