from instructor import OpenAISchema
from instructor.dsl import IterableModel
from typing import cast


def test_multi_task():
    class Search(OpenAISchema):
        """This is the search docstring"""

        id: int
        query: str

    IterableSearch = cast(type[OpenAISchema], IterableModel(Search))
    assert IterableSearch.openai_schema["name"] == "IterableSearch"
    assert (
        IterableSearch.openai_schema["description"]
        == "Correct segmentation of `Search` tasks"
    )