from pathlib import Path

import pytest

from haiku.rag.config import get_config


@pytest.fixture(scope="module")
def vcr_cassette_dir(request):
    """Cassettes sit with the rest, under `tests/cassettes/multi_db/`."""
    module = request.module.__name__.rsplit(".", 1)[-1]
    return str(Path(__file__).parent.parent / "cassettes" / "multi_db" / module)


@pytest.fixture
def query_embedding(monkeypatch):
    """Vector search with no embedder behind it, recording the queries
    embedded."""
    from haiku.rag.embeddings import EmbedderWrapper

    embedded: list[str] = []

    async def embed_query(self, text):
        embedded.append(text)
        return [0.1] * get_config().embeddings.model.vector_dim

    monkeypatch.setattr(EmbedderWrapper, "embed_query", embed_query)
    return embedded