diff --git a/backend/apps/system/schemas/ai_model_schema.py b/backend/apps/system/schemas/ai_model_schema.py new file mode 100644 index 0000000..8b5946b --- /dev/null +++ b/backend/apps/system/schemas/ai_model_schema.py @@ -0,0 +1,29 @@ + +from typing import List +from pydantic import BaseModel + +from common.core.schemas import BaseCreatorDTO + +class AiModelItem(BaseModel): + name: str + model_type: int + base_model: str + supplier: int + protocol: int + default_model: bool = False + +class AiModelGridItem(AiModelItem, BaseCreatorDTO): + pass + +class AiModelConfigItem(BaseModel): + key: str + val: object + name: str + +class AiModelCreator(AiModelItem): + api_domain: str + api_key: str + config_list: List[AiModelConfigItem] + +class AiModelEditor(AiModelCreator, BaseCreatorDTO): + pass \ No newline at end of file