File tree Expand file tree Collapse file tree 1 file changed +10
-5
lines changed Expand file tree Collapse file tree 1 file changed +10
-5
lines changed Original file line number Diff line number Diff line change 11import argparse
22import subprocess
33from commitgen .gitdiff import get_git_diff
4- from commitgen .generator import generate_commit_message
5-
4+ from commitgen .generator import generate_commit_message , LLM_MODEL
65
76# --- constant
8- parser = argparse .ArgumentParser (description = "Generate a commit message using GPT." )
7+ parser = argparse .ArgumentParser (description = f "Generate a commit message using GPT." )
98parser .add_argument (
109 "-l" , "--language" ,
1110 default = "korean" ,
1211 help = "Specify the language for the commit message (default: english)"
1312)
13+ parser .add_argument (
14+ "-m" , "--model" ,
15+ default = LLM_MODEL ,
16+ help = f"Specify the LLM model to use (default: { LLM_MODEL } )"
17+ )
1418args = parser .parse_args ()
1519
1620
@@ -39,10 +43,11 @@ def main():
3943 print ("⚠️ No changes to commit. Please run `git add` first." )
4044 return
4145
42- print (f"🤖 Generating commit message in { args .language } ...\n " )
46+ print (f"🤖 Generating commit message in { args .language } using GPT (model name: { args . model } ) ...\n " )
4347 message = generate_commit_message (
4448 git_diff_text = git_diff_text ,
45- language = args .language
49+ language = args .language ,
50+ llm_model = LLM_MODEL ,
4651 )
4752
4853 print ("\n ✅ Generated commit message:\n " )
You can’t perform that action at this time.
0 commit comments