main.py

import logging
import os

import flask
import functions_framework

from bugbug import generative_model_tool
from bugbug.tools.release_notes import ReleaseNotesCommitsSelector
from bugbug.utils import get_secret

logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)

os.environ["OPENAI_API_KEY"] = get_secret("OPENAI_API_KEY")

tool: ReleaseNotesCommitsSelector | None = None

DEFAULT_CHUNK_SIZE = 1000


@functions_framework.http
def handle_release_notes(request: flask.Request):
    global tool

    if request.method != "GET":
        return "Only GET requests are allowed", 405

    version = request.args.get("version")
    if not version:
        return "Missing 'version' query parameter", 400

    if tool is None:
        logger.info("Initializing new ReleaseNotesCommitsSelector...")

        llm = generative_model_tool.create_openai_llm()
        tool = ReleaseNotesCommitsSelector(chunk_size=DEFAULT_CHUNK_SIZE, llm=llm)
        tool.llm_name = "openai"
        tool.chunk_size = DEFAULT_CHUNK_SIZE

    notes = tool.get_final_release_notes_commits(version=version)

    if not notes:
        return {"commits": []}, 200

    return {"commits": notes}, 200