From b97d4b7895b79da3b1d5aab29a82049733da413d Mon Sep 17 00:00:00 2001 From: GeekMaster Date: Tue, 22 Apr 2025 15:47:21 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8DDALLE=20function=20call=20bug?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- api/service/dalle/service.go | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/api/service/dalle/service.go b/api/service/dalle/service.go index 2306f446..ba716bdc 100644 --- a/api/service/dalle/service.go +++ b/api/service/dalle/service.go @@ -132,7 +132,11 @@ func (s *Service) Image(task types.DallTask, sync bool) (string, error) { } var chatModel model.ChatModel - s.db.Where("id = ?", task.ModelId).First(&chatModel) + if task.ModelId > 0 { + s.db.Where("id", task.ModelId).First(&chatModel) + } else { + s.db.Where("value", task.ModelName).First(&chatModel) + } // get image generation API KEY var apiKey model.ApiKey