from __future__ import annotations

from typing_extensions import assert_type

import openai
from pydantic import BaseModel

from instructor import (
    AsyncInstructor,
    Instructor,
    Maybe,
    Partial,
    from_anyscale,
    from_databricks,
    from_deepseek,
    from_openai,
    from_provider,
    from_together,
    response_schema,
)
from instructor.v2.dsl.maybe import MaybeBase


class User(BaseModel):
    name: str


def check_public_factories(
    sync_client: openai.OpenAI,
    async_client: openai.AsyncOpenAI,
) -> None:
    assert_type(from_openai(sync_client), Instructor)
    assert_type(from_openai(async_client), AsyncInstructor)
    assert_type(from_anyscale("model"), Instructor)
    assert_type(from_anyscale("model", async_client=True), AsyncInstructor)
    assert_type(from_together("model"), Instructor)
    assert_type(from_together("model", async_client=True), AsyncInstructor)
    assert_type(from_databricks("model"), Instructor)
    assert_type(from_databricks("model", async_client=True), AsyncInstructor)
    assert_type(from_deepseek("model"), Instructor)
    assert_type(from_deepseek("model", async_client=True), AsyncInstructor)
    assert_type(from_provider("openai/gpt-4o-mini"), Instructor)
    assert_type(from_provider("openai/gpt-4o-mini", async_client=True), AsyncInstructor)


def check_public_model_helpers() -> None:
    assert_type(response_schema(User), type[User])
    assert_type(Maybe(User), type[MaybeBase[User]])
    assert_type(Partial[User], type[User])