from openai import OpenAI


class OpenrouterProvider:
    """Generates content using OpenRouter (OpenAI-compatible API)."""

    BASE_URL = "https://openrouter.ai/api/v1"

    def generate(
        self, prompt: str, api_key: str, model: str = "openai/gpt-3.5-turbo", **kwargs
    ) -> str:
        """Generates a response using OpenRouter."""
        if not api_key:
            raise ValueError("OpenRouter API key is required.")

        try:
            client = OpenAI(
                base_url=self.BASE_URL,
                api_key=api_key,
            )

            response = client.chat.completions.create(
                model=model,
                messages=[{"role": "user", "content": prompt}],
                extra_headers={
                    "HTTP-Referer": "https://github.com/S4NKALP/devgen",  # Optional
                    "X-Title": "devgen CLI",  # Optional
                },
                **kwargs,
            )
            return response.choices[0].message.content.strip()
        except Exception as e:
            raise RuntimeError(f"OpenRouter generation failed: {e}")