"""Inference adapters for the execution-feedback agent."""

from __future__ import annotations

from typing import Any


class OpenAICompatibleGenerator:
    """Callable adapter for MiniMind, vLLM, SGLang, or other OpenAI-style APIs."""

    def __init__(
        self,
        *,
        base_url: str = "http://localhost:8998/v1",
        api_key: str = "not-needed",
        model: str = "minimind",
        temperature: float = 0.7,
        max_tokens: int = 1024,
        open_thinking: bool = False,
        client: Any = None,
    ) -> None:
        if client is None:
            try:
                from openai import OpenAI
            except ImportError as exc:
                raise RuntimeError("Install the project 'openai' dependency to use this backend.") from exc
            client = OpenAI(api_key=api_key, base_url=base_url)
        self.client = client
        self.model = model
        self.temperature = temperature
        self.max_tokens = max_tokens
        self.open_thinking = open_thinking

    def __call__(self, prompt: str) -> str:
        response = self.client.chat.completions.create(
            model=self.model,
            messages=[{"role": "user", "content": prompt}],
            temperature=self.temperature,
            max_tokens=self.max_tokens,
            stream=False,
            extra_body={"open_thinking": self.open_thinking},
        )
        if not response.choices or response.choices[0].message is None:
            raise RuntimeError("Model API returned no completion choice.")
        content = response.choices[0].message.content
        if not content:
            raise RuntimeError("Model API returned an empty completion.")
        return content