from __future__ import annotations

import pytest

from django_anyllm.registry import ModelRegistry

from .models import Author, Book


@pytest.fixture
def registry(db) -> ModelRegistry:
    """A fresh ModelRegistry with Author and Book registered."""
    reg = ModelRegistry()
    reg.register(Author)
    reg.register(Book)
    return reg
