mirror of
https://github.com/JefferyHcool/BiliNote.git
synced 2026-05-21 16:20:40 +08:00
fix(frontend): 修复供应商开关切换不能实时生效的问题
This commit is contained in:
@@ -1,5 +1,6 @@
|
||||
from app.db.engine import get_db
|
||||
from app.db.models.models import Model
|
||||
from app.db.models.providers import Provider
|
||||
|
||||
|
||||
def get_model_by_provider_and_name(provider_id: int, model_name: str):
|
||||
@@ -58,7 +59,8 @@ def delete_model(model_id: int):
|
||||
def get_all_models():
|
||||
db = next(get_db())
|
||||
try:
|
||||
models = db.query(Model).all()
|
||||
# 只查询启用状态供应商的模型
|
||||
models = db.query(Model).join(Provider, Model.provider_id == Provider.id).filter(Provider.enabled == 1).all()
|
||||
return [
|
||||
{"id": m.id, "provider_id": m.provider_id, "model_name": m.model_name}
|
||||
for m in models
|
||||
|
||||
@@ -77,11 +77,14 @@ def update_provider(data: ProviderUpdateRequest):
|
||||
):
|
||||
return R.error(msg='请至少填写一个参数')
|
||||
|
||||
provider_id =ProviderService.update_provider(
|
||||
updated_provider =ProviderService.update_provider(
|
||||
id=data.id,
|
||||
data=dict(data)
|
||||
)
|
||||
return R.success(msg='更新模型供应商成功',data={'id': provider_id})
|
||||
if updated_provider:
|
||||
return R.success(msg='更新模型供应商成功', data=updated_provider)
|
||||
else:
|
||||
return R.error(msg='更新模型供应商失败')
|
||||
except Exception as e:
|
||||
print(e)
|
||||
return R.error(msg=str(e))
|
||||
|
||||
@@ -123,7 +123,12 @@ class ProviderService:
|
||||
filtered_data = {k: v for k, v in data.items() if v is not None and k != 'id'}
|
||||
print('更新模型供应商',filtered_data)
|
||||
update_provider(id, **filtered_data)
|
||||
return id
|
||||
# 获取更新后的供应商信息
|
||||
updated_provider = get_provider_by_id(id)
|
||||
return {
|
||||
'id': id,
|
||||
'enabled': updated_provider.enabled,
|
||||
}
|
||||
|
||||
except Exception as e:
|
||||
print('更新模型供应商失败:',e)
|
||||
|
||||
Reference in New Issue
Block a user