llm-connect/tests/conftest.py

"""
Shared pytest fixtures for llm-connect tests.
"""

import pytest

from llm_connect.adapter import MockLLMAdapter
from llm_connect.models import LLMResponse, RunConfig


@pytest.fixture
def run_config():
    """Default RunConfig for tests."""
    return RunConfig()


@pytest.fixture
def mock_adapter():
    """MockLLMAdapter with a predictable response."""
    return MockLLMAdapter(mock_response="test response")


@pytest.fixture
def sample_response():
    """A minimal valid LLMResponse."""
    return LLMResponse(content="hello", model="test-model")