from ..schemas.llm import ChatRequest, ChatResponse
from ..providers.base import LLMProvider


class LLMService:
    def __init__(self, provider: LLMProvider) -> None:
        self._provider = provider

    def chat(self, req: ChatRequest) -> ChatResponse:
        return self._provider.chat(
            req.messages,
            model=req.model,
            temperature=req.temperature,
            max_tokens=req.max_tokens,
        )