mirror of
https://github.com/langbot-app/LangBot.git
synced 2026-06-02 12:05:54 +00:00
30 lines
507 B
Python
30 lines
507 B
Python
from __future__ import annotations
|
|
|
|
import typing
|
|
|
|
import pydantic
|
|
|
|
from . import api
|
|
from . import token, tokenizer
|
|
|
|
|
|
class LLMModelInfo(pydantic.BaseModel):
|
|
"""模型"""
|
|
|
|
name: str
|
|
|
|
model_name: typing.Optional[str] = None
|
|
|
|
token_mgr: token.TokenManager
|
|
|
|
requester: api.LLMAPIRequester
|
|
|
|
tokenizer: 'tokenizer.LLMTokenizer'
|
|
|
|
tool_call_supported: typing.Optional[bool] = False
|
|
|
|
max_tokens: typing.Optional[int] = 2048
|
|
|
|
class Config:
|
|
arbitrary_types_allowed = True
|