From db0ba0d9a09be672e5201fb2e62e8de0983fab86 Mon Sep 17 00:00:00 2001 From: RockYang Date: Thu, 11 Apr 2024 17:32:34 +0800 Subject: [PATCH] feat: use custom mode for mj upscale and variation operarions --- api/service/mj/plus_client.go | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/api/service/mj/plus_client.go b/api/service/mj/plus_client.go index 89bca627..52846208 100644 --- a/api/service/mj/plus_client.go +++ b/api/service/mj/plus_client.go @@ -163,7 +163,7 @@ func (c *PlusClient) Upscale(task types.MjTask) (ImageRes, error) { "customId": fmt.Sprintf("MJ::JOB::upsample::%d::%s", task.Index, task.MessageHash), "taskId": task.MessageId, } - apiURL := fmt.Sprintf("%s/mj/submit/action", c.apiURL) + apiURL := fmt.Sprintf("%s/mj-%s/mj/submit/action", c.Config.Mode, c.apiURL) var res ImageRes var errRes ErrRes r, err := c.client.R(). @@ -189,7 +189,7 @@ func (c *PlusClient) Variation(task types.MjTask) (ImageRes, error) { "customId": fmt.Sprintf("MJ::JOB::variation::%d::%s", task.Index, task.MessageHash), "taskId": task.MessageId, } - apiURL := fmt.Sprintf("%s/mj/submit/action", c.apiURL) + apiURL := fmt.Sprintf("%s/mj-%s/mj/submit/action", c.Config.Mode, c.apiURL) var res ImageRes var errRes ErrRes r, err := req.C().R().