llm_client.py

import requests


class LLMClient:
    def __init__(self, api_key: str, model: str, url: str):
        self.url = url
        self.headers = {
            "Authorization": f"Bearer {api_key}",
            "Content-Type": "application/json",
        }
        self.model = model

    def review_diff(self, diff: str) -> str:
        prompt = (
            "Review this PR diff. Respond in markdown with sections: "
            "Bugs, Improvements, Positives.\n\n"
            f"{diff[:8000]}"
        )
        payload = {
            "model": self.model,
            "messages": [
                {"role": "system", "content": "You are a senior code reviewer."},
                {"role": "user", "content": prompt},
            ],
            "temperature": 0.2,
        }
        r = requests.post(self.url, headers=self.headers, json=payload, timeout=60)
        r.raise_for_status()
        data = r.json()
        return data["choices"][0]["message"]["content"].strip()