修改模型、功能查找判断逻辑

This commit is contained in:
junleea 2025-03-28 15:21:57 +08:00
parent eb67ddb45b
commit 63f81760ae
2 changed files with 4 additions and 4 deletions

View File

@ -199,11 +199,11 @@ func doReceiveGenChatMessage(userId int, sessionID *uint, data *proto.WSMessageR
var model dao.Model
//查看请求功能类型
models, funcs := service.FindFuncModelByFunction(data.Function, userId)
log.Println("find function model by function:", models, funcs)
if funcs == nil || len(funcs) == 0 {
//log.Println("find function model by function:", models, funcs)
if len(funcs) == 0 {
return errors.New("function not exist")
}
if models == nil || len(models) == 0 {
if len(models) == 0 {
return errors.New("model not exist")
}
model = models[0]

View File

@ -112,7 +112,7 @@ func FindFuncModelByFunction(function string, userID int) ([]dao.Model, []dao.Fu
//先查找对应功能
funcModels := dao.FindFunctionModelByFunction(function)
if len(funcModels) == 0 {
return nil, nil
return []dao.Model{}, funcModels
}
modelID := funcModels[0].ModelID
//再查找对应模型