llm.py

from google import genai
import os

client = genai.Client(api_key=os.getenv("GEMINI_API_KEY"))

def suggest_label(issue: dict) -> str:
    try:

        prompt = (
            f"Given this GitHub issue, suggest one label from this list: "
            f"bug, enhancement, documentation, question.\n\n"
            f"Title: {issue['title']}\n"
            f"Body: {issue['body']}\n\n"
            f"Reply with just the label name."
        )
        response = client.models.generate_content(
            model="gemini-1.5-flash-8b",
            contents=prompt
        )
        return (response.text or "question").strip()
    except Exception:
        return "question"