from ai.openai_provider import OpenAIProvider


class OllamaProvider(OpenAIProvider):
    def __init__(self, base_url: str = "http://host.docker.internal:11434", model: str = "llama3.2"):
        super().__init__(
            api_key="ollama",
            model=model,
            base_url=base_url.rstrip("/") + "/v1",
        )
