fix(model):

1. when adding a model API key to the model list, a tenant_id uniqueness check needs to be added;
2.the Model Square has cancelled custom models;
3. optimization of the interface logic for customizing model configurations in the model list
This commit is contained in:
Timebomb2018
2026-02-09 10:02:34 +08:00
parent 4d98bace87
commit ebad5e00a3
4 changed files with 43 additions and 22 deletions

View File

@@ -48,13 +48,17 @@ class ModelConfigRepository:
raise
@staticmethod
def get_by_name(db: Session, name: str, tenant_id: uuid.UUID | None = None) -> Optional[ModelConfig]:
"""根据名称获取模型配置"""
db_logger.debug(f"根据名称查询模型配置: name={name}, tenant_id={tenant_id}")
def get_by_name(db: Session, name: str, provider: str | None = None, tenant_id: uuid.UUID | None = None) -> Optional[ModelConfig]:
"""根据名称和供应商获取模型配置"""
db_logger.debug(f"根据名称查询模型配置: name={name}, provider={provider}, tenant_id={tenant_id}")
try:
query = db.query(ModelConfig).filter(ModelConfig.name == name)
# 添加供应商过滤
if provider:
query = query.filter(ModelConfig.provider == provider)
# 添加租户过滤
if tenant_id:
query = query.filter(
@@ -69,7 +73,7 @@ class ModelConfigRepository:
db_logger.debug(f"模型配置查询成功: {model.name}")
return model
except Exception as e:
db_logger.error(f"根据名称查询模型配置失败: name={name} - {str(e)}")
db_logger.error(f"根据名称查询模型配置失败: name={name}, provider={provider} - {str(e)}")
raise
@staticmethod