from __future__ import annotations

import typing

import pydantic

from . import api
from . import token, tokenizer


class LLMModelInfo(pydantic.BaseModel):
    """模型"""

    name: str

    provider: str

    token_mgr: token.TokenManager

    requester: api.LLMAPIRequester

    tokenizer: 'tokenizer.LLMTokenizer'

    tool_call_supported: typing.Optional[bool] = False

    class Config:
        arbitrary_types_allowed = True