15 lines
565 B
Python
15 lines
565 B
Python
import requests
|
|
|
|
def generate_commit_message(prompt, model="qwen2.5-coder:3b", base_url="http://localhost:11434"):
|
|
"""Generate commit message using Ollama."""
|
|
try:
|
|
response = requests.post(
|
|
f"{base_url}/api/generate",
|
|
json={"model": model, "prompt": prompt, "stream": False},
|
|
timeout=60
|
|
)
|
|
response.raise_for_status()
|
|
return response.json().get('response', '').strip()
|
|
except requests.exceptions.RequestException as e:
|
|
raise ConnectionError(f"Failed to connect to Ollama: {e}")
|