from __future__ import annotations

import pytest

from livekit.plugins.perplexity import __version__, responses
from livekit.plugins.perplexity.responses.llm import PERPLEXITY_RESPONSES_BASE_URL

pytestmark = pytest.mark.plugin("perplexity")


def test_default_model_base_url_and_transport(monkeypatch: pytest.MonkeyPatch) -> None:
    monkeypatch.setenv("PERPLEXITY_API_KEY", "test-key")
    llm = responses.LLM()
    assert llm.model == "sonar-pro"
    assert llm._opts.use_websocket is False
    assert PERPLEXITY_RESPONSES_BASE_URL == "https://api.perplexity.ai/v1"
    assert str(llm._client.base_url).startswith("https://api.perplexity.ai/v1")


def test_attribution_header_attached(monkeypatch: pytest.MonkeyPatch) -> None:
    monkeypatch.setenv("PERPLEXITY_API_KEY", "test-key")
    llm = responses.LLM()
    assert llm._client.default_headers["X-Pplx-Integration"] == f"livekit-agents/{__version__}"


def test_missing_api_key_raises(monkeypatch: pytest.MonkeyPatch) -> None:
    monkeypatch.delenv("PERPLEXITY_API_KEY", raising=False)
    with pytest.raises(ValueError, match="PERPLEXITY_API_KEY"):
        responses.LLM()


def test_responses_submodule_exported() -> None:
    assert responses.LLM is not None