"""Live smoke test against the real LiteLLM proxy."""

import pytest

from nisse_dev.canonical import AssistantText, StoredEvent, UserMessage
from nisse_dev.catalog import ModelCatalog
from nisse_dev.config import AgentConfig
from nisse_dev.context import BudgetedClient
from nisse_dev.llm import LiteLLMClient

pytestmark = pytest.mark.integration


def test_the_driver_model_completes_via_litellm(
    litellm: LiteLLMClient, live_config: AgentConfig
) -> None:
    events = [
        StoredEvent(
            id=1,
            ts="t",
            event=UserMessage(
                text="Reply with exactly one word: pong", source="smoke-test"
            ),
        )
    ]
    result = litellm.complete(events, tools=None, model=live_config.roles.driver)
    texts = [e.text for e in result if isinstance(e, AssistantText)]
    assert texts, f"no assistant text in {result!r}"
    # A plumbing check, so any answer will do: what the model chooses to say
    # is not under test, and asserting the word made CI flaky (it once
    # replied "ping").
    assert texts[0].strip(), f"empty assistant text in {result!r}"


def test_declared_limits_reach_a_budgeted_request(
    litellm: LiteLLMClient, live_config: AgentConfig
) -> None:
    """The whole chain: LiteLLM's model_info, the catalog, the budget, and a
    real request through the proxy with the ceiling the budget chose."""
    catalog = ModelCatalog.discover(live_config.llm_base_url, live_config.llm_token)
    roles = live_config.roles.pinned()
    catalog.require_limits(roles)  # every test role has a declared window

    model = live_config.roles.driver
    facts = catalog.facts(model)
    client = BudgetedClient(
        litellm,
        lambda m: (
            catalog.facts(m).max_input_tokens,
            catalog.facts(m).max_output_tokens,
        ),
        max_output_tokens=live_config.max_output_tokens,
    )
    assert client.budget_for(model).window == facts.max_input_tokens

    events = [
        StoredEvent(
            id=1, ts="t", event=UserMessage(text="Reply with exactly: pong", source="t")
        )
    ]
    produced = client.complete(events, None, model, system="Be terse.")
    texts = [e.text for e in produced if isinstance(e, AssistantText)]
    assert texts and "pong" in texts[0].lower()
    assert client.last_usage.get("input", 0) > 0