pq-web-ai/app/providers/base.py

from typing import List, Optional, Protocol

from ..schemas.base import ChatMessage, ChatResponse, DataResponse


class LLMProvider(Protocol):
    def chat(
        self,
        messages: List[ChatMessage],
        *,
        model: Optional[str] = None,
        temperature: Optional[float] = None,
        max_tokens: Optional[int] = None,
    ) -> ChatResponse:
        ...

class SpeechProvider(Protocol):
    def text_to_speech(
        self,
        text: str,
        *,
        model: Optional[str] = None,
    ) -> DataResponse:
        ...