fix: resolve CI build failures
This commit is contained in:
@@ -1,109 +0,0 @@
|
||||
"""Prompt management for Git Commit AI."""
|
||||
|
||||
from pathlib import Path
|
||||
from typing import Optional
|
||||
|
||||
from git_commit_ai.core.config import Config, get_config
|
||||
|
||||
|
||||
class PromptBuilder:
|
||||
"""Builder for commit message prompts."""
|
||||
|
||||
DEFAULT_PROMPT = """You are a helpful assistant that generates git commit messages.
|
||||
|
||||
Analyze the following git diff and generate a concise, descriptive commit message.
|
||||
The message should:
|
||||
- Be clear and descriptive
|
||||
- Explain what changed and why
|
||||
- Be in present tense
|
||||
- Not exceed 72 characters for the first line if possible
|
||||
|
||||
Git diff:
|
||||
```
|
||||
{diff}
|
||||
```
|
||||
|
||||
{few_shot}
|
||||
|
||||
Generate 3 different commit message suggestions, one per line.
|
||||
Format: Just the commit messages, one per line, nothing else.
|
||||
|
||||
Suggestions:
|
||||
"""
|
||||
|
||||
CONVENTIONAL_PROMPT = """You are a helpful assistant that generates conventional git commit messages.
|
||||
|
||||
Generate a commit message in the conventional commit format:
|
||||
- type(scope): description
|
||||
- Examples: feat(auth): add login, fix: resolve memory leak
|
||||
|
||||
Valid types: feat, fix, docs, style, refactor, perf, test, build, ci, chore, revert
|
||||
|
||||
Analyze the following git diff and generate commit messages.
|
||||
Git diff:
|
||||
```
|
||||
{diff}
|
||||
```
|
||||
|
||||
{few_shot}
|
||||
|
||||
Generate 3 different commit message suggestions, one per line.
|
||||
Format: Just the commit messages, one per line, nothing else.
|
||||
|
||||
Suggestions:
|
||||
"""
|
||||
|
||||
SYSTEM_DEFAULT = "You are a helpful assistant that generates clear and concise git commit messages."
|
||||
|
||||
SYSTEM_CONVENTIONAL = "You are a helpful assistant that generates conventional git commit messages. Always use the format: type(scope): description. Valid types: feat, fix, docs, style, refactor, perf, test, build, ci, chore, revert"
|
||||
|
||||
def __init__(self, config: Optional[Config] = None):
|
||||
self.config = config or get_config()
|
||||
self._prompt_dir = Path(self.config.prompt_directory)
|
||||
|
||||
def build_prompt(self, diff: str, context: Optional[str] = None, conventional: bool = False) -> str:
|
||||
few_shot = self._build_few_shot(context)
|
||||
|
||||
if conventional:
|
||||
template = self._get_conventional_template()
|
||||
else:
|
||||
template = self._get_default_template()
|
||||
|
||||
prompt = template.format(diff=diff[:10000] if len(diff) > 10000 else diff, few_shot=few_shot)
|
||||
return prompt
|
||||
|
||||
def _get_default_template(self) -> str:
|
||||
custom_path = self._prompt_dir / "default.txt"
|
||||
if custom_path.exists():
|
||||
return custom_path.read_text()
|
||||
return self.DEFAULT_PROMPT
|
||||
|
||||
def _get_conventional_template(self) -> str:
|
||||
custom_path = self._prompt_dir / "conventional.txt"
|
||||
if custom_path.exists():
|
||||
return custom_path.read_text()
|
||||
return self.CONVENTIONAL_PROMPT
|
||||
|
||||
def _build_few_shot(self, context: Optional[str]) -> str:
|
||||
if not context:
|
||||
return ""
|
||||
return f"\n\nRecent commit history for context:\n{context}"
|
||||
|
||||
def get_system_prompt(self, conventional: bool = False) -> str:
|
||||
if conventional:
|
||||
custom_path = self._prompt_dir / "system_conventional.txt"
|
||||
if custom_path.exists():
|
||||
return custom_path.read_text()
|
||||
return self.SYSTEM_CONVENTIONAL
|
||||
|
||||
custom_path = self._prompt_dir / "system_default.txt"
|
||||
if custom_path.exists():
|
||||
return custom_path.read_text()
|
||||
return self.SYSTEM_DEFAULT
|
||||
|
||||
def get_supported_languages(self) -> list[str]:
|
||||
return ["Python", "JavaScript", "TypeScript", "Java", "Go", "Rust", "Ruby", "PHP", "C", "C++", "C#", "Swift", "Kotlin", "Scala", "HTML", "CSS", "SQL", "Shell"]
|
||||
|
||||
|
||||
def get_prompt_builder(config: Optional[Config] = None) -> PromptBuilder:
|
||||
return PromptBuilder(config)
|
||||
|
||||
Reference in New Issue
Block a user