from __future__ import annotations

from collections.abc import Callable
from typing import Any

import pytest

from tests.infra.agent.middlewares.helpers import DummyModelRequest


@pytest.fixture
def dummy_model_request() -> Callable[..., DummyModelRequest]:
    def _factory(
        *,
        system_prompt: str = "",
        model: Any = None,
        runtime_context: Any = None,
    ) -> DummyModelRequest:
        return DummyModelRequest(
            system_prompt=system_prompt,
            model=model,
            runtime_context=runtime_context,
        )

    return _factory