mirror of
https://github.com/JefferyHcool/BiliNote.git
synced 2026-05-11 18:10:06 +08:00
171dea5e0d
### v1.1.0 - #### Added - 新增 AI 笔记风格选择 - 新增 AI 笔记返回格式选择 - 添加 AI 自定义笔记备注 Prompt - 添加任务失败重试 - 添加全局设置页，可在设置页进行模型设置 - #### Optimize - 优化前端样式，优化用户体验 - 增加生成中间产物，可用于失败后加快生成速度 - #### Fix - 修复视频截图视频过早删除错误
100 lines
2.7 KiB
Python
100 lines
2.7 KiB
Python
from typing import Optional
|
|
from fastapi import APIRouter
|
|
from pydantic import BaseModel
|
|
|
|
from app.models.model_config import ModelConfig
|
|
from app.services.model import ModelService
|
|
from app.utils.response import ResponseWrapper as R
|
|
from app.services.provider import ProviderService
|
|
|
|
router = APIRouter()
|
|
|
|
# ✅ 新增 type 字段
|
|
class ProviderRequest(BaseModel):
|
|
name: str
|
|
api_key: str
|
|
base_url: str
|
|
logo: Optional[str] = None
|
|
type: str
|
|
|
|
class TestRequest(BaseModel):
|
|
|
|
api_key: str
|
|
base_url:str
|
|
class ProviderUpdateRequest(BaseModel):
|
|
id: str
|
|
name: Optional[str] = None
|
|
api_key: Optional[str] = None
|
|
base_url: Optional[str] = None
|
|
logo: Optional[str] = None
|
|
type: Optional[str] = None
|
|
enabled:Optional[int] = None
|
|
|
|
@router.post("/add_provider")
|
|
def add_provider(data: ProviderRequest):
|
|
try:
|
|
ProviderService.add_provider(
|
|
name=data.name,
|
|
api_key=data.api_key,
|
|
base_url=data.base_url,
|
|
logo=data.logo,
|
|
type_=data.type
|
|
)
|
|
return R.success(msg='添加模型供应商成功')
|
|
except Exception as e:
|
|
return R.error(msg=e)
|
|
|
|
@router.get("/get_all_providers")
|
|
def get_all_providers():
|
|
try:
|
|
res = ProviderService.get_all_providers()
|
|
return R.success(data=res)
|
|
except Exception as e:
|
|
return R.error(msg=e)
|
|
|
|
@router.get("/get_provider_by_id/{id}")
|
|
def get_provider_by_id(id: str):
|
|
try:
|
|
res = ProviderService.get_provider_by_id(id)
|
|
return R.success(data=res)
|
|
except Exception as e:
|
|
return R.error(msg=e)
|
|
|
|
@router.get("/get_provider_by_name/{name}")
|
|
def get_provider_by_name(name: str):
|
|
try:
|
|
res = ProviderService.get_provider_by_name(name)
|
|
return R.success(data=res)
|
|
except Exception as e:
|
|
return R.error(msg=e)
|
|
|
|
|
|
@router.post("/update_provider")
|
|
def update_provider(data: ProviderUpdateRequest):
|
|
try:
|
|
if all(
|
|
field is None
|
|
for field in [data.name, data.api_key, data.base_url, data.logo, data.type,data.enabled]
|
|
):
|
|
return R.error(msg='请至少填写一个参数')
|
|
|
|
ProviderService.update_provider(
|
|
id=data.id,
|
|
data=dict(data)
|
|
)
|
|
return R.success(msg='更新模型供应商成功')
|
|
except Exception as e:
|
|
print(e)
|
|
return R.error(msg=e)
|
|
|
|
@router.post('/connect_test')
|
|
def gpt_connect_test(data:TestRequest):
|
|
try:
|
|
|
|
res= ModelService().connect_test(data.api_key,data.base_url)
|
|
if not res:
|
|
return R.error(msg='连接失败')
|
|
return R.success(msg='连接成功')
|
|
except Exception as e:
|
|
print(e)
|
|
return R.error(msg=e) |