From 5570814f89da3399ce4542f3ce639df8e6318959 Mon Sep 17 00:00:00 2001 From: Zhang Minghan Date: Mon, 29 Jan 2024 17:04:05 +0800 Subject: [PATCH] fix: fix unnecessary progress bar residue in midjourney format --- adapter/midjourney/chat.go | 18 ++++++++++-------- 1 file changed, 10 insertions(+), 8 deletions(-) diff --git a/adapter/midjourney/chat.go b/adapter/midjourney/chat.go index ee0c120..118e438 100644 --- a/adapter/midjourney/chat.go +++ b/adapter/midjourney/chat.go @@ -59,18 +59,20 @@ func (c *ChatInstance) CreateStreamChatRequest(props *ChatProps, callback global return fmt.Errorf("format error: please provide available prompt") } - if err := callback("```progress\n"); err != nil { - return err - } - url, err := c.CreateStreamImagineTask(prompt, func(progress int) error { + if progress == 0 { + if err := callback("```progress\n"); err != nil { + return err + } + } else if progress == 100 { + if err := callback("```\n"); err != nil { + return err + } + } + return callback(fmt.Sprintf("%d\n", progress)) }) - if err := callback("```\n"); err != nil { - return err - } - if err != nil { return fmt.Errorf("error from midjourney: %s", err.Error()) }