mirror of
https://github.com/RockChinQ/QChatGPT.git
synced 2024-11-16 03:32:33 +08:00
28 lines
463 B
Python
28 lines
463 B
Python
from __future__ import annotations
|
|
|
|
import typing
|
|
|
|
import pydantic
|
|
|
|
from . import api
|
|
from . import token
|
|
|
|
|
|
class LLMModelInfo(pydantic.BaseModel):
|
|
"""模型"""
|
|
|
|
name: str
|
|
|
|
model_name: typing.Optional[str] = None
|
|
|
|
token_mgr: token.TokenManager
|
|
|
|
requester: api.LLMAPIRequester
|
|
|
|
tool_call_supported: typing.Optional[bool] = False
|
|
|
|
vision_supported: typing.Optional[bool] = False
|
|
|
|
class Config:
|
|
arbitrary_types_allowed = True
|