(FEAT)[Refactor commit message generation]: enhance the generate_commit_message function to utilize configuration settings and improve error handling.

src/generator.py

@@ -22,9 +22,30 @@ def load_phi2(model_path: str, context_window: int = 2048):
 
 
 # Generation Function
-def generate_commit_message(llm, prompt: str, max_tokens: int = 512):
-    response = llm(prompt, max_tokens=max_tokens)
-    return response["choices"][0]["text"].strip()
+def generate_commit_message(prompt: str, config: dict, max_tokens: int = None):
+    """
+    Generate a commit message using the Phi-2 model.
+
+    Args:
+        prompt (str): The input prompt for generation
+        config (dict): Configuration dictionary containing model settings
+        max_tokens (int, optional): Maximum tokens to generate
+
+    Returns:
+        str: Generated commit message
+    """
+    if max_tokens is None:
+        max_tokens = config.get("generator", {}).get("max_tokens", 512)
+
+    model_path = config["generator"]["model_path"]
+    n_ctx = config.get("generator", {}).get("n_ctx", 2048)
+
+    try:
+        llm = Llama(model_path=model_path, n_ctx=n_ctx, verbose=False)
+        response = llm(prompt, max_tokens=max_tokens)
+        return response["choices"][0]["text"].strip()
+    except Exception as e:
+        raise RuntimeError(f"Failed to generate with Phi-2: {e}") from e
 
 
 # Fallback Logic