mirror of
https://github.com/JefferyHcool/BiliNote.git
synced 2026-05-10 17:43:40 +08:00
171dea5e0d
### v1.1.0 - #### Added - 新增 AI 笔记风格选择 - 新增 AI 笔记返回格式选择 - 添加 AI 自定义笔记备注 Prompt - 添加任务失败重试 - 添加全局设置页，可在设置页进行模型设置 - #### Optimize - 优化前端样式，优化用户体验 - 增加生成中间产物，可用于失败后加快生成速度 - #### Fix - 修复视频截图视频过早删除错误
37 lines
1.0 KiB
Python
37 lines
1.0 KiB
Python
from fastapi import APIRouter
|
|
from pydantic import BaseModel
|
|
|
|
from app.services.model import ModelService
|
|
from app.utils.response import ResponseWrapper as R
|
|
router = APIRouter()
|
|
modelService = ModelService()
|
|
class CreateModelRequest(BaseModel):
|
|
provider_id: str
|
|
model_name: str
|
|
|
|
# 返回体：模型信息
|
|
class ModelItem(BaseModel):
|
|
id: int
|
|
model_name: str
|
|
@router.get("/model_list")
|
|
def model_list():
|
|
try:
|
|
return R.success(modelService.get_all_models(True),msg="获取模型列表成功")
|
|
except Exception as e:
|
|
return R.error(e)
|
|
|
|
@router.get("/model_list/{provider_id}")
|
|
def model_list(provider_id):
|
|
try:
|
|
return R.success(modelService.get_all_models_by_id(provider_id))
|
|
except Exception as e:
|
|
return R.error(e)
|
|
|
|
@router.post("/models")
|
|
def create_model(data: CreateModelRequest):
|
|
success = ModelService.add_new_model(data.provider_id, data.model_name)
|
|
if not success:
|
|
raise R.error("模型添加失败")
|
|
return R.success(msg="模型添加成功")
|
|
|