mirror of
https://gh-proxy.org/https://github.com/klinsmaya/SQLBot.git
synced 2026-11-09 01:36:57 +08:00
30 lines
1.3 KiB
Python
30 lines
1.3 KiB
Python
|
|
from typing import List
|
|
from pydantic import BaseModel, Field
|
|
|
|
from apps.swagger.i18n import PLACEHOLDER_PREFIX
|
|
from common.core.schemas import BaseCreatorDTO
|
|
|
|
class AiModelItem(BaseModel):
|
|
name: str = Field(description=f"{PLACEHOLDER_PREFIX}model_name")
|
|
model_type: int = Field(description=f"{PLACEHOLDER_PREFIX}model_type")
|
|
base_model: str = Field(description=f"{PLACEHOLDER_PREFIX}base_model")
|
|
supplier: int = Field(description=f"{PLACEHOLDER_PREFIX}supplier")
|
|
protocol: int = Field(description=f"{PLACEHOLDER_PREFIX}protocol")
|
|
default_model: bool = Field(default=False, description=f"{PLACEHOLDER_PREFIX}default_model")
|
|
|
|
class AiModelGridItem(AiModelItem, BaseCreatorDTO):
|
|
ws_mapping_count: int = Field(default=0, description="workspace mapping count")
|
|
|
|
class AiModelConfigItem(BaseModel):
|
|
key: str = Field(description=f"{PLACEHOLDER_PREFIX}arg_name")
|
|
val: object = Field(description=f"{PLACEHOLDER_PREFIX}arg_val")
|
|
name: str = Field(description=f"{PLACEHOLDER_PREFIX}arg_show_name")
|
|
|
|
class AiModelCreator(AiModelItem):
|
|
api_domain: str = Field(description=f"{PLACEHOLDER_PREFIX}api_domain")
|
|
api_key: str = Field(description=f"{PLACEHOLDER_PREFIX}api_key")
|
|
config_list: List[AiModelConfigItem] = Field(description=f"{PLACEHOLDER_PREFIX}config_list")
|
|
|
|
class AiModelEditor(AiModelCreator, BaseCreatorDTO):
|
|
pass |