Class: Ace::GitCommit::Molecules::MessageGenerator
- Inherits:
-
Object
- Object
- Ace::GitCommit::Molecules::MessageGenerator
- Defined in:
- lib/ace/git_commit/molecules/message_generator.rb
Overview
MessageGenerator generates commit messages using LLM
Defined Under Namespace
Classes: BatchParseError
Constant Summary collapse
- DEFAULT_MODEL =
"glite"- MAX_TOKENS =
8192- SYSTEM_PROMPT_PATH =
"ace-git-commit/handbook/prompts/git-commit.system.md"- COMMIT_HEADER_PATTERN =
/\A(feat|fix|docs|style|refactor|test|chore|spec|perf|build|ci|revert)(\([^)]+\))?:\s+\S+/
Instance Method Summary collapse
-
#generate(diff, intention: nil, files: [], config: nil) ⇒ String
Generate a commit message from diff.
-
#generate_batch(groups_context, intention: nil, config: nil) ⇒ Hash
Generate commit messages for multiple groups in one LLM call.
-
#initialize(config = nil) ⇒ MessageGenerator
constructor
A new instance of MessageGenerator.
Constructor Details
#initialize(config = nil) ⇒ MessageGenerator
18 19 20 21 |
# File 'lib/ace/git_commit/molecules/message_generator.rb', line 18 def initialize(config = nil) @config = config || {} @model = @config.fetch("model", DEFAULT_MODEL) end |
Instance Method Details
#generate(diff, intention: nil, files: [], config: nil) ⇒ String
Generate a commit message from diff
29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 |
# File 'lib/ace/git_commit/molecules/message_generator.rb', line 29 def generate(diff, intention: nil, files: [], config: nil) system_prompt = load_system_prompt user_prompt = build_user_prompt(diff, intention, files) # Use QueryInterface with named parameters matching CLI response = Ace::LLM::QueryInterface.query( resolve_model(config), user_prompt, system: system_prompt, temperature: 0.7, timeout: 60, max_tokens: MAX_TOKENS ) (response[:text]) rescue Ace::LLM::Error => e raise Error, "Failed to generate commit message: #{e.message}" end |
#generate_batch(groups_context, intention: nil, config: nil) ⇒ Hash
Generate commit messages for multiple groups in one LLM call
53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 |
# File 'lib/ace/git_commit/molecules/message_generator.rb', line 53 def generate_batch(groups_context, intention: nil, config: nil) return {messages: [], order: []} if groups_context.empty? if groups_context.length == 1 msg = generate(groups_context.first[:diff], intention: intention, files: groups_context.first[:files], config: config) return {messages: [msg], order: [groups_context.first[:scope_name]]} end system_prompt = load_batch_system_prompt user_prompt = build_batch_user_prompt(groups_context, intention) response = Ace::LLM::QueryInterface.query( resolve_model(config), user_prompt, system: system_prompt, temperature: 0.7, timeout: 120, max_tokens: MAX_TOKENS ) parse_batch_response(response[:text], groups_context) rescue BatchParseError => e repaired = retry_batch_parse(groups_context, intention, response[:text], e., config) return repaired if repaired raise Error, "Failed to generate batch commit messages: #{e.message}" rescue Ace::LLM::Error => e raise Error, "Failed to generate batch commit messages: #{e.message}" end |