zlei6/ragflow
1
0
Fork 0
mirror of https://github.com/infiniflow/ragflow.git synced 2026-07-23 17:06:42 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix(go): check model availability (#17156)

Browse Source
This commit is contained in:
Lynn
2026-07-21 19:02:14 +08:00
committed by GitHub
parent a5488d035a
commit 3f5b765a5f
1 changed files with 1 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
internal/service/tenant.go
View File

@@ -732,16 +732,7 @@ func (s *TenantService) checkModelAvailable(tenantID, providerName, instanceName
return err
}
modelSchema, err := dao.GetModelProviderManager().GetModelByName(providerName, modelName)
if err != nil {
return err
}
factoryModelType := factoryModelTypeName(modelType)
if !modelSchema.ModelTypeMap[factoryModelType] {
return fmt.Errorf("model %s isn't a %s model", modelName, modelType)
}
// Validate model availability through the DB (TenantModel table)
modelEntity, err := s.modelDAO.GetModelByProviderIDAndInstanceIDAndModelName(modelProvider.ID, modelInstance.ID, modelName)
if err != nil {
if dao.IsNotFoundErr(err) {