feat: 添加更多LLM模型

2026-07-19 19:06:07 +00:00 · 2024-01-30 16:29:54 +08:00
parent e9e458c877
commit 28bd232dda
3 changed files with 178 additions and 6 deletions
@@ -13,7 +13,7 @@ class LLMModelInfo(pydantic.BaseModel):

    name: str

-    provider: str
+    model_name: typing.Optional[str] = None

    token_mgr: token.TokenManager

@@ -23,5 +23,7 @@ class LLMModelInfo(pydantic.BaseModel):

    tool_call_supported: typing.Optional[bool] = False

+    max_tokens: typing.Optional[int] = 2048
+
    class Config:
        arbitrary_types_allowed = True