Skip to content

Commit ede03d8

Browse files
committed
feat: 커밋 메시지 생성을 위한 LLM 모델 선택 기능 추가
1 parent 9c1e5fc commit ede03d8

File tree

1 file changed

+10
-5
lines changed

1 file changed

+10
-5
lines changed

main.py

Lines changed: 10 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,16 +1,20 @@
11
import argparse
22
import subprocess
33
from 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.")
98
parser.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+
)
1418
args = 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")

0 commit comments

Comments
 (0)