feat(AI): add done channel for faster response

openai/chat.go

@@ -51,6 +51,7 @@ type ChatResponseStream struct {
 	Created int
 	Model   string
 	Stream  chan string
+	Done    chan struct{}
 	Err     error
 }
 

openai/client.go

@@ -77,11 +77,15 @@ func (c *Client) ChatCompletionStream(request ChatRequest) (*ChatResponseStream,
 		return nil, errors.Errorf("status code: %d, body: %q", resp.StatusCode(), respBodyStr)
 	}
 
-	ret := &ChatResponseStream{Stream: make(chan string, 1024)}
+	ret := &ChatResponseStream{
+		Stream: make(chan string, 1024),
+		Done:   make(chan struct{}),
+	}
 	go func() {
 		defer func() {
 			rbody.Close()
 			close(ret.Stream)
+			close(ret.Done)
 		}()
 
 		var contentBegan bool